feat(tarko-agent): add messageId to thinking events for proper session correlation (#1282)

Author: ulivz

multimodal/tarko/agent-interface/src/agent-event-stream.ts

@@ -154,6 +154,12 @@ export namespace AgentEventStream {
 
     /** Whether the thinking process is complete */
     isComplete?: boolean;
+
+    /**
+     * Unique message identifier that links thinking messages to their session
+     * This allows clients to correlate incremental updates with complete thinking
+     */
+    messageId?: string;
   }
 
   /**
@@ -186,6 +192,12 @@ export namespace AgentEventStream {
 
     /** Whether this is the final reasoning chunk */
     isComplete?: boolean;
+
+    /**
+     * Unique message identifier that links streaming thinking messages to their final thinking
+     * This allows clients to correlate incremental updates with complete thinking
+     */
+    messageId?: string;
   }
 
   /**

multimodal/tarko/agent/src/agent/runner/llm-processor.ts

@@ -337,6 +337,7 @@ export class LLMProcessor {
             {
               content: chunkResult.reasoningContent,
               isComplete: Boolean(processingState.finishReason),
+              messageId: messageId,
             },
           );
           this.eventStream.sendEvent(thinkingEvent);
@@ -482,6 +483,7 @@ export class LLMProcessor {
       const thinkingEvent = this.eventStream.createEvent('assistant_thinking_message', {
         content: reasoningBuffer,
         isComplete: true,
+        messageId: messageId,
       });
 
       this.eventStream.sendEvent(thinkingEvent);