@j0hanz/memory-mcp 1.0.0 → 1.0.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +32 -25
- package/dist/completions/index.d.ts.map +1 -1
- package/dist/completions/index.js +8 -5
- package/dist/completions/index.js.map +1 -1
- package/dist/db/index.d.ts.map +1 -1
- package/dist/db/index.js +68 -8
- package/dist/db/index.js.map +1 -1
- package/dist/db/typed.d.ts +9 -3
- package/dist/db/typed.d.ts.map +1 -1
- package/dist/db/typed.js +33 -4
- package/dist/db/typed.js.map +1 -1
- package/dist/index.js +13 -8
- package/dist/index.js.map +1 -1
- package/dist/instructions.md +102 -56
- package/dist/lib/errors.d.ts +3 -0
- package/dist/lib/errors.d.ts.map +1 -1
- package/dist/lib/errors.js +11 -4
- package/dist/lib/errors.js.map +1 -1
- package/dist/lib/hash.d.ts.map +1 -1
- package/dist/lib/hash.js +9 -2
- package/dist/lib/hash.js.map +1 -1
- package/dist/lib/instructions.d.ts +2 -0
- package/dist/lib/instructions.d.ts.map +1 -0
- package/dist/lib/instructions.js +27 -0
- package/dist/lib/instructions.js.map +1 -0
- package/dist/lib/mcp-utils.d.ts +6 -0
- package/dist/lib/mcp-utils.d.ts.map +1 -0
- package/dist/lib/mcp-utils.js +26 -0
- package/dist/lib/mcp-utils.js.map +1 -0
- package/dist/lib/pagination.d.ts.map +1 -1
- package/dist/lib/pagination.js +14 -7
- package/dist/lib/pagination.js.map +1 -1
- package/dist/lib/search-cursor.d.ts +13 -0
- package/dist/lib/search-cursor.d.ts.map +1 -0
- package/dist/lib/search-cursor.js +89 -0
- package/dist/lib/search-cursor.js.map +1 -0
- package/dist/lib/search.d.ts +16 -0
- package/dist/lib/search.d.ts.map +1 -1
- package/dist/lib/search.js +53 -12
- package/dist/lib/search.js.map +1 -1
- package/dist/lib/tool-response.d.ts +1 -5
- package/dist/lib/tool-response.d.ts.map +1 -1
- package/dist/lib/tool-response.js +9 -7
- package/dist/lib/tool-response.js.map +1 -1
- package/dist/lib/types.d.ts.map +1 -1
- package/dist/lib/types.js +10 -9
- package/dist/lib/types.js.map +1 -1
- package/dist/prompts/index.d.ts.map +1 -1
- package/dist/prompts/index.js +21 -35
- package/dist/prompts/index.js.map +1 -1
- package/dist/resources/index.d.ts.map +1 -1
- package/dist/resources/index.js +12 -32
- package/dist/resources/index.js.map +1 -1
- package/dist/schemas/index.d.ts.map +1 -1
- package/dist/schemas/index.js +1 -1
- package/dist/schemas/index.js.map +1 -1
- package/dist/schemas/inputs.d.ts +5 -5
- package/dist/schemas/inputs.d.ts.map +1 -1
- package/dist/schemas/inputs.js +21 -14
- package/dist/schemas/inputs.js.map +1 -1
- package/dist/schemas/outputs.d.ts +94 -148
- package/dist/schemas/outputs.d.ts.map +1 -1
- package/dist/schemas/outputs.js +59 -66
- package/dist/schemas/outputs.js.map +1 -1
- package/dist/server.d.ts.map +1 -1
- package/dist/server.js +20 -18
- package/dist/server.js.map +1 -1
- package/dist/tools/create-relationship.d.ts.map +1 -1
- package/dist/tools/create-relationship.js +28 -17
- package/dist/tools/create-relationship.js.map +1 -1
- package/dist/tools/delete-memories.d.ts.map +1 -1
- package/dist/tools/delete-memories.js +23 -20
- package/dist/tools/delete-memories.js.map +1 -1
- package/dist/tools/delete-memory.d.ts.map +1 -1
- package/dist/tools/delete-memory.js +13 -13
- package/dist/tools/delete-memory.js.map +1 -1
- package/dist/tools/delete-relationship.d.ts.map +1 -1
- package/dist/tools/delete-relationship.js +18 -12
- package/dist/tools/delete-relationship.js.map +1 -1
- package/dist/tools/get-memory.d.ts.map +1 -1
- package/dist/tools/get-memory.js +16 -8
- package/dist/tools/get-memory.js.map +1 -1
- package/dist/tools/get-relationships.d.ts.map +1 -1
- package/dist/tools/get-relationships.js +27 -16
- package/dist/tools/get-relationships.js.map +1 -1
- package/dist/tools/index.d.ts +6 -14
- package/dist/tools/index.d.ts.map +1 -1
- package/dist/tools/index.js +5 -0
- package/dist/tools/index.js.map +1 -1
- package/dist/tools/memory-stats.d.ts.map +1 -1
- package/dist/tools/memory-stats.js +26 -20
- package/dist/tools/memory-stats.js.map +1 -1
- package/dist/tools/progress.d.ts +34 -0
- package/dist/tools/progress.d.ts.map +1 -0
- package/dist/tools/progress.js +129 -0
- package/dist/tools/progress.js.map +1 -0
- package/dist/tools/recall.d.ts.map +1 -1
- package/dist/tools/recall.js +150 -98
- package/dist/tools/recall.js.map +1 -1
- package/dist/tools/result.d.ts +5 -0
- package/dist/tools/result.d.ts.map +1 -0
- package/dist/tools/result.js +14 -0
- package/dist/tools/result.js.map +1 -0
- package/dist/tools/retrieve-context.d.ts.map +1 -1
- package/dist/tools/retrieve-context.js +99 -26
- package/dist/tools/retrieve-context.js.map +1 -1
- package/dist/tools/search-memories.d.ts.map +1 -1
- package/dist/tools/search-memories.js +28 -40
- package/dist/tools/search-memories.js.map +1 -1
- package/dist/tools/store-memories.d.ts.map +1 -1
- package/dist/tools/store-memories.js +30 -18
- package/dist/tools/store-memories.js.map +1 -1
- package/dist/tools/store-memory.d.ts.map +1 -1
- package/dist/tools/store-memory.js +32 -13
- package/dist/tools/store-memory.js.map +1 -1
- package/dist/tools/update-memory.d.ts.map +1 -1
- package/dist/tools/update-memory.js +35 -16
- package/dist/tools/update-memory.js.map +1 -1
- package/package.json +1 -2
- package/dist/tools/helpers.d.ts +0 -13
- package/dist/tools/helpers.d.ts.map +0 -1
- package/dist/tools/helpers.js +0 -49
- package/dist/tools/helpers.js.map +0 -1
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"create-relationship.js","sourceRoot":"","sources":["../../src/tools/create-relationship.ts"],"names":[],"mappings":"AAKA,OAAO,
|
|
1
|
+
{"version":3,"file":"create-relationship.js","sourceRoot":"","sources":["../../src/tools/create-relationship.ts"],"names":[],"mappings":"AAKA,OAAO,EACL,WAAW,EACX,SAAS,EACT,eAAe,EACf,eAAe,GAChB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,6BAA6B,EAAE,MAAM,sBAAsB,CAAC;AACrE,OAAO,EAAE,8BAA8B,EAAE,MAAM,uBAAuB,CAAC;AACvE,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAGhD,MAAM,uBAAuB,GAAG;sBACV,CAAC;AACvB,MAAM,iBAAiB,GACrB,wDAAwD,CAAC;AAE3D,SAAS,oBAAoB,CAAC,IAAyB,EAAE,IAAY;IACnE,OAAO,GAAG,IAAI,sBAAsB,IAAI,EAAE,CAAC;AAC7C,CAAC;AAED,SAAS,kBAAkB,CACzB,EAAW,EACX,MAAqD;IAErD,MAAM,IAAI,GAAG,EAAE;SACZ,WAAW,CAAmB,iBAAiB,CAAC;SAChD,GAAG,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;IACzC,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;IAEnD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;QACjC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC;IACpD,CAAC;IAED,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QAC/B,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;IAClD,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,MAAiB,EACjB,EAAW;IAEX,MAAM,CAAC,YAAY,CACjB,qBAAqB,EACrB;QACE,KAAK,EAAE,qBAAqB;QAC5B,WAAW,EACT,oHAAoH;QACtH,WAAW,EAAE,6BAA6B;QAC1C,YAAY,EAAE,8BAA8B;QAC5C,WAAW,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE;KAC5D,EACD,eAAe,CACb,KAAK,EAAE,MAAsB,EAAE,EAAE;QAC/B,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,kBAAkB,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;YAC/C,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,mBAAmB,CACxB,WAAW,EACX,oBAAoB,CAAC,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CACjD,CAAC;YACJ,CAAC;YAED,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YACrC,MAAM,MAAM,GAAG,EAAE;iBACd,WAAW,CAAC,uBAAuB,CAAC;iBACpC,GAAG,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC;YAEpE,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,GAAG,CAAC,CAAC;YAEnC,MAAM,YAAY,CAAC,MAAM,EAAE,qBAAqB,EAAE;gBAChD,QAAQ,EAAE,MAAM,CAAC,SAAS;gBAC1B,MAAM,EAAE,MAAM,CAAC,OAAO;gBACtB,YAAY,EAAE,MAAM,CAAC,aAAa;gBAClC,OAAO;aACR,CAAC,CAAC;YAEH,OAAO,kBAAkB,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;QACzC,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;QACE,eAAe,EAAE,CAAC,MAAsB,EAAE,EAAE,CAC1C,0BAA0B,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,UAAU,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,QAAQ,MAAM,CAAC,aAAa,GAAG;KAC5H,CACF,CACF,CAAC;AACJ,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"delete-memories.d.ts","sourceRoot":"","sources":["../../src/tools/delete-memories.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;
|
|
1
|
+
{"version":3,"file":"delete-memories.d.ts","sourceRoot":"","sources":["../../src/tools/delete-memories.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AA8B9C,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CAsD3E"}
|
|
@@ -1,12 +1,18 @@
|
|
|
1
|
-
import { E_UNKNOWN, getErrorMessage } from '../lib/errors.js';
|
|
1
|
+
import { E_UNKNOWN, getErrorMessage, rethrowMcpError } from '../lib/errors.js';
|
|
2
|
+
import { logToolEvent, notifyMemoryResourceUpdated } from '../lib/mcp-utils.js';
|
|
2
3
|
import { createErrorResponse, createToolResponse, } from '../lib/tool-response.js';
|
|
3
4
|
import { DeleteMemoriesInputSchema } from '../schemas/inputs.js';
|
|
4
5
|
import { BatchResultSchema } from '../schemas/outputs.js';
|
|
5
|
-
import {
|
|
6
|
+
import { wrapToolHandler } from './progress.js';
|
|
6
7
|
const DELETE_MEMORY_SQL = 'DELETE FROM memories WHERE hash = ?';
|
|
7
|
-
function
|
|
8
|
-
|
|
9
|
-
|
|
8
|
+
function countDeleted(items) {
|
|
9
|
+
return items.reduce((count, item) => (item.deleted ? count + 1 : count), 0);
|
|
10
|
+
}
|
|
11
|
+
async function notifyDeletedResources(server, items) {
|
|
12
|
+
const notifications = items
|
|
13
|
+
.filter((item) => item.deleted)
|
|
14
|
+
.map((item) => notifyMemoryResourceUpdated(server, item.hash));
|
|
15
|
+
await Promise.allSettled(notifications);
|
|
10
16
|
}
|
|
11
17
|
export function registerDeleteMemories(server, db) {
|
|
12
18
|
server.registerTool('delete_memories', {
|
|
@@ -15,11 +21,11 @@ export function registerDeleteMemories(server, db) {
|
|
|
15
21
|
inputSchema: DeleteMemoriesInputSchema,
|
|
16
22
|
outputSchema: BatchResultSchema,
|
|
17
23
|
annotations: { destructiveHint: true, openWorldHint: false },
|
|
18
|
-
}, async (params) => {
|
|
24
|
+
}, wrapToolHandler(async (params) => {
|
|
19
25
|
try {
|
|
20
|
-
const results =
|
|
26
|
+
const results = db.transaction(() => {
|
|
21
27
|
const items = [];
|
|
22
|
-
const stmt = db.
|
|
28
|
+
const stmt = db.prepareOnce(DELETE_MEMORY_SQL);
|
|
23
29
|
for (const hash of params.hashes) {
|
|
24
30
|
const result = stmt.run(hash);
|
|
25
31
|
items.push({
|
|
@@ -31,25 +37,22 @@ export function registerDeleteMemories(server, db) {
|
|
|
31
37
|
}
|
|
32
38
|
return items;
|
|
33
39
|
});
|
|
34
|
-
const deleted = results
|
|
35
|
-
const
|
|
40
|
+
const deleted = countDeleted(results);
|
|
41
|
+
const succeeded = results.reduce((count, item) => (item.ok ? count + 1 : count), 0);
|
|
42
|
+
const failed = results.length - succeeded;
|
|
36
43
|
await logToolEvent(server, 'delete_memories', {
|
|
37
44
|
total: params.hashes.length,
|
|
38
45
|
deleted,
|
|
39
46
|
});
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
await notifyMemoryResourceUpdated(server, item.hash);
|
|
43
|
-
}
|
|
44
|
-
}
|
|
45
|
-
return createToolResponse({
|
|
46
|
-
ok: true,
|
|
47
|
-
result: { items: results, succeeded, failed },
|
|
48
|
-
});
|
|
47
|
+
await notifyDeletedResources(server, results);
|
|
48
|
+
return createToolResponse({ items: results, succeeded, failed });
|
|
49
49
|
}
|
|
50
50
|
catch (err) {
|
|
51
|
+
rethrowMcpError(err);
|
|
51
52
|
return createErrorResponse(E_UNKNOWN, getErrorMessage(err));
|
|
52
53
|
}
|
|
53
|
-
}
|
|
54
|
+
}, {
|
|
55
|
+
progressMessage: (params) => `⊖ delete_memories: ${params.hashes.length} hashes [batch]`,
|
|
56
|
+
}));
|
|
54
57
|
}
|
|
55
58
|
//# sourceMappingURL=delete-memories.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"delete-memories.js","sourceRoot":"","sources":["../../src/tools/delete-memories.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;
|
|
1
|
+
{"version":3,"file":"delete-memories.js","sourceRoot":"","sources":["../../src/tools/delete-memories.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAC/E,OAAO,EAAE,YAAY,EAAE,2BAA2B,EAAE,MAAM,qBAAqB,CAAC;AAChF,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AAEjC,OAAO,EAAE,yBAAyB,EAAE,MAAM,sBAAsB,CAAC;AACjE,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAIhD,MAAM,iBAAiB,GAAG,qCAAqC,CAAC;AAEhE,SAAS,YAAY,CAAC,KAAiC;IACrD,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;AAC9E,CAAC;AAED,KAAK,UAAU,sBAAsB,CACnC,MAAiB,EACjB,KAAiC;IAEjC,MAAM,aAAa,GAAG,KAAK;SACxB,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC;SAC9B,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,2BAA2B,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IACjE,MAAM,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;AAC1C,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,MAAiB,EAAE,EAAW;IACnE,MAAM,CAAC,YAAY,CACjB,iBAAiB,EACjB;QACE,KAAK,EAAE,yBAAyB;QAChC,WAAW,EACT,iIAAiI;QACnI,WAAW,EAAE,yBAAyB;QACtC,YAAY,EAAE,iBAAiB;QAC/B,WAAW,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE;KAC7D,EACD,eAAe,CACb,KAAK,EAAE,MAA2B,EAAE,EAAE;QACpC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE;gBAClC,MAAM,KAAK,GAAsB,EAAE,CAAC;gBACpC,MAAM,IAAI,GAAG,EAAE,CAAC,WAAW,CAAU,iBAAiB,CAAC,CAAC;gBACxD,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;oBACjC,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;oBAC9B,KAAK,CAAC,IAAI,CAAC;wBACT,IAAI;wBACJ,EAAE,EAAE,IAAI;wBACR,OAAO,EAAE,KAAK;wBACd,OAAO,EAAE,MAAM,CAAC,OAAO,GAAG,CAAC;qBAC5B,CAAC,CAAC;gBACL,CAAC;gBACD,OAAO,KAAK,CAAC;YACf,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;YACtC,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,CAC9B,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,EAC9C,CAAC,CACF,CAAC;YACF,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;YAE1C,MAAM,YAAY,CAAC,MAAM,EAAE,iBAAiB,EAAE;gBAC5C,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM;gBAC3B,OAAO;aACR,CAAC,CAAC;YACH,MAAM,sBAAsB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAE9C,OAAO,kBAAkB,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,CAAC;QACnE,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;QACE,eAAe,EAAE,CAAC,MAA2B,EAAE,EAAE,CAC/C,sBAAsB,MAAM,CAAC,MAAM,CAAC,MAAM,iBAAiB;KAC9D,CACF,CACF,CAAC;AACJ,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"delete-memory.d.ts","sourceRoot":"","sources":["../../src/tools/delete-memory.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;
|
|
1
|
+
{"version":3,"file":"delete-memory.d.ts","sourceRoot":"","sources":["../../src/tools/delete-memory.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAwB9C,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CAoCzE"}
|
|
@@ -1,11 +1,12 @@
|
|
|
1
|
-
import { E_NOT_FOUND, E_UNKNOWN, getErrorMessage } from '../lib/errors.js';
|
|
1
|
+
import { E_NOT_FOUND, E_UNKNOWN, getErrorMessage, rethrowMcpError, } from '../lib/errors.js';
|
|
2
|
+
import { logToolEvent, notifyMemoryResourceUpdated } from '../lib/mcp-utils.js';
|
|
2
3
|
import { createErrorResponse, createToolResponse, } from '../lib/tool-response.js';
|
|
3
4
|
import { DeleteMemoryInputSchema } from '../schemas/inputs.js';
|
|
4
5
|
import { DeleteResultSchema } from '../schemas/outputs.js';
|
|
5
|
-
import {
|
|
6
|
+
import { wrapToolHandler } from './progress.js';
|
|
6
7
|
const DELETE_MEMORY_SQL = 'DELETE FROM memories WHERE hash = ?';
|
|
7
|
-
function
|
|
8
|
-
return
|
|
8
|
+
function deleteByHash(db, hash) {
|
|
9
|
+
return db.prepareOnce(DELETE_MEMORY_SQL).run(hash).changes > 0;
|
|
9
10
|
}
|
|
10
11
|
export function registerDeleteMemory(server, db) {
|
|
11
12
|
server.registerTool('delete_memory', {
|
|
@@ -14,22 +15,21 @@ export function registerDeleteMemory(server, db) {
|
|
|
14
15
|
inputSchema: DeleteMemoryInputSchema,
|
|
15
16
|
outputSchema: DeleteResultSchema,
|
|
16
17
|
annotations: { destructiveHint: true, openWorldHint: false },
|
|
17
|
-
}, async (params) => {
|
|
18
|
+
}, wrapToolHandler(async (params) => {
|
|
18
19
|
try {
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
return createErrorResponse(E_NOT_FOUND, formatMemoryNotFound(params.hash));
|
|
20
|
+
if (!deleteByHash(db, params.hash)) {
|
|
21
|
+
return createErrorResponse(E_NOT_FOUND, `Memory not found: ${params.hash}`);
|
|
22
22
|
}
|
|
23
23
|
await logToolEvent(server, 'delete', { hash: params.hash });
|
|
24
24
|
await notifyMemoryResourceUpdated(server, params.hash);
|
|
25
|
-
return createToolResponse({
|
|
26
|
-
ok: true,
|
|
27
|
-
result: { deleted: true, hash: params.hash },
|
|
28
|
-
});
|
|
25
|
+
return createToolResponse({ deleted: true, hash: params.hash });
|
|
29
26
|
}
|
|
30
27
|
catch (err) {
|
|
28
|
+
rethrowMcpError(err);
|
|
31
29
|
return createErrorResponse(E_UNKNOWN, getErrorMessage(err));
|
|
32
30
|
}
|
|
33
|
-
}
|
|
31
|
+
}, {
|
|
32
|
+
progressMessage: (params) => `⊖ delete_memory: ${params.hash.slice(0, 12)}... [single]`,
|
|
33
|
+
}));
|
|
34
34
|
}
|
|
35
35
|
//# sourceMappingURL=delete-memory.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"delete-memory.js","sourceRoot":"","sources":["../../src/tools/delete-memory.ts"],"names":[],"mappings":"AAKA,OAAO,
|
|
1
|
+
{"version":3,"file":"delete-memory.js","sourceRoot":"","sources":["../../src/tools/delete-memory.ts"],"names":[],"mappings":"AAKA,OAAO,EACL,WAAW,EACX,SAAS,EACT,eAAe,EACf,eAAe,GAChB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,YAAY,EAAE,2BAA2B,EAAE,MAAM,qBAAqB,CAAC;AAChF,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AAC/D,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAIhD,MAAM,iBAAiB,GAAG,qCAAqC,CAAC;AAEhE,SAAS,YAAY,CAAC,EAAW,EAAE,IAAY;IAC7C,OAAO,EAAE,CAAC,WAAW,CAAC,iBAAiB,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,OAAO,GAAG,CAAC,CAAC;AACjE,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,MAAiB,EAAE,EAAW;IACjE,MAAM,CAAC,YAAY,CACjB,eAAe,EACf;QACE,KAAK,EAAE,eAAe;QACtB,WAAW,EACT,0FAA0F;QAC5F,WAAW,EAAE,uBAAuB;QACpC,YAAY,EAAE,kBAAkB;QAChC,WAAW,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE;KAC7D,EACD,eAAe,CACb,KAAK,EAAE,MAAmB,EAAE,EAAE;QAC5B,IAAI,CAAC;YACH,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;gBACnC,OAAO,mBAAmB,CACxB,WAAW,EACX,qBAAqB,MAAM,CAAC,IAAI,EAAE,CACnC,CAAC;YACJ,CAAC;YAED,MAAM,YAAY,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;YAC5D,MAAM,2BAA2B,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;YAEvD,OAAO,kBAAkB,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;QAClE,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;QACE,eAAe,EAAE,CAAC,MAAmB,EAAE,EAAE,CACvC,oBAAoB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,cAAc;KAC7D,CACF,CACF,CAAC;AACJ,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"delete-relationship.d.ts","sourceRoot":"","sources":["../../src/tools/delete-relationship.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;
|
|
1
|
+
{"version":3,"file":"delete-relationship.d.ts","sourceRoot":"","sources":["../../src/tools/delete-relationship.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AA4C9C,wBAAgB,0BAA0B,CACxC,MAAM,EAAE,SAAS,EACjB,EAAE,EAAE,OAAO,GACV,IAAI,CAgCN"}
|
|
@@ -1,12 +1,21 @@
|
|
|
1
|
-
import { E_NOT_FOUND, E_UNKNOWN, getErrorMessage } from '../lib/errors.js';
|
|
1
|
+
import { E_NOT_FOUND, E_UNKNOWN, getErrorMessage, rethrowMcpError, } from '../lib/errors.js';
|
|
2
2
|
import { createErrorResponse, createToolResponse, } from '../lib/tool-response.js';
|
|
3
3
|
import { DeleteRelationshipInputSchema } from '../schemas/inputs.js';
|
|
4
4
|
import { DeleteRelationshipResultSchema } from '../schemas/outputs.js';
|
|
5
|
+
import { wrapToolHandler } from './progress.js';
|
|
5
6
|
const DELETE_RELATIONSHIP_SQL = 'DELETE FROM relationships WHERE from_hash = ? AND to_hash = ? AND relation_type = ?';
|
|
6
7
|
const DELETE_RELATIONSHIP_RESULT = { deleted: true };
|
|
7
8
|
function formatRelationship(params) {
|
|
8
9
|
return `${params.from_hash} -[${params.relation_type}]-> ${params.to_hash}`;
|
|
9
10
|
}
|
|
11
|
+
function createNotFoundRelationshipMessage(params) {
|
|
12
|
+
return `Relationship not found: ${formatRelationship(params)}`;
|
|
13
|
+
}
|
|
14
|
+
function deleteRelationship(db, params) {
|
|
15
|
+
return (db
|
|
16
|
+
.prepareOnce(DELETE_RELATIONSHIP_SQL)
|
|
17
|
+
.run(params.from_hash, params.to_hash, params.relation_type).changes > 0);
|
|
18
|
+
}
|
|
10
19
|
export function registerDeleteRelationship(server, db) {
|
|
11
20
|
server.registerTool('delete_relationship', {
|
|
12
21
|
title: 'Delete Relationship',
|
|
@@ -14,22 +23,19 @@ export function registerDeleteRelationship(server, db) {
|
|
|
14
23
|
inputSchema: DeleteRelationshipInputSchema,
|
|
15
24
|
outputSchema: DeleteRelationshipResultSchema,
|
|
16
25
|
annotations: { destructiveHint: true, openWorldHint: false },
|
|
17
|
-
}, (params) => {
|
|
26
|
+
}, wrapToolHandler((params) => {
|
|
18
27
|
try {
|
|
19
|
-
|
|
20
|
-
|
|
21
|
-
.run(params.from_hash, params.to_hash, params.relation_type);
|
|
22
|
-
if (result.changes === 0) {
|
|
23
|
-
return createErrorResponse(E_NOT_FOUND, `Relationship not found: ${formatRelationship(params)}`);
|
|
28
|
+
if (!deleteRelationship(db, params)) {
|
|
29
|
+
return createErrorResponse(E_NOT_FOUND, createNotFoundRelationshipMessage(params));
|
|
24
30
|
}
|
|
25
|
-
return createToolResponse(
|
|
26
|
-
ok: true,
|
|
27
|
-
result: DELETE_RELATIONSHIP_RESULT,
|
|
28
|
-
});
|
|
31
|
+
return createToolResponse(DELETE_RELATIONSHIP_RESULT);
|
|
29
32
|
}
|
|
30
33
|
catch (err) {
|
|
34
|
+
rethrowMcpError(err);
|
|
31
35
|
return createErrorResponse(E_UNKNOWN, getErrorMessage(err));
|
|
32
36
|
}
|
|
33
|
-
}
|
|
37
|
+
}, {
|
|
38
|
+
progressMessage: (params) => `⊖ delete_relationship: ${params.from_hash.slice(0, 8)}... -> ${params.to_hash.slice(0, 8)}... [${params.relation_type}]`,
|
|
39
|
+
}));
|
|
34
40
|
}
|
|
35
41
|
//# sourceMappingURL=delete-relationship.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"delete-relationship.js","sourceRoot":"","sources":["../../src/tools/delete-relationship.ts"],"names":[],"mappings":"AAKA,OAAO,
|
|
1
|
+
{"version":3,"file":"delete-relationship.js","sourceRoot":"","sources":["../../src/tools/delete-relationship.ts"],"names":[],"mappings":"AAKA,OAAO,EACL,WAAW,EACX,SAAS,EACT,eAAe,EACf,eAAe,GAChB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,6BAA6B,EAAE,MAAM,sBAAsB,CAAC;AACrE,OAAO,EAAE,8BAA8B,EAAE,MAAM,uBAAuB,CAAC;AACvE,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAIhD,MAAM,uBAAuB,GAC3B,qFAAqF,CAAC;AACxF,MAAM,0BAA0B,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;AAErD,SAAS,kBAAkB,CACzB,MAAuE;IAEvE,OAAO,GAAG,MAAM,CAAC,SAAS,MAAM,MAAM,CAAC,aAAa,OAAO,MAAM,CAAC,OAAO,EAAE,CAAC;AAC9E,CAAC;AAED,SAAS,iCAAiC,CACxC,MAAuE;IAEvE,OAAO,2BAA2B,kBAAkB,CAAC,MAAM,CAAC,EAAE,CAAC;AACjE,CAAC;AAED,SAAS,kBAAkB,CACzB,EAAW,EACX,MAAuE;IAEvE,OAAO,CACL,EAAE;SACC,WAAW,CAAC,uBAAuB,CAAC;SACpC,GAAG,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC,OAAO,GAAG,CAAC,CAC3E,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,MAAiB,EACjB,EAAW;IAEX,MAAM,CAAC,YAAY,CACjB,qBAAqB,EACrB;QACE,KAAK,EAAE,qBAAqB;QAC5B,WAAW,EAAE,sDAAsD;QACnE,WAAW,EAAE,6BAA6B;QAC1C,YAAY,EAAE,8BAA8B;QAC5C,WAAW,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE;KAC7D,EACD,eAAe,CACb,CAAC,MAAsB,EAAE,EAAE;QACzB,IAAI,CAAC;YACH,IAAI,CAAC,kBAAkB,CAAC,EAAE,EAAE,MAAM,CAAC,EAAE,CAAC;gBACpC,OAAO,mBAAmB,CACxB,WAAW,EACX,iCAAiC,CAAC,MAAM,CAAC,CAC1C,CAAC;YACJ,CAAC;YAED,OAAO,kBAAkB,CAAC,0BAA0B,CAAC,CAAC;QACxD,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;QACE,eAAe,EAAE,CAAC,MAAsB,EAAE,EAAE,CAC1C,0BAA0B,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,UAAU,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,QAAQ,MAAM,CAAC,aAAa,GAAG;KAC5H,CACF,CACF,CAAC;AACJ,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"get-memory.d.ts","sourceRoot":"","sources":["../../src/tools/get-memory.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;
|
|
1
|
+
{"version":3,"file":"get-memory.d.ts","sourceRoot":"","sources":["../../src/tools/get-memory.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AA4B9C,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CAgCtE"}
|
package/dist/tools/get-memory.js
CHANGED
|
@@ -1,11 +1,15 @@
|
|
|
1
|
-
import { E_NOT_FOUND, E_UNKNOWN, getErrorMessage } from '../lib/errors.js';
|
|
1
|
+
import { E_NOT_FOUND, E_UNKNOWN, getErrorMessage, rethrowMcpError, } from '../lib/errors.js';
|
|
2
2
|
import { createErrorResponse, createToolResponse, } from '../lib/tool-response.js';
|
|
3
3
|
import { parseMemoryRow } from '../lib/types.js';
|
|
4
4
|
import { GetMemoryInputSchema } from '../schemas/inputs.js';
|
|
5
5
|
import { MemoryResultSchema } from '../schemas/outputs.js';
|
|
6
|
-
import {
|
|
7
|
-
|
|
8
|
-
|
|
6
|
+
import { wrapToolHandler } from './progress.js';
|
|
7
|
+
const SELECT_MEMORY_BY_HASH_SQL = 'SELECT * FROM memories WHERE hash = ?';
|
|
8
|
+
function getMemoryRow(db, hash) {
|
|
9
|
+
return db.prepareOnce(SELECT_MEMORY_BY_HASH_SQL).get(hash);
|
|
10
|
+
}
|
|
11
|
+
function notFound(hash) {
|
|
12
|
+
return createErrorResponse(E_NOT_FOUND, `Memory not found: ${hash}`);
|
|
9
13
|
}
|
|
10
14
|
export function registerGetMemory(server, db) {
|
|
11
15
|
server.registerTool('get_memory', {
|
|
@@ -14,17 +18,21 @@ export function registerGetMemory(server, db) {
|
|
|
14
18
|
inputSchema: GetMemoryInputSchema,
|
|
15
19
|
outputSchema: MemoryResultSchema,
|
|
16
20
|
annotations: { readOnlyHint: true, openWorldHint: false },
|
|
17
|
-
}, (params) => {
|
|
21
|
+
}, wrapToolHandler((params) => {
|
|
18
22
|
try {
|
|
19
23
|
const row = getMemoryRow(db, params.hash);
|
|
20
24
|
if (!row) {
|
|
21
|
-
return
|
|
25
|
+
return notFound(params.hash);
|
|
22
26
|
}
|
|
23
|
-
|
|
27
|
+
const memory = parseMemoryRow(row);
|
|
28
|
+
return createToolResponse({ ...memory });
|
|
24
29
|
}
|
|
25
30
|
catch (err) {
|
|
31
|
+
rethrowMcpError(err);
|
|
26
32
|
return createErrorResponse(E_UNKNOWN, getErrorMessage(err));
|
|
27
33
|
}
|
|
28
|
-
}
|
|
34
|
+
}, {
|
|
35
|
+
progressMessage: (params) => `⊙ get_memory: ${params.hash.slice(0, 12)}... [single]`,
|
|
36
|
+
}));
|
|
29
37
|
}
|
|
30
38
|
//# sourceMappingURL=get-memory.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"get-memory.js","sourceRoot":"","sources":["../../src/tools/get-memory.ts"],"names":[],"mappings":"AAKA,OAAO,
|
|
1
|
+
{"version":3,"file":"get-memory.js","sourceRoot":"","sources":["../../src/tools/get-memory.ts"],"names":[],"mappings":"AAKA,OAAO,EACL,WAAW,EACX,SAAS,EACT,eAAe,EACf,eAAe,GAChB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAkB,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjE,OAAO,EAAE,oBAAoB,EAAE,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAIhD,MAAM,yBAAyB,GAAG,uCAAuC,CAAC;AAE1E,SAAS,YAAY,CAAC,EAAW,EAAE,IAAY;IAC7C,OAAO,EAAE,CAAC,WAAW,CAAY,yBAAyB,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;AACxE,CAAC;AAED,SAAS,QAAQ,CAAC,IAAY;IAC5B,OAAO,mBAAmB,CAAC,WAAW,EAAE,qBAAqB,IAAI,EAAE,CAAC,CAAC;AACvE,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,MAAiB,EAAE,EAAW;IAC9D,MAAM,CAAC,YAAY,CACjB,YAAY,EACZ;QACE,KAAK,EAAE,YAAY;QACnB,WAAW,EAAE,+CAA+C;QAC5D,WAAW,EAAE,oBAAoB;QACjC,YAAY,EAAE,kBAAkB;QAChC,WAAW,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE;KAC1D,EACD,eAAe,CACb,CAAC,MAAgB,EAAE,EAAE;QACnB,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,YAAY,CAAC,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;YAE1C,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,OAAO,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC/B,CAAC;YAED,MAAM,MAAM,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;YACnC,OAAO,kBAAkB,CAAC,EAAE,GAAG,MAAM,EAAE,CAAC,CAAC;QAC3C,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;QACE,eAAe,EAAE,CAAC,MAAgB,EAAE,EAAE,CACpC,iBAAiB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,cAAc;KAC1D,CACF,CACF,CAAC;AACJ,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"get-relationships.d.ts","sourceRoot":"","sources":["../../src/tools/get-relationships.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;
|
|
1
|
+
{"version":3,"file":"get-relationships.d.ts","sourceRoot":"","sources":["../../src/tools/get-relationships.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAgF9C,wBAAgB,wBAAwB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CA4C7E"}
|
|
@@ -1,9 +1,14 @@
|
|
|
1
|
-
import { E_NOT_FOUND, E_UNKNOWN, getErrorMessage } from '../lib/errors.js';
|
|
1
|
+
import { E_NOT_FOUND, E_UNKNOWN, getErrorMessage, rethrowMcpError, } from '../lib/errors.js';
|
|
2
2
|
import { createErrorResponse, createToolResponse, } from '../lib/tool-response.js';
|
|
3
3
|
import { parseTags } from '../lib/types.js';
|
|
4
4
|
import { GetRelationshipsInputSchema } from '../schemas/inputs.js';
|
|
5
5
|
import { RelationshipResultSchema } from '../schemas/outputs.js';
|
|
6
|
-
import {
|
|
6
|
+
import { wrapToolHandler } from './progress.js';
|
|
7
|
+
const SELECT_MEMORY_HASH_SQL = 'SELECT hash FROM memories WHERE hash = ?';
|
|
8
|
+
function memoryExists(db, hash) {
|
|
9
|
+
return (db.prepareOnce(SELECT_MEMORY_HASH_SQL).get(hash) !==
|
|
10
|
+
undefined);
|
|
11
|
+
}
|
|
7
12
|
const DIRECTIONS_BY_MODE = {
|
|
8
13
|
outgoing: ['outgoing'],
|
|
9
14
|
incoming: ['incoming'],
|
|
@@ -13,7 +18,7 @@ function loadRelationships(db, hash, direction) {
|
|
|
13
18
|
const joinCondition = direction === 'outgoing' ? 'r.to_hash = m.hash' : 'r.from_hash = m.hash';
|
|
14
19
|
const whereColumn = direction === 'outgoing' ? 'r.from_hash' : 'r.to_hash';
|
|
15
20
|
return db
|
|
16
|
-
.
|
|
21
|
+
.prepareOnce(`SELECT r.from_hash, r.to_hash, r.relation_type, r.created_at,
|
|
17
22
|
m.hash AS linked_hash, m.content AS linked_content, m.tags AS linked_tags
|
|
18
23
|
FROM relationships r
|
|
19
24
|
JOIN memories m ON ${joinCondition}
|
|
@@ -21,6 +26,17 @@ function loadRelationships(db, hash, direction) {
|
|
|
21
26
|
ORDER BY r.created_at DESC`)
|
|
22
27
|
.all(hash);
|
|
23
28
|
}
|
|
29
|
+
function toRelationshipWithMemory(row) {
|
|
30
|
+
return {
|
|
31
|
+
from_hash: row.from_hash,
|
|
32
|
+
to_hash: row.to_hash,
|
|
33
|
+
relation_type: row.relation_type,
|
|
34
|
+
created_at: row.created_at,
|
|
35
|
+
linked_hash: row.linked_hash,
|
|
36
|
+
linked_content: row.linked_content,
|
|
37
|
+
linked_tags: parseTags(row.linked_tags),
|
|
38
|
+
};
|
|
39
|
+
}
|
|
24
40
|
export function registerGetRelationships(server, db) {
|
|
25
41
|
server.registerTool('get_relationships', {
|
|
26
42
|
title: 'Get Relationships',
|
|
@@ -28,30 +44,25 @@ export function registerGetRelationships(server, db) {
|
|
|
28
44
|
inputSchema: GetRelationshipsInputSchema,
|
|
29
45
|
outputSchema: RelationshipResultSchema,
|
|
30
46
|
annotations: { readOnlyHint: true, openWorldHint: false },
|
|
31
|
-
}, (params) => {
|
|
47
|
+
}, wrapToolHandler((params) => {
|
|
32
48
|
try {
|
|
33
49
|
if (!memoryExists(db, params.hash)) {
|
|
34
50
|
return createErrorResponse(E_NOT_FOUND, `Memory not found: ${params.hash}`);
|
|
35
51
|
}
|
|
36
52
|
const directions = DIRECTIONS_BY_MODE[params.direction];
|
|
37
53
|
const rows = directions.flatMap((direction) => loadRelationships(db, params.hash, direction));
|
|
38
|
-
const relationships = rows.map(
|
|
39
|
-
from_hash: r.from_hash,
|
|
40
|
-
to_hash: r.to_hash,
|
|
41
|
-
relation_type: r.relation_type,
|
|
42
|
-
created_at: r.created_at,
|
|
43
|
-
linked_hash: r.linked_hash,
|
|
44
|
-
linked_content: r.linked_content,
|
|
45
|
-
linked_tags: parseTags(r.linked_tags),
|
|
46
|
-
}));
|
|
54
|
+
const relationships = rows.map(toRelationshipWithMemory);
|
|
47
55
|
return createToolResponse({
|
|
48
|
-
|
|
49
|
-
|
|
56
|
+
relationships,
|
|
57
|
+
count: relationships.length,
|
|
50
58
|
});
|
|
51
59
|
}
|
|
52
60
|
catch (err) {
|
|
61
|
+
rethrowMcpError(err);
|
|
53
62
|
return createErrorResponse(E_UNKNOWN, getErrorMessage(err));
|
|
54
63
|
}
|
|
55
|
-
}
|
|
64
|
+
}, {
|
|
65
|
+
progressMessage: (params) => `⊙ get_relationships: ${params.hash.slice(0, 12)}... [${params.direction}]`,
|
|
66
|
+
}));
|
|
56
67
|
}
|
|
57
68
|
//# sourceMappingURL=get-relationships.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"get-relationships.js","sourceRoot":"","sources":["../../src/tools/get-relationships.ts"],"names":[],"mappings":"AAKA,OAAO,
|
|
1
|
+
{"version":3,"file":"get-relationships.js","sourceRoot":"","sources":["../../src/tools/get-relationships.ts"],"names":[],"mappings":"AAKA,OAAO,EACL,WAAW,EACX,SAAS,EACT,eAAe,EACf,eAAe,GAChB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAE5C,OAAO,EAAE,2BAA2B,EAAE,MAAM,sBAAsB,CAAC;AACnE,OAAO,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAC;AACjE,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAIhD,MAAM,sBAAsB,GAAG,0CAA0C,CAAC;AAE1E,SAAS,YAAY,CAAC,EAAW,EAAE,IAAY;IAC7C,OAAO,CACL,EAAE,CAAC,WAAW,CAAmB,sBAAsB,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC;QAClE,SAAS,CACV,CAAC;AACJ,CAAC;AAWD,MAAM,kBAAkB,GAGpB;IACF,QAAQ,EAAE,CAAC,UAAU,CAAC;IACtB,QAAQ,EAAE,CAAC,UAAU,CAAC;IACtB,IAAI,EAAE,CAAC,UAAU,EAAE,UAAU,CAAC;CAC/B,CAAC;AAEF,SAAS,iBAAiB,CACxB,EAAW,EACX,IAAY,EACZ,SAAgC;IAEhC,MAAM,aAAa,GACjB,SAAS,KAAK,UAAU,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,sBAAsB,CAAC;IAC3E,MAAM,WAAW,GAAG,SAAS,KAAK,UAAU,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,WAAW,CAAC;IAC3E,OAAO,EAAE;SACN,WAAW,CACV;;;4BAGsB,aAAa;eAC1B,WAAW;kCACQ,CAC7B;SACA,GAAG,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED,SAAS,wBAAwB,CAC/B,GAAwB;IAExB,OAAO;QACL,SAAS,EAAE,GAAG,CAAC,SAAS;QACxB,OAAO,EAAE,GAAG,CAAC,OAAO;QACpB,aAAa,EAAE,GAAG,CAAC,aAAa;QAChC,UAAU,EAAE,GAAG,CAAC,UAAU;QAC1B,WAAW,EAAE,GAAG,CAAC,WAAW;QAC5B,cAAc,EAAE,GAAG,CAAC,cAAc;QAClC,WAAW,EAAE,SAAS,CAAC,GAAG,CAAC,WAAW,CAAC;KACxC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,MAAiB,EAAE,EAAW;IACrE,MAAM,CAAC,YAAY,CACjB,mBAAmB,EACnB;QACE,KAAK,EAAE,mBAAmB;QAC1B,WAAW,EACT,6HAA6H;QAC/H,WAAW,EAAE,2BAA2B;QACxC,YAAY,EAAE,wBAAwB;QACtC,WAAW,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE;KAC1D,EACD,eAAe,CACb,CAAC,MAAmB,EAAE,EAAE;QACtB,IAAI,CAAC;YACH,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;gBACnC,OAAO,mBAAmB,CACxB,WAAW,EACX,qBAAqB,MAAM,CAAC,IAAI,EAAE,CACnC,CAAC;YACJ,CAAC;YAED,MAAM,UAAU,GAAG,kBAAkB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YACxD,MAAM,IAAI,GAA0B,UAAU,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE,CACnE,iBAAiB,CAAC,EAAE,EAAE,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,CAC9C,CAAC;YACF,MAAM,aAAa,GAA6B,IAAI,CAAC,GAAG,CACtD,wBAAwB,CACzB,CAAC;YAEF,OAAO,kBAAkB,CAAC;gBACxB,aAAa;gBACb,KAAK,EAAE,aAAa,CAAC,MAAM;aAC5B,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;QACE,eAAe,EAAE,CAAC,MAAmB,EAAE,EAAE,CACvC,wBAAwB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,MAAM,CAAC,SAAS,GAAG;KAC9E,CACF,CACF,CAAC;AACJ,CAAC"}
|
package/dist/tools/index.d.ts
CHANGED
|
@@ -1,15 +1,7 @@
|
|
|
1
|
-
import {
|
|
2
|
-
import {
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
|
|
6
|
-
|
|
7
|
-
import { registerMemoryStats } from './memory-stats.js';
|
|
8
|
-
import { registerRecall } from './recall.js';
|
|
9
|
-
import { registerRetrieveContext } from './retrieve-context.js';
|
|
10
|
-
import { registerSearchMemories } from './search-memories.js';
|
|
11
|
-
import { registerStoreMemories } from './store-memories.js';
|
|
12
|
-
import { registerStoreMemory } from './store-memory.js';
|
|
13
|
-
import { registerUpdateMemory } from './update-memory.js';
|
|
14
|
-
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];
|
|
1
|
+
import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
|
|
2
|
+
import type { TypedDb } from '../db/typed.js';
|
|
3
|
+
type ToolRegistrar = (server: McpServer, db: TypedDb) => void;
|
|
4
|
+
export declare const TOOL_REGISTRARS: readonly ToolRegistrar[];
|
|
5
|
+
export declare function registerAllTools(server: McpServer, db: TypedDb): void;
|
|
6
|
+
export {};
|
|
15
7
|
//# sourceMappingURL=index.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,
|
|
1
|
+
{"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
CHANGED
|
@@ -26,4 +26,9 @@ export const TOOL_REGISTRARS = [
|
|
|
26
26
|
registerRecall,
|
|
27
27
|
registerRetrieveContext,
|
|
28
28
|
];
|
|
29
|
+
export function registerAllTools(server, db) {
|
|
30
|
+
for (const registerTool of TOOL_REGISTRARS) {
|
|
31
|
+
registerTool(server, db);
|
|
32
|
+
}
|
|
33
|
+
}
|
|
29
34
|
//# sourceMappingURL=index.js.map
|
package/dist/tools/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"
|
|
1
|
+
{"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"}
|
|
@@ -1 +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;
|
|
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;AAgD9C,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CA8CxE"}
|
|
@@ -1,7 +1,8 @@
|
|
|
1
|
-
import { E_UNKNOWN, getErrorMessage } from '../lib/errors.js';
|
|
1
|
+
import { E_UNKNOWN, getErrorMessage, rethrowMcpError } from '../lib/errors.js';
|
|
2
2
|
import { createErrorResponse, createToolResponse, } from '../lib/tool-response.js';
|
|
3
3
|
import { MemoryStatsInputSchema } from '../schemas/inputs.js';
|
|
4
4
|
import { StatsResultSchema } from '../schemas/outputs.js';
|
|
5
|
+
import { wrapToolHandler } from './progress.js';
|
|
5
6
|
const TOTAL_MEMORIES_SQL = 'SELECT COUNT(*) AS total FROM memories';
|
|
6
7
|
const TOTAL_RELATIONSHIPS_SQL = 'SELECT COUNT(*) AS total FROM relationships';
|
|
7
8
|
const TYPE_COUNTS_SQL = 'SELECT memory_type, COUNT(*) AS count FROM memories GROUP BY memory_type ORDER BY count DESC';
|
|
@@ -16,7 +17,15 @@ function toTypeCounts(rows) {
|
|
|
16
17
|
return byType;
|
|
17
18
|
}
|
|
18
19
|
function getTotalCount(db, sql) {
|
|
19
|
-
return db.
|
|
20
|
+
return db.prepareOnce(sql).get()?.total ?? 0;
|
|
21
|
+
}
|
|
22
|
+
function getMemoryExtremes(db) {
|
|
23
|
+
const oldestRow = db.prepareOnce(OLDEST_MEMORY_SQL).get();
|
|
24
|
+
const newestRow = db.prepareOnce(NEWEST_MEMORY_SQL).get();
|
|
25
|
+
return {
|
|
26
|
+
oldest: oldestRow?.oldest ?? null,
|
|
27
|
+
newest: newestRow?.newest ?? null,
|
|
28
|
+
};
|
|
20
29
|
}
|
|
21
30
|
export function registerMemoryStats(server, db) {
|
|
22
31
|
server.registerTool('memory_stats', {
|
|
@@ -25,36 +34,33 @@ export function registerMemoryStats(server, db) {
|
|
|
25
34
|
inputSchema: MemoryStatsInputSchema,
|
|
26
35
|
outputSchema: StatsResultSchema,
|
|
27
36
|
annotations: { readOnlyHint: true, openWorldHint: false },
|
|
28
|
-
}, () => {
|
|
37
|
+
}, wrapToolHandler(() => {
|
|
29
38
|
try {
|
|
30
39
|
const totalMemories = getTotalCount(db, TOTAL_MEMORIES_SQL);
|
|
31
40
|
const totalRelationships = getTotalCount(db, TOTAL_RELATIONSHIPS_SQL);
|
|
32
|
-
const typeRows = db.
|
|
33
|
-
const
|
|
34
|
-
const newestRow = db.prepare(NEWEST_MEMORY_SQL).get();
|
|
41
|
+
const typeRows = db.prepareOnce(TYPE_COUNTS_SQL).all();
|
|
42
|
+
const { oldest, newest } = getMemoryExtremes(db);
|
|
35
43
|
const avgImportanceRow = db
|
|
36
|
-
.
|
|
44
|
+
.prepareOnce(AVG_IMPORTANCE_SQL)
|
|
37
45
|
.get();
|
|
38
46
|
const byType = toTypeCounts(typeRows);
|
|
39
47
|
return createToolResponse({
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
relationships: {
|
|
49
|
-
total: totalRelationships,
|
|
50
|
-
},
|
|
51
|
-
by_type: byType,
|
|
48
|
+
memories: {
|
|
49
|
+
total: totalMemories,
|
|
50
|
+
oldest,
|
|
51
|
+
newest,
|
|
52
|
+
avg_importance: avgImportanceRow?.avg_importance ?? null,
|
|
53
|
+
},
|
|
54
|
+
relationships: {
|
|
55
|
+
total: totalRelationships,
|
|
52
56
|
},
|
|
57
|
+
by_type: byType,
|
|
53
58
|
});
|
|
54
59
|
}
|
|
55
60
|
catch (err) {
|
|
61
|
+
rethrowMcpError(err);
|
|
56
62
|
return createErrorResponse(E_UNKNOWN, getErrorMessage(err));
|
|
57
63
|
}
|
|
58
|
-
});
|
|
64
|
+
}, { progressMessage: () => '⊙ memory_stats: [aggregate]' }));
|
|
59
65
|
}
|
|
60
66
|
//# sourceMappingURL=memory-stats.js.map
|
|
@@ -1 +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;
|
|
1
|
+
{"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,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AAEjC,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAMhD,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,WAAW,CAAW,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,KAAK,IAAI,CAAC,CAAC;AACzD,CAAC;AAED,SAAS,iBAAiB,CAAC,EAAW;IAIpC,MAAM,SAAS,GAAG,EAAE,CAAC,WAAW,CAAY,iBAAiB,CAAC,CAAC,GAAG,EAAE,CAAC;IACrE,MAAM,SAAS,GAAG,EAAE,CAAC,WAAW,CAAY,iBAAiB,CAAC,CAAC,GAAG,EAAE,CAAC;IACrE,OAAO;QACL,MAAM,EAAE,SAAS,EAAE,MAAM,IAAI,IAAI;QACjC,MAAM,EAAE,SAAS,EAAE,MAAM,IAAI,IAAI;KAClC,CAAC;AACJ,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,eAAe,CACb,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,WAAW,CAAU,eAAe,CAAC,CAAC,GAAG,EAAE,CAAC;YAChE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAEjD,MAAM,gBAAgB,GAAG,EAAE;iBACxB,WAAW,CAAmB,kBAAkB,CAAC;iBACjD,GAAG,EAAE,CAAC;YAET,MAAM,MAAM,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;YAEtC,OAAO,kBAAkB,CAAC;gBACxB,QAAQ,EAAE;oBACR,KAAK,EAAE,aAAa;oBACpB,MAAM;oBACN,MAAM;oBACN,cAAc,EAAE,gBAAgB,EAAE,cAAc,IAAI,IAAI;iBACzD;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"}
|
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
import type { CallToolResult, ProgressNotification } from '@modelcontextprotocol/sdk/types.js';
|
|
2
|
+
interface ProgressMeta {
|
|
3
|
+
progressToken?: unknown;
|
|
4
|
+
}
|
|
5
|
+
interface ProgressContext {
|
|
6
|
+
_meta?: ProgressMeta;
|
|
7
|
+
sendNotification?: (notification: ProgressNotification) => Promise<void>;
|
|
8
|
+
}
|
|
9
|
+
interface ProgressUpdate {
|
|
10
|
+
current: number;
|
|
11
|
+
total?: number;
|
|
12
|
+
message?: string;
|
|
13
|
+
}
|
|
14
|
+
interface ProgressReporterOptions {
|
|
15
|
+
minIntervalMs?: number;
|
|
16
|
+
}
|
|
17
|
+
interface ProgressSnapshot {
|
|
18
|
+
current: number;
|
|
19
|
+
total?: number;
|
|
20
|
+
}
|
|
21
|
+
export type ProgressReporter<TProgress extends ProgressSnapshot = ProgressUpdate> = ((progress: TProgress) => void) & {
|
|
22
|
+
flush: () => Promise<void>;
|
|
23
|
+
};
|
|
24
|
+
interface WrappedHandlerOptions<TArgs> {
|
|
25
|
+
completionMessage?: (args: TArgs, result: CallToolResult) => string;
|
|
26
|
+
progressMessage: (args: TArgs) => string;
|
|
27
|
+
}
|
|
28
|
+
type ToolHandler<TArgs> = (args: TArgs, extra: ProgressContext) => Promise<CallToolResult> | CallToolResult;
|
|
29
|
+
export declare function notifyProgress(extra: ProgressContext, progress: ProgressUpdate): Promise<void>;
|
|
30
|
+
export declare function createProgressReporter(extra: ProgressContext, options?: ProgressReporterOptions): ProgressReporter;
|
|
31
|
+
export declare function progressWithMessage(reporter: ProgressReporter, getMessage: (progress: ProgressSnapshot) => string): ProgressReporter<ProgressSnapshot>;
|
|
32
|
+
export declare function wrapToolHandler<TArgs>(handler: ToolHandler<TArgs>, options: WrappedHandlerOptions<TArgs>): ToolHandler<TArgs>;
|
|
33
|
+
export {};
|
|
34
|
+
//# sourceMappingURL=progress.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
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;AAI5C,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;AAsD9C,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,CAoCpB"}
|