@j0hanz/memory-mcp 1.0.0 → 1.1.0

package/dist/tools/index.d.ts

@@ -1,15 +1,7 @@
-import { registerCreateRelationship } from './create-relationship.js';
-import { registerDeleteMemories } from './delete-memories.js';
-import { registerDeleteMemory } from './delete-memory.js';
-import { registerDeleteRelationship } from './delete-relationship.js';
-import { registerGetMemory } from './get-memory.js';
-import { registerGetRelationships } from './get-relationships.js';
-import { registerMemoryStats } from './memory-stats.js';
-import { registerRecall } from './recall.js';
-import { registerRetrieveContext } from './retrieve-context.js';
-import { registerSearchMemories } from './search-memories.js';
-import { registerStoreMemories } from './store-memories.js';
-import { registerStoreMemory } from './store-memory.js';
-import { registerUpdateMemory } from './update-memory.js';
-export declare const TOOL_REGISTRARS: readonly [typeof registerStoreMemory, typeof registerGetMemory, typeof registerUpdateMemory, typeof registerDeleteMemory, typeof registerMemoryStats, typeof registerStoreMemories, typeof registerDeleteMemories, typeof registerSearchMemories, typeof registerCreateRelationship, typeof registerDeleteRelationship, typeof registerGetRelationships, typeof registerRecall, typeof registerRetrieveContext];
+import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import type { TypedDb } from '../db/typed.js';
+type ToolRegistrar = (server: McpServer, db: TypedDb) => void;
+export declare const TOOL_REGISTRARS: readonly ToolRegistrar[];
+export declare function registerAllTools(server: McpServer, db: TypedDb): void;
+export {};
 //# sourceMappingURL=index.d.ts.map

package/dist/tools/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,EAAE,wBAAwB,EAAE,MAAM,wBAAwB,CAAC;AAClE,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAE1D,eAAO,MAAM,eAAe,iZAclB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAe9C,KAAK,aAAa,GAAG,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,KAAK,IAAI,CAAC;AAE9D,eAAO,MAAM,eAAe,EAAE,SAAS,aAAa,EAcnD,CAAC;AAEF,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CAIrE"}

package/dist/tools/index.js

@@ -26,4 +26,9 @@ export const TOOL_REGISTRARS = [
     registerRecall,
     registerRetrieveContext,
 ];
+export function registerAllTools(server, db) {
+    for (const registerTool of TOOL_REGISTRARS) {
+        registerTool(server, db);
+    }
+}
 //# sourceMappingURL=index.js.map

package/dist/tools/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,EAAE,wBAAwB,EAAE,MAAM,wBAAwB,CAAC;AAClE,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAE1D,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B,mBAAmB;IACnB,iBAAiB;IACjB,oBAAoB;IACpB,oBAAoB;IACpB,mBAAmB;IACnB,qBAAqB;IACrB,sBAAsB;IACtB,sBAAsB;IACtB,0BAA0B;IAC1B,0BAA0B;IAC1B,wBAAwB;IACxB,cAAc;IACd,uBAAuB;CACf,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,EAAE,wBAAwB,EAAE,MAAM,wBAAwB,CAAC;AAClE,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAC7C,OAAO,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AAI1D,MAAM,CAAC,MAAM,eAAe,GAA6B;IACvD,mBAAmB;IACnB,iBAAiB;IACjB,oBAAoB;IACpB,oBAAoB;IACpB,mBAAmB;IACnB,qBAAqB;IACrB,sBAAsB;IACtB,sBAAsB;IACtB,0BAA0B;IAC1B,0BAA0B;IAC1B,wBAAwB;IACxB,cAAc;IACd,uBAAuB;CACxB,CAAC;AAEF,MAAM,UAAU,gBAAgB,CAAC,MAAiB,EAAE,EAAW;IAC7D,KAAK,MAAM,YAAY,IAAI,eAAe,EAAE,CAAC;QAC3C,YAAY,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IAC3B,CAAC;AACH,CAAC"}

package/dist/tools/memory-stats.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"memory-stats.d.ts","sourceRoot":"","sources":["../../src/tools/memory-stats.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAmC9C,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CAgDxE"}
+{"version":3,"file":"memory-stats.d.ts","sourceRoot":"","sources":["../../src/tools/memory-stats.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAEzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAgC9C,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CA4CxE"}

package/dist/tools/memory-stats.js

@@ -1,13 +1,10 @@
-import { E_UNKNOWN, getErrorMessage } from '../lib/errors.js';
+import { E_UNKNOWN, getErrorMessage, rethrowMcpError } from '../lib/errors.js';
+import { MEMORY_AGGREGATE_SQL, RELATIONSHIP_COUNT_SQL, TYPE_COUNTS_SQL, } from '../lib/sql.js';
+import { getToolContract } from '../lib/tool-contracts.js';
 import { createErrorResponse, createToolResponse, } from '../lib/tool-response.js';
-import { MemoryStatsInputSchema } from '../schemas/inputs.js';
-import { StatsResultSchema } from '../schemas/outputs.js';
-const TOTAL_MEMORIES_SQL = 'SELECT COUNT(*) AS total FROM memories';
-const TOTAL_RELATIONSHIPS_SQL = 'SELECT COUNT(*) AS total FROM relationships';
-const TYPE_COUNTS_SQL = 'SELECT memory_type, COUNT(*) AS count FROM memories GROUP BY memory_type ORDER BY count DESC';
-const OLDEST_MEMORY_SQL = 'SELECT MIN(created_at) AS oldest FROM memories';
-const NEWEST_MEMORY_SQL = 'SELECT MAX(created_at) AS newest FROM memories';
-const AVG_IMPORTANCE_SQL = 'SELECT AVG(importance) AS avg_importance FROM memories';
+import {} from '../schemas/inputs.js';
+import {} from '../schemas/outputs.js';
+import { wrapToolHandler } from './progress.js';
 function toTypeCounts(rows) {
     const byType = {};
     for (const row of rows) {
@@ -15,46 +12,39 @@ function toTypeCounts(rows) {
     }
     return byType;
 }
-function getTotalCount(db, sql) {
-    return db.prepare(sql).get()?.total ?? 0;
-}
 export function registerMemoryStats(server, db) {
-    server.registerTool('memory_stats', {
-        title: 'Memory Stats',
-        description: 'Return aggregate statistics about the memory store (counts, oldest/newest timestamps, breakdown by type).',
-        inputSchema: MemoryStatsInputSchema,
-        outputSchema: StatsResultSchema,
-        annotations: { readOnlyHint: true, openWorldHint: false },
-    }, () => {
+    const contract = getToolContract('memory_stats');
+    server.registerTool(contract.name, {
+        title: contract.title,
+        description: contract.description,
+        inputSchema: contract.inputSchema,
+        outputSchema: contract.outputSchema,
+        annotations: contract.annotations,
+    }, wrapToolHandler(() => {
         try {
-            const totalMemories = getTotalCount(db, TOTAL_MEMORIES_SQL);
-            const totalRelationships = getTotalCount(db, TOTAL_RELATIONSHIPS_SQL);
-            const typeRows = db.prepare(TYPE_COUNTS_SQL).all();
-            const oldestRow = db.prepare(OLDEST_MEMORY_SQL).get();
-            const newestRow = db.prepare(NEWEST_MEMORY_SQL).get();
-            const avgImportanceRow = db
-                .prepare(AVG_IMPORTANCE_SQL)
+            const aggregate = db
+                .prepareOnce(MEMORY_AGGREGATE_SQL)
                 .get();
+            const totalRelationships = db.prepareOnce(RELATIONSHIP_COUNT_SQL).get()?.total ?? 0;
+            const typeRows = db.prepareOnce(TYPE_COUNTS_SQL).all();
             const byType = toTypeCounts(typeRows);
             return createToolResponse({
-                ok: true,
-                result: {
-                    memories: {
-                        total: totalMemories,
-                        oldest: oldestRow?.oldest ?? null,
-                        newest: newestRow?.newest ?? null,
-                        avg_importance: avgImportanceRow?.avg_importance ?? null,
-                    },
-                    relationships: {
-                        total: totalRelationships,
-                    },
-                    by_type: byType,
+                memories: {
+                    total: aggregate?.total ?? 0,
+                    oldest: aggregate?.oldest ?? null,
+                    newest: aggregate?.newest ?? null,
+                    avg_importance: aggregate?.avg_importance ?? null,
+                },
+                relationships: {
+                    total: totalRelationships,
                 },
+                by_type: byType,
             });
         }
         catch (err) {
+            rethrowMcpError(err);
             return createErrorResponse(E_UNKNOWN, getErrorMessage(err));
         }
-    });
+    }, { progressMessage: () => '⊙ memory_stats: [aggregate]' }));
 }
 //# sourceMappingURL=memory-stats.js.map

package/dist/tools/memory-stats.js.map

@@ -1 +1 @@
-{"version":3,"file":"memory-stats.js","sourceRoot":"","sources":["../../src/tools/memory-stats.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAC9D,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AAEjC,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAM1D,MAAM,kBAAkB,GAAG,wCAAwC,CAAC;AACpE,MAAM,uBAAuB,GAAG,6CAA6C,CAAC;AAC9E,MAAM,eAAe,GACnB,8FAA8F,CAAC;AACjG,MAAM,iBAAiB,GAAG,gDAAgD,CAAC;AAC3E,MAAM,iBAAiB,GAAG,gDAAgD,CAAC;AAC3E,MAAM,kBAAkB,GACtB,wDAAwD,CAAC;AAE3D,SAAS,YAAY,CAAC,IAAe;IACnC,MAAM,MAAM,GAA2B,EAAE,CAAC;IAC1C,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC;IACtC,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,aAAa,CAAC,EAAW,EAAE,GAAW;IAC7C,OAAO,EAAE,CAAC,OAAO,CAAW,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,KAAK,IAAI,CAAC,CAAC;AACrD,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAAiB,EAAE,EAAW;IAChE,MAAM,CAAC,YAAY,CACjB,cAAc,EACd;QACE,KAAK,EAAE,cAAc;QACrB,WAAW,EACT,2GAA2G;QAC7G,WAAW,EAAE,sBAAsB;QACnC,YAAY,EAAE,iBAAiB;QAC/B,WAAW,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE;KAC1D,EACD,GAAG,EAAE;QACH,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,aAAa,CAAC,EAAE,EAAE,kBAAkB,CAAC,CAAC;YAC5D,MAAM,kBAAkB,GAAG,aAAa,CAAC,EAAE,EAAE,uBAAuB,CAAC,CAAC;YAEtE,MAAM,QAAQ,GAAG,EAAE,CAAC,OAAO,CAAU,eAAe,CAAC,CAAC,GAAG,EAAE,CAAC;YAE5D,MAAM,SAAS,GAAG,EAAE,CAAC,OAAO,CAAY,iBAAiB,CAAC,CAAC,GAAG,EAAE,CAAC;YAEjE,MAAM,SAAS,GAAG,EAAE,CAAC,OAAO,CAAY,iBAAiB,CAAC,CAAC,GAAG,EAAE,CAAC;YAEjE,MAAM,gBAAgB,GAAG,EAAE;iBACxB,OAAO,CAAmB,kBAAkB,CAAC;iBAC7C,GAAG,EAAE,CAAC;YAET,MAAM,MAAM,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;YAEtC,OAAO,kBAAkB,CAAC;gBACxB,EAAE,EAAE,IAAI;gBACR,MAAM,EAAE;oBACN,QAAQ,EAAE;wBACR,KAAK,EAAE,aAAa;wBACpB,MAAM,EAAE,SAAS,EAAE,MAAM,IAAI,IAAI;wBACjC,MAAM,EAAE,SAAS,EAAE,MAAM,IAAI,IAAI;wBACjC,cAAc,EAAE,gBAAgB,EAAE,cAAc,IAAI,IAAI;qBACzD;oBACD,aAAa,EAAE;wBACb,KAAK,EAAE,kBAAkB;qBAC1B;oBACD,OAAO,EAAE,MAAM;iBAChB;aACF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,mBAAmB,CAAC,SAAS,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC"}
+{"version":3,"file":"memory-stats.js","sourceRoot":"","sources":["../../src/tools/memory-stats.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAC/E,OAAO,EACL,oBAAoB,EACpB,sBAAsB,EACtB,eAAe,GAChB,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAC3D,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AAEjC,OAAO,EAA+B,MAAM,sBAAsB,CAAC;AACnE,OAAO,EAA0B,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAShD,SAAS,YAAY,CAAC,IAAe;IACnC,MAAM,MAAM,GAA2B,EAAE,CAAC;IAC1C,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC;IACtC,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAAiB,EAAE,EAAW;IAChE,MAAM,QAAQ,GAAG,eAAe,CAAC,cAAc,CAAC,CAAC;IACjD,MAAM,CAAC,YAAY,CACjB,QAAQ,CAAC,IAAI,EACb;QACE,KAAK,EAAE,QAAQ,CAAC,KAAK;QACrB,WAAW,EAAE,QAAQ,CAAC,WAAW;QACjC,WAAW,EAAE,QAAQ,CAAC,WAA4C;QAClE,YAAY,EAAE,QAAQ,CAAC,YAAwC;QAC/D,WAAW,EAAE,QAAQ,CAAC,WAAW;KAClC,EACD,eAAe,CACb,GAAG,EAAE;QACH,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,EAAE;iBACjB,WAAW,CAAqB,oBAAoB,CAAC;iBACrD,GAAG,EAAE,CAAC;YAET,MAAM,kBAAkB,GACtB,EAAE,CAAC,WAAW,CAAW,sBAAsB,CAAC,CAAC,GAAG,EAAE,EAAE,KAAK,IAAI,CAAC,CAAC;YAErE,MAAM,QAAQ,GAAG,EAAE,CAAC,WAAW,CAAU,eAAe,CAAC,CAAC,GAAG,EAAE,CAAC;YAChE,MAAM,MAAM,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;YAEtC,OAAO,kBAAkB,CAAC;gBACxB,QAAQ,EAAE;oBACR,KAAK,EAAE,SAAS,EAAE,KAAK,IAAI,CAAC;oBAC5B,MAAM,EAAE,SAAS,EAAE,MAAM,IAAI,IAAI;oBACjC,MAAM,EAAE,SAAS,EAAE,MAAM,IAAI,IAAI;oBACjC,cAAc,EAAE,SAAS,EAAE,cAAc,IAAI,IAAI;iBAClD;gBACD,aAAa,EAAE;oBACb,KAAK,EAAE,kBAAkB;iBAC1B;gBACD,OAAO,EAAE,MAAM;aAChB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,eAAe,CAAC,GAAG,CAAC,CAAC;YACrB,OAAO,mBAAmB,CAAC,SAAS,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC,EACD,EAAE,eAAe,EAAE,GAAG,EAAE,CAAC,6BAA6B,EAAE,CACzD,CACF,CAAC;AACJ,CAAC"}

package/dist/tools/progress.d.ts

@@ -0,0 +1,34 @@
+import type { CallToolResult, ProgressNotification } from '@modelcontextprotocol/sdk/types.js';
+interface ProgressMeta {
+    progressToken?: unknown;
+}
+interface ProgressContext {
+    _meta?: ProgressMeta;
+    sendNotification?: (notification: ProgressNotification) => Promise<void>;
+}
+interface ProgressUpdate {
+    current: number;
+    total?: number;
+    message?: string;
+}
+interface ProgressReporterOptions {
+    minIntervalMs?: number;
+}
+interface ProgressSnapshot {
+    current: number;
+    total?: number;
+}
+export type ProgressReporter<TProgress extends ProgressSnapshot = ProgressUpdate> = ((progress: TProgress) => void) & {
+    flush: () => Promise<void>;
+};
+interface WrappedHandlerOptions<TArgs> {
+    completionMessage?: (args: TArgs, result: CallToolResult) => string;
+    progressMessage: (args: TArgs) => string;
+}
+type ToolHandler<TArgs> = (args: TArgs, extra: ProgressContext) => Promise<CallToolResult> | CallToolResult;
+export declare function notifyProgress(extra: ProgressContext, progress: ProgressUpdate): Promise<void>;
+export declare function createProgressReporter(extra: ProgressContext, options?: ProgressReporterOptions): ProgressReporter;
+export declare function progressWithMessage(reporter: ProgressReporter, getMessage: (progress: ProgressSnapshot) => string): ProgressReporter<ProgressSnapshot>;
+export declare function wrapToolHandler<TArgs>(handler: ToolHandler<TArgs>, options: WrappedHandlerOptions<TArgs>): ToolHandler<TArgs>;
+export {};
+//# sourceMappingURL=progress.d.ts.map

package/dist/tools/progress.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"progress.d.ts","sourceRoot":"","sources":["../../src/tools/progress.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,cAAc,EACd,oBAAoB,EACrB,MAAM,oCAAoC,CAAC;AAM5C,UAAU,YAAY;IACpB,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB;AAED,UAAU,eAAe;IACvB,KAAK,CAAC,EAAE,YAAY,CAAC;IACrB,gBAAgB,CAAC,EAAE,CAAC,YAAY,EAAE,oBAAoB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1E;AAED,UAAU,cAAc;IACtB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,UAAU,uBAAuB;IAC/B,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,UAAU,gBAAgB;IACxB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,MAAM,gBAAgB,CAC1B,SAAS,SAAS,gBAAgB,GAAG,cAAc,IACjD,CAAC,CAAC,QAAQ,EAAE,SAAS,KAAK,IAAI,CAAC,GAAG;IACpC,KAAK,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CAC5B,CAAC;AAEF,UAAU,qBAAqB,CAAC,KAAK;IACnC,iBAAiB,CAAC,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,cAAc,KAAK,MAAM,CAAC;IACpE,eAAe,EAAE,CAAC,IAAI,EAAE,KAAK,KAAK,MAAM,CAAC;CAC1C;AAED,KAAK,WAAW,CAAC,KAAK,IAAI,CACxB,IAAI,EAAE,KAAK,EACX,KAAK,EAAE,eAAe,KACnB,OAAO,CAAC,cAAc,CAAC,GAAG,cAAc,CAAC;AA4D9C,wBAAsB,cAAc,CAClC,KAAK,EAAE,eAAe,EACtB,QAAQ,EAAE,cAAc,GACvB,OAAO,CAAC,IAAI,CAAC,CAkBf;AAED,wBAAgB,sBAAsB,CACpC,KAAK,EAAE,eAAe,EACtB,OAAO,GAAE,uBAA4B,GACpC,gBAAgB,CA+ClB;AAED,wBAAgB,mBAAmB,CACjC,QAAQ,EAAE,gBAAgB,EAC1B,UAAU,EAAE,CAAC,QAAQ,EAAE,gBAAgB,KAAK,MAAM,GACjD,gBAAgB,CAAC,gBAAgB,CAAC,CAepC;AAED,wBAAgB,eAAe,CAAC,KAAK,EACnC,OAAO,EAAE,WAAW,CAAC,KAAK,CAAC,EAC3B,OAAO,EAAE,qBAAqB,CAAC,KAAK,CAAC,GACpC,WAAW,CAAC,KAAK,CAAC,CAsCpB"}

package/dist/tools/progress.js

@@ -0,0 +1,136 @@
+import { E_CANCELLED } from '../lib/errors.js';
+const DEFAULT_PROGRESS_INTERVAL_MS = 250;
+function toProgressToken(value) {
+    if (typeof value === 'string' || typeof value === 'number') {
+        return value;
+    }
+    return undefined;
+}
+function toNotificationParams(progressToken, progress) {
+    const params = {
+        progressToken,
+        progress: progress.current,
+    };
+    if (progress.total !== undefined) {
+        params.total = progress.total;
+    }
+    if (progress.message !== undefined) {
+        params.message = progress.message;
+    }
+    return params;
+}
+function getResultOutcome(result) {
+    if (result.structuredContent?.error) {
+        const error = result.structuredContent.error;
+        if (error.code === E_CANCELLED) {
+            return 'cancelled';
+        }
+    }
+    if (result.isError) {
+        return 'failed';
+    }
+    if (result.structuredContent?.ok === false) {
+        return 'failed';
+    }
+    return 'completed';
+}
+function defaultCompletionMessage(startMessage, result) {
+    return `${startMessage} • ${getResultOutcome(result)}`;
+}
+export async function notifyProgress(extra, progress) {
+    const progressToken = toProgressToken(extra._meta?.progressToken);
+    if (progressToken === undefined) {
+        return;
+    }
+    if (typeof extra.sendNotification !== 'function') {
+        return;
+    }
+    try {
+        await extra.sendNotification({
+            method: 'notifications/progress',
+            params: toNotificationParams(progressToken, progress),
+        });
+    }
+    catch {
+        // best-effort progress
+    }
+}
+export function createProgressReporter(extra, options = {}) {
+    const minIntervalMs = options.minIntervalMs ?? DEFAULT_PROGRESS_INTERVAL_MS;
+    let lastCurrent = 0;
+    let lastReportedAt = 0;
+    let isCompleted = false;
+    let notificationChain = Promise.resolve();
+    const reporter = ((progress) => {
+        if (isCompleted) {
+            return;
+        }
+        const monotonicCurrent = progress.current >= lastCurrent ? progress.current : lastCurrent;
+        const now = Date.now();
+        const shouldFlush = now - lastReportedAt >= minIntervalMs ||
+            (progress.total !== undefined && monotonicCurrent >= progress.total);
+        lastCurrent = monotonicCurrent;
+        if (!shouldFlush) {
+            return;
+        }
+        lastReportedAt = now;
+        isCompleted =
+            progress.total !== undefined && monotonicCurrent >= progress.total;
+        const payload = { current: monotonicCurrent };
+        if (progress.total !== undefined) {
+            payload.total = progress.total;
+        }
+        if (progress.message !== undefined) {
+            payload.message = progress.message;
+        }
+        notificationChain = notificationChain.then(() => notifyProgress(extra, payload));
+    });
+    reporter.flush = async () => {
+        await notificationChain;
+    };
+    return reporter;
+}
+export function progressWithMessage(reporter, getMessage) {
+    const wrapped = (({ current, total }) => {
+        const message = getMessage(total === undefined ? { current } : { current, total });
+        reporter({
+            current,
+            ...(total === undefined ? {} : { total }),
+            message,
+        });
+    });
+    wrapped.flush = () => reporter.flush();
+    return wrapped;
+}
+export function wrapToolHandler(handler, options) {
+    return async (args, extra) => {
+        const startMessage = options.progressMessage(args);
+        await notifyProgress(extra, {
+            current: 0,
+            total: 1,
+            message: startMessage,
+        });
+        let result;
+        try {
+            result = await handler(args, extra);
+        }
+        catch (error) {
+            const isCancelled = error instanceof Error && error.message === E_CANCELLED;
+            await notifyProgress(extra, {
+                current: 1,
+                total: 1,
+                message: `${startMessage} • ${isCancelled ? 'cancelled' : 'failed'}`,
+            });
+            throw error;
+        }
+        const completionMessage = options.completionMessage?.(args, result) ??
+            defaultCompletionMessage(startMessage, result);
+        await notifyProgress(extra, {
+            current: 1,
+            total: 1,
+            message: completionMessage,
+        });
+        return result;
+    };
+}
+//# sourceMappingURL=progress.js.map

package/dist/tools/progress.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"progress.js","sourceRoot":"","sources":["../../src/tools/progress.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AA4C/C,MAAM,4BAA4B,GAAG,GAAG,CAAC;AAEzC,SAAS,eAAe,CAAC,KAAc;IACrC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC3D,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,oBAAoB,CAC3B,aAA4B,EAC5B,QAAwB;IAExB,MAAM,MAAM,GAAmC;QAC7C,aAAa;QACb,QAAQ,EAAE,QAAQ,CAAC,OAAO;KAC3B,CAAC;IAEF,IAAI,QAAQ,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QACjC,MAAM,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;IAChC,CAAC;IAED,IAAI,QAAQ,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;QACnC,MAAM,CAAC,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC;IACpC,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,gBAAgB,CACvB,MAAsB;IAEtB,IAAI,MAAM,CAAC,iBAAiB,EAAE,KAAK,EAAE,CAAC;QACpC,MAAM,KAAK,GAAG,MAAM,CAAC,iBAAiB,CAAC,KAA0B,CAAC;QAClE,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC/B,OAAO,WAAW,CAAC;QACrB,CAAC;IACH,CAAC;IAED,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,IAAI,MAAM,CAAC,iBAAiB,EAAE,EAAE,KAAK,KAAK,EAAE,CAAC;QAC3C,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,SAAS,wBAAwB,CAC/B,YAAoB,EACpB,MAAsB;IAEtB,OAAO,GAAG,YAAY,MAAM,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC;AACzD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,KAAsB,EACtB,QAAwB;IAExB,MAAM,aAAa,GAAG,eAAe,CAAC,KAAK,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;IAClE,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;QAChC,OAAO;IACT,CAAC;IAED,IAAI,OAAO,KAAK,CAAC,gBAAgB,KAAK,UAAU,EAAE,CAAC;QACjD,OAAO;IACT,CAAC;IAED,IAAI,CAAC;QACH,MAAM,KAAK,CAAC,gBAAgB,CAAC;YAC3B,MAAM,EAAE,wBAAwB;YAChC,MAAM,EAAE,oBAAoB,CAAC,aAAa,EAAE,QAAQ,CAAC;SACtD,CAAC,CAAC;IACL,CAAC;IAAC,MAAM,CAAC;QACP,uBAAuB;IACzB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,KAAsB,EACtB,UAAmC,EAAE;IAErC,MAAM,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,4BAA4B,CAAC;IAC5E,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,cAAc,GAAG,CAAC,CAAC;IACvB,IAAI,WAAW,GAAG,KAAK,CAAC;IACxB,IAAI,iBAAiB,GAAkB,OAAO,CAAC,OAAO,EAAE,CAAC;IAEzD,MAAM,QAAQ,GAAG,CAAC,CAAC,QAAwB,EAAQ,EAAE;QACnD,IAAI,WAAW,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QAED,MAAM,gBAAgB,GACpB,QAAQ,CAAC,OAAO,IAAI,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC;QACnE,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,WAAW,GACf,GAAG,GAAG,cAAc,IAAI,aAAa;YACrC,CAAC,QAAQ,CAAC,KAAK,KAAK,SAAS,IAAI,gBAAgB,IAAI,QAAQ,CAAC,KAAK,CAAC,CAAC;QAEvE,WAAW,GAAG,gBAAgB,CAAC;QAE/B,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QAED,cAAc,GAAG,GAAG,CAAC;QACrB,WAAW;YACT,QAAQ,CAAC,KAAK,KAAK,SAAS,IAAI,gBAAgB,IAAI,QAAQ,CAAC,KAAK,CAAC;QAErE,MAAM,OAAO,GAAmB,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;QAC9D,IAAI,QAAQ,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YACjC,OAAO,CAAC,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC;QACjC,CAAC;QACD,IAAI,QAAQ,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YACnC,OAAO,CAAC,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC;QACrC,CAAC;QAED,iBAAiB,GAAG,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,CAC9C,cAAc,CAAC,KAAK,EAAE,OAAO,CAAC,CAC/B,CAAC;IACJ,CAAC,CAAqB,CAAC;IAEvB,QAAQ,CAAC,KAAK,GAAG,KAAK,IAAmB,EAAE;QACzC,MAAM,iBAAiB,CAAC;IAC1B,CAAC,CAAC;IAEF,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,QAA0B,EAC1B,UAAkD;IAElD,MAAM,OAAO,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAoB,EAAQ,EAAE;QAC9D,MAAM,OAAO,GAAG,UAAU,CACxB,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CACvD,CAAC;QACF,QAAQ,CAAC;YACP,OAAO;YACP,GAAG,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC;YACzC,OAAO;SACR,CAAC,CAAC;IACL,CAAC,CAAuC,CAAC;IAEzC,OAAO,CAAC,KAAK,GAAG,GAAkB,EAAE,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IAEtD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,eAAe,CAC7B,OAA2B,EAC3B,OAAqC;IAErC,OAAO,KAAK,EACV,IAAW,EACX,KAAsB,EACG,EAAE;QAC3B,MAAM,YAAY,GAAG,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QACnD,MAAM,cAAc,CAAC,KAAK,EAAE;YAC1B,OAAO,EAAE,CAAC;YACV,KAAK,EAAE,CAAC;YACR,OAAO,EAAE,YAAY;SACtB,CAAC,CAAC;QAEH,IAAI,MAAsB,CAAC;QAC3B,IAAI,CAAC;YACH,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,WAAW,GACf,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,OAAO,KAAK,WAAW,CAAC;YAC1D,MAAM,cAAc,CAAC,KAAK,EAAE;gBAC1B,OAAO,EAAE,CAAC;gBACV,KAAK,EAAE,CAAC;gBACR,OAAO,EAAE,GAAG,YAAY,MAAM,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,EAAE;aACrE,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;QAED,MAAM,iBAAiB,GACrB,OAAO,CAAC,iBAAiB,EAAE,CAAC,IAAI,EAAE,MAAM,CAAC;YACzC,wBAAwB,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;QAEjD,MAAM,cAAc,CAAC,KAAK,EAAE;YAC1B,OAAO,EAAE,CAAC;YACV,KAAK,EAAE,CAAC;YACR,OAAO,EAAE,iBAAiB;SAC3B,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;AACJ,CAAC"}

package/dist/tools/recall.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"recall.d.ts","sourceRoot":"","sources":["../../src/tools/recall.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAMzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAqO9C,wBAAgB,cAAc,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CA8DnE"}
+{"version":3,"file":"recall.d.ts","sourceRoot":"","sources":["../../src/tools/recall.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAUzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAmO9C,wBAAgB,cAAc,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CAgInE"}