feat: add interactive config setup and baseURL support
This commit is contained in:
parent
d9bffdec2b
commit
3c9a5e625a
4 changed files with 60 additions and 25 deletions
26
index.ts
26
index.ts
|
|
@ -2,7 +2,7 @@ import { parseArgs } from "node:util";
|
|||
import { createOpenAICompatible } from "@ai-sdk/openai-compatible";
|
||||
import { streamText } from "ai";
|
||||
|
||||
import { loadConfig, saveConfig } from "./src/config";
|
||||
import { initConfig, loadConfig } from "./src/config";
|
||||
import { Conversation } from "./src/conversation";
|
||||
import Renderer from "./src/render";
|
||||
|
||||
|
|
@ -13,8 +13,8 @@ const { values: args, positionals } = parseArgs({
|
|||
m: { type: "string" },
|
||||
new: { type: "boolean" },
|
||||
n: { type: "boolean" },
|
||||
save: { type: "boolean" },
|
||||
s: { type: "boolean" },
|
||||
config: { type: "boolean" },
|
||||
c: { type: "boolean" },
|
||||
plain: { type: "boolean" },
|
||||
plainOut: { type: "boolean" },
|
||||
plainErr: { type: "boolean" },
|
||||
|
|
@ -24,26 +24,18 @@ const { values: args, positionals } = parseArgs({
|
|||
});
|
||||
|
||||
async function main() {
|
||||
const config = await loadConfig();
|
||||
|
||||
const newConvo = args.new ?? args.n;
|
||||
const plainOut = args.plainOut ?? args.plain ?? !process.stdout.isTTY;
|
||||
const plainErr = args.plainErr ?? args.plain ?? !process.stderr.isTTY;
|
||||
const modelFlag = args.model ?? args.m;
|
||||
const saveConfigFlag = args.save ?? args.s;
|
||||
const configFlag = args.config ?? args.c;
|
||||
|
||||
const config = await loadConfig();
|
||||
|
||||
const model = modelFlag ?? config.model;
|
||||
|
||||
if (!model) {
|
||||
console.error(
|
||||
"Please run 'lmc --model <model> --save' to set a default model.",
|
||||
);
|
||||
process.exit(1);
|
||||
}
|
||||
|
||||
if (saveConfigFlag) {
|
||||
config.model = modelFlag ?? config.model;
|
||||
await saveConfig(config);
|
||||
if (configFlag) {
|
||||
await initConfig(config);
|
||||
return;
|
||||
}
|
||||
|
||||
|
|
@ -52,7 +44,7 @@ async function main() {
|
|||
const provider = createOpenAICompatible({
|
||||
name: "llama.cpp",
|
||||
apiKey: "local",
|
||||
baseURL: "http://127.0.0.1:8080/v1",
|
||||
baseURL: config.baseURL,
|
||||
includeUsage: true, // Include usage information in streaming responses
|
||||
});
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue