@cognistore/mcp-server 2.2.1 → 2.2.3

package/dist/index.js

@@ -15,13 +15,13 @@ import * as sqliteVec from "sqlite-vec";
 
 // ../../packages/shared/dist/types/knowledge.js
 var KnowledgeType;
-(function(KnowledgeType2) {
-  KnowledgeType2["DECISION"] = "decision";
-  KnowledgeType2["PATTERN"] = "pattern";
-  KnowledgeType2["FIX"] = "fix";
-  KnowledgeType2["CONSTRAINT"] = "constraint";
-  KnowledgeType2["GOTCHA"] = "gotcha";
-  KnowledgeType2["SYSTEM"] = "system";
+(function(KnowledgeType3) {
+  KnowledgeType3["DECISION"] = "decision";
+  KnowledgeType3["PATTERN"] = "pattern";
+  KnowledgeType3["FIX"] = "fix";
+  KnowledgeType3["CONSTRAINT"] = "constraint";
+  KnowledgeType3["GOTCHA"] = "gotcha";
+  KnowledgeType3["SYSTEM"] = "system";
 })(KnowledgeType || (KnowledgeType = {}));
 var KnowledgeStatus;
 (function(KnowledgeStatus2) {
@@ -140,6 +140,24 @@ var mergeTagsBatchSchema = z.object({
     to: z.string().min(1, "to is required")
   })).min(1, "at least one merge is required").max(50, "at most 50 merges per batch")
 });
+var MAX_IMPORT_ITEMS = 5e4;
+var importKnowledgeEntrySchema = z.object({
+  title: z.string().max(2e3).optional(),
+  content: z.string().min(1).max(5e5),
+  tags: z.array(z.string().max(200)).max(200).optional().default([]),
+  type: knowledgeTypeSchema,
+  scope: z.string().min(1).max(300),
+  source: z.string().max(300).optional(),
+  confidenceScore: z.number().min(0).max(1).optional(),
+  expiresAt: z.string().max(64).nullable().optional(),
+  relatedIds: z.array(z.string().max(64)).nullable().optional(),
+  agentId: z.string().max(200).nullable().optional()
+});
+var importSchema = z.object({
+  include: z.array(z.string().max(50)).max(10).optional().default([]),
+  knowledge: z.array(importKnowledgeEntrySchema).max(MAX_IMPORT_ITEMS).optional(),
+  plans: z.array(z.record(z.unknown())).max(MAX_IMPORT_ITEMS).optional()
+});
 
 // ../../packages/core/dist/db/schema/index.js
 var schema_exports = {};
@@ -2538,12 +2556,16 @@ var OllamaEmbeddingClient = class {
   dimensions;
   maxRetries;
   maxInputChars;
+  requestTimeoutMs;
+  healthTimeoutMs;
   constructor(config) {
     this.host = config?.host ?? (process.env.OLLAMA_HOST ?? DEFAULT_OLLAMA_HOST);
     this.model = config?.model ?? (process.env.OLLAMA_MODEL ?? DEFAULT_EMBEDDING_MODEL);
     this.dimensions = config?.dimensions ?? (Number(process.env.EMBEDDING_DIMENSIONS) || DEFAULT_EMBEDDING_DIMENSIONS);
     this.maxRetries = config?.maxRetries ?? 3;
     this.maxInputChars = config?.maxInputChars ?? 2e3;
+    this.requestTimeoutMs = config?.requestTimeoutMs ?? 3e4;
+    this.healthTimeoutMs = config?.healthTimeoutMs ?? 1e4;
   }
   truncateText(text2, maxChars) {
     if (text2.length <= maxChars)
@@ -2590,7 +2612,7 @@ var OllamaEmbeddingClient = class {
   }
   async isHealthy() {
     try {
-      const response = await fetch(`${this.host}/api/tags`);
+      const response = await fetch(`${this.host}/api/tags`, { signal: AbortSignal.timeout(this.healthTimeoutMs) });
       return response.ok;
     } catch {
       return false;
@@ -2598,7 +2620,7 @@ var OllamaEmbeddingClient = class {
   }
   async isModelAvailable() {
     try {
-      const response = await fetch(`${this.host}/api/tags`);
+      const response = await fetch(`${this.host}/api/tags`, { signal: AbortSignal.timeout(this.healthTimeoutMs) });
       if (!response.ok)
         return false;
       const data = await response.json();
@@ -2608,20 +2630,43 @@ var OllamaEmbeddingClient = class {
     }
   }
   async pullModel() {
-    const response = await fetch(`${this.host}/api/pull`, {
-      method: "POST",
-      headers: { "Content-Type": "application/json" },
-      body: JSON.stringify({ name: this.model })
-    });
+    const controller = new AbortController();
+    const connectTimer = setTimeout(() => controller.abort(), 3e4);
+    let response;
+    try {
+      response = await fetch(`${this.host}/api/pull`, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ name: this.model }),
+        signal: controller.signal
+      });
+    } finally {
+      clearTimeout(connectTimer);
+    }
     if (!response.ok) {
       throw new Error(`Failed to pull model ${this.model}: ${response.statusText}`);
     }
     const reader = response.body?.getReader();
     if (reader) {
-      while (true) {
-        const { done } = await reader.read();
-        if (done)
-          break;
+      const IDLE_MS = 12e4;
+      try {
+        while (true) {
+          let timer;
+          const idle = new Promise((_, reject) => {
+            timer = setTimeout(() => reject(new Error(`Model pull stalled (no data for ${IDLE_MS / 1e3}s)`)), IDLE_MS);
+          });
+          try {
+            const { done } = await Promise.race([reader.read(), idle]);
+            if (done)
+              break;
+          } finally {
+            clearTimeout(timer);
+          }
+        }
+      } catch (err) {
+        await reader.cancel().catch(() => {
+        });
+        throw err;
       }
     }
   }
@@ -2653,7 +2698,7 @@ var OllamaEmbeddingClient = class {
     let lastError;
     for (let attempt = 0; attempt < this.maxRetries; attempt++) {
       try {
-        return await fetch(url, init);
+        return await fetch(url, { ...init, signal: init.signal ?? AbortSignal.timeout(this.requestTimeoutMs) });
       } catch (error) {
         lastError = error instanceof Error ? error : new Error(String(error));
         if (attempt < this.maxRetries - 1) {
@@ -4232,7 +4277,7 @@ function createServer(sdk) {
     },
     WRITE,
     async (params) => {
-      const result = sdk.updatePlan(params.planId, { status: "archived" });
+      const result = sdk.updatePlan(params.planId, { status: KnowledgeStatus.ARCHIVED });
       if (!result) return { content: [{ type: "text", text: JSON.stringify({ error: "not_found", type: "plan", id: params.planId }) }] };
       return { content: [{ type: "text", text: JSON.stringify({ archived: true, id: result.id, status: result.status }, null, 2) }] };
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cognistore/mcp-server",
-  "version": "2.2.1",
+  "version": "2.2.3",
   "private": false,
   "type": "module",
   "description": "MCP server for CogniStore — integrates with Claude Code and GitHub Copilot",
@@ -35,6 +35,7 @@
     "build:deps": "pnpm --filter @cognistore/shared build && pnpm --filter @cognistore/core build && pnpm --filter @cognistore/embeddings build && pnpm --filter @cognistore/sdk build",
     "dev": "tsx src/index.ts",
     "clean": "rm -rf dist",
-    "test": "if find . -name '*.test.ts' -o -name '*.spec.ts' | grep -q .; then npx playwright test; else echo 'No tests found, skipping'; fi"
+    "test": "if find . -name '*.test.ts' -o -name '*.spec.ts' | grep -q .; then npx playwright test; else echo 'No tests found, skipping'; fi",
+    "type-check": "tsc --noEmit"
   }
 }