@tanstack/ai-client 0.7.1 → 0.7.4

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  AnyClientTool,\n  ContentPart,\n  ModelMessage,\n  StreamChunk,\n} from '@tanstack/ai'\nimport type { ConnectionAdapter } from './connection-adapters'\nimport type { ChatClientEventEmitter } from './events'\nimport type {\n  ChatClientOptions,\n  ChatClientState,\n  MessagePart,\n  MultimodalContent,\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 pendingMessageBody: Record<string, any> | undefined = undefined\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  // Tracks whether a queued checkForContinuation was skipped because\n  // continuationPending was true (chained approval scenario)\n  private continuationSkipped = 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      onCustomEvent: (\n        eventType: string,\n        data: unknown,\n        context: { toolCallId?: string },\n      ) => 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        onCustomEvent: options.onCustomEvent || (() => {}),\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        onCustomEvent: (\n          eventType: string,\n          data: unknown,\n          context: { toolCallId?: string },\n        ) => {\n          this.callbacksRef.current.onCustomEvent(eventType, data, context)\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 | null> {\n    // Generate a stream ID for this streaming operation\n    this.currentStreamId = this.generateUniqueId('stream')\n\n    // Prepare for a new assistant message (created lazily on first content)\n    this.processor.prepareAssistantMessage()\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      // Track the message ID once the processor lazily creates it\n      if (!this.currentMessageId) {\n        const newMessageId =\n          this.processor.getCurrentAssistantMessageId() ?? null\n        if (newMessageId) {\n          this.currentMessageId = newMessageId\n          // Emit message appended event now that the assistant message exists\n          const assistantMessage = this.processor\n            .getMessages()\n            .find((m: UIMessage) => m.id === newMessageId)\n          if (assistantMessage) {\n            this.events.messageAppended(\n              assistantMessage,\n              this.currentStreamId || undefined,\n            )\n          }\n        }\n      }\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    // Get the message ID (may be null if no content arrived)\n    const messageId = this.processor.getCurrentAssistantMessageId()\n\n    // Clear the current stream and message IDs\n    this.currentStreamId = null\n    this.currentMessageId = null\n\n    // Return the assistant message if one was created\n    if (messageId) {\n      const messages = this.processor.getMessages()\n      return messages.find((m: UIMessage) => m.id === messageId) || null\n    }\n\n    return null\n  }\n\n  /**\n   * Send a message and stream the response.\n   * Supports both simple string content and multimodal content (images, audio, video, documents).\n   *\n   * @param content - The message content. Can be:\n   *   - A simple string for text-only messages\n   *   - A MultimodalContent object with content array and optional custom ID\n   * @param body - Optional body parameters to merge with the client's base body for this request.\n   *               Uses shallow merge with per-message body taking priority.\n   *\n   * @example\n   * ```ts\n   * // Simple text message\n   * await client.sendMessage('Hello!')\n   *\n   * // Text message with custom body params\n   * await client.sendMessage('Hello!', { temperature: 0.7 })\n   *\n   * // Multimodal message with image\n   * await client.sendMessage({\n   *   content: [\n   *     { type: 'text', content: 'What is in this image?' },\n   *     { type: 'image', source: { type: 'url', value: 'https://example.com/photo.jpg' } }\n   *   ]\n   * })\n   *\n   * // Multimodal message with custom ID and body params\n   * await client.sendMessage(\n   *   {\n   *     content: [\n   *       { type: 'text', content: 'Describe this audio' },\n   *       { type: 'audio', source: { type: 'data', value: 'base64...' } }\n   *     ],\n   *     id: 'custom-message-id'\n   *   },\n   *   { model: 'gpt-4-audio' }\n   * )\n   * ```\n   */\n  async sendMessage(\n    content: string | MultimodalContent,\n    body?: Record<string, any>,\n  ): Promise<void> {\n    const emptyMessage = typeof content === 'string' && !content.trim()\n    if (emptyMessage || this.isLoading) {\n      return\n    }\n    // Normalize input to extract content, id, and validate\n    const normalizedContent = this.normalizeMessageInput(content)\n\n    // Store the per-message body for use in streamResponse\n    this.pendingMessageBody = body\n\n    // Add user message via processor\n    const userMessage = this.processor.addUserMessage(\n      normalizedContent.content,\n      normalizedContent.id,\n    )\n    this.events.messageSent(userMessage.id, normalizedContent.content)\n\n    await this.streamResponse()\n  }\n\n  /**\n   * Normalize the message input to extract content and optional id.\n   * Trims string content automatically.\n   */\n  private normalizeMessageInput(input: string | MultimodalContent): {\n    content: string | Array<ContentPart>\n    id?: string\n  } {\n    if (typeof input === 'string') {\n      return { content: input.trim() }\n    }\n    return { content: input.content, id: input.id }\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(async () => {\n        await this.streamResponse()\n      })\n      return\n    }\n\n    await this.streamResponse()\n  }\n\n  /**\n   * Stream a response from the LLM.\n   * Returns true if the stream completed successfully, false on abort or error.\n   */\n  private async streamResponse(): Promise<boolean> {\n    // Guard against concurrent streams - if already loading, skip\n    if (this.isLoading) {\n      return false\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      // Merge body: base body + per-message body (per-message takes priority)\n      // Include conversationId for server-side event correlation\n      const mergedBody = {\n        ...this.body,\n        ...this.pendingMessageBody,\n        conversationId: this.uniqueId,\n      }\n\n      // Clear the pending message body after use\n      this.pendingMessageBody = undefined\n\n      // Connect and stream\n      const stream = this.connection.connect(\n        messages,\n        mergedBody,\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 false\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      this.pendingMessageBody = undefined // Ensure it's cleared even on error\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    return streamCompletedSuccessfully\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      this.continuationSkipped = true\n      return\n    }\n\n    if (this.shouldAutoSend()) {\n      this.continuationPending = true\n      this.continuationSkipped = false\n      let succeeded = false\n      try {\n        succeeded = await this.streamResponse()\n      } finally {\n        this.continuationPending = false\n      }\n      // If a queued check was skipped while continuationPending was true\n      // (e.g. a chained approval responded to during the stream), re-evaluate\n      // now that the flag is cleared. Only replay after a successful stream —\n      // aborted or errored streams should not trigger further continuation.\n      // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition -- mutated asynchronously during await\n      if (this.continuationSkipped && succeeded) {\n        this.continuationSkipped = false\n        await this.checkForContinuation()\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    onCustomEvent?: (\n      eventType: string,\n      data: unknown,\n      context: { toolCallId?: string },\n    ) => 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    if (options.onCustomEvent !== undefined) {\n      this.callbacksRef.current.onCustomEvent = options.onCustomEvent\n    }\n  }\n}\n"],"names":[],"mappings":";;AAuBO,MAAM,WAAW;AAAA,EAyCtB,YAAY,SAA4B;AArCxC,SAAQ,OAA4B,CAAA;AACpC,SAAQ,qBAAsD;AAC9D,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;AAG9B,SAAQ,sBAAsB;AAqB5B,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,QAClD,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;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,QACA,eAAe,CACb,WACA,MACA,YACG;AACH,eAAK,aAAa,QAAQ,cAAc,WAAW,MAAM,OAAO;AAAA,QAClE;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,QAC2B;AAE3B,SAAK,kBAAkB,KAAK,iBAAiB,QAAQ;AAGrD,SAAK,UAAU,wBAAA;AAGf,qBAAiB,SAAS,QAAQ;AAChC,WAAK,aAAa,QAAQ,QAAQ,KAAK;AACvC,WAAK,UAAU,aAAa,KAAK;AAGjC,UAAI,CAAC,KAAK,kBAAkB;AAC1B,cAAM,eACJ,KAAK,UAAU,6BAAA,KAAkC;AACnD,YAAI,cAAc;AAChB,eAAK,mBAAmB;AAExB,gBAAM,mBAAmB,KAAK,UAC3B,YAAA,EACA,KAAK,CAAC,MAAiB,EAAE,OAAO,YAAY;AAC/C,cAAI,kBAAkB;AACpB,iBAAK,OAAO;AAAA,cACV;AAAA,cACA,KAAK,mBAAmB;AAAA,YAAA;AAAA,UAE5B;AAAA,QACF;AAAA,MACF;AAGA,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,UAAM,YAAY,KAAK,UAAU,6BAAA;AAGjC,SAAK,kBAAkB;AACvB,SAAK,mBAAmB;AAGxB,QAAI,WAAW;AACb,YAAM,WAAW,KAAK,UAAU,YAAA;AAChC,aAAO,SAAS,KAAK,CAAC,MAAiB,EAAE,OAAO,SAAS,KAAK;AAAA,IAChE;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAyCA,MAAM,YACJ,SACA,MACe;AACf,UAAM,eAAe,OAAO,YAAY,YAAY,CAAC,QAAQ,KAAA;AAC7D,QAAI,gBAAgB,KAAK,WAAW;AAClC;AAAA,IACF;AAEA,UAAM,oBAAoB,KAAK,sBAAsB,OAAO;AAG5D,SAAK,qBAAqB;AAG1B,UAAM,cAAc,KAAK,UAAU;AAAA,MACjC,kBAAkB;AAAA,MAClB,kBAAkB;AAAA,IAAA;AAEpB,SAAK,OAAO,YAAY,YAAY,IAAI,kBAAkB,OAAO;AAEjE,UAAM,KAAK,eAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA,EAMQ,sBAAsB,OAG5B;AACA,QAAI,OAAO,UAAU,UAAU;AAC7B,aAAO,EAAE,SAAS,MAAM,OAAK;AAAA,IAC/B;AACA,WAAO,EAAE,SAAS,MAAM,SAAS,IAAI,MAAM,GAAA;AAAA,EAC7C;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,YAAY;AACrC,cAAM,KAAK,eAAA;AAAA,MACb,CAAC;AACD;AAAA,IACF;AAEA,UAAM,KAAK,eAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAc,iBAAmC;AAE/C,QAAI,KAAK,WAAW;AAClB,aAAO;AAAA,IACT;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;AAIhC,YAAM,aAAa;AAAA,QACjB,GAAG,KAAK;AAAA,QACR,GAAG,KAAK;AAAA,QACR,gBAAgB,KAAK;AAAA,MAAA;AAIvB,WAAK,qBAAqB;AAG1B,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,iBAAO;AAAA,QACT;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;AACvB,WAAK,qBAAqB;AAG1B,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;AAEA,WAAO;AAAA,EACT;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,WAAK,sBAAsB;AAC3B;AAAA,IACF;AAEA,QAAI,KAAK,kBAAkB;AACzB,WAAK,sBAAsB;AAC3B,WAAK,sBAAsB;AAC3B,UAAI,YAAY;AAChB,UAAI;AACF,oBAAY,MAAM,KAAK,eAAA;AAAA,MACzB,UAAA;AACE,aAAK,sBAAsB;AAAA,MAC7B;AAMA,UAAI,KAAK,uBAAuB,WAAW;AACzC,aAAK,sBAAsB;AAC3B,cAAM,KAAK,qBAAA;AAAA,MACb;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,SAaL;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;AACA,QAAI,QAAQ,kBAAkB,QAAW;AACvC,WAAK,aAAa,QAAQ,gBAAgB,QAAQ;AAAA,IACpD;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 { normalizeConnectionAdapter } from './connection-adapters'\nimport type {\n  AnyClientTool,\n  ContentPart,\n  ModelMessage,\n  StreamChunk,\n} from '@tanstack/ai'\nimport type {\n  ConnectionAdapter,\n  SubscribeConnectionAdapter,\n} from './connection-adapters'\nimport type { ChatClientEventEmitter } from './events'\nimport type {\n  ChatClientOptions,\n  ChatClientState,\n  ConnectionStatus,\n  MessagePart,\n  MultimodalContent,\n  ToolCallPart,\n  UIMessage,\n} from './types'\n\nexport class ChatClient {\n  private processor: StreamProcessor\n  private connection: SubscribeConnectionAdapter\n  private uniqueId: string\n  private body: Record<string, any> = {}\n  private pendingMessageBody: Record<string, any> | undefined = undefined\n  private isLoading = false\n  private isSubscribed = false\n  private error: Error | undefined = undefined\n  private status: ChatClientState = 'ready'\n  private connectionStatus: ConnectionStatus = 'disconnected'\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  private subscriptionAbortController: AbortController | null = null\n  private processingResolve: (() => void) | null = null\n  private errorReportedGeneration: number | null = null\n  private streamGeneration = 0\n  // Tracks whether a queued checkForContinuation was skipped because\n  // continuationPending was true (chained approval scenario)\n  private continuationSkipped = false\n  private sessionGenerating = false\n  private activeRunIds = new Set<string>()\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      onSubscriptionChange: (isSubscribed: boolean) => void\n      onConnectionStatusChange: (status: ConnectionStatus) => void\n      onSessionGeneratingChange: (isGenerating: boolean) => void\n      onCustomEvent: (\n        eventType: string,\n        data: unknown,\n        context: { toolCallId?: string },\n      ) => void\n    }\n  }\n\n  constructor(options: ChatClientOptions) {\n    this.uniqueId = options.id || this.generateUniqueId('chat')\n    this.body = options.body || {}\n    this.connection = normalizeConnectionAdapter(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        onSubscriptionChange: options.onSubscriptionChange || (() => {}),\n        onConnectionStatusChange:\n          options.onConnectionStatusChange || (() => {}),\n        onSessionGeneratingChange:\n          options.onSessionGeneratingChange || (() => {}),\n        onCustomEvent: options.onCustomEvent || (() => {}),\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          const assistantMessageId =\n            this.processor.getCurrentAssistantMessageId()\n          if (!assistantMessageId) {\n            return\n          }\n          const messages = this.processor.getMessages()\n          const assistantMessage = messages.find(\n            (m: UIMessage) => m.id === assistantMessageId,\n          )\n          if (assistantMessage) {\n            this.currentMessageId = assistantMessage.id\n            this.events.messageAppended(\n              assistantMessage,\n              this.currentStreamId || undefined,\n            )\n          }\n        },\n        onStreamEnd: (message: UIMessage) => {\n          this.callbacksRef.current.onFinish(message)\n          this.setStatus('ready')\n          // Resolve the processing-complete promise so streamResponse can continue\n          this.resolveProcessing()\n        },\n        onError: (error: Error) => {\n          this.reportStreamError(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        onCustomEvent: (\n          eventType: string,\n          data: unknown,\n          context: { toolCallId?: string },\n        ) => {\n          this.callbacksRef.current.onCustomEvent(eventType, data, context)\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 setIsSubscribed(isSubscribed: boolean): void {\n    this.isSubscribed = isSubscribed\n    this.callbacksRef.current.onSubscriptionChange(isSubscribed)\n  }\n\n  private setConnectionStatus(status: ConnectionStatus): void {\n    this.connectionStatus = status\n    this.callbacksRef.current.onConnectionStatusChange(status)\n  }\n\n  private setSessionGenerating(isGenerating: boolean): void {\n    if (this.sessionGenerating === isGenerating) return\n    this.sessionGenerating = isGenerating\n    this.callbacksRef.current.onSessionGeneratingChange(isGenerating)\n  }\n\n  private resetSessionGenerating(): void {\n    this.activeRunIds.clear()\n    this.setSessionGenerating(false)\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  private abortSubscriptionLoop(): void {\n    this.subscriptionAbortController?.abort()\n    this.subscriptionAbortController = null\n  }\n\n  private resolveProcessing(): void {\n    this.processingResolve?.()\n    this.processingResolve = null\n  }\n\n  private cancelInFlightStream(options?: {\n    setReadyStatus?: boolean\n    abortSubscription?: boolean\n  }): void {\n    this.abortController?.abort()\n    this.abortController = null\n    if (options?.abortSubscription) {\n      this.abortSubscriptionLoop()\n    }\n    this.resolveProcessing()\n    this.setIsLoading(false)\n    if (options?.setReadyStatus) {\n      this.setStatus('ready')\n    }\n  }\n\n  private reportStreamError(error: Error): void {\n    const alreadyReported =\n      this.errorReportedGeneration === this.streamGeneration\n    this.setError(error)\n    // Preserve request-level error semantics even if a RUN_ERROR arrives\n    // slightly after loading flips false during stream teardown.\n    if (\n      this.isLoading ||\n      this.status === 'submitted' ||\n      this.status === 'streaming'\n    ) {\n      this.setStatus('error')\n    }\n    if (!alreadyReported) {\n      this.errorReportedGeneration = this.streamGeneration\n      this.callbacksRef.current.onError(error)\n    }\n  }\n\n  /**\n   * Start the background subscription loop.\n   */\n  private startSubscription(): void {\n    this.subscriptionAbortController = new AbortController()\n    const signal = this.subscriptionAbortController.signal\n\n    this.consumeSubscription(signal)\n      .catch((err) => {\n        if (err instanceof Error && err.name !== 'AbortError') {\n          this.setConnectionStatus('error')\n          this.resetSessionGenerating()\n          this.setIsSubscribed(false)\n          this.reportStreamError(err)\n        }\n        // Resolve pending processing so streamResponse doesn't hang\n        this.resolveProcessing()\n      })\n      .finally(() => {\n        // Ignore stale loops that were superseded by a restart.\n        if (this.subscriptionAbortController?.signal !== signal) {\n          return\n        }\n        this.subscriptionAbortController = null\n        if (!signal.aborted && this.isSubscribed) {\n          this.setIsSubscribed(false)\n          if (this.connectionStatus !== 'error') {\n            this.setConnectionStatus('disconnected')\n          }\n        }\n      })\n  }\n\n  /**\n   * Consume chunks from the connection subscription.\n   */\n  private async consumeSubscription(signal: AbortSignal): Promise<void> {\n    const stream = this.connection.subscribe(signal)\n    for await (const chunk of stream) {\n      if (signal.aborted) break\n      if (this.connectionStatus === 'connecting') {\n        this.setConnectionStatus('connected')\n      }\n      this.callbacksRef.current.onChunk(chunk)\n      this.processor.processChunk(chunk)\n      if (chunk.type === 'RUN_STARTED') {\n        this.activeRunIds.add(chunk.runId)\n        this.setSessionGenerating(true)\n      }\n      // RUN_FINISHED / RUN_ERROR signal run completion — resolve processing\n      // (redundant if onStreamEnd already resolved it, harmless)\n      if (chunk.type === 'RUN_FINISHED' || chunk.type === 'RUN_ERROR') {\n        if (chunk.runId) {\n          this.activeRunIds.delete(chunk.runId)\n        } else if (chunk.type === 'RUN_ERROR') {\n          // RUN_ERROR without runId is a session-level error; clear all runs\n          this.activeRunIds.clear()\n        }\n        this.setSessionGenerating(this.activeRunIds.size > 0)\n        this.resolveProcessing()\n      }\n      // Yield control back to event loop for UI updates\n      await new Promise((resolve) => setTimeout(resolve, 0))\n    }\n  }\n\n  /**\n   * Ensure subscription loop is running, starting it if needed.\n   */\n  private ensureSubscription(): void {\n    if (!this.isSubscribed) {\n      this.subscribe()\n      return\n    }\n    if (\n      !this.subscriptionAbortController ||\n      this.subscriptionAbortController.signal.aborted\n    ) {\n      this.subscribe({ restart: true })\n    }\n  }\n\n  /**\n   * Create a promise that resolves when onStreamEnd fires.\n   * Used by streamResponse to await processing completion.\n   */\n  private waitForProcessing(): Promise<void> {\n    // Resolve any stale promise (e.g., from a previous aborted request)\n    this.resolveProcessing()\n    return new Promise<void>((resolve) => {\n      this.processingResolve = resolve\n    })\n  }\n\n  /**\n   * Send a message and stream the response.\n   * Supports both simple string content and multimodal content (images, audio, video, documents).\n   *\n   * @param content - The message content. Can be:\n   *   - A simple string for text-only messages\n   *   - A MultimodalContent object with content array and optional custom ID\n   * @param body - Optional body parameters to merge with the client's base body for this request.\n   *               Uses shallow merge with per-message body taking priority.\n   *\n   * @example\n   * ```ts\n   * // Simple text message\n   * await client.sendMessage('Hello!')\n   *\n   * // Text message with custom body params\n   * await client.sendMessage('Hello!', { temperature: 0.7 })\n   *\n   * // Multimodal message with image\n   * await client.sendMessage({\n   *   content: [\n   *     { type: 'text', content: 'What is in this image?' },\n   *     { type: 'image', source: { type: 'url', value: 'https://example.com/photo.jpg' } }\n   *   ]\n   * })\n   *\n   * // Multimodal message with custom ID and body params\n   * await client.sendMessage(\n   *   {\n   *     content: [\n   *       { type: 'text', content: 'Describe this audio' },\n   *       { type: 'audio', source: { type: 'data', value: 'base64...' } }\n   *     ],\n   *     id: 'custom-message-id'\n   *   },\n   *   { model: 'gpt-4-audio' }\n   * )\n   * ```\n   */\n  async sendMessage(\n    content: string | MultimodalContent,\n    body?: Record<string, any>,\n  ): Promise<void> {\n    const emptyMessage = typeof content === 'string' && !content.trim()\n    if (emptyMessage || this.isLoading) {\n      return\n    }\n    // Normalize input to extract content, id, and validate\n    const normalizedContent = this.normalizeMessageInput(content)\n\n    // Store the per-message body for use in streamResponse\n    this.pendingMessageBody = body\n\n    // Add user message via processor\n    const userMessage = this.processor.addUserMessage(\n      normalizedContent.content,\n      normalizedContent.id,\n    )\n    this.events.messageSent(userMessage.id, normalizedContent.content)\n\n    await this.streamResponse()\n  }\n\n  /**\n   * Normalize the message input to extract content and optional id.\n   * Trims string content automatically.\n   */\n  private normalizeMessageInput(input: string | MultimodalContent): {\n    content: string | Array<ContentPart>\n    id?: string\n  } {\n    if (typeof input === 'string') {\n      return { content: input.trim() }\n    }\n    return { content: input.content, id: input.id }\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(async () => {\n        await this.streamResponse()\n      })\n      return\n    }\n\n    await this.streamResponse()\n  }\n\n  /**\n   * Stream a response from the LLM.\n   * Returns true if the stream completed successfully, false on abort or error.\n   */\n  private async streamResponse(): Promise<boolean> {\n    // Guard against concurrent streams - if already loading, skip\n    if (this.isLoading) {\n      return false\n    }\n\n    // Track generation so a superseded stream's cleanup doesn't clobber the new one\n    const generation = ++this.streamGeneration\n\n    this.setIsLoading(true)\n    this.setStatus('submitted')\n    this.setError(undefined)\n    this.errorReportedGeneration = null\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      // Merge body: base body + per-message body (per-message takes priority)\n      // Include conversationId for server-side event correlation\n      const mergedBody = {\n        ...this.body,\n        ...this.pendingMessageBody,\n        conversationId: this.uniqueId,\n      }\n\n      // Clear the pending message body after use\n      this.pendingMessageBody = undefined\n\n      // Generate stream ID — assistant message will be created by stream events\n      this.currentStreamId = this.generateUniqueId('stream')\n      this.currentMessageId = null\n\n      // Reset processor stream state for new response — prevents stale\n      // messageStates entries (from a previous stream) from blocking\n      // creation of a new assistant message (e.g. after reload).\n      this.processor.prepareAssistantMessage()\n\n      // Ensure subscription loop is running\n      this.ensureSubscription()\n\n      // Set up promise that resolves when onStreamEnd fires\n      const processingComplete = this.waitForProcessing()\n\n      // Send through normalized connection (pushes chunks to subscription queue)\n      await this.connection.send(\n        messages,\n        mergedBody,\n        this.abortController.signal,\n      )\n\n      // Wait for subscription loop to finish processing all chunks\n      await processingComplete\n\n      // If this stream was superseded (e.g. by reload()), bail out —\n      // the new stream owns the processor and processingResolve now.\n      if (generation !== this.streamGeneration) {\n        return false\n      }\n\n      // A RUN_ERROR from the stream transitions status to error.\n      // Do not treat this stream as a successful completion.\n      if (this.status === 'error') {\n        return false\n      }\n\n      // Wait for pending client tool executions\n      if (this.pendingToolExecutions.size > 0) {\n        await Promise.all(this.pendingToolExecutions.values())\n      }\n\n      // Finalize (idempotent — may already be done by RUN_FINISHED handler)\n      this.processor.finalizeStream()\n      streamCompletedSuccessfully = true\n    } catch (err) {\n      if (err instanceof Error) {\n        if (err.name === 'AbortError') {\n          return false\n        }\n        if (generation === this.streamGeneration) {\n          this.reportStreamError(err)\n        }\n      }\n    } finally {\n      // Only clean up if this is still the active stream.\n      // A superseded stream (e.g. reload() started a new one) must not\n      // clobber the new stream's abortController or isLoading state.\n      if (generation === this.streamGeneration) {\n        this.currentStreamId = null\n        this.currentMessageId = null\n        this.abortController = null\n        this.setIsLoading(false)\n        this.pendingMessageBody = undefined // Ensure it's cleared even on error\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    return streamCompletedSuccessfully\n  }\n\n  /**\n   * Start the client subscription loop.\n   * This controls the connection lifecycle independently from request lifecycle.\n   */\n  subscribe(options?: { restart?: boolean }): void {\n    const restart = options?.restart === true\n    if (this.isSubscribed && !restart) {\n      return\n    }\n\n    if (this.isSubscribed && restart) {\n      this.abortSubscriptionLoop()\n    }\n\n    this.setIsSubscribed(true)\n    this.setConnectionStatus('connecting')\n    this.startSubscription()\n  }\n\n  /**\n   * Unsubscribe and fully tear down live behavior.\n   * This aborts an in-flight request and the subscription loop.\n   */\n  unsubscribe(): void {\n    this.cancelInFlightStream({\n      setReadyStatus: true,\n      abortSubscription: true,\n    })\n    this.resetSessionGenerating()\n    this.setIsSubscribed(false)\n    this.setConnectionStatus('disconnected')\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    // Cancel any active stream before reloading\n    if (this.isLoading) {\n      this.cancelInFlightStream()\n    }\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    this.cancelInFlightStream({ setReadyStatus: true })\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      this.continuationSkipped = true\n      return\n    }\n\n    if (this.shouldAutoSend()) {\n      this.continuationPending = true\n      this.continuationSkipped = false\n      let succeeded = false\n      try {\n        succeeded = await this.streamResponse()\n      } finally {\n        this.continuationPending = false\n      }\n      // If a queued check was skipped while continuationPending was true\n      // (e.g. a chained approval responded to during the stream), re-evaluate\n      // now that the flag is cleared. Only replay after a successful stream —\n      // aborted or errored streams should not trigger further continuation.\n      // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition -- mutated asynchronously during await\n      if (this.continuationSkipped && succeeded) {\n        this.continuationSkipped = false\n        await this.checkForContinuation()\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 whether the subscription loop is active\n   */\n  getIsSubscribed(): boolean {\n    return this.isSubscribed\n  }\n\n  /**\n   * Get current connection lifecycle status\n   */\n  getConnectionStatus(): ConnectionStatus {\n    return this.connectionStatus\n  }\n\n  /**\n   * Whether the shared session is actively generating.\n   * Derived from stream run events (RUN_STARTED / RUN_FINISHED / RUN_ERROR).\n   * Unlike `isLoading` (request-local), this reflects shared generation\n   * activity visible to all subscribers (e.g. across tabs/devices).\n   */\n  getSessionGenerating(): boolean {\n    return this.sessionGenerating\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    onSubscriptionChange?: (isSubscribed: boolean) => void\n    onConnectionStatusChange?: (status: ConnectionStatus) => void\n    onSessionGeneratingChange?: (isGenerating: boolean) => void\n    onCustomEvent?: (\n      eventType: string,\n      data: unknown,\n      context: { toolCallId?: string },\n    ) => void\n  }): void {\n    if (options.connection !== undefined) {\n      const wasSubscribed = this.isSubscribed\n\n      if (this.isLoading) {\n        this.cancelInFlightStream({\n          setReadyStatus: true,\n          abortSubscription: true,\n        })\n      } else if (wasSubscribed) {\n        this.abortSubscriptionLoop()\n      }\n\n      this.resetSessionGenerating()\n      this.setIsSubscribed(false)\n      this.setConnectionStatus('disconnected')\n      this.connection = normalizeConnectionAdapter(options.connection)\n\n      if (wasSubscribed) {\n        this.subscribe()\n      }\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    if (options.onSubscriptionChange !== undefined) {\n      this.callbacksRef.current.onSubscriptionChange =\n        options.onSubscriptionChange\n    }\n    if (options.onConnectionStatusChange !== undefined) {\n      this.callbacksRef.current.onConnectionStatusChange =\n        options.onConnectionStatusChange\n    }\n    if (options.onSessionGeneratingChange !== undefined) {\n      this.callbacksRef.current.onSessionGeneratingChange =\n        options.onSessionGeneratingChange\n    }\n    if (options.onCustomEvent !== undefined) {\n      this.callbacksRef.current.onCustomEvent = options.onCustomEvent\n    }\n  }\n}\n"],"names":[],"mappings":";;;AA4BO,MAAM,WAAW;AAAA,EAoDtB,YAAY,SAA4B;AAhDxC,SAAQ,OAA4B,CAAA;AACpC,SAAQ,qBAAsD;AAC9D,SAAQ,YAAY;AACpB,SAAQ,eAAe;AACvB,SAAQ,QAA2B;AACnC,SAAQ,SAA0B;AAClC,SAAQ,mBAAqC;AAC7C,SAAQ,kBAA0C;AAGlD,SAAQ,kBAAiC;AACzC,SAAQ,mBAAkC;AAC1C,SAAQ,oBAAgD,CAAA;AAExD,SAAQ,4CAAwD,IAAA;AAEhE,SAAQ,sBAAsB;AAC9B,SAAQ,8BAAsD;AAC9D,SAAQ,oBAAyC;AACjD,SAAQ,0BAAyC;AACjD,SAAQ,mBAAmB;AAG3B,SAAQ,sBAAsB;AAC9B,SAAQ,oBAAoB;AAC5B,SAAQ,mCAAmB,IAAA;AAwBzB,SAAK,WAAW,QAAQ,MAAM,KAAK,iBAAiB,MAAM;AAC1D,SAAK,OAAO,QAAQ,QAAQ,CAAA;AAC5B,SAAK,aAAa,2BAA2B,QAAQ,UAAU;AAC/D,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,QAClD,sBAAsB,QAAQ,yBAAyB,MAAM;AAAA,QAAC;AAAA,QAC9D,0BACE,QAAQ,6BAA6B,MAAM;AAAA,QAAC;AAAA,QAC9C,2BACE,QAAQ,8BAA8B,MAAM;AAAA,QAAC;AAAA,QAC/C,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;AACnB,eAAK,UAAU,WAAW;AAC1B,gBAAM,qBACJ,KAAK,UAAU,6BAAA;AACjB,cAAI,CAAC,oBAAoB;AACvB;AAAA,UACF;AACA,gBAAM,WAAW,KAAK,UAAU,YAAA;AAChC,gBAAM,mBAAmB,SAAS;AAAA,YAChC,CAAC,MAAiB,EAAE,OAAO;AAAA,UAAA;AAE7B,cAAI,kBAAkB;AACpB,iBAAK,mBAAmB,iBAAiB;AACzC,iBAAK,OAAO;AAAA,cACV;AAAA,cACA,KAAK,mBAAmB;AAAA,YAAA;AAAA,UAE5B;AAAA,QACF;AAAA,QACA,aAAa,CAAC,YAAuB;AACnC,eAAK,aAAa,QAAQ,SAAS,OAAO;AAC1C,eAAK,UAAU,OAAO;AAEtB,eAAK,kBAAA;AAAA,QACP;AAAA,QACA,SAAS,CAAC,UAAiB;AACzB,eAAK,kBAAkB,KAAK;AAAA,QAC9B;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,QACA,eAAe,CACb,WACA,MACA,YACG;AACH,eAAK,aAAa,QAAQ,cAAc,WAAW,MAAM,OAAO;AAAA,QAClE;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,gBAAgB,cAA6B;AACnD,SAAK,eAAe;AACpB,SAAK,aAAa,QAAQ,qBAAqB,YAAY;AAAA,EAC7D;AAAA,EAEQ,oBAAoB,QAAgC;AAC1D,SAAK,mBAAmB;AACxB,SAAK,aAAa,QAAQ,yBAAyB,MAAM;AAAA,EAC3D;AAAA,EAEQ,qBAAqB,cAA6B;AACxD,QAAI,KAAK,sBAAsB,aAAc;AAC7C,SAAK,oBAAoB;AACzB,SAAK,aAAa,QAAQ,0BAA0B,YAAY;AAAA,EAClE;AAAA,EAEQ,yBAA+B;AACrC,SAAK,aAAa,MAAA;AAClB,SAAK,qBAAqB,KAAK;AAAA,EACjC;AAAA,EAEQ,SAAS,OAAgC;AAC/C,SAAK,QAAQ;AACb,SAAK,aAAa,QAAQ,cAAc,KAAK;AAC7C,SAAK,OAAO,aAAa,OAAO,WAAW,IAAI;AAAA,EACjD;AAAA,EAEQ,wBAA8B;AACpC,SAAK,6BAA6B,MAAA;AAClC,SAAK,8BAA8B;AAAA,EACrC;AAAA,EAEQ,oBAA0B;AAChC,SAAK,oBAAA;AACL,SAAK,oBAAoB;AAAA,EAC3B;AAAA,EAEQ,qBAAqB,SAGpB;AACP,SAAK,iBAAiB,MAAA;AACtB,SAAK,kBAAkB;AACvB,QAAI,SAAS,mBAAmB;AAC9B,WAAK,sBAAA;AAAA,IACP;AACA,SAAK,kBAAA;AACL,SAAK,aAAa,KAAK;AACvB,QAAI,SAAS,gBAAgB;AAC3B,WAAK,UAAU,OAAO;AAAA,IACxB;AAAA,EACF;AAAA,EAEQ,kBAAkB,OAAoB;AAC5C,UAAM,kBACJ,KAAK,4BAA4B,KAAK;AACxC,SAAK,SAAS,KAAK;AAGnB,QACE,KAAK,aACL,KAAK,WAAW,eAChB,KAAK,WAAW,aAChB;AACA,WAAK,UAAU,OAAO;AAAA,IACxB;AACA,QAAI,CAAC,iBAAiB;AACpB,WAAK,0BAA0B,KAAK;AACpC,WAAK,aAAa,QAAQ,QAAQ,KAAK;AAAA,IACzC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,oBAA0B;AAChC,SAAK,8BAA8B,IAAI,gBAAA;AACvC,UAAM,SAAS,KAAK,4BAA4B;AAEhD,SAAK,oBAAoB,MAAM,EAC5B,MAAM,CAAC,QAAQ;AACd,UAAI,eAAe,SAAS,IAAI,SAAS,cAAc;AACrD,aAAK,oBAAoB,OAAO;AAChC,aAAK,uBAAA;AACL,aAAK,gBAAgB,KAAK;AAC1B,aAAK,kBAAkB,GAAG;AAAA,MAC5B;AAEA,WAAK,kBAAA;AAAA,IACP,CAAC,EACA,QAAQ,MAAM;AAEb,UAAI,KAAK,6BAA6B,WAAW,QAAQ;AACvD;AAAA,MACF;AACA,WAAK,8BAA8B;AACnC,UAAI,CAAC,OAAO,WAAW,KAAK,cAAc;AACxC,aAAK,gBAAgB,KAAK;AAC1B,YAAI,KAAK,qBAAqB,SAAS;AACrC,eAAK,oBAAoB,cAAc;AAAA,QACzC;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACL;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,oBAAoB,QAAoC;AACpE,UAAM,SAAS,KAAK,WAAW,UAAU,MAAM;AAC/C,qBAAiB,SAAS,QAAQ;AAChC,UAAI,OAAO,QAAS;AACpB,UAAI,KAAK,qBAAqB,cAAc;AAC1C,aAAK,oBAAoB,WAAW;AAAA,MACtC;AACA,WAAK,aAAa,QAAQ,QAAQ,KAAK;AACvC,WAAK,UAAU,aAAa,KAAK;AACjC,UAAI,MAAM,SAAS,eAAe;AAChC,aAAK,aAAa,IAAI,MAAM,KAAK;AACjC,aAAK,qBAAqB,IAAI;AAAA,MAChC;AAGA,UAAI,MAAM,SAAS,kBAAkB,MAAM,SAAS,aAAa;AAC/D,YAAI,MAAM,OAAO;AACf,eAAK,aAAa,OAAO,MAAM,KAAK;AAAA,QACtC,WAAW,MAAM,SAAS,aAAa;AAErC,eAAK,aAAa,MAAA;AAAA,QACpB;AACA,aAAK,qBAAqB,KAAK,aAAa,OAAO,CAAC;AACpD,aAAK,kBAAA;AAAA,MACP;AAEA,YAAM,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,CAAC,CAAC;AAAA,IACvD;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,qBAA2B;AACjC,QAAI,CAAC,KAAK,cAAc;AACtB,WAAK,UAAA;AACL;AAAA,IACF;AACA,QACE,CAAC,KAAK,+BACN,KAAK,4BAA4B,OAAO,SACxC;AACA,WAAK,UAAU,EAAE,SAAS,KAAA,CAAM;AAAA,IAClC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA,EAMQ,oBAAmC;AAEzC,SAAK,kBAAA;AACL,WAAO,IAAI,QAAc,CAAC,YAAY;AACpC,WAAK,oBAAoB;AAAA,IAC3B,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAyCA,MAAM,YACJ,SACA,MACe;AACf,UAAM,eAAe,OAAO,YAAY,YAAY,CAAC,QAAQ,KAAA;AAC7D,QAAI,gBAAgB,KAAK,WAAW;AAClC;AAAA,IACF;AAEA,UAAM,oBAAoB,KAAK,sBAAsB,OAAO;AAG5D,SAAK,qBAAqB;AAG1B,UAAM,cAAc,KAAK,UAAU;AAAA,MACjC,kBAAkB;AAAA,MAClB,kBAAkB;AAAA,IAAA;AAEpB,SAAK,OAAO,YAAY,YAAY,IAAI,kBAAkB,OAAO;AAEjE,UAAM,KAAK,eAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA,EAMQ,sBAAsB,OAG5B;AACA,QAAI,OAAO,UAAU,UAAU;AAC7B,aAAO,EAAE,SAAS,MAAM,OAAK;AAAA,IAC/B;AACA,WAAO,EAAE,SAAS,MAAM,SAAS,IAAI,MAAM,GAAA;AAAA,EAC7C;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,YAAY;AACrC,cAAM,KAAK,eAAA;AAAA,MACb,CAAC;AACD;AAAA,IACF;AAEA,UAAM,KAAK,eAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAc,iBAAmC;AAE/C,QAAI,KAAK,WAAW;AAClB,aAAO;AAAA,IACT;AAGA,UAAM,aAAa,EAAE,KAAK;AAE1B,SAAK,aAAa,IAAI;AACtB,SAAK,UAAU,WAAW;AAC1B,SAAK,SAAS,MAAS;AACvB,SAAK,0BAA0B;AAC/B,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;AAIhC,YAAM,aAAa;AAAA,QACjB,GAAG,KAAK;AAAA,QACR,GAAG,KAAK;AAAA,QACR,gBAAgB,KAAK;AAAA,MAAA;AAIvB,WAAK,qBAAqB;AAG1B,WAAK,kBAAkB,KAAK,iBAAiB,QAAQ;AACrD,WAAK,mBAAmB;AAKxB,WAAK,UAAU,wBAAA;AAGf,WAAK,mBAAA;AAGL,YAAM,qBAAqB,KAAK,kBAAA;AAGhC,YAAM,KAAK,WAAW;AAAA,QACpB;AAAA,QACA;AAAA,QACA,KAAK,gBAAgB;AAAA,MAAA;AAIvB,YAAM;AAIN,UAAI,eAAe,KAAK,kBAAkB;AACxC,eAAO;AAAA,MACT;AAIA,UAAI,KAAK,WAAW,SAAS;AAC3B,eAAO;AAAA,MACT;AAGA,UAAI,KAAK,sBAAsB,OAAO,GAAG;AACvC,cAAM,QAAQ,IAAI,KAAK,sBAAsB,QAAQ;AAAA,MACvD;AAGA,WAAK,UAAU,eAAA;AACf,oCAA8B;AAAA,IAChC,SAAS,KAAK;AACZ,UAAI,eAAe,OAAO;AACxB,YAAI,IAAI,SAAS,cAAc;AAC7B,iBAAO;AAAA,QACT;AACA,YAAI,eAAe,KAAK,kBAAkB;AACxC,eAAK,kBAAkB,GAAG;AAAA,QAC5B;AAAA,MACF;AAAA,IACF,UAAA;AAIE,UAAI,eAAe,KAAK,kBAAkB;AACxC,aAAK,kBAAkB;AACvB,aAAK,mBAAmB;AACxB,aAAK,kBAAkB;AACvB,aAAK,aAAa,KAAK;AACvB,aAAK,qBAAqB;AAG1B,cAAM,KAAK,uBAAA;AAGX,YAAI,6BAA6B;AAC/B,gBAAM,WAAW,KAAK,UAAU,YAAA;AAChC,gBAAM,WAAW,SAAS,GAAG,EAAE,GAAG,MAAM,GAAG,EAAE;AAE7C,cAAI,UAAU,SAAS,iBAAiB,KAAK,kBAAkB;AAC7D,gBAAI;AACF,oBAAM,KAAK,qBAAA;AAAA,YACb,SAAS,OAAO;AACd,sBAAQ,MAAM,8CAA8C,KAAK;AAAA,YACnE;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,UAAU,SAAuC;AAC/C,UAAM,UAAU,SAAS,YAAY;AACrC,QAAI,KAAK,gBAAgB,CAAC,SAAS;AACjC;AAAA,IACF;AAEA,QAAI,KAAK,gBAAgB,SAAS;AAChC,WAAK,sBAAA;AAAA,IACP;AAEA,SAAK,gBAAgB,IAAI;AACzB,SAAK,oBAAoB,YAAY;AACrC,SAAK,kBAAA;AAAA,EACP;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,cAAoB;AAClB,SAAK,qBAAqB;AAAA,MACxB,gBAAgB;AAAA,MAChB,mBAAmB;AAAA,IAAA,CACpB;AACD,SAAK,uBAAA;AACL,SAAK,gBAAgB,KAAK;AAC1B,SAAK,oBAAoB,cAAc;AAAA,EACzC;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;AAGjC,QAAI,KAAK,WAAW;AAClB,WAAK,qBAAA;AAAA,IACP;AAEA,SAAK,OAAO,SAAS,oBAAoB;AAGzC,SAAK,UAAU,oBAAoB,oBAAoB;AAGvD,UAAM,KAAK,eAAA;AAAA,EACb;AAAA;AAAA;AAAA;AAAA,EAKA,OAAa;AACX,SAAK,qBAAqB,EAAE,gBAAgB,KAAA,CAAM;AAClD,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,WAAK,sBAAsB;AAC3B;AAAA,IACF;AAEA,QAAI,KAAK,kBAAkB;AACzB,WAAK,sBAAsB;AAC3B,WAAK,sBAAsB;AAC3B,UAAI,YAAY;AAChB,UAAI;AACF,oBAAY,MAAM,KAAK,eAAA;AAAA,MACzB,UAAA;AACE,aAAK,sBAAsB;AAAA,MAC7B;AAMA,UAAI,KAAK,uBAAuB,WAAW;AACzC,aAAK,sBAAsB;AAC3B,cAAM,KAAK,qBAAA;AAAA,MACb;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,kBAA2B;AACzB,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAKA,sBAAwC;AACtC,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,uBAAgC;AAC9B,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,SAgBL;AACP,QAAI,QAAQ,eAAe,QAAW;AACpC,YAAM,gBAAgB,KAAK;AAE3B,UAAI,KAAK,WAAW;AAClB,aAAK,qBAAqB;AAAA,UACxB,gBAAgB;AAAA,UAChB,mBAAmB;AAAA,QAAA,CACpB;AAAA,MACH,WAAW,eAAe;AACxB,aAAK,sBAAA;AAAA,MACP;AAEA,WAAK,uBAAA;AACL,WAAK,gBAAgB,KAAK;AAC1B,WAAK,oBAAoB,cAAc;AACvC,WAAK,aAAa,2BAA2B,QAAQ,UAAU;AAE/D,UAAI,eAAe;AACjB,aAAK,UAAA;AAAA,MACP;AAAA,IACF;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;AACA,QAAI,QAAQ,yBAAyB,QAAW;AAC9C,WAAK,aAAa,QAAQ,uBACxB,QAAQ;AAAA,IACZ;AACA,QAAI,QAAQ,6BAA6B,QAAW;AAClD,WAAK,aAAa,QAAQ,2BACxB,QAAQ;AAAA,IACZ;AACA,QAAI,QAAQ,8BAA8B,QAAW;AACnD,WAAK,aAAa,QAAQ,4BACxB,QAAQ;AAAA,IACZ;AACA,QAAI,QAAQ,kBAAkB,QAAW;AACvC,WAAK,aAAa,QAAQ,gBAAgB,QAAQ;AAAA,IACpD;AAAA,EACF;AACF;"}

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

@@ -1,16 +1,32 @@
 import { ModelMessage, StreamChunk, UIMessage } from '@tanstack/ai';
-/**
- * Connection adapter interface - converts a connection into a stream of chunks
- */
-export interface ConnectionAdapter {
+export interface ConnectConnectionAdapter {
     /**
-     * Connect and return an async iterable of StreamChunks
-     * @param messages - The messages to send (UIMessages or ModelMessages)
-     * @param data - Additional data to send
-     * @param abortSignal - Optional abort signal for request cancellation
+     * Connect and return an async iterable of StreamChunks.
      */
     connect: (messages: Array<UIMessage> | Array<ModelMessage>, data?: Record<string, any>, abortSignal?: AbortSignal) => AsyncIterable<StreamChunk>;
 }
+export interface SubscribeConnectionAdapter {
+    /**
+     * Subscribe to stream chunks.
+     */
+    subscribe: (abortSignal?: AbortSignal) => AsyncIterable<StreamChunk>;
+    /**
+     * Send a request; chunks arrive through subscribe().
+     */
+    send: (messages: Array<UIMessage> | Array<ModelMessage>, data?: Record<string, any>, abortSignal?: AbortSignal) => Promise<void>;
+}
+/**
+ * Connection adapter union.
+ * Provide either `connect`, or `subscribe` + `send`.
+ */
+export type ConnectionAdapter = ConnectConnectionAdapter | SubscribeConnectionAdapter;
+/**
+ * Normalize a ConnectionAdapter to subscribe/send operations.
+ *
+ * If a connection provides native subscribe/send, that mode is used.
+ * Otherwise, connect() is wrapped using an async queue.
+ */
+export declare function normalizeConnectionAdapter(connection: ConnectionAdapter | undefined): SubscribeConnectionAdapter;
 /**
  * Options for fetch-based connection adapters
  */
@@ -55,7 +71,7 @@ export interface FetchConnectionOptions {
  * }));
  * ```
  */
-export declare function fetchServerSentEvents(url: string | (() => string), options?: FetchConnectionOptions | (() => FetchConnectionOptions | Promise<FetchConnectionOptions>)): ConnectionAdapter;
+export declare function fetchServerSentEvents(url: string | (() => string), options?: FetchConnectionOptions | (() => FetchConnectionOptions | Promise<FetchConnectionOptions>)): ConnectConnectionAdapter;
 /**
  * Create an HTTP streaming connection adapter (for raw streaming without SSE format)
  *
@@ -90,7 +106,7 @@ export declare function fetchServerSentEvents(url: string | (() => string), opti
  * }));
  * ```
  */
-export declare function fetchHttpStream(url: string | (() => string), options?: FetchConnectionOptions | (() => FetchConnectionOptions | Promise<FetchConnectionOptions>)): ConnectionAdapter;
+export declare function fetchHttpStream(url: string | (() => string), options?: FetchConnectionOptions | (() => FetchConnectionOptions | Promise<FetchConnectionOptions>)): ConnectConnectionAdapter;
 /**
  * Create a direct stream connection adapter (for server functions or direct streams)
  *
@@ -105,7 +121,7 @@ export declare function fetchHttpStream(url: string | (() => string), options?:
  * const client = new ChatClient({ connection });
  * ```
  */
-export declare function stream(streamFactory: (messages: Array<UIMessage> | 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>): ConnectConnectionAdapter;
 /**
  * Create an RPC stream connection adapter (for RPC-based streaming like Cap'n Web RPC)
  *
@@ -122,4 +138,4 @@ export declare function stream(streamFactory: (messages: Array<UIMessage> | Arra
  * const client = new ChatClient({ connection });
  * ```
  */
-export declare function rpcStream(rpcCall: (messages: Array<UIMessage> | 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>): ConnectConnectionAdapter;

package/dist/esm/connection-adapters.js

@@ -37,6 +37,98 @@ async function* readStreamLines(reader, abortSignal) {
     reader.releaseLock();
   }
 }
+function normalizeConnectionAdapter(connection) {
+  if (!connection) {
+    throw new Error("Connection adapter is required");
+  }
+  const hasConnect = "connect" in connection;
+  const hasSubscribe = "subscribe" in connection;
+  const hasSend = "send" in connection;
+  if (hasConnect && (hasSubscribe || hasSend)) {
+    throw new Error(
+      "Connection adapter must provide either connect or both subscribe and send, not both modes"
+    );
+  }
+  if (hasSubscribe && hasSend) {
+    return {
+      subscribe: connection.subscribe.bind(connection),
+      send: connection.send.bind(connection)
+    };
+  }
+  if (!hasConnect) {
+    throw new Error(
+      "Connection adapter must provide either connect or both subscribe and send"
+    );
+  }
+  let activeBuffer = [];
+  let activeWaiters = [];
+  function push(chunk) {
+    const waiter = activeWaiters.shift();
+    if (waiter) {
+      waiter(chunk);
+    } else {
+      activeBuffer.push(chunk);
+    }
+  }
+  return {
+    subscribe(abortSignal) {
+      const myBuffer = activeBuffer.splice(0);
+      const myWaiters = [];
+      activeBuffer = myBuffer;
+      activeWaiters = myWaiters;
+      return (async function* () {
+        while (!abortSignal?.aborted) {
+          let chunk;
+          if (myBuffer.length > 0) {
+            chunk = myBuffer.shift();
+          } else {
+            chunk = await new Promise((resolve) => {
+              const onAbort = () => resolve(null);
+              myWaiters.push((c) => {
+                abortSignal?.removeEventListener("abort", onAbort);
+                resolve(c);
+              });
+              abortSignal?.addEventListener("abort", onAbort, { once: true });
+            });
+          }
+          if (chunk !== null) yield chunk;
+        }
+      })();
+    },
+    async send(messages, data, abortSignal) {
+      let hasTerminalEvent = false;
+      try {
+        const stream2 = connection.connect(messages, data, abortSignal);
+        for await (const chunk of stream2) {
+          if (chunk.type === "RUN_FINISHED" || chunk.type === "RUN_ERROR") {
+            hasTerminalEvent = true;
+          }
+          push(chunk);
+        }
+        if (!abortSignal?.aborted && !hasTerminalEvent) {
+          push({
+            type: "RUN_FINISHED",
+            runId: `run-${Date.now()}`,
+            model: "connect-wrapper",
+            timestamp: Date.now(),
+            finishReason: "stop"
+          });
+        }
+      } catch (err) {
+        if (!abortSignal?.aborted && !hasTerminalEvent) {
+          push({
+            type: "RUN_ERROR",
+            timestamp: Date.now(),
+            error: {
+              message: err instanceof Error ? err.message : "Unknown error in connect()"
+            }
+          });
+        }
+        throw err;
+      }
+    }
+  };
+}
 function fetchServerSentEvents(url, options = {}) {
   return {
     async *connect(messages, data, abortSignal) {
@@ -140,6 +232,7 @@ function rpcStream(rpcCall) {
 export {
   fetchHttpStream,
   fetchServerSentEvents,
+  normalizeConnectionAdapter,
   rpcStream,
   stream
 };

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

@@ -1 +1 @@
-{"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;"}
+{"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\nexport interface ConnectConnectionAdapter {\n  /**\n   * Connect and return an async iterable of StreamChunks.\n   */\n  connect: (\n    messages: Array<UIMessage> | Array<ModelMessage>,\n    data?: Record<string, any>,\n    abortSignal?: AbortSignal,\n  ) => AsyncIterable<StreamChunk>\n}\n\nexport interface SubscribeConnectionAdapter {\n  /**\n   * Subscribe to stream chunks.\n   */\n  subscribe: (abortSignal?: AbortSignal) => AsyncIterable<StreamChunk>\n  /**\n   * Send a request; chunks arrive through subscribe().\n   */\n  send: (\n    messages: Array<UIMessage> | Array<ModelMessage>,\n    data?: Record<string, any>,\n    abortSignal?: AbortSignal,\n  ) => Promise<void>\n}\n\n/**\n * Connection adapter union.\n * Provide either `connect`, or `subscribe` + `send`.\n */\nexport type ConnectionAdapter =\n  | ConnectConnectionAdapter\n  | SubscribeConnectionAdapter\n\n/**\n * Normalize a ConnectionAdapter to subscribe/send operations.\n *\n * If a connection provides native subscribe/send, that mode is used.\n * Otherwise, connect() is wrapped using an async queue.\n */\nexport function normalizeConnectionAdapter(\n  connection: ConnectionAdapter | undefined,\n): SubscribeConnectionAdapter {\n  if (!connection) {\n    throw new Error('Connection adapter is required')\n  }\n\n  const hasConnect = 'connect' in connection\n  const hasSubscribe = 'subscribe' in connection\n  const hasSend = 'send' in connection\n\n  if (hasConnect && (hasSubscribe || hasSend)) {\n    throw new Error(\n      'Connection adapter must provide either connect or both subscribe and send, not both modes',\n    )\n  }\n\n  if (hasSubscribe && hasSend) {\n    return {\n      subscribe: connection.subscribe.bind(connection),\n      send: connection.send.bind(connection),\n    }\n  }\n\n  if (!hasConnect) {\n    throw new Error(\n      'Connection adapter must provide either connect or both subscribe and send',\n    )\n  }\n\n  // Legacy connect() wrapper\n  let activeBuffer: Array<StreamChunk> = []\n  let activeWaiters: Array<(chunk: StreamChunk | null) => void> = []\n\n  function push(chunk: StreamChunk): void {\n    const waiter = activeWaiters.shift()\n    if (waiter) {\n      waiter(chunk)\n    } else {\n      activeBuffer.push(chunk)\n    }\n  }\n\n  return {\n    subscribe(abortSignal?: AbortSignal): AsyncIterable<StreamChunk> {\n      // Transfer ownership to the latest subscriber so only one active\n      // subscribe() call receives chunks from the shared connect-wrapper queue.\n      const myBuffer: Array<StreamChunk> = activeBuffer.splice(0)\n      const myWaiters: Array<(chunk: StreamChunk | null) => void> = []\n      activeBuffer = myBuffer\n      activeWaiters = myWaiters\n\n      return (async function* () {\n        while (!abortSignal?.aborted) {\n          let chunk: StreamChunk | null\n          if (myBuffer.length > 0) {\n            chunk = myBuffer.shift()!\n          } else {\n            chunk = await new Promise<StreamChunk | null>((resolve) => {\n              const onAbort = () => resolve(null)\n              myWaiters.push((c) => {\n                abortSignal?.removeEventListener('abort', onAbort)\n                resolve(c)\n              })\n              abortSignal?.addEventListener('abort', onAbort, { once: true })\n            })\n          }\n          if (chunk !== null) yield chunk\n        }\n      })()\n    },\n    async send(messages, data, abortSignal) {\n      let hasTerminalEvent = false\n      try {\n        const stream = connection.connect(messages, data, abortSignal)\n        for await (const chunk of stream) {\n          if (chunk.type === 'RUN_FINISHED' || chunk.type === 'RUN_ERROR') {\n            hasTerminalEvent = true\n          }\n          push(chunk)\n        }\n\n        // If the connect stream ended cleanly without a terminal event,\n        // synthesize RUN_FINISHED so request-scoped consumers can complete.\n        if (!abortSignal?.aborted && !hasTerminalEvent) {\n          push({\n            type: 'RUN_FINISHED',\n            runId: `run-${Date.now()}`,\n            model: 'connect-wrapper',\n            timestamp: Date.now(),\n            finishReason: 'stop',\n          })\n        }\n      } catch (err) {\n        if (!abortSignal?.aborted && !hasTerminalEvent) {\n          push({\n            type: 'RUN_ERROR',\n            timestamp: Date.now(),\n            error: {\n              message:\n                err instanceof Error\n                  ? err.message\n                  : 'Unknown error in connect()',\n            },\n          })\n        }\n        throw err\n      }\n    },\n  }\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): ConnectConnectionAdapter {\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): ConnectConnectionAdapter {\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): ConnectConnectionAdapter {\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): ConnectConnectionAdapter {\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":["stream","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;AA0CO,SAAS,2BACd,YAC4B;AAC5B,MAAI,CAAC,YAAY;AACf,UAAM,IAAI,MAAM,gCAAgC;AAAA,EAClD;AAEA,QAAM,aAAa,aAAa;AAChC,QAAM,eAAe,eAAe;AACpC,QAAM,UAAU,UAAU;AAE1B,MAAI,eAAe,gBAAgB,UAAU;AAC3C,UAAM,IAAI;AAAA,MACR;AAAA,IAAA;AAAA,EAEJ;AAEA,MAAI,gBAAgB,SAAS;AAC3B,WAAO;AAAA,MACL,WAAW,WAAW,UAAU,KAAK,UAAU;AAAA,MAC/C,MAAM,WAAW,KAAK,KAAK,UAAU;AAAA,IAAA;AAAA,EAEzC;AAEA,MAAI,CAAC,YAAY;AACf,UAAM,IAAI;AAAA,MACR;AAAA,IAAA;AAAA,EAEJ;AAGA,MAAI,eAAmC,CAAA;AACvC,MAAI,gBAA4D,CAAA;AAEhE,WAAS,KAAK,OAA0B;AACtC,UAAM,SAAS,cAAc,MAAA;AAC7B,QAAI,QAAQ;AACV,aAAO,KAAK;AAAA,IACd,OAAO;AACL,mBAAa,KAAK,KAAK;AAAA,IACzB;AAAA,EACF;AAEA,SAAO;AAAA,IACL,UAAU,aAAuD;AAG/D,YAAM,WAA+B,aAAa,OAAO,CAAC;AAC1D,YAAM,YAAwD,CAAA;AAC9D,qBAAe;AACf,sBAAgB;AAEhB,cAAQ,mBAAmB;AACzB,eAAO,CAAC,aAAa,SAAS;AAC5B,cAAI;AACJ,cAAI,SAAS,SAAS,GAAG;AACvB,oBAAQ,SAAS,MAAA;AAAA,UACnB,OAAO;AACL,oBAAQ,MAAM,IAAI,QAA4B,CAAC,YAAY;AACzD,oBAAM,UAAU,MAAM,QAAQ,IAAI;AAClC,wBAAU,KAAK,CAAC,MAAM;AACpB,6BAAa,oBAAoB,SAAS,OAAO;AACjD,wBAAQ,CAAC;AAAA,cACX,CAAC;AACD,2BAAa,iBAAiB,SAAS,SAAS,EAAE,MAAM,MAAM;AAAA,YAChE,CAAC;AAAA,UACH;AACA,cAAI,UAAU,KAAM,OAAM;AAAA,QAC5B;AAAA,MACF,GAAA;AAAA,IACF;AAAA,IACA,MAAM,KAAK,UAAU,MAAM,aAAa;AACtC,UAAI,mBAAmB;AACvB,UAAI;AACF,cAAMA,UAAS,WAAW,QAAQ,UAAU,MAAM,WAAW;AAC7D,yBAAiB,SAASA,SAAQ;AAChC,cAAI,MAAM,SAAS,kBAAkB,MAAM,SAAS,aAAa;AAC/D,+BAAmB;AAAA,UACrB;AACA,eAAK,KAAK;AAAA,QACZ;AAIA,YAAI,CAAC,aAAa,WAAW,CAAC,kBAAkB;AAC9C,eAAK;AAAA,YACH,MAAM;AAAA,YACN,OAAO,OAAO,KAAK,IAAA,CAAK;AAAA,YACxB,OAAO;AAAA,YACP,WAAW,KAAK,IAAA;AAAA,YAChB,cAAc;AAAA,UAAA,CACf;AAAA,QACH;AAAA,MACF,SAAS,KAAK;AACZ,YAAI,CAAC,aAAa,WAAW,CAAC,kBAAkB;AAC9C,eAAK;AAAA,YACH,MAAM;AAAA,YACN,WAAW,KAAK,IAAA;AAAA,YAChB,OAAO;AAAA,cACL,SACE,eAAe,QACX,IAAI,UACJ;AAAA,YAAA;AAAA,UACR,CACD;AAAA,QACH;AACA,cAAM;AAAA,MACR;AAAA,IACF;AAAA,EAAA;AAEJ;AA+CO,SAAS,sBACd,KACA,UAEuE,IAC7C;AAC1B,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,cAAMC,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,IAC7C;AAC1B,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,eAI0B;AAC1B,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM;AAG7B,aAAO,cAAc,UAAU,IAAI;AAAA,IACrC;AAAA,EAAA;AAEJ;AAkBO,SAAS,UACd,SAI0B;AAC1B,SAAO;AAAA,IACL,OAAO,QAAQ,UAAU,MAAM;AAG7B,aAAO,QAAQ,UAAU,IAAI;AAAA,IAC/B;AAAA,EAAA;AAEJ;"}

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

@@ -1,11 +1,11 @@
-import { ConnectionAdapter } from './connection-adapters.js';
+import { ConnectConnectionAdapter } from './connection-adapters.js';
 import { GenerationClientOptions, GenerationClientState, GenerationFetcher } from './generation-types.js';
 /**
  * A lightweight, generic client for one-shot generation tasks
  * (image, speech, transcription, summarize).
  *
  * Supports two transport modes:
- * - **ConnectionAdapter** — Streaming transport (SSE, HTTP stream, custom).
+ * - **ConnectConnectionAdapter** — Streaming transport (SSE, HTTP stream, custom).
  *   Server wraps results in StreamChunk events with CUSTOM event names.
  * - **Fetcher** — Direct async function call. No streaming protocol needed.
  *
@@ -14,7 +14,7 @@ import { GenerationClientOptions, GenerationClientState, GenerationFetcher } fro
  *
  * @example
  * ```typescript
- * // With ConnectionAdapter (streaming)
+ * // With streaming connection adapter
  * const client = new GenerationClient<ImageGenerateInput, ImageGenerationResult>({
  *   connection: fetchServerSentEvents('/api/generate/image'),
  *   onResultChange: setResult,
@@ -46,7 +46,7 @@ export declare class GenerationClient<TInput extends Record<string, any>, TResul
     private abortController;
     private callbacksRef;
     constructor(options: GenerationClientOptions<TInput, TResult, TOutput> & ({
-        connection: ConnectionAdapter;
+        connection: ConnectConnectionAdapter;
         fetcher?: never;
     } | {
         fetcher: GenerationFetcher<TInput, TResult>;
@@ -59,7 +59,7 @@ export declare class GenerationClient<TInput extends Record<string, any>, TResul
      */
     generate(input: TInput): Promise<void>;
     /**
-     * Process a stream of AG-UI events from the ConnectionAdapter.
+     * Process a stream of AG-UI events from the streaming connection adapter.
      */
     private processStream;
     /**

package/dist/esm/generation-client.js

@@ -65,7 +65,7 @@ class GenerationClient {
     }
   }
   /**
-   * Process a stream of AG-UI events from the ConnectionAdapter.
+   * Process a stream of AG-UI events from the streaming connection adapter.
    */
   async processStream(source) {
     for await (const chunk of source) {

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

@@ -1 +1 @@
-{"version":3,"file":"generation-client.js","sources":["../../src/generation-client.ts"],"sourcesContent":["import { GENERATION_EVENTS } from './generation-types'\nimport { parseSSEResponse } from './sse-parser'\nimport type { StreamChunk } from '@tanstack/ai'\nimport type { ConnectionAdapter } from './connection-adapters'\nimport type {\n  GenerationClientOptions,\n  GenerationClientState,\n  GenerationFetcher,\n} from './generation-types'\n\n/**\n * Callbacks stored in a ref so hooks can update them without recreating the client.\n */\ninterface GenerationCallbacks<TResult, TOutput> {\n  onResult?: (result: TResult) => TOutput | null | void\n  onError?: (error: Error) => void\n  onProgress?: (progress: number, message?: string) => void\n  onChunk?: (chunk: StreamChunk) => void\n  onResultChange?: (result: TOutput | null) => void\n  onLoadingChange?: (isLoading: boolean) => void\n  onErrorChange?: (error: Error | undefined) => void\n  onStatusChange?: (status: GenerationClientState) => void\n}\n\n/**\n * A lightweight, generic client for one-shot generation tasks\n * (image, speech, transcription, summarize).\n *\n * Supports two transport modes:\n * - **ConnectionAdapter** — Streaming transport (SSE, HTTP stream, custom).\n *   Server wraps results in StreamChunk events with CUSTOM event names.\n * - **Fetcher** — Direct async function call. No streaming protocol needed.\n *\n * @template TInput - The input type for the generation request\n * @template TResult - The result type returned by the generation\n *\n * @example\n * ```typescript\n * // With ConnectionAdapter (streaming)\n * const client = new GenerationClient<ImageGenerateInput, ImageGenerationResult>({\n *   connection: fetchServerSentEvents('/api/generate/image'),\n *   onResultChange: setResult,\n *   onLoadingChange: setIsLoading,\n * })\n *\n * // With fetcher (direct)\n * const client = new GenerationClient<ImageGenerateInput, ImageGenerationResult>({\n *   fetcher: async (input) => {\n *     const res = await fetch('/api/generate/image', {\n *       method: 'POST',\n *       body: JSON.stringify(input),\n *     })\n *     return res.json()\n *   },\n * })\n *\n * await client.generate({ prompt: 'A sunset over mountains' })\n * ```\n */\nexport class GenerationClient<\n  TInput extends Record<string, any>,\n  TResult,\n  TOutput = TResult,\n> {\n  private connection: ConnectionAdapter | undefined\n  private fetcher: GenerationFetcher<TInput, TResult> | undefined\n  private body: Record<string, any>\n  private result: TOutput | null = null\n  private isLoading = false\n  private error: Error | undefined = undefined\n  private status: GenerationClientState = 'idle'\n  private abortController: AbortController | null = null\n  private callbacksRef: GenerationCallbacks<TResult, TOutput>\n\n  constructor(\n    options: GenerationClientOptions<TInput, TResult, TOutput> &\n      (\n        | { connection: ConnectionAdapter; fetcher?: never }\n        | {\n            fetcher: GenerationFetcher<TInput, TResult>\n            connection?: never\n          }\n      ),\n  ) {\n    this.connection = options.connection\n    this.fetcher = options.fetcher\n    this.body = options.body ?? {}\n\n    this.callbacksRef = {\n      onResult: options.onResult,\n      onError: options.onError,\n      onProgress: options.onProgress,\n      onChunk: options.onChunk,\n      onResultChange: options.onResultChange,\n      onLoadingChange: options.onLoadingChange,\n      onErrorChange: options.onErrorChange,\n      onStatusChange: options.onStatusChange,\n    }\n  }\n\n  /**\n   * Trigger a generation request.\n   * Only one generation can be in-flight at a time; calling generate()\n   * while already generating will be a no-op.\n   */\n  async generate(input: TInput): Promise<void> {\n    if (this.isLoading) return\n\n    this.setIsLoading(true)\n    this.setStatus('generating')\n    this.setError(undefined)\n\n    const abortController = new AbortController()\n    this.abortController = abortController\n    const { signal } = abortController\n\n    try {\n      if (this.fetcher) {\n        // Direct fetch path\n        const result = await this.fetcher(input, { signal })\n        if (signal.aborted) return\n        if (result instanceof Response) {\n          // Server function returned SSE Response — parse stream\n          await this.processStream(parseSSEResponse(result, signal))\n        } else {\n          this.setResult(result)\n          this.setStatus('success')\n        }\n      } else if (this.connection) {\n        // ConnectionAdapter streaming path\n        const mergedData = { ...this.body, ...input }\n        const stream = this.connection.connect([], mergedData, signal)\n        await this.processStream(stream)\n      } else {\n        throw new Error(\n          'GenerationClient requires either a connection or fetcher option',\n        )\n      }\n    } catch (err: any) {\n      if (signal.aborted) return\n      const error = err instanceof Error ? err : new Error(String(err))\n      this.setError(error)\n      this.setStatus('error')\n      this.callbacksRef.onError?.(error)\n    } finally {\n      this.abortController = null\n      this.setIsLoading(false)\n    }\n  }\n\n  /**\n   * Process a stream of AG-UI events from the ConnectionAdapter.\n   */\n  private async processStream(\n    source: AsyncIterable<StreamChunk>,\n  ): Promise<void> {\n    for await (const chunk of source) {\n      if (this.abortController?.signal.aborted) break\n\n      this.callbacksRef.onChunk?.(chunk)\n\n      switch (chunk.type) {\n        case 'CUSTOM': {\n          if (chunk.name === GENERATION_EVENTS.RESULT) {\n            this.setResult(chunk.value as TResult)\n          } else if (chunk.name === GENERATION_EVENTS.PROGRESS) {\n            const { progress, message } = chunk.value as {\n              progress: number\n              message?: string\n            }\n            this.callbacksRef.onProgress?.(progress, message)\n          }\n          break\n        }\n        case 'RUN_FINISHED': {\n          this.setStatus('success')\n          break\n        }\n        case 'RUN_ERROR': {\n          throw new Error(chunk.error.message)\n        }\n      }\n    }\n  }\n\n  /**\n   * Abort any in-flight generation request.\n   */\n  stop(): void {\n    if (this.abortController) {\n      this.abortController.abort()\n      this.abortController = null\n    }\n    this.setIsLoading(false)\n    if (this.status === 'generating') {\n      this.setStatus('idle')\n    }\n  }\n\n  /**\n   * Clear the result, error, and return to idle state.\n   */\n  reset(): void {\n    this.stop()\n    this.setResult(null)\n    this.setError(undefined)\n    this.setStatus('idle')\n  }\n\n  /**\n   * Update options without recreating the client.\n   */\n  updateOptions(\n    options: Partial<\n      Pick<\n        GenerationClientOptions<TInput, TResult, TOutput>,\n        'body' | 'onResult' | 'onError' | 'onProgress' | 'onChunk'\n      >\n    >,\n  ): void {\n    if (options.body !== undefined) {\n      this.body = options.body ?? {}\n    }\n    if (options.onResult !== undefined) {\n      this.callbacksRef.onResult = options.onResult\n    }\n    if (options.onError !== undefined) {\n      this.callbacksRef.onError = options.onError\n    }\n    if (options.onProgress !== undefined) {\n      this.callbacksRef.onProgress = options.onProgress\n    }\n    if (options.onChunk !== undefined) {\n      this.callbacksRef.onChunk = options.onChunk\n    }\n  }\n\n  // ===========================\n  // Getters\n  // ===========================\n\n  getResult(): TOutput | null {\n    return this.result\n  }\n\n  getIsLoading(): boolean {\n    return this.isLoading\n  }\n\n  getError(): Error | undefined {\n    return this.error\n  }\n\n  getStatus(): GenerationClientState {\n    return this.status\n  }\n\n  // ===========================\n  // Private state setters\n  // ===========================\n\n  private setResult(rawResult: TResult | null): void {\n    if (rawResult === null) {\n      this.result = null\n      this.callbacksRef.onResultChange?.(null)\n      return\n    }\n\n    if (this.callbacksRef.onResult) {\n      const transformed = this.callbacksRef.onResult(rawResult)\n      if (transformed === null) {\n        // null return → keep previous result unchanged\n        return\n      }\n      if (transformed !== undefined) {\n        // Non-null, non-undefined → use transformed value\n        this.result = transformed\n        this.callbacksRef.onResultChange?.(this.result)\n        return\n      }\n    }\n\n    // No onResult callback, or callback returned void → use raw value\n    this.result = rawResult as unknown as TOutput\n    this.callbacksRef.onResultChange?.(this.result)\n  }\n\n  private setIsLoading(isLoading: boolean): void {\n    this.isLoading = isLoading\n    this.callbacksRef.onLoadingChange?.(isLoading)\n  }\n\n  private setError(error: Error | undefined): void {\n    this.error = error\n    this.callbacksRef.onErrorChange?.(error)\n  }\n\n  private setStatus(status: GenerationClientState): void {\n    this.status = status\n    this.callbacksRef.onStatusChange?.(status)\n  }\n}\n"],"names":[],"mappings":";;AA2DO,MAAM,iBAIX;AAAA,EAWA,YACE,SAQA;AAhBF,SAAQ,SAAyB;AACjC,SAAQ,YAAY;AACpB,SAAQ,QAA2B;AACnC,SAAQ,SAAgC;AACxC,SAAQ,kBAA0C;AAahD,SAAK,aAAa,QAAQ;AAC1B,SAAK,UAAU,QAAQ;AACvB,SAAK,OAAO,QAAQ,QAAQ,CAAA;AAE5B,SAAK,eAAe;AAAA,MAClB,UAAU,QAAQ;AAAA,MAClB,SAAS,QAAQ;AAAA,MACjB,YAAY,QAAQ;AAAA,MACpB,SAAS,QAAQ;AAAA,MACjB,gBAAgB,QAAQ;AAAA,MACxB,iBAAiB,QAAQ;AAAA,MACzB,eAAe,QAAQ;AAAA,MACvB,gBAAgB,QAAQ;AAAA,IAAA;AAAA,EAE5B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,SAAS,OAA8B;AAC3C,QAAI,KAAK,UAAW;AAEpB,SAAK,aAAa,IAAI;AACtB,SAAK,UAAU,YAAY;AAC3B,SAAK,SAAS,MAAS;AAEvB,UAAM,kBAAkB,IAAI,gBAAA;AAC5B,SAAK,kBAAkB;AACvB,UAAM,EAAE,WAAW;AAEnB,QAAI;AACF,UAAI,KAAK,SAAS;AAEhB,cAAM,SAAS,MAAM,KAAK,QAAQ,OAAO,EAAE,QAAQ;AACnD,YAAI,OAAO,QAAS;AACpB,YAAI,kBAAkB,UAAU;AAE9B,gBAAM,KAAK,cAAc,iBAAiB,QAAQ,MAAM,CAAC;AAAA,QAC3D,OAAO;AACL,eAAK,UAAU,MAAM;AACrB,eAAK,UAAU,SAAS;AAAA,QAC1B;AAAA,MACF,WAAW,KAAK,YAAY;AAE1B,cAAM,aAAa,EAAE,GAAG,KAAK,MAAM,GAAG,MAAA;AACtC,cAAM,SAAS,KAAK,WAAW,QAAQ,CAAA,GAAI,YAAY,MAAM;AAC7D,cAAM,KAAK,cAAc,MAAM;AAAA,MACjC,OAAO;AACL,cAAM,IAAI;AAAA,UACR;AAAA,QAAA;AAAA,MAEJ;AAAA,IACF,SAAS,KAAU;AACjB,UAAI,OAAO,QAAS;AACpB,YAAM,QAAQ,eAAe,QAAQ,MAAM,IAAI,MAAM,OAAO,GAAG,CAAC;AAChE,WAAK,SAAS,KAAK;AACnB,WAAK,UAAU,OAAO;AACtB,WAAK,aAAa,UAAU,KAAK;AAAA,IACnC,UAAA;AACE,WAAK,kBAAkB;AACvB,WAAK,aAAa,KAAK;AAAA,IACzB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,cACZ,QACe;AACf,qBAAiB,SAAS,QAAQ;AAChC,UAAI,KAAK,iBAAiB,OAAO,QAAS;AAE1C,WAAK,aAAa,UAAU,KAAK;AAEjC,cAAQ,MAAM,MAAA;AAAA,QACZ,KAAK,UAAU;AACb,cAAI,MAAM,SAAS,kBAAkB,QAAQ;AAC3C,iBAAK,UAAU,MAAM,KAAgB;AAAA,UACvC,WAAW,MAAM,SAAS,kBAAkB,UAAU;AACpD,kBAAM,EAAE,UAAU,QAAA,IAAY,MAAM;AAIpC,iBAAK,aAAa,aAAa,UAAU,OAAO;AAAA,UAClD;AACA;AAAA,QACF;AAAA,QACA,KAAK,gBAAgB;AACnB,eAAK,UAAU,SAAS;AACxB;AAAA,QACF;AAAA,QACA,KAAK,aAAa;AAChB,gBAAM,IAAI,MAAM,MAAM,MAAM,OAAO;AAAA,QACrC;AAAA,MAAA;AAAA,IAEJ;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,OAAa;AACX,QAAI,KAAK,iBAAiB;AACxB,WAAK,gBAAgB,MAAA;AACrB,WAAK,kBAAkB;AAAA,IACzB;AACA,SAAK,aAAa,KAAK;AACvB,QAAI,KAAK,WAAW,cAAc;AAChC,WAAK,UAAU,MAAM;AAAA,IACvB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,QAAc;AACZ,SAAK,KAAA;AACL,SAAK,UAAU,IAAI;AACnB,SAAK,SAAS,MAAS;AACvB,SAAK,UAAU,MAAM;AAAA,EACvB;AAAA;AAAA;AAAA;AAAA,EAKA,cACE,SAMM;AACN,QAAI,QAAQ,SAAS,QAAW;AAC9B,WAAK,OAAO,QAAQ,QAAQ,CAAA;AAAA,IAC9B;AACA,QAAI,QAAQ,aAAa,QAAW;AAClC,WAAK,aAAa,WAAW,QAAQ;AAAA,IACvC;AACA,QAAI,QAAQ,YAAY,QAAW;AACjC,WAAK,aAAa,UAAU,QAAQ;AAAA,IACtC;AACA,QAAI,QAAQ,eAAe,QAAW;AACpC,WAAK,aAAa,aAAa,QAAQ;AAAA,IACzC;AACA,QAAI,QAAQ,YAAY,QAAW;AACjC,WAAK,aAAa,UAAU,QAAQ;AAAA,IACtC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAMA,YAA4B;AAC1B,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,eAAwB;AACtB,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,WAA8B;AAC5B,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,YAAmC;AACjC,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAMQ,UAAU,WAAiC;AACjD,QAAI,cAAc,MAAM;AACtB,WAAK,SAAS;AACd,WAAK,aAAa,iBAAiB,IAAI;AACvC;AAAA,IACF;AAEA,QAAI,KAAK,aAAa,UAAU;AAC9B,YAAM,cAAc,KAAK,aAAa,SAAS,SAAS;AACxD,UAAI,gBAAgB,MAAM;AAExB;AAAA,MACF;AACA,UAAI,gBAAgB,QAAW;AAE7B,aAAK,SAAS;AACd,aAAK,aAAa,iBAAiB,KAAK,MAAM;AAC9C;AAAA,MACF;AAAA,IACF;AAGA,SAAK,SAAS;AACd,SAAK,aAAa,iBAAiB,KAAK,MAAM;AAAA,EAChD;AAAA,EAEQ,aAAa,WAA0B;AAC7C,SAAK,YAAY;AACjB,SAAK,aAAa,kBAAkB,SAAS;AAAA,EAC/C;AAAA,EAEQ,SAAS,OAAgC;AAC/C,SAAK,QAAQ;AACb,SAAK,aAAa,gBAAgB,KAAK;AAAA,EACzC;AAAA,EAEQ,UAAU,QAAqC;AACrD,SAAK,SAAS;AACd,SAAK,aAAa,iBAAiB,MAAM;AAAA,EAC3C;AACF;"}
+{"version":3,"file":"generation-client.js","sources":["../../src/generation-client.ts"],"sourcesContent":["import { GENERATION_EVENTS } from './generation-types'\nimport { parseSSEResponse } from './sse-parser'\nimport type { StreamChunk } from '@tanstack/ai'\nimport type { ConnectConnectionAdapter } from './connection-adapters'\nimport type {\n  GenerationClientOptions,\n  GenerationClientState,\n  GenerationFetcher,\n} from './generation-types'\n\n/**\n * Callbacks stored in a ref so hooks can update them without recreating the client.\n */\ninterface GenerationCallbacks<TResult, TOutput> {\n  onResult?: (result: TResult) => TOutput | null | void\n  onError?: (error: Error) => void\n  onProgress?: (progress: number, message?: string) => void\n  onChunk?: (chunk: StreamChunk) => void\n  onResultChange?: (result: TOutput | null) => void\n  onLoadingChange?: (isLoading: boolean) => void\n  onErrorChange?: (error: Error | undefined) => void\n  onStatusChange?: (status: GenerationClientState) => void\n}\n\n/**\n * A lightweight, generic client for one-shot generation tasks\n * (image, speech, transcription, summarize).\n *\n * Supports two transport modes:\n * - **ConnectConnectionAdapter** — Streaming transport (SSE, HTTP stream, custom).\n *   Server wraps results in StreamChunk events with CUSTOM event names.\n * - **Fetcher** — Direct async function call. No streaming protocol needed.\n *\n * @template TInput - The input type for the generation request\n * @template TResult - The result type returned by the generation\n *\n * @example\n * ```typescript\n * // With streaming connection adapter\n * const client = new GenerationClient<ImageGenerateInput, ImageGenerationResult>({\n *   connection: fetchServerSentEvents('/api/generate/image'),\n *   onResultChange: setResult,\n *   onLoadingChange: setIsLoading,\n * })\n *\n * // With fetcher (direct)\n * const client = new GenerationClient<ImageGenerateInput, ImageGenerationResult>({\n *   fetcher: async (input) => {\n *     const res = await fetch('/api/generate/image', {\n *       method: 'POST',\n *       body: JSON.stringify(input),\n *     })\n *     return res.json()\n *   },\n * })\n *\n * await client.generate({ prompt: 'A sunset over mountains' })\n * ```\n */\nexport class GenerationClient<\n  TInput extends Record<string, any>,\n  TResult,\n  TOutput = TResult,\n> {\n  private connection: ConnectConnectionAdapter | undefined\n  private fetcher: GenerationFetcher<TInput, TResult> | undefined\n  private body: Record<string, any>\n  private result: TOutput | null = null\n  private isLoading = false\n  private error: Error | undefined = undefined\n  private status: GenerationClientState = 'idle'\n  private abortController: AbortController | null = null\n  private callbacksRef: GenerationCallbacks<TResult, TOutput>\n\n  constructor(\n    options: GenerationClientOptions<TInput, TResult, TOutput> &\n      (\n        | { connection: ConnectConnectionAdapter; fetcher?: never }\n        | {\n            fetcher: GenerationFetcher<TInput, TResult>\n            connection?: never\n          }\n      ),\n  ) {\n    this.connection = options.connection\n    this.fetcher = options.fetcher\n    this.body = options.body ?? {}\n\n    this.callbacksRef = {\n      onResult: options.onResult,\n      onError: options.onError,\n      onProgress: options.onProgress,\n      onChunk: options.onChunk,\n      onResultChange: options.onResultChange,\n      onLoadingChange: options.onLoadingChange,\n      onErrorChange: options.onErrorChange,\n      onStatusChange: options.onStatusChange,\n    }\n  }\n\n  /**\n   * Trigger a generation request.\n   * Only one generation can be in-flight at a time; calling generate()\n   * while already generating will be a no-op.\n   */\n  async generate(input: TInput): Promise<void> {\n    if (this.isLoading) return\n\n    this.setIsLoading(true)\n    this.setStatus('generating')\n    this.setError(undefined)\n\n    const abortController = new AbortController()\n    this.abortController = abortController\n    const { signal } = abortController\n\n    try {\n      if (this.fetcher) {\n        // Direct fetch path\n        const result = await this.fetcher(input, { signal })\n        if (signal.aborted) return\n        if (result instanceof Response) {\n          // Server function returned SSE Response — parse stream\n          await this.processStream(parseSSEResponse(result, signal))\n        } else {\n          this.setResult(result)\n          this.setStatus('success')\n        }\n      } else if (this.connection) {\n        // Streaming adapter path\n        const mergedData = { ...this.body, ...input }\n        const stream = this.connection.connect([], mergedData, signal)\n        await this.processStream(stream)\n      } else {\n        throw new Error(\n          'GenerationClient requires either a connection or fetcher option',\n        )\n      }\n    } catch (err: any) {\n      if (signal.aborted) return\n      const error = err instanceof Error ? err : new Error(String(err))\n      this.setError(error)\n      this.setStatus('error')\n      this.callbacksRef.onError?.(error)\n    } finally {\n      this.abortController = null\n      this.setIsLoading(false)\n    }\n  }\n\n  /**\n   * Process a stream of AG-UI events from the streaming connection adapter.\n   */\n  private async processStream(\n    source: AsyncIterable<StreamChunk>,\n  ): Promise<void> {\n    for await (const chunk of source) {\n      if (this.abortController?.signal.aborted) break\n\n      this.callbacksRef.onChunk?.(chunk)\n\n      switch (chunk.type) {\n        case 'CUSTOM': {\n          if (chunk.name === GENERATION_EVENTS.RESULT) {\n            this.setResult(chunk.value as TResult)\n          } else if (chunk.name === GENERATION_EVENTS.PROGRESS) {\n            const { progress, message } = chunk.value as {\n              progress: number\n              message?: string\n            }\n            this.callbacksRef.onProgress?.(progress, message)\n          }\n          break\n        }\n        case 'RUN_FINISHED': {\n          this.setStatus('success')\n          break\n        }\n        case 'RUN_ERROR': {\n          throw new Error(chunk.error.message)\n        }\n      }\n    }\n  }\n\n  /**\n   * Abort any in-flight generation request.\n   */\n  stop(): void {\n    if (this.abortController) {\n      this.abortController.abort()\n      this.abortController = null\n    }\n    this.setIsLoading(false)\n    if (this.status === 'generating') {\n      this.setStatus('idle')\n    }\n  }\n\n  /**\n   * Clear the result, error, and return to idle state.\n   */\n  reset(): void {\n    this.stop()\n    this.setResult(null)\n    this.setError(undefined)\n    this.setStatus('idle')\n  }\n\n  /**\n   * Update options without recreating the client.\n   */\n  updateOptions(\n    options: Partial<\n      Pick<\n        GenerationClientOptions<TInput, TResult, TOutput>,\n        'body' | 'onResult' | 'onError' | 'onProgress' | 'onChunk'\n      >\n    >,\n  ): void {\n    if (options.body !== undefined) {\n      this.body = options.body ?? {}\n    }\n    if (options.onResult !== undefined) {\n      this.callbacksRef.onResult = options.onResult\n    }\n    if (options.onError !== undefined) {\n      this.callbacksRef.onError = options.onError\n    }\n    if (options.onProgress !== undefined) {\n      this.callbacksRef.onProgress = options.onProgress\n    }\n    if (options.onChunk !== undefined) {\n      this.callbacksRef.onChunk = options.onChunk\n    }\n  }\n\n  // ===========================\n  // Getters\n  // ===========================\n\n  getResult(): TOutput | null {\n    return this.result\n  }\n\n  getIsLoading(): boolean {\n    return this.isLoading\n  }\n\n  getError(): Error | undefined {\n    return this.error\n  }\n\n  getStatus(): GenerationClientState {\n    return this.status\n  }\n\n  // ===========================\n  // Private state setters\n  // ===========================\n\n  private setResult(rawResult: TResult | null): void {\n    if (rawResult === null) {\n      this.result = null\n      this.callbacksRef.onResultChange?.(null)\n      return\n    }\n\n    if (this.callbacksRef.onResult) {\n      const transformed = this.callbacksRef.onResult(rawResult)\n      if (transformed === null) {\n        // null return → keep previous result unchanged\n        return\n      }\n      if (transformed !== undefined) {\n        // Non-null, non-undefined → use transformed value\n        this.result = transformed\n        this.callbacksRef.onResultChange?.(this.result)\n        return\n      }\n    }\n\n    // No onResult callback, or callback returned void → use raw value\n    this.result = rawResult as unknown as TOutput\n    this.callbacksRef.onResultChange?.(this.result)\n  }\n\n  private setIsLoading(isLoading: boolean): void {\n    this.isLoading = isLoading\n    this.callbacksRef.onLoadingChange?.(isLoading)\n  }\n\n  private setError(error: Error | undefined): void {\n    this.error = error\n    this.callbacksRef.onErrorChange?.(error)\n  }\n\n  private setStatus(status: GenerationClientState): void {\n    this.status = status\n    this.callbacksRef.onStatusChange?.(status)\n  }\n}\n"],"names":[],"mappings":";;AA2DO,MAAM,iBAIX;AAAA,EAWA,YACE,SAQA;AAhBF,SAAQ,SAAyB;AACjC,SAAQ,YAAY;AACpB,SAAQ,QAA2B;AACnC,SAAQ,SAAgC;AACxC,SAAQ,kBAA0C;AAahD,SAAK,aAAa,QAAQ;AAC1B,SAAK,UAAU,QAAQ;AACvB,SAAK,OAAO,QAAQ,QAAQ,CAAA;AAE5B,SAAK,eAAe;AAAA,MAClB,UAAU,QAAQ;AAAA,MAClB,SAAS,QAAQ;AAAA,MACjB,YAAY,QAAQ;AAAA,MACpB,SAAS,QAAQ;AAAA,MACjB,gBAAgB,QAAQ;AAAA,MACxB,iBAAiB,QAAQ;AAAA,MACzB,eAAe,QAAQ;AAAA,MACvB,gBAAgB,QAAQ;AAAA,IAAA;AAAA,EAE5B;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,SAAS,OAA8B;AAC3C,QAAI,KAAK,UAAW;AAEpB,SAAK,aAAa,IAAI;AACtB,SAAK,UAAU,YAAY;AAC3B,SAAK,SAAS,MAAS;AAEvB,UAAM,kBAAkB,IAAI,gBAAA;AAC5B,SAAK,kBAAkB;AACvB,UAAM,EAAE,WAAW;AAEnB,QAAI;AACF,UAAI,KAAK,SAAS;AAEhB,cAAM,SAAS,MAAM,KAAK,QAAQ,OAAO,EAAE,QAAQ;AACnD,YAAI,OAAO,QAAS;AACpB,YAAI,kBAAkB,UAAU;AAE9B,gBAAM,KAAK,cAAc,iBAAiB,QAAQ,MAAM,CAAC;AAAA,QAC3D,OAAO;AACL,eAAK,UAAU,MAAM;AACrB,eAAK,UAAU,SAAS;AAAA,QAC1B;AAAA,MACF,WAAW,KAAK,YAAY;AAE1B,cAAM,aAAa,EAAE,GAAG,KAAK,MAAM,GAAG,MAAA;AACtC,cAAM,SAAS,KAAK,WAAW,QAAQ,CAAA,GAAI,YAAY,MAAM;AAC7D,cAAM,KAAK,cAAc,MAAM;AAAA,MACjC,OAAO;AACL,cAAM,IAAI;AAAA,UACR;AAAA,QAAA;AAAA,MAEJ;AAAA,IACF,SAAS,KAAU;AACjB,UAAI,OAAO,QAAS;AACpB,YAAM,QAAQ,eAAe,QAAQ,MAAM,IAAI,MAAM,OAAO,GAAG,CAAC;AAChE,WAAK,SAAS,KAAK;AACnB,WAAK,UAAU,OAAO;AACtB,WAAK,aAAa,UAAU,KAAK;AAAA,IACnC,UAAA;AACE,WAAK,kBAAkB;AACvB,WAAK,aAAa,KAAK;AAAA,IACzB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,cACZ,QACe;AACf,qBAAiB,SAAS,QAAQ;AAChC,UAAI,KAAK,iBAAiB,OAAO,QAAS;AAE1C,WAAK,aAAa,UAAU,KAAK;AAEjC,cAAQ,MAAM,MAAA;AAAA,QACZ,KAAK,UAAU;AACb,cAAI,MAAM,SAAS,kBAAkB,QAAQ;AAC3C,iBAAK,UAAU,MAAM,KAAgB;AAAA,UACvC,WAAW,MAAM,SAAS,kBAAkB,UAAU;AACpD,kBAAM,EAAE,UAAU,QAAA,IAAY,MAAM;AAIpC,iBAAK,aAAa,aAAa,UAAU,OAAO;AAAA,UAClD;AACA;AAAA,QACF;AAAA,QACA,KAAK,gBAAgB;AACnB,eAAK,UAAU,SAAS;AACxB;AAAA,QACF;AAAA,QACA,KAAK,aAAa;AAChB,gBAAM,IAAI,MAAM,MAAM,MAAM,OAAO;AAAA,QACrC;AAAA,MAAA;AAAA,IAEJ;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,OAAa;AACX,QAAI,KAAK,iBAAiB;AACxB,WAAK,gBAAgB,MAAA;AACrB,WAAK,kBAAkB;AAAA,IACzB;AACA,SAAK,aAAa,KAAK;AACvB,QAAI,KAAK,WAAW,cAAc;AAChC,WAAK,UAAU,MAAM;AAAA,IACvB;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,QAAc;AACZ,SAAK,KAAA;AACL,SAAK,UAAU,IAAI;AACnB,SAAK,SAAS,MAAS;AACvB,SAAK,UAAU,MAAM;AAAA,EACvB;AAAA;AAAA;AAAA;AAAA,EAKA,cACE,SAMM;AACN,QAAI,QAAQ,SAAS,QAAW;AAC9B,WAAK,OAAO,QAAQ,QAAQ,CAAA;AAAA,IAC9B;AACA,QAAI,QAAQ,aAAa,QAAW;AAClC,WAAK,aAAa,WAAW,QAAQ;AAAA,IACvC;AACA,QAAI,QAAQ,YAAY,QAAW;AACjC,WAAK,aAAa,UAAU,QAAQ;AAAA,IACtC;AACA,QAAI,QAAQ,eAAe,QAAW;AACpC,WAAK,aAAa,aAAa,QAAQ;AAAA,IACzC;AACA,QAAI,QAAQ,YAAY,QAAW;AACjC,WAAK,aAAa,UAAU,QAAQ;AAAA,IACtC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAMA,YAA4B;AAC1B,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,eAAwB;AACtB,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,WAA8B;AAC5B,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,YAAmC;AACjC,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA,EAMQ,UAAU,WAAiC;AACjD,QAAI,cAAc,MAAM;AACtB,WAAK,SAAS;AACd,WAAK,aAAa,iBAAiB,IAAI;AACvC;AAAA,IACF;AAEA,QAAI,KAAK,aAAa,UAAU;AAC9B,YAAM,cAAc,KAAK,aAAa,SAAS,SAAS;AACxD,UAAI,gBAAgB,MAAM;AAExB;AAAA,MACF;AACA,UAAI,gBAAgB,QAAW;AAE7B,aAAK,SAAS;AACd,aAAK,aAAa,iBAAiB,KAAK,MAAM;AAC9C;AAAA,MACF;AAAA,IACF;AAGA,SAAK,SAAS;AACd,SAAK,aAAa,iBAAiB,KAAK,MAAM;AAAA,EAChD;AAAA,EAEQ,aAAa,WAA0B;AAC7C,SAAK,YAAY;AACjB,SAAK,aAAa,kBAAkB,SAAS;AAAA,EAC/C;AAAA,EAEQ,SAAS,OAAgC;AAC/C,SAAK,QAAQ;AACb,SAAK,aAAa,gBAAgB,KAAK;AAAA,EACzC;AAAA,EAEQ,UAAU,QAAqC;AACrD,SAAK,SAAS;AACd,SAAK,aAAa,iBAAiB,MAAM;AAAA,EAC3C;AACF;"}

package/dist/esm/generation-types.d.ts

@@ -1,5 +1,5 @@
 import { StreamChunk } from '@tanstack/ai';
-import { ConnectionAdapter } from './connection-adapters.js';
+import { ConnectConnectionAdapter } from './connection-adapters.js';
 /**
  * Infers the output type from an `onResult` callback's return type.
  *
@@ -50,10 +50,10 @@ export interface GenerationFetcherOptions {
 export type GenerationFetcher<TInput, TResult> = (input: TInput, options?: GenerationFetcherOptions) => Promise<TResult | Response>;
 /**
  * Transport configuration for generation clients.
- * Supports either a ConnectionAdapter (streaming) or a direct fetcher function.
+ * Supports either a connect-based streaming adapter or a direct fetcher function.
  */
 export type GenerationTransport<TInput, TResult> = {
-    connection: ConnectionAdapter;
+    connection: ConnectConnectionAdapter;
     fetcher?: never;
 } | {
     fetcher: GenerationFetcher<TInput, TResult>;
@@ -69,7 +69,7 @@ export type GenerationTransport<TInput, TResult> = {
 export interface GenerationClientOptions<_TInput, TResult, TOutput = TResult> {
     /** Unique identifier for this generation client instance */
     id?: string;
-    /** Additional body parameters to send with ConnectionAdapter requests */
+    /** Additional body parameters to send with connect-based adapter requests */
     body?: Record<string, any>;
     /**
      * Callback when a result is received. Can optionally return a transformed value
@@ -84,7 +84,7 @@ export interface GenerationClientOptions<_TInput, TResult, TOutput = TResult> {
     onError?: (error: Error) => void;
     /** Callback when progress is reported (0-100) */
     onProgress?: (progress: number, message?: string) => void;
-    /** Callback for each stream chunk (ConnectionAdapter mode only) */
+    /** Callback for each stream chunk (connect-based adapter mode only) */
     onChunk?: (chunk: StreamChunk) => void;
     /** @internal Called when result changes */
     onResultChange?: (result: TOutput | null) => void;