@cognistore/mcp-server 2.2.3 → 2.3.0

package/dist/index.js

@@ -74,7 +74,8 @@ var createKnowledgeSchema = z.object({
   confidenceScore: z.number().min(0).max(1).optional().default(1),
   expiresAt: z.date().nullable().optional().default(null),
   relatedIds: z.array(z.string().uuid()).nullable().optional().default(null),
-  agentId: z.string().nullable().optional().default(null)
+  agentId: z.string().max(64).nullable().optional().default(null),
+  platform: z.string().max(64).nullable().optional().default(null)
 });
 var updateKnowledgeSchema = z.object({
   title: z.string().min(1).optional(),
@@ -86,7 +87,8 @@ var updateKnowledgeSchema = z.object({
   confidenceScore: z.number().min(0).max(1).optional(),
   expiresAt: z.date().nullable().optional(),
   relatedIds: z.array(z.string().uuid()).nullable().optional(),
-  agentId: z.string().nullable().optional()
+  agentId: z.string().max(64).nullable().optional(),
+  platform: z.string().max(64).nullable().optional()
 });
 var searchOptionsSchema = z.object({
   tags: z.array(z.string()).optional(),
@@ -105,6 +107,8 @@ var createPlanSchema = z.object({
   source: z.string().min(1, "Source is required"),
   status: knowledgeStatusSchema.optional().default(KnowledgeStatus.DRAFT),
   planFilePath: z.string().min(1).nullable().optional(),
+  agentId: z.string().max(64).nullable().optional(),
+  platform: z.string().max(64).nullable().optional(),
   tasks: z.array(z.object({
     description: z.string().min(1),
     priority: z.enum(["low", "medium", "high"]).optional()
@@ -117,7 +121,9 @@ var updatePlanSchema = z.object({
   scope: scopeSchema.optional(),
   status: knowledgeStatusSchema.optional(),
   source: z.string().min(1).optional(),
-  planFilePath: z.string().min(1).nullable().optional()
+  planFilePath: z.string().min(1).nullable().optional(),
+  agentId: z.string().max(64).nullable().optional(),
+  platform: z.string().max(64).nullable().optional()
 });
 var createPlanTaskSchema = z.object({
   planId: z.string().min(1),
@@ -151,7 +157,8 @@ var importKnowledgeEntrySchema = z.object({
   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()
+  agentId: z.string().max(200).nullable().optional(),
+  platform: z.string().max(200).nullable().optional()
 });
 var importSchema = z.object({
   include: z.array(z.string().max(50)).max(10).optional().default([]),
@@ -192,11 +199,13 @@ var knowledgeEntries = sqliteTable("knowledge_entries", {
   confidenceScore: real("confidence_score").notNull().default(1),
   relatedIds: text("related_ids", { mode: "json" }).$type(),
   agentId: text("agent_id"),
+  platform: text("platform"),
   createdAt: text("created_at").notNull().$defaultFn(() => (/* @__PURE__ */ new Date()).toISOString()),
   updatedAt: text("updated_at").notNull().$defaultFn(() => (/* @__PURE__ */ new Date()).toISOString())
 }, (table) => [
   index("idx_type").on(table.type),
-  index("idx_scope").on(table.scope)
+  index("idx_scope").on(table.scope),
+  index("idx_knowledge_platform").on(table.platform)
 ]);
 
 // ../../packages/core/dist/db/schema/sqlite-vec.js
@@ -437,6 +446,16 @@ ALTER TABLE plans ADD COLUMN plan_file_path TEXT;
 `,
   "2.1.0": `
 CREATE VIRTUAL TABLE IF NOT EXISTS knowledge_fts USING fts5(id UNINDEXED, title, content, tags);
+`,
+  // v2.3.0: provenance tracking — platform (auto) + agent_id (caller) on entries and plans.
+  // knowledge_entries.agent_id already exists since 0.8.0; only platform is new there.
+  "2.3.0": `
+ALTER TABLE knowledge_entries ADD COLUMN platform TEXT;
+ALTER TABLE plans ADD COLUMN agent_id TEXT;
+ALTER TABLE plans ADD COLUMN platform TEXT;
+CREATE INDEX IF NOT EXISTS idx_knowledge_platform ON knowledge_entries(platform);
+CREATE INDEX IF NOT EXISTS idx_plans_platform ON plans(platform);
+CREATE INDEX IF NOT EXISTS idx_plans_agent ON plans(agent_id);
 `
 };
 function runMigrations(sqlite, migrationsDir) {
@@ -581,7 +600,7 @@ function createPlansEmbeddingsTable(sqlite, dimensions = DEFAULT_EMBEDDING_DIMEN
 
 // ../../packages/core/dist/repositories/knowledge.repository.js
 import { eq, ne, sql, and, or, isNull } from "drizzle-orm";
-var OPERATIONS_RETENTION_DAYS = 30;
+var OPERATIONS_RETENTION_DAYS = 800;
 var KnowledgeRepository = class _KnowledgeRepository {
   db;
   sqlite;
@@ -603,8 +622,8 @@ var KnowledgeRepository = class _KnowledgeRepository {
     const now = (/* @__PURE__ */ new Date()).toISOString();
     const insertTxn = this.sqlite.transaction(() => {
       this.sqlite.prepare(`INSERT INTO knowledge_entries
-           (id, title, content, tags, type, scope, source, version, expires_at, confidence_score, related_ids, agent_id, created_at, updated_at)
-         VALUES (?, ?, ?, ?, ?, ?, ?, 1, ?, ?, ?, ?, ?, ?)`).run(id, input.title, input.content, JSON.stringify(input.tags ?? []), input.type, input.scope, input.source, input.expiresAt ? input.expiresAt.toISOString() : null, input.confidenceScore ?? 1, input.relatedIds ? JSON.stringify(input.relatedIds) : null, input.agentId ?? null, now, now);
+           (id, title, content, tags, type, scope, source, version, expires_at, confidence_score, related_ids, agent_id, platform, created_at, updated_at)
+         VALUES (?, ?, ?, ?, ?, ?, ?, 1, ?, ?, ?, ?, ?, ?, ?)`).run(id, input.title, input.content, JSON.stringify(input.tags ?? []), input.type, input.scope, input.source, input.expiresAt ? input.expiresAt.toISOString() : null, input.confidenceScore ?? 1, input.relatedIds ? JSON.stringify(input.relatedIds) : null, input.agentId ?? null, input.platform ?? null, now, now);
       insertEmbedding(this.sqlite, id, input.embedding);
       try {
         insertFts(this.sqlite, { id, title: input.title, content: input.content, tags: ftsTags(input.tags) });
@@ -712,6 +731,12 @@ var KnowledgeRepository = class _KnowledgeRepository {
       conditions.push(sql`${knowledgeEntries.type} = ${filters.type}`);
     if (filters?.scope)
       conditions.push(sql`${knowledgeEntries.scope} = ${filters.scope}`);
+    if (filters?.agent) {
+      conditions.push(filters.agent === "unspecified" ? isNull(knowledgeEntries.agentId) : sql`${knowledgeEntries.agentId} = ${filters.agent}`);
+    }
+    if (filters?.platform) {
+      conditions.push(filters.platform === "unknown" ? or(isNull(knowledgeEntries.platform), sql`${knowledgeEntries.platform} = 'unknown'`) : sql`${knowledgeEntries.platform} = ${filters.platform}`);
+    }
     if (filters?.tags && filters.tags.length > 0) {
       const tagConditions = filters.tags.map((tag) => sql`EXISTS (SELECT 1 FROM json_each(${knowledgeEntries.tags}) WHERE value = ${tag})`);
       conditions.push(or(...tagConditions));
@@ -962,6 +987,39 @@ var KnowledgeRepository = class _KnowledgeRepository {
     }).from(knowledgeEntries).where(and(...conditions)).groupBy(knowledgeEntries.scope);
     return results.map((r) => ({ scope: r.scope, count: Number(r.count) }));
   }
+  /**
+   * Count entries by the agent that created them. agent_id is caller-provided and
+   * NULL when not passed; COALESCE collapses NULL into a single "unspecified"
+   * bucket (the chart label the frontend round-trips back to IS NULL on filter).
+   */
+  async countByAgent(opts = {}) {
+    const { from, to } = opts;
+    const conditions = [ne(knowledgeEntries.type, "system")];
+    if (from && to) {
+      conditions.push(sql`${knowledgeEntries.createdAt} >= ${from}`);
+      conditions.push(sql`${knowledgeEntries.createdAt} < ${to}`);
+    }
+    const agentLabel = sql`COALESCE(${knowledgeEntries.agentId}, 'unspecified')`;
+    const results = await this.db.select({ agent: agentLabel, count: sql`count(*)` }).from(knowledgeEntries).where(and(...conditions)).groupBy(agentLabel);
+    return results.map((r) => ({ agent: r.agent, count: Number(r.count) }));
+  }
+  /**
+   * Count entries by the platform that created them. platform is NULL for
+   * pre-2.3.0 / dashboard-created rows and the literal "unknown" for MCP rows
+   * whose client couldn't be resolved; COALESCE merges both into one "unknown"
+   * bucket so they don't render as two slices.
+   */
+  async countByPlatform(opts = {}) {
+    const { from, to } = opts;
+    const conditions = [ne(knowledgeEntries.type, "system")];
+    if (from && to) {
+      conditions.push(sql`${knowledgeEntries.createdAt} >= ${from}`);
+      conditions.push(sql`${knowledgeEntries.createdAt} < ${to}`);
+    }
+    const platformLabel = sql`COALESCE(${knowledgeEntries.platform}, 'unknown')`;
+    const results = await this.db.select({ platform: platformLabel, count: sql`count(*)` }).from(knowledgeEntries).where(and(...conditions)).groupBy(platformLabel);
+    return results.map((r) => ({ platform: r.platform, count: Number(r.count) }));
+  }
   async listAll() {
     return this.db.select().from(knowledgeEntries).where(ne(knowledgeEntries.type, "system")).orderBy(sql`${knowledgeEntries.createdAt} DESC`);
   }
@@ -976,7 +1034,7 @@ var KnowledgeRepository = class _KnowledgeRepository {
   createPlan(input) {
     const id = crypto.randomUUID();
     const now = (/* @__PURE__ */ new Date()).toISOString();
-    this.sqlite.prepare("INSERT INTO plans (id, title, content, tags, scope, status, source, plan_file_path, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)").run(id, input.title, input.content, JSON.stringify(input.tags), input.scope, input.status ?? "draft", input.source, input.planFilePath ?? null, now, now);
+    this.sqlite.prepare("INSERT INTO plans (id, title, content, tags, scope, status, source, plan_file_path, agent_id, platform, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)").run(id, input.title, input.content, JSON.stringify(input.tags), input.scope, input.status ?? "draft", input.source, input.planFilePath ?? null, input.agentId ?? null, input.platform ?? null, now, now);
     try {
       insertPlanEmbedding(this.sqlite, id, input.embedding);
     } catch {
@@ -1248,6 +1306,28 @@ var KnowledgeRepository = class _KnowledgeRepository {
     }
     return Object.entries(map).map(([date, counts]) => ({ date, ...counts }));
   }
+  /** Plans created per day, zero-filled for every day in the window (mirrors getOperationsByDay). */
+  getPlansByDay(days = 15) {
+    const cutoff = new Date(Date.now() - days * 24 * 60 * 60 * 1e3).toISOString();
+    const rows = this.sqlite.prepare(`
+      SELECT date(created_at) as date, COUNT(*) as count
+      FROM plans
+      WHERE created_at >= ?
+      GROUP BY date(created_at)
+      ORDER BY date(created_at)
+    `).all(cutoff);
+    const map = {};
+    const now = /* @__PURE__ */ new Date();
+    for (let i = days - 1; i >= 0; i--) {
+      const d = new Date(now);
+      d.setDate(d.getDate() - i);
+      map[d.toISOString().split("T")[0]] = 0;
+    }
+    for (const row of rows) {
+      map[row.date] = row.count;
+    }
+    return Object.entries(map).map(([date, count]) => ({ date, count }));
+  }
   cleanupOldOperations() {
     const cutoff = new Date(Date.now() - OPERATIONS_RETENTION_DAYS * 24 * 60 * 60 * 1e3).toISOString();
     return this.sqlite.prepare("DELETE FROM operations_log WHERE created_at < ?").run(cutoff).changes;
@@ -1364,7 +1444,13 @@ var KnowledgeService = class {
         });
         if (similar.length > 0) {
           const existing = similar[0].entry;
-          const updated = await this.repository.update(existing.id, { ...rest, embedding });
+          const { platform, agentId, ...keep } = rest;
+          const prov = {};
+          if (platform != null)
+            prov.platform = platform;
+          if (agentId != null)
+            prov.agentId = agentId;
+          const updated = await this.repository.update(existing.id, { ...keep, ...prov, embedding });
           this.logOp("write");
           return { ...this.toKnowledgeEntry(updated), deduplicated: true };
         }
@@ -1749,6 +1835,12 @@ var KnowledgeService = class {
   async countByScope(opts = {}) {
     return this.repository.countByScope(opts);
   }
+  async countByAgent(opts = {}) {
+    return this.repository.countByAgent(opts);
+  }
+  async countByPlatform(opts = {}) {
+    return this.repository.countByPlatform(opts);
+  }
   // ─── Plans (separate entity) ────────────────────────────────
   async createPlan(input) {
     const { tasks, skipDedup, ...planInput } = input;
@@ -1996,6 +2088,9 @@ var KnowledgeService = class {
   getOperationsByDay(days = 15) {
     return this.repository.getOperationsByDay(days);
   }
+  getPlansByDay(days = 15) {
+    return this.repository.getPlansByDay(days);
+  }
   cleanupOldOperations() {
     return this.repository.cleanupOldOperations();
   }
@@ -2017,7 +2112,8 @@ var KnowledgeService = class {
       expiresAt: row.expiresAt ? new Date(row.expiresAt) : null,
       confidenceScore: row.confidenceScore,
       relatedIds: row.relatedIds ? Array.isArray(row.relatedIds) ? row.relatedIds : JSON.parse(row.relatedIds) : null,
-      agentId: row.agentId,
+      agentId: row.agentId ?? row.agent_id ?? null,
+      platform: row.platform ?? null,
       createdAt: new Date(row.createdAt ?? row.created_at),
       updatedAt: new Date(row.updatedAt ?? row.updated_at)
     };
@@ -2032,6 +2128,8 @@ var KnowledgeService = class {
       status: row.status,
       source: row.source ?? "",
       planFilePath: row.plan_file_path ?? row.planFilePath ?? null,
+      agentId: row.agent_id ?? row.agentId ?? null,
+      platform: row.platform ?? null,
       createdAt: new Date(row.created_at ?? row.createdAt),
       updatedAt: new Date(row.updated_at ?? row.updatedAt)
     };
@@ -3578,6 +3676,22 @@ var KnowledgeSDK = class {
       throw this.wrapError(error, "Failed to count by scope");
     }
   }
+  async countByAgent(opts = {}) {
+    this.ensureInitialized();
+    try {
+      return await this.service.countByAgent(opts);
+    } catch (error) {
+      throw this.wrapError(error, "Failed to count by agent");
+    }
+  }
+  async countByPlatform(opts = {}) {
+    this.ensureInitialized();
+    try {
+      return await this.service.countByPlatform(opts);
+    } catch (error) {
+      throw this.wrapError(error, "Failed to count by platform");
+    }
+  }
   async getStats() {
     this.ensureInitialized();
     try {
@@ -3733,6 +3847,10 @@ var KnowledgeSDK = class {
     this.ensureInitialized();
     return this.service.getOperationsByDay(days);
   }
+  getPlansByDay(days = 15) {
+    this.ensureInitialized();
+    return this.service.getPlansByDay(days);
+  }
   cleanupOldOperations() {
     if (!this.initialized || !this.service)
       return 0;
@@ -3879,6 +3997,21 @@ function createServer(sdk) {
     version: "1.0.0"
   });
   let lastSearchResultIds = [];
+  const PLATFORM_ALLOWLIST = /* @__PURE__ */ new Set(["claude-code", "copilot", "opencode"]);
+  function normalizeProvenance(s) {
+    if (typeof s !== "string") return void 0;
+    const cleaned = Array.from(s).filter((c) => {
+      const n = c.charCodeAt(0);
+      return n >= 32 && n !== 127;
+    }).join("").trim().slice(0, 64);
+    return cleaned.length > 0 ? cleaned : void 0;
+  }
+  function resolvePlatform() {
+    const raw = normalizeProvenance(process.env.COGNISTORE_PLATFORM) ?? normalizeProvenance(server.server.getClientVersion()?.name);
+    if (!raw) return "unknown";
+    const mapped = raw.toLowerCase() === "claude" ? "claude-code" : raw.toLowerCase();
+    return PLATFORM_ALLOWLIST.has(mapped) ? mapped : "unknown";
+  }
   const knowledgeEntrySchema = z3.object({
     title: z3.string().describe("Short descriptive title"),
     content: z3.string().describe("The knowledge content text"),
@@ -3887,7 +4020,7 @@ function createServer(sdk) {
     scope: z3.string().describe('Scope: "global" or "workspace:<project-name>"'),
     source: z3.string().describe("Source of the knowledge"),
     confidenceScore: z3.number().min(0).max(1).optional().describe("Confidence score 0-1"),
-    agentId: z3.string().optional().describe("ID of the agent that created this"),
+    agentId: z3.string().optional().describe("Your agent/role name (e.g. 'documentation', 'code-reviewer') so entries can be summarized per agent. Pass it whenever you are a named/custom agent."),
     planId: z3.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) {
@@ -3899,7 +4032,8 @@ function createServer(sdk) {
       scope: params.scope,
       source: params.source,
       confidenceScore: params.confidenceScore,
-      agentId: params.agentId
+      agentId: normalizeProvenance(params.agentId),
+      platform: resolvePlatform()
     });
     let linked = false;
     let linkWarning = "";
@@ -4102,6 +4236,7 @@ function createServer(sdk) {
       scope: z3.string().describe('Scope: "global" or "workspace:<project-name>"'),
       source: z3.string().describe("Source/context of the plan"),
       planFilePath: z3.string().optional().describe("ABSOLUTE path to the local plan file you wrote (e.g. a plan-mode file like /home/user/.claude/plans/<name>.md). REQUIRED whenever you persisted the plan to a file \u2014 always link it so the CogniStore plan points back to the on-disk file."),
+      agentId: z3.string().optional().describe("Your agent/role name (e.g. 'documentation') so plans can be summarized per agent. Pass it whenever you are a named/custom agent."),
       relatedKnowledgeIds: z3.array(z3.string()).optional().describe("IDs of knowledge entries consulted during planning (auto-linked as input)"),
       tasks: z3.array(z3.object({
         description: z3.string(),
@@ -4121,6 +4256,8 @@ function createServer(sdk) {
         scope: params.scope,
         source: params.source,
         planFilePath: params.planFilePath,
+        agentId: normalizeProvenance(params.agentId),
+        platform: resolvePlatform(),
         relatedKnowledgeIds: inputIds.size > 0 ? [...inputIds] : void 0,
         tasks: params.tasks
       });

package/package.json

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