@j0hanz/memory-mcp 0.1.1 → 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.
Files changed (123) hide show
  1. package/README.md +32 -25
  2. package/dist/completions/index.d.ts.map +1 -1
  3. package/dist/completions/index.js +8 -5
  4. package/dist/completions/index.js.map +1 -1
  5. package/dist/db/index.d.ts.map +1 -1
  6. package/dist/db/index.js +68 -8
  7. package/dist/db/index.js.map +1 -1
  8. package/dist/db/typed.d.ts +9 -3
  9. package/dist/db/typed.d.ts.map +1 -1
  10. package/dist/db/typed.js +33 -4
  11. package/dist/db/typed.js.map +1 -1
  12. package/dist/index.js +13 -8
  13. package/dist/index.js.map +1 -1
  14. package/dist/instructions.md +102 -56
  15. package/dist/lib/errors.d.ts +3 -0
  16. package/dist/lib/errors.d.ts.map +1 -1
  17. package/dist/lib/errors.js +11 -4
  18. package/dist/lib/errors.js.map +1 -1
  19. package/dist/lib/hash.d.ts.map +1 -1
  20. package/dist/lib/hash.js +9 -2
  21. package/dist/lib/hash.js.map +1 -1
  22. package/dist/lib/instructions.d.ts +2 -0
  23. package/dist/lib/instructions.d.ts.map +1 -0
  24. package/dist/lib/instructions.js +27 -0
  25. package/dist/lib/instructions.js.map +1 -0
  26. package/dist/lib/mcp-utils.d.ts +6 -0
  27. package/dist/lib/mcp-utils.d.ts.map +1 -0
  28. package/dist/lib/mcp-utils.js +26 -0
  29. package/dist/lib/mcp-utils.js.map +1 -0
  30. package/dist/lib/pagination.d.ts.map +1 -1
  31. package/dist/lib/pagination.js +14 -7
  32. package/dist/lib/pagination.js.map +1 -1
  33. package/dist/lib/search-cursor.d.ts +13 -0
  34. package/dist/lib/search-cursor.d.ts.map +1 -0
  35. package/dist/lib/search-cursor.js +89 -0
  36. package/dist/lib/search-cursor.js.map +1 -0
  37. package/dist/lib/search.d.ts +16 -0
  38. package/dist/lib/search.d.ts.map +1 -1
  39. package/dist/lib/search.js +53 -12
  40. package/dist/lib/search.js.map +1 -1
  41. package/dist/lib/tool-response.d.ts +1 -5
  42. package/dist/lib/tool-response.d.ts.map +1 -1
  43. package/dist/lib/tool-response.js +9 -7
  44. package/dist/lib/tool-response.js.map +1 -1
  45. package/dist/lib/types.d.ts.map +1 -1
  46. package/dist/lib/types.js +10 -9
  47. package/dist/lib/types.js.map +1 -1
  48. package/dist/prompts/index.d.ts.map +1 -1
  49. package/dist/prompts/index.js +21 -35
  50. package/dist/prompts/index.js.map +1 -1
  51. package/dist/resources/index.d.ts.map +1 -1
  52. package/dist/resources/index.js +12 -32
  53. package/dist/resources/index.js.map +1 -1
  54. package/dist/schemas/index.d.ts.map +1 -1
  55. package/dist/schemas/index.js +1 -1
  56. package/dist/schemas/index.js.map +1 -1
  57. package/dist/schemas/inputs.d.ts +5 -5
  58. package/dist/schemas/inputs.d.ts.map +1 -1
  59. package/dist/schemas/inputs.js +21 -14
  60. package/dist/schemas/inputs.js.map +1 -1
  61. package/dist/schemas/outputs.d.ts +94 -148
  62. package/dist/schemas/outputs.d.ts.map +1 -1
  63. package/dist/schemas/outputs.js +59 -66
  64. package/dist/schemas/outputs.js.map +1 -1
  65. package/dist/server.d.ts.map +1 -1
  66. package/dist/server.js +20 -18
  67. package/dist/server.js.map +1 -1
  68. package/dist/tools/create-relationship.d.ts.map +1 -1
  69. package/dist/tools/create-relationship.js +28 -17
  70. package/dist/tools/create-relationship.js.map +1 -1
  71. package/dist/tools/delete-memories.d.ts.map +1 -1
  72. package/dist/tools/delete-memories.js +23 -20
  73. package/dist/tools/delete-memories.js.map +1 -1
  74. package/dist/tools/delete-memory.d.ts.map +1 -1
  75. package/dist/tools/delete-memory.js +13 -13
  76. package/dist/tools/delete-memory.js.map +1 -1
  77. package/dist/tools/delete-relationship.d.ts.map +1 -1
  78. package/dist/tools/delete-relationship.js +18 -12
  79. package/dist/tools/delete-relationship.js.map +1 -1
  80. package/dist/tools/get-memory.d.ts.map +1 -1
  81. package/dist/tools/get-memory.js +16 -8
  82. package/dist/tools/get-memory.js.map +1 -1
  83. package/dist/tools/get-relationships.d.ts.map +1 -1
  84. package/dist/tools/get-relationships.js +27 -16
  85. package/dist/tools/get-relationships.js.map +1 -1
  86. package/dist/tools/index.d.ts +6 -14
  87. package/dist/tools/index.d.ts.map +1 -1
  88. package/dist/tools/index.js +5 -0
  89. package/dist/tools/index.js.map +1 -1
  90. package/dist/tools/memory-stats.d.ts.map +1 -1
  91. package/dist/tools/memory-stats.js +26 -20
  92. package/dist/tools/memory-stats.js.map +1 -1
  93. package/dist/tools/progress.d.ts +34 -0
  94. package/dist/tools/progress.d.ts.map +1 -0
  95. package/dist/tools/progress.js +129 -0
  96. package/dist/tools/progress.js.map +1 -0
  97. package/dist/tools/recall.d.ts.map +1 -1
  98. package/dist/tools/recall.js +150 -98
  99. package/dist/tools/recall.js.map +1 -1
  100. package/dist/tools/result.d.ts +5 -0
  101. package/dist/tools/result.d.ts.map +1 -0
  102. package/dist/tools/result.js +14 -0
  103. package/dist/tools/result.js.map +1 -0
  104. package/dist/tools/retrieve-context.d.ts.map +1 -1
  105. package/dist/tools/retrieve-context.js +99 -26
  106. package/dist/tools/retrieve-context.js.map +1 -1
  107. package/dist/tools/search-memories.d.ts.map +1 -1
  108. package/dist/tools/search-memories.js +28 -40
  109. package/dist/tools/search-memories.js.map +1 -1
  110. package/dist/tools/store-memories.d.ts.map +1 -1
  111. package/dist/tools/store-memories.js +30 -18
  112. package/dist/tools/store-memories.js.map +1 -1
  113. package/dist/tools/store-memory.d.ts.map +1 -1
  114. package/dist/tools/store-memory.js +32 -13
  115. package/dist/tools/store-memory.js.map +1 -1
  116. package/dist/tools/update-memory.d.ts.map +1 -1
  117. package/dist/tools/update-memory.js +35 -16
  118. package/dist/tools/update-memory.js.map +1 -1
  119. package/package.json +1 -2
  120. package/dist/tools/helpers.d.ts +0 -13
  121. package/dist/tools/helpers.d.ts.map +0 -1
  122. package/dist/tools/helpers.js +0 -49
  123. package/dist/tools/helpers.js.map +0 -1
@@ -1,19 +1,23 @@
1
- import { E_UNKNOWN, getErrorMessage } from '../lib/errors.js';
1
+ import { ErrorCode, McpError } from '@modelcontextprotocol/sdk/types.js';
2
+ import { E_UNKNOWN, getErrorMessage, rethrowMcpError } from '../lib/errors.js';
2
3
  import { sanitizeFtsQuery } from '../lib/search.js';
3
4
  import { createErrorResponse, createToolResponse, } from '../lib/tool-response.js';
4
5
  import { parseMemoryRow } from '../lib/types.js';
5
6
  import { RetrieveContextInputSchema } from '../schemas/inputs.js';
6
7
  import { RetrieveContextResultSchema } from '../schemas/outputs.js';
8
+ import { createProgressReporter, notifyProgress, progressWithMessage, } from './progress.js';
9
+ import { getToolResultPayload, isOkStructuredToolResult } from './result.js';
7
10
  const RETRIEVE_CONTEXT_LIMIT = 200;
8
11
  const ESTIMATED_CHARS_PER_TOKEN = 4;
9
- function getOrderBy(strategy) {
10
- if (strategy === 'importance') {
11
- return 'm.importance DESC, memories_fts.rank';
12
- }
13
- if (strategy === 'recency') {
14
- return 'm.created_at DESC, memories_fts.rank';
15
- }
16
- return 'memories_fts.rank';
12
+ const RETRIEVE_CONTEXT_PROGRESS_MILESTONE = 25;
13
+ const ORDER_BY_MAP = {
14
+ importance: 'm.importance DESC, memories_fts.rank',
15
+ recency: 'm.created_at DESC, memories_fts.rank',
16
+ relevance: 'memories_fts.rank',
17
+ };
18
+ function countPayloadArrayItems(payload, key) {
19
+ const value = payload[key];
20
+ return Array.isArray(value) ? value.length : 0;
17
21
  }
18
22
  function estimateTokens(content) {
19
23
  return Math.ceil(content.length / ESTIMATED_CHARS_PER_TOKEN);
@@ -21,13 +25,48 @@ function estimateTokens(content) {
21
25
  function loadContextRows(db, query, orderBy) {
22
26
  const ftsQuery = sanitizeFtsQuery(query);
23
27
  return db
24
- .prepare(`SELECT m.*, memories_fts.rank AS rank FROM memories m
28
+ .prepareOnce(`SELECT m.*, memories_fts.rank AS rank FROM memories m
25
29
  JOIN memories_fts ON memories_fts.rowid = m.rowid
26
30
  WHERE memories_fts MATCH ?
27
31
  ORDER BY ${orderBy}
28
32
  LIMIT ${RETRIEVE_CONTEXT_LIMIT + 1}`)
29
33
  .all(ftsQuery);
30
34
  }
35
+ function reportSelectionProgress(onProgress, current, total, force = false) {
36
+ if (!onProgress || current === 0) {
37
+ return;
38
+ }
39
+ if (!force && current % RETRIEVE_CONTEXT_PROGRESS_MILESTONE !== 0) {
40
+ return;
41
+ }
42
+ onProgress({ current, total });
43
+ }
44
+ function formatCompletionMessage(query, result) {
45
+ const failedMessage = `⊙ retrieve_context: ${query} • failed`;
46
+ if (result.isError) {
47
+ return failedMessage;
48
+ }
49
+ if (!isOkStructuredToolResult(result)) {
50
+ return failedMessage;
51
+ }
52
+ const payload = getToolResultPayload(result);
53
+ if (!payload) {
54
+ return `⊙ retrieve_context: ${query} • completed`;
55
+ }
56
+ const memoriesCount = countPayloadArrayItems(payload, 'memories');
57
+ const estimatedTokens = 'estimated_tokens' in payload &&
58
+ typeof payload.estimated_tokens === 'number'
59
+ ? payload.estimated_tokens
60
+ : 0;
61
+ const truncated = 'truncated' in payload && payload.truncated === true ? ' [truncated]' : '';
62
+ return `⊙ retrieve_context: ${query} • ${memoriesCount} memories, ${estimatedTokens} tokens${truncated}`;
63
+ }
64
+ function throwIfAborted(signal) {
65
+ if (!signal?.aborted) {
66
+ return;
67
+ }
68
+ throw new McpError(ErrorCode.RequestTimeout, 'Request cancelled');
69
+ }
31
70
  export function registerRetrieveContext(server, db) {
32
71
  server.registerTool('retrieve_context', {
33
72
  title: 'Retrieve Context',
@@ -35,22 +74,41 @@ export function registerRetrieveContext(server, db) {
35
74
  inputSchema: RetrieveContextInputSchema,
36
75
  outputSchema: RetrieveContextResultSchema,
37
76
  annotations: { readOnlyHint: true, openWorldHint: false },
38
- }, (params) => {
77
+ }, async (params, extra) => {
78
+ const { query, strategy } = params;
79
+ const tokenBudget = params.token_budget;
80
+ const contextLabel = `⊙ retrieve_context: ${query} [${strategy}]`;
81
+ let completionCurrent = 1;
82
+ await notifyProgress(extra, {
83
+ current: 0,
84
+ message: `${contextLabel} [budget ${tokenBudget}]`,
85
+ });
86
+ const loopProgress = progressWithMessage(createProgressReporter(extra), ({ current, total }) => `${contextLabel} [scan ${current}/${Math.max((total ?? current) - 1, current)}]`);
87
+ let result;
88
+ let thrownError;
39
89
  try {
40
- const { query, strategy } = params;
41
- const tokenBudget = params.token_budget;
42
- const orderBy = getOrderBy(strategy);
90
+ throwIfAborted(extra.signal);
91
+ const orderBy = ORDER_BY_MAP[strategy];
43
92
  const rows = loadContextRows(db, query, orderBy);
44
93
  const rowCapExceeded = rows.length > RETRIEVE_CONTEXT_LIMIT;
45
- const candidateRows = rowCapExceeded
46
- ? rows.slice(0, RETRIEVE_CONTEXT_LIMIT)
47
- : rows;
94
+ const candidateCount = rowCapExceeded
95
+ ? RETRIEVE_CONTEXT_LIMIT
96
+ : rows.length;
97
+ completionCurrent = candidateCount + 1;
48
98
  let estimatedTokens = 0;
49
99
  let truncated = rowCapExceeded;
50
100
  const selected = [];
51
- for (const row of candidateRows) {
101
+ let scanned = 0;
102
+ for (let i = 0; i < candidateCount; i += 1) {
103
+ throwIfAborted(extra.signal);
104
+ const row = rows[i];
105
+ if (row === undefined) {
106
+ break;
107
+ }
52
108
  const mem = parseMemoryRow(row);
53
109
  const tokens = estimateTokens(mem.content);
110
+ scanned += 1;
111
+ reportSelectionProgress(loopProgress, scanned, completionCurrent, false);
54
112
  if (estimatedTokens + tokens > tokenBudget) {
55
113
  truncated = true;
56
114
  break;
@@ -58,18 +116,33 @@ export function registerRetrieveContext(server, db) {
58
116
  estimatedTokens += tokens;
59
117
  selected.push(mem);
60
118
  }
61
- return createToolResponse({
62
- ok: true,
63
- result: {
64
- memories: selected,
65
- estimated_tokens: estimatedTokens,
66
- truncated,
67
- },
119
+ reportSelectionProgress(loopProgress, scanned, completionCurrent, true);
120
+ result = createToolResponse({
121
+ memories: selected,
122
+ estimated_tokens: estimatedTokens,
123
+ truncated,
68
124
  });
69
125
  }
70
126
  catch (err) {
71
- return createErrorResponse(E_UNKNOWN, getErrorMessage(err));
127
+ if (err instanceof McpError) {
128
+ thrownError = err;
129
+ }
130
+ else {
131
+ rethrowMcpError(err);
132
+ result = createErrorResponse(E_UNKNOWN, getErrorMessage(err));
133
+ }
134
+ }
135
+ await loopProgress.flush();
136
+ const completionResult = result ?? createErrorResponse(E_UNKNOWN, getErrorMessage(thrownError));
137
+ await notifyProgress(extra, {
138
+ current: completionCurrent,
139
+ total: completionCurrent,
140
+ message: formatCompletionMessage(query, completionResult),
141
+ });
142
+ if (thrownError) {
143
+ throw thrownError;
72
144
  }
145
+ return completionResult;
73
146
  });
74
147
  }
75
148
  //# sourceMappingURL=retrieve-context.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"retrieve-context.js","sourceRoot":"","sources":["../../src/tools/retrieve-context.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAC9D,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAEjD,OAAO,EAAE,0BAA0B,EAAE,MAAM,sBAAsB,CAAC;AAClE,OAAO,EAAE,2BAA2B,EAAE,MAAM,uBAAuB,CAAC;AAKpE,MAAM,sBAAsB,GAAG,GAAG,CAAC;AACnC,MAAM,yBAAyB,GAAG,CAAC,CAAC;AAEpC,SAAS,UAAU,CAAC,QAAyB;IAC3C,IAAI,QAAQ,KAAK,YAAY,EAAE,CAAC;QAC9B,OAAO,sCAAsC,CAAC;IAChD,CAAC;IACD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,OAAO,sCAAsC,CAAC;IAChD,CAAC;IACD,OAAO,mBAAmB,CAAC;AAC7B,CAAC;AAED,SAAS,cAAc,CAAC,OAAe;IACrC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,yBAAyB,CAAC,CAAC;AAC/D,CAAC;AAED,SAAS,eAAe,CACtB,EAAW,EACX,KAAa,EACb,OAAe;IAEf,MAAM,QAAQ,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;IACzC,OAAO,EAAE;SACN,OAAO,CACN;;;kBAGY,OAAO;eACV,sBAAsB,GAAG,CAAC,EAAE,CACtC;SACA,GAAG,CAAC,QAAQ,CAAC,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,MAAiB,EAAE,EAAW;IACpE,MAAM,CAAC,YAAY,CACjB,kBAAkB,EAClB;QACE,KAAK,EAAE,kBAAkB;QACzB,WAAW,EACT,qLAAqL;QACvL,WAAW,EAAE,0BAA0B;QACvC,YAAY,EAAE,2BAA2B;QACzC,WAAW,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE;KAC1D,EACD,CAAC,MAA4B,EAAE,EAAE;QAC/B,IAAI,CAAC;YACH,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;YACnC,MAAM,WAAW,GAAG,MAAM,CAAC,YAAY,CAAC;YACxC,MAAM,OAAO,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC;YACrC,MAAM,IAAI,GAAG,eAAe,CAAC,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;YACjD,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,GAAG,sBAAsB,CAAC;YAC5D,MAAM,aAAa,GAAG,cAAc;gBAClC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,sBAAsB,CAAC;gBACvC,CAAC,CAAC,IAAI,CAAC;YAET,IAAI,eAAe,GAAG,CAAC,CAAC;YACxB,IAAI,SAAS,GAAG,cAAc,CAAC;YAC/B,MAAM,QAAQ,GAAa,EAAE,CAAC;YAE9B,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;gBAChC,MAAM,GAAG,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;gBAChC,MAAM,MAAM,GAAG,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC3C,IAAI,eAAe,GAAG,MAAM,GAAG,WAAW,EAAE,CAAC;oBAC3C,SAAS,GAAG,IAAI,CAAC;oBACjB,MAAM;gBACR,CAAC;gBACD,eAAe,IAAI,MAAM,CAAC;gBAC1B,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrB,CAAC;YAED,OAAO,kBAAkB,CAAC;gBACxB,EAAE,EAAE,IAAI;gBACR,MAAM,EAAE;oBACN,QAAQ,EAAE,QAAQ;oBAClB,gBAAgB,EAAE,eAAe;oBACjC,SAAS;iBACV;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"}
1
+ {"version":3,"file":"retrieve-context.js","sourceRoot":"","sources":["../../src/tools/retrieve-context.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,oCAAoC,CAAC;AAKzE,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAC/E,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAEjD,OAAO,EAAE,0BAA0B,EAAE,MAAM,sBAAsB,CAAC;AAClE,OAAO,EAAE,2BAA2B,EAAE,MAAM,uBAAuB,CAAC;AACpE,OAAO,EACL,sBAAsB,EACtB,cAAc,EACd,mBAAmB,GACpB,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,oBAAoB,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AAK7E,MAAM,sBAAsB,GAAG,GAAG,CAAC;AACnC,MAAM,yBAAyB,GAAG,CAAC,CAAC;AACpC,MAAM,mCAAmC,GAAG,EAAE,CAAC;AAE/C,MAAM,YAAY,GAAG;IACnB,UAAU,EAAE,sCAAsC;IAClD,OAAO,EAAE,sCAAsC;IAC/C,SAAS,EAAE,mBAAmB;CACoB,CAAC;AAErD,SAAS,sBAAsB,CAC7B,OAAgC,EAChC,GAAW;IAEX,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;IAC3B,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;AACjD,CAAC;AAED,SAAS,cAAc,CAAC,OAAe;IACrC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,yBAAyB,CAAC,CAAC;AAC/D,CAAC;AAED,SAAS,eAAe,CACtB,EAAW,EACX,KAAa,EACb,OAAe;IAEf,MAAM,QAAQ,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;IACzC,OAAO,EAAE;SACN,WAAW,CACV;;;kBAGY,OAAO;eACV,sBAAsB,GAAG,CAAC,EAAE,CACtC;SACA,GAAG,CAAC,QAAQ,CAAC,CAAC;AACnB,CAAC;AAED,SAAS,uBAAuB,CAC9B,UAEa,EACb,OAAe,EACf,KAAa,EACb,KAAK,GAAG,KAAK;IAEb,IAAI,CAAC,UAAU,IAAI,OAAO,KAAK,CAAC,EAAE,CAAC;QACjC,OAAO;IACT,CAAC;IACD,IAAI,CAAC,KAAK,IAAI,OAAO,GAAG,mCAAmC,KAAK,CAAC,EAAE,CAAC;QAClE,OAAO;IACT,CAAC;IACD,UAAU,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,uBAAuB,CAC9B,KAAa,EACb,MAAsB;IAEtB,MAAM,aAAa,GAAG,uBAAuB,KAAK,WAAW,CAAC;IAC9D,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,OAAO,aAAa,CAAC;IACvB,CAAC;IACD,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,EAAE,CAAC;QACtC,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,MAAM,OAAO,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;IAC7C,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,uBAAuB,KAAK,cAAc,CAAC;IACpD,CAAC;IAED,MAAM,aAAa,GAAG,sBAAsB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IAClE,MAAM,eAAe,GACnB,kBAAkB,IAAI,OAAO;QAC7B,OAAO,OAAO,CAAC,gBAAgB,KAAK,QAAQ;QAC1C,CAAC,CAAC,OAAO,CAAC,gBAAgB;QAC1B,CAAC,CAAC,CAAC,CAAC;IACR,MAAM,SAAS,GACb,WAAW,IAAI,OAAO,IAAI,OAAO,CAAC,SAAS,KAAK,IAAI,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC;IAE7E,OAAO,uBAAuB,KAAK,MAAM,aAAa,cAAc,eAAe,UAAU,SAAS,EAAE,CAAC;AAC3G,CAAC;AAED,SAAS,cAAc,CAAC,MAAoB;IAC1C,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;QACrB,OAAO;IACT,CAAC;IAED,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,cAAc,EAAE,mBAAmB,CAAC,CAAC;AACpE,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,MAAiB,EAAE,EAAW;IACpE,MAAM,CAAC,YAAY,CACjB,kBAAkB,EAClB;QACE,KAAK,EAAE,kBAAkB;QACzB,WAAW,EACT,qLAAqL;QACvL,WAAW,EAAE,0BAA0B;QACvC,YAAY,EAAE,2BAA2B;QACzC,WAAW,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE;KAC1D,EACD,KAAK,EAAE,MAA4B,EAAE,KAAK,EAAE,EAAE;QAC5C,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;QACnC,MAAM,WAAW,GAAG,MAAM,CAAC,YAAY,CAAC;QACxC,MAAM,YAAY,GAAG,uBAAuB,KAAK,KAAK,QAAQ,GAAG,CAAC;QAClE,IAAI,iBAAiB,GAAG,CAAC,CAAC;QAE1B,MAAM,cAAc,CAAC,KAAK,EAAE;YAC1B,OAAO,EAAE,CAAC;YACV,OAAO,EAAE,GAAG,YAAY,YAAY,WAAW,GAAG;SACnD,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,mBAAmB,CACtC,sBAAsB,CAAC,KAAK,CAAC,EAC7B,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CACrB,GAAG,YAAY,UAAU,OAAO,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,IAAI,OAAO,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,GAAG,CACnF,CAAC;QAEF,IAAI,MAAkC,CAAC;QACvC,IAAI,WAAiC,CAAC;QACtC,IAAI,CAAC;YACH,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC7B,MAAM,OAAO,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;YACvC,MAAM,IAAI,GAAG,eAAe,CAAC,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;YACjD,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,GAAG,sBAAsB,CAAC;YAC5D,MAAM,cAAc,GAAG,cAAc;gBACnC,CAAC,CAAC,sBAAsB;gBACxB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;YAChB,iBAAiB,GAAG,cAAc,GAAG,CAAC,CAAC;YAEvC,IAAI,eAAe,GAAG,CAAC,CAAC;YACxB,IAAI,SAAS,GAAG,cAAc,CAAC;YAC/B,MAAM,QAAQ,GAAa,EAAE,CAAC;YAC9B,IAAI,OAAO,GAAG,CAAC,CAAC;YAEhB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,cAAc,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC3C,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;gBAC7B,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;gBACpB,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;oBACtB,MAAM;gBACR,CAAC;gBACD,MAAM,GAAG,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;gBAChC,MAAM,MAAM,GAAG,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAC3C,OAAO,IAAI,CAAC,CAAC;gBACb,uBAAuB,CACrB,YAAY,EACZ,OAAO,EACP,iBAAiB,EACjB,KAAK,CACN,CAAC;gBACF,IAAI,eAAe,GAAG,MAAM,GAAG,WAAW,EAAE,CAAC;oBAC3C,SAAS,GAAG,IAAI,CAAC;oBACjB,MAAM;gBACR,CAAC;gBACD,eAAe,IAAI,MAAM,CAAC;gBAC1B,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrB,CAAC;YAED,uBAAuB,CAAC,YAAY,EAAE,OAAO,EAAE,iBAAiB,EAAE,IAAI,CAAC,CAAC;YAExE,MAAM,GAAG,kBAAkB,CAAC;gBAC1B,QAAQ,EAAE,QAAQ;gBAClB,gBAAgB,EAAE,eAAe;gBACjC,SAAS;aACV,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,GAAG,YAAY,QAAQ,EAAE,CAAC;gBAC5B,WAAW,GAAG,GAAG,CAAC;YACpB,CAAC;iBAAM,CAAC;gBACN,eAAe,CAAC,GAAG,CAAC,CAAC;gBACrB,MAAM,GAAG,mBAAmB,CAAC,SAAS,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC;YAChE,CAAC;QACH,CAAC;QAED,MAAM,YAAY,CAAC,KAAK,EAAE,CAAC;QAE3B,MAAM,gBAAgB,GACpB,MAAM,IAAI,mBAAmB,CAAC,SAAS,EAAE,eAAe,CAAC,WAAW,CAAC,CAAC,CAAC;QAEzE,MAAM,cAAc,CAAC,KAAK,EAAE;YAC1B,OAAO,EAAE,iBAAiB;YAC1B,KAAK,EAAE,iBAAiB;YACxB,OAAO,EAAE,uBAAuB,CAAC,KAAK,EAAE,gBAAgB,CAAC;SAC1D,CAAC,CAAC;QAEH,IAAI,WAAW,EAAE,CAAC;YAChB,MAAM,WAAW,CAAC;QACpB,CAAC;QAED,OAAO,gBAAgB,CAAC;IAC1B,CAAC,CACF,CAAC;AACJ,CAAC"}
@@ -1 +1 @@
1
- {"version":3,"file":"search-memories.d.ts","sourceRoot":"","sources":["../../src/tools/search-memories.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAuD9C,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CAwC3E"}
1
+ {"version":3,"file":"search-memories.d.ts","sourceRoot":"","sources":["../../src/tools/search-memories.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAqB9C,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CAuD3E"}
@@ -1,62 +1,50 @@
1
- import { E_UNKNOWN, getErrorMessage } from '../lib/errors.js';
2
- import { decodeCursor, encodeCursor, splitPage } from '../lib/pagination.js';
3
- import { buildAndWhereClause, buildFilterClauses, sanitizeFtsQuery, } from '../lib/search.js';
1
+ import { E_UNKNOWN, getErrorMessage, rethrowMcpError } from '../lib/errors.js';
2
+ import { splitPage } from '../lib/pagination.js';
3
+ import { buildSearchCursorScope, decodeSearchCursor, encodeSearchCursor, } from '../lib/search-cursor.js';
4
+ import { loadRankedSearchRows, toMemoryFilters } from '../lib/search.js';
4
5
  import { createErrorResponse, createToolResponse, } from '../lib/tool-response.js';
5
6
  import { parseMemoryRow } from '../lib/types.js';
6
7
  import { SearchMemoriesInputSchema } from '../schemas/inputs.js';
7
8
  import { SearchResultSchema } from '../schemas/outputs.js';
8
- function toMemoryFilters(params) {
9
- const filters = {};
10
- if (params.min_importance != null) {
11
- filters.min_importance = params.min_importance;
12
- }
13
- if (params.max_importance != null) {
14
- filters.max_importance = params.max_importance;
15
- }
16
- if (params.memory_type != null) {
17
- filters.memory_type = params.memory_type;
18
- }
19
- return filters;
20
- }
21
- function loadSearchRows(db, query, limit, offset, filters) {
22
- const ftsQuery = sanitizeFtsQuery(query);
23
- const filter = buildFilterClauses(filters);
24
- const whereExtra = buildAndWhereClause(filter.clauses);
25
- return db
26
- .prepare(`SELECT m.*, memories_fts.rank AS rank FROM memories m
27
- JOIN memories_fts ON memories_fts.rowid = m.rowid
28
- WHERE memories_fts MATCH ?${whereExtra}
29
- ORDER BY memories_fts.rank
30
- LIMIT ? OFFSET ?`)
31
- .all(ftsQuery, ...filter.params, limit + 1, offset);
32
- }
9
+ import { wrapToolHandler } from './progress.js';
33
10
  export function registerSearchMemories(server, db) {
34
11
  server.registerTool('search_memories', {
35
12
  title: 'Search Memories',
36
- description: 'Full-text search over memory content and tags using FTS5. Returns ranked results with pagination support via cursor.',
13
+ description: 'Full-text search over memory content and tags using FTS5. Returns ranked results with pagination support via cursor. Query terms are matched individually; FTS5 phrase operators and negation are not supported.',
37
14
  inputSchema: SearchMemoriesInputSchema,
38
15
  outputSchema: SearchResultSchema,
39
16
  annotations: { readOnlyHint: true, openWorldHint: false },
40
- }, (params) => {
17
+ }, wrapToolHandler((params) => {
41
18
  try {
42
19
  const { limit, cursor } = params;
43
- const offset = cursor ? decodeCursor(cursor) : 0;
44
- const rows = loadSearchRows(db, params.query, limit, offset, toMemoryFilters(params));
20
+ const filters = toMemoryFilters(params);
21
+ const scope = buildSearchCursorScope(params.query, filters);
22
+ const decodedCursor = cursor
23
+ ? decodeSearchCursor(cursor, scope)
24
+ : undefined;
25
+ const rows = loadRankedSearchRows(db, params.query, limit, decodedCursor, filters);
45
26
  const { page: pageRows, hasMore } = splitPage(rows, limit);
46
27
  const memories = pageRows.map(parseMemoryRow);
47
- const nextCursor = hasMore ? encodeCursor(offset + limit) : undefined;
28
+ let nextCursor;
29
+ if (hasMore && pageRows.length > 0) {
30
+ const lastRow = pageRows[pageRows.length - 1];
31
+ if (lastRow !== undefined) {
32
+ const rank = lastRow.rank ?? 0;
33
+ nextCursor = encodeSearchCursor(scope, rank, lastRow.hash);
34
+ }
35
+ }
48
36
  return createToolResponse({
49
- ok: true,
50
- result: {
51
- memories,
52
- total_returned: memories.length,
53
- ...(nextCursor ? { nextCursor } : {}),
54
- },
37
+ memories,
38
+ total_returned: memories.length,
39
+ ...(nextCursor ? { nextCursor } : {}),
55
40
  });
56
41
  }
57
42
  catch (err) {
43
+ rethrowMcpError(err);
58
44
  return createErrorResponse(E_UNKNOWN, getErrorMessage(err));
59
45
  }
60
- });
46
+ }, {
47
+ progressMessage: (params) => `⊙ search_memories: ${params.query} [limit ${params.limit}]`,
48
+ }));
61
49
  }
62
50
  //# sourceMappingURL=search-memories.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"search-memories.js","sourceRoot":"","sources":["../../src/tools/search-memories.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAC9D,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AAC7E,OAAO,EACL,mBAAmB,EACnB,kBAAkB,EAClB,gBAAgB,GACjB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAEjD,OAAO,EAAE,yBAAyB,EAAE,MAAM,sBAAsB,CAAC;AACjE,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAI3D,SAAS,eAAe,CAAC,MAAmB;IAC1C,MAAM,OAAO,GAAkB,EAAE,CAAC;IAClC,IAAI,MAAM,CAAC,cAAc,IAAI,IAAI,EAAE,CAAC;QAClC,OAAO,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC;IACjD,CAAC;IACD,IAAI,MAAM,CAAC,cAAc,IAAI,IAAI,EAAE,CAAC;QAClC,OAAO,CAAC,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC;IACjD,CAAC;IACD,IAAI,MAAM,CAAC,WAAW,IAAI,IAAI,EAAE,CAAC;QAC/B,OAAO,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;IAC3C,CAAC;IACD,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,cAAc,CACrB,EAAW,EACX,KAAa,EACb,KAAa,EACb,MAAc,EACd,OAAsB;IAEtB,MAAM,QAAQ,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAC;IACzC,MAAM,MAAM,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;IAC3C,MAAM,UAAU,GAAG,mBAAmB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACvD,OAAO,EAAE;SACN,OAAO,CACN;;mCAE6B,UAAU;;wBAErB,CACnB;SACA,GAAG,CAAC,QAAQ,EAAE,GAAG,MAAM,CAAC,MAAM,EAAE,KAAK,GAAG,CAAC,EAAE,MAAM,CAAC,CAAC;AACxD,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,MAAiB,EAAE,EAAW;IACnE,MAAM,CAAC,YAAY,CACjB,iBAAiB,EACjB;QACE,KAAK,EAAE,iBAAiB;QACxB,WAAW,EACT,sHAAsH;QACxH,WAAW,EAAE,yBAAyB;QACtC,YAAY,EAAE,kBAAkB;QAChC,WAAW,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE;KAC1D,EACD,CAAC,MAAmB,EAAE,EAAE;QACtB,IAAI,CAAC;YACH,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;YACjC,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACjD,MAAM,IAAI,GAAG,cAAc,CACzB,EAAE,EACF,MAAM,CAAC,KAAK,EACZ,KAAK,EACL,MAAM,EACN,eAAe,CAAC,MAAM,CAAC,CACxB,CAAC;YACF,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,SAAS,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAE3D,MAAM,QAAQ,GAAa,QAAQ,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YACxD,MAAM,UAAU,GAAG,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YAEtE,OAAO,kBAAkB,CAAC;gBACxB,EAAE,EAAE,IAAI;gBACR,MAAM,EAAE;oBACN,QAAQ;oBACR,cAAc,EAAE,QAAQ,CAAC,MAAM;oBAC/B,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACtC;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"}
1
+ {"version":3,"file":"search-memories.js","sourceRoot":"","sources":["../../src/tools/search-memories.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAC/E,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjD,OAAO,EACL,sBAAsB,EACtB,kBAAkB,EAClB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,oBAAoB,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACzE,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAEjD,OAAO,EAAE,yBAAyB,EAAE,MAAM,sBAAsB,CAAC;AACjE,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAIhD,MAAM,UAAU,sBAAsB,CAAC,MAAiB,EAAE,EAAW;IACnE,MAAM,CAAC,YAAY,CACjB,iBAAiB,EACjB;QACE,KAAK,EAAE,iBAAiB;QACxB,WAAW,EACT,kNAAkN;QACpN,WAAW,EAAE,yBAAyB;QACtC,YAAY,EAAE,kBAAkB;QAChC,WAAW,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE;KAC1D,EACD,eAAe,CACb,CAAC,MAAmB,EAAE,EAAE;QACtB,IAAI,CAAC;YACH,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;YACjC,MAAM,OAAO,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;YACxC,MAAM,KAAK,GAAG,sBAAsB,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YAC5D,MAAM,aAAa,GAAG,MAAM;gBAC1B,CAAC,CAAC,kBAAkB,CAAC,MAAM,EAAE,KAAK,CAAC;gBACnC,CAAC,CAAC,SAAS,CAAC;YACd,MAAM,IAAI,GAAG,oBAAoB,CAC/B,EAAE,EACF,MAAM,CAAC,KAAK,EACZ,KAAK,EACL,aAAa,EACb,OAAO,CACR,CAAC;YACF,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,SAAS,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YAE3D,MAAM,QAAQ,GAAa,QAAQ,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YACxD,IAAI,UAA8B,CAAC;YACnC,IAAI,OAAO,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACnC,MAAM,OAAO,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;gBAC9C,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;oBAC1B,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,CAAC,CAAC;oBAC/B,UAAU,GAAG,kBAAkB,CAAC,KAAK,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;gBAC7D,CAAC;YACH,CAAC;YAED,OAAO,kBAAkB,CAAC;gBACxB,QAAQ;gBACR,cAAc,EAAE,QAAQ,CAAC,MAAM;gBAC/B,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACtC,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,sBAAsB,MAAM,CAAC,KAAK,WAAW,MAAM,CAAC,KAAK,GAAG;KAC/D,CACF,CACF,CAAC;AACJ,CAAC"}
@@ -1 +1 @@
1
- {"version":3,"file":"store-memories.d.ts","sourceRoot":"","sources":["../../src/tools/store-memories.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AA6B9C,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CAqD1E"}
1
+ {"version":3,"file":"store-memories.d.ts","sourceRoot":"","sources":["../../src/tools/store-memories.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AA+B9C,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CAmE1E"}
@@ -1,14 +1,20 @@
1
- import { E_UNKNOWN, getErrorMessage } from '../lib/errors.js';
1
+ import { E_UNKNOWN, getErrorMessage, rethrowMcpError } from '../lib/errors.js';
2
2
  import { computeMemoryHash } from '../lib/hash.js';
3
+ import { logToolEvent, notifyMemoryResourceUpdated } from '../lib/mcp-utils.js';
3
4
  import { createErrorResponse, createToolResponse, } from '../lib/tool-response.js';
4
5
  import { StoreMemoriesInputSchema } from '../schemas/inputs.js';
5
6
  import { BatchResultSchema } from '../schemas/outputs.js';
6
- import { logToolEvent, normalizeMemoryType, nowIso, withImmediateTransaction, } from './helpers.js';
7
+ import { wrapToolHandler } from './progress.js';
7
8
  const INSERT_MEMORY_SQL = `INSERT OR IGNORE INTO memories (hash, content, tags, memory_type, importance, created_at, updated_at)
8
9
  VALUES (?, ?, ?, ?, ?, ?, ?)`;
9
- function summarizeBatch(items) {
10
- const succeeded = items.filter((item) => item.ok).length;
11
- return { succeeded, failed: items.length - succeeded };
10
+ function countCreated(items) {
11
+ return items.reduce((count, item) => (item.created ? count + 1 : count), 0);
12
+ }
13
+ async function notifyCreatedResources(server, items) {
14
+ const notifications = items
15
+ .filter((item) => item.created)
16
+ .map((item) => notifyMemoryResourceUpdated(server, item.hash));
17
+ await Promise.allSettled(notifications);
12
18
  }
13
19
  export function registerStoreMemories(server, db) {
14
20
  server.registerTool('store_memories', {
@@ -17,36 +23,42 @@ export function registerStoreMemories(server, db) {
17
23
  inputSchema: StoreMemoriesInputSchema,
18
24
  outputSchema: BatchResultSchema,
19
25
  annotations: { idempotentHint: true, openWorldHint: false },
20
- }, async (params) => {
26
+ }, wrapToolHandler(async (params) => {
21
27
  try {
22
- const now = nowIso();
23
- const results = withImmediateTransaction(db, () => {
28
+ const now = new Date().toISOString();
29
+ const results = db.transaction(() => {
24
30
  const items = [];
25
- const stmt = db.prepare(INSERT_MEMORY_SQL);
31
+ const stmt = db.prepareOnce(INSERT_MEMORY_SQL);
26
32
  for (const item of params.items) {
27
33
  const { importance, memory_type: rawMemoryType } = item;
28
- const memoryType = normalizeMemoryType(rawMemoryType);
34
+ const memoryType = rawMemoryType ?? 'general';
29
35
  const hash = computeMemoryHash(item.content, item.tags);
30
36
  const tagsJson = JSON.stringify(item.tags);
31
37
  const result = stmt.run(hash, item.content, tagsJson, memoryType, importance, now, now);
32
- items.push({ hash, ok: true, created: result.changes > 0 });
38
+ items.push({
39
+ hash,
40
+ ok: true,
41
+ created: result.changes > 0,
42
+ });
33
43
  }
34
44
  return items;
35
45
  });
36
- const created = results.filter((r) => r.created).length;
37
- const { succeeded, failed } = summarizeBatch(results);
46
+ const created = countCreated(results);
47
+ const succeeded = results.reduce((count, item) => (item.ok ? count + 1 : count), 0);
48
+ const failed = results.length - succeeded;
38
49
  await logToolEvent(server, 'store_memories', {
39
50
  total: results.length,
40
51
  created,
41
52
  });
42
- return createToolResponse({
43
- ok: true,
44
- result: { items: results, succeeded, failed },
45
- });
53
+ await notifyCreatedResources(server, results);
54
+ return createToolResponse({ items: results, succeeded, failed });
46
55
  }
47
56
  catch (err) {
57
+ rethrowMcpError(err);
48
58
  return createErrorResponse(E_UNKNOWN, getErrorMessage(err));
49
59
  }
50
- });
60
+ }, {
61
+ progressMessage: (params) => `⊕ store_memories: ${params.items.length} items [batch]`,
62
+ }));
51
63
  }
52
64
  //# sourceMappingURL=store-memories.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"store-memories.js","sourceRoot":"","sources":["../../src/tools/store-memories.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAC9D,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AAEjC,OAAO,EAAE,wBAAwB,EAAE,MAAM,sBAAsB,CAAC;AAChE,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EACL,YAAY,EACZ,mBAAmB,EACnB,MAAM,EACN,wBAAwB,GACzB,MAAM,cAAc,CAAC;AAGtB,MAAM,iBAAiB,GAAG;+BACK,CAAC;AAEhC,SAAS,cAAc,CAAC,KAAiC;IAIvD,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC;IACzD,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,GAAG,SAAS,EAAE,CAAC;AACzD,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,MAAiB,EAAE,EAAW;IAClE,MAAM,CAAC,YAAY,CACjB,gBAAgB,EAChB;QACE,KAAK,EAAE,wBAAwB;QAC/B,WAAW,EACT,2KAA2K;QAC7K,WAAW,EAAE,wBAAwB;QACrC,YAAY,EAAE,iBAAiB;QAC/B,WAAW,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE;KAC5D,EACD,KAAK,EAAE,MAA0B,EAAE,EAAE;QACnC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;YACrB,MAAM,OAAO,GAAG,wBAAwB,CAAC,EAAE,EAAE,GAAG,EAAE;gBAChD,MAAM,KAAK,GAAsB,EAAE,CAAC;gBACpC,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAU,iBAAiB,CAAC,CAAC;gBAEpD,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;oBAChC,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,aAAa,EAAE,GAAG,IAAI,CAAC;oBACxD,MAAM,UAAU,GAAG,mBAAmB,CAAC,aAAa,CAAC,CAAC;oBACtD,MAAM,IAAI,GAAG,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;oBACxD,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBAC3C,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CACrB,IAAI,EACJ,IAAI,CAAC,OAAO,EACZ,QAAQ,EACR,UAAU,EACV,UAAU,EACV,GAAG,EACH,GAAG,CACJ,CAAC;oBACF,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,CAAC,OAAO,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC9D,CAAC;gBACD,OAAO,KAAK,CAAC;YACf,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC;YACxD,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;YACtD,MAAM,YAAY,CAAC,MAAM,EAAE,gBAAgB,EAAE;gBAC3C,KAAK,EAAE,OAAO,CAAC,MAAM;gBACrB,OAAO;aACR,CAAC,CAAC;YAEH,OAAO,kBAAkB,CAAC;gBACxB,EAAE,EAAE,IAAI;gBACR,MAAM,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE;aAC9C,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"}
1
+ {"version":3,"file":"store-memories.js","sourceRoot":"","sources":["../../src/tools/store-memories.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAC/E,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,2BAA2B,EAAE,MAAM,qBAAqB,CAAC;AAChF,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AAEjC,OAAO,EAAE,wBAAwB,EAAE,MAAM,sBAAsB,CAAC;AAChE,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAGhD,MAAM,iBAAiB,GAAG;+BACK,CAAC;AAEhC,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,qBAAqB,CAAC,MAAiB,EAAE,EAAW;IAClE,MAAM,CAAC,YAAY,CACjB,gBAAgB,EAChB;QACE,KAAK,EAAE,wBAAwB;QAC/B,WAAW,EACT,2KAA2K;QAC7K,WAAW,EAAE,wBAAwB;QACrC,YAAY,EAAE,iBAAiB;QAC/B,WAAW,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE;KAC5D,EACD,eAAe,CACb,KAAK,EAAE,MAA0B,EAAE,EAAE;QACnC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YACrC,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;gBAExD,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;oBAChC,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,aAAa,EAAE,GAAG,IAAI,CAAC;oBACxD,MAAM,UAAU,GAAG,aAAa,IAAI,SAAS,CAAC;oBAC9C,MAAM,IAAI,GAAG,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;oBACxD,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBAC3C,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CACrB,IAAI,EACJ,IAAI,CAAC,OAAO,EACZ,QAAQ,EACR,UAAU,EACV,UAAU,EACV,GAAG,EACH,GAAG,CACJ,CAAC;oBACF,KAAK,CAAC,IAAI,CAAC;wBACT,IAAI;wBACJ,EAAE,EAAE,IAAI;wBACR,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,gBAAgB,EAAE;gBAC3C,KAAK,EAAE,OAAO,CAAC,MAAM;gBACrB,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,MAA0B,EAAE,EAAE,CAC9C,qBAAqB,MAAM,CAAC,KAAK,CAAC,MAAM,gBAAgB;KAC3D,CACF,CACF,CAAC;AACJ,CAAC"}
@@ -1 +1 @@
1
- {"version":3,"file":"store-memory.d.ts","sourceRoot":"","sources":["../../src/tools/store-memory.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAe9C,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CAwCxE"}
1
+ {"version":3,"file":"store-memory.d.ts","sourceRoot":"","sources":["../../src/tools/store-memory.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AA8C9C,wBAAgB,mBAAmB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CAmCxE"}
@@ -1,11 +1,29 @@
1
- import { E_UNKNOWN, getErrorMessage } from '../lib/errors.js';
1
+ import { E_UNKNOWN, getErrorMessage, rethrowMcpError } from '../lib/errors.js';
2
2
  import { computeMemoryHash } from '../lib/hash.js';
3
+ import { logToolEvent, notifyMemoryResourceUpdated } from '../lib/mcp-utils.js';
3
4
  import { createErrorResponse, createToolResponse, } from '../lib/tool-response.js';
4
5
  import { StoreMemoryInputSchema } from '../schemas/inputs.js';
5
6
  import { StoreResultSchema } from '../schemas/outputs.js';
6
- import { logToolEvent, normalizeMemoryType, nowIso } from './helpers.js';
7
+ import { wrapToolHandler } from './progress.js';
7
8
  const INSERT_MEMORY_SQL = `INSERT OR IGNORE INTO memories (hash, content, tags, memory_type, importance, created_at, updated_at)
8
9
  VALUES (?, ?, ?, ?, ?, ?, ?)`;
10
+ function toInsertParams(params, hash, memoryType, now) {
11
+ return [
12
+ hash,
13
+ params.content,
14
+ JSON.stringify(params.tags),
15
+ memoryType,
16
+ params.importance,
17
+ now,
18
+ now,
19
+ ];
20
+ }
21
+ function insertMemory(db, params, hash, memoryType, now) {
22
+ const insertResult = db
23
+ .prepareOnce(INSERT_MEMORY_SQL)
24
+ .run(...toInsertParams(params, hash, memoryType, now));
25
+ return insertResult.changes > 0;
26
+ }
9
27
  export function registerStoreMemory(server, db) {
10
28
  server.registerTool('store_memory', {
11
29
  title: 'Store Memory',
@@ -13,23 +31,24 @@ export function registerStoreMemory(server, db) {
13
31
  inputSchema: StoreMemoryInputSchema,
14
32
  outputSchema: StoreResultSchema,
15
33
  annotations: { idempotentHint: true, openWorldHint: false },
16
- }, async (params) => {
34
+ }, wrapToolHandler(async (params) => {
17
35
  try {
18
- const { importance } = params;
19
- const memoryType = normalizeMemoryType(params.memory_type);
36
+ const memoryType = params.memory_type ?? 'general';
20
37
  const hash = computeMemoryHash(params.content, params.tags);
21
- const now = nowIso();
22
- const tagsJson = JSON.stringify(params.tags);
23
- const insertResult = db
24
- .prepare(INSERT_MEMORY_SQL)
25
- .run(hash, params.content, tagsJson, memoryType, importance, now, now);
26
- const created = insertResult.changes > 0;
38
+ const now = new Date().toISOString();
39
+ const created = insertMemory(db, params, hash, memoryType, now);
27
40
  await logToolEvent(server, 'store', { hash, created });
28
- return createToolResponse({ ok: true, result: { hash, created } });
41
+ if (created) {
42
+ await notifyMemoryResourceUpdated(server, hash);
43
+ }
44
+ return createToolResponse({ hash, created });
29
45
  }
30
46
  catch (err) {
47
+ rethrowMcpError(err);
31
48
  return createErrorResponse(E_UNKNOWN, getErrorMessage(err));
32
49
  }
33
- });
50
+ }, {
51
+ progressMessage: (params) => `⊕ store_memory: ${params.tags.length} tags [single]`,
52
+ }));
34
53
  }
35
54
  //# sourceMappingURL=store-memory.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"store-memory.js","sourceRoot":"","sources":["../../src/tools/store-memory.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAC9D,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,YAAY,EAAE,mBAAmB,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAGzE,MAAM,iBAAiB,GAAG;+BACK,CAAC;AAEhC,MAAM,UAAU,mBAAmB,CAAC,MAAiB,EAAE,EAAW;IAChE,MAAM,CAAC,YAAY,CACjB,cAAc,EACd;QACE,KAAK,EAAE,cAAc;QACrB,WAAW,EACT,yLAAyL;QAC3L,WAAW,EAAE,sBAAsB;QACnC,YAAY,EAAE,iBAAiB;QAC/B,WAAW,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE;KAC5D,EACD,KAAK,EAAE,MAAkB,EAAE,EAAE;QAC3B,IAAI,CAAC;YACH,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,CAAC;YAC9B,MAAM,UAAU,GAAG,mBAAmB,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YAC3D,MAAM,IAAI,GAAG,iBAAiB,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;YAC5D,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;YACrB,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAE7C,MAAM,YAAY,GAAG,EAAE;iBACpB,OAAO,CAAC,iBAAiB,CAAC;iBAC1B,GAAG,CACF,IAAI,EACJ,MAAM,CAAC,OAAO,EACd,QAAQ,EACR,UAAU,EACV,UAAU,EACV,GAAG,EACH,GAAG,CACJ,CAAC;YAEJ,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO,GAAG,CAAC,CAAC;YACzC,MAAM,YAAY,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;YAEvD,OAAO,kBAAkB,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;QACrE,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"}
1
+ {"version":3,"file":"store-memory.js","sourceRoot":"","sources":["../../src/tools/store-memory.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAC/E,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,2BAA2B,EAAE,MAAM,qBAAqB,CAAC;AAChF,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAGhD,MAAM,iBAAiB,GAAG;+BACK,CAAC;AAEhC,SAAS,cAAc,CACrB,MAA2D,EAC3D,IAAY,EACZ,UAAkB,EAClB,GAAW;IAEX,OAAO;QACL,IAAI;QACJ,MAAM,CAAC,OAAO;QACd,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC;QAC3B,UAAU;QACV,MAAM,CAAC,UAAU;QACjB,GAAG;QACH,GAAG;KACJ,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CACnB,EAAW,EACX,MAAkB,EAClB,IAAY,EACZ,UAAkB,EAClB,GAAW;IAEX,MAAM,YAAY,GAAG,EAAE;SACpB,WAAW,CAAC,iBAAiB,CAAC;SAC9B,GAAG,CAAC,GAAG,cAAc,CAAC,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,CAAC,CAAC,CAAC;IACzD,OAAO,YAAY,CAAC,OAAO,GAAG,CAAC,CAAC;AAClC,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,MAAiB,EAAE,EAAW;IAChE,MAAM,CAAC,YAAY,CACjB,cAAc,EACd;QACE,KAAK,EAAE,cAAc;QACrB,WAAW,EACT,yLAAyL;QAC3L,WAAW,EAAE,sBAAsB;QACnC,YAAY,EAAE,iBAAiB;QAC/B,WAAW,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE;KAC5D,EACD,eAAe,CACb,KAAK,EAAE,MAAkB,EAAE,EAAE;QAC3B,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,MAAM,CAAC,WAAW,IAAI,SAAS,CAAC;YACnD,MAAM,IAAI,GAAG,iBAAiB,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;YAC5D,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YACrC,MAAM,OAAO,GAAG,YAAY,CAAC,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,CAAC,CAAC;YAChE,MAAM,YAAY,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;YACvD,IAAI,OAAO,EAAE,CAAC;gBACZ,MAAM,2BAA2B,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YAClD,CAAC;YAED,OAAO,kBAAkB,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAC/C,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,MAAkB,EAAE,EAAE,CACtC,mBAAmB,MAAM,CAAC,IAAI,CAAC,MAAM,gBAAgB;KACxD,CACF,CACF,CAAC;AACJ,CAAC"}
@@ -1 +1 @@
1
- {"version":3,"file":"update-memory.d.ts","sourceRoot":"","sources":["../../src/tools/update-memory.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AA0B9C,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CAoDzE"}
1
+ {"version":3,"file":"update-memory.d.ts","sourceRoot":"","sources":["../../src/tools/update-memory.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAIzE,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAsC9C,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,GAAG,IAAI,CAyEzE"}
@@ -1,15 +1,22 @@
1
- import { E_NOT_FOUND, E_UNKNOWN, getErrorMessage } from '../lib/errors.js';
1
+ import { E_CONFLICT, E_NOT_FOUND, E_UNKNOWN, getErrorMessage, rethrowMcpError, } from '../lib/errors.js';
2
2
  import { computeMemoryHash } from '../lib/hash.js';
3
+ import { logToolEvent, notifyMemoryResourceUpdated } from '../lib/mcp-utils.js';
3
4
  import { createErrorResponse, createToolResponse, } from '../lib/tool-response.js';
4
5
  import { parseTags } from '../lib/types.js';
5
6
  import { UpdateMemoryInputSchema } from '../schemas/inputs.js';
6
7
  import { UpdateResultSchema } from '../schemas/outputs.js';
7
- import { getMemoryRow, logToolEvent, notifyMemoryResourceUpdated, nowIso, withImmediateTransaction, } from './helpers.js';
8
+ import { wrapToolHandler } from './progress.js';
8
9
  const UPDATE_MEMORY_SQL = `UPDATE memories
9
10
  SET hash = ?, content = ?, tags = ?, updated_at = ?
10
11
  WHERE hash = ?`;
11
- function formatMemoryNotFound(hash) {
12
- return `Memory not found: ${hash}`;
12
+ const SELECT_MEMORY_BY_HASH_SQL = 'SELECT * FROM memories WHERE hash = ?';
13
+ const SELECT_MEMORY_HASH_SQL = 'SELECT hash FROM memories WHERE hash = ?';
14
+ async function notifyUpdatedMemoryResources(server, oldHash, newHash) {
15
+ const notifications = [notifyMemoryResourceUpdated(server, oldHash)];
16
+ if (newHash !== oldHash) {
17
+ notifications.push(notifyMemoryResourceUpdated(server, newHash));
18
+ }
19
+ await Promise.allSettled(notifications);
13
20
  }
14
21
  export function registerUpdateMemory(server, db) {
15
22
  server.registerTool('update_memory', {
@@ -18,32 +25,44 @@ export function registerUpdateMemory(server, db) {
18
25
  inputSchema: UpdateMemoryInputSchema,
19
26
  outputSchema: UpdateResultSchema,
20
27
  annotations: { destructiveHint: true, openWorldHint: false },
21
- }, async (params) => {
28
+ }, wrapToolHandler(async (params) => {
22
29
  try {
23
- const existing = getMemoryRow(db, params.hash);
30
+ const existing = db
31
+ .prepareOnce(SELECT_MEMORY_BY_HASH_SQL)
32
+ .get(params.hash);
24
33
  if (!existing) {
25
- return createErrorResponse(E_NOT_FOUND, formatMemoryNotFound(params.hash));
34
+ return createErrorResponse(E_NOT_FOUND, `Memory not found: ${params.hash}`);
26
35
  }
27
- const existingTags = parseTags(existing.tags);
28
- const newTags = params.tags ?? existingTags;
36
+ const newTags = params.tags ?? parseTags(existing.tags);
29
37
  const newHash = computeMemoryHash(params.content, newTags);
30
- const now = nowIso();
31
- withImmediateTransaction(db, () => {
32
- db.prepare(UPDATE_MEMORY_SQL).run(newHash, params.content, JSON.stringify(newTags), now, params.hash);
38
+ if (newHash !== params.hash) {
39
+ const collision = db
40
+ .prepareOnce(SELECT_MEMORY_HASH_SQL)
41
+ .get(newHash);
42
+ if (collision) {
43
+ return createErrorResponse(E_CONFLICT, `Memory already exists for target content/tags: ${newHash}`);
44
+ }
45
+ }
46
+ const now = new Date().toISOString();
47
+ db.transaction(() => {
48
+ db.prepareOnce(UPDATE_MEMORY_SQL).run(newHash, params.content, JSON.stringify(newTags), now, params.hash);
33
49
  });
34
50
  await logToolEvent(server, 'update', {
35
51
  oldHash: params.hash,
36
52
  newHash,
37
53
  });
38
- await notifyMemoryResourceUpdated(server, params.hash);
54
+ await notifyUpdatedMemoryResources(server, params.hash, newHash);
39
55
  return createToolResponse({
40
- ok: true,
41
- result: { old_hash: params.hash, new_hash: newHash },
56
+ old_hash: params.hash,
57
+ new_hash: newHash,
42
58
  });
43
59
  }
44
60
  catch (err) {
61
+ rethrowMcpError(err);
45
62
  return createErrorResponse(E_UNKNOWN, getErrorMessage(err));
46
63
  }
47
- });
64
+ }, {
65
+ progressMessage: (params) => `⊜ update_memory: ${params.hash.slice(0, 12)}... [replace content]`,
66
+ }));
48
67
  }
49
68
  //# sourceMappingURL=update-memory.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"update-memory.js","sourceRoot":"","sources":["../../src/tools/update-memory.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAC3E,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AAC/D,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,EACL,YAAY,EACZ,YAAY,EACZ,2BAA2B,EAC3B,MAAM,EACN,wBAAwB,GACzB,MAAM,cAAc,CAAC;AAGtB,MAAM,iBAAiB,GAAG;;iBAET,CAAC;AAClB,SAAS,oBAAoB,CAAC,IAAY;IACxC,OAAO,qBAAqB,IAAI,EAAE,CAAC;AACrC,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,MAAiB,EAAE,EAAW;IACjE,MAAM,CAAC,YAAY,CACjB,eAAe,EACf;QACE,KAAK,EAAE,eAAe;QACtB,WAAW,EACT,0HAA0H;QAC5H,WAAW,EAAE,uBAAuB;QACpC,YAAY,EAAE,kBAAkB;QAChC,WAAW,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE;KAC7D,EACD,KAAK,EAAE,MAAmB,EAAE,EAAE;QAC5B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,YAAY,CAAC,EAAE,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;YAE/C,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO,mBAAmB,CACxB,WAAW,EACX,oBAAoB,CAAC,MAAM,CAAC,IAAI,CAAC,CAClC,CAAC;YACJ,CAAC;YAED,MAAM,YAAY,GAAG,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC9C,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,IAAI,YAAY,CAAC;YAC5C,MAAM,OAAO,GAAG,iBAAiB,CAAC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC3D,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;YAErB,wBAAwB,CAAC,EAAE,EAAE,GAAG,EAAE;gBAChC,EAAE,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,GAAG,CAC/B,OAAO,EACP,MAAM,CAAC,OAAO,EACd,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EACvB,GAAG,EACH,MAAM,CAAC,IAAI,CACZ,CAAC;YACJ,CAAC,CAAC,CAAC;YACH,MAAM,YAAY,CAAC,MAAM,EAAE,QAAQ,EAAE;gBACnC,OAAO,EAAE,MAAM,CAAC,IAAI;gBACpB,OAAO;aACR,CAAC,CAAC;YAEH,MAAM,2BAA2B,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;YAEvD,OAAO,kBAAkB,CAAC;gBACxB,EAAE,EAAE,IAAI;gBACR,MAAM,EAAE,EAAE,QAAQ,EAAE,MAAM,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE;aACrD,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"}
1
+ {"version":3,"file":"update-memory.js","sourceRoot":"","sources":["../../src/tools/update-memory.ts"],"names":[],"mappings":"AAKA,OAAO,EACL,UAAU,EACV,WAAW,EACX,SAAS,EACT,eAAe,EACf,eAAe,GAChB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,iBAAiB,EAAE,MAAM,gBAAgB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,2BAA2B,EAAE,MAAM,qBAAqB,CAAC;AAChF,OAAO,EACL,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAkB,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5D,OAAO,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AAC/D,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAC3D,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAGhD,MAAM,iBAAiB,GAAG;;iBAET,CAAC;AAClB,MAAM,yBAAyB,GAAG,uCAAuC,CAAC;AAC1E,MAAM,sBAAsB,GAAG,0CAA0C,CAAC;AAE1E,KAAK,UAAU,4BAA4B,CACzC,MAAiB,EACjB,OAAe,EACf,OAAe;IAEf,MAAM,aAAa,GAAG,CAAC,2BAA2B,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IACrE,IAAI,OAAO,KAAK,OAAO,EAAE,CAAC;QACxB,aAAa,CAAC,IAAI,CAAC,2BAA2B,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IACnE,CAAC;IACD,MAAM,OAAO,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;AAC1C,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,MAAiB,EAAE,EAAW;IACjE,MAAM,CAAC,YAAY,CACjB,eAAe,EACf;QACE,KAAK,EAAE,eAAe;QACtB,WAAW,EACT,0HAA0H;QAC5H,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,MAAM,QAAQ,GAAG,EAAE;iBAChB,WAAW,CAAY,yBAAyB,CAAC;iBACjD,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAEpB,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO,mBAAmB,CACxB,WAAW,EACX,qBAAqB,MAAM,CAAC,IAAI,EAAE,CACnC,CAAC;YACJ,CAAC;YAED,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,IAAI,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YACxD,MAAM,OAAO,GAAG,iBAAiB,CAAC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAE3D,IAAI,OAAO,KAAK,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5B,MAAM,SAAS,GAAG,EAAE;qBACjB,WAAW,CAAC,sBAAsB,CAAC;qBACnC,GAAG,CAAC,OAAO,CAAC,CAAC;gBAChB,IAAI,SAAS,EAAE,CAAC;oBACd,OAAO,mBAAmB,CACxB,UAAU,EACV,kDAAkD,OAAO,EAAE,CAC5D,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YAErC,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE;gBAClB,EAAE,CAAC,WAAW,CAAC,iBAAiB,CAAC,CAAC,GAAG,CACnC,OAAO,EACP,MAAM,CAAC,OAAO,EACd,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EACvB,GAAG,EACH,MAAM,CAAC,IAAI,CACZ,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,MAAM,YAAY,CAAC,MAAM,EAAE,QAAQ,EAAE;gBACnC,OAAO,EAAE,MAAM,CAAC,IAAI;gBACpB,OAAO;aACR,CAAC,CAAC;YACH,MAAM,4BAA4B,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;YAEjE,OAAO,kBAAkB,CAAC;gBACxB,QAAQ,EAAE,MAAM,CAAC,IAAI;gBACrB,QAAQ,EAAE,OAAO;aAClB,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,oBAAoB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,uBAAuB;KACtE,CACF,CACF,CAAC;AACJ,CAAC"}
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@j0hanz/memory-mcp",
3
- "version": "0.1.1",
3
+ "version": "1.0.1",
4
4
  "mcpName": "io.github.j0hanz/memory-mcp",
5
5
  "author": "Johanz",
6
6
  "license": "MIT",
@@ -50,7 +50,6 @@
50
50
  "type-check:trace": "node -e \"require('fs').rmSync('.ts-trace',{recursive:true,force:true})\" && tsc --noEmit --generateTrace .ts-trace",
51
51
  "lint": "eslint .",
52
52
  "lint:fix": "eslint . --fix",
53
- "check:docs:mcp": "node scripts/check-mcp-docs-quality.mjs",
54
53
  "test": "node scripts/tasks.mjs test",
55
54
  "test:fast": "node --test --import tsx/esm src/__tests__/**/*.test.ts",
56
55
  "test:coverage": "node scripts/tasks.mjs test --coverage",