feat(ai): link memories to source messages

apps/cli/src/ai/agent-stream-runner.ts

@@ -160,6 +160,7 @@ export function createCliRunAgentStream(
         maxToolResultTokens: resolveCrossChatToolResultTokenBudget(contextWindow),
         memoryService: options.memoryService,
         aiChatId,
+        reportMemoryChange: (memoryId) => onEvent({ type: 'memory_change', memoryId }),
       })
       await runCrossChatAgent({
         userMessage,

apps/cli/src/ai/cross-chat-tool-adapter.ts

@@ -23,6 +23,7 @@ export function createCliCrossChatTools(options: {
   maxToolResultTokens: number
   memoryService: AIMemoryService
   aiChatId: string
+  reportMemoryChange?: (memoryId: string) => void
 }): AgentTool<any, any>[] {
   const context: Omit<CrossChatToolExecutionContext, 'abortSignal' | 'reportProgress'> = {
     locale: options.locale,
@@ -30,6 +31,7 @@ export function createCliCrossChatTools(options: {
     analysisService: options.analysisService,
     memoryService: options.memoryService,
     aiChatId: options.aiChatId,
+    reportMemoryChange: options.reportMemoryChange,
     maxToolResultTokens: options.maxToolResultTokens,
     countTokens,
     preprocessMessagesBySession: (sessionId, messages) =>

apps/desktop/main/ai/agent-stream-runner.ts

@@ -155,6 +155,7 @@ export function createElectronRunAgentStream(
         maxToolResultTokens: resolveCrossChatToolResultTokenBudget(resolvedContextWindow),
         memoryService: crossChatOptions.memoryService,
         aiChatId,
+        reportMemoryChange: (memoryId) => onEvent({ type: 'memory_change', memoryId }),
       })
       await runCrossChatAgent({
         userMessage,

apps/desktop/main/ai/cross-chat-tool-adapter.ts

@@ -23,6 +23,7 @@ export function createElectronCrossChatTools(options: {
   maxToolResultTokens: number
   memoryService: AIMemoryService
   aiChatId: string
+  reportMemoryChange?: (memoryId: string) => void
 }): AgentTool<any, any>[] {
   const context: Omit<CrossChatToolExecutionContext, 'abortSignal' | 'reportProgress'> = {
     locale: options.locale,
@@ -30,6 +31,7 @@ export function createElectronCrossChatTools(options: {
     analysisService: options.analysisService,
     memoryService: options.memoryService,
     aiChatId: options.aiChatId,
+    reportMemoryChange: options.reportMemoryChange,
     maxToolResultTokens: options.maxToolResultTokens,
     countTokens,
     preprocessMessagesBySession: (sessionId, messages) =>