@tanstack/ai-client 0.3.0 → 0.4.0

package/dist/esm/chat-client.d.ts

@@ -1,6 +1,6 @@
-import { ChatClientOptions, UIMessage } from './types.js';
 import { AnyClientTool, ModelMessage, StreamChunk } from '@tanstack/ai';
 import { ConnectionAdapter } from './connection-adapters.js';
+import { ChatClientOptions, ChatClientState, UIMessage } from './types.js';
 export declare class ChatClient {
     private processor;
     private connection;
@@ -8,16 +8,20 @@ export declare class ChatClient {
     private body;
     private isLoading;
     private error;
+    private status;
     private abortController;
     private events;
     private clientToolsRef;
     private currentStreamId;
     private currentMessageId;
     private postStreamActions;
+    private pendingToolExecutions;
+    private continuationPending;
     private callbacksRef;
     constructor(options: ChatClientOptions);
     private generateUniqueId;
     private setIsLoading;
+    private setStatus;
     private setError;
     /**
      * Process a stream through the StreamProcessor
@@ -88,6 +92,10 @@ export declare class ChatClient {
      * Get loading state
      */
     getIsLoading(): boolean;
+    /**
+     * Get current status
+     */
+    getStatus(): ChatClientState;
     /**
      * Get current error
      */

package/dist/esm/chat-client.js

@@ -5,10 +5,13 @@ class ChatClient {
     this.body = {};
     this.isLoading = false;
     this.error = void 0;
+    this.status = "ready";
     this.abortController = null;
     this.currentStreamId = null;
     this.currentMessageId = null;
     this.postStreamActions = [];
+    this.pendingToolExecutions = /* @__PURE__ */ new Map();
+    this.continuationPending = false;
     this.uniqueId = options.id || this.generateUniqueId("chat");
     this.body = options.body || {};
     this.connection = options.connection;
@@ -34,6 +37,8 @@ class ChatClient {
         onLoadingChange: options.onLoadingChange || (() => {
         }),
         onErrorChange: options.onErrorChange || (() => {
+        }),
+        onStatusChange: options.onStatusChange || (() => {
         })
       }
     };
@@ -45,12 +50,15 @@ class ChatClient {
           this.callbacksRef.current.onMessagesChange(messages);
         },
         onStreamStart: () => {
+          this.setStatus("streaming");
         },
         onStreamEnd: (message) => {
           this.callbacksRef.current.onFinish(message);
+          this.setStatus("ready");
         },
         onError: (error) => {
           this.setError(error);
+          this.setStatus("error");
           this.callbacksRef.current.onError(error);
         },
         onTextUpdate: (messageId, content) => {
@@ -85,26 +93,32 @@ class ChatClient {
             );
           }
         },
-        onToolCall: async (args) => {
+        onToolCall: (args) => {
           const clientTool = this.clientToolsRef.current.get(args.toolName);
-          if (clientTool?.execute) {
-            try {
-              const output = await clientTool.execute(args.input);
-              await this.addToolResult({
-                toolCallId: args.toolCallId,
-                tool: args.toolName,
-                output,
-                state: "output-available"
-              });
-            } catch (error) {
-              await this.addToolResult({
-                toolCallId: args.toolCallId,
-                tool: args.toolName,
-                output: null,
-                state: "output-error",
-                errorText: error.message
-              });
-            }
+          const executeFunc = clientTool?.execute;
+          if (executeFunc) {
+            const executionPromise = (async () => {
+              try {
+                const output = await executeFunc(args.input);
+                await this.addToolResult({
+                  toolCallId: args.toolCallId,
+                  tool: args.toolName,
+                  output,
+                  state: "output-available"
+                });
+              } catch (error) {
+                await this.addToolResult({
+                  toolCallId: args.toolCallId,
+                  tool: args.toolName,
+                  output: null,
+                  state: "output-error",
+                  errorText: error.message
+                });
+              } finally {
+                this.pendingToolExecutions.delete(args.toolCallId);
+              }
+            })();
+            this.pendingToolExecutions.set(args.toolCallId, executionPromise);
           }
         },
         onApprovalRequest: (args) => {
@@ -131,6 +145,10 @@ class ChatClient {
     this.callbacksRef.current.onLoadingChange(isLoading);
     this.events.loadingChanged(isLoading);
   }
+  setStatus(status) {
+    this.status = status;
+    this.callbacksRef.current.onStatusChange(status);
+  }
   setError(error) {
     this.error = error;
     this.callbacksRef.current.onErrorChange(error);
@@ -158,6 +176,9 @@ class ChatClient {
       this.processor.processChunk(chunk);
       await new Promise((resolve) => setTimeout(resolve, 0));
     }
+    if (this.pendingToolExecutions.size > 0) {
+      await Promise.all(this.pendingToolExecutions.values());
+    }
     this.processor.finalizeStream();
     this.currentStreamId = null;
     this.currentMessageId = null;
@@ -195,40 +216,63 @@ class ChatClient {
     this.events.messageAppended(uiMessage);
     const messages = this.processor.getMessages();
     this.processor.setMessages([...messages, uiMessage]);
+    if (this.isLoading) {
+      this.queuePostStreamAction(() => this.streamResponse());
+      return;
+    }
     await this.streamResponse();
   }
   /**
    * Stream a response from the LLM
    */
   async streamResponse() {
+    if (this.isLoading) {
+      return;
+    }
     this.setIsLoading(true);
+    this.setStatus("submitted");
     this.setError(void 0);
     this.abortController = new AbortController();
+    this.pendingToolExecutions.clear();
+    let streamCompletedSuccessfully = false;
     try {
-      const modelMessages = this.processor.toModelMessages();
+      const messages = this.processor.getMessages();
       await this.callbacksRef.current.onResponse();
       const bodyWithConversationId = {
         ...this.body,
         conversationId: this.uniqueId
       };
       const stream = this.connection.connect(
-        modelMessages,
+        messages,
         bodyWithConversationId,
         this.abortController.signal
       );
       await this.processStream(stream);
+      streamCompletedSuccessfully = true;
     } catch (err) {
       if (err instanceof Error) {
         if (err.name === "AbortError") {
           return;
         }
         this.setError(err);
+        this.setStatus("error");
         this.callbacksRef.current.onError(err);
       }
     } finally {
       this.abortController = null;
       this.setIsLoading(false);
       await this.drainPostStreamActions();
+      if (streamCompletedSuccessfully) {
+        const messages = this.processor.getMessages();
+        const lastPart = messages.at(-1)?.parts.at(-1);
+        if (lastPart?.type === "tool-result" && this.shouldAutoSend()) {
+          try {
+            await this.checkForContinuation();
+          } catch (error) {
+            console.error("Failed to continue flow after tool result:", error);
+          }
+        }
+      }
     }
   }
   /**
@@ -254,6 +298,7 @@ class ChatClient {
       this.abortController = null;
     }
     this.setIsLoading(false);
+    this.setStatus("ready");
     this.events.stopped();
   }
   /**
@@ -333,8 +378,16 @@ class ChatClient {
    * Check if we should continue the flow and do so if needed
    */
   async checkForContinuation() {
+    if (this.continuationPending || this.isLoading) {
+      return;
+    }
     if (this.shouldAutoSend()) {
-      await this.streamResponse();
+      this.continuationPending = true;
+      try {
+        await this.streamResponse();
+      } finally {
+        this.continuationPending = false;
+      }
     }
   }
   /**
@@ -355,6 +408,12 @@ class ChatClient {
   getIsLoading() {
     return this.isLoading;
   }
+  /**
+   * Get current status
+   */
+  getStatus() {
+    return this.status;
+  }
   /**
    * Get current error
    */

package/dist/esm/chat-client.js.map

@@ -1 +1 @@
-{"version":3,"file":"chat-client.js","sources":["../../src/chat-client.ts"],"sourcesContent":["import {\n  StreamProcessor,\n  generateMessageId,\n  normalizeToUIMessage,\n} from '@tanstack/ai'\nimport { DefaultChatClientEventEmitter } from './events'\nimport type {\n  ChatClientOptions,\n  MessagePart,\n  ToolCallPart,\n  UIMessage,\n} from './types'\nimport type { AnyClientTool, ModelMessage, StreamChunk } from '@tanstack/ai'\nimport type { ConnectionAdapter } from './connection-adapters'\nimport type { ChatClientEventEmitter } from './events'\n\nexport class ChatClient {\n  private processor: StreamProcessor\n  private connection: ConnectionAdapter\n  private uniqueId: string\n  private body: Record<string, any> = {}\n  private isLoading = false\n  private error: Error | undefined = undefined\n  private abortController: AbortController | null = null\n  private events: ChatClientEventEmitter\n  private clientToolsRef: { current: Map<string, AnyClientTool> }\n  private currentStreamId: string | null = null\n  private currentMessageId: string | null = null\n  private postStreamActions: Array<() => Promise<void>> = []\n\n  private callbacksRef: {\n    current: {\n      onResponse: (response?: Response) => void | Promise<void>\n      onChunk: (chunk: StreamChunk) => void\n      onFinish: (message: UIMessage) => void\n      onError: (error: Error) => void\n      onMessagesChange: (messages: Array<UIMessage>) => void\n      onLoadingChange: (isLoading: boolean) => void\n      onErrorChange: (error: Error | undefined) => void\n    }\n  }\n\n  constructor(options: ChatClientOptions) {\n    this.uniqueId = options.id || this.generateUniqueId('chat')\n    this.body = options.body || {}\n    this.connection = options.connection\n    this.events = new DefaultChatClientEventEmitter(this.uniqueId)\n\n    // Build client tools map\n    this.clientToolsRef = { current: new Map() }\n    if (options.tools) {\n      for (const tool of options.tools) {\n        this.clientToolsRef.current.set(tool.name, tool)\n      }\n    }\n\n    this.callbacksRef = {\n      current: {\n        onResponse: options.onResponse || (() => {}),\n        onChunk: options.onChunk || (() => {}),\n        onFinish: options.onFinish || (() => {}),\n        onError: options.onError || (() => {}),\n        onMessagesChange: options.onMessagesChange || (() => {}),\n        onLoadingChange: options.onLoadingChange || (() => {}),\n        onErrorChange: options.onErrorChange || (() => {}),\n      },\n    }\n\n    // Create StreamProcessor with event handlers\n    this.processor = new StreamProcessor({\n      chunkStrategy: options.streamProcessor?.chunkStrategy,\n      initialMessages: options.initialMessages,\n      events: {\n        onMessagesChange: (messages: Array<UIMessage>) => {\n          this.callbacksRef.current.onMessagesChange(messages)\n        },\n        onStreamStart: () => {\n          // Stream started\n        },\n        onStreamEnd: (message: UIMessage) => {\n          this.callbacksRef.current.onFinish(message)\n        },\n        onError: (error: Error) => {\n          this.setError(error)\n          this.callbacksRef.current.onError(error)\n        },\n        onTextUpdate: (messageId: string, content: string) => {\n          // Emit text update to devtools\n          if (this.currentStreamId) {\n            this.events.textUpdated(this.currentStreamId, messageId, content)\n          }\n        },\n        onThinkingUpdate: (messageId: string, content: string) => {\n          // Emit thinking update to devtools\n          if (this.currentStreamId) {\n            this.events.thinkingUpdated(\n              this.currentStreamId,\n              messageId,\n              content,\n            )\n          }\n        },\n        onToolCallStateChange: (\n          messageId: string,\n          toolCallId: string,\n          state: string,\n          args: string,\n        ) => {\n          // Get the tool name from the messages\n          const messages = this.processor.getMessages()\n          const message = messages.find((m: UIMessage) => m.id === messageId)\n          const toolCallPart = message?.parts.find(\n            (p: MessagePart): p is ToolCallPart =>\n              p.type === 'tool-call' && p.id === toolCallId,\n          )\n          const toolName = toolCallPart?.name || 'unknown'\n\n          // Emit tool call state change to devtools\n          if (this.currentStreamId) {\n            this.events.toolCallStateChanged(\n              this.currentStreamId,\n              messageId,\n              toolCallId,\n              toolName,\n              state,\n              args,\n            )\n          }\n        },\n        onToolCall: async (args: {\n          toolCallId: string\n          toolName: string\n          input: any\n        }) => {\n          // Handle client-side tool execution automatically\n          const clientTool = this.clientToolsRef.current.get(args.toolName)\n          if (clientTool?.execute) {\n            try {\n              const output = await clientTool.execute(args.input)\n              await this.addToolResult({\n                toolCallId: args.toolCallId,\n                tool: args.toolName,\n                output,\n                state: 'output-available',\n              })\n            } catch (error: any) {\n              await this.addToolResult({\n                toolCallId: args.toolCallId,\n                tool: args.toolName,\n                output: null,\n                state: 'output-error',\n                errorText: error.message,\n              })\n            }\n          }\n        },\n        onApprovalRequest: (args: {\n          toolCallId: string\n          toolName: string\n          input: any\n          approvalId: string\n        }) => {\n          if (this.currentStreamId) {\n            this.events.approvalRequested(\n              this.currentStreamId,\n              this.currentMessageId || '',\n              args.toolCallId,\n              args.toolName,\n              args.input,\n              args.approvalId,\n            )\n          }\n        },\n      },\n    })\n\n    this.events.clientCreated(this.processor.getMessages().length)\n  }\n\n  private generateUniqueId(prefix: string): string {\n    return `${prefix}-${Date.now()}-${Math.random().toString(36).substring(7)}`\n  }\n\n  private setIsLoading(isLoading: boolean): void {\n    this.isLoading = isLoading\n    this.callbacksRef.current.onLoadingChange(isLoading)\n    this.events.loadingChanged(isLoading)\n  }\n\n  private setError(error: Error | undefined): void {\n    this.error = error\n    this.callbacksRef.current.onErrorChange(error)\n    this.events.errorChanged(error?.message || null)\n  }\n\n  /**\n   * Process a stream through the StreamProcessor\n   */\n  private async processStream(\n    source: AsyncIterable<StreamChunk>,\n  ): Promise<UIMessage> {\n    // Generate a stream ID for this streaming operation\n    this.currentStreamId = this.generateUniqueId('stream')\n\n    // Start a new assistant message\n    const messageId = this.processor.startAssistantMessage()\n    this.currentMessageId = messageId\n\n    // Emit message appended event for the new assistant message\n    const assistantMessage: UIMessage = {\n      id: messageId,\n      role: 'assistant',\n      parts: [],\n      createdAt: new Date(),\n    }\n    this.events.messageAppended(\n      assistantMessage,\n      this.currentStreamId || undefined,\n    )\n\n    // Process each chunk\n    for await (const chunk of source) {\n      this.callbacksRef.current.onChunk(chunk)\n      this.processor.processChunk(chunk)\n\n      // Yield control back to event loop to allow UI updates\n      await new Promise((resolve) => setTimeout(resolve, 0))\n    }\n\n    // Finalize the stream\n    this.processor.finalizeStream()\n\n    // Clear the current stream and message IDs\n    this.currentStreamId = null\n    this.currentMessageId = null\n\n    // Return the assistant message\n    const messages = this.processor.getMessages()\n    const finalAssistantMessage = messages.find(\n      (m: UIMessage) => m.id === messageId,\n    )\n\n    return (\n      finalAssistantMessage || {\n        id: messageId,\n        role: 'assistant',\n        parts: [],\n        createdAt: new Date(),\n      }\n    )\n  }\n\n  /**\n   * Send a message and stream the response\n   */\n  async sendMessage(content: string): Promise<void> {\n    if (!content.trim() || this.isLoading) {\n      return\n    }\n\n    // Add user message via processor\n    const userMessage = this.processor.addUserMessage(content.trim())\n    this.events.messageSent(userMessage.id, content.trim())\n\n    await this.streamResponse()\n  }\n\n  /**\n   * Append a message and stream the response\n   */\n  async append(message: UIMessage | ModelMessage): Promise<void> {\n    // Normalize the message to ensure it has id and createdAt\n    const normalizedMessage = normalizeToUIMessage(message, generateMessageId)\n\n    // Skip system messages - they're handled via systemPrompts, not UIMessages\n    if (normalizedMessage.role === 'system') {\n      return\n    }\n\n    // Type assertion: after checking for system, we know it's user or assistant\n    const uiMessage = normalizedMessage as UIMessage\n\n    // Emit message appended event\n    this.events.messageAppended(uiMessage)\n\n    // Add to messages\n    const messages = this.processor.getMessages()\n    this.processor.setMessages([...messages, uiMessage])\n\n    await this.streamResponse()\n  }\n\n  /**\n   * Stream a response from the LLM\n   */\n  private async streamResponse(): Promise<void> {\n    this.setIsLoading(true)\n    this.setError(undefined)\n    this.abortController = new AbortController()\n\n    try {\n      // Get model messages for the LLM\n      const modelMessages = this.processor.toModelMessages()\n\n      // Call onResponse callback\n      await this.callbacksRef.current.onResponse()\n\n      // Include conversationId in the body for server-side event correlation\n      const bodyWithConversationId = {\n        ...this.body,\n        conversationId: this.uniqueId,\n      }\n\n      // Connect and stream\n      const stream = this.connection.connect(\n        modelMessages,\n        bodyWithConversationId,\n        this.abortController.signal,\n      )\n\n      await this.processStream(stream)\n    } catch (err) {\n      if (err instanceof Error) {\n        if (err.name === 'AbortError') {\n          return\n        }\n        this.setError(err)\n        this.callbacksRef.current.onError(err)\n      }\n    } finally {\n      this.abortController = null\n      this.setIsLoading(false)\n\n      // Drain any actions that were queued while the stream was in progress\n      await this.drainPostStreamActions()\n    }\n  }\n\n  /**\n   * Reload the last assistant message\n   */\n  async reload(): Promise<void> {\n    const messages = this.processor.getMessages()\n    if (messages.length === 0) return\n\n    // Find the last user message\n    const lastUserMessageIndex = messages.findLastIndex(\n      (m: UIMessage) => m.role === 'user',\n    )\n\n    if (lastUserMessageIndex === -1) return\n\n    this.events.reloaded(lastUserMessageIndex)\n\n    // Remove all messages after the last user message\n    this.processor.removeMessagesAfter(lastUserMessageIndex)\n\n    // Resend\n    await this.streamResponse()\n  }\n\n  /**\n   * Stop the current stream\n   */\n  stop(): void {\n    if (this.abortController) {\n      this.abortController.abort()\n      this.abortController = null\n    }\n    this.setIsLoading(false)\n    this.events.stopped()\n  }\n\n  /**\n   * Clear all messages\n   */\n  clear(): void {\n    this.processor.clearMessages()\n    this.setError(undefined)\n    this.events.messagesCleared()\n  }\n\n  /**\n   * Add the result of a client-side tool execution\n   */\n  async addToolResult(result: {\n    toolCallId: string\n    tool: string\n    output: any\n    state?: 'output-available' | 'output-error'\n    errorText?: string\n  }): Promise<void> {\n    this.events.toolResultAdded(\n      result.toolCallId,\n      result.tool,\n      result.output,\n      result.state || 'output-available',\n    )\n\n    // Add result via processor\n    this.processor.addToolResult(\n      result.toolCallId,\n      result.output,\n      result.errorText,\n    )\n\n    // If stream is in progress, queue continuation check for after it ends\n    if (this.isLoading) {\n      this.queuePostStreamAction(() => this.checkForContinuation())\n      return\n    }\n\n    await this.checkForContinuation()\n  }\n\n  /**\n   * Respond to a tool approval request\n   */\n  async addToolApprovalResponse(response: {\n    id: string // approval.id, not toolCallId\n    approved: boolean\n  }): Promise<void> {\n    // Find the tool call ID from the approval ID\n    const messages = this.processor.getMessages()\n    let foundToolCallId: string | undefined\n\n    for (const msg of messages) {\n      const toolCallPart = msg.parts.find(\n        (p: MessagePart): p is ToolCallPart =>\n          p.type === 'tool-call' && p.approval?.id === response.id,\n      )\n      if (toolCallPart) {\n        foundToolCallId = toolCallPart.id\n        break\n      }\n    }\n\n    if (foundToolCallId) {\n      this.events.toolApprovalResponded(\n        response.id,\n        foundToolCallId,\n        response.approved,\n      )\n    }\n\n    // Add response via processor\n    this.processor.addToolApprovalResponse(response.id, response.approved)\n\n    // If stream is in progress, queue continuation check for after it ends\n    if (this.isLoading) {\n      this.queuePostStreamAction(() => this.checkForContinuation())\n      return\n    }\n\n    await this.checkForContinuation()\n  }\n\n  /**\n   * Queue an action to be executed after the current stream ends\n   */\n  private queuePostStreamAction(action: () => Promise<void>): void {\n    this.postStreamActions.push(action)\n  }\n\n  /**\n   * Drain and execute all queued post-stream actions\n   */\n  private async drainPostStreamActions(): Promise<void> {\n    while (this.postStreamActions.length > 0) {\n      const action = this.postStreamActions.shift()!\n      await action()\n    }\n  }\n\n  /**\n   * Check if we should continue the flow and do so if needed\n   */\n  private async checkForContinuation(): Promise<void> {\n    if (this.shouldAutoSend()) {\n      await this.streamResponse()\n    }\n  }\n\n  /**\n   * Check if all tool calls are complete and we should auto-send\n   */\n  private shouldAutoSend(): boolean {\n    return this.processor.areAllToolsComplete()\n  }\n\n  /**\n   * Get current messages\n   */\n  getMessages(): Array<UIMessage> {\n    return this.processor.getMessages()\n  }\n\n  /**\n   * Get loading state\n   */\n  getIsLoading(): boolean {\n    return this.isLoading\n  }\n\n  /**\n   * Get current error\n   */\n  getError(): Error | undefined {\n    return this.error\n  }\n\n  /**\n   * Manually set messages\n   */\n  setMessagesManually(messages: Array<UIMessage>): void {\n    this.processor.setMessages(messages)\n  }\n\n  /**\n   * Update options refs (for use in React hooks to avoid recreating client)\n   */\n  updateOptions(options: {\n    connection?: ConnectionAdapter\n    body?: Record<string, any>\n    tools?: ReadonlyArray<AnyClientTool>\n    onResponse?: (response?: Response) => void | Promise<void>\n    onChunk?: (chunk: StreamChunk) => void\n    onFinish?: (message: UIMessage) => void\n    onError?: (error: Error) => void\n  }): void {\n    if (options.connection !== undefined) {\n      this.connection = options.connection\n    }\n    if (options.body !== undefined) {\n      this.body = options.body\n    }\n    if (options.tools !== undefined) {\n      this.clientToolsRef.current = new Map()\n      for (const tool of options.tools) {\n        this.clientToolsRef.current.set(tool.name, tool)\n      }\n    }\n    if (options.onResponse !== undefined) {\n      this.callbacksRef.current.onResponse = options.onResponse\n    }\n    if (options.onChunk !== undefined) {\n      this.callbacksRef.current.onChunk = options.onChunk\n    }\n    if (options.onFinish !== undefined) {\n      this.callbacksRef.current.onFinish = options.onFinish\n    }\n    if (options.onError !== undefined) {\n      this.callbacksRef.current.onError = options.onError\n    }\n  }\n}\n"],"names":[],"mappings":";;AAgBO,MAAM,WAAW;AAAA,EA0BtB,YAAY,SAA4B;AAtBxC,SAAQ,OAA4B,CAAA;AACpC,SAAQ,YAAY;AACpB,SAAQ,QAA2B;AACnC,SAAQ,kBAA0C;AAGlD,SAAQ,kBAAiC;AACzC,SAAQ,mBAAkC;AAC1C,SAAQ,oBAAgD,CAAA;AAetD,SAAK,WAAW,QAAQ,MAAM,KAAK,iBAAiB,MAAM;AAC1D,SAAK,OAAO,QAAQ,QAAQ,CAAA;AAC5B,SAAK,aAAa,QAAQ;AAC1B,SAAK,SAAS,IAAI,8BAA8B,KAAK,QAAQ;AAG7D,SAAK,iBAAiB,EAAE,SAAS,oBAAI,MAAI;AACzC,QAAI,QAAQ,OAAO;AACjB,iBAAW,QAAQ,QAAQ,OAAO;AAChC,aAAK,eAAe,QAAQ,IAAI,KAAK,MAAM,IAAI;AAAA,MACjD;AAAA,IACF;AAEA,SAAK,eAAe;AAAA,MAClB,SAAS;AAAA,QACP,YAAY,QAAQ,eAAe,MAAM;AAAA,QAAC;AAAA,QAC1C,SAAS,QAAQ,YAAY,MAAM;AAAA,QAAC;AAAA,QACpC,UAAU,QAAQ,aAAa,MAAM;AAAA,QAAC;AAAA,QACtC,SAAS,QAAQ,YAAY,MAAM;AAAA,QAAC;AAAA,QACpC,kBAAkB,QAAQ,qBAAqB,MAAM;AAAA,QAAC;AAAA,QACtD,iBAAiB,QAAQ,oBAAoB,MAAM;AAAA,QAAC;AAAA,QACpD,eAAe,QAAQ,kBAAkB,MAAM;AAAA,QAAC;AAAA,MAAA;AAAA,IAClD;AAIF,SAAK,YAAY,IAAI,gBAAgB;AAAA,MACnC,eAAe,QAAQ,iBAAiB;AAAA,MACxC,iBAAiB,QAAQ;AAAA,MACzB,QAAQ;AAAA,QACN,kBAAkB,CAAC,aAA+B;AAChD,eAAK,aAAa,QAAQ,iBAAiB,QAAQ;AAAA,QACrD;AAAA,QACA,eAAe,MAAM;AAAA,QAErB;AAAA,QACA,aAAa,CAAC,YAAuB;AACnC,eAAK,aAAa,QAAQ,SAAS,OAAO;AAAA,QAC5C;AAAA,QACA,SAAS,CAAC,UAAiB;AACzB,eAAK,SAAS,KAAK;AACnB,eAAK,aAAa,QAAQ,QAAQ,KAAK;AAAA,QACzC;AAAA,QACA,cAAc,CAAC,WAAmB,YAAoB;AAEpD,cAAI,KAAK,iBAAiB;AACxB,iBAAK,OAAO,YAAY,KAAK,iBAAiB,WAAW,OAAO;AAAA,UAClE;AAAA,QACF;AAAA,QACA,kBAAkB,CAAC,WAAmB,YAAoB;AAExD,cAAI,KAAK,iBAAiB;AACxB,iBAAK,OAAO;AAAA,cACV,KAAK;AAAA,cACL;AAAA,cACA;AAAA,YAAA;AAAA,UAEJ;AAAA,QACF;AAAA,QACA,uBAAuB,CACrB,WACA,YACA,OACA,SACG;AAEH,gBAAM,WAAW,KAAK,UAAU,YAAA;AAChC,gBAAM,UAAU,SAAS,KAAK,CAAC,MAAiB,EAAE,OAAO,SAAS;AAClE,gBAAM,eAAe,SAAS,MAAM;AAAA,YAClC,CAAC,MACC,EAAE,SAAS,eAAe,EAAE,OAAO;AAAA,UAAA;AAEvC,gBAAM,WAAW,cAAc,QAAQ;AAGvC,cAAI,KAAK,iBAAiB;AACxB,iBAAK,OAAO;AAAA,cACV,KAAK;AAAA,cACL;AAAA,cACA;AAAA,cACA;AAAA,cACA;AAAA,cACA;AAAA,YAAA;AAAA,UAEJ;AAAA,QACF;AAAA,QACA,YAAY,OAAO,SAIb;AAEJ,gBAAM,aAAa,KAAK,eAAe,QAAQ,IAAI,KAAK,QAAQ;AAChE,cAAI,YAAY,SAAS;AACvB,gBAAI;AACF,oBAAM,SAAS,MAAM,WAAW,QAAQ,KAAK,KAAK;AAClD,oBAAM,KAAK,cAAc;AAAA,gBACvB,YAAY,KAAK;AAAA,gBACjB,MAAM,KAAK;AAAA,gBACX;AAAA,gBACA,OAAO;AAAA,cAAA,CACR;AAAA,YACH,SAAS,OAAY;AACnB,oBAAM,KAAK,cAAc;AAAA,gBACvB,YAAY,KAAK;AAAA,gBACjB,MAAM,KAAK;AAAA,gBACX,QAAQ;AAAA,gBACR,OAAO;AAAA,gBACP,WAAW,MAAM;AAAA,cAAA,CAClB;AAAA,YACH;AAAA,UACF;AAAA,QACF;AAAA,QACA,mBAAmB,CAAC,SAKd;AACJ,cAAI,KAAK,iBAAiB;AACxB,iBAAK,OAAO;AAAA,cACV,KAAK;AAAA,cACL,KAAK,oBAAoB;AAAA,cACzB,KAAK;AAAA,cACL,KAAK;AAAA,cACL,KAAK;AAAA,cACL,KAAK;AAAA,YAAA;AAAA,UAET;AAAA,QACF;AAAA,MAAA;AAAA,IACF,CACD;AAED,SAAK,OAAO,cAAc,KAAK,UAAU,YAAA,EAAc,MAAM;AAAA,EAC/D;AAAA,EAEQ,iBAAiB,QAAwB;AAC/C,WAAO,GAAG,MAAM,IAAI,KAAK,KAAK,IAAI,KAAK,OAAA,EAAS,SAAS,EAAE,EAAE,UAAU,CAAC,CAAC;AAAA,EAC3E;AAAA,EAEQ,aAAa,WAA0B;AAC7C,SAAK,YAAY;AACjB,SAAK,aAAa,QAAQ,gBAAgB,SAAS;AACnD,SAAK,OAAO,eAAe,SAAS;AAAA,EACtC;AAAA,EAEQ,SAAS,OAAgC;AAC/C,SAAK,QAAQ;AACb,SAAK,aAAa,QAAQ,cAAc,KAAK;AAC7C,SAAK,OAAO,aAAa,OAAO,WAAW,IAAI;AAAA,EACjD;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,cACZ,QACoB;AAEpB,SAAK,kBAAkB,KAAK,iBAAiB,QAAQ;AAGrD,UAAM,YAAY,KAAK,UAAU,sBAAA;AACjC,SAAK,mBAAmB;AAGxB,UAAM,mBAA8B;AAAA,MAClC,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,OAAO,CAAA;AAAA,MACP,+BAAe,KAAA;AAAA,IAAK;AAEtB,SAAK,OAAO;AAAA,MACV;AAAA,MACA,KAAK,mBAAmB;AAAA,IAAA;AAI1B,qBAAiB,SAAS,QAAQ;AAChC,WAAK,aAAa,QAAQ,QAAQ,KAAK;AACvC,WAAK,UAAU,aAAa,KAAK;AAGjC,YAAM,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,CAAC,CAAC;AAAA,IACvD;AAGA,SAAK,UAAU,eAAA;AAGf,SAAK,kBAAkB;AACvB,SAAK,mBAAmB;AAGxB,UAAM,WAAW,KAAK,UAAU,YAAA;AAChC,UAAM,wBAAwB,SAAS;AAAA,MACrC,CAAC,MAAiB,EAAE,OAAO;AAAA,IAAA;AAG7B,WACE,yBAAyB;AAAA,MACvB,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,OAAO,CAAA;AAAA,MACP,+BAAe,KAAA;AAAA,IAAK;AAAA,EAG1B;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,YAAY,SAAgC;AAChD,QAAI,CAAC,QAAQ,UAAU,KAAK,WAAW;AACrC;AAAA,IACF;AAGA,UAAM,cAAc,KAAK,UAAU,eAAe,QAAQ,MAAM;AAChE,SAAK,OAAO,YAAY,YAAY,IAAI,QAAQ,MAAM;AAEtD,UAAM,KAAK,eAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,OAAO,SAAkD;AAE7D,UAAM,oBAAoB,qBAAqB,SAAS,iBAAiB;AAGzE,QAAI,kBAAkB,SAAS,UAAU;AACvC;AAAA,IACF;AAGA,UAAM,YAAY;AAGlB,SAAK,OAAO,gBAAgB,SAAS;AAGrC,UAAM,WAAW,KAAK,UAAU,YAAA;AAChC,SAAK,UAAU,YAAY,CAAC,GAAG,UAAU,SAAS,CAAC;AAEnD,UAAM,KAAK,eAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,iBAAgC;AAC5C,SAAK,aAAa,IAAI;AACtB,SAAK,SAAS,MAAS;AACvB,SAAK,kBAAkB,IAAI,gBAAA;AAE3B,QAAI;AAEF,YAAM,gBAAgB,KAAK,UAAU,gBAAA;AAGrC,YAAM,KAAK,aAAa,QAAQ,WAAA;AAGhC,YAAM,yBAAyB;AAAA,QAC7B,GAAG,KAAK;AAAA,QACR,gBAAgB,KAAK;AAAA,MAAA;AAIvB,YAAM,SAAS,KAAK,WAAW;AAAA,QAC7B;AAAA,QACA;AAAA,QACA,KAAK,gBAAgB;AAAA,MAAA;AAGvB,YAAM,KAAK,cAAc,MAAM;AAAA,IACjC,SAAS,KAAK;AACZ,UAAI,eAAe,OAAO;AACxB,YAAI,IAAI,SAAS,cAAc;AAC7B;AAAA,QACF;AACA,aAAK,SAAS,GAAG;AACjB,aAAK,aAAa,QAAQ,QAAQ,GAAG;AAAA,MACvC;AAAA,IACF,UAAA;AACE,WAAK,kBAAkB;AACvB,WAAK,aAAa,KAAK;AAGvB,YAAM,KAAK,uBAAA;AAAA,IACb;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,SAAwB;AAC5B,UAAM,WAAW,KAAK,UAAU,YAAA;AAChC,QAAI,SAAS,WAAW,EAAG;AAG3B,UAAM,uBAAuB,SAAS;AAAA,MACpC,CAAC,MAAiB,EAAE,SAAS;AAAA,IAAA;AAG/B,QAAI,yBAAyB,GAAI;AAEjC,SAAK,OAAO,SAAS,oBAAoB;AAGzC,SAAK,UAAU,oBAAoB,oBAAoB;AAGvD,UAAM,KAAK,eAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA,EAKA,OAAa;AACX,QAAI,KAAK,iBAAiB;AACxB,WAAK,gBAAgB,MAAA;AACrB,WAAK,kBAAkB;AAAA,IACzB;AACA,SAAK,aAAa,KAAK;AACvB,SAAK,OAAO,QAAA;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAKA,QAAc;AACZ,SAAK,UAAU,cAAA;AACf,SAAK,SAAS,MAAS;AACvB,SAAK,OAAO,gBAAA;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,cAAc,QAMF;AAChB,SAAK,OAAO;AAAA,MACV,OAAO;AAAA,MACP,OAAO;AAAA,MACP,OAAO;AAAA,MACP,OAAO,SAAS;AAAA,IAAA;AAIlB,SAAK,UAAU;AAAA,MACb,OAAO;AAAA,MACP,OAAO;AAAA,MACP,OAAO;AAAA,IAAA;AAIT,QAAI,KAAK,WAAW;AAClB,WAAK,sBAAsB,MAAM,KAAK,qBAAA,CAAsB;AAC5D;AAAA,IACF;AAEA,UAAM,KAAK,qBAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,wBAAwB,UAGZ;AAEhB,UAAM,WAAW,KAAK,UAAU,YAAA;AAChC,QAAI;AAEJ,eAAW,OAAO,UAAU;AAC1B,YAAM,eAAe,IAAI,MAAM;AAAA,QAC7B,CAAC,MACC,EAAE,SAAS,eAAe,EAAE,UAAU,OAAO,SAAS;AAAA,MAAA;AAE1D,UAAI,cAAc;AAChB,0BAAkB,aAAa;AAC/B;AAAA,MACF;AAAA,IACF;AAEA,QAAI,iBAAiB;AACnB,WAAK,OAAO;AAAA,QACV,SAAS;AAAA,QACT;AAAA,QACA,SAAS;AAAA,MAAA;AAAA,IAEb;AAGA,SAAK,UAAU,wBAAwB,SAAS,IAAI,SAAS,QAAQ;AAGrE,QAAI,KAAK,WAAW;AAClB,WAAK,sBAAsB,MAAM,KAAK,qBAAA,CAAsB;AAC5D;AAAA,IACF;AAEA,UAAM,KAAK,qBAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA,EAKQ,sBAAsB,QAAmC;AAC/D,SAAK,kBAAkB,KAAK,MAAM;AAAA,EACpC;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,yBAAwC;AACpD,WAAO,KAAK,kBAAkB,SAAS,GAAG;AACxC,YAAM,SAAS,KAAK,kBAAkB,MAAA;AACtC,YAAM,OAAA;AAAA,IACR;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,uBAAsC;AAClD,QAAI,KAAK,kBAAkB;AACzB,YAAM,KAAK,eAAA;AAAA,IACb;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,iBAA0B;AAChC,WAAO,KAAK,UAAU,oBAAA;AAAA,EACxB;AAAA;AAAA;AAAA;AAAA,EAKA,cAAgC;AAC9B,WAAO,KAAK,UAAU,YAAA;AAAA,EACxB;AAAA;AAAA;AAAA;AAAA,EAKA,eAAwB;AACtB,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAKA,WAA8B;AAC5B,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAKA,oBAAoB,UAAkC;AACpD,SAAK,UAAU,YAAY,QAAQ;AAAA,EACrC;AAAA;AAAA;AAAA;AAAA,EAKA,cAAc,SAQL;AACP,QAAI,QAAQ,eAAe,QAAW;AACpC,WAAK,aAAa,QAAQ;AAAA,IAC5B;AACA,QAAI,QAAQ,SAAS,QAAW;AAC9B,WAAK,OAAO,QAAQ;AAAA,IACtB;AACA,QAAI,QAAQ,UAAU,QAAW;AAC/B,WAAK,eAAe,UAAU,oBAAI,IAAA;AAClC,iBAAW,QAAQ,QAAQ,OAAO;AAChC,aAAK,eAAe,QAAQ,IAAI,KAAK,MAAM,IAAI;AAAA,MACjD;AAAA,IACF;AACA,QAAI,QAAQ,eAAe,QAAW;AACpC,WAAK,aAAa,QAAQ,aAAa,QAAQ;AAAA,IACjD;AACA,QAAI,QAAQ,YAAY,QAAW;AACjC,WAAK,aAAa,QAAQ,UAAU,QAAQ;AAAA,IAC9C;AACA,QAAI,QAAQ,aAAa,QAAW;AAClC,WAAK,aAAa,QAAQ,WAAW,QAAQ;AAAA,IAC/C;AACA,QAAI,QAAQ,YAAY,QAAW;AACjC,WAAK,aAAa,QAAQ,UAAU,QAAQ;AAAA,IAC9C;AAAA,EACF;AACF;"}
+{"version":3,"file":"chat-client.js","sources":["../../src/chat-client.ts"],"sourcesContent":["import {\n  StreamProcessor,\n  generateMessageId,\n  normalizeToUIMessage,\n} from '@tanstack/ai'\nimport { DefaultChatClientEventEmitter } from './events'\nimport type { AnyClientTool, ModelMessage, StreamChunk } from '@tanstack/ai'\nimport type { ConnectionAdapter } from './connection-adapters'\nimport type { ChatClientEventEmitter } from './events'\nimport type {\n  ChatClientOptions,\n  ChatClientState,\n  MessagePart,\n  ToolCallPart,\n  UIMessage,\n} from './types'\n\nexport class ChatClient {\n  private processor: StreamProcessor\n  private connection: ConnectionAdapter\n  private uniqueId: string\n  private body: Record<string, any> = {}\n  private isLoading = false\n  private error: Error | undefined = undefined\n  private status: ChatClientState = 'ready'\n  private abortController: AbortController | null = null\n  private events: ChatClientEventEmitter\n  private clientToolsRef: { current: Map<string, AnyClientTool> }\n  private currentStreamId: string | null = null\n  private currentMessageId: string | null = null\n  private postStreamActions: Array<() => Promise<void>> = []\n  // Track pending client tool executions to await them before stream finalization\n  private pendingToolExecutions: Map<string, Promise<void>> = new Map()\n  // Flag to deduplicate continuation checks during action draining\n  private continuationPending = false\n\n  private callbacksRef: {\n    current: {\n      onResponse: (response?: Response) => void | Promise<void>\n      onChunk: (chunk: StreamChunk) => void\n      onFinish: (message: UIMessage) => void\n      onError: (error: Error) => void\n      onMessagesChange: (messages: Array<UIMessage>) => void\n      onLoadingChange: (isLoading: boolean) => void\n      onErrorChange: (error: Error | undefined) => void\n      onStatusChange: (status: ChatClientState) => void\n    }\n  }\n\n  constructor(options: ChatClientOptions) {\n    this.uniqueId = options.id || this.generateUniqueId('chat')\n    this.body = options.body || {}\n    this.connection = options.connection\n    this.events = new DefaultChatClientEventEmitter(this.uniqueId)\n\n    // Build client tools map\n    this.clientToolsRef = { current: new Map() }\n    if (options.tools) {\n      for (const tool of options.tools) {\n        this.clientToolsRef.current.set(tool.name, tool)\n      }\n    }\n\n    this.callbacksRef = {\n      current: {\n        onResponse: options.onResponse || (() => {}),\n        onChunk: options.onChunk || (() => {}),\n        onFinish: options.onFinish || (() => {}),\n        onError: options.onError || (() => {}),\n        onMessagesChange: options.onMessagesChange || (() => {}),\n        onLoadingChange: options.onLoadingChange || (() => {}),\n        onErrorChange: options.onErrorChange || (() => {}),\n        onStatusChange: options.onStatusChange || (() => {}),\n      },\n    }\n\n    // Create StreamProcessor with event handlers\n    this.processor = new StreamProcessor({\n      chunkStrategy: options.streamProcessor?.chunkStrategy,\n      initialMessages: options.initialMessages,\n      events: {\n        onMessagesChange: (messages: Array<UIMessage>) => {\n          this.callbacksRef.current.onMessagesChange(messages)\n        },\n        onStreamStart: () => {\n          this.setStatus('streaming')\n        },\n        onStreamEnd: (message: UIMessage) => {\n          this.callbacksRef.current.onFinish(message)\n          this.setStatus('ready')\n        },\n        onError: (error: Error) => {\n          this.setError(error)\n          this.setStatus('error')\n          this.callbacksRef.current.onError(error)\n        },\n        onTextUpdate: (messageId: string, content: string) => {\n          // Emit text update to devtools\n          if (this.currentStreamId) {\n            this.events.textUpdated(this.currentStreamId, messageId, content)\n          }\n        },\n        onThinkingUpdate: (messageId: string, content: string) => {\n          // Emit thinking update to devtools\n          if (this.currentStreamId) {\n            this.events.thinkingUpdated(\n              this.currentStreamId,\n              messageId,\n              content,\n            )\n          }\n        },\n        onToolCallStateChange: (\n          messageId: string,\n          toolCallId: string,\n          state: string,\n          args: string,\n        ) => {\n          // Get the tool name from the messages\n          const messages = this.processor.getMessages()\n          const message = messages.find((m: UIMessage) => m.id === messageId)\n          const toolCallPart = message?.parts.find(\n            (p: MessagePart): p is ToolCallPart =>\n              p.type === 'tool-call' && p.id === toolCallId,\n          )\n          const toolName = toolCallPart?.name || 'unknown'\n\n          // Emit tool call state change to devtools\n          if (this.currentStreamId) {\n            this.events.toolCallStateChanged(\n              this.currentStreamId,\n              messageId,\n              toolCallId,\n              toolName,\n              state,\n              args,\n            )\n          }\n        },\n        onToolCall: (args: {\n          toolCallId: string\n          toolName: string\n          input: any\n        }) => {\n          // Handle client-side tool execution automatically\n          const clientTool = this.clientToolsRef.current.get(args.toolName)\n          const executeFunc = clientTool?.execute\n          if (executeFunc) {\n            // Create and track the execution promise\n            const executionPromise = (async () => {\n              try {\n                const output = await executeFunc(args.input)\n                await this.addToolResult({\n                  toolCallId: args.toolCallId,\n                  tool: args.toolName,\n                  output,\n                  state: 'output-available',\n                })\n              } catch (error: any) {\n                await this.addToolResult({\n                  toolCallId: args.toolCallId,\n                  tool: args.toolName,\n                  output: null,\n                  state: 'output-error',\n                  errorText: error.message,\n                })\n              } finally {\n                // Remove from pending when complete\n                this.pendingToolExecutions.delete(args.toolCallId)\n              }\n            })()\n\n            // Track the pending execution\n            this.pendingToolExecutions.set(args.toolCallId, executionPromise)\n          }\n        },\n        onApprovalRequest: (args: {\n          toolCallId: string\n          toolName: string\n          input: any\n          approvalId: string\n        }) => {\n          if (this.currentStreamId) {\n            this.events.approvalRequested(\n              this.currentStreamId,\n              this.currentMessageId || '',\n              args.toolCallId,\n              args.toolName,\n              args.input,\n              args.approvalId,\n            )\n          }\n        },\n      },\n    })\n\n    this.events.clientCreated(this.processor.getMessages().length)\n  }\n\n  private generateUniqueId(prefix: string): string {\n    return `${prefix}-${Date.now()}-${Math.random().toString(36).substring(7)}`\n  }\n\n  private setIsLoading(isLoading: boolean): void {\n    this.isLoading = isLoading\n    this.callbacksRef.current.onLoadingChange(isLoading)\n    this.events.loadingChanged(isLoading)\n  }\n\n  private setStatus(status: ChatClientState): void {\n    this.status = status\n    this.callbacksRef.current.onStatusChange(status)\n  }\n\n  private setError(error: Error | undefined): void {\n    this.error = error\n    this.callbacksRef.current.onErrorChange(error)\n    this.events.errorChanged(error?.message || null)\n  }\n\n  /**\n   * Process a stream through the StreamProcessor\n   */\n  private async processStream(\n    source: AsyncIterable<StreamChunk>,\n  ): Promise<UIMessage> {\n    // Generate a stream ID for this streaming operation\n    this.currentStreamId = this.generateUniqueId('stream')\n\n    // Start a new assistant message\n    const messageId = this.processor.startAssistantMessage()\n    this.currentMessageId = messageId\n\n    // Emit message appended event for the new assistant message\n    const assistantMessage: UIMessage = {\n      id: messageId,\n      role: 'assistant',\n      parts: [],\n      createdAt: new Date(),\n    }\n    this.events.messageAppended(\n      assistantMessage,\n      this.currentStreamId || undefined,\n    )\n\n    // Process each chunk\n    for await (const chunk of source) {\n      this.callbacksRef.current.onChunk(chunk)\n      this.processor.processChunk(chunk)\n\n      // Yield control back to event loop to allow UI updates\n      await new Promise((resolve) => setTimeout(resolve, 0))\n    }\n\n    // Wait for all pending tool executions to complete before finalizing\n    // This ensures client tools finish before we check for continuation\n    if (this.pendingToolExecutions.size > 0) {\n      await Promise.all(this.pendingToolExecutions.values())\n    }\n\n    // Finalize the stream\n    this.processor.finalizeStream()\n\n    // Clear the current stream and message IDs\n    this.currentStreamId = null\n    this.currentMessageId = null\n\n    // Return the assistant message\n    const messages = this.processor.getMessages()\n    const finalAssistantMessage = messages.find(\n      (m: UIMessage) => m.id === messageId,\n    )\n\n    return (\n      finalAssistantMessage || {\n        id: messageId,\n        role: 'assistant',\n        parts: [],\n        createdAt: new Date(),\n      }\n    )\n  }\n\n  /**\n   * Send a message and stream the response\n   */\n  async sendMessage(content: string): Promise<void> {\n    if (!content.trim() || this.isLoading) {\n      return\n    }\n\n    // Add user message via processor\n    const userMessage = this.processor.addUserMessage(content.trim())\n    this.events.messageSent(userMessage.id, content.trim())\n\n    await this.streamResponse()\n  }\n\n  /**\n   * Append a message and stream the response\n   */\n  async append(message: UIMessage | ModelMessage): Promise<void> {\n    // Normalize the message to ensure it has id and createdAt\n    const normalizedMessage = normalizeToUIMessage(message, generateMessageId)\n\n    // Skip system messages - they're handled via systemPrompts, not UIMessages\n    if (normalizedMessage.role === 'system') {\n      return\n    }\n\n    // Type assertion: after checking for system, we know it's user or assistant\n    const uiMessage = normalizedMessage as UIMessage\n\n    // Emit message appended event\n    this.events.messageAppended(uiMessage)\n\n    // Add to messages\n    const messages = this.processor.getMessages()\n    this.processor.setMessages([...messages, uiMessage])\n\n    // If stream is in progress, queue the response for after it ends\n    if (this.isLoading) {\n      this.queuePostStreamAction(() => this.streamResponse())\n      return\n    }\n\n    await this.streamResponse()\n  }\n\n  /**\n   * Stream a response from the LLM\n   */\n  private async streamResponse(): Promise<void> {\n    // Guard against concurrent streams - if already loading, skip\n    if (this.isLoading) {\n      return\n    }\n\n    this.setIsLoading(true)\n    this.setStatus('submitted')\n    this.setError(undefined)\n    this.abortController = new AbortController()\n    // Reset pending tool executions for the new stream\n    this.pendingToolExecutions.clear()\n    let streamCompletedSuccessfully = false\n\n    try {\n      // Get UIMessages with parts (preserves approval state and client tool results)\n      const messages = this.processor.getMessages()\n\n      // Call onResponse callback\n      await this.callbacksRef.current.onResponse()\n\n      // Include conversationId in the body for server-side event correlation\n      const bodyWithConversationId = {\n        ...this.body,\n        conversationId: this.uniqueId,\n      }\n\n      // Connect and stream\n      const stream = this.connection.connect(\n        messages,\n        bodyWithConversationId,\n        this.abortController.signal,\n      )\n\n      await this.processStream(stream)\n      streamCompletedSuccessfully = true\n    } catch (err) {\n      if (err instanceof Error) {\n        if (err.name === 'AbortError') {\n          return\n        }\n        this.setError(err)\n        this.setStatus('error')\n        this.callbacksRef.current.onError(err)\n      }\n    } finally {\n      this.abortController = null\n      this.setIsLoading(false)\n\n      // Drain any actions that were queued while the stream was in progress\n      await this.drainPostStreamActions()\n\n      // Continue conversation if the stream ended with a tool result (server tool completed)\n      if (streamCompletedSuccessfully) {\n        const messages = this.processor.getMessages()\n        const lastPart = messages.at(-1)?.parts.at(-1)\n\n        if (lastPart?.type === 'tool-result' && this.shouldAutoSend()) {\n          try {\n            await this.checkForContinuation()\n          } catch (error) {\n            console.error('Failed to continue flow after tool result:', error)\n          }\n        }\n      }\n    }\n  }\n\n  /**\n   * Reload the last assistant message\n   */\n  async reload(): Promise<void> {\n    const messages = this.processor.getMessages()\n    if (messages.length === 0) return\n\n    // Find the last user message\n    const lastUserMessageIndex = messages.findLastIndex(\n      (m: UIMessage) => m.role === 'user',\n    )\n\n    if (lastUserMessageIndex === -1) return\n\n    this.events.reloaded(lastUserMessageIndex)\n\n    // Remove all messages after the last user message\n    this.processor.removeMessagesAfter(lastUserMessageIndex)\n\n    // Resend\n    await this.streamResponse()\n  }\n\n  /**\n   * Stop the current stream\n   */\n  stop(): void {\n    if (this.abortController) {\n      this.abortController.abort()\n      this.abortController = null\n    }\n    this.setIsLoading(false)\n    this.setStatus('ready')\n    this.events.stopped()\n  }\n\n  /**\n   * Clear all messages\n   */\n  clear(): void {\n    this.processor.clearMessages()\n    this.setError(undefined)\n    this.events.messagesCleared()\n  }\n\n  /**\n   * Add the result of a client-side tool execution\n   */\n  async addToolResult(result: {\n    toolCallId: string\n    tool: string\n    output: any\n    state?: 'output-available' | 'output-error'\n    errorText?: string\n  }): Promise<void> {\n    this.events.toolResultAdded(\n      result.toolCallId,\n      result.tool,\n      result.output,\n      result.state || 'output-available',\n    )\n\n    // Add result via processor\n    this.processor.addToolResult(\n      result.toolCallId,\n      result.output,\n      result.errorText,\n    )\n\n    // If stream is in progress, queue continuation check for after it ends\n    if (this.isLoading) {\n      this.queuePostStreamAction(() => this.checkForContinuation())\n      return\n    }\n\n    await this.checkForContinuation()\n  }\n\n  /**\n   * Respond to a tool approval request\n   */\n  async addToolApprovalResponse(response: {\n    id: string // approval.id, not toolCallId\n    approved: boolean\n  }): Promise<void> {\n    // Find the tool call ID from the approval ID\n    const messages = this.processor.getMessages()\n    let foundToolCallId: string | undefined\n\n    for (const msg of messages) {\n      const toolCallPart = msg.parts.find(\n        (p: MessagePart): p is ToolCallPart =>\n          p.type === 'tool-call' && p.approval?.id === response.id,\n      )\n      if (toolCallPart) {\n        foundToolCallId = toolCallPart.id\n        break\n      }\n    }\n\n    if (foundToolCallId) {\n      this.events.toolApprovalResponded(\n        response.id,\n        foundToolCallId,\n        response.approved,\n      )\n    }\n\n    // Add response via processor\n    this.processor.addToolApprovalResponse(response.id, response.approved)\n\n    // If stream is in progress, queue continuation check for after it ends\n    if (this.isLoading) {\n      this.queuePostStreamAction(() => this.checkForContinuation())\n      return\n    }\n\n    await this.checkForContinuation()\n  }\n\n  /**\n   * Queue an action to be executed after the current stream ends\n   */\n  private queuePostStreamAction(action: () => Promise<void>): void {\n    this.postStreamActions.push(action)\n  }\n\n  /**\n   * Drain and execute all queued post-stream actions\n   */\n  private async drainPostStreamActions(): Promise<void> {\n    while (this.postStreamActions.length > 0) {\n      const action = this.postStreamActions.shift()!\n      await action()\n    }\n  }\n\n  /**\n   * Check if we should continue the flow and do so if needed\n   */\n  private async checkForContinuation(): Promise<void> {\n    // Prevent duplicate continuation attempts\n    if (this.continuationPending || this.isLoading) {\n      return\n    }\n\n    if (this.shouldAutoSend()) {\n      this.continuationPending = true\n      try {\n        await this.streamResponse()\n      } finally {\n        this.continuationPending = false\n      }\n    }\n  }\n\n  /**\n   * Check if all tool calls are complete and we should auto-send\n   */\n  private shouldAutoSend(): boolean {\n    return this.processor.areAllToolsComplete()\n  }\n\n  /**\n   * Get current messages\n   */\n  getMessages(): Array<UIMessage> {\n    return this.processor.getMessages()\n  }\n\n  /**\n   * Get loading state\n   */\n  getIsLoading(): boolean {\n    return this.isLoading\n  }\n\n  /**\n   * Get current status\n   */\n  getStatus(): ChatClientState {\n    return this.status\n  }\n\n  /**\n   * Get current error\n   */\n  getError(): Error | undefined {\n    return this.error\n  }\n\n  /**\n   * Manually set messages\n   */\n  setMessagesManually(messages: Array<UIMessage>): void {\n    this.processor.setMessages(messages)\n  }\n\n  /**\n   * Update options refs (for use in React hooks to avoid recreating client)\n   */\n  updateOptions(options: {\n    connection?: ConnectionAdapter\n    body?: Record<string, any>\n    tools?: ReadonlyArray<AnyClientTool>\n    onResponse?: (response?: Response) => void | Promise<void>\n    onChunk?: (chunk: StreamChunk) => void\n    onFinish?: (message: UIMessage) => void\n    onError?: (error: Error) => void\n  }): void {\n    if (options.connection !== undefined) {\n      this.connection = options.connection\n    }\n    if (options.body !== undefined) {\n      this.body = options.body\n    }\n    if (options.tools !== undefined) {\n      this.clientToolsRef.current = new Map()\n      for (const tool of options.tools) {\n        this.clientToolsRef.current.set(tool.name, tool)\n      }\n    }\n    if (options.onResponse !== undefined) {\n      this.callbacksRef.current.onResponse = options.onResponse\n    }\n    if (options.onChunk !== undefined) {\n      this.callbacksRef.current.onChunk = options.onChunk\n    }\n    if (options.onFinish !== undefined) {\n      this.callbacksRef.current.onFinish = options.onFinish\n    }\n    if (options.onError !== undefined) {\n      this.callbacksRef.current.onError = options.onError\n    }\n  }\n}\n"],"names":[],"mappings":";;AAiBO,MAAM,WAAW;AAAA,EAgCtB,YAAY,SAA4B;AA5BxC,SAAQ,OAA4B,CAAA;AACpC,SAAQ,YAAY;AACpB,SAAQ,QAA2B;AACnC,SAAQ,SAA0B;AAClC,SAAQ,kBAA0C;AAGlD,SAAQ,kBAAiC;AACzC,SAAQ,mBAAkC;AAC1C,SAAQ,oBAAgD,CAAA;AAExD,SAAQ,4CAAwD,IAAA;AAEhE,SAAQ,sBAAsB;AAgB5B,SAAK,WAAW,QAAQ,MAAM,KAAK,iBAAiB,MAAM;AAC1D,SAAK,OAAO,QAAQ,QAAQ,CAAA;AAC5B,SAAK,aAAa,QAAQ;AAC1B,SAAK,SAAS,IAAI,8BAA8B,KAAK,QAAQ;AAG7D,SAAK,iBAAiB,EAAE,SAAS,oBAAI,MAAI;AACzC,QAAI,QAAQ,OAAO;AACjB,iBAAW,QAAQ,QAAQ,OAAO;AAChC,aAAK,eAAe,QAAQ,IAAI,KAAK,MAAM,IAAI;AAAA,MACjD;AAAA,IACF;AAEA,SAAK,eAAe;AAAA,MAClB,SAAS;AAAA,QACP,YAAY,QAAQ,eAAe,MAAM;AAAA,QAAC;AAAA,QAC1C,SAAS,QAAQ,YAAY,MAAM;AAAA,QAAC;AAAA,QACpC,UAAU,QAAQ,aAAa,MAAM;AAAA,QAAC;AAAA,QACtC,SAAS,QAAQ,YAAY,MAAM;AAAA,QAAC;AAAA,QACpC,kBAAkB,QAAQ,qBAAqB,MAAM;AAAA,QAAC;AAAA,QACtD,iBAAiB,QAAQ,oBAAoB,MAAM;AAAA,QAAC;AAAA,QACpD,eAAe,QAAQ,kBAAkB,MAAM;AAAA,QAAC;AAAA,QAChD,gBAAgB,QAAQ,mBAAmB,MAAM;AAAA,QAAC;AAAA,MAAA;AAAA,IACpD;AAIF,SAAK,YAAY,IAAI,gBAAgB;AAAA,MACnC,eAAe,QAAQ,iBAAiB;AAAA,MACxC,iBAAiB,QAAQ;AAAA,MACzB,QAAQ;AAAA,QACN,kBAAkB,CAAC,aAA+B;AAChD,eAAK,aAAa,QAAQ,iBAAiB,QAAQ;AAAA,QACrD;AAAA,QACA,eAAe,MAAM;AACnB,eAAK,UAAU,WAAW;AAAA,QAC5B;AAAA,QACA,aAAa,CAAC,YAAuB;AACnC,eAAK,aAAa,QAAQ,SAAS,OAAO;AAC1C,eAAK,UAAU,OAAO;AAAA,QACxB;AAAA,QACA,SAAS,CAAC,UAAiB;AACzB,eAAK,SAAS,KAAK;AACnB,eAAK,UAAU,OAAO;AACtB,eAAK,aAAa,QAAQ,QAAQ,KAAK;AAAA,QACzC;AAAA,QACA,cAAc,CAAC,WAAmB,YAAoB;AAEpD,cAAI,KAAK,iBAAiB;AACxB,iBAAK,OAAO,YAAY,KAAK,iBAAiB,WAAW,OAAO;AAAA,UAClE;AAAA,QACF;AAAA,QACA,kBAAkB,CAAC,WAAmB,YAAoB;AAExD,cAAI,KAAK,iBAAiB;AACxB,iBAAK,OAAO;AAAA,cACV,KAAK;AAAA,cACL;AAAA,cACA;AAAA,YAAA;AAAA,UAEJ;AAAA,QACF;AAAA,QACA,uBAAuB,CACrB,WACA,YACA,OACA,SACG;AAEH,gBAAM,WAAW,KAAK,UAAU,YAAA;AAChC,gBAAM,UAAU,SAAS,KAAK,CAAC,MAAiB,EAAE,OAAO,SAAS;AAClE,gBAAM,eAAe,SAAS,MAAM;AAAA,YAClC,CAAC,MACC,EAAE,SAAS,eAAe,EAAE,OAAO;AAAA,UAAA;AAEvC,gBAAM,WAAW,cAAc,QAAQ;AAGvC,cAAI,KAAK,iBAAiB;AACxB,iBAAK,OAAO;AAAA,cACV,KAAK;AAAA,cACL;AAAA,cACA;AAAA,cACA;AAAA,cACA;AAAA,cACA;AAAA,YAAA;AAAA,UAEJ;AAAA,QACF;AAAA,QACA,YAAY,CAAC,SAIP;AAEJ,gBAAM,aAAa,KAAK,eAAe,QAAQ,IAAI,KAAK,QAAQ;AAChE,gBAAM,cAAc,YAAY;AAChC,cAAI,aAAa;AAEf,kBAAM,oBAAoB,YAAY;AACpC,kBAAI;AACF,sBAAM,SAAS,MAAM,YAAY,KAAK,KAAK;AAC3C,sBAAM,KAAK,cAAc;AAAA,kBACvB,YAAY,KAAK;AAAA,kBACjB,MAAM,KAAK;AAAA,kBACX;AAAA,kBACA,OAAO;AAAA,gBAAA,CACR;AAAA,cACH,SAAS,OAAY;AACnB,sBAAM,KAAK,cAAc;AAAA,kBACvB,YAAY,KAAK;AAAA,kBACjB,MAAM,KAAK;AAAA,kBACX,QAAQ;AAAA,kBACR,OAAO;AAAA,kBACP,WAAW,MAAM;AAAA,gBAAA,CAClB;AAAA,cACH,UAAA;AAEE,qBAAK,sBAAsB,OAAO,KAAK,UAAU;AAAA,cACnD;AAAA,YACF,GAAA;AAGA,iBAAK,sBAAsB,IAAI,KAAK,YAAY,gBAAgB;AAAA,UAClE;AAAA,QACF;AAAA,QACA,mBAAmB,CAAC,SAKd;AACJ,cAAI,KAAK,iBAAiB;AACxB,iBAAK,OAAO;AAAA,cACV,KAAK;AAAA,cACL,KAAK,oBAAoB;AAAA,cACzB,KAAK;AAAA,cACL,KAAK;AAAA,cACL,KAAK;AAAA,cACL,KAAK;AAAA,YAAA;AAAA,UAET;AAAA,QACF;AAAA,MAAA;AAAA,IACF,CACD;AAED,SAAK,OAAO,cAAc,KAAK,UAAU,YAAA,EAAc,MAAM;AAAA,EAC/D;AAAA,EAEQ,iBAAiB,QAAwB;AAC/C,WAAO,GAAG,MAAM,IAAI,KAAK,KAAK,IAAI,KAAK,OAAA,EAAS,SAAS,EAAE,EAAE,UAAU,CAAC,CAAC;AAAA,EAC3E;AAAA,EAEQ,aAAa,WAA0B;AAC7C,SAAK,YAAY;AACjB,SAAK,aAAa,QAAQ,gBAAgB,SAAS;AACnD,SAAK,OAAO,eAAe,SAAS;AAAA,EACtC;AAAA,EAEQ,UAAU,QAA+B;AAC/C,SAAK,SAAS;AACd,SAAK,aAAa,QAAQ,eAAe,MAAM;AAAA,EACjD;AAAA,EAEQ,SAAS,OAAgC;AAC/C,SAAK,QAAQ;AACb,SAAK,aAAa,QAAQ,cAAc,KAAK;AAC7C,SAAK,OAAO,aAAa,OAAO,WAAW,IAAI;AAAA,EACjD;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,cACZ,QACoB;AAEpB,SAAK,kBAAkB,KAAK,iBAAiB,QAAQ;AAGrD,UAAM,YAAY,KAAK,UAAU,sBAAA;AACjC,SAAK,mBAAmB;AAGxB,UAAM,mBAA8B;AAAA,MAClC,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,OAAO,CAAA;AAAA,MACP,+BAAe,KAAA;AAAA,IAAK;AAEtB,SAAK,OAAO;AAAA,MACV;AAAA,MACA,KAAK,mBAAmB;AAAA,IAAA;AAI1B,qBAAiB,SAAS,QAAQ;AAChC,WAAK,aAAa,QAAQ,QAAQ,KAAK;AACvC,WAAK,UAAU,aAAa,KAAK;AAGjC,YAAM,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,CAAC,CAAC;AAAA,IACvD;AAIA,QAAI,KAAK,sBAAsB,OAAO,GAAG;AACvC,YAAM,QAAQ,IAAI,KAAK,sBAAsB,QAAQ;AAAA,IACvD;AAGA,SAAK,UAAU,eAAA;AAGf,SAAK,kBAAkB;AACvB,SAAK,mBAAmB;AAGxB,UAAM,WAAW,KAAK,UAAU,YAAA;AAChC,UAAM,wBAAwB,SAAS;AAAA,MACrC,CAAC,MAAiB,EAAE,OAAO;AAAA,IAAA;AAG7B,WACE,yBAAyB;AAAA,MACvB,IAAI;AAAA,MACJ,MAAM;AAAA,MACN,OAAO,CAAA;AAAA,MACP,+BAAe,KAAA;AAAA,IAAK;AAAA,EAG1B;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,YAAY,SAAgC;AAChD,QAAI,CAAC,QAAQ,UAAU,KAAK,WAAW;AACrC;AAAA,IACF;AAGA,UAAM,cAAc,KAAK,UAAU,eAAe,QAAQ,MAAM;AAChE,SAAK,OAAO,YAAY,YAAY,IAAI,QAAQ,MAAM;AAEtD,UAAM,KAAK,eAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,OAAO,SAAkD;AAE7D,UAAM,oBAAoB,qBAAqB,SAAS,iBAAiB;AAGzE,QAAI,kBAAkB,SAAS,UAAU;AACvC;AAAA,IACF;AAGA,UAAM,YAAY;AAGlB,SAAK,OAAO,gBAAgB,SAAS;AAGrC,UAAM,WAAW,KAAK,UAAU,YAAA;AAChC,SAAK,UAAU,YAAY,CAAC,GAAG,UAAU,SAAS,CAAC;AAGnD,QAAI,KAAK,WAAW;AAClB,WAAK,sBAAsB,MAAM,KAAK,eAAA,CAAgB;AACtD;AAAA,IACF;AAEA,UAAM,KAAK,eAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,iBAAgC;AAE5C,QAAI,KAAK,WAAW;AAClB;AAAA,IACF;AAEA,SAAK,aAAa,IAAI;AACtB,SAAK,UAAU,WAAW;AAC1B,SAAK,SAAS,MAAS;AACvB,SAAK,kBAAkB,IAAI,gBAAA;AAE3B,SAAK,sBAAsB,MAAA;AAC3B,QAAI,8BAA8B;AAElC,QAAI;AAEF,YAAM,WAAW,KAAK,UAAU,YAAA;AAGhC,YAAM,KAAK,aAAa,QAAQ,WAAA;AAGhC,YAAM,yBAAyB;AAAA,QAC7B,GAAG,KAAK;AAAA,QACR,gBAAgB,KAAK;AAAA,MAAA;AAIvB,YAAM,SAAS,KAAK,WAAW;AAAA,QAC7B;AAAA,QACA;AAAA,QACA,KAAK,gBAAgB;AAAA,MAAA;AAGvB,YAAM,KAAK,cAAc,MAAM;AAC/B,oCAA8B;AAAA,IAChC,SAAS,KAAK;AACZ,UAAI,eAAe,OAAO;AACxB,YAAI,IAAI,SAAS,cAAc;AAC7B;AAAA,QACF;AACA,aAAK,SAAS,GAAG;AACjB,aAAK,UAAU,OAAO;AACtB,aAAK,aAAa,QAAQ,QAAQ,GAAG;AAAA,MACvC;AAAA,IACF,UAAA;AACE,WAAK,kBAAkB;AACvB,WAAK,aAAa,KAAK;AAGvB,YAAM,KAAK,uBAAA;AAGX,UAAI,6BAA6B;AAC/B,cAAM,WAAW,KAAK,UAAU,YAAA;AAChC,cAAM,WAAW,SAAS,GAAG,EAAE,GAAG,MAAM,GAAG,EAAE;AAE7C,YAAI,UAAU,SAAS,iBAAiB,KAAK,kBAAkB;AAC7D,cAAI;AACF,kBAAM,KAAK,qBAAA;AAAA,UACb,SAAS,OAAO;AACd,oBAAQ,MAAM,8CAA8C,KAAK;AAAA,UACnE;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,SAAwB;AAC5B,UAAM,WAAW,KAAK,UAAU,YAAA;AAChC,QAAI,SAAS,WAAW,EAAG;AAG3B,UAAM,uBAAuB,SAAS;AAAA,MACpC,CAAC,MAAiB,EAAE,SAAS;AAAA,IAAA;AAG/B,QAAI,yBAAyB,GAAI;AAEjC,SAAK,OAAO,SAAS,oBAAoB;AAGzC,SAAK,UAAU,oBAAoB,oBAAoB;AAGvD,UAAM,KAAK,eAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA,EAKA,OAAa;AACX,QAAI,KAAK,iBAAiB;AACxB,WAAK,gBAAgB,MAAA;AACrB,WAAK,kBAAkB;AAAA,IACzB;AACA,SAAK,aAAa,KAAK;AACvB,SAAK,UAAU,OAAO;AACtB,SAAK,OAAO,QAAA;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAKA,QAAc;AACZ,SAAK,UAAU,cAAA;AACf,SAAK,SAAS,MAAS;AACvB,SAAK,OAAO,gBAAA;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,cAAc,QAMF;AAChB,SAAK,OAAO;AAAA,MACV,OAAO;AAAA,MACP,OAAO;AAAA,MACP,OAAO;AAAA,MACP,OAAO,SAAS;AAAA,IAAA;AAIlB,SAAK,UAAU;AAAA,MACb,OAAO;AAAA,MACP,OAAO;AAAA,MACP,OAAO;AAAA,IAAA;AAIT,QAAI,KAAK,WAAW;AAClB,WAAK,sBAAsB,MAAM,KAAK,qBAAA,CAAsB;AAC5D;AAAA,IACF;AAEA,UAAM,KAAK,qBAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA,EAKA,MAAM,wBAAwB,UAGZ;AAEhB,UAAM,WAAW,KAAK,UAAU,YAAA;AAChC,QAAI;AAEJ,eAAW,OAAO,UAAU;AAC1B,YAAM,eAAe,IAAI,MAAM;AAAA,QAC7B,CAAC,MACC,EAAE,SAAS,eAAe,EAAE,UAAU,OAAO,SAAS;AAAA,MAAA;AAE1D,UAAI,cAAc;AAChB,0BAAkB,aAAa;AAC/B;AAAA,MACF;AAAA,IACF;AAEA,QAAI,iBAAiB;AACnB,WAAK,OAAO;AAAA,QACV,SAAS;AAAA,QACT;AAAA,QACA,SAAS;AAAA,MAAA;AAAA,IAEb;AAGA,SAAK,UAAU,wBAAwB,SAAS,IAAI,SAAS,QAAQ;AAGrE,QAAI,KAAK,WAAW;AAClB,WAAK,sBAAsB,MAAM,KAAK,qBAAA,CAAsB;AAC5D;AAAA,IACF;AAEA,UAAM,KAAK,qBAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA,EAKQ,sBAAsB,QAAmC;AAC/D,SAAK,kBAAkB,KAAK,MAAM;AAAA,EACpC;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,yBAAwC;AACpD,WAAO,KAAK,kBAAkB,SAAS,GAAG;AACxC,YAAM,SAAS,KAAK,kBAAkB,MAAA;AACtC,YAAM,OAAA;AAAA,IACR;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,uBAAsC;AAElD,QAAI,KAAK,uBAAuB,KAAK,WAAW;AAC9C;AAAA,IACF;AAEA,QAAI,KAAK,kBAAkB;AACzB,WAAK,sBAAsB;AAC3B,UAAI;AACF,cAAM,KAAK,eAAA;AAAA,MACb,UAAA;AACE,aAAK,sBAAsB;AAAA,MAC7B;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,iBAA0B;AAChC,WAAO,KAAK,UAAU,oBAAA;AAAA,EACxB;AAAA;AAAA;AAAA;AAAA,EAKA,cAAgC;AAC9B,WAAO,KAAK,UAAU,YAAA;AAAA,EACxB;AAAA;AAAA;AAAA;AAAA,EAKA,eAAwB;AACtB,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAKA,YAA6B;AAC3B,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAKA,WAA8B;AAC5B,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAKA,oBAAoB,UAAkC;AACpD,SAAK,UAAU,YAAY,QAAQ;AAAA,EACrC;AAAA;AAAA;AAAA;AAAA,EAKA,cAAc,SAQL;AACP,QAAI,QAAQ,eAAe,QAAW;AACpC,WAAK,aAAa,QAAQ;AAAA,IAC5B;AACA,QAAI,QAAQ,SAAS,QAAW;AAC9B,WAAK,OAAO,QAAQ;AAAA,IACtB;AACA,QAAI,QAAQ,UAAU,QAAW;AAC/B,WAAK,eAAe,UAAU,oBAAI,IAAA;AAClC,iBAAW,QAAQ,QAAQ,OAAO;AAChC,aAAK,eAAe,QAAQ,IAAI,KAAK,MAAM,IAAI;AAAA,MACjD;AAAA,IACF;AACA,QAAI,QAAQ,eAAe,QAAW;AACpC,WAAK,aAAa,QAAQ,aAAa,QAAQ;AAAA,IACjD;AACA,QAAI,QAAQ,YAAY,QAAW;AACjC,WAAK,aAAa,QAAQ,UAAU,QAAQ;AAAA,IAC9C;AACA,QAAI,QAAQ,aAAa,QAAW;AAClC,WAAK,aAAa,QAAQ,WAAW,QAAQ;AAAA,IAC/C;AACA,QAAI,QAAQ,YAAY,QAAW;AACjC,WAAK,aAAa,QAAQ,UAAU,QAAQ;AAAA,IAC9C;AAAA,EACF;AACF;"}

package/dist/esm/connection-adapters.d.ts

@@ -105,7 +105,7 @@ export declare function fetchHttpStream(url: string | (() => string), options?:
  * const client = new ChatClient({ connection });
  * ```
  */
-export declare function stream(streamFactory: (messages: Array<ModelMessage>, data?: Record<string, any>) => AsyncIterable<StreamChunk>): ConnectionAdapter;
+export declare function stream(streamFactory: (messages: Array<UIMessage> | Array<ModelMessage>, data?: Record<string, any>) => AsyncIterable<StreamChunk>): ConnectionAdapter;
 /**
  * Create an RPC stream connection adapter (for RPC-based streaming like Cap'n Web RPC)
  *
@@ -122,4 +122,4 @@ export declare function stream(streamFactory: (messages: Array<ModelMessage>, da
  * const client = new ChatClient({ connection });
  * ```
  */
-export declare function rpcStream(rpcCall: (messages: Array<ModelMessage>, data?: Record<string, any>) => AsyncIterable<StreamChunk>): ConnectionAdapter;
+export declare function rpcStream(rpcCall: (messages: Array<UIMessage> | Array<ModelMessage>, data?: Record<string, any>) => AsyncIterable<StreamChunk>): ConnectionAdapter;

package/dist/esm/connection-adapters.js

@@ -1,4 +1,3 @@
-import { convertMessagesToModelMessages } from "@tanstack/ai";
 function mergeHeaders(customHeaders) {
   if (!customHeaders) {
     return {};
@@ -43,13 +42,12 @@ function fetchServerSentEvents(url, options = {}) {
     async *connect(messages, data, abortSignal) {
       const resolvedUrl = typeof url === "function" ? url() : url;
       const resolvedOptions = typeof options === "function" ? await options() : options;
-      const modelMessages = convertMessagesToModelMessages(messages);
       const requestHeaders = {
         "Content-Type": "application/json",
         ...mergeHeaders(resolvedOptions.headers)
       };
       const requestBody = {
-        messages: modelMessages,
+        messages,
         data,
         ...resolvedOptions.body
       };
@@ -88,13 +86,12 @@ function fetchHttpStream(url, options = {}) {
     async *connect(messages, data, abortSignal) {
       const resolvedUrl = typeof url === "function" ? url() : url;
       const resolvedOptions = typeof options === "function" ? await options() : options;
-      const modelMessages = convertMessagesToModelMessages(messages);
       const requestHeaders = {
         "Content-Type": "application/json",
         ...mergeHeaders(resolvedOptions.headers)
       };
       const requestBody = {
-        messages: modelMessages,
+        messages,
         data,
         ...resolvedOptions.body
       };
@@ -129,16 +126,14 @@ function fetchHttpStream(url, options = {}) {
 function stream(streamFactory) {
   return {
     async *connect(messages, data) {
-      const modelMessages = convertMessagesToModelMessages(messages);
-      yield* streamFactory(modelMessages, data);
+      yield* streamFactory(messages, data);
     }
   };
 }
 function rpcStream(rpcCall) {
   return {
     async *connect(messages, data) {
-      const modelMessages = convertMessagesToModelMessages(messages);
-      yield* rpcCall(modelMessages, data);
+      yield* rpcCall(messages, data);
     }
   };
 }

package/dist/esm/connection-adapters.js.map

@@ -1 +1 @@
-{"version":3,"file":"connection-adapters.js","sources":["../../src/connection-adapters.ts"],"sourcesContent":["import { convertMessagesToModelMessages } from '@tanstack/ai'\nimport type { ModelMessage, StreamChunk, UIMessage } from '@tanstack/ai'\n\n/**\n * Merge custom headers into request headers\n */\nfunction mergeHeaders(\n  customHeaders?: Record<string, string> | Headers,\n): Record<string, string> {\n  if (!customHeaders) {\n    return {}\n  }\n  if (customHeaders instanceof Headers) {\n    const result: Record<string, string> = {}\n    customHeaders.forEach((value, key) => {\n      result[key] = value\n    })\n    return result\n  }\n  return customHeaders\n}\n\n/**\n * Read lines from a stream (newline-delimited)\n */\nasync function* readStreamLines(\n  reader: ReadableStreamDefaultReader<Uint8Array>,\n  abortSignal?: AbortSignal,\n): AsyncGenerator<string> {\n  try {\n    const decoder = new TextDecoder()\n    let buffer = ''\n\n    // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition\n    while (true) {\n      // Check if aborted before reading\n      if (abortSignal?.aborted) {\n        break\n      }\n\n      const { done, value } = await reader.read()\n      if (done) break\n\n      buffer += decoder.decode(value, { stream: true })\n      const lines = buffer.split('\\n')\n\n      // Keep the last incomplete line in the buffer\n      buffer = lines.pop() || ''\n\n      for (const line of lines) {\n        if (line.trim()) {\n          yield line\n        }\n      }\n    }\n\n    // Process any remaining data in the buffer\n    if (buffer.trim()) {\n      yield buffer\n    }\n  } finally {\n    reader.releaseLock()\n  }\n}\n\n/**\n * Connection adapter interface - converts a connection into a stream of chunks\n */\nexport interface ConnectionAdapter {\n  /**\n   * Connect and return an async iterable of StreamChunks\n   * @param messages - The messages to send (UIMessages or ModelMessages)\n   * @param data - Additional data to send\n   * @param abortSignal - Optional abort signal for request cancellation\n   */\n  connect: (\n    messages: Array<UIMessage> | Array<ModelMessage>,\n    data?: Record<string, any>,\n    abortSignal?: AbortSignal,\n  ) => AsyncIterable<StreamChunk>\n}\n\n/**\n * Options for fetch-based connection adapters\n */\nexport interface FetchConnectionOptions {\n  headers?: Record<string, string> | Headers\n  credentials?: RequestCredentials\n  signal?: AbortSignal\n  body?: Record<string, any>\n  fetchClient?: typeof globalThis.fetch\n}\n\n/**\n * Create a Server-Sent Events connection adapter\n *\n * @param url - The API endpoint URL (or a function that returns the URL)\n * @param options - Fetch options (headers, credentials, body, etc.) or a function that returns options (can be async)\n * @returns A connection adapter for SSE streams\n *\n * @example\n * ```typescript\n * // Static URL\n * const connection = fetchServerSentEvents('/api/chat');\n *\n * // Dynamic URL\n * const connection = fetchServerSentEvents(() => `/api/chat?user=${userId}`);\n *\n * // With options\n * const connection = fetchServerSentEvents('/api/chat', {\n *   headers: { 'Authorization': 'Bearer token' }\n * });\n *\n * // With dynamic options\n * const connection = fetchServerSentEvents('/api/chat', () => ({\n *   headers: { 'Authorization': `Bearer ${getToken()}` }\n * }));\n *\n * // With additional body data\n * const connection = fetchServerSentEvents('/api/chat', async () => ({\n *   body: {\n *     provider: 'openai',\n *     model: 'gpt-4o',\n *   }\n * }));\n * ```\n */\nexport function fetchServerSentEvents(\n  url: string | (() => string),\n  options:\n    | FetchConnectionOptions\n    | (() => FetchConnectionOptions | Promise<FetchConnectionOptions>) = {},\n): ConnectionAdapter {\n  return {\n    async *connect(messages, data, abortSignal) {\n      // Resolve URL and options if they are functions\n      const resolvedUrl = typeof url === 'function' ? url() : url\n      const resolvedOptions =\n        typeof options === 'function' ? await options() : options\n\n      const modelMessages = convertMessagesToModelMessages(messages)\n\n      const requestHeaders: Record<string, string> = {\n        'Content-Type': 'application/json',\n        ...mergeHeaders(resolvedOptions.headers),\n      }\n\n      // Merge body from options with messages and data\n      const requestBody = {\n        messages: modelMessages,\n        data,\n        ...resolvedOptions.body,\n      }\n\n      const fetchClient = resolvedOptions.fetchClient ?? fetch\n      const response = await fetchClient(resolvedUrl, {\n        method: 'POST',\n        headers: requestHeaders,\n        body: JSON.stringify(requestBody),\n        credentials: resolvedOptions.credentials || 'same-origin',\n        signal: abortSignal || resolvedOptions.signal,\n      })\n\n      if (!response.ok) {\n        throw new Error(\n          `HTTP error! status: ${response.status} ${response.statusText}`,\n        )\n      }\n\n      // Parse Server-Sent Events format\n      const reader = response.body?.getReader()\n      if (!reader) {\n        throw new Error('Response body is not readable')\n      }\n\n      for await (const line of readStreamLines(reader, abortSignal)) {\n        // Handle Server-Sent Events format\n        const data = line.startsWith('data: ') ? line.slice(6) : line\n\n        if (data === '[DONE]') continue\n\n        try {\n          const parsed: StreamChunk = JSON.parse(data)\n          yield parsed\n        } catch (parseError) {\n          // Skip non-JSON lines or malformed chunks\n          console.warn('Failed to parse SSE chunk:', data)\n        }\n      }\n    },\n  }\n}\n\n/**\n * Create an HTTP streaming connection adapter (for raw streaming without SSE format)\n *\n * @param url - The API endpoint URL (or a function that returns the URL)\n * @param options - Fetch options (headers, credentials, body, etc.) or a function that returns options (can be async)\n * @returns A connection adapter for HTTP streams\n *\n * @example\n * ```typescript\n * // Static URL\n * const connection = fetchHttpStream('/api/chat');\n *\n * // Dynamic URL\n * const connection = fetchHttpStream(() => `/api/chat?user=${userId}`);\n *\n * // With options\n * const connection = fetchHttpStream('/api/chat', {\n *   headers: { 'Authorization': 'Bearer token' }\n * });\n *\n * // With dynamic options\n * const connection = fetchHttpStream('/api/chat', () => ({\n *   headers: { 'Authorization': `Bearer ${getToken()}` }\n * }));\n *\n * // With additional body data\n * const connection = fetchHttpStream('/api/chat', async () => ({\n *   body: {\n *     provider: 'openai',\n *     model: 'gpt-4o',\n *   }\n * }));\n * ```\n */\nexport function fetchHttpStream(\n  url: string | (() => string),\n  options:\n    | FetchConnectionOptions\n    | (() => FetchConnectionOptions | Promise<FetchConnectionOptions>) = {},\n): ConnectionAdapter {\n  return {\n    async *connect(messages, data, abortSignal) {\n      // Resolve URL and options if they are functions\n      const resolvedUrl = typeof url === 'function' ? url() : url\n      const resolvedOptions =\n        typeof options === 'function' ? await options() : options\n\n      // Convert UIMessages to ModelMessages if needed\n      const modelMessages = convertMessagesToModelMessages(messages)\n\n      const requestHeaders: Record<string, string> = {\n        'Content-Type': 'application/json',\n        ...mergeHeaders(resolvedOptions.headers),\n      }\n\n      // Merge body from options with messages and data\n      const requestBody = {\n        messages: modelMessages,\n        data,\n        ...resolvedOptions.body,\n      }\n\n      const fetchClient = resolvedOptions.fetchClient ?? fetch\n      const response = await fetchClient(resolvedUrl, {\n        method: 'POST',\n        headers: requestHeaders,\n        body: JSON.stringify(requestBody),\n        credentials: resolvedOptions.credentials || 'same-origin',\n        signal: abortSignal || resolvedOptions.signal,\n      })\n\n      if (!response.ok) {\n        throw new Error(\n          `HTTP error! status: ${response.status} ${response.statusText}`,\n        )\n      }\n\n      // Parse raw HTTP stream (newline-delimited JSON)\n      const reader = response.body?.getReader()\n      if (!reader) {\n        throw new Error('Response body is not readable')\n      }\n\n      for await (const line of readStreamLines(reader, abortSignal)) {\n        try {\n          const parsed: StreamChunk = JSON.parse(line)\n          yield parsed\n        } catch (parseError) {\n          console.warn('Failed to parse HTTP stream chunk:', line)\n        }\n      }\n    },\n  }\n}\n\n/**\n * Create a direct stream connection adapter (for server functions or direct streams)\n *\n * @param streamFactory - A function that returns an async iterable of StreamChunks\n * @returns A connection adapter for direct streams\n *\n * @example\n * ```typescript\n * // With TanStack Start server function\n * const connection = stream(() => serverFunction({ messages }));\n *\n * const client = new ChatClient({ connection });\n * ```\n */\nexport function stream(\n  streamFactory: (\n    messages: Array<ModelMessage>,\n    data?: Record<string, any>,\n  ) => AsyncIterable<StreamChunk>,\n): ConnectionAdapter {\n  return {\n    async *connect(messages, data) {\n      const modelMessages = convertMessagesToModelMessages(messages)\n      yield* streamFactory(modelMessages, data)\n    },\n  }\n}\n\n/**\n * Create an RPC stream connection adapter (for RPC-based streaming like Cap'n Web RPC)\n *\n * @param rpcCall - A function that accepts messages and returns an async iterable of StreamChunks\n * @returns A connection adapter for RPC streams\n *\n * @example\n * ```typescript\n * // With Cap'n Web RPC\n * const connection = rpcStream((messages, data) =>\n *   api.streamMurfResponse(messages, data)\n * );\n *\n * const client = new ChatClient({ connection });\n * ```\n */\nexport function rpcStream(\n  rpcCall: (\n    messages: Array<ModelMessage>,\n    data?: Record<string, any>,\n  ) => AsyncIterable<StreamChunk>,\n): ConnectionAdapter {\n  return {\n    async *connect(messages, data) {\n      const modelMessages = convertMessagesToModelMessages(messages)\n      // Simply yield from the RPC call\n      // The RPC layer handles WebSocket transport\n      yield* rpcCall(modelMessages, data)\n    },\n  }\n}\n"],"names":["data"],"mappings":";AAMA,SAAS,aACP,eACwB;AACxB,MAAI,CAAC,eAAe;AAClB,WAAO,CAAA;AAAA,EACT;AACA,MAAI,yBAAyB,SAAS;AACpC,UAAM,SAAiC,CAAA;AACvC,kBAAc,QAAQ,CAAC,OAAO,QAAQ;AACpC,aAAO,GAAG,IAAI;AAAA,IAChB,CAAC;AACD,WAAO;AAAA,EACT;AACA,SAAO;AACT;AAKA,gBAAgB,gBACd,QACA,aACwB;AACxB,MAAI;AACF,UAAM,UAAU,IAAI,YAAA;AACpB,QAAI,SAAS;AAGb,WAAO,MAAM;AAEX,UAAI,aAAa,SAAS;AACxB;AAAA,MACF;AAEA,YAAM,EAAE,MAAM,MAAA,IAAU,MAAM,OAAO,KAAA;AACrC,UAAI,KAAM;AAEV,gBAAU,QAAQ,OAAO,OAAO,EAAE,QAAQ,MAAM;AAChD,YAAM,QAAQ,OAAO,MAAM,IAAI;AAG/B,eAAS,MAAM,SAAS;AAExB,iBAAW,QAAQ,OAAO;AACxB,YAAI,KAAK,QAAQ;AACf,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAGA,QAAI,OAAO,QAAQ;AACjB,YAAM;AAAA,IACR;AAAA,EACF,UAAA;AACE,WAAO,YAAA;AAAA,EACT;AACF;AAgEO,SAAS,sBACd,KACA,UAEuE,IACpD;AACnB,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM,aAAa;AAE1C,YAAM,cAAc,OAAO,QAAQ,aAAa,QAAQ;AACxD,YAAM,kBACJ,OAAO,YAAY,aAAa,MAAM,YAAY;AAEpD,YAAM,gBAAgB,+BAA+B,QAAQ;AAE7D,YAAM,iBAAyC;AAAA,QAC7C,gBAAgB;AAAA,QAChB,GAAG,aAAa,gBAAgB,OAAO;AAAA,MAAA;AAIzC,YAAM,cAAc;AAAA,QAClB,UAAU;AAAA,QACV;AAAA,QACA,GAAG,gBAAgB;AAAA,MAAA;AAGrB,YAAM,cAAc,gBAAgB,eAAe;AACnD,YAAM,WAAW,MAAM,YAAY,aAAa;AAAA,QAC9C,QAAQ;AAAA,QACR,SAAS;AAAA,QACT,MAAM,KAAK,UAAU,WAAW;AAAA,QAChC,aAAa,gBAAgB,eAAe;AAAA,QAC5C,QAAQ,eAAe,gBAAgB;AAAA,MAAA,CACxC;AAED,UAAI,CAAC,SAAS,IAAI;AAChB,cAAM,IAAI;AAAA,UACR,uBAAuB,SAAS,MAAM,IAAI,SAAS,UAAU;AAAA,QAAA;AAAA,MAEjE;AAGA,YAAM,SAAS,SAAS,MAAM,UAAA;AAC9B,UAAI,CAAC,QAAQ;AACX,cAAM,IAAI,MAAM,+BAA+B;AAAA,MACjD;AAEA,uBAAiB,QAAQ,gBAAgB,QAAQ,WAAW,GAAG;AAE7D,cAAMA,QAAO,KAAK,WAAW,QAAQ,IAAI,KAAK,MAAM,CAAC,IAAI;AAEzD,YAAIA,UAAS,SAAU;AAEvB,YAAI;AACF,gBAAM,SAAsB,KAAK,MAAMA,KAAI;AAC3C,gBAAM;AAAA,QACR,SAAS,YAAY;AAEnB,kBAAQ,KAAK,8BAA8BA,KAAI;AAAA,QACjD;AAAA,MACF;AAAA,IACF;AAAA,EAAA;AAEJ;AAoCO,SAAS,gBACd,KACA,UAEuE,IACpD;AACnB,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM,aAAa;AAE1C,YAAM,cAAc,OAAO,QAAQ,aAAa,QAAQ;AACxD,YAAM,kBACJ,OAAO,YAAY,aAAa,MAAM,YAAY;AAGpD,YAAM,gBAAgB,+BAA+B,QAAQ;AAE7D,YAAM,iBAAyC;AAAA,QAC7C,gBAAgB;AAAA,QAChB,GAAG,aAAa,gBAAgB,OAAO;AAAA,MAAA;AAIzC,YAAM,cAAc;AAAA,QAClB,UAAU;AAAA,QACV;AAAA,QACA,GAAG,gBAAgB;AAAA,MAAA;AAGrB,YAAM,cAAc,gBAAgB,eAAe;AACnD,YAAM,WAAW,MAAM,YAAY,aAAa;AAAA,QAC9C,QAAQ;AAAA,QACR,SAAS;AAAA,QACT,MAAM,KAAK,UAAU,WAAW;AAAA,QAChC,aAAa,gBAAgB,eAAe;AAAA,QAC5C,QAAQ,eAAe,gBAAgB;AAAA,MAAA,CACxC;AAED,UAAI,CAAC,SAAS,IAAI;AAChB,cAAM,IAAI;AAAA,UACR,uBAAuB,SAAS,MAAM,IAAI,SAAS,UAAU;AAAA,QAAA;AAAA,MAEjE;AAGA,YAAM,SAAS,SAAS,MAAM,UAAA;AAC9B,UAAI,CAAC,QAAQ;AACX,cAAM,IAAI,MAAM,+BAA+B;AAAA,MACjD;AAEA,uBAAiB,QAAQ,gBAAgB,QAAQ,WAAW,GAAG;AAC7D,YAAI;AACF,gBAAM,SAAsB,KAAK,MAAM,IAAI;AAC3C,gBAAM;AAAA,QACR,SAAS,YAAY;AACnB,kBAAQ,KAAK,sCAAsC,IAAI;AAAA,QACzD;AAAA,MACF;AAAA,IACF;AAAA,EAAA;AAEJ;AAgBO,SAAS,OACd,eAImB;AACnB,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM;AAC7B,YAAM,gBAAgB,+BAA+B,QAAQ;AAC7D,aAAO,cAAc,eAAe,IAAI;AAAA,IAC1C;AAAA,EAAA;AAEJ;AAkBO,SAAS,UACd,SAImB;AACnB,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM;AAC7B,YAAM,gBAAgB,+BAA+B,QAAQ;AAG7D,aAAO,QAAQ,eAAe,IAAI;AAAA,IACpC;AAAA,EAAA;AAEJ;"}
+{"version":3,"file":"connection-adapters.js","sources":["../../src/connection-adapters.ts"],"sourcesContent":["import type { ModelMessage, StreamChunk, UIMessage } from '@tanstack/ai'\n\n/**\n * Merge custom headers into request headers\n */\nfunction mergeHeaders(\n  customHeaders?: Record<string, string> | Headers,\n): Record<string, string> {\n  if (!customHeaders) {\n    return {}\n  }\n  if (customHeaders instanceof Headers) {\n    const result: Record<string, string> = {}\n    customHeaders.forEach((value, key) => {\n      result[key] = value\n    })\n    return result\n  }\n  return customHeaders\n}\n\n/**\n * Read lines from a stream (newline-delimited)\n */\nasync function* readStreamLines(\n  reader: ReadableStreamDefaultReader<Uint8Array>,\n  abortSignal?: AbortSignal,\n): AsyncGenerator<string> {\n  try {\n    const decoder = new TextDecoder()\n    let buffer = ''\n\n    // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition\n    while (true) {\n      // Check if aborted before reading\n      if (abortSignal?.aborted) {\n        break\n      }\n\n      const { done, value } = await reader.read()\n      if (done) break\n\n      buffer += decoder.decode(value, { stream: true })\n      const lines = buffer.split('\\n')\n\n      // Keep the last incomplete line in the buffer\n      buffer = lines.pop() || ''\n\n      for (const line of lines) {\n        if (line.trim()) {\n          yield line\n        }\n      }\n    }\n\n    // Process any remaining data in the buffer\n    if (buffer.trim()) {\n      yield buffer\n    }\n  } finally {\n    reader.releaseLock()\n  }\n}\n\n/**\n * Connection adapter interface - converts a connection into a stream of chunks\n */\nexport interface ConnectionAdapter {\n  /**\n   * Connect and return an async iterable of StreamChunks\n   * @param messages - The messages to send (UIMessages or ModelMessages)\n   * @param data - Additional data to send\n   * @param abortSignal - Optional abort signal for request cancellation\n   */\n  connect: (\n    messages: Array<UIMessage> | Array<ModelMessage>,\n    data?: Record<string, any>,\n    abortSignal?: AbortSignal,\n  ) => AsyncIterable<StreamChunk>\n}\n\n/**\n * Options for fetch-based connection adapters\n */\nexport interface FetchConnectionOptions {\n  headers?: Record<string, string> | Headers\n  credentials?: RequestCredentials\n  signal?: AbortSignal\n  body?: Record<string, any>\n  fetchClient?: typeof globalThis.fetch\n}\n\n/**\n * Create a Server-Sent Events connection adapter\n *\n * @param url - The API endpoint URL (or a function that returns the URL)\n * @param options - Fetch options (headers, credentials, body, etc.) or a function that returns options (can be async)\n * @returns A connection adapter for SSE streams\n *\n * @example\n * ```typescript\n * // Static URL\n * const connection = fetchServerSentEvents('/api/chat');\n *\n * // Dynamic URL\n * const connection = fetchServerSentEvents(() => `/api/chat?user=${userId}`);\n *\n * // With options\n * const connection = fetchServerSentEvents('/api/chat', {\n *   headers: { 'Authorization': 'Bearer token' }\n * });\n *\n * // With dynamic options\n * const connection = fetchServerSentEvents('/api/chat', () => ({\n *   headers: { 'Authorization': `Bearer ${getToken()}` }\n * }));\n *\n * // With additional body data\n * const connection = fetchServerSentEvents('/api/chat', async () => ({\n *   body: {\n *     provider: 'openai',\n *     model: 'gpt-4o',\n *   }\n * }));\n * ```\n */\nexport function fetchServerSentEvents(\n  url: string | (() => string),\n  options:\n    | FetchConnectionOptions\n    | (() => FetchConnectionOptions | Promise<FetchConnectionOptions>) = {},\n): ConnectionAdapter {\n  return {\n    async *connect(messages, data, abortSignal) {\n      // Resolve URL and options if they are functions\n      const resolvedUrl = typeof url === 'function' ? url() : url\n      const resolvedOptions =\n        typeof options === 'function' ? await options() : options\n\n      const requestHeaders: Record<string, string> = {\n        'Content-Type': 'application/json',\n        ...mergeHeaders(resolvedOptions.headers),\n      }\n\n      // Send messages as-is (UIMessages with parts preserved)\n      // Server-side TextEngine handles conversion to ModelMessages\n      const requestBody = {\n        messages,\n        data,\n        ...resolvedOptions.body,\n      }\n\n      const fetchClient = resolvedOptions.fetchClient ?? fetch\n      const response = await fetchClient(resolvedUrl, {\n        method: 'POST',\n        headers: requestHeaders,\n        body: JSON.stringify(requestBody),\n        credentials: resolvedOptions.credentials || 'same-origin',\n        signal: abortSignal || resolvedOptions.signal,\n      })\n\n      if (!response.ok) {\n        throw new Error(\n          `HTTP error! status: ${response.status} ${response.statusText}`,\n        )\n      }\n\n      // Parse Server-Sent Events format\n      const reader = response.body?.getReader()\n      if (!reader) {\n        throw new Error('Response body is not readable')\n      }\n\n      for await (const line of readStreamLines(reader, abortSignal)) {\n        // Handle Server-Sent Events format\n        const data = line.startsWith('data: ') ? line.slice(6) : line\n\n        if (data === '[DONE]') continue\n\n        try {\n          const parsed: StreamChunk = JSON.parse(data)\n          yield parsed\n        } catch (parseError) {\n          // Skip non-JSON lines or malformed chunks\n          console.warn('Failed to parse SSE chunk:', data)\n        }\n      }\n    },\n  }\n}\n\n/**\n * Create an HTTP streaming connection adapter (for raw streaming without SSE format)\n *\n * @param url - The API endpoint URL (or a function that returns the URL)\n * @param options - Fetch options (headers, credentials, body, etc.) or a function that returns options (can be async)\n * @returns A connection adapter for HTTP streams\n *\n * @example\n * ```typescript\n * // Static URL\n * const connection = fetchHttpStream('/api/chat');\n *\n * // Dynamic URL\n * const connection = fetchHttpStream(() => `/api/chat?user=${userId}`);\n *\n * // With options\n * const connection = fetchHttpStream('/api/chat', {\n *   headers: { 'Authorization': 'Bearer token' }\n * });\n *\n * // With dynamic options\n * const connection = fetchHttpStream('/api/chat', () => ({\n *   headers: { 'Authorization': `Bearer ${getToken()}` }\n * }));\n *\n * // With additional body data\n * const connection = fetchHttpStream('/api/chat', async () => ({\n *   body: {\n *     provider: 'openai',\n *     model: 'gpt-4o',\n *   }\n * }));\n * ```\n */\nexport function fetchHttpStream(\n  url: string | (() => string),\n  options:\n    | FetchConnectionOptions\n    | (() => FetchConnectionOptions | Promise<FetchConnectionOptions>) = {},\n): ConnectionAdapter {\n  return {\n    async *connect(messages, data, abortSignal) {\n      // Resolve URL and options if they are functions\n      const resolvedUrl = typeof url === 'function' ? url() : url\n      const resolvedOptions =\n        typeof options === 'function' ? await options() : options\n\n      const requestHeaders: Record<string, string> = {\n        'Content-Type': 'application/json',\n        ...mergeHeaders(resolvedOptions.headers),\n      }\n\n      // Send messages as-is (UIMessages with parts preserved)\n      // Server-side TextEngine handles conversion to ModelMessages\n      const requestBody = {\n        messages,\n        data,\n        ...resolvedOptions.body,\n      }\n\n      const fetchClient = resolvedOptions.fetchClient ?? fetch\n      const response = await fetchClient(resolvedUrl, {\n        method: 'POST',\n        headers: requestHeaders,\n        body: JSON.stringify(requestBody),\n        credentials: resolvedOptions.credentials || 'same-origin',\n        signal: abortSignal || resolvedOptions.signal,\n      })\n\n      if (!response.ok) {\n        throw new Error(\n          `HTTP error! status: ${response.status} ${response.statusText}`,\n        )\n      }\n\n      // Parse raw HTTP stream (newline-delimited JSON)\n      const reader = response.body?.getReader()\n      if (!reader) {\n        throw new Error('Response body is not readable')\n      }\n\n      for await (const line of readStreamLines(reader, abortSignal)) {\n        try {\n          const parsed: StreamChunk = JSON.parse(line)\n          yield parsed\n        } catch (parseError) {\n          console.warn('Failed to parse HTTP stream chunk:', line)\n        }\n      }\n    },\n  }\n}\n\n/**\n * Create a direct stream connection adapter (for server functions or direct streams)\n *\n * @param streamFactory - A function that returns an async iterable of StreamChunks\n * @returns A connection adapter for direct streams\n *\n * @example\n * ```typescript\n * // With TanStack Start server function\n * const connection = stream(() => serverFunction({ messages }));\n *\n * const client = new ChatClient({ connection });\n * ```\n */\nexport function stream(\n  streamFactory: (\n    messages: Array<UIMessage> | Array<ModelMessage>,\n    data?: Record<string, any>,\n  ) => AsyncIterable<StreamChunk>,\n): ConnectionAdapter {\n  return {\n    async *connect(messages, data) {\n      // Pass messages as-is (UIMessages with parts preserved)\n      // Server-side chat() handles conversion to ModelMessages\n      yield* streamFactory(messages, data)\n    },\n  }\n}\n\n/**\n * Create an RPC stream connection adapter (for RPC-based streaming like Cap'n Web RPC)\n *\n * @param rpcCall - A function that accepts messages and returns an async iterable of StreamChunks\n * @returns A connection adapter for RPC streams\n *\n * @example\n * ```typescript\n * // With Cap'n Web RPC\n * const connection = rpcStream((messages, data) =>\n *   api.streamMurfResponse(messages, data)\n * );\n *\n * const client = new ChatClient({ connection });\n * ```\n */\nexport function rpcStream(\n  rpcCall: (\n    messages: Array<UIMessage> | Array<ModelMessage>,\n    data?: Record<string, any>,\n  ) => AsyncIterable<StreamChunk>,\n): ConnectionAdapter {\n  return {\n    async *connect(messages, data) {\n      // Pass messages as-is (UIMessages with parts preserved)\n      // Server-side chat() handles conversion to ModelMessages\n      yield* rpcCall(messages, data)\n    },\n  }\n}\n"],"names":["data"],"mappings":"AAKA,SAAS,aACP,eACwB;AACxB,MAAI,CAAC,eAAe;AAClB,WAAO,CAAA;AAAA,EACT;AACA,MAAI,yBAAyB,SAAS;AACpC,UAAM,SAAiC,CAAA;AACvC,kBAAc,QAAQ,CAAC,OAAO,QAAQ;AACpC,aAAO,GAAG,IAAI;AAAA,IAChB,CAAC;AACD,WAAO;AAAA,EACT;AACA,SAAO;AACT;AAKA,gBAAgB,gBACd,QACA,aACwB;AACxB,MAAI;AACF,UAAM,UAAU,IAAI,YAAA;AACpB,QAAI,SAAS;AAGb,WAAO,MAAM;AAEX,UAAI,aAAa,SAAS;AACxB;AAAA,MACF;AAEA,YAAM,EAAE,MAAM,MAAA,IAAU,MAAM,OAAO,KAAA;AACrC,UAAI,KAAM;AAEV,gBAAU,QAAQ,OAAO,OAAO,EAAE,QAAQ,MAAM;AAChD,YAAM,QAAQ,OAAO,MAAM,IAAI;AAG/B,eAAS,MAAM,SAAS;AAExB,iBAAW,QAAQ,OAAO;AACxB,YAAI,KAAK,QAAQ;AACf,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAGA,QAAI,OAAO,QAAQ;AACjB,YAAM;AAAA,IACR;AAAA,EACF,UAAA;AACE,WAAO,YAAA;AAAA,EACT;AACF;AAgEO,SAAS,sBACd,KACA,UAEuE,IACpD;AACnB,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM,aAAa;AAE1C,YAAM,cAAc,OAAO,QAAQ,aAAa,QAAQ;AACxD,YAAM,kBACJ,OAAO,YAAY,aAAa,MAAM,YAAY;AAEpD,YAAM,iBAAyC;AAAA,QAC7C,gBAAgB;AAAA,QAChB,GAAG,aAAa,gBAAgB,OAAO;AAAA,MAAA;AAKzC,YAAM,cAAc;AAAA,QAClB;AAAA,QACA;AAAA,QACA,GAAG,gBAAgB;AAAA,MAAA;AAGrB,YAAM,cAAc,gBAAgB,eAAe;AACnD,YAAM,WAAW,MAAM,YAAY,aAAa;AAAA,QAC9C,QAAQ;AAAA,QACR,SAAS;AAAA,QACT,MAAM,KAAK,UAAU,WAAW;AAAA,QAChC,aAAa,gBAAgB,eAAe;AAAA,QAC5C,QAAQ,eAAe,gBAAgB;AAAA,MAAA,CACxC;AAED,UAAI,CAAC,SAAS,IAAI;AAChB,cAAM,IAAI;AAAA,UACR,uBAAuB,SAAS,MAAM,IAAI,SAAS,UAAU;AAAA,QAAA;AAAA,MAEjE;AAGA,YAAM,SAAS,SAAS,MAAM,UAAA;AAC9B,UAAI,CAAC,QAAQ;AACX,cAAM,IAAI,MAAM,+BAA+B;AAAA,MACjD;AAEA,uBAAiB,QAAQ,gBAAgB,QAAQ,WAAW,GAAG;AAE7D,cAAMA,QAAO,KAAK,WAAW,QAAQ,IAAI,KAAK,MAAM,CAAC,IAAI;AAEzD,YAAIA,UAAS,SAAU;AAEvB,YAAI;AACF,gBAAM,SAAsB,KAAK,MAAMA,KAAI;AAC3C,gBAAM;AAAA,QACR,SAAS,YAAY;AAEnB,kBAAQ,KAAK,8BAA8BA,KAAI;AAAA,QACjD;AAAA,MACF;AAAA,IACF;AAAA,EAAA;AAEJ;AAoCO,SAAS,gBACd,KACA,UAEuE,IACpD;AACnB,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM,aAAa;AAE1C,YAAM,cAAc,OAAO,QAAQ,aAAa,QAAQ;AACxD,YAAM,kBACJ,OAAO,YAAY,aAAa,MAAM,YAAY;AAEpD,YAAM,iBAAyC;AAAA,QAC7C,gBAAgB;AAAA,QAChB,GAAG,aAAa,gBAAgB,OAAO;AAAA,MAAA;AAKzC,YAAM,cAAc;AAAA,QAClB;AAAA,QACA;AAAA,QACA,GAAG,gBAAgB;AAAA,MAAA;AAGrB,YAAM,cAAc,gBAAgB,eAAe;AACnD,YAAM,WAAW,MAAM,YAAY,aAAa;AAAA,QAC9C,QAAQ;AAAA,QACR,SAAS;AAAA,QACT,MAAM,KAAK,UAAU,WAAW;AAAA,QAChC,aAAa,gBAAgB,eAAe;AAAA,QAC5C,QAAQ,eAAe,gBAAgB;AAAA,MAAA,CACxC;AAED,UAAI,CAAC,SAAS,IAAI;AAChB,cAAM,IAAI;AAAA,UACR,uBAAuB,SAAS,MAAM,IAAI,SAAS,UAAU;AAAA,QAAA;AAAA,MAEjE;AAGA,YAAM,SAAS,SAAS,MAAM,UAAA;AAC9B,UAAI,CAAC,QAAQ;AACX,cAAM,IAAI,MAAM,+BAA+B;AAAA,MACjD;AAEA,uBAAiB,QAAQ,gBAAgB,QAAQ,WAAW,GAAG;AAC7D,YAAI;AACF,gBAAM,SAAsB,KAAK,MAAM,IAAI;AAC3C,gBAAM;AAAA,QACR,SAAS,YAAY;AACnB,kBAAQ,KAAK,sCAAsC,IAAI;AAAA,QACzD;AAAA,MACF;AAAA,IACF;AAAA,EAAA;AAEJ;AAgBO,SAAS,OACd,eAImB;AACnB,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM;AAG7B,aAAO,cAAc,UAAU,IAAI;AAAA,IACrC;AAAA,EAAA;AAEJ;AAkBO,SAAS,UACd,SAImB;AACnB,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM;AAG7B,aAAO,QAAQ,UAAU,IAAI;AAAA,IAC/B;AAAA,EAAA;AAEJ;"}

package/dist/esm/index.d.ts

@@ -1,5 +1,5 @@
 export { ChatClient } from './chat-client.js';
-export type { UIMessage, MessagePart, TextPart, ToolCallPart, ToolResultPart, ThinkingPart, ChatClientOptions, ChatRequestBody, InferChatMessages, } from './types.js';
+export type { UIMessage, MessagePart, TextPart, ToolCallPart, ToolResultPart, ThinkingPart, ChatClientOptions, ChatRequestBody, InferChatMessages, ChatClientState, } from './types.js';
 export { clientTools, createChatClientOptions } from './types.js';
 export type { ExtractToolNames, ExtractToolInput, ExtractToolOutput, } from './tool-types.js';
 export type { AnyClientTool } from '@tanstack/ai';

package/dist/esm/types.d.ts

@@ -8,6 +8,10 @@ export type ToolCallState = 'awaiting-input' | 'input-streaming' | 'input-comple
  * Tool result states - track the lifecycle of a tool result
  */
 export type ToolResultState = 'streaming' | 'complete' | 'error';
+/**
+ * ChatClient state - track the lifecycle of a chat
+ */
+export type ChatClientState = 'ready' | 'submitted' | 'streaming' | 'error';
 /**
  * Message parts - building blocks of UIMessage
  */
@@ -138,6 +142,10 @@ export interface ChatClientOptions<TTools extends ReadonlyArray<AnyClientTool> =
      * Callback when error state changes
      */
     onErrorChange?: (error: Error | undefined) => void;
+    /**
+     * Callback when chat status changes
+     */
+    onStatusChange?: (status: ChatClientState) => void;
     /**
      * Client-side tools with execution logic
      * When provided, tools with execute functions will be called automatically

package/dist/esm/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","sources":["../../src/types.ts"],"sourcesContent":["import type {\n  AnyClientTool,\n  ChunkStrategy,\n  InferToolInput,\n  InferToolOutput,\n  ModelMessage,\n  StreamChunk,\n} from '@tanstack/ai'\nimport type { ConnectionAdapter } from './connection-adapters'\n\n/**\n * Tool call states - track the lifecycle of a tool call\n */\nexport type ToolCallState =\n  | 'awaiting-input' // Received start but no arguments yet\n  | 'input-streaming' // Partial arguments received\n  | 'input-complete' // All arguments received\n  | 'approval-requested' // Waiting for user approval\n  | 'approval-responded' // User has approved/denied\n\n/**\n * Tool result states - track the lifecycle of a tool result\n */\nexport type ToolResultState =\n  | 'streaming' // Placeholder for future streamed output\n  | 'complete' // Result is complete\n  | 'error' // Error occurred\n\n/**\n * Message parts - building blocks of UIMessage\n */\nexport interface TextPart {\n  type: 'text'\n  content: string\n}\n\n/**\n * Helper type that creates a tool-call part for a specific tool.\n * This is a conditional type to enable proper distribution over union types,\n * creating a discriminated union where `name` is the discriminant.\n */\ntype ToolCallPartForTool<T> = T extends AnyClientTool\n  ? {\n      type: 'tool-call'\n      id: string\n      name: T['name']\n      arguments: string // JSON string (may be incomplete)\n      /** Parsed tool input (typed from inputSchema) */\n      input?: InferToolInput<T>\n      state: ToolCallState\n      /** Approval metadata if tool requires user approval */\n      approval?: {\n        id: string // Unique approval ID\n        needsApproval: boolean // Always true if present\n        approved?: boolean // User's decision (undefined until responded)\n      }\n      /** Tool execution output (for client tools or after approval) */\n      output?: InferToolOutput<T>\n    }\n  : never\n\n/**\n * Fallback tool-call part type when tools are not typed\n */\ntype UntypedToolCallPart = {\n  type: 'tool-call'\n  id: string\n  name: string\n  arguments: string\n  input?: any\n  state: ToolCallState\n  approval?: {\n    id: string\n    needsApproval: boolean\n    approved?: boolean\n  }\n  output?: any\n}\n\n/**\n * Tool call part that creates a proper discriminated union.\n * When TTools is typed, checking `part.name === 'toolName'` will narrow\n * `part.output` to the correct type for that tool.\n *\n * The discriminant is `name`, so code like:\n * ```ts\n * if (part.name === 'recommendGuitar') {\n *   // part.output is now typed to the recommendGuitar tool's output\n * }\n * ```\n */\nexport type ToolCallPart<TTools extends ReadonlyArray<AnyClientTool> = any> =\n  // Check if we have a concrete tools array (not 'any' or 'never')\n  [TTools] extends [never]\n    ? UntypedToolCallPart\n    : unknown extends TTools\n      ? UntypedToolCallPart\n      : TTools extends ReadonlyArray<infer Tool>\n        ? Tool extends AnyClientTool\n          ? ToolCallPartForTool<Tool>\n          : UntypedToolCallPart\n        : UntypedToolCallPart\n\nexport interface ToolResultPart {\n  type: 'tool-result'\n  toolCallId: string\n  content: string\n  state: ToolResultState\n  error?: string // Error message if state is \"error\"\n}\n\nexport interface ThinkingPart {\n  type: 'thinking'\n  content: string\n}\n\nexport type MessagePart<TTools extends ReadonlyArray<AnyClientTool> = any> =\n  | TextPart\n  | ToolCallPart<TTools>\n  | ToolResultPart\n  | ThinkingPart\n\n/**\n * UIMessage - Domain-specific message format optimized for building chat UIs\n * Contains parts that can be text, tool calls, or tool results\n */\nexport interface UIMessage<TTools extends ReadonlyArray<AnyClientTool> = any> {\n  id: string\n  role: 'system' | 'user' | 'assistant'\n  parts: Array<MessagePart<TTools>>\n  createdAt?: Date\n}\n\nexport interface ChatClientOptions<\n  TTools extends ReadonlyArray<AnyClientTool> = any,\n> {\n  /**\n   * Connection adapter for streaming\n   * Use fetchServerSentEvents(), fetchHttpStream(), or stream() to create adapters\n   */\n  connection: ConnectionAdapter\n\n  /**\n   * Initial messages to populate the chat\n   */\n  initialMessages?: Array<UIMessage<TTools>>\n\n  /**\n   * Unique identifier for this chat instance\n   * Used for managing multiple chats\n   */\n  id?: string\n\n  /**\n   * Additional body parameters to send\n   */\n  body?: Record<string, any>\n\n  /**\n   * Callback when a response is received\n   */\n  onResponse?: (response?: Response) => void | Promise<void>\n\n  /**\n   * Callback when a stream chunk is received\n   */\n  onChunk?: (chunk: StreamChunk) => void\n\n  /**\n   * Callback when the response is finished\n   */\n  onFinish?: (message: UIMessage<TTools>) => void\n\n  /**\n   * Callback when an error occurs\n   */\n  onError?: (error: Error) => void\n\n  /**\n   * Callback when messages change\n   */\n  onMessagesChange?: (messages: Array<UIMessage<TTools>>) => void\n\n  /**\n   * Callback when loading state changes\n   */\n  onLoadingChange?: (isLoading: boolean) => void\n\n  /**\n   * Callback when error state changes\n   */\n  onErrorChange?: (error: Error | undefined) => void\n\n  /**\n   * Client-side tools with execution logic\n   * When provided, tools with execute functions will be called automatically\n   */\n  tools?: TTools\n\n  /**\n   * Stream processing options (optional)\n   * Configure chunking strategy\n   */\n  streamProcessor?: {\n    /**\n     * Strategy for when to emit text updates\n     * Defaults to ImmediateStrategy (every chunk)\n     */\n    chunkStrategy?: ChunkStrategy\n  }\n}\n\nexport interface ChatRequestBody {\n  messages: Array<ModelMessage>\n  data?: Record<string, any>\n}\n\n/**\n * Create a typed array of client tools with proper type inference.\n * This eliminates the need for `as const` when defining tool arrays.\n *\n * @example\n * ```ts\n * const tools = clientTools(\n *   myTool1.client(() => result1),\n *   myTool2.client(() => result2),\n * )\n *\n * // tools is now properly typed as a tuple with literal tool names\n * // This enables type narrowing when checking part.name === 'toolName'\n * ```\n */\nexport function clientTools<const T extends Array<AnyClientTool>>(\n  ...tools: T\n): T {\n  return tools\n}\n\n/**\n * Helper to create typed chat client options\n * Use this to get proper type inference for messages\n *\n * @example\n * ```ts\n * const tools = clientTools(myTool1, myTool2)\n *\n * const chatOptions = createChatClientOptions({\n *   connection: fetchServerSentEvents('/api/chat'),\n *   tools,\n * })\n *\n * type MyMessages = InferChatMessages<typeof chatOptions>\n * ```\n */\nexport function createChatClientOptions<\n  const TTools extends ReadonlyArray<AnyClientTool>,\n>(options: ChatClientOptions<TTools>): ChatClientOptions<TTools> {\n  return options\n}\n\n/**\n * Extract the message type from chat options\n *\n * @example\n * ```ts\n * const chatOptions = createChatClientOptions({\n *   connection: fetchServerSentEvents('/api/chat'),\n *   tools: [myTool1, myTool2],\n * })\n *\n * type MyMessages = InferChatMessages<typeof chatOptions>\n * // MyMessages is now Array<UIMessage<[typeof myTool1, typeof myTool2]>>\n * ```\n */\nexport type InferChatMessages<T> =\n  T extends ChatClientOptions<infer TTools> ? Array<UIMessage<TTools>> : never\n"],"names":[],"mappings":"AAwOO,SAAS,eACX,OACA;AACH,SAAO;AACT;AAkBO,SAAS,wBAEd,SAA+D;AAC/D,SAAO;AACT;"}
+{"version":3,"file":"types.js","sources":["../../src/types.ts"],"sourcesContent":["import type {\n  AnyClientTool,\n  ChunkStrategy,\n  InferToolInput,\n  InferToolOutput,\n  ModelMessage,\n  StreamChunk,\n} from '@tanstack/ai'\nimport type { ConnectionAdapter } from './connection-adapters'\n\n/**\n * Tool call states - track the lifecycle of a tool call\n */\nexport type ToolCallState =\n  | 'awaiting-input' // Received start but no arguments yet\n  | 'input-streaming' // Partial arguments received\n  | 'input-complete' // All arguments received\n  | 'approval-requested' // Waiting for user approval\n  | 'approval-responded' // User has approved/denied\n\n/**\n * Tool result states - track the lifecycle of a tool result\n */\nexport type ToolResultState =\n  | 'streaming' // Placeholder for future streamed output\n  | 'complete' // Result is complete\n  | 'error' // Error occurred\n\n/**\n * ChatClient state - track the lifecycle of a chat\n */\nexport type ChatClientState = 'ready' | 'submitted' | 'streaming' | 'error'\n\n/**\n * Message parts - building blocks of UIMessage\n */\nexport interface TextPart {\n  type: 'text'\n  content: string\n}\n\n/**\n * Helper type that creates a tool-call part for a specific tool.\n * This is a conditional type to enable proper distribution over union types,\n * creating a discriminated union where `name` is the discriminant.\n */\ntype ToolCallPartForTool<T> = T extends AnyClientTool\n  ? {\n      type: 'tool-call'\n      id: string\n      name: T['name']\n      arguments: string // JSON string (may be incomplete)\n      /** Parsed tool input (typed from inputSchema) */\n      input?: InferToolInput<T>\n      state: ToolCallState\n      /** Approval metadata if tool requires user approval */\n      approval?: {\n        id: string // Unique approval ID\n        needsApproval: boolean // Always true if present\n        approved?: boolean // User's decision (undefined until responded)\n      }\n      /** Tool execution output (for client tools or after approval) */\n      output?: InferToolOutput<T>\n    }\n  : never\n\n/**\n * Fallback tool-call part type when tools are not typed\n */\ntype UntypedToolCallPart = {\n  type: 'tool-call'\n  id: string\n  name: string\n  arguments: string\n  input?: any\n  state: ToolCallState\n  approval?: {\n    id: string\n    needsApproval: boolean\n    approved?: boolean\n  }\n  output?: any\n}\n\n/**\n * Tool call part that creates a proper discriminated union.\n * When TTools is typed, checking `part.name === 'toolName'` will narrow\n * `part.output` to the correct type for that tool.\n *\n * The discriminant is `name`, so code like:\n * ```ts\n * if (part.name === 'recommendGuitar') {\n *   // part.output is now typed to the recommendGuitar tool's output\n * }\n * ```\n */\nexport type ToolCallPart<TTools extends ReadonlyArray<AnyClientTool> = any> =\n  // Check if we have a concrete tools array (not 'any' or 'never')\n  [TTools] extends [never]\n    ? UntypedToolCallPart\n    : unknown extends TTools\n      ? UntypedToolCallPart\n      : TTools extends ReadonlyArray<infer Tool>\n        ? Tool extends AnyClientTool\n          ? ToolCallPartForTool<Tool>\n          : UntypedToolCallPart\n        : UntypedToolCallPart\n\nexport interface ToolResultPart {\n  type: 'tool-result'\n  toolCallId: string\n  content: string\n  state: ToolResultState\n  error?: string // Error message if state is \"error\"\n}\n\nexport interface ThinkingPart {\n  type: 'thinking'\n  content: string\n}\n\nexport type MessagePart<TTools extends ReadonlyArray<AnyClientTool> = any> =\n  | TextPart\n  | ToolCallPart<TTools>\n  | ToolResultPart\n  | ThinkingPart\n\n/**\n * UIMessage - Domain-specific message format optimized for building chat UIs\n * Contains parts that can be text, tool calls, or tool results\n */\nexport interface UIMessage<TTools extends ReadonlyArray<AnyClientTool> = any> {\n  id: string\n  role: 'system' | 'user' | 'assistant'\n  parts: Array<MessagePart<TTools>>\n  createdAt?: Date\n}\n\nexport interface ChatClientOptions<\n  TTools extends ReadonlyArray<AnyClientTool> = any,\n> {\n  /**\n   * Connection adapter for streaming\n   * Use fetchServerSentEvents(), fetchHttpStream(), or stream() to create adapters\n   */\n  connection: ConnectionAdapter\n\n  /**\n   * Initial messages to populate the chat\n   */\n  initialMessages?: Array<UIMessage<TTools>>\n\n  /**\n   * Unique identifier for this chat instance\n   * Used for managing multiple chats\n   */\n  id?: string\n\n  /**\n   * Additional body parameters to send\n   */\n  body?: Record<string, any>\n\n  /**\n   * Callback when a response is received\n   */\n  onResponse?: (response?: Response) => void | Promise<void>\n\n  /**\n   * Callback when a stream chunk is received\n   */\n  onChunk?: (chunk: StreamChunk) => void\n\n  /**\n   * Callback when the response is finished\n   */\n  onFinish?: (message: UIMessage<TTools>) => void\n\n  /**\n   * Callback when an error occurs\n   */\n  onError?: (error: Error) => void\n\n  /**\n   * Callback when messages change\n   */\n  onMessagesChange?: (messages: Array<UIMessage<TTools>>) => void\n\n  /**\n   * Callback when loading state changes\n   */\n  onLoadingChange?: (isLoading: boolean) => void\n\n  /**\n   * Callback when error state changes\n   */\n  onErrorChange?: (error: Error | undefined) => void\n\n  /**\n   * Callback when chat status changes\n   */\n  onStatusChange?: (status: ChatClientState) => void\n\n  /**\n   * Client-side tools with execution logic\n   * When provided, tools with execute functions will be called automatically\n   */\n  tools?: TTools\n\n  /**\n   * Stream processing options (optional)\n   * Configure chunking strategy\n   */\n  streamProcessor?: {\n    /**\n     * Strategy for when to emit text updates\n     * Defaults to ImmediateStrategy (every chunk)\n     */\n    chunkStrategy?: ChunkStrategy\n  }\n}\n\nexport interface ChatRequestBody {\n  messages: Array<ModelMessage>\n  data?: Record<string, any>\n}\n\n/**\n * Create a typed array of client tools with proper type inference.\n * This eliminates the need for `as const` when defining tool arrays.\n *\n * @example\n * ```ts\n * const tools = clientTools(\n *   myTool1.client(() => result1),\n *   myTool2.client(() => result2),\n * )\n *\n * // tools is now properly typed as a tuple with literal tool names\n * // This enables type narrowing when checking part.name === 'toolName'\n * ```\n */\nexport function clientTools<const T extends Array<AnyClientTool>>(\n  ...tools: T\n): T {\n  return tools\n}\n\n/**\n * Helper to create typed chat client options\n * Use this to get proper type inference for messages\n *\n * @example\n * ```ts\n * const tools = clientTools(myTool1, myTool2)\n *\n * const chatOptions = createChatClientOptions({\n *   connection: fetchServerSentEvents('/api/chat'),\n *   tools,\n * })\n *\n * type MyMessages = InferChatMessages<typeof chatOptions>\n * ```\n */\nexport function createChatClientOptions<\n  const TTools extends ReadonlyArray<AnyClientTool>,\n>(options: ChatClientOptions<TTools>): ChatClientOptions<TTools> {\n  return options\n}\n\n/**\n * Extract the message type from chat options\n *\n * @example\n * ```ts\n * const chatOptions = createChatClientOptions({\n *   connection: fetchServerSentEvents('/api/chat'),\n *   tools: [myTool1, myTool2],\n * })\n *\n * type MyMessages = InferChatMessages<typeof chatOptions>\n * // MyMessages is now Array<UIMessage<[typeof myTool1, typeof myTool2]>>\n * ```\n */\nexport type InferChatMessages<T> =\n  T extends ChatClientOptions<infer TTools> ? Array<UIMessage<TTools>> : never\n"],"names":[],"mappings":"AAkPO,SAAS,eACX,OACA;AACH,SAAO;AACT;AAkBO,SAAS,wBAEd,SAA+D;AAC/D,SAAO;AACT;"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tanstack/ai-client",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "description": "Framework-agnostic headless client for TanStack AI",
   "author": "",
   "license": "MIT",
@@ -31,7 +31,7 @@
     "src"
   ],
   "dependencies": {
-    "@tanstack/ai": "0.3.0"
+    "@tanstack/ai": "0.3.1"
   },
   "devDependencies": {
     "@vitest/coverage-v8": "4.0.14",

package/src/chat-client.ts

@@ -4,15 +4,16 @@ import {
   normalizeToUIMessage,
 } from '@tanstack/ai'
 import { DefaultChatClientEventEmitter } from './events'
+import type { AnyClientTool, ModelMessage, StreamChunk } from '@tanstack/ai'
+import type { ConnectionAdapter } from './connection-adapters'
+import type { ChatClientEventEmitter } from './events'
 import type {
   ChatClientOptions,
+  ChatClientState,
   MessagePart,
   ToolCallPart,
   UIMessage,
 } from './types'
-import type { AnyClientTool, ModelMessage, StreamChunk } from '@tanstack/ai'
-import type { ConnectionAdapter } from './connection-adapters'
-import type { ChatClientEventEmitter } from './events'
 
 export class ChatClient {
   private processor: StreamProcessor
@@ -21,12 +22,17 @@ export class ChatClient {
   private body: Record<string, any> = {}
   private isLoading = false
   private error: Error | undefined = undefined
+  private status: ChatClientState = 'ready'
   private abortController: AbortController | null = null
   private events: ChatClientEventEmitter
   private clientToolsRef: { current: Map<string, AnyClientTool> }
   private currentStreamId: string | null = null
   private currentMessageId: string | null = null
   private postStreamActions: Array<() => Promise<void>> = []
+  // Track pending client tool executions to await them before stream finalization
+  private pendingToolExecutions: Map<string, Promise<void>> = new Map()
+  // Flag to deduplicate continuation checks during action draining
+  private continuationPending = false
 
   private callbacksRef: {
     current: {
@@ -37,6 +43,7 @@ export class ChatClient {
       onMessagesChange: (messages: Array<UIMessage>) => void
       onLoadingChange: (isLoading: boolean) => void
       onErrorChange: (error: Error | undefined) => void
+      onStatusChange: (status: ChatClientState) => void
     }
   }
 
@@ -63,6 +70,7 @@ export class ChatClient {
         onMessagesChange: options.onMessagesChange || (() => {}),
         onLoadingChange: options.onLoadingChange || (() => {}),
         onErrorChange: options.onErrorChange || (() => {}),
+        onStatusChange: options.onStatusChange || (() => {}),
       },
     }
 
@@ -75,13 +83,15 @@ export class ChatClient {
           this.callbacksRef.current.onMessagesChange(messages)
         },
         onStreamStart: () => {
-          // Stream started
+          this.setStatus('streaming')
         },
         onStreamEnd: (message: UIMessage) => {
           this.callbacksRef.current.onFinish(message)
+          this.setStatus('ready')
         },
         onError: (error: Error) => {
           this.setError(error)
+          this.setStatus('error')
           this.callbacksRef.current.onError(error)
         },
         onTextUpdate: (messageId: string, content: string) => {
@@ -127,31 +137,41 @@ export class ChatClient {
             )
           }
         },
-        onToolCall: async (args: {
+        onToolCall: (args: {
           toolCallId: string
           toolName: string
           input: any
         }) => {
           // Handle client-side tool execution automatically
           const clientTool = this.clientToolsRef.current.get(args.toolName)
-          if (clientTool?.execute) {
-            try {
-              const output = await clientTool.execute(args.input)
-              await this.addToolResult({
-                toolCallId: args.toolCallId,
-                tool: args.toolName,
-                output,
-                state: 'output-available',
-              })
-            } catch (error: any) {
-              await this.addToolResult({
-                toolCallId: args.toolCallId,
-                tool: args.toolName,
-                output: null,
-                state: 'output-error',
-                errorText: error.message,
-              })
-            }
+          const executeFunc = clientTool?.execute
+          if (executeFunc) {
+            // Create and track the execution promise
+            const executionPromise = (async () => {
+              try {
+                const output = await executeFunc(args.input)
+                await this.addToolResult({
+                  toolCallId: args.toolCallId,
+                  tool: args.toolName,
+                  output,
+                  state: 'output-available',
+                })
+              } catch (error: any) {
+                await this.addToolResult({
+                  toolCallId: args.toolCallId,
+                  tool: args.toolName,
+                  output: null,
+                  state: 'output-error',
+                  errorText: error.message,
+                })
+              } finally {
+                // Remove from pending when complete
+                this.pendingToolExecutions.delete(args.toolCallId)
+              }
+            })()
+
+            // Track the pending execution
+            this.pendingToolExecutions.set(args.toolCallId, executionPromise)
           }
         },
         onApprovalRequest: (args: {
@@ -187,6 +207,11 @@ export class ChatClient {
     this.events.loadingChanged(isLoading)
   }
 
+  private setStatus(status: ChatClientState): void {
+    this.status = status
+    this.callbacksRef.current.onStatusChange(status)
+  }
+
   private setError(error: Error | undefined): void {
     this.error = error
     this.callbacksRef.current.onErrorChange(error)
@@ -227,6 +252,12 @@ export class ChatClient {
       await new Promise((resolve) => setTimeout(resolve, 0))
     }
 
+    // Wait for all pending tool executions to complete before finalizing
+    // This ensures client tools finish before we check for continuation
+    if (this.pendingToolExecutions.size > 0) {
+      await Promise.all(this.pendingToolExecutions.values())
+    }
+
     // Finalize the stream
     this.processor.finalizeStream()
 
@@ -287,6 +318,12 @@ export class ChatClient {
     const messages = this.processor.getMessages()
     this.processor.setMessages([...messages, uiMessage])
 
+    // If stream is in progress, queue the response for after it ends
+    if (this.isLoading) {
+      this.queuePostStreamAction(() => this.streamResponse())
+      return
+    }
+
     await this.streamResponse()
   }
 
@@ -294,13 +331,22 @@ export class ChatClient {
    * Stream a response from the LLM
    */
   private async streamResponse(): Promise<void> {
+    // Guard against concurrent streams - if already loading, skip
+    if (this.isLoading) {
+      return
+    }
+
     this.setIsLoading(true)
+    this.setStatus('submitted')
     this.setError(undefined)
     this.abortController = new AbortController()
+    // Reset pending tool executions for the new stream
+    this.pendingToolExecutions.clear()
+    let streamCompletedSuccessfully = false
 
     try {
-      // Get model messages for the LLM
-      const modelMessages = this.processor.toModelMessages()
+      // Get UIMessages with parts (preserves approval state and client tool results)
+      const messages = this.processor.getMessages()
 
       // Call onResponse callback
       await this.callbacksRef.current.onResponse()
@@ -313,18 +359,20 @@ export class ChatClient {
 
       // Connect and stream
       const stream = this.connection.connect(
-        modelMessages,
+        messages,
         bodyWithConversationId,
         this.abortController.signal,
       )
 
       await this.processStream(stream)
+      streamCompletedSuccessfully = true
     } catch (err) {
       if (err instanceof Error) {
         if (err.name === 'AbortError') {
           return
         }
         this.setError(err)
+        this.setStatus('error')
         this.callbacksRef.current.onError(err)
       }
     } finally {
@@ -333,6 +381,20 @@ export class ChatClient {
 
       // Drain any actions that were queued while the stream was in progress
       await this.drainPostStreamActions()
+
+      // Continue conversation if the stream ended with a tool result (server tool completed)
+      if (streamCompletedSuccessfully) {
+        const messages = this.processor.getMessages()
+        const lastPart = messages.at(-1)?.parts.at(-1)
+
+        if (lastPart?.type === 'tool-result' && this.shouldAutoSend()) {
+          try {
+            await this.checkForContinuation()
+          } catch (error) {
+            console.error('Failed to continue flow after tool result:', error)
+          }
+        }
+      }
     }
   }
 
@@ -368,6 +430,7 @@ export class ChatClient {
       this.abortController = null
     }
     this.setIsLoading(false)
+    this.setStatus('ready')
     this.events.stopped()
   }
 
@@ -476,8 +539,18 @@ export class ChatClient {
    * Check if we should continue the flow and do so if needed
    */
   private async checkForContinuation(): Promise<void> {
+    // Prevent duplicate continuation attempts
+    if (this.continuationPending || this.isLoading) {
+      return
+    }
+
     if (this.shouldAutoSend()) {
-      await this.streamResponse()
+      this.continuationPending = true
+      try {
+        await this.streamResponse()
+      } finally {
+        this.continuationPending = false
+      }
     }
   }
 
@@ -502,6 +575,13 @@ export class ChatClient {
     return this.isLoading
   }
 
+  /**
+   * Get current status
+   */
+  getStatus(): ChatClientState {
+    return this.status
+  }
+
   /**
    * Get current error
    */

package/src/connection-adapters.ts

@@ -1,4 +1,3 @@
-import { convertMessagesToModelMessages } from '@tanstack/ai'
 import type { ModelMessage, StreamChunk, UIMessage } from '@tanstack/ai'
 
 /**
@@ -138,16 +137,15 @@ export function fetchServerSentEvents(
       const resolvedOptions =
         typeof options === 'function' ? await options() : options
 
-      const modelMessages = convertMessagesToModelMessages(messages)
-
       const requestHeaders: Record<string, string> = {
         'Content-Type': 'application/json',
         ...mergeHeaders(resolvedOptions.headers),
       }
 
-      // Merge body from options with messages and data
+      // Send messages as-is (UIMessages with parts preserved)
+      // Server-side TextEngine handles conversion to ModelMessages
       const requestBody = {
-        messages: modelMessages,
+        messages,
         data,
         ...resolvedOptions.body,
       }
@@ -238,17 +236,15 @@ export function fetchHttpStream(
       const resolvedOptions =
         typeof options === 'function' ? await options() : options
 
-      // Convert UIMessages to ModelMessages if needed
-      const modelMessages = convertMessagesToModelMessages(messages)
-
       const requestHeaders: Record<string, string> = {
         'Content-Type': 'application/json',
         ...mergeHeaders(resolvedOptions.headers),
       }
 
-      // Merge body from options with messages and data
+      // Send messages as-is (UIMessages with parts preserved)
+      // Server-side TextEngine handles conversion to ModelMessages
       const requestBody = {
-        messages: modelMessages,
+        messages,
         data,
         ...resolvedOptions.body,
       }
@@ -302,14 +298,15 @@ export function fetchHttpStream(
  */
 export function stream(
   streamFactory: (
-    messages: Array<ModelMessage>,
+    messages: Array<UIMessage> | Array<ModelMessage>,
     data?: Record<string, any>,
   ) => AsyncIterable<StreamChunk>,
 ): ConnectionAdapter {
   return {
     async *connect(messages, data) {
-      const modelMessages = convertMessagesToModelMessages(messages)
-      yield* streamFactory(modelMessages, data)
+      // Pass messages as-is (UIMessages with parts preserved)
+      // Server-side chat() handles conversion to ModelMessages
+      yield* streamFactory(messages, data)
     },
   }
 }
@@ -332,16 +329,15 @@ export function stream(
  */
 export function rpcStream(
   rpcCall: (
-    messages: Array<ModelMessage>,
+    messages: Array<UIMessage> | Array<ModelMessage>,
     data?: Record<string, any>,
   ) => AsyncIterable<StreamChunk>,
 ): ConnectionAdapter {
   return {
     async *connect(messages, data) {
-      const modelMessages = convertMessagesToModelMessages(messages)
-      // Simply yield from the RPC call
-      // The RPC layer handles WebSocket transport
-      yield* rpcCall(modelMessages, data)
+      // Pass messages as-is (UIMessages with parts preserved)
+      // Server-side chat() handles conversion to ModelMessages
+      yield* rpcCall(messages, data)
     },
   }
 }

package/src/index.ts

@@ -11,6 +11,7 @@ export type {
   ChatClientOptions,
   ChatRequestBody,
   InferChatMessages,
+  ChatClientState,
 } from './types'
 export { clientTools, createChatClientOptions } from './types'
 export type {

package/src/types.ts

@@ -26,6 +26,11 @@ export type ToolResultState =
   | 'complete' // Result is complete
   | 'error' // Error occurred
 
+/**
+ * ChatClient state - track the lifecycle of a chat
+ */
+export type ChatClientState = 'ready' | 'submitted' | 'streaming' | 'error'
+
 /**
  * Message parts - building blocks of UIMessage
  */
@@ -191,6 +196,11 @@ export interface ChatClientOptions<
    */
   onErrorChange?: (error: Error | undefined) => void
 
+  /**
+   * Callback when chat status changes
+   */
+  onStatusChange?: (status: ChatClientState) => void
+
   /**
    * Client-side tools with execution logic
    * When provided, tools with execute functions will be called automatically