@cognistore/mcp-server 1.0.2 → 1.0.3

package/dist/index.js

@@ -1601,88 +1601,64 @@ function createServer(sdk) {
     version: "1.0.0"
   });
   let lastSearchResultIds = [];
-  server.tool(
-    "addKnowledge",
-    "Store a knowledge entry. If you have an active plan, ALWAYS pass planId to auto-link as output.",
-    {
-      title: z2.string().describe("Short descriptive title"),
-      content: z2.string().describe("The knowledge content text"),
-      tags: z2.array(z2.string()).describe("Categorical tags for filtering"),
-      type: z2.enum(knowledgeTypeValues).describe("Type: decision, pattern, fix, constraint, or gotcha"),
-      scope: z2.string().describe('Scope: "global" or "workspace:<project-name>"'),
-      source: z2.string().describe("Source of the knowledge"),
-      confidenceScore: z2.number().min(0).max(1).optional().describe("Confidence score 0-1"),
-      agentId: z2.string().optional().describe("ID of the agent that created this"),
-      planId: z2.string().optional().describe("Plan ID to auto-link this knowledge as output. ALWAYS pass this if you have an active plan.")
-    },
-    WRITE,
-    async (params) => {
-      const result = await sdk.addKnowledge({
-        title: params.title,
-        content: params.content,
-        tags: params.tags,
-        type: params.type,
-        scope: params.scope,
-        source: params.source,
-        confidenceScore: params.confidenceScore,
-        agentId: params.agentId
-      });
-      let linked = false;
-      let linkWarning = "";
-      if (params.planId && result.type !== "system") {
-        try {
-          await sdk.addPlanRelation(params.planId, result.id, "output");
-          linked = true;
-        } catch (e) {
-          linkWarning = e instanceof Error ? e.message : "Unknown linking error";
-        }
-      }
-      const response = { entry: result };
-      if (params.planId) {
-        response.linked = linked;
-        response.planId = params.planId;
-        if (linkWarning) response.linkWarning = linkWarning;
+  const knowledgeEntrySchema = z2.object({
+    title: z2.string().describe("Short descriptive title"),
+    content: z2.string().describe("The knowledge content text"),
+    tags: z2.array(z2.string()).describe("Categorical tags for filtering"),
+    type: z2.enum(knowledgeTypeValues).describe("Type: decision, pattern, fix, constraint, or gotcha"),
+    scope: z2.string().describe('Scope: "global" or "workspace:<project-name>"'),
+    source: z2.string().describe("Source of the knowledge"),
+    confidenceScore: z2.number().min(0).max(1).optional().describe("Confidence score 0-1"),
+    agentId: z2.string().optional().describe("ID of the agent that created this"),
+    planId: z2.string().optional().describe("Plan ID to auto-link this knowledge as output. ALWAYS pass this if you have an active plan.")
+  });
+  async function createEntry(params) {
+    const entry = await sdk.addKnowledge({
+      title: params.title,
+      content: params.content,
+      tags: params.tags,
+      type: params.type,
+      scope: params.scope,
+      source: params.source,
+      confidenceScore: params.confidenceScore,
+      agentId: params.agentId
+    });
+    let linked = false;
+    let linkWarning = "";
+    if (params.planId && entry.type !== "system") {
+      try {
+        await sdk.addPlanRelation(params.planId, entry.id, "output");
+        linked = true;
+      } catch (e) {
+        linkWarning = e instanceof Error ? e.message : "Unknown linking error";
       }
-      return { content: [{ type: "text", text: JSON.stringify(response, null, 2) }] };
     }
-  );
+    const result = { entry };
+    if (params.planId) {
+      result.linked = linked;
+      result.planId = params.planId;
+      if (linkWarning) result.linkWarning = linkWarning;
+    }
+    return result;
+  }
   server.tool(
-    "addKnowledgeBatch",
-    "Create multiple knowledge entries at once. Pass planId per entry for output linking. Reduces tool calls.",
+    "addKnowledge",
+    "Store one or multiple knowledge entries. Pass a single object or an array. If you have an active plan, ALWAYS pass planId to auto-link as output.",
     {
-      entries: z2.array(z2.object({
-        title: z2.string(),
-        content: z2.string(),
-        tags: z2.array(z2.string()),
-        type: z2.enum(knowledgeTypeValues),
-        scope: z2.string(),
-        source: z2.string(),
-        planId: z2.string().optional()
-      })).describe("Array of knowledge entries to create")
+      entries: z2.union([
+        knowledgeEntrySchema,
+        z2.array(knowledgeEntrySchema)
+      ]).describe("A single knowledge entry object, or an array of entries")
     },
     WRITE,
     async (params) => {
+      const items = Array.isArray(params.entries) ? params.entries : [params.entries];
       const results = [];
-      for (const e of params.entries) {
-        const entry = await sdk.addKnowledge({
-          title: e.title,
-          content: e.content,
-          tags: e.tags,
-          type: e.type,
-          scope: e.scope,
-          source: e.source
-        });
-        let linked = false;
-        let linkWarning = "";
-        if (e.planId && entry.type !== "system") {
-          try {
-            await sdk.addPlanRelation(e.planId, entry.id, "output");
-            linked = true;
-          } catch (err) {
-            linkWarning = err instanceof Error ? err.message : "Unknown linking error";
-          }
-        }
-        results.push({ entry, linked, ...e.planId ? { planId: e.planId } : {}, ...linkWarning ? { linkWarning } : {} });
+      for (const item of items) {
+        results.push(await createEntry(item));
+      }
+      if (results.length === 1) {
+        return { content: [{ type: "text", text: JSON.stringify(results[0], null, 2) }] };
       }
       return { content: [{ type: "text", text: JSON.stringify({ created: results.length, entries: results }, null, 2) }] };
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cognistore/mcp-server",
-  "version": "1.0.2",
+  "version": "1.0.3",
   "private": false,
   "type": "module",
   "description": "MCP server for CogniStore — integrates with Claude Code and GitHub Copilot",