@vpxa/kb 0.1.1 → 0.1.2

package/packages/server/dist/curated-manager.js

@@ -1,357 +1,10 @@
-import { createHash } from "node:crypto";
-import { mkdir, readdir, readFile, stat, unlink, writeFile } from "node:fs/promises";
-import { dirname, isAbsolute, join } from "node:path";
-const MAX_CONTENT_SIZE = 50 * 1024;
-class CuratedKnowledgeManager {
-  constructor(curatedDir, store, embedder) {
-    this.curatedDir = curatedDir;
-    this.store = store;
-    this.embedder = embedder;
-  }
-  async remember(title, content, category, tags = []) {
-    this.validateCategoryName(category);
-    this.validateContentSize(content);
-    const slug = this.slugify(title);
-    const relativePath = await this.uniqueRelativePath(category, slug);
-    const filePath = join(this.curatedDir, relativePath);
-    const now = (/* @__PURE__ */ new Date()).toISOString();
-    const frontmatter = {
-      title,
-      category,
-      tags,
-      created: now,
-      updated: now,
-      version: 1,
-      origin: "curated",
-      changelog: [{ version: 1, date: now, reason: "Initial creation" }]
-    };
-    const fileContent = this.serializeFile(content, frontmatter);
-    await mkdir(dirname(filePath), { recursive: true });
-    await writeFile(filePath, fileContent, "utf-8");
-    await this.indexCuratedFile(relativePath, content, frontmatter);
-    return { path: relativePath };
-  }
-  async update(relativePath, newContent, reason) {
-    this.guardPath(relativePath);
-    this.validateContentSize(newContent);
-    const filePath = join(this.curatedDir, relativePath);
-    const raw = await readFile(filePath, "utf-8");
-    const { frontmatter } = this.parseFile(raw);
-    const newVersion = (frontmatter.version ?? 1) + 1;
-    const now = (/* @__PURE__ */ new Date()).toISOString();
-    frontmatter.version = newVersion;
-    frontmatter.updated = now;
-    frontmatter.changelog = [
-      ...frontmatter.changelog ?? [],
-      { version: newVersion, date: now, reason }
-    ];
-    const fileContent = this.serializeFile(newContent, frontmatter);
-    await writeFile(filePath, fileContent, "utf-8");
-    await this.indexCuratedFile(relativePath, newContent, frontmatter);
-    return { path: relativePath, version: newVersion };
-  }
-  async forget(relativePath, _reason) {
-    this.guardPath(relativePath);
-    const filePath = join(this.curatedDir, relativePath);
-    const storePath = `curated/${relativePath}`;
-    await this.store.deleteBySourcePath(storePath);
-    await unlink(filePath);
-    return { path: relativePath };
-  }
-  async read(relativePath) {
-    this.guardPath(relativePath);
-    const filePath = join(this.curatedDir, relativePath);
-    const raw = await readFile(filePath, "utf-8");
-    const { frontmatter, content } = this.parseFile(raw);
-    const category = relativePath.split("/")[0];
-    return {
-      path: relativePath,
-      title: frontmatter.title ?? relativePath,
-      category,
-      tags: frontmatter.tags ?? [],
-      version: frontmatter.version ?? 1,
-      created: frontmatter.created ?? "",
-      updated: frontmatter.updated ?? "",
-      contentPreview: content.slice(0, 200),
-      content
-    };
-  }
-  async list(filters) {
-    const entries = [];
-    const categories = filters?.category ? [filters.category] : await this.discoverCategories();
-    for (const cat of categories) {
-      const catDir = join(this.curatedDir, cat);
-      try {
-        const files = await readdir(catDir);
-        for (const file of files) {
-          if (!file.endsWith(".md")) continue;
-          const filePath = join(catDir, file);
-          const raw = await readFile(filePath, "utf-8");
-          const { frontmatter, content } = this.parseFile(raw);
-          if (filters?.tag && !(frontmatter.tags ?? []).includes(filters.tag)) continue;
-          entries.push({
-            path: `${cat}/${file}`,
-            title: frontmatter.title ?? file,
-            category: cat,
-            tags: frontmatter.tags ?? [],
-            version: frontmatter.version ?? 1,
-            created: frontmatter.created ?? "",
-            updated: frontmatter.updated ?? "",
-            contentPreview: content.slice(0, 200)
-          });
-        }
-      } catch {
-      }
-    }
-    return entries;
-  }
-  /**
-   * Re-index all curated files into the vector store.
-   * Call this after the indexer has run to restore curated vectors
-   * that may have been lost, or to bulk-sync disk state to vectors.
-   */
-  async reindexAll() {
-    const categories = await this.discoverCategories();
-    const errors = [];
-    const pending = [];
-    for (const cat of categories) {
-      const catDir = join(this.curatedDir, cat);
-      let files;
-      try {
-        files = (await readdir(catDir)).filter((f) => f.endsWith(".md"));
-      } catch {
-        continue;
-      }
-      for (const file of files) {
-        const relativePath = `${cat}/${file}`;
-        const filePath = join(catDir, file);
-        try {
-          const raw = await readFile(filePath, "utf-8");
-          const { frontmatter, content } = this.parseFile(raw);
-          pending.push({ relativePath, content, frontmatter });
-        } catch (err) {
-          errors.push(`${relativePath}: ${err.message}`);
-        }
-      }
-    }
-    if (pending.length === 0) return { indexed: 0, errors };
-    const vectors = await this.embedder.embedBatch(pending.map((p) => p.content));
-    const now = (/* @__PURE__ */ new Date()).toISOString();
-    const records = pending.map((p) => {
-      const storePath = `curated/${p.relativePath}`;
-      return {
-        id: this.hashId(storePath, 0),
-        content: p.content,
-        sourcePath: storePath,
-        contentType: "curated-knowledge",
-        headingPath: p.frontmatter.title,
-        chunkIndex: 0,
-        totalChunks: 1,
-        startLine: 1,
-        endLine: p.content.split("\n").length,
-        fileHash: this.hash(p.content),
-        indexedAt: now,
-        origin: "curated",
-        tags: p.frontmatter.tags,
-        category: p.frontmatter.category,
-        version: p.frontmatter.version
-      };
-    });
-    await this.store.upsert(records, vectors);
-    return { indexed: pending.length, errors };
-  }
-  // --- Private helpers ---
-  async indexCuratedFile(relativePath, content, frontmatter) {
-    const vector = await this.embedder.embed(content);
-    const storePath = `curated/${relativePath}`;
-    const now = (/* @__PURE__ */ new Date()).toISOString();
-    const record = {
-      id: this.hashId(storePath, 0),
-      content,
-      sourcePath: storePath,
-      contentType: "curated-knowledge",
-      headingPath: frontmatter.title,
-      chunkIndex: 0,
-      totalChunks: 1,
-      startLine: 1,
-      endLine: content.split("\n").length,
-      fileHash: this.hash(content),
-      indexedAt: now,
-      origin: "curated",
-      tags: frontmatter.tags,
-      category: frontmatter.category,
-      version: frontmatter.version
-    };
-    await this.store.upsert([record], [vector]);
-  }
-  async discoverCategories() {
-    try {
-      const entries = await readdir(this.curatedDir, { withFileTypes: true });
-      return entries.filter((e) => e.isDirectory() && /^[a-z][a-z0-9-]*$/.test(e.name)).map((e) => e.name);
-    } catch {
-      return [];
-    }
-  }
-  guardPath(relativePath) {
-    if (relativePath.includes("..") || isAbsolute(relativePath)) {
-      throw new Error(`Invalid path: ${relativePath}. Must be relative within curated/ directory.`);
-    }
-    const category = relativePath.split("/")[0];
-    this.validateCategoryName(category);
-  }
-  validateCategoryName(category) {
-    if (!/^[a-z][a-z0-9-]*$/.test(category)) {
-      throw new Error(
-        `Invalid category name: "${category}". Must be lowercase kebab-case (e.g., "decisions", "api-contracts").`
-      );
-    }
-  }
-  validateContentSize(content) {
-    if (Buffer.byteLength(content, "utf-8") > MAX_CONTENT_SIZE) {
-      throw new Error(`Content exceeds maximum size of ${MAX_CONTENT_SIZE / 1024}KB`);
-    }
-  }
-  slugify(title) {
-    return title.toLowerCase().replace(/[^a-z0-9]+/g, "-").replace(/^-|-$/g, "").slice(0, 80);
-  }
-  /** Return a unique `category/slug.md` path, appending `-2`, `-3`, … on collision. */
-  async uniqueRelativePath(category, slug) {
-    const base = `${category}/${slug}.md`;
-    const basePath = join(this.curatedDir, base);
-    try {
-      await stat(basePath);
-    } catch {
-      return base;
-    }
-    for (let i = 2; i <= 100; i++) {
-      const candidate = `${category}/${slug}-${i}.md`;
-      try {
-        await stat(join(this.curatedDir, candidate));
-      } catch {
-        return candidate;
-      }
-    }
-    throw new Error(`Too many entries with slug "${slug}" in category "${category}"`);
-  }
-  hash(content) {
-    return createHash("sha256").update(content).digest("hex").slice(0, 16);
-  }
-  hashId(sourcePath, chunkIndex) {
-    return this.hash(`${sourcePath}::${chunkIndex}`);
-  }
-  /**
-   * Simple YAML frontmatter serializer (no gray-matter dependency).
-   */
-  serializeFile(content, frontmatter) {
-    const yaml = [
-      "---",
-      `title: "${frontmatter.title.replace(/"/g, '\\"')}"`,
-      `category: ${frontmatter.category}`,
-      `tags: [${frontmatter.tags.map((t) => `"${t}"`).join(", ")}]`,
-      `created: ${frontmatter.created}`,
-      `updated: ${frontmatter.updated}`,
-      `version: ${frontmatter.version}`,
-      `origin: ${frontmatter.origin}`,
-      "changelog:",
-      ...frontmatter.changelog.map(
-        (c) => `  - version: ${c.version}
-    date: ${c.date}
-    reason: "${c.reason.replace(/"/g, '\\"')}"`
-      ),
-      "---"
-    ].join("\n");
-    return `${yaml}
+import{createHash as C}from"node:crypto";import{mkdir as b,readdir as f,readFile as m,stat as y,unlink as P,writeFile as v}from"node:fs/promises";import{dirname as F,isAbsolute as I,join as u}from"node:path";const w=50*1024;class E{constructor(t,e,n){this.curatedDir=t;this.store=e;this.embedder=n}async remember(t,e,n,r=[]){this.validateCategoryName(n),this.validateContentSize(e);const a=this.slugify(t),s=await this.uniqueRelativePath(n,a),i=u(this.curatedDir,s),g=new Date().toISOString(),c={title:t,category:n,tags:r,created:g,updated:g,version:1,origin:"curated",changelog:[{version:1,date:g,reason:"Initial creation"}]},d=this.serializeFile(e,c);return await b(F(i),{recursive:!0}),await v(i,d,"utf-8"),await this.indexCuratedFile(s,e,c),{path:s}}async update(t,e,n){this.guardPath(t),this.validateContentSize(e);const r=u(this.curatedDir,t),a=await m(r,"utf-8"),{frontmatter:s}=this.parseFile(a),i=(s.version??1)+1,g=new Date().toISOString();s.version=i,s.updated=g,s.changelog=[...s.changelog??[],{version:i,date:g,reason:n}];const c=this.serializeFile(e,s);return await v(r,c,"utf-8"),await this.indexCuratedFile(t,e,s),{path:t,version:i}}async forget(t,e){this.guardPath(t);const n=u(this.curatedDir,t),r=`curated/${t}`;return await this.store.deleteBySourcePath(r),await P(n),{path:t}}async read(t){this.guardPath(t);const e=u(this.curatedDir,t),n=await m(e,"utf-8"),{frontmatter:r,content:a}=this.parseFile(n),s=t.split("/")[0];return{path:t,title:r.title??t,category:s,tags:r.tags??[],version:r.version??1,created:r.created??"",updated:r.updated??"",contentPreview:a.slice(0,200),content:a}}async list(t){const e=[],n=t?.category?[t.category]:await this.discoverCategories();for(const r of n){const a=u(this.curatedDir,r);try{const s=await f(a);for(const i of s){if(!i.endsWith(".md"))continue;const g=u(a,i),c=await m(g,"utf-8"),{frontmatter:d,content:h}=this.parseFile(c);t?.tag&&!(d.tags??[]).includes(t.tag)||e.push({path:`${r}/${i}`,title:d.title??i,category:r,tags:d.tags??[],version:d.version??1,created:d.created??"",updated:d.updated??"",contentPreview:h.slice(0,200)})}}catch{}}return e}async reindexAll(){const t=await this.discoverCategories(),e=[],n=[];for(const i of t){const g=u(this.curatedDir,i);let c;try{c=(await f(g)).filter(d=>d.endsWith(".md"))}catch{continue}for(const d of c){const h=`${i}/${d}`,p=u(g,d);try{const o=await m(p,"utf-8"),{frontmatter:l,content:$}=this.parseFile(o);n.push({relativePath:h,content:$,frontmatter:l})}catch(o){e.push(`${h}: ${o.message}`)}}}if(n.length===0)return{indexed:0,errors:e};const r=await this.embedder.embedBatch(n.map(i=>i.content)),a=new Date().toISOString(),s=n.map(i=>{const g=`curated/${i.relativePath}`;return{id:this.hashId(g,0),content:i.content,sourcePath:g,contentType:"curated-knowledge",headingPath:i.frontmatter.title,chunkIndex:0,totalChunks:1,startLine:1,endLine:i.content.split(`
+`).length,fileHash:this.hash(i.content),indexedAt:a,origin:"curated",tags:i.frontmatter.tags,category:i.frontmatter.category,version:i.frontmatter.version}});return await this.store.upsert(s,r),{indexed:n.length,errors:e}}async indexCuratedFile(t,e,n){const r=await this.embedder.embed(e),a=`curated/${t}`,s=new Date().toISOString(),i={id:this.hashId(a,0),content:e,sourcePath:a,contentType:"curated-knowledge",headingPath:n.title,chunkIndex:0,totalChunks:1,startLine:1,endLine:e.split(`
+`).length,fileHash:this.hash(e),indexedAt:s,origin:"curated",tags:n.tags,category:n.category,version:n.version};await this.store.upsert([i],[r])}async discoverCategories(){try{return(await f(this.curatedDir,{withFileTypes:!0})).filter(e=>e.isDirectory()&&/^[a-z][a-z0-9-]*$/.test(e.name)).map(e=>e.name)}catch{return[]}}guardPath(t){if(t.includes("..")||I(t))throw new Error(`Invalid path: ${t}. Must be relative within curated/ directory.`);const e=t.split("/")[0];this.validateCategoryName(e)}validateCategoryName(t){if(!/^[a-z][a-z0-9-]*$/.test(t))throw new Error(`Invalid category name: "${t}". Must be lowercase kebab-case (e.g., "decisions", "api-contracts").`)}validateContentSize(t){if(Buffer.byteLength(t,"utf-8")>w)throw new Error(`Content exceeds maximum size of ${w/1024}KB`)}slugify(t){return t.toLowerCase().replace(/[^a-z0-9]+/g,"-").replace(/^-|-$/g,"").slice(0,80)}async uniqueRelativePath(t,e){const n=`${t}/${e}.md`,r=u(this.curatedDir,n);try{await y(r)}catch{return n}for(let a=2;a<=100;a++){const s=`${t}/${e}-${a}.md`;try{await y(u(this.curatedDir,s))}catch{return s}}throw new Error(`Too many entries with slug "${e}" in category "${t}"`)}hash(t){return C("sha256").update(t).digest("hex").slice(0,16)}hashId(t,e){return this.hash(`${t}::${e}`)}serializeFile(t,e){return`${["---",`title: "${e.title.replace(/"/g,'\\"')}"`,`category: ${e.category}`,`tags: [${e.tags.map(r=>`"${r}"`).join(", ")}]`,`created: ${e.created}`,`updated: ${e.updated}`,`version: ${e.version}`,`origin: ${e.origin}`,"changelog:",...e.changelog.map(r=>`  - version: ${r.version}
+    date: ${r.date}
+    reason: "${r.reason.replace(/"/g,'\\"')}"`),"---"].join(`
+`)}
 
-${content}
-`;
-  }
-  /**
-   * Simple YAML frontmatter parser (no gray-matter dependency).
-   */
-  parseFile(raw) {
-    const fmMatch = raw.match(/^---\n([\s\S]*?)\n---\n\n?([\s\S]*)$/);
-    if (!fmMatch) {
-      return {
-        frontmatter: {
-          title: "Untitled",
-          category: "notes",
-          tags: [],
-          created: "",
-          updated: "",
-          version: 1,
-          origin: "curated",
-          changelog: []
-        },
-        content: raw
-      };
-    }
-    const yamlStr = fmMatch[1];
-    const content = fmMatch[2].trim();
-    const fm = {};
-    const changelog = [];
-    const lines = yamlStr.split("\n");
-    let inChangelog = false;
-    let currentEntry = {};
-    for (const line of lines) {
-      if (/^changelog:\s*$/.test(line)) {
-        inChangelog = true;
-        continue;
-      }
-      if (inChangelog) {
-        const itemMatch = line.match(/^\s+-\s+version:\s*(\d+)$/);
-        if (itemMatch) {
-          if (currentEntry.version != null) changelog.push(currentEntry);
-          currentEntry = { version: parseInt(itemMatch[1], 10) };
-          continue;
-        }
-        const dateMatch = line.match(/^\s+date:\s*(.+)$/);
-        if (dateMatch) {
-          currentEntry.date = dateMatch[1].trim();
-          continue;
-        }
-        const reasonMatch = line.match(/^\s+reason:\s*"?(.*?)"?\s*$/);
-        if (reasonMatch) {
-          currentEntry.reason = reasonMatch[1];
-          continue;
-        }
-        if (/^\w/.test(line)) {
-          inChangelog = false;
-          if (currentEntry.version != null) changelog.push(currentEntry);
-          currentEntry = {};
-        }
-        continue;
-      }
-      const kvMatch = line.match(/^(\w+):\s*(.*)$/);
-      if (kvMatch) {
-        const key = kvMatch[1];
-        let value = kvMatch[2];
-        if (typeof value === "string" && value.startsWith("[") && value.endsWith("]")) {
-          value = value.slice(1, -1).split(",").map((s) => s.trim().replace(/^"|"$/g, "")).filter((s) => s.length > 0);
-        } else if (typeof value === "string" && /^\d+$/.test(value)) {
-          value = parseInt(value, 10);
-        } else if (typeof value === "string" && value.startsWith('"') && value.endsWith('"')) {
-          value = value.slice(1, -1);
-        }
-        fm[key] = value;
-      }
-    }
-    if (currentEntry.version != null) changelog.push(currentEntry);
-    return {
-      frontmatter: {
-        title: fm.title ?? "Untitled",
-        category: fm.category ?? "notes",
-        tags: fm.tags ?? [],
-        created: fm.created ?? "",
-        updated: fm.updated ?? "",
-        version: fm.version ?? 1,
-        origin: "curated",
-        changelog
-      },
-      content
-    };
-  }
-}
-export {
-  CuratedKnowledgeManager
-};
-//# sourceMappingURL=curated-manager.js.map
+${t}
+`}parseFile(t){const e=t.match(/^---\n([\s\S]*?)\n---\n\n?([\s\S]*)$/);if(!e)return{frontmatter:{title:"Untitled",category:"notes",tags:[],created:"",updated:"",version:1,origin:"curated",changelog:[]},content:t};const n=e[1],r=e[2].trim(),a={},s=[],i=n.split(`
+`);let g=!1,c={};for(const d of i){if(/^changelog:\s*$/.test(d)){g=!0;continue}if(g){const p=d.match(/^\s+-\s+version:\s*(\d+)$/);if(p){c.version!=null&&s.push(c),c={version:parseInt(p[1],10)};continue}const o=d.match(/^\s+date:\s*(.+)$/);if(o){c.date=o[1].trim();continue}const l=d.match(/^\s+reason:\s*"?(.*?)"?\s*$/);if(l){c.reason=l[1];continue}/^\w/.test(d)&&(g=!1,c.version!=null&&s.push(c),c={});continue}const h=d.match(/^(\w+):\s*(.*)$/);if(h){const p=h[1];let o=h[2];typeof o=="string"&&o.startsWith("[")&&o.endsWith("]")?o=o.slice(1,-1).split(",").map(l=>l.trim().replace(/^"|"$/g,"")).filter(l=>l.length>0):typeof o=="string"&&/^\d+$/.test(o)?o=parseInt(o,10):typeof o=="string"&&o.startsWith('"')&&o.endsWith('"')&&(o=o.slice(1,-1)),a[p]=o}}return c.version!=null&&s.push(c),{frontmatter:{title:a.title??"Untitled",category:a.category??"notes",tags:a.tags??[],created:a.created??"",updated:a.updated??"",version:a.version??1,origin:"curated",changelog:s},content:r}}}export{E as CuratedKnowledgeManager};

package/packages/server/dist/index.js

@@ -1,134 +1 @@
-import { parseArgs } from "node:util";
-import { loadConfig } from "./config.js";
-import { createMcpServer, createServer, initializeKnowledgeBase } from "./server.js";
-const { values } = parseArgs({
-  options: {
-    transport: { type: "string", default: process.env.KB_TRANSPORT ?? "stdio" },
-    port: { type: "string", default: process.env.KB_PORT ?? "3210" }
-  }
-});
-async function main() {
-  console.error("[KB] Starting MCP Knowledge Base server...");
-  const config = loadConfig();
-  console.error(
-    `[KB] Config loaded: ${config.sources.length} source(s), store at ${config.store.path}`
-  );
-  if (values.transport === "http") {
-    const { StreamableHTTPServerTransport } = await import("@modelcontextprotocol/sdk/server/streamableHttp.js");
-    const express = (await import("express")).default;
-    const kb = await initializeKnowledgeBase(config);
-    const mcpServer = createMcpServer(kb, config);
-    console.error("[KB] MCP server configured with 46 tools and 2 resources");
-    const app = express();
-    app.use(express.json());
-    app.use((_req, res, next) => {
-      res.setHeader("Access-Control-Allow-Origin", process.env.KB_CORS_ORIGIN ?? "*");
-      res.setHeader("Access-Control-Allow-Methods", "GET, POST, DELETE, OPTIONS");
-      res.setHeader("Access-Control-Allow-Headers", "Content-Type, Authorization");
-      if (_req.method === "OPTIONS") {
-        res.status(204).end();
-        return;
-      }
-      next();
-    });
-    app.get("/health", (_req, res) => {
-      res.json({ status: "ok" });
-    });
-    app.post("/mcp", async (req, res) => {
-      try {
-        const transport = new StreamableHTTPServerTransport({ sessionIdGenerator: void 0 });
-        await mcpServer.connect(transport);
-        await transport.handleRequest(req, res, req.body);
-        res.on("close", () => {
-          transport.close();
-        });
-      } catch (err) {
-        console.error("[KB] MCP handler error:", err);
-        if (!res.headersSent) {
-          res.status(500).json({
-            jsonrpc: "2.0",
-            error: { code: -32603, message: "Internal server error" },
-            id: null
-          });
-        }
-      }
-    });
-    app.get("/mcp", (_req, res) => {
-      res.writeHead(405).end(
-        JSON.stringify({
-          jsonrpc: "2.0",
-          error: { code: -32e3, message: "Method not allowed." },
-          id: null
-        })
-      );
-    });
-    app.delete("/mcp", (_req, res) => {
-      res.writeHead(405).end(
-        JSON.stringify({
-          jsonrpc: "2.0",
-          error: { code: -32e3, message: "Method not allowed." },
-          id: null
-        })
-      );
-    });
-    const port = Number(values.port);
-    const httpServer = app.listen(port, () => {
-      console.error(`[KB] MCP server listening on http://0.0.0.0:${port}/mcp`);
-      const runInitialIndex = async () => {
-        try {
-          const sourcePaths = config.sources.map((s) => s.path).join(", ");
-          console.error(`[KB] Running initial index for sources: ${sourcePaths}`);
-          const result = await kb.indexer.index(config, (p) => {
-            if (p.phase === "crawling" || p.phase === "done") return;
-            if (p.phase === "chunking" && p.currentFile) {
-              console.error(`[KB]   [${p.filesProcessed + 1}/${p.filesTotal}] ${p.currentFile}`);
-            }
-          });
-          console.error(
-            `[KB] Indexed ${result.filesProcessed} files (${result.filesSkipped} skipped, ${result.chunksCreated} chunks) in ${(result.durationMs / 1e3).toFixed(1)}s`
-          );
-          try {
-            const curatedResult = await kb.curated.reindexAll();
-            console.error(
-              `[KB] Curated re-index: ${curatedResult.indexed} entries restored to vector store`
-            );
-          } catch (curatedErr) {
-            console.error("[KB] Curated re-index failed:", curatedErr);
-          }
-        } catch (err) {
-          console.error("[KB] Initial index failed (will retry on kb_reindex):", err);
-        }
-      };
-      runInitialIndex();
-    });
-    const shutdown = async (signal) => {
-      console.error(`[KB] ${signal} received \u2014 shutting down...`);
-      httpServer.close();
-      await mcpServer.close();
-      await kb.store.close();
-      await kb.embedder.shutdown();
-      process.exit(0);
-    };
-    process.on("SIGINT", () => shutdown("SIGINT"));
-    process.on("SIGTERM", () => shutdown("SIGTERM"));
-  } else {
-    const { server, runInitialIndex } = await createServer(config);
-    const { StdioServerTransport } = await import("@modelcontextprotocol/sdk/server/stdio.js");
-    const transport = new StdioServerTransport();
-    await server.connect(transport);
-    console.error("[KB] MCP server started (stdio)");
-    const autoIndex = process.env.KB_AUTO_INDEX !== "false";
-    if (autoIndex) {
-      runInitialIndex();
-    } else {
-      console.error(
-        "[KB] Auto-index disabled (KB_AUTO_INDEX=false). Use kb_reindex to index manually."
-      );
-    }
-  }
-}
-main().catch((err) => {
-  console.error("[KB] Fatal error:", err);
-  process.exit(1);
-});
-//# sourceMappingURL=index.js.map
+import{parseArgs as m}from"node:util";import{loadConfig as f}from"./config.js";import{createMcpServer as w,createServer as x,initializeKnowledgeBase as g}from"./server.js";const{values:u}=m({options:{transport:{type:"string",default:process.env.KB_TRANSPORT??"stdio"},port:{type:"string",default:process.env.KB_PORT??"3210"}}});async function I(){console.error("[KB] Starting MCP Knowledge Base server...");const t=f();if(console.error(`[KB] Config loaded: ${t.sources.length} source(s), store at ${t.store.path}`),u.transport==="http"){const{StreamableHTTPServerTransport:d}=await import("@modelcontextprotocol/sdk/server/streamableHttp.js"),a=(await import("express")).default,i=await g(t),c=w(i,t);console.error("[KB] MCP server configured with 46 tools and 2 resources");const n=a();n.use(a.json()),n.use((s,e,o)=>{if(e.setHeader("Access-Control-Allow-Origin",process.env.KB_CORS_ORIGIN??"*"),e.setHeader("Access-Control-Allow-Methods","GET, POST, DELETE, OPTIONS"),e.setHeader("Access-Control-Allow-Headers","Content-Type, Authorization"),s.method==="OPTIONS"){e.status(204).end();return}o()}),n.get("/health",(s,e)=>{e.json({status:"ok"})}),n.post("/mcp",async(s,e)=>{try{const o=new d({sessionIdGenerator:void 0});await c.connect(o),await o.handleRequest(s,e,s.body),e.on("close",()=>{o.close()})}catch(o){console.error("[KB] MCP handler error:",o),e.headersSent||e.status(500).json({jsonrpc:"2.0",error:{code:-32603,message:"Internal server error"},id:null})}}),n.get("/mcp",(s,e)=>{e.writeHead(405).end(JSON.stringify({jsonrpc:"2.0",error:{code:-32e3,message:"Method not allowed."},id:null}))}),n.delete("/mcp",(s,e)=>{e.writeHead(405).end(JSON.stringify({jsonrpc:"2.0",error:{code:-32e3,message:"Method not allowed."},id:null}))});const l=Number(u.port),h=n.listen(l,()=>{console.error(`[KB] MCP server listening on http://0.0.0.0:${l}/mcp`),(async()=>{try{const e=t.sources.map(r=>r.path).join(", ");console.error(`[KB] Running initial index for sources: ${e}`);const o=await i.indexer.index(t,r=>{r.phase==="crawling"||r.phase==="done"||r.phase==="chunking"&&r.currentFile&&console.error(`[KB]   [${r.filesProcessed+1}/${r.filesTotal}] ${r.currentFile}`)});console.error(`[KB] Indexed ${o.filesProcessed} files (${o.filesSkipped} skipped, ${o.chunksCreated} chunks) in ${(o.durationMs/1e3).toFixed(1)}s`);try{const r=await i.curated.reindexAll();console.error(`[KB] Curated re-index: ${r.indexed} entries restored to vector store`)}catch(r){console.error("[KB] Curated re-index failed:",r)}}catch(e){console.error("[KB] Initial index failed (will retry on kb_reindex):",e)}})()}),p=async s=>{console.error(`[KB] ${s} received \u2014 shutting down...`),h.close(),await c.close(),await i.store.close(),await i.embedder.shutdown(),process.exit(0)};process.on("SIGINT",()=>p("SIGINT")),process.on("SIGTERM",()=>p("SIGTERM"))}else{const{server:d,runInitialIndex:a}=await x(t),{StdioServerTransport:i}=await import("@modelcontextprotocol/sdk/server/stdio.js"),c=new i;await d.connect(c),console.error("[KB] MCP server started (stdio)"),process.env.KB_AUTO_INDEX!=="false"?a():console.error("[KB] Auto-index disabled (KB_AUTO_INDEX=false). Use kb_reindex to index manually.")}}I().catch(t=>{console.error("[KB] Fatal error:",t),process.exit(1)});

package/packages/server/dist/replay-interceptor.js

@@ -1,38 +1 @@
-import { replayAppend } from "@kb/tools";
-function installReplayInterceptor(server) {
-  const original = server.registerTool.bind(server);
-  server.registerTool = (name, config, handler) => {
-    const wrappedHandler = async (args, extra) => {
-      const start = Date.now();
-      try {
-        const result = await handler(args, extra);
-        replayAppend({
-          ts: (/* @__PURE__ */ new Date()).toISOString(),
-          source: "mcp",
-          tool: name,
-          input: JSON.stringify(args),
-          durationMs: Date.now() - start,
-          status: "ok",
-          output: JSON.stringify(result).slice(0, 200)
-        });
-        return result;
-      } catch (err) {
-        replayAppend({
-          ts: (/* @__PURE__ */ new Date()).toISOString(),
-          source: "mcp",
-          tool: name,
-          input: JSON.stringify(args),
-          durationMs: Date.now() - start,
-          status: "error",
-          output: err instanceof Error ? err.message : String(err)
-        });
-        throw err;
-      }
-    };
-    return original(name, config, wrappedHandler);
-  };
-}
-export {
-  installReplayInterceptor
-};
-//# sourceMappingURL=replay-interceptor.js.map
+import{replayAppend as s}from"@kb/tools";function w(n){const i=n.registerTool.bind(n);n.registerTool=(o,a,p)=>i(o,a,async(r,c)=>{const e=Date.now();try{const t=await p(r,c);return s({ts:new Date().toISOString(),source:"mcp",tool:o,input:JSON.stringify(r),durationMs:Date.now()-e,status:"ok",output:JSON.stringify(t).slice(0,200)}),t}catch(t){throw s({ts:new Date().toISOString(),source:"mcp",tool:o,input:JSON.stringify(r),durationMs:Date.now()-e,status:"error",output:t instanceof Error?t.message:String(t)}),t}})}export{w as installReplayInterceptor};

package/packages/server/dist/resources/resources.js

@@ -1,40 +1,2 @@
-function registerResources(server, store) {
-  server.resource(
-    "kb-status",
-    "kb://status",
-    { description: "Current knowledge base status and statistics", mimeType: "text/plain" },
-    async () => {
-      const stats = await store.getStats();
-      return {
-        contents: [
-          {
-            uri: "kb://status",
-            text: `Knowledge Base: ${stats.totalRecords} records from ${stats.totalFiles} files. Last indexed: ${stats.lastIndexedAt ?? "Never"}`,
-            mimeType: "text/plain"
-          }
-        ]
-      };
-    }
-  );
-  server.resource(
-    "kb-file-tree",
-    "kb://file-tree",
-    { description: "List of all indexed source files", mimeType: "text/plain" },
-    async () => {
-      const paths = await store.listSourcePaths();
-      return {
-        contents: [
-          {
-            uri: "kb://file-tree",
-            text: paths.sort().join("\n"),
-            mimeType: "text/plain"
-          }
-        ]
-      };
-    }
-  );
-}
-export {
-  registerResources
-};
-//# sourceMappingURL=resources.js.map
+function r(t,s){t.resource("kb-status","kb://status",{description:"Current knowledge base status and statistics",mimeType:"text/plain"},async()=>{const e=await s.getStats();return{contents:[{uri:"kb://status",text:`Knowledge Base: ${e.totalRecords} records from ${e.totalFiles} files. Last indexed: ${e.lastIndexedAt??"Never"}`,mimeType:"text/plain"}]}}),t.resource("kb-file-tree","kb://file-tree",{description:"List of all indexed source files",mimeType:"text/plain"},async()=>({contents:[{uri:"kb://file-tree",text:(await s.listSourcePaths()).sort().join(`
+`),mimeType:"text/plain"}]}))}export{r as registerResources};