@tanstack/ai 0.8.0 → 0.8.1

package/src/activities/chat/stream/types.ts

@@ -45,6 +45,24 @@ export interface ChunkStrategy {
   reset?: () => void
 }
 
+/**
+ * Per-message streaming state.
+ * Tracks the accumulation of text, tool calls, and thinking content
+ * for a single message in the stream.
+ */
+export interface MessageStreamState {
+  id: string
+  role: 'user' | 'assistant' | 'system'
+  totalTextContent: string
+  currentSegmentText: string
+  lastEmittedText: string
+  thinkingContent: string
+  toolCalls: Map<string, InternalToolCallState>
+  toolCallOrder: Array<string>
+  hasToolCallsSinceTextStart: boolean
+  isComplete: boolean
+}
+
 /**
  * Result from processing a stream
  */

package/src/types.ts

@@ -730,6 +730,7 @@ export type AGUIEventType =
   | 'TOOL_CALL_END'
   | 'STEP_STARTED'
   | 'STEP_FINISHED'
+  | 'MESSAGES_SNAPSHOT'
   | 'STATE_SNAPSHOT'
   | 'STATE_DELTA'
   | 'CUSTOM'
@@ -806,8 +807,8 @@ export interface TextMessageStartEvent extends BaseAGUIEvent {
   type: 'TEXT_MESSAGE_START'
   /** Unique identifier for this message */
   messageId: string
-  /** Role is always assistant for generated messages */
-  role: 'assistant'
+  /** Role of the message sender */
+  role: 'user' | 'assistant' | 'system' | 'tool'
 }
 
 /**
@@ -841,6 +842,8 @@ export interface ToolCallStartEvent extends BaseAGUIEvent {
   toolCallId: string
   /** Name of the tool being called */
   toolName: string
+  /** ID of the parent message that initiated this tool call */
+  parentMessageId?: string
   /** Index for parallel tool calls */
   index?: number
 }
@@ -897,6 +900,19 @@ export interface StepFinishedEvent extends BaseAGUIEvent {
   content?: string
 }
 
+/**
+ * Emitted to provide a snapshot of all messages in a conversation.
+ *
+ * Unlike StateSnapshot (which carries arbitrary application state),
+ * MessagesSnapshot specifically delivers the conversation transcript.
+ * This is a first-class AG-UI event type.
+ */
+export interface MessagesSnapshotEvent extends BaseAGUIEvent {
+  type: 'MESSAGES_SNAPSHOT'
+  /** Complete array of messages in the conversation */
+  messages: Array<UIMessage>
+}
+
 /**
  * Emitted to provide a full state snapshot.
  */
@@ -941,6 +957,7 @@ export type AGUIEvent =
   | ToolCallEndEvent
   | StepStartedEvent
   | StepFinishedEvent
+  | MessagesSnapshotEvent
   | StateSnapshotEvent
   | StateDeltaEvent
   | CustomEvent