indusagi 0.12.13 → 0.12.16

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 (344) hide show
  1. package/dist/agent/agent-session.d.ts +352 -0
  2. package/dist/agent/agent-session.d.ts.map +1 -0
  3. package/dist/agent/agent-session.js +870 -0
  4. package/dist/agent/agent-session.js.map +1 -0
  5. package/dist/agent/agent-session.test.d.ts +5 -0
  6. package/dist/agent/agent-session.test.d.ts.map +1 -0
  7. package/dist/agent/agent-session.test.js +129 -0
  8. package/dist/agent/agent-session.test.js.map +1 -0
  9. package/dist/agent/agent.d.ts +108 -2
  10. package/dist/agent/agent.d.ts.map +1 -1
  11. package/dist/agent/agent.js +210 -17
  12. package/dist/agent/agent.js.map +1 -1
  13. package/dist/agent/bash-executor.d.ts +33 -0
  14. package/dist/agent/bash-executor.d.ts.map +1 -0
  15. package/dist/agent/bash-executor.js +197 -0
  16. package/dist/agent/bash-executor.js.map +1 -0
  17. package/dist/agent/compaction/compaction.d.ts +87 -0
  18. package/dist/agent/compaction/compaction.d.ts.map +1 -0
  19. package/dist/agent/compaction/compaction.js +598 -0
  20. package/dist/agent/compaction/compaction.js.map +1 -0
  21. package/dist/agent/compaction/index.d.ts +6 -0
  22. package/dist/agent/compaction/index.d.ts.map +1 -0
  23. package/dist/agent/compaction/index.js +6 -0
  24. package/dist/agent/compaction/index.js.map +1 -0
  25. package/dist/agent/compaction/utils.d.ts +41 -0
  26. package/dist/agent/compaction/utils.d.ts.map +1 -0
  27. package/dist/agent/compaction/utils.js +240 -0
  28. package/dist/agent/compaction/utils.js.map +1 -0
  29. package/dist/agent/event-bus.js +1 -3
  30. package/dist/agent/event-bus.js.map +1 -1
  31. package/dist/agent/index.d.ts +11 -5
  32. package/dist/agent/index.d.ts.map +1 -1
  33. package/dist/agent/index.js +19 -8
  34. package/dist/agent/index.js.map +1 -1
  35. package/dist/agent/messages.d.ts +2 -48
  36. package/dist/agent/messages.d.ts.map +1 -1
  37. package/dist/agent/messages.js.map +1 -1
  38. package/dist/agent/proxy.js.map +1 -1
  39. package/dist/agent/session-manager.d.ts +2 -3
  40. package/dist/agent/session-manager.d.ts.map +1 -1
  41. package/dist/agent/session-manager.js +10 -6
  42. package/dist/agent/session-manager.js.map +1 -1
  43. package/dist/agent/settings-manager.d.ts +204 -0
  44. package/dist/agent/settings-manager.d.ts.map +1 -0
  45. package/dist/agent/settings-manager.js +594 -0
  46. package/dist/agent/settings-manager.js.map +1 -0
  47. package/dist/agent/state-manager.js +1 -0
  48. package/dist/agent/state-manager.js.map +1 -1
  49. package/dist/agent/telemetry.js +1 -3
  50. package/dist/agent/telemetry.js.map +1 -1
  51. package/dist/agent/tools/bash.js +1 -0
  52. package/dist/agent/tools/bash.js.map +1 -1
  53. package/dist/agent/tools/edit.test.d.ts +6 -0
  54. package/dist/agent/tools/edit.test.d.ts.map +1 -0
  55. package/dist/agent/tools/edit.test.js +196 -0
  56. package/dist/agent/tools/edit.test.js.map +1 -0
  57. package/dist/agent/tools/index.d.ts +26 -26
  58. package/dist/agent/tools/registry.d.ts +1 -1
  59. package/dist/agent/tools/registry.d.ts.map +1 -1
  60. package/dist/agent/tools/registry.js +2 -4
  61. package/dist/agent/tools/registry.js.map +1 -1
  62. package/dist/agent/tools/todo-store.js +2 -1
  63. package/dist/agent/tools/todo-store.js.map +1 -1
  64. package/dist/agent/tools/utils/hook-runner.js +1 -3
  65. package/dist/agent/tools/utils/hook-runner.js.map +1 -1
  66. package/dist/agent/types.d.ts +139 -1
  67. package/dist/agent/types.d.ts.map +1 -1
  68. package/dist/agent/types.js.map +1 -1
  69. package/dist/ai/api-registry.js +1 -3
  70. package/dist/ai/api-registry.js.map +1 -1
  71. package/dist/ai/models.generated.d.ts +621 -53
  72. package/dist/ai/models.generated.d.ts.map +1 -1
  73. package/dist/ai/models.generated.js +548 -0
  74. package/dist/ai/models.generated.js.map +1 -1
  75. package/dist/ai/models.js +1 -0
  76. package/dist/ai/models.js.map +1 -1
  77. package/dist/ai/providers/amazon-bedrock.d.ts +6 -0
  78. package/dist/ai/providers/amazon-bedrock.d.ts.map +1 -1
  79. package/dist/ai/providers/amazon-bedrock.js +47 -9
  80. package/dist/ai/providers/amazon-bedrock.js.map +1 -1
  81. package/dist/ai/providers/anthropic.d.ts +8 -0
  82. package/dist/ai/providers/anthropic.d.ts.map +1 -1
  83. package/dist/ai/providers/anthropic.js +28 -12
  84. package/dist/ai/providers/anthropic.js.map +1 -1
  85. package/dist/ai/providers/azure-openai-responses.js +1 -0
  86. package/dist/ai/providers/azure-openai-responses.js.map +1 -1
  87. package/dist/ai/providers/github-copilot-headers.d.ts +33 -0
  88. package/dist/ai/providers/github-copilot-headers.d.ts.map +1 -0
  89. package/dist/ai/providers/github-copilot-headers.js +51 -0
  90. package/dist/ai/providers/github-copilot-headers.js.map +1 -0
  91. package/dist/ai/providers/google-shared.d.ts.map +1 -1
  92. package/dist/ai/providers/google-shared.js +16 -22
  93. package/dist/ai/providers/google-shared.js.map +1 -1
  94. package/dist/ai/providers/google-vertex.d.ts.map +1 -1
  95. package/dist/ai/providers/google-vertex.js +28 -5
  96. package/dist/ai/providers/google-vertex.js.map +1 -1
  97. package/dist/ai/providers/google.js +1 -0
  98. package/dist/ai/providers/google.js.map +1 -1
  99. package/dist/ai/providers/kimi.js +1 -0
  100. package/dist/ai/providers/kimi.js.map +1 -1
  101. package/dist/ai/providers/mistral.d.ts +22 -0
  102. package/dist/ai/providers/mistral.d.ts.map +1 -0
  103. package/dist/ai/providers/mistral.js +497 -0
  104. package/dist/ai/providers/mistral.js.map +1 -0
  105. package/dist/ai/providers/openai-codex-responses.js +1 -0
  106. package/dist/ai/providers/openai-codex-responses.js.map +1 -1
  107. package/dist/ai/providers/openai-completions.js +2 -0
  108. package/dist/ai/providers/openai-completions.js.map +1 -1
  109. package/dist/ai/providers/openai-responses.d.ts.map +1 -1
  110. package/dist/ai/providers/openai-responses.js +30 -0
  111. package/dist/ai/providers/openai-responses.js.map +1 -1
  112. package/dist/ai/providers/register-builtins.d.ts.map +1 -1
  113. package/dist/ai/providers/register-builtins.js +6 -0
  114. package/dist/ai/providers/register-builtins.js.map +1 -1
  115. package/dist/ai/providers/simple-options.js +2 -0
  116. package/dist/ai/providers/simple-options.js.map +1 -1
  117. package/dist/ai/providers/transform-messages.js +3 -9
  118. package/dist/ai/providers/transform-messages.js.map +1 -1
  119. package/dist/ai/stream.js +1 -3
  120. package/dist/ai/stream.js.map +1 -1
  121. package/dist/ai/types.d.ts +34 -0
  122. package/dist/ai/types.d.ts.map +1 -1
  123. package/dist/ai/types.js.map +1 -1
  124. package/dist/ai/utils/base-stream-handler.js +1 -0
  125. package/dist/ai/utils/base-stream-handler.js.map +1 -1
  126. package/dist/ai/utils/event-stream.js +9 -4
  127. package/dist/ai/utils/event-stream.js.map +1 -1
  128. package/dist/ai/utils/hash.d.ts +6 -0
  129. package/dist/ai/utils/hash.d.ts.map +1 -0
  130. package/dist/ai/utils/hash.js +17 -0
  131. package/dist/ai/utils/hash.js.map +1 -0
  132. package/dist/ai/utils/index.d.ts +1 -0
  133. package/dist/ai/utils/index.d.ts.map +1 -1
  134. package/dist/ai/utils/index.js +1 -0
  135. package/dist/ai/utils/index.js.map +1 -1
  136. package/dist/ai/utils/oauth/github-copilot.d.ts.map +1 -1
  137. package/dist/ai/utils/oauth/github-copilot.js +19 -8
  138. package/dist/ai/utils/oauth/github-copilot.js.map +1 -1
  139. package/dist/ai/utils/overflow.d.ts.map +1 -1
  140. package/dist/ai/utils/overflow.js +4 -0
  141. package/dist/ai/utils/overflow.js.map +1 -1
  142. package/dist/ai/utils/provider-adapter.js +9 -0
  143. package/dist/ai/utils/provider-adapter.js.map +1 -1
  144. package/dist/ai/utils/provider-client-builder.js +1 -1
  145. package/dist/ai/utils/provider-client-builder.js.map +1 -1
  146. package/dist/ai/utils/provider-errors.js +2 -0
  147. package/dist/ai/utils/provider-errors.js.map +1 -1
  148. package/dist/ai/utils/stream-event-helper.js +3 -0
  149. package/dist/ai/utils/stream-event-helper.js.map +1 -1
  150. package/dist/ai/utils/stream-handler-types.js +5 -0
  151. package/dist/ai/utils/stream-handler-types.js.map +1 -1
  152. package/dist/ai/utils/streaming-state-manager.js +4 -2
  153. package/dist/ai/utils/streaming-state-manager.js.map +1 -1
  154. package/dist/index.d.ts +1 -0
  155. package/dist/index.d.ts.map +1 -1
  156. package/dist/index.js +1 -0
  157. package/dist/index.js.map +1 -1
  158. package/dist/mcp/client-pool.d.ts +106 -0
  159. package/dist/mcp/client-pool.d.ts.map +1 -0
  160. package/dist/mcp/client-pool.js +234 -0
  161. package/dist/mcp/client-pool.js.map +1 -0
  162. package/dist/mcp/client.d.ts +158 -0
  163. package/dist/mcp/client.d.ts.map +1 -0
  164. package/dist/mcp/client.js +599 -0
  165. package/dist/mcp/client.js.map +1 -0
  166. package/dist/mcp/config.d.ts +61 -0
  167. package/dist/mcp/config.d.ts.map +1 -0
  168. package/dist/mcp/config.js +250 -0
  169. package/dist/mcp/config.js.map +1 -0
  170. package/dist/mcp/errors.d.ts +104 -0
  171. package/dist/mcp/errors.d.ts.map +1 -0
  172. package/dist/mcp/errors.js +146 -0
  173. package/dist/mcp/errors.js.map +1 -0
  174. package/dist/mcp/index.d.ts +56 -0
  175. package/dist/mcp/index.d.ts.map +1 -0
  176. package/dist/mcp/index.js +83 -0
  177. package/dist/mcp/index.js.map +1 -0
  178. package/dist/mcp/schema-converter.d.ts +68 -0
  179. package/dist/mcp/schema-converter.d.ts.map +1 -0
  180. package/dist/mcp/schema-converter.js +230 -0
  181. package/dist/mcp/schema-converter.js.map +1 -0
  182. package/dist/mcp/server.d.ts +111 -0
  183. package/dist/mcp/server.d.ts.map +1 -0
  184. package/dist/mcp/server.js +305 -0
  185. package/dist/mcp/server.js.map +1 -0
  186. package/dist/mcp/tool-factory.d.ts +63 -0
  187. package/dist/mcp/tool-factory.d.ts.map +1 -0
  188. package/dist/mcp/tool-factory.js +228 -0
  189. package/dist/mcp/tool-factory.js.map +1 -0
  190. package/dist/mcp/types.d.ts +289 -0
  191. package/dist/mcp/types.d.ts.map +1 -0
  192. package/dist/mcp/types.js +8 -0
  193. package/dist/mcp/types.js.map +1 -0
  194. package/dist/mcp.d.ts +6 -0
  195. package/dist/mcp.d.ts.map +1 -0
  196. package/dist/mcp.js +6 -0
  197. package/dist/mcp.js.map +1 -0
  198. package/dist/memory/embedder/base.d.ts +41 -0
  199. package/dist/memory/embedder/base.d.ts.map +1 -0
  200. package/dist/memory/embedder/base.js +10 -0
  201. package/dist/memory/embedder/base.js.map +1 -0
  202. package/dist/memory/embedder/index.d.ts +8 -0
  203. package/dist/memory/embedder/index.d.ts.map +1 -0
  204. package/dist/memory/embedder/index.js +6 -0
  205. package/dist/memory/embedder/index.js.map +1 -0
  206. package/dist/memory/embedder/openai.d.ts +35 -0
  207. package/dist/memory/embedder/openai.d.ts.map +1 -0
  208. package/dist/memory/embedder/openai.js +109 -0
  209. package/dist/memory/embedder/openai.js.map +1 -0
  210. package/dist/memory/index.d.ts +33 -0
  211. package/dist/memory/index.d.ts.map +1 -0
  212. package/dist/memory/index.js +31 -0
  213. package/dist/memory/index.js.map +1 -0
  214. package/dist/memory/memory.d.ts +126 -0
  215. package/dist/memory/memory.d.ts.map +1 -0
  216. package/dist/memory/memory.js +285 -0
  217. package/dist/memory/memory.js.map +1 -0
  218. package/dist/memory/processors/base.d.ts +42 -0
  219. package/dist/memory/processors/base.d.ts.map +1 -0
  220. package/dist/memory/processors/base.js +6 -0
  221. package/dist/memory/processors/base.js.map +1 -0
  222. package/dist/memory/processors/index.d.ts +16 -0
  223. package/dist/memory/processors/index.d.ts.map +1 -0
  224. package/dist/memory/processors/index.js +18 -0
  225. package/dist/memory/processors/index.js.map +1 -0
  226. package/dist/memory/processors/message-history.d.ts +35 -0
  227. package/dist/memory/processors/message-history.d.ts.map +1 -0
  228. package/dist/memory/processors/message-history.js +53 -0
  229. package/dist/memory/processors/message-history.js.map +1 -0
  230. package/dist/memory/processors/observational-memory/index.d.ts +82 -0
  231. package/dist/memory/processors/observational-memory/index.d.ts.map +1 -0
  232. package/dist/memory/processors/observational-memory/index.js +239 -0
  233. package/dist/memory/processors/observational-memory/index.js.map +1 -0
  234. package/dist/memory/processors/observational-memory/observer-agent.d.ts +64 -0
  235. package/dist/memory/processors/observational-memory/observer-agent.d.ts.map +1 -0
  236. package/dist/memory/processors/observational-memory/observer-agent.js +362 -0
  237. package/dist/memory/processors/observational-memory/observer-agent.js.map +1 -0
  238. package/dist/memory/processors/observational-memory/reflector-agent.d.ts +38 -0
  239. package/dist/memory/processors/observational-memory/reflector-agent.d.ts.map +1 -0
  240. package/dist/memory/processors/observational-memory/reflector-agent.js +213 -0
  241. package/dist/memory/processors/observational-memory/reflector-agent.js.map +1 -0
  242. package/dist/memory/processors/observational-memory/token-counter.d.ts +35 -0
  243. package/dist/memory/processors/observational-memory/token-counter.d.ts.map +1 -0
  244. package/dist/memory/processors/observational-memory/token-counter.js +90 -0
  245. package/dist/memory/processors/observational-memory/token-counter.js.map +1 -0
  246. package/dist/memory/processors/semantic-recall.d.ts +55 -0
  247. package/dist/memory/processors/semantic-recall.d.ts.map +1 -0
  248. package/dist/memory/processors/semantic-recall.js +152 -0
  249. package/dist/memory/processors/semantic-recall.js.map +1 -0
  250. package/dist/memory/processors/working-memory.d.ts +41 -0
  251. package/dist/memory/processors/working-memory.d.ts.map +1 -0
  252. package/dist/memory/processors/working-memory.js +87 -0
  253. package/dist/memory/processors/working-memory.js.map +1 -0
  254. package/dist/memory/storage/base.d.ts +288 -0
  255. package/dist/memory/storage/base.d.ts.map +1 -0
  256. package/dist/memory/storage/base.js +209 -0
  257. package/dist/memory/storage/base.js.map +1 -0
  258. package/dist/memory/storage/index.d.ts +9 -0
  259. package/dist/memory/storage/index.d.ts.map +1 -0
  260. package/dist/memory/storage/index.js +7 -0
  261. package/dist/memory/storage/index.js.map +1 -0
  262. package/dist/memory/storage/inmemory.d.ts +93 -0
  263. package/dist/memory/storage/inmemory.d.ts.map +1 -0
  264. package/dist/memory/storage/inmemory.js +647 -0
  265. package/dist/memory/storage/inmemory.js.map +1 -0
  266. package/dist/memory/tools/working-memory.d.ts +100 -0
  267. package/dist/memory/tools/working-memory.d.ts.map +1 -0
  268. package/dist/memory/tools/working-memory.js +237 -0
  269. package/dist/memory/tools/working-memory.js.map +1 -0
  270. package/dist/memory/types.d.ts +386 -0
  271. package/dist/memory/types.d.ts.map +1 -0
  272. package/dist/memory/types.js +58 -0
  273. package/dist/memory/types.js.map +1 -0
  274. package/dist/memory/vector/base.d.ts +145 -0
  275. package/dist/memory/vector/base.d.ts.map +1 -0
  276. package/dist/memory/vector/base.js +81 -0
  277. package/dist/memory/vector/base.js.map +1 -0
  278. package/dist/memory/vector/index.d.ts +8 -0
  279. package/dist/memory/vector/index.d.ts.map +1 -0
  280. package/dist/memory/vector/index.js +7 -0
  281. package/dist/memory/vector/index.js.map +1 -0
  282. package/dist/memory/vector/inmemory.d.ts +47 -0
  283. package/dist/memory/vector/inmemory.d.ts.map +1 -0
  284. package/dist/memory/vector/inmemory.js +231 -0
  285. package/dist/memory/vector/inmemory.js.map +1 -0
  286. package/dist/tui/autocomplete.js +5 -1
  287. package/dist/tui/autocomplete.js.map +1 -1
  288. package/dist/tui/components/box.js +9 -1
  289. package/dist/tui/components/box.js.map +1 -1
  290. package/dist/tui/components/cancellable-loader.js +3 -4
  291. package/dist/tui/components/cancellable-loader.js.map +1 -1
  292. package/dist/tui/components/editor.d.ts +39 -3
  293. package/dist/tui/components/editor.d.ts.map +1 -1
  294. package/dist/tui/components/editor.js +162 -133
  295. package/dist/tui/components/editor.js.map +1 -1
  296. package/dist/tui/components/image.js +8 -0
  297. package/dist/tui/components/image.js.map +1 -1
  298. package/dist/tui/components/input.js +10 -11
  299. package/dist/tui/components/input.js.map +1 -1
  300. package/dist/tui/components/loader.js +7 -4
  301. package/dist/tui/components/loader.js.map +1 -1
  302. package/dist/tui/components/markdown.js +10 -0
  303. package/dist/tui/components/markdown.js.map +1 -1
  304. package/dist/tui/components/select-list.js +8 -4
  305. package/dist/tui/components/select-list.js.map +1 -1
  306. package/dist/tui/components/settings-list.js +12 -4
  307. package/dist/tui/components/settings-list.js.map +1 -1
  308. package/dist/tui/components/spacer.js +1 -0
  309. package/dist/tui/components/spacer.js.map +1 -1
  310. package/dist/tui/components/text.js +5 -0
  311. package/dist/tui/components/text.js.map +1 -1
  312. package/dist/tui/components/truncated-text.js +3 -0
  313. package/dist/tui/components/truncated-text.js.map +1 -1
  314. package/dist/tui/fuzzy.js +4 -1
  315. package/dist/tui/fuzzy.js.map +1 -1
  316. package/dist/tui/keybindings.js +3 -2
  317. package/dist/tui/keybindings.js.map +1 -1
  318. package/dist/tui/keys.d.ts +17 -1
  319. package/dist/tui/keys.d.ts.map +1 -1
  320. package/dist/tui/keys.js +56 -0
  321. package/dist/tui/keys.js.map +1 -1
  322. package/dist/tui/kill-ring.d.ts +28 -0
  323. package/dist/tui/kill-ring.d.ts.map +1 -0
  324. package/dist/tui/kill-ring.js +44 -0
  325. package/dist/tui/kill-ring.js.map +1 -0
  326. package/dist/tui/stdin-buffer.js +6 -5
  327. package/dist/tui/stdin-buffer.js.map +1 -1
  328. package/dist/tui/terminal-image.js +1 -3
  329. package/dist/tui/terminal-image.js.map +1 -1
  330. package/dist/tui/terminal.d.ts +13 -0
  331. package/dist/tui/terminal.d.ts.map +1 -1
  332. package/dist/tui/terminal.js +70 -7
  333. package/dist/tui/terminal.js.map +1 -1
  334. package/dist/tui/tui.d.ts +36 -0
  335. package/dist/tui/tui.d.ts.map +1 -1
  336. package/dist/tui/tui.js +101 -29
  337. package/dist/tui/tui.js.map +1 -1
  338. package/dist/tui/undo-stack.d.ts +17 -0
  339. package/dist/tui/undo-stack.d.ts.map +1 -0
  340. package/dist/tui/undo-stack.js +25 -0
  341. package/dist/tui/undo-stack.js.map +1 -0
  342. package/dist/tui/utils.js +12 -16
  343. package/dist/tui/utils.js.map +1 -1
  344. package/package.json +32 -2
@@ -0,0 +1,90 @@
1
+ /**
2
+ * Token Counter Utility
3
+ * Based on Mastra TokenCounter
4
+ */
5
+ /**
6
+ * Simple token counter using character-based estimation.
7
+ * For production, integrate with js-tiktoken for accurate counting.
8
+ */
9
+ export class TokenCounter {
10
+ // Average characters per token (OpenAI's estimation)
11
+ static CHARS_PER_TOKEN = 4;
12
+ // Per-message overhead for role tokens, framing, separators
13
+ static TOKENS_PER_MESSAGE = 3.8;
14
+ // Conversation-level overhead
15
+ static TOKENS_PER_CONVERSATION = 24;
16
+ /**
17
+ * Count tokens in a plain string
18
+ */
19
+ countString(text) {
20
+ if (!text)
21
+ return 0;
22
+ // Simple estimation: characters / 4
23
+ // For production, use tiktoken for accurate counting
24
+ return Math.ceil(text.length / TokenCounter.CHARS_PER_TOKEN);
25
+ }
26
+ /**
27
+ * Count tokens in a single message
28
+ */
29
+ countMessage(message) {
30
+ let tokenString = message.role;
31
+ let overhead = TokenCounter.TOKENS_PER_MESSAGE;
32
+ if (typeof message.content === "string") {
33
+ tokenString += message.content;
34
+ }
35
+ else if (message.content && typeof message.content === "object") {
36
+ const content = message.content;
37
+ if (content.content && !Array.isArray(content.parts)) {
38
+ tokenString += content.content;
39
+ }
40
+ else if (Array.isArray(content.parts)) {
41
+ for (const part of content.parts) {
42
+ if (part.type === "text") {
43
+ tokenString += part.text;
44
+ }
45
+ else if (part.type === "tool-call") {
46
+ tokenString += part.toolName;
47
+ if (part.args) {
48
+ tokenString += JSON.stringify(part.args);
49
+ overhead -= 12; // Compensate for JSON overhead
50
+ }
51
+ }
52
+ else if (part.type === "tool-result") {
53
+ if (part.result !== undefined) {
54
+ tokenString += typeof part.result === "string"
55
+ ? part.result
56
+ : JSON.stringify(part.result);
57
+ overhead -= 12;
58
+ }
59
+ }
60
+ }
61
+ }
62
+ }
63
+ return Math.round(this.countString(tokenString) + overhead);
64
+ }
65
+ /**
66
+ * Count tokens in an array of messages
67
+ */
68
+ countMessages(messages) {
69
+ if (!messages || messages.length === 0)
70
+ return 0;
71
+ let total = TokenCounter.TOKENS_PER_CONVERSATION;
72
+ for (const message of messages) {
73
+ total += this.countMessage(message);
74
+ }
75
+ return total;
76
+ }
77
+ /**
78
+ * Count tokens in observations string
79
+ */
80
+ countObservations(observations) {
81
+ return this.countString(observations);
82
+ }
83
+ }
84
+ /**
85
+ * Create a default token counter instance
86
+ */
87
+ export function createTokenCounter() {
88
+ return new TokenCounter();
89
+ }
90
+ //# sourceMappingURL=token-counter.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"token-counter.js","sourceRoot":"","sources":["../../../../src/memory/processors/observational-memory/token-counter.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAIH;;;GAGG;AACH,MAAM,OAAO,YAAY;IACvB,qDAAqD;IAC7C,MAAM,CAAU,eAAe,GAAG,CAAC,CAAC;IAE5C,4DAA4D;IACpD,MAAM,CAAU,kBAAkB,GAAG,GAAG,CAAC;IAEjD,8BAA8B;IACtB,MAAM,CAAU,uBAAuB,GAAG,EAAE,CAAC;IAErD;;OAEG;IACH,WAAW,CAAC,IAAY;QACtB,IAAI,CAAC,IAAI;YAAE,OAAO,CAAC,CAAC;QACpB,oCAAoC;QACpC,qDAAqD;QACrD,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,YAAY,CAAC,eAAe,CAAC,CAAC;IAC/D,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,OAAoB;QAC/B,IAAI,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC;QAC/B,IAAI,QAAQ,GAAG,YAAY,CAAC,kBAAkB,CAAC;QAE/C,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YACxC,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC;QACjC,CAAC;aAAM,IAAI,OAAO,CAAC,OAAO,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YAClE,MAAM,OAAO,GAAG,OAAO,CAAC,OAA2B,CAAC;YAEpD,IAAI,OAAO,CAAC,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBACrD,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC;YACjC,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBACxC,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;oBACjC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;wBACzB,WAAW,IAAI,IAAI,CAAC,IAAI,CAAC;oBAC3B,CAAC;yBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;wBACrC,WAAW,IAAI,IAAI,CAAC,QAAQ,CAAC;wBAC7B,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;4BACd,WAAW,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;4BACzC,QAAQ,IAAI,EAAE,CAAC,CAAC,+BAA+B;wBACjD,CAAC;oBACH,CAAC;yBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;wBACvC,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;4BAC9B,WAAW,IAAI,OAAO,IAAI,CAAC,MAAM,KAAK,QAAQ;gCAC5C,CAAC,CAAC,IAAI,CAAC,MAAM;gCACb,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;4BAChC,QAAQ,IAAI,EAAE,CAAC;wBACjB,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,GAAG,QAAQ,CAAC,CAAC;IAC9D,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,QAAuB;QACnC,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QAEjD,IAAI,KAAK,GAAG,YAAY,CAAC,uBAAuB,CAAC;QACjD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QACtC,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,YAAoB;QACpC,OAAO,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;IACxC,CAAC;;AAGH;;GAEG;AACH,MAAM,UAAU,kBAAkB;IAChC,OAAO,IAAI,YAAY,EAAE,CAAC;AAC5B,CAAC"}
@@ -0,0 +1,55 @@
1
+ /**
2
+ * Semantic Recall Processor
3
+ * Based on Mastra SemanticRecall processor
4
+ */
5
+ import type { MemoryProcessor, ProcessorContext } from "./base.js";
6
+ import type { CoreMessage, EmbedderOptions } from "../types.js";
7
+ import type { MemoryStorage } from "../storage/base.js";
8
+ import type { VectorStore } from "../vector/base.js";
9
+ import type { Embedder } from "../embedder/base.js";
10
+ /**
11
+ * Semantic recall processor configuration
12
+ */
13
+ export interface SemanticRecallProcessorConfig {
14
+ storage: MemoryStorage;
15
+ vector: VectorStore;
16
+ embedder: Embedder;
17
+ embedderOptions?: EmbedderOptions;
18
+ indexName: string;
19
+ topK?: number;
20
+ messageRange?: number | {
21
+ before: number;
22
+ after: number;
23
+ };
24
+ scope?: "thread" | "resource";
25
+ threshold?: number;
26
+ }
27
+ /**
28
+ * Semantic Recall Processor
29
+ *
30
+ * Retrieves relevant past messages using vector similarity search.
31
+ */
32
+ export declare class SemanticRecall implements MemoryProcessor {
33
+ readonly id = "semantic-recall";
34
+ readonly name = "Semantic Recall Processor";
35
+ private storage;
36
+ private vector;
37
+ private embedder;
38
+ private embedderOptions?;
39
+ private indexName;
40
+ private topK;
41
+ private messageRange;
42
+ private scope;
43
+ private threshold;
44
+ constructor(config: SemanticRecallProcessorConfig);
45
+ processInputStep(args: {
46
+ messages: CoreMessage[];
47
+ context: ProcessorContext;
48
+ }): Promise<CoreMessage[]>;
49
+ processOutputResult(args: {
50
+ messages: CoreMessage[];
51
+ context: ProcessorContext;
52
+ }): Promise<CoreMessage[]>;
53
+ private messageToText;
54
+ }
55
+ //# sourceMappingURL=semantic-recall.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"semantic-recall.d.ts","sourceRoot":"","sources":["../../../src/memory/processors/semantic-recall.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AACnE,OAAO,KAAK,EAAE,WAAW,EAA0C,eAAe,EAAE,MAAM,aAAa,CAAC;AACxG,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACxD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAEpD;;GAEG;AACH,MAAM,WAAW,6BAA6B;IAC5C,OAAO,EAAE,aAAa,CAAC;IACvB,MAAM,EAAE,WAAW,CAAC;IACpB,QAAQ,EAAE,QAAQ,CAAC;IACnB,eAAe,CAAC,EAAE,eAAe,CAAC;IAClC,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,GAAG;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IAC1D,KAAK,CAAC,EAAE,QAAQ,GAAG,UAAU,CAAC;IAC9B,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAQD;;;;GAIG;AACH,qBAAa,cAAe,YAAW,eAAe;IACpD,QAAQ,CAAC,EAAE,qBAAqB;IAChC,QAAQ,CAAC,IAAI,+BAA+B;IAE5C,OAAO,CAAC,OAAO,CAAgB;IAC/B,OAAO,CAAC,MAAM,CAAc;IAC5B,OAAO,CAAC,QAAQ,CAAW;IAC3B,OAAO,CAAC,eAAe,CAAC,CAAkB;IAC1C,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,IAAI,CAAS;IACrB,OAAO,CAAC,YAAY,CAA6C;IACjE,OAAO,CAAC,KAAK,CAAwB;IACrC,OAAO,CAAC,SAAS,CAAS;gBAEd,MAAM,EAAE,6BAA6B;IAY3C,gBAAgB,CAAC,IAAI,EAAE;QAC3B,QAAQ,EAAE,WAAW,EAAE,CAAC;QACxB,OAAO,EAAE,gBAAgB,CAAC;KAC3B,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAkFpB,mBAAmB,CAAC,IAAI,EAAE;QAC9B,QAAQ,EAAE,WAAW,EAAE,CAAC;QACxB,OAAO,EAAE,gBAAgB,CAAC;KAC3B,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAsC1B,OAAO,CAAC,aAAa;CAatB"}
@@ -0,0 +1,152 @@
1
+ /**
2
+ * Semantic Recall Processor
3
+ * Based on Mastra SemanticRecall processor
4
+ */
5
+ /**
6
+ * Default message range configuration
7
+ */
8
+ const DEFAULT_MESSAGE_RANGE = { before: 1, after: 1 };
9
+ const DEFAULT_TOP_K = 4;
10
+ /**
11
+ * Semantic Recall Processor
12
+ *
13
+ * Retrieves relevant past messages using vector similarity search.
14
+ */
15
+ export class SemanticRecall {
16
+ id = "semantic-recall";
17
+ name = "Semantic Recall Processor";
18
+ storage;
19
+ vector;
20
+ embedder;
21
+ embedderOptions;
22
+ indexName;
23
+ topK;
24
+ messageRange;
25
+ scope;
26
+ threshold;
27
+ constructor(config) {
28
+ this.storage = config.storage;
29
+ this.vector = config.vector;
30
+ this.embedder = config.embedder;
31
+ this.embedderOptions = config.embedderOptions;
32
+ this.indexName = config.indexName;
33
+ this.topK = config.topK ?? DEFAULT_TOP_K;
34
+ this.messageRange = config.messageRange ?? DEFAULT_MESSAGE_RANGE;
35
+ this.scope = config.scope ?? "resource";
36
+ this.threshold = config.threshold ?? 0.7;
37
+ }
38
+ async processInputStep(args) {
39
+ const { messages, context } = args;
40
+ // Get the last user message as query
41
+ const lastUserMessage = [...messages].reverse().find(m => m.role === "user");
42
+ if (!lastUserMessage)
43
+ return messages;
44
+ const queryText = this.messageToText(lastUserMessage);
45
+ // Embed the query
46
+ const { vector: queryVector } = await this.embedder.embed(queryText, this.embedderOptions);
47
+ // Search for similar messages
48
+ const results = await this.vector.query({
49
+ indexName: this.indexName,
50
+ queryVector,
51
+ topK: this.topK,
52
+ minScore: this.threshold,
53
+ filter: {
54
+ resourceId: context.resourceId,
55
+ ...(this.scope === "thread" && { threadId: context.threadId }),
56
+ },
57
+ });
58
+ if (results.length === 0)
59
+ return messages;
60
+ // Get message context (before/after each match)
61
+ const messageRange = typeof this.messageRange === "number"
62
+ ? { before: this.messageRange, after: this.messageRange }
63
+ : this.messageRange;
64
+ const retrievedMessages = [];
65
+ const retrievedIds = new Set();
66
+ for (const result of results) {
67
+ const msgId = result.metadata?.messageId;
68
+ const msgThreadId = result.metadata?.threadId;
69
+ if (!msgId || !msgThreadId)
70
+ continue;
71
+ // Get surrounding messages
72
+ const { messages: threadMessages } = await this.storage.listMessages({
73
+ threadId: msgThreadId,
74
+ perPage: 1000, // Get all messages to find context
75
+ });
76
+ const msgIndex = threadMessages.findIndex(m => m.id === msgId);
77
+ if (msgIndex === -1)
78
+ continue;
79
+ const start = Math.max(0, msgIndex - messageRange.before);
80
+ const end = Math.min(threadMessages.length, msgIndex + messageRange.after + 1);
81
+ for (let i = start; i < end; i++) {
82
+ const msg = threadMessages[i];
83
+ if (msg && !retrievedIds.has(msg.id)) {
84
+ retrievedMessages.push(msg);
85
+ retrievedIds.add(msg.id);
86
+ }
87
+ }
88
+ }
89
+ if (retrievedMessages.length === 0)
90
+ return messages;
91
+ // Sort retrieved messages by date
92
+ retrievedMessages.sort((a, b) => new Date(a.createdAt).getTime() - new Date(b.createdAt).getTime());
93
+ // Remove duplicates from original messages
94
+ const filteredMessages = messages.filter(m => !retrievedIds.has(m.id));
95
+ // Insert semantic recall marker
96
+ const recallMarker = {
97
+ id: `semantic-recall-${Date.now()}`,
98
+ role: "system",
99
+ content: `<semantic_recall>\nRelevant past conversation retrieved based on semantic similarity:\n</semantic_recall>`,
100
+ createdAt: new Date(),
101
+ };
102
+ return [recallMarker, ...retrievedMessages, ...filteredMessages];
103
+ }
104
+ async processOutputResult(args) {
105
+ const { messages, context } = args;
106
+ // Embed and store new messages
107
+ for (const message of messages) {
108
+ if (message.role !== "user" && message.role !== "assistant")
109
+ continue;
110
+ if (message.id.startsWith("working-memory-") || message.id.startsWith("semantic-recall-"))
111
+ continue;
112
+ const text = this.messageToText(message);
113
+ if (!text.trim())
114
+ continue;
115
+ try {
116
+ const { vector } = await this.embedder.embed(text, this.embedderOptions);
117
+ await this.vector.upsert({
118
+ indexName: this.indexName,
119
+ vectors: [vector],
120
+ metadata: [{
121
+ messageId: message.id,
122
+ threadId: context.threadId,
123
+ resourceId: context.resourceId,
124
+ role: message.role,
125
+ timestamp: new Date(message.createdAt).getTime(),
126
+ }],
127
+ ids: [`${context.threadId}-${message.id}`],
128
+ });
129
+ }
130
+ catch (error) {
131
+ console.error(`Failed to embed message ${message.id}:`, error);
132
+ }
133
+ }
134
+ return messages;
135
+ }
136
+ // ========================================================================
137
+ // Private Helpers
138
+ // ========================================================================
139
+ messageToText(message) {
140
+ if (typeof message.content === "string")
141
+ return message.content;
142
+ const content = message.content;
143
+ if (content.parts) {
144
+ return content.parts
145
+ .filter((p) => p.type === "text")
146
+ .map(p => p.text)
147
+ .join("\n");
148
+ }
149
+ return content.content ?? "";
150
+ }
151
+ }
152
+ //# sourceMappingURL=semantic-recall.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"semantic-recall.js","sourceRoot":"","sources":["../../../src/memory/processors/semantic-recall.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAuBH;;GAEG;AACH,MAAM,qBAAqB,GAAG,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;AACtD,MAAM,aAAa,GAAG,CAAC,CAAC;AAExB;;;;GAIG;AACH,MAAM,OAAO,cAAc;IAChB,EAAE,GAAG,iBAAiB,CAAC;IACvB,IAAI,GAAG,2BAA2B,CAAC;IAEpC,OAAO,CAAgB;IACvB,MAAM,CAAc;IACpB,QAAQ,CAAW;IACnB,eAAe,CAAmB;IAClC,SAAS,CAAS;IAClB,IAAI,CAAS;IACb,YAAY,CAA6C;IACzD,KAAK,CAAwB;IAC7B,SAAS,CAAS;IAE1B,YAAY,MAAqC;QAC/C,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAC9B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,eAAe,GAAG,MAAM,CAAC,eAAe,CAAC;QAC9C,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QAClC,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,aAAa,CAAC;QACzC,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,qBAAqB,CAAC;QACjE,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,UAAU,CAAC;QACxC,IAAI,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,GAAG,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,IAGtB;QACC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;QAEnC,qCAAqC;QACrC,MAAM,eAAe,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QAC7E,IAAI,CAAC,eAAe;YAAE,OAAO,QAAQ,CAAC;QAEtC,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC;QAEtD,kBAAkB;QAClB,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,SAAS,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;QAE3F,8BAA8B;QAC9B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;YACtC,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,WAAW;YACX,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,QAAQ,EAAE,IAAI,CAAC,SAAS;YACxB,MAAM,EAAE;gBACN,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,GAAG,CAAC,IAAI,CAAC,KAAK,KAAK,QAAQ,IAAI,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC;aAC/D;SACF,CAAC,CAAC;QAEH,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,QAAQ,CAAC;QAE1C,gDAAgD;QAChD,MAAM,YAAY,GAAG,OAAO,IAAI,CAAC,YAAY,KAAK,QAAQ;YACxD,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,CAAC,YAAY,EAAE;YACzD,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC;QAEtB,MAAM,iBAAiB,GAAkB,EAAE,CAAC;QAC5C,MAAM,YAAY,GAAG,IAAI,GAAG,EAAU,CAAC;QAEvC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,EAAE,SAAmB,CAAC;YACnD,MAAM,WAAW,GAAG,MAAM,CAAC,QAAQ,EAAE,QAAkB,CAAC;YAExD,IAAI,CAAC,KAAK,IAAI,CAAC,WAAW;gBAAE,SAAS;YAErC,2BAA2B;YAC3B,MAAM,EAAE,QAAQ,EAAE,cAAc,EAAE,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC;gBACnE,QAAQ,EAAE,WAAW;gBACrB,OAAO,EAAE,IAAI,EAAE,mCAAmC;aACnD,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,cAAc,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,KAAK,CAAC,CAAC;YAC/D,IAAI,QAAQ,KAAK,CAAC,CAAC;gBAAE,SAAS;YAE9B,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;YAC1D,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,cAAc,CAAC,MAAM,EAAE,QAAQ,GAAG,YAAY,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;YAE/E,KAAK,IAAI,CAAC,GAAG,KAAK,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;gBACjC,MAAM,GAAG,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;gBAC9B,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;oBACrC,iBAAiB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBAC5B,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC3B,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,QAAQ,CAAC;QAEpD,kCAAkC;QAClC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAC9B,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAClE,CAAC;QAEF,2CAA2C;QAC3C,MAAM,gBAAgB,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAEvE,gCAAgC;QAChC,MAAM,YAAY,GAAgB;YAChC,EAAE,EAAE,mBAAmB,IAAI,CAAC,GAAG,EAAE,EAAE;YACnC,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,2GAA2G;YACpH,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;QAEF,OAAO,CAAC,YAAY,EAAE,GAAG,iBAAiB,EAAE,GAAG,gBAAgB,CAAC,CAAC;IACnE,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,IAGzB;QACC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;QAEnC,+BAA+B;QAC/B,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW;gBAAE,SAAS;YACtE,IAAI,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,iBAAiB,CAAC,IAAI,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,kBAAkB,CAAC;gBAAE,SAAS;YAEpG,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;YACzC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;gBAAE,SAAS;YAE3B,IAAI,CAAC;gBACH,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;gBAEzE,MAAM,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;oBACvB,SAAS,EAAE,IAAI,CAAC,SAAS;oBACzB,OAAO,EAAE,CAAC,MAAM,CAAC;oBACjB,QAAQ,EAAE,CAAC;4BACT,SAAS,EAAE,OAAO,CAAC,EAAE;4BACrB,QAAQ,EAAE,OAAO,CAAC,QAAQ;4BAC1B,UAAU,EAAE,OAAO,CAAC,UAAU;4BAC9B,IAAI,EAAE,OAAO,CAAC,IAAI;4BAClB,SAAS,EAAE,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE;yBACjD,CAAC;oBACF,GAAG,EAAE,CAAC,GAAG,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,EAAE,EAAE,CAAC;iBAC3C,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,2BAA2B,OAAO,CAAC,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC;YACjE,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,2EAA2E;IAC3E,kBAAkB;IAClB,2EAA2E;IAEnE,aAAa,CAAC,OAAoB;QACxC,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ;YAAE,OAAO,OAAO,CAAC,OAAO,CAAC;QAEhE,MAAM,OAAO,GAAG,OAAO,CAAC,OAA2B,CAAC;QACpD,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,OAAO,OAAO,CAAC,KAAK;iBACjB,MAAM,CAAC,CAAC,CAAC,EAAuC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;iBACrE,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;iBAChB,IAAI,CAAC,IAAI,CAAC,CAAC;QAChB,CAAC;QAED,OAAO,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;IAC/B,CAAC;CACF"}
@@ -0,0 +1,41 @@
1
+ /**
2
+ * Working Memory Processor
3
+ * Based on Mastra WorkingMemory processor
4
+ */
5
+ import type { MemoryProcessor, ProcessorContext } from "./base.js";
6
+ import type { CoreMessage, WorkingMemoryTemplate } from "../types.js";
7
+ import type { MemoryStorage } from "../storage/base.js";
8
+ /**
9
+ * Working memory processor configuration
10
+ */
11
+ export interface WorkingMemoryProcessorConfig {
12
+ storage: MemoryStorage;
13
+ template?: WorkingMemoryTemplate;
14
+ scope?: "thread" | "resource";
15
+ useVNext?: boolean;
16
+ templateProvider?: {
17
+ getWorkingMemoryTemplate(args: {
18
+ memoryConfig?: import("../types.js").MemoryConfig;
19
+ }): Promise<WorkingMemoryTemplate | null>;
20
+ };
21
+ }
22
+ /**
23
+ * Working Memory Processor
24
+ *
25
+ * Injects structured context (user profile, preferences) into the conversation.
26
+ */
27
+ export declare class WorkingMemory implements MemoryProcessor {
28
+ readonly id = "working-memory";
29
+ readonly name = "Working Memory Processor";
30
+ private storage;
31
+ private template?;
32
+ private scope;
33
+ private useVNext;
34
+ private templateProvider?;
35
+ constructor(config: WorkingMemoryProcessorConfig);
36
+ processInputStep(args: {
37
+ messages: CoreMessage[];
38
+ context: ProcessorContext;
39
+ }): Promise<CoreMessage[]>;
40
+ }
41
+ //# sourceMappingURL=working-memory.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"working-memory.d.ts","sourceRoot":"","sources":["../../../src/memory/processors/working-memory.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AACnE,OAAO,KAAK,EAAE,WAAW,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AACtE,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AAExD;;GAEG;AACH,MAAM,WAAW,4BAA4B;IAC3C,OAAO,EAAE,aAAa,CAAC;IACvB,QAAQ,CAAC,EAAE,qBAAqB,CAAC;IACjC,KAAK,CAAC,EAAE,QAAQ,GAAG,UAAU,CAAC;IAC9B,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,gBAAgB,CAAC,EAAE;QACjB,wBAAwB,CAAC,IAAI,EAAE;YAAE,YAAY,CAAC,EAAE,OAAO,aAAa,EAAE,YAAY,CAAA;SAAE,GAAG,OAAO,CAAC,qBAAqB,GAAG,IAAI,CAAC,CAAC;KAC9H,CAAC;CACH;AAqBD;;;;GAIG;AACH,qBAAa,aAAc,YAAW,eAAe;IACnD,QAAQ,CAAC,EAAE,oBAAoB;IAC/B,QAAQ,CAAC,IAAI,8BAA8B;IAE3C,OAAO,CAAC,OAAO,CAAgB;IAC/B,OAAO,CAAC,QAAQ,CAAC,CAAwB;IACzC,OAAO,CAAC,KAAK,CAAwB;IACrC,OAAO,CAAC,QAAQ,CAAU;IAC1B,OAAO,CAAC,gBAAgB,CAAC,CAAmD;gBAEhE,MAAM,EAAE,4BAA4B;IAQ1C,gBAAgB,CAAC,IAAI,EAAE;QAC3B,QAAQ,EAAE,WAAW,EAAE,CAAC;QACxB,OAAO,EAAE,gBAAgB,CAAC;KAC3B,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;CAmD3B"}
@@ -0,0 +1,87 @@
1
+ /**
2
+ * Working Memory Processor
3
+ * Based on Mastra WorkingMemory processor
4
+ */
5
+ /**
6
+ * Default working memory template
7
+ */
8
+ const DEFAULT_TEMPLATE = {
9
+ format: "markdown",
10
+ content: `
11
+ # User Information
12
+ - **First Name**:
13
+ - **Last Name**:
14
+ - **Location**:
15
+ - **Occupation**:
16
+ - **Interests**:
17
+ - **Goals**:
18
+ - **Events**:
19
+ - **Facts**:
20
+ - **Projects**:
21
+ `,
22
+ };
23
+ /**
24
+ * Working Memory Processor
25
+ *
26
+ * Injects structured context (user profile, preferences) into the conversation.
27
+ */
28
+ export class WorkingMemory {
29
+ id = "working-memory";
30
+ name = "Working Memory Processor";
31
+ storage;
32
+ template;
33
+ scope;
34
+ useVNext;
35
+ templateProvider;
36
+ constructor(config) {
37
+ this.storage = config.storage;
38
+ this.template = config.template;
39
+ this.scope = config.scope ?? "resource";
40
+ this.useVNext = config.useVNext ?? false;
41
+ this.templateProvider = config.templateProvider;
42
+ }
43
+ async processInputStep(args) {
44
+ const { messages, context } = args;
45
+ const { threadId, resourceId, memoryConfig } = context;
46
+ // Determine scope from config
47
+ const scope = memoryConfig?.workingMemory &&
48
+ typeof memoryConfig.workingMemory === "object" &&
49
+ "scope" in memoryConfig.workingMemory
50
+ ? memoryConfig.workingMemory.scope
51
+ : this.scope;
52
+ let workingMemory;
53
+ // Get working memory from storage
54
+ if (scope === "thread") {
55
+ const thread = await this.storage.getThreadById({ threadId });
56
+ workingMemory = thread?.metadata?.workingMemory;
57
+ }
58
+ else {
59
+ const resource = await this.storage.getResourceById({ resourceId });
60
+ workingMemory = resource?.workingMemory;
61
+ }
62
+ // If no working memory exists, use template
63
+ if (!workingMemory) {
64
+ // Try to get template from provider or use local template
65
+ let template = this.template;
66
+ if (this.templateProvider && memoryConfig) {
67
+ const providerTemplate = await this.templateProvider.getWorkingMemoryTemplate({ memoryConfig });
68
+ if (providerTemplate) {
69
+ template = providerTemplate;
70
+ }
71
+ }
72
+ if (!template) {
73
+ template = DEFAULT_TEMPLATE;
74
+ }
75
+ workingMemory = template.content;
76
+ }
77
+ // Inject as system message at the start
78
+ const workingMemoryMessage = {
79
+ id: `working-memory-${Date.now()}`,
80
+ role: "system",
81
+ content: `<working_memory>\n${workingMemory}\n</working_memory>`,
82
+ createdAt: new Date(),
83
+ };
84
+ return [workingMemoryMessage, ...messages];
85
+ }
86
+ }
87
+ //# sourceMappingURL=working-memory.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"working-memory.js","sourceRoot":"","sources":["../../../src/memory/processors/working-memory.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAmBH;;GAEG;AACH,MAAM,gBAAgB,GAA0B;IAC9C,MAAM,EAAE,UAAU;IAClB,OAAO,EAAE;;;;;;;;;;;CAWV;CACA,CAAC;AAEF;;;;GAIG;AACH,MAAM,OAAO,aAAa;IACf,EAAE,GAAG,gBAAgB,CAAC;IACtB,IAAI,GAAG,0BAA0B,CAAC;IAEnC,OAAO,CAAgB;IACvB,QAAQ,CAAyB;IACjC,KAAK,CAAwB;IAC7B,QAAQ,CAAU;IAClB,gBAAgB,CAAoD;IAE5E,YAAY,MAAoC;QAC9C,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QAC9B,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,UAAU,CAAC;QACxC,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,IAAI,KAAK,CAAC;QACzC,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,CAAC;IAClD,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,IAGtB;QACC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC;QACnC,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,YAAY,EAAE,GAAG,OAAO,CAAC;QAEvD,8BAA8B;QAC9B,MAAM,KAAK,GAAG,YAAY,EAAE,aAAa;YACvC,OAAO,YAAY,CAAC,aAAa,KAAK,QAAQ;YAC9C,OAAO,IAAI,YAAY,CAAC,aAAa;YACnC,CAAC,CAAC,YAAY,CAAC,aAAa,CAAC,KAAK;YAClC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC;QAEjB,IAAI,aAAwC,CAAC;QAE7C,kCAAkC;QAClC,IAAI,KAAK,KAAK,QAAQ,EAAE,CAAC;YACvB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC9D,aAAa,GAAG,MAAM,EAAE,QAAQ,EAAE,aAAmC,CAAC;QACxE,CAAC;aAAM,CAAC;YACN,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC;YACpE,aAAa,GAAG,QAAQ,EAAE,aAAa,CAAC;QAC1C,CAAC;QAED,4CAA4C;QAC5C,IAAI,CAAC,aAAa,EAAE,CAAC;YACnB,0DAA0D;YAC1D,IAAI,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC;YAE7B,IAAI,IAAI,CAAC,gBAAgB,IAAI,YAAY,EAAE,CAAC;gBAC1C,MAAM,gBAAgB,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,wBAAwB,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC;gBAChG,IAAI,gBAAgB,EAAE,CAAC;oBACrB,QAAQ,GAAG,gBAAgB,CAAC;gBAC9B,CAAC;YACH,CAAC;YAED,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,QAAQ,GAAG,gBAAgB,CAAC;YAC9B,CAAC;YAED,aAAa,GAAG,QAAQ,CAAC,OAAO,CAAC;QACnC,CAAC;QAED,wCAAwC;QACxC,MAAM,oBAAoB,GAAgB;YACxC,EAAE,EAAE,kBAAkB,IAAI,CAAC,GAAG,EAAE,EAAE;YAClC,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,qBAAqB,aAAa,qBAAqB;YAChE,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;QAEF,OAAO,CAAC,oBAAoB,EAAE,GAAG,QAAQ,CAAC,CAAC;IAC7C,CAAC;CACF"}