@ganziliang/kb-model-setup 0.1.0

package/dist/index.d.ts

@@ -0,0 +1,24 @@
+export type ApiProtocol = "openai-responses" | "anthropic-messages";
+export type ModelConfig = {
+    provider: string;
+    api: ApiProtocol;
+    model: string;
+    baseURL: string;
+    apiKey: string;
+};
+export type ModelConfigStore = {
+    load(): Promise<ModelConfig[]>;
+    save(models: ModelConfig[]): Promise<void>;
+};
+export type SetupPrompter = {
+    ask(question: string, initial?: string): Promise<string>;
+    select(question: string, choices: string[], initial?: number): Promise<number>;
+    confirm(question: string): Promise<boolean>;
+    notice(message: string): Promise<void>;
+};
+export type ModelSetupExtension = {
+    configure(prompter: SetupPrompter, existing?: ModelConfig): Promise<ModelConfig>;
+    validate(config: ModelConfig): string[];
+};
+export declare function createModelSetupExtension(): ModelSetupExtension;
+export declare function configFromEnvironment(env?: NodeJS.ProcessEnv): ModelConfig | undefined;

package/dist/index.js

@@ -0,0 +1,44 @@
+const protocols = ["openai-responses", "anthropic-messages"];
+export function createModelSetupExtension() {
+    return {
+        async configure(prompter, existing) {
+            const provider = await prompter.ask("Provider", existing?.provider ?? "");
+            const apiIndex = await prompter.select("API protocol", protocols, Math.max(0, protocols.indexOf(existing?.api ?? protocols[0])));
+            const model = await prompter.ask("Model", existing?.model ?? "");
+            const baseURL = await prompter.ask("Base URL", existing?.baseURL ?? "");
+            const apiKey = await prompter.ask("API key", existing?.apiKey ?? "");
+            const config = { provider, api: protocols[apiIndex], model, baseURL, apiKey };
+            const errors = this.validate(config);
+            if (errors.length) {
+                await prompter.notice(errors.join("\n"));
+                return this.configure(prompter, config);
+            }
+            return config;
+        },
+        validate(config) {
+            const errors = [];
+            if (!config.provider.trim())
+                errors.push("Provider is required.");
+            if (!protocols.includes(config.api))
+                errors.push("API protocol is invalid.");
+            if (!config.model.trim())
+                errors.push("Model is required.");
+            if (!config.baseURL.trim())
+                errors.push("Base URL is required.");
+            if (!config.apiKey.trim())
+                errors.push("API key is required.");
+            return errors;
+        },
+    };
+}
+export function configFromEnvironment(env = process.env) {
+    const provider = env.KB_PROVIDER;
+    const api = env.KB_API;
+    const model = env.KB_MODEL;
+    const baseURL = env.KB_BASE_URL;
+    const apiKey = env.KB_API_KEY;
+    if (!provider || !api || !model || !baseURL || !apiKey)
+        return undefined;
+    const config = { provider, api, model, baseURL, apiKey };
+    return createModelSetupExtension().validate(config).length ? undefined : config;
+}

package/package.json

@@ -0,0 +1,25 @@
+{
+  "name": "@ganziliang/kb-model-setup",
+  "version": "0.1.0",
+  "description": "Interactive model configuration for the kb local knowledge base agent",
+  "type": "module",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "files": ["dist"],
+  "scripts": {
+    "build": "tsc -p tsconfig.json",
+    "typecheck": "tsc -p tsconfig.json --noEmit",
+    "test": "npm run build && node --test"
+  },
+  "engines": { "node": ">=22.5.0" },
+  "devDependencies": {
+    "@types/node": "^22.10.0",
+    "typescript": "^5.7.2"
+  }
+}