@dikolab/kbdb 0.3.0 → 0.3.2

package/dist/cli.mjs

@@ -78,6 +78,9 @@ function parseArgs(argv) {
           parseInt(argv[++i] ?? "", 10) || 0
         );
         break;
+      case "--algo":
+        options.algo = argv[++i];
+        break;
       case "--title":
         options.title = argv[++i];
         break;
@@ -207,7 +210,8 @@ async function runSearch(client, options) {
     query,
     mode: options.mode,
     limit: options.limit,
-    offset: options.offset
+    offset: options.offset,
+    algo: options.algo
   });
   return {
     items: results.items.map((r) => ({
@@ -2375,6 +2379,11 @@ function createToolDefinitions() {
           offset: {
             type: "number",
             description: "Number of results to skip for pagination (default: 0)"
+          },
+          algo: {
+            type: "string",
+            enum: ["lexical", "vector", "hybrid"],
+            description: "Search algorithm (default: lexical)"
           }
         },
         required: ["query"]
@@ -2788,6 +2797,27 @@ function createToolDefinitions() {
         idempotentHint: true,
         openWorldHint: false
       }
+    },
+    {
+      name: "auto-capture-review",
+      description: "Review and approve pending auto-captured knowledge before storage.",
+      inputSchema: {
+        type: "object",
+        properties: {
+          action: {
+            type: "string",
+            enum: ["list", "approve", "reject"],
+            description: "Action to perform on pending captures"
+          }
+        }
+      },
+      annotations: {
+        title: "Auto-Capture Review",
+        readOnlyHint: false,
+        destructiveHint: false,
+        idempotentHint: true,
+        openWorldHint: false
+      }
     }
   ];
 }
@@ -2861,7 +2891,7 @@ async function handleAgentCreate(client, args) {
     description
   });
   const allKbids = [metaResult.kbid, ...skills];
-  const docid = await client.groupSections(name, allKbids);
+  const docid = await client.groupSections(name, allKbids, "agent");
   return {
     content: [
       {
@@ -3074,6 +3104,8 @@ async function handleToolCall(client, name, args) {
         return await handleAgentGet(client, args);
       case "agent-delete":
         return await handleAgentDelete(client, args);
+      case "auto-capture-review":
+        return handleAutoCaptureReview();
       default:
         return {
           content: [
@@ -3102,7 +3134,8 @@ async function handleSearch2(client, args) {
     query: asString2(args["query"], ""),
     mode: args["mode"],
     limit: args["limit"],
-    offset: args["offset"]
+    offset: args["offset"],
+    algo: args["algo"]
   });
   const stripped = {
     ...results,
@@ -3158,7 +3191,8 @@ async function handleLearn2(client, args) {
     warnings: result.warnings ?? [],
     near_duplicates: (result.near_duplicates ?? []).map((d) => ({
       kbid: d.kbid,
-      similarity: d.similarity
+      similarity: d.similarity,
+      ...d.method !== void 0 ? { method: d.method } : {}
     }))
   };
   return {
@@ -3406,6 +3440,19 @@ async function handleSkillDelete(client, args) {
     ]
   };
 }
+function handleAutoCaptureReview() {
+  return {
+    content: [
+      {
+        type: "text",
+        text: JSON.stringify({
+          pending: [],
+          message: "No pending auto-captured knowledge entries."
+        })
+      }
+    ]
+  };
+}
 async function handleExport2(client, args) {
   const path = typeof args["path"] === "string" ? args["path"] : void 0;
   const result = await client.export(
@@ -3516,8 +3563,12 @@ function buildSkillsText(skillSections) {
 ${lines.join("\n")}`;
 }
 async function handleAgentPrompt(name, client) {
-  const docid = name.slice(6);
-  const doc = await client.retrieveDocument(docid);
+  const agentName = name.slice(6);
+  const agents = await client.listDocumentsByType("agent");
+  const doc = agents.find((a) => a.title === agentName);
+  if (!doc) {
+    throw new Error("agent not found: " + agentName);
+  }
   const sections = doc.sections;
   const metaSection = sections.at(0);
   const skillRefs = sections.slice(1);
@@ -3675,13 +3726,16 @@ var McpServer = class {
       input: process.stdin,
       terminal: false
     });
+    let pending = Promise.resolve();
     await new Promise((resolve4) => {
       rl.on("line", (line) => {
-        void this.dispatchLine(line);
+        pending = pending.then(() => this.dispatchLine(line));
       });
       rl.on("close", () => {
-        this.client.disconnect();
-        resolve4();
+        void pending.then(() => {
+          this.client.disconnect();
+          resolve4();
+        });
       });
     });
   }