llm-kb 0.4.1 → 0.5.0

package/README.md

@@ -39,7 +39,7 @@ llm-kb run ./my-documents
 ```
 
 ```
-llm-kb v0.4.0
+llm-kb v0.5.0
 
 Scanning ./my-documents...
   Found 9 files (9 PDF)

package/bin/{chunk-DHOXVEIR.js → chunk-G4UEKXXU.js}

@@ -7112,6 +7112,40 @@ if (canUseRuntimeCodegen()) {
   }
 }
 
+// src/model-resolver.ts
+import { AuthStorage } from "@mariozechner/pi-coding-agent";
+var ANTHROPIC_TO_OPENROUTER = {
+  "claude-haiku-4-5": "anthropic/claude-haiku-4.5",
+  "claude-sonnet-4-6": "anthropic/claude-sonnet-4.6",
+  "claude-sonnet-4-5": "anthropic/claude-sonnet-4.5",
+  "claude-sonnet-4-0": "anthropic/claude-sonnet-4",
+  "claude-opus-4-5": "anthropic/claude-opus-4.5"
+};
+var PROVIDERS = ["anthropic", "openrouter"];
+async function resolveModel(modelId, authStorage) {
+  const storage = authStorage ?? AuthStorage.create();
+  const anthropicKey = await storage.getApiKey("anthropic");
+  if (anthropicKey) {
+    const model = getModels("anthropic").find((m) => m.id === modelId);
+    if (model) return model;
+  }
+  const openrouterKey = await storage.getApiKey("openrouter");
+  if (openrouterKey) {
+    const orId = ANTHROPIC_TO_OPENROUTER[modelId] ?? modelId;
+    const model = getModels("openrouter").find((m) => m.id === orId);
+    if (model) return model;
+  }
+  return void 0;
+}
+async function resolveApiKey(authStorage) {
+  const storage = authStorage ?? AuthStorage.create();
+  for (const provider of PROVIDERS) {
+    const key = await storage.getApiKey(provider);
+    if (key) return { key, provider };
+  }
+  return void 0;
+}
+
 // src/indexer.ts
 import { readdir, readFile } from "fs/promises";
 
@@ -7211,7 +7245,7 @@ async function buildIndex(folder, sourcesDir, onOutput, authStorage, modelId) {
     })
   });
   await loader.reload();
-  const model = modelId ? getModels("anthropic").find((m) => m.id === modelId) : void 0;
+  const model = modelId ? await resolveModel(modelId, authStorage) : void 0;
   const { session } = await createAgentSession({
     cwd: folder,
     resourceLoader: loader,
@@ -7254,6 +7288,8 @@ Add a total word count estimate at the bottom.`;
 
 export {
   completeSimple,
+  resolveModel,
+  resolveApiKey,
   continueKBSession,
   createKBSession,
   getNodeModulesPath,

package/bin/cli.js

@@ -4,13 +4,13 @@ import {
   completeSimple,
   continueKBSession,
   createKBSession,
-  getNodeModulesPath
-} from "./chunk-DHOXVEIR.js";
+  getNodeModulesPath,
+  resolveApiKey,
+  resolveModel
+} from "./chunk-G4UEKXXU.js";
 import "./chunk-3YMNGUZZ.js";
 import "./chunk-LDHOKBJA.js";
-import {
-  getModels
-} from "./chunk-5PYKQQLA.js";
+import "./chunk-5PYKQQLA.js";
 import "./chunk-EAQYK3U2.js";
 
 // src/cli.ts
@@ -335,20 +335,12 @@ function extractText(content) {
 }
 
 // src/wiki-updater.ts
-import { AuthStorage } from "@mariozechner/pi-coding-agent";
 import { existsSync as existsSync2 } from "fs";
 import { readFile as readFile2, writeFile as writeFile3, mkdir as mkdir3 } from "fs/promises";
 import { join as join4 } from "path";
-import { homedir } from "os";
-async function resolveApiKey(authStorage) {
-  if (authStorage) {
-    return authStorage.getApiKey("anthropic");
-  }
-  const piAuthPath = join4(homedir(), ".pi", "agent", "auth.json");
-  if (existsSync2(piAuthPath)) {
-    const storage = AuthStorage.create(piAuthPath);
-    return storage.getApiKey("anthropic");
-  }
+async function resolveApiKey2(authStorage) {
+  const result = await resolveApiKey(authStorage);
+  if (result) return result.key;
   return process.env.ANTHROPIC_API_KEY;
 }
 function buildPrompt(question, answer, sources, date, currentWiki) {
@@ -413,9 +405,9 @@ async function updateWiki(kbRoot, trace, authStorage, indexModelId = "claude-hai
   const sources = trace.filesRead.map((f) => f.split(/[\\/]/).pop() ?? f).filter((f) => f.endsWith(".md") && f !== "index.md" && f !== "wiki.md").join(", ") || "unknown";
   const date = new Date(trace.timestamp).toISOString().slice(0, 10);
   const prompt = buildPrompt(trace.question, trace.answer, sources, date, currentWiki);
-  const apiKey = await resolveApiKey(authStorage);
+  const apiKey = await resolveApiKey2(authStorage);
   if (!apiKey) return;
-  const model = getModels("anthropic").find((m) => m.id === indexModelId);
+  const model = await resolveModel(indexModelId, authStorage);
   if (!model) return;
   const result = await completeSimple(
     model,
@@ -923,7 +915,7 @@ async function createChat(folder, options) {
     createBashTool(folder),
     createWriteTool(folder)
   ];
-  const model = options.modelId ? getModels("anthropic").find((m) => m.id === options.modelId) : void 0;
+  const model = options.modelId ? await resolveModel(options.modelId, options.authStorage) : void 0;
   const { session } = await createAgentSession({
     cwd: folder,
     resourceLoader: loader,
@@ -952,17 +944,15 @@ async function query(folder, question, options) {
   session.dispose();
   if (options.save) {
     const sourcesDir = join6(folder, ".llm-kb", "wiki", "sources");
-    const { buildIndex: buildIndex2 } = await import("./indexer-KSYRIVVN.js");
+    const { buildIndex: buildIndex2 } = await import("./indexer-Q44ALZ7N.js");
     await buildIndex2(folder, sourcesDir, void 0, options.authStorage);
   }
 }
 
 // src/eval.ts
-import { AuthStorage as AuthStorage3 } from "@mariozechner/pi-coding-agent";
 import { readFile as readFile5, readdir as readdir5, writeFile as writeFile5, mkdir as mkdir6 } from "fs/promises";
 import { existsSync as existsSync5 } from "fs";
 import { join as join7, basename as basename5 } from "path";
-import { homedir as homedir2 } from "os";
 async function parseSessionsForEval(sessionsDir, sourcesDir, limit) {
   if (!existsSync5(sessionsDir)) return [];
   const sessionFiles = (await readdir5(sessionsDir)).filter((f) => f.endsWith(".jsonl")).sort().reverse();
@@ -1102,18 +1092,14 @@ function calculateMetrics(qas) {
     wastedReads
   };
 }
-async function resolveApiKey2(authStorage) {
-  if (authStorage) return authStorage.getApiKey("anthropic");
-  const piAuthPath = join7(homedir2(), ".pi", "agent", "auth.json");
-  if (existsSync5(piAuthPath)) {
-    const storage = AuthStorage3.create(piAuthPath);
-    return storage.getApiKey("anthropic");
-  }
+async function resolveApiKey3(authStorage) {
+  const result = await resolveApiKey(authStorage);
+  if (result) return result.key;
   return process.env.ANTHROPIC_API_KEY;
 }
 async function judgeQA(qa, apiKey, modelId) {
   const issues = [];
-  const model = getModels("anthropic").find((m) => m.id === modelId);
+  const model = await resolveModel(modelId);
   if (!model) return issues;
   const filesSummary = qa.filesRead.map((f) => `File: ${basename5(f.path)}
 Content (first 2000 chars):
@@ -1303,7 +1289,7 @@ async function runEval(kbRoot, options) {
   }
   log("Calculating metrics...");
   const metrics = calculateMetrics(qas);
-  const apiKey = await resolveApiKey2(options.authStorage);
+  const apiKey = await resolveApiKey3(options.authStorage);
   const allIssues = [];
   if (apiKey) {
     const modelId = "claude-haiku-4-5";
@@ -5476,16 +5462,17 @@ function resolveKnowledgeBase(startDir) {
 // src/auth.ts
 import { existsSync as existsSync7 } from "fs";
 import { join as join11 } from "path";
-import { homedir as homedir4 } from "os";
-import { AuthStorage as AuthStorage4 } from "@mariozechner/pi-coding-agent";
+import { homedir as homedir2 } from "os";
+import { AuthStorage as AuthStorage2 } from "@mariozechner/pi-coding-agent";
 import chalk5 from "chalk";
 function checkAuth() {
-  const piAuthPath = join11(homedir4(), ".pi", "agent", "auth.json");
+  const piAuthPath = join11(homedir2(), ".pi", "agent", "auth.json");
   if (existsSync7(piAuthPath)) {
-    return { ok: true, method: "pi-sdk" };
+    const authStorage = AuthStorage2.create();
+    return { ok: true, method: "pi-sdk", authStorage };
   }
   if (process.env.ANTHROPIC_API_KEY) {
-    const authStorage = AuthStorage4.inMemory({
+    const authStorage = AuthStorage2.inMemory({
       anthropic: { type: "api_key", key: process.env.ANTHROPIC_API_KEY }
     });
     return { ok: true, method: "api-key", authStorage };
@@ -5543,14 +5530,20 @@ async function ensureConfig(kbRoot) {
 
 // src/cli.ts
 import { existsSync as existsSync9 } from "fs";
+import { readFileSync } from "fs";
 import { mkdir as mkdir8, readdir as readdir6, stat as stat2 } from "fs/promises";
-import { resolve as resolve3, join as join13 } from "path";
+import { resolve as resolve3, join as join13, dirname as dirname3 } from "path";
+import { fileURLToPath } from "url";
 import chalk6 from "chalk";
+var __filename = fileURLToPath(import.meta.url);
+var __dirname = dirname3(__filename);
+var pkg = JSON.parse(readFileSync(join13(__dirname, "..", "package.json"), "utf-8"));
+var VERSION = pkg.version;
 var program = new Command();
-program.name("llm-kb").description("Drop files into a folder. Get a knowledge base you can query.").version("0.4.0");
+program.name("llm-kb").description("Drop files into a folder. Get a knowledge base you can query.").version(VERSION);
 program.command("run").description("Scan, parse, index, and watch a folder").argument("<folder>", "Path to your documents folder").action(async (folder) => {
   console.log(`
-${chalk6.bold("llm-kb")} v0.4.0
+${chalk6.bold("llm-kb")} v${VERSION}
 `);
   const auth = checkAuth();
   if (!auth.ok) exitWithAuthError();

package/bin/{indexer-KSYRIVVN.js → indexer-Q44ALZ7N.js}

@@ -1,6 +1,6 @@
 import {
   buildIndex
-} from "./chunk-DHOXVEIR.js";
+} from "./chunk-G4UEKXXU.js";
 import "./chunk-3YMNGUZZ.js";
 import "./chunk-LDHOKBJA.js";
 import "./chunk-5PYKQQLA.js";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "llm-kb",
-  "version": "0.4.1",
+  "version": "0.5.0",
   "description": "LLM-powered knowledge base. Drop documents, build a wiki, ask questions. Inspired by Karpathy.",
   "bin": {
     "llm-kb": "./bin/cli.js"