@tanstack/ai-react 0.27.3 → 0.28.0

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

@@ -1 +1 @@
-{"version":3,"file":"use-chat.js","names":[],"sources":["../../src/use-chat.ts"],"sourcesContent":["import { ChatClient } from '@tanstack/ai-client'\nimport { createChatDevtoolsBridge } from '@tanstack/ai-client/devtools'\nimport { useCallback, useEffect, useId, useMemo, useRef, useState } from 'react'\nimport type {\n  AnyClientTool,\n  InferSchemaType,\n  InterruptDefinition,\n  ModelMessage,\n  RunAgentResumeItem,\n  SchemaInput,\n  StreamChunk,\n} from '@tanstack/ai/client'\nimport type {\n  ChatClientState,\n  ResolvableChatInterrupt,\n  ChatInterruptState,\n  ChatResumeState,\n  ConnectionStatus,\n  InferredClientContext,\n  QueuedMessage,\n  SendMessageOptions,\n  StructuredOutputPart,\n} from '@tanstack/ai-client'\n\nimport type {\n  DeepPartial,\n  MultimodalContent,\n  UIMessage,\n  UseChatOptions,\n  UseChatReturn,\n} from './types'\n\nconst EMPTY_INTERRUPTS = Object.freeze([])\nconst EMPTY_INTERRUPT_ERRORS = Object.freeze([])\n\nexport function useChat<\n  const TTools extends ReadonlyArray<AnyClientTool> = any,\n  TSchema extends SchemaInput | undefined = undefined,\n  TContext = InferredClientContext<TTools>,\n  const TInterrupts extends ReadonlyArray<\n    InterruptDefinition<any, any, any, any>\n  > = readonly [],\n>(\n  options: UseChatOptions<TTools, TSchema, TContext, TInterrupts>,\n): UseChatReturn<TTools, TSchema, TInterrupts> {\n  // The hook's identity is its `threadId`. Reload with the same `threadId`\n  // restores the same conversation. `hookId` is only a React recreation key\n  // when no `threadId` is given. It is never sent on the wire.\n  const hookId = useId()\n  const clientId = options.threadId ?? hookId\n\n  const [messages, setMessages] = useState<Array<UIMessage<TTools>>>(\n    options.initialMessages || [],\n  )\n  const [isLoading, setIsLoading] = useState(false)\n  const [hasOlderMessages, setHasOlderMessages] = useState(false)\n  const [error, setError] = useState<Error | undefined>(undefined)\n  const [status, setStatus] = useState<ChatClientState>('ready')\n  const [isSubscribed, setIsSubscribed] = useState(false)\n  const [connectionStatus, setConnectionStatus] =\n    useState<ConnectionStatus>('disconnected')\n  const [sessionGenerating, setSessionGenerating] = useState(false)\n  const [queue, setQueue] = useState<Array<QueuedMessage>>([])\n  const [runId, setRunId] = useState<string | null>(null)\n  const [interruptState, setInterruptState] = useState<\n    ChatInterruptState<TTools, TInterrupts>\n  >(() => ({\n    interrupts: EMPTY_INTERRUPTS,\n    pendingInterrupts: EMPTY_INTERRUPTS,\n    interruptErrors: EMPTY_INTERRUPT_ERRORS,\n    resuming: false,\n  }))\n\n  type Partial = DeepPartial<InferSchemaType<NonNullable<TSchema>>>\n  type Final = InferSchemaType<NonNullable<TSchema>>\n\n  // Track current messages in a ref to preserve them when client is recreated\n  const messagesRef = useRef<Array<UIMessage<TTools>>>(\n    options.initialMessages || [],\n  )\n  const isFirstMountRef = useRef(true)\n  const subscribedRef = useRef(false)\n  const activeClientRef = useRef<ChatClient | null>(null)\n  const cleanupInvalidationRef = useRef<ReturnType<typeof setTimeout> | null>(\n    null,\n  )\n  const cleanupDisposalRef = useRef<{\n    client: ChatClient\n    timeout: ReturnType<typeof setTimeout>\n  } | null>(null)\n\n  // Update ref synchronously during render so it's always current when useMemo runs.\n  messagesRef.current = messages\n\n  // Track current options in a ref to avoid recreating client when options change\n  const optionsRef =\n    useRef<UseChatOptions<TTools, TSchema, TContext, TInterrupts>>(options)\n  optionsRef.current = options\n\n  const syncResumeState = useCallback((target: ChatClient | null) => {\n    if (!target) return\n    setRunId(target.getCurrentRunId())\n    setInterruptState(target.getInterruptState())\n  }, [])\n\n  // Create ChatClient instance with callbacks to sync state\n  const { client, initialization } = useMemo(() => {\n    const messagesToUse = options.initialMessages || []\n    isFirstMountRef.current = false\n\n    // Build options with conditional spreads for fields whose source\n    // type is `T | undefined` but the ChatClient target uses a strict\n    // optional (`field?: T`) — `exactOptionalPropertyTypes` rejects\n    // assigning `undefined` to those, so we omit the key when absent.\n    const initialOptions = optionsRef.current\n    const transport = initialOptions.connection\n      ? { connection: initialOptions.connection }\n      : { fetcher: initialOptions.fetcher }\n\n    const instanceHolder: {\n      current: ChatClient<TTools, TContext, TInterrupts> | undefined\n    } = { current: undefined }\n    const getActiveInstance = () => {\n      const currentInstance = instanceHolder.current\n      if (!currentInstance || activeClientRef.current !== currentInstance) {\n        return undefined\n      }\n      return currentInstance\n    }\n    // ChatClient may publish while its constructor is running or while async\n    // persistence resolves before commit. Preserve those exact notifications\n    // until this render commits; invoking them here would run state setters and\n    // user callbacks for a client React may abandon.\n    const initializationState = {\n      ready: false,\n      callbacks: [] as Array<() => void>,\n    }\n    const runOrQueueForActiveInstance = (callback: () => void) => {\n      if (!initializationState.ready) {\n        initializationState.callbacks.push(callback)\n        return\n      }\n      const currentInstance = instanceHolder.current\n      if (!currentInstance || activeClientRef.current !== currentInstance)\n        return\n      callback()\n    }\n    const instance = new ChatClient<TTools, TContext, TInterrupts>({\n      devtoolsBridgeFactory: createChatDevtoolsBridge,\n      ...transport,\n      initialMessages: messagesToUse,\n      ...(typeof initialOptions.threadId === 'string' &&\n      initialOptions.persistence === true\n        ? {\n            persistence: true,\n            threadId: initialOptions.threadId,\n            ...(initialOptions.history !== undefined && {\n              history: initialOptions.history,\n            }),\n          }\n        : typeof initialOptions.threadId === 'string' &&\n            initialOptions.persistence\n          ? {\n              persistence: initialOptions.persistence,\n              threadId: initialOptions.threadId,\n            }\n          : {\n              ...(initialOptions.threadId !== undefined && {\n                threadId: initialOptions.threadId,\n              }),\n            }),\n      ...(initialOptions.body !== undefined && { body: initialOptions.body }),\n      ...(initialOptions.forwardedProps !== undefined && {\n        forwardedProps: initialOptions.forwardedProps,\n      }),\n      ...(initialOptions.byok !== undefined && { byok: initialOptions.byok }),\n      byokProvider: () => optionsRef.current.byokProvider?.(),\n      ...(initialOptions.initialResumeSnapshot !== undefined && {\n        initialResumeSnapshot: initialOptions.initialResumeSnapshot,\n      }),\n      ...(initialOptions.context !== undefined && {\n        context: initialOptions.context,\n      }),\n      devtools: {\n        ...initialOptions.devtools,\n        framework: 'react',\n        hookName: 'useChat',\n        outputKind: initialOptions.outputSchema ? 'structured' : 'chat',\n      },\n      onResponse: (response) => {\n        // ChatClient awaits this return value. Queuing would drop the promise.\n        if (!getActiveInstance()) return\n        return optionsRef.current.onResponse?.(response)\n      },\n      onChunk: (chunk: StreamChunk) => {\n        runOrQueueForActiveInstance(() => {\n          optionsRef.current.onChunk?.(chunk)\n        })\n      },\n      onFinish: (message: UIMessage<TTools>) => {\n        runOrQueueForActiveInstance(() => {\n          optionsRef.current.onFinish?.(message)\n        })\n      },\n      onError: (error: Error) => {\n        runOrQueueForActiveInstance(() => {\n          optionsRef.current.onError?.(error)\n        })\n      },\n      ...(initialOptions.tools !== undefined && {\n        tools: initialOptions.tools,\n      }),\n      ...(initialOptions.interrupts !== undefined && {\n        interrupts: initialOptions.interrupts,\n      }),\n      onCustomEvent: (eventType, data, context) => {\n        runOrQueueForActiveInstance(() => {\n          optionsRef.current.onCustomEvent?.(eventType, data, context)\n        })\n      },\n      ...(options.streamProcessor !== undefined && {\n        streamProcessor: options.streamProcessor,\n      }),\n      onMessagesChange: (newMessages: Array<UIMessage<TTools>>) => {\n        runOrQueueForActiveInstance(() => {\n          setMessages(newMessages)\n          const currentInstance = getActiveInstance()\n          if (!currentInstance) return\n          setHasOlderMessages(currentInstance.getHasOlderMessages())\n        })\n      },\n      onLoadingChange: (newIsLoading: boolean) => {\n        runOrQueueForActiveInstance(() => {\n          const currentInstance = getActiveInstance()\n          if (!currentInstance) return\n          setIsLoading(newIsLoading)\n          syncResumeState(currentInstance)\n        })\n      },\n      onErrorChange: (newError: Error | undefined) => {\n        runOrQueueForActiveInstance(() => {\n          setError(newError)\n        })\n      },\n      onStatusChange: (status: ChatClientState) => {\n        runOrQueueForActiveInstance(() => {\n          setStatus(status)\n        })\n      },\n      onSubscriptionChange: (nextIsSubscribed: boolean) => {\n        runOrQueueForActiveInstance(() => {\n          setIsSubscribed(nextIsSubscribed)\n        })\n      },\n      onConnectionStatusChange: (nextStatus: ConnectionStatus) => {\n        runOrQueueForActiveInstance(() => {\n          setConnectionStatus(nextStatus)\n        })\n      },\n      onSessionGeneratingChange: (isGenerating: boolean) => {\n        runOrQueueForActiveInstance(() => {\n          setSessionGenerating(isGenerating)\n        })\n      },\n      ...(optionsRef.current.queue !== undefined && {\n        queue: optionsRef.current.queue,\n      }),\n      onQueueChange: (nextQueue: Array<QueuedMessage>) => {\n        runOrQueueForActiveInstance(() => {\n          setQueue(nextQueue)\n        })\n      },\n      onRunIdChange: (nextRunId) => {\n        runOrQueueForActiveInstance(() => {\n          setRunId(nextRunId)\n        })\n      },\n      onResumeStateChange: (_nextResumeState, nextPendingInterrupts) => {\n        runOrQueueForActiveInstance(() => {\n          setInterruptState((current) => ({\n            ...current,\n            interrupts: nextPendingInterrupts,\n            pendingInterrupts: nextPendingInterrupts,\n          }))\n        })\n      },\n      onInterruptStateChange: (nextInterruptState, context) => {\n        runOrQueueForActiveInstance(() => {\n          setInterruptState(nextInterruptState)\n          optionsRef.current.onInterruptStateChange?.(\n            nextInterruptState,\n            context,\n          )\n        })\n      },\n    })\n    instanceHolder.current = instance\n    return { client: instance, initialization: initializationState }\n  }, [clientId, syncResumeState])\n\n  useEffect(() => {\n    activeClientRef.current = client\n    try {\n      // Keep initialization closed while draining so callbacks published by a\n      // queued callback are appended and delivered in the same commit.\n      while (initialization.callbacks.length > 0) {\n        if (activeClientRef.current !== client) {\n          initialization.callbacks.length = 0\n          break\n        }\n        initialization.callbacks.shift()?.()\n      }\n    } finally {\n      // A throw from a queued user callback must not leave the queue closed.\n      initialization.ready = true\n    }\n  }, [client, initialization])\n\n  useEffect(() => {\n    const clientMessages = client.getMessages()\n    if (clientMessages !== messagesRef.current) {\n      setMessages(clientMessages)\n    }\n    setHasOlderMessages(client.getHasOlderMessages())\n  }, [client])\n\n  // Sync each wire-payload slot in its own effect so an unrelated option\n  // changing doesn't re-run the others. `updateOptions` declares strict-optional\n  // fields and rejects explicit `undefined` under EOPT, so guard the optional\n  // slots before passing them.\n  useEffect(() => {\n    client.updateOptions({ body: options.body })\n  }, [client, options.body])\n\n  useEffect(() => {\n    if (options.forwardedProps !== undefined) {\n      client.updateOptions({ forwardedProps: options.forwardedProps })\n    }\n  }, [client, options.forwardedProps])\n\n  useEffect(() => {\n    if (options.tools !== undefined) {\n      client.updateOptions({ tools: options.tools })\n    }\n  }, [client, options.tools])\n\n  useEffect(() => {\n    client.updateOptions({ context: options.context })\n  }, [client, options.context])\n\n  useEffect(() => {\n    if (options.queue !== undefined) {\n      client.updateOptions({ queue: options.queue })\n    }\n  }, [client, options.queue])\n\n  useEffect(() => {\n    if (options.live) {\n      client.subscribe()\n      subscribedRef.current = true\n    } else if (subscribedRef.current) {\n      // Only tear down a subscription we actually started. Calling\n      // `unsubscribe()` on initial mount (when `live` was never enabled) would\n      // abort an in-flight delivery resume — `resumeInFlightRun` is kicked off\n      // in the client constructor, and `unsubscribe()` cancels the shared\n      // in-flight stream — so a mid-stream reload would drop its rejoin before\n      // it delivers a single chunk. This is exactly why a reload froze instead\n      // of continuing.\n      client.unsubscribe()\n      subscribedRef.current = false\n    }\n  }, [client, options.live])\n\n  // ONLY THE VIEW ON SCREEN HOLDS A STREAM.\n  //\n  // A page can own many chats — 40 sandboxes, 40 conversations — and a browser\n  // allows only ~6 connections per origin. One long-lived stream per chat reaches\n  // that ceiling after a handful of views, and every request after it QUEUES:\n  // measured, an in-page fetch took over two minutes while the same request from\n  // outside the browser took 17ms. So the connection follows the view.\n  //\n  // Immediate, not deferred: the deferred teardown below can be skipped when the\n  // same client remounts, which is right for disposal but useless for a\n  // connection. `attach` is idempotent and `detach` keeps the transcript and the\n  // resume pointer, so a Strict Mode remount is just detach-then-attach and the\n  // run is picked straight back up from the durable log.\n  useEffect(() => {\n    client.attach()\n    return () => {\n      client.detach()\n    }\n  }, [client])\n\n  useEffect(() => {\n    if (cleanupDisposalRef.current?.client === client) {\n      clearTimeout(cleanupDisposalRef.current.timeout)\n      cleanupDisposalRef.current = null\n    }\n    if (cleanupInvalidationRef.current) {\n      clearTimeout(cleanupInvalidationRef.current)\n      cleanupInvalidationRef.current = null\n    }\n    client.mountDevtools()\n    // Delivery-durability resume is transparent: the resumable SSE connection\n    // adapter re-attaches via the browser's native Last-Event-ID on reconnect.\n    // We only seed interrupt (state) resume from the client here.\n    syncResumeState(client)\n\n    return () => {\n      cleanupInvalidationRef.current = setTimeout(() => {\n        if (activeClientRef.current === client) {\n          activeClientRef.current = null\n        }\n        cleanupInvalidationRef.current = null\n      }, 0)\n      // Soft cleanup only: do NOT stop/unsubscribe here. React Strict Mode\n      // remounts fire this cleanup then re-attach the same client one tick\n      // later; calling `stop()` would abort a constructor rejoin\n      // (`resumeInFlightRun`) and can wipe the durable resume pointer before\n      // the first chunk. Real teardown lives in the deferred dispose path\n      // below, which only runs when the client is not remounted.\n      // Subscribe/unsubscribe on `options.live` is still owned by the\n      // dedicated effect above for live toggles.\n      const disposal = {\n        client,\n        timeout: setTimeout(() => {\n          if (optionsRef.current.live) {\n            client.unsubscribe()\n          } else {\n            client.stop()\n          }\n          client.dispose()\n          if (cleanupDisposalRef.current === disposal) {\n            cleanupDisposalRef.current = null\n          }\n        }, 0),\n      }\n      cleanupDisposalRef.current = disposal\n    }\n  }, [client, syncResumeState])\n\n  const sendMessage = useCallback(\n    async (\n      content: string | MultimodalContent,\n      sendOptions?: SendMessageOptions,\n    ) => {\n      try {\n        await client.sendMessage(content, undefined, sendOptions)\n      } finally {\n        syncResumeState(client)\n      }\n    },\n    [client, syncResumeState],\n  )\n\n  const cancelQueued = useCallback(\n    (id: string) => {\n      client.cancelQueued(id)\n    },\n    [client, syncResumeState],\n  )\n\n  const append = useCallback(\n    async (message: ModelMessage | UIMessage) => {\n      try {\n        await client.append(message)\n      } finally {\n        syncResumeState(client)\n      }\n    },\n    [client, syncResumeState],\n  )\n\n  const reload = useCallback(async () => {\n    try {\n      await client.reload()\n    } finally {\n      syncResumeState(client)\n    }\n  }, [client, syncResumeState])\n\n  const loadOlderMessages = useCallback(async () => {\n    await client.loadOlderMessages()\n    setHasOlderMessages(client.getHasOlderMessages())\n  }, [client])\n\n  const stop = useCallback(() => {\n    client.stop()\n  }, [client])\n\n  const clear = useCallback(() => {\n    client.clear()\n    syncResumeState(client)\n  }, [client, syncResumeState])\n\n  const setMessagesManually = useCallback(\n    (newMessages: Array<UIMessage<TTools>>) => {\n      client.setMessagesManually(newMessages)\n    },\n    [client],\n  )\n\n  const addToolResult = useCallback(\n    async (result: {\n      toolCallId: string\n      tool: string\n      output: any\n      state?: 'output-available' | 'output-error'\n      errorText?: string\n    }) => {\n      await client.addToolResult(result)\n    },\n    [client],\n  )\n\n  const addToolApprovalResponse = useCallback(\n    async (response: { id: string; approved: boolean }) => {\n      await client.addToolApprovalResponse(response)\n      syncResumeState(client)\n    },\n    [client, syncResumeState],\n  )\n\n  const resumeInterrupts = useCallback(\n    async (resumeItems: Array<RunAgentResumeItem>, state?: ChatResumeState) => {\n      const result = await client.resumeInterrupts(\n        resumeItems,\n        state ?? undefined,\n      )\n      syncResumeState(client)\n      return result\n    },\n    [client, syncResumeState],\n  )\n\n  const resolveInterrupts = useCallback(\n    (\n      resolution:\n        | boolean\n        | ((\n            interrupt: ResolvableChatInterrupt<TTools, TInterrupts>,\n          ) => undefined),\n    ) => {\n      if (typeof resolution === 'boolean') {\n        client.resolveInterrupts(resolution)\n      } else {\n        client.resolveInterrupts(resolution)\n      }\n    },\n    [client],\n  )\n\n  const cancelInterrupts = useCallback(() => {\n    client.cancelInterrupts()\n  }, [client])\n\n  const retryInterrupts = useCallback(() => {\n    client.retryInterrupts()\n  }, [client])\n\n  const resumeInterruptsUnsafe = useCallback(\n    (resumeItems: Array<RunAgentResumeItem>, state?: ChatResumeState) =>\n      client.resumeInterruptsUnsafe(resumeItems, state),\n    [client],\n  )\n\n  // The \"active\" structured-output part is the one on the assistant message\n  // that follows the latest user message. No such message exists between\n  // sendMessage() and the first chunk, so partial/final naturally read as\n  // cleared. Historical parts on earlier assistant messages remain available\n  // via `messages` directly.\n  //\n  // When there is NO user message yet (e.g. `initialMessages` contains only\n  // a stale assistant turn or a system prompt) we deliberately return null\n  // rather than scanning historical assistants — otherwise a `final` from a\n  // previous session would leak into the hook value on first render.\n  const renderedMessages = client.getMessages()\n\n  const activeStructuredPart = useMemo<StructuredOutputPart | null>(() => {\n    let lastUserIndex = -1\n    for (let i = renderedMessages.length - 1; i >= 0; i--) {\n      if (renderedMessages[i]?.role === 'user') {\n        lastUserIndex = i\n        break\n      }\n    }\n    if (lastUserIndex === -1) return null\n    for (let i = renderedMessages.length - 1; i > lastUserIndex; i--) {\n      const m = renderedMessages[i]\n      if (m?.role !== 'assistant') continue\n      const part = m.parts.find(\n        (p): p is StructuredOutputPart => p.type === 'structured-output',\n      )\n      if (part) return part\n    }\n    return null\n  }, [renderedMessages])\n\n  const partial = useMemo<Partial>(() => {\n    if (!activeStructuredPart) return {} as Partial\n    const v = activeStructuredPart.partial ?? activeStructuredPart.data\n    return (v ?? {}) as Partial\n  }, [activeStructuredPart])\n\n  const final = useMemo<Final | null>(() => {\n    if (!activeStructuredPart || activeStructuredPart.status !== 'complete') {\n      return null\n    }\n    return activeStructuredPart.data as Final\n  }, [activeStructuredPart])\n\n  // The runtime shape unconditionally exposes partial/final; the public\n  // return type hides them when no outputSchema was supplied. TS can't\n  // structurally narrow across that conditional, so the `as` is the seam.\n  // oxlint-disable-next-line eslint-js/no-restricted-syntax -- hook return shape diverges from generic UseChatReturn<TTools, TSchema> due to conditional type on TSchema; TS can't structurally narrow\n  return {\n    messages: renderedMessages,\n    sendMessage,\n    append,\n    reload,\n    stop,\n    isLoading,\n    hasOlderMessages,\n    loadOlderMessages,\n    error,\n    status,\n    isSubscribed,\n    connectionStatus,\n    sessionGenerating,\n    setMessages: setMessagesManually,\n    clear,\n    addToolResult,\n    addToolApprovalResponse,\n    queue,\n    cancelQueued,\n    runId,\n    interrupts: interruptState.interrupts,\n    pendingInterrupts: interruptState.pendingInterrupts,\n    interruptErrors: interruptState.interruptErrors,\n    resuming: interruptState.resuming,\n    resolveInterrupts,\n    cancelInterrupts,\n    retryInterrupts,\n    resumeInterruptsUnsafe,\n    resumeInterrupts,\n    partial,\n    final,\n  } as unknown as UseChatReturn<TTools, TSchema, TInterrupts>\n}\n"],"mappings":";;;;AAgCA,IAAM,mBAAmB,OAAO,OAAO,CAAC,CAAC;AACzC,IAAM,yBAAyB,OAAO,OAAO,CAAC,CAAC;AAE/C,SAAgB,QAQd,SAC6C;CAI7C,MAAM,SAAS,MAAM;CACrB,MAAM,WAAW,QAAQ,YAAY;CAErC,MAAM,CAAC,UAAU,eAAe,SAC9B,QAAQ,mBAAmB,CAAC,CAC9B;CACA,MAAM,CAAC,WAAW,gBAAgB,SAAS,KAAK;CAChD,MAAM,CAAC,kBAAkB,uBAAuB,SAAS,KAAK;CAC9D,MAAM,CAAC,OAAO,YAAY,SAA4B,KAAA,CAAS;CAC/D,MAAM,CAAC,QAAQ,aAAa,SAA0B,OAAO;CAC7D,MAAM,CAAC,cAAc,mBAAmB,SAAS,KAAK;CACtD,MAAM,CAAC,kBAAkB,uBACvB,SAA2B,cAAc;CAC3C,MAAM,CAAC,mBAAmB,wBAAwB,SAAS,KAAK;CAChE,MAAM,CAAC,OAAO,YAAY,SAA+B,CAAC,CAAC;CAC3D,MAAM,CAAC,OAAO,YAAY,SAAwB,IAAI;CACtD,MAAM,CAAC,gBAAgB,qBAAqB,gBAEnC;EACP,YAAY;EACZ,mBAAmB;EACnB,iBAAiB;EACjB,UAAU;CACZ,EAAE;CAMF,MAAM,cAAc,OAClB,QAAQ,mBAAmB,CAAC,CAC9B;CACA,MAAM,kBAAkB,OAAO,IAAI;CACnC,MAAM,gBAAgB,OAAO,KAAK;CAClC,MAAM,kBAAkB,OAA0B,IAAI;CACtD,MAAM,yBAAyB,OAC7B,IACF;CACA,MAAM,qBAAqB,OAGjB,IAAI;CAGd,YAAY,UAAU;CAGtB,MAAM,aACJ,OAA+D,OAAO;CACxE,WAAW,UAAU;CAErB,MAAM,kBAAkB,aAAa,WAA8B;EACjE,IAAI,CAAC,QAAQ;EACb,SAAS,OAAO,gBAAgB,CAAC;EACjC,kBAAkB,OAAO,kBAAkB,CAAC;CAC9C,GAAG,CAAC,CAAC;CAGL,MAAM,EAAE,QAAQ,mBAAmB,cAAc;EAC/C,MAAM,gBAAgB,QAAQ,mBAAmB,CAAC;EAClD,gBAAgB,UAAU;EAM1B,MAAM,iBAAiB,WAAW;EAClC,MAAM,YAAY,eAAe,aAC7B,EAAE,YAAY,eAAe,WAAW,IACxC,EAAE,SAAS,eAAe,QAAQ;EAEtC,MAAM,iBAEF,EAAE,SAAS,KAAA,EAAU;EACzB,MAAM,0BAA0B;GAC9B,MAAM,kBAAkB,eAAe;GACvC,IAAI,CAAC,mBAAmB,gBAAgB,YAAY,iBAClD;GAEF,OAAO;EACT;EAKA,MAAM,sBAAsB;GAC1B,OAAO;GACP,WAAW,CAAC;EACd;EACA,MAAM,+BAA+B,aAAyB;GAC5D,IAAI,CAAC,oBAAoB,OAAO;IAC9B,oBAAoB,UAAU,KAAK,QAAQ;IAC3C;GACF;GACA,MAAM,kBAAkB,eAAe;GACvC,IAAI,CAAC,mBAAmB,gBAAgB,YAAY,iBAClD;GACF,SAAS;EACX;EACA,MAAM,WAAW,IAAI,WAA0C;GAC7D,uBAAuB;GACvB,GAAG;GACH,iBAAiB;GACjB,GAAI,OAAO,eAAe,aAAa,YACvC,eAAe,gBAAgB,OAC3B;IACE,aAAa;IACb,UAAU,eAAe;IACzB,GAAI,eAAe,YAAY,KAAA,KAAa,EAC1C,SAAS,eAAe,QAC1B;GACF,IACA,OAAO,eAAe,aAAa,YACjC,eAAe,cACf;IACE,aAAa,eAAe;IAC5B,UAAU,eAAe;GAC3B,IACA,EACE,GAAI,eAAe,aAAa,KAAA,KAAa,EAC3C,UAAU,eAAe,SAC3B,EACF;GACN,GAAI,eAAe,SAAS,KAAA,KAAa,EAAE,MAAM,eAAe,KAAK;GACrE,GAAI,eAAe,mBAAmB,KAAA,KAAa,EACjD,gBAAgB,eAAe,eACjC;GACA,GAAI,eAAe,SAAS,KAAA,KAAa,EAAE,MAAM,eAAe,KAAK;GACrE,oBAAoB,WAAW,QAAQ,eAAe;GACtD,GAAI,eAAe,0BAA0B,KAAA,KAAa,EACxD,uBAAuB,eAAe,sBACxC;GACA,GAAI,eAAe,YAAY,KAAA,KAAa,EAC1C,SAAS,eAAe,QAC1B;GACA,UAAU;IACR,GAAG,eAAe;IAClB,WAAW;IACX,UAAU;IACV,YAAY,eAAe,eAAe,eAAe;GAC3D;GACA,aAAa,aAAa;IAExB,IAAI,CAAC,kBAAkB,GAAG;IAC1B,OAAO,WAAW,QAAQ,aAAa,QAAQ;GACjD;GACA,UAAU,UAAuB;IAC/B,kCAAkC;KAChC,WAAW,QAAQ,UAAU,KAAK;IACpC,CAAC;GACH;GACA,WAAW,YAA+B;IACxC,kCAAkC;KAChC,WAAW,QAAQ,WAAW,OAAO;IACvC,CAAC;GACH;GACA,UAAU,UAAiB;IACzB,kCAAkC;KAChC,WAAW,QAAQ,UAAU,KAAK;IACpC,CAAC;GACH;GACA,GAAI,eAAe,UAAU,KAAA,KAAa,EACxC,OAAO,eAAe,MACxB;GACA,GAAI,eAAe,eAAe,KAAA,KAAa,EAC7C,YAAY,eAAe,WAC7B;GACA,gBAAgB,WAAW,MAAM,YAAY;IAC3C,kCAAkC;KAChC,WAAW,QAAQ,gBAAgB,WAAW,MAAM,OAAO;IAC7D,CAAC;GACH;GACA,GAAI,QAAQ,oBAAoB,KAAA,KAAa,EAC3C,iBAAiB,QAAQ,gBAC3B;GACA,mBAAmB,gBAA0C;IAC3D,kCAAkC;KAChC,YAAY,WAAW;KACvB,MAAM,kBAAkB,kBAAkB;KAC1C,IAAI,CAAC,iBAAiB;KACtB,oBAAoB,gBAAgB,oBAAoB,CAAC;IAC3D,CAAC;GACH;GACA,kBAAkB,iBAA0B;IAC1C,kCAAkC;KAChC,MAAM,kBAAkB,kBAAkB;KAC1C,IAAI,CAAC,iBAAiB;KACtB,aAAa,YAAY;KACzB,gBAAgB,eAAe;IACjC,CAAC;GACH;GACA,gBAAgB,aAAgC;IAC9C,kCAAkC;KAChC,SAAS,QAAQ;IACnB,CAAC;GACH;GACA,iBAAiB,WAA4B;IAC3C,kCAAkC;KAChC,UAAU,MAAM;IAClB,CAAC;GACH;GACA,uBAAuB,qBAA8B;IACnD,kCAAkC;KAChC,gBAAgB,gBAAgB;IAClC,CAAC;GACH;GACA,2BAA2B,eAAiC;IAC1D,kCAAkC;KAChC,oBAAoB,UAAU;IAChC,CAAC;GACH;GACA,4BAA4B,iBAA0B;IACpD,kCAAkC;KAChC,qBAAqB,YAAY;IACnC,CAAC;GACH;GACA,GAAI,WAAW,QAAQ,UAAU,KAAA,KAAa,EAC5C,OAAO,WAAW,QAAQ,MAC5B;GACA,gBAAgB,cAAoC;IAClD,kCAAkC;KAChC,SAAS,SAAS;IACpB,CAAC;GACH;GACA,gBAAgB,cAAc;IAC5B,kCAAkC;KAChC,SAAS,SAAS;IACpB,CAAC;GACH;GACA,sBAAsB,kBAAkB,0BAA0B;IAChE,kCAAkC;KAChC,mBAAmB,aAAa;MAC9B,GAAG;MACH,YAAY;MACZ,mBAAmB;KACrB,EAAE;IACJ,CAAC;GACH;GACA,yBAAyB,oBAAoB,YAAY;IACvD,kCAAkC;KAChC,kBAAkB,kBAAkB;KACpC,WAAW,QAAQ,yBACjB,oBACA,OACF;IACF,CAAC;GACH;EACF,CAAC;EACD,eAAe,UAAU;EACzB,OAAO;GAAE,QAAQ;GAAU,gBAAgB;EAAoB;CACjE,GAAG,CAAC,UAAU,eAAe,CAAC;CAE9B,gBAAgB;EACd,gBAAgB,UAAU;EAC1B,IAAI;GAGF,OAAO,eAAe,UAAU,SAAS,GAAG;IAC1C,IAAI,gBAAgB,YAAY,QAAQ;KACtC,eAAe,UAAU,SAAS;KAClC;IACF;IACA,eAAe,UAAU,MAAM,CAAC,GAAG;GACrC;EACF,UAAU;GAER,eAAe,QAAQ;EACzB;CACF,GAAG,CAAC,QAAQ,cAAc,CAAC;CAE3B,gBAAgB;EACd,MAAM,iBAAiB,OAAO,YAAY;EAC1C,IAAI,mBAAmB,YAAY,SACjC,YAAY,cAAc;EAE5B,oBAAoB,OAAO,oBAAoB,CAAC;CAClD,GAAG,CAAC,MAAM,CAAC;CAMX,gBAAgB;EACd,OAAO,cAAc,EAAE,MAAM,QAAQ,KAAK,CAAC;CAC7C,GAAG,CAAC,QAAQ,QAAQ,IAAI,CAAC;CAEzB,gBAAgB;EACd,IAAI,QAAQ,mBAAmB,KAAA,GAC7B,OAAO,cAAc,EAAE,gBAAgB,QAAQ,eAAe,CAAC;CAEnE,GAAG,CAAC,QAAQ,QAAQ,cAAc,CAAC;CAEnC,gBAAgB;EACd,IAAI,QAAQ,UAAU,KAAA,GACpB,OAAO,cAAc,EAAE,OAAO,QAAQ,MAAM,CAAC;CAEjD,GAAG,CAAC,QAAQ,QAAQ,KAAK,CAAC;CAE1B,gBAAgB;EACd,OAAO,cAAc,EAAE,SAAS,QAAQ,QAAQ,CAAC;CACnD,GAAG,CAAC,QAAQ,QAAQ,OAAO,CAAC;CAE5B,gBAAgB;EACd,IAAI,QAAQ,UAAU,KAAA,GACpB,OAAO,cAAc,EAAE,OAAO,QAAQ,MAAM,CAAC;CAEjD,GAAG,CAAC,QAAQ,QAAQ,KAAK,CAAC;CAE1B,gBAAgB;EACd,IAAI,QAAQ,MAAM;GAChB,OAAO,UAAU;GACjB,cAAc,UAAU;EAC1B,OAAO,IAAI,cAAc,SAAS;GAQhC,OAAO,YAAY;GACnB,cAAc,UAAU;EAC1B;CACF,GAAG,CAAC,QAAQ,QAAQ,IAAI,CAAC;CAezB,gBAAgB;EACd,OAAO,OAAO;EACd,aAAa;GACX,OAAO,OAAO;EAChB;CACF,GAAG,CAAC,MAAM,CAAC;CAEX,gBAAgB;EACd,IAAI,mBAAmB,SAAS,WAAW,QAAQ;GACjD,aAAa,mBAAmB,QAAQ,OAAO;GAC/C,mBAAmB,UAAU;EAC/B;EACA,IAAI,uBAAuB,SAAS;GAClC,aAAa,uBAAuB,OAAO;GAC3C,uBAAuB,UAAU;EACnC;EACA,OAAO,cAAc;EAIrB,gBAAgB,MAAM;EAEtB,aAAa;GACX,uBAAuB,UAAU,iBAAiB;IAChD,IAAI,gBAAgB,YAAY,QAC9B,gBAAgB,UAAU;IAE5B,uBAAuB,UAAU;GACnC,GAAG,CAAC;GASJ,MAAM,WAAW;IACf;IACA,SAAS,iBAAiB;KACxB,IAAI,WAAW,QAAQ,MACrB,OAAO,YAAY;UAEnB,OAAO,KAAK;KAEd,OAAO,QAAQ;KACf,IAAI,mBAAmB,YAAY,UACjC,mBAAmB,UAAU;IAEjC,GAAG,CAAC;GACN;GACA,mBAAmB,UAAU;EAC/B;CACF,GAAG,CAAC,QAAQ,eAAe,CAAC;CAE5B,MAAM,cAAc,YAClB,OACE,SACA,gBACG;EACH,IAAI;GACF,MAAM,OAAO,YAAY,SAAS,KAAA,GAAW,WAAW;EAC1D,UAAU;GACR,gBAAgB,MAAM;EACxB;CACF,GACA,CAAC,QAAQ,eAAe,CAC1B;CAEA,MAAM,eAAe,aAClB,OAAe;EACd,OAAO,aAAa,EAAE;CACxB,GACA,CAAC,QAAQ,eAAe,CAC1B;CAEA,MAAM,SAAS,YACb,OAAO,YAAsC;EAC3C,IAAI;GACF,MAAM,OAAO,OAAO,OAAO;EAC7B,UAAU;GACR,gBAAgB,MAAM;EACxB;CACF,GACA,CAAC,QAAQ,eAAe,CAC1B;CAEA,MAAM,SAAS,YAAY,YAAY;EACrC,IAAI;GACF,MAAM,OAAO,OAAO;EACtB,UAAU;GACR,gBAAgB,MAAM;EACxB;CACF,GAAG,CAAC,QAAQ,eAAe,CAAC;CAE5B,MAAM,oBAAoB,YAAY,YAAY;EAChD,MAAM,OAAO,kBAAkB;EAC/B,oBAAoB,OAAO,oBAAoB,CAAC;CAClD,GAAG,CAAC,MAAM,CAAC;CAEX,MAAM,OAAO,kBAAkB;EAC7B,OAAO,KAAK;CACd,GAAG,CAAC,MAAM,CAAC;CAEX,MAAM,QAAQ,kBAAkB;EAC9B,OAAO,MAAM;EACb,gBAAgB,MAAM;CACxB,GAAG,CAAC,QAAQ,eAAe,CAAC;CAE5B,MAAM,sBAAsB,aACzB,gBAA0C;EACzC,OAAO,oBAAoB,WAAW;CACxC,GACA,CAAC,MAAM,CACT;CAEA,MAAM,gBAAgB,YACpB,OAAO,WAMD;EACJ,MAAM,OAAO,cAAc,MAAM;CACnC,GACA,CAAC,MAAM,CACT;CAEA,MAAM,0BAA0B,YAC9B,OAAO,aAAgD;EACrD,MAAM,OAAO,wBAAwB,QAAQ;EAC7C,gBAAgB,MAAM;CACxB,GACA,CAAC,QAAQ,eAAe,CAC1B;CAEA,MAAM,mBAAmB,YACvB,OAAO,aAAwC,UAA4B;EACzE,MAAM,SAAS,MAAM,OAAO,iBAC1B,aACA,SAAS,KAAA,CACX;EACA,gBAAgB,MAAM;EACtB,OAAO;CACT,GACA,CAAC,QAAQ,eAAe,CAC1B;CAEA,MAAM,oBAAoB,aAEtB,eAKG;EACH,IAAI,OAAO,eAAe,WACxB,OAAO,kBAAkB,UAAU;OAEnC,OAAO,kBAAkB,UAAU;CAEvC,GACA,CAAC,MAAM,CACT;CAEA,MAAM,mBAAmB,kBAAkB;EACzC,OAAO,iBAAiB;CAC1B,GAAG,CAAC,MAAM,CAAC;CAEX,MAAM,kBAAkB,kBAAkB;EACxC,OAAO,gBAAgB;CACzB,GAAG,CAAC,MAAM,CAAC;CAEX,MAAM,yBAAyB,aAC5B,aAAwC,UACvC,OAAO,uBAAuB,aAAa,KAAK,GAClD,CAAC,MAAM,CACT;CAYA,MAAM,mBAAmB,OAAO,YAAY;CAE5C,MAAM,uBAAuB,cAA2C;EACtE,IAAI,gBAAgB;EACpB,KAAK,IAAI,IAAI,iBAAiB,SAAS,GAAG,KAAK,GAAG,KAChD,IAAI,iBAAiB,EAAE,EAAE,SAAS,QAAQ;GACxC,gBAAgB;GAChB;EACF;EAEF,IAAI,kBAAkB,IAAI,OAAO;EACjC,KAAK,IAAI,IAAI,iBAAiB,SAAS,GAAG,IAAI,eAAe,KAAK;GAChE,MAAM,IAAI,iBAAiB;GAC3B,IAAI,GAAG,SAAS,aAAa;GAC7B,MAAM,OAAO,EAAE,MAAM,MAClB,MAAiC,EAAE,SAAS,mBAC/C;GACA,IAAI,MAAM,OAAO;EACnB;EACA,OAAO;CACT,GAAG,CAAC,gBAAgB,CAAC;CAErB,MAAM,UAAU,cAAuB;EACrC,IAAI,CAAC,sBAAsB,OAAO,CAAC;EAEnC,OADU,qBAAqB,WAAW,qBAAqB,QAClD,CAAC;CAChB,GAAG,CAAC,oBAAoB,CAAC;CAEzB,MAAM,QAAQ,cAA4B;EACxC,IAAI,CAAC,wBAAwB,qBAAqB,WAAW,YAC3D,OAAO;EAET,OAAO,qBAAqB;CAC9B,GAAG,CAAC,oBAAoB,CAAC;CAMzB,OAAO;EACL,UAAU;EACV;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA,aAAa;EACb;EACA;EACA;EACA;EACA;EACA;EACA,YAAY,eAAe;EAC3B,mBAAmB,eAAe;EAClC,iBAAiB,eAAe;EAChC,UAAU,eAAe;EACzB;EACA;EACA;EACA;EACA;EACA;EACA;CACF;AACF"}
+{"version":3,"file":"use-chat.js","names":[],"sources":["../../src/use-chat.ts"],"sourcesContent":["import { ChatClient } from '@tanstack/ai-client'\nimport { createChatDevtoolsBridge } from '@tanstack/ai-client/devtools'\nimport { useCallback, useEffect, useId, useMemo, useRef, useState } from 'react'\nimport type {\n  AnyClientTool,\n  InferSchemaType,\n  InterruptDefinition,\n  ModelMessage,\n  RunAgentResumeItem,\n  SchemaInput,\n  StreamChunk,\n} from '@tanstack/ai/client'\nimport type {\n  ChatClientState,\n  ResolvableChatInterrupt,\n  ChatInterruptState,\n  ChatResumeState,\n  ConnectionStatus,\n  InferredClientContext,\n  QueuedMessage,\n  SendMessageOptions,\n  StructuredOutputPart,\n} from '@tanstack/ai-client'\n\nimport type {\n  DeepPartial,\n  MultimodalContent,\n  UIMessage,\n  UseChatOptions,\n  UseChatReturn,\n} from './types'\n\nconst EMPTY_INTERRUPTS = Object.freeze([])\nconst EMPTY_INTERRUPT_ERRORS = Object.freeze([])\n\nexport function useChat<\n  const TTools extends ReadonlyArray<AnyClientTool> = any,\n  TSchema extends SchemaInput | undefined = undefined,\n  TContext = InferredClientContext<TTools>,\n  const TInterrupts extends ReadonlyArray<\n    InterruptDefinition<any, any, any, any>\n  > = readonly [],\n  const TSubagents extends ReadonlyArray<{ name: string }> | undefined =\n    undefined,\n>(\n  options: UseChatOptions<TTools, TSchema, TContext, TInterrupts, TSubagents>,\n): UseChatReturn<TTools, TSchema, TInterrupts, TSubagents> {\n  // The hook's identity is its `threadId`. Reload with the same `threadId`\n  // restores the same conversation. `hookId` is only a React recreation key\n  // when no `threadId` is given. It is never sent on the wire.\n  const hookId = useId()\n  const clientId = options.threadId ?? hookId\n\n  const [messages, setMessages] = useState<Array<UIMessage<TTools>>>(\n    options.initialMessages || [],\n  )\n  const [isLoading, setIsLoading] = useState(false)\n  const [hasOlderMessages, setHasOlderMessages] = useState(false)\n  const [error, setError] = useState<Error | undefined>(undefined)\n  const [status, setStatus] = useState<ChatClientState>('ready')\n  const [isSubscribed, setIsSubscribed] = useState(false)\n  const [connectionStatus, setConnectionStatus] =\n    useState<ConnectionStatus>('disconnected')\n  const [sessionGenerating, setSessionGenerating] = useState(false)\n  const [queue, setQueue] = useState<Array<QueuedMessage>>([])\n  const [runId, setRunId] = useState<string | null>(null)\n  const [interruptState, setInterruptState] = useState<\n    ChatInterruptState<TTools, TInterrupts>\n  >(() => ({\n    interrupts: EMPTY_INTERRUPTS,\n    pendingInterrupts: EMPTY_INTERRUPTS,\n    interruptErrors: EMPTY_INTERRUPT_ERRORS,\n    resuming: false,\n  }))\n\n  type Partial = DeepPartial<InferSchemaType<NonNullable<TSchema>>>\n  type Final = InferSchemaType<NonNullable<TSchema>>\n\n  // Track current messages in a ref to preserve them when client is recreated\n  const messagesRef = useRef<Array<UIMessage<TTools>>>(\n    options.initialMessages || [],\n  )\n  const isFirstMountRef = useRef(true)\n  const subscribedRef = useRef(false)\n  const activeClientRef = useRef<ChatClient | null>(null)\n  const cleanupInvalidationRef = useRef<ReturnType<typeof setTimeout> | null>(\n    null,\n  )\n  const cleanupDisposalRef = useRef<{\n    client: ChatClient\n    timeout: ReturnType<typeof setTimeout>\n  } | null>(null)\n\n  // Update ref synchronously during render so it's always current when useMemo runs.\n  messagesRef.current = messages\n\n  // Track current options in a ref to avoid recreating client when options change\n  const optionsRef =\n    useRef<UseChatOptions<TTools, TSchema, TContext, TInterrupts, TSubagents>>(\n      options,\n    )\n  optionsRef.current = options\n\n  const syncResumeState = useCallback((target: ChatClient | null) => {\n    if (!target) return\n    setRunId(target.getCurrentRunId())\n    setInterruptState(target.getInterruptState())\n  }, [])\n\n  // Create ChatClient instance with callbacks to sync state\n  const { client, initialization } = useMemo(() => {\n    const messagesToUse = options.initialMessages || []\n    isFirstMountRef.current = false\n\n    // Build options with conditional spreads for fields whose source\n    // type is `T | undefined` but the ChatClient target uses a strict\n    // optional (`field?: T`) — `exactOptionalPropertyTypes` rejects\n    // assigning `undefined` to those, so we omit the key when absent.\n    const initialOptions = optionsRef.current\n    const transport = initialOptions.connection\n      ? { connection: initialOptions.connection }\n      : { fetcher: initialOptions.fetcher }\n\n    const instanceHolder: {\n      current: ChatClient<TTools, TContext, TInterrupts> | undefined\n    } = { current: undefined }\n    const getActiveInstance = () => {\n      const currentInstance = instanceHolder.current\n      if (!currentInstance || activeClientRef.current !== currentInstance) {\n        return undefined\n      }\n      return currentInstance\n    }\n    // ChatClient may publish while its constructor is running or while async\n    // persistence resolves before commit. Preserve those exact notifications\n    // until this render commits; invoking them here would run state setters and\n    // user callbacks for a client React may abandon.\n    const initializationState = {\n      ready: false,\n      callbacks: [] as Array<() => void>,\n    }\n    const runOrQueueForActiveInstance = (callback: () => void) => {\n      if (!initializationState.ready) {\n        initializationState.callbacks.push(callback)\n        return\n      }\n      const currentInstance = instanceHolder.current\n      if (!currentInstance || activeClientRef.current !== currentInstance)\n        return\n      callback()\n    }\n    const instance = new ChatClient<TTools, TContext, TInterrupts>({\n      devtoolsBridgeFactory: createChatDevtoolsBridge,\n      ...transport,\n      initialMessages: messagesToUse,\n      ...(typeof initialOptions.threadId === 'string' &&\n      initialOptions.persistence === true\n        ? {\n            persistence: true,\n            threadId: initialOptions.threadId,\n            ...(initialOptions.history !== undefined && {\n              history: initialOptions.history,\n            }),\n          }\n        : typeof initialOptions.threadId === 'string' &&\n            initialOptions.persistence\n          ? {\n              persistence: initialOptions.persistence,\n              threadId: initialOptions.threadId,\n            }\n          : {\n              ...(initialOptions.threadId !== undefined && {\n                threadId: initialOptions.threadId,\n              }),\n            }),\n      ...(initialOptions.body !== undefined && { body: initialOptions.body }),\n      ...(initialOptions.forwardedProps !== undefined && {\n        forwardedProps: initialOptions.forwardedProps,\n      }),\n      ...(initialOptions.byok !== undefined && { byok: initialOptions.byok }),\n      byokProvider: () => optionsRef.current.byokProvider?.(),\n      ...(initialOptions.initialResumeSnapshot !== undefined && {\n        initialResumeSnapshot: initialOptions.initialResumeSnapshot,\n      }),\n      ...(initialOptions.context !== undefined && {\n        context: initialOptions.context,\n      }),\n      devtools: {\n        ...initialOptions.devtools,\n        framework: 'react',\n        hookName: 'useChat',\n        outputKind: initialOptions.outputSchema ? 'structured' : 'chat',\n      },\n      onResponse: (response) => {\n        // ChatClient awaits this return value. Queuing would drop the promise.\n        if (!getActiveInstance()) return\n        return optionsRef.current.onResponse?.(response)\n      },\n      onChunk: (chunk: StreamChunk) => {\n        runOrQueueForActiveInstance(() => {\n          optionsRef.current.onChunk?.(chunk)\n        })\n      },\n      onFinish: (message: UIMessage<TTools>) => {\n        runOrQueueForActiveInstance(() => {\n          optionsRef.current.onFinish?.(message)\n        })\n      },\n      onError: (error: Error) => {\n        runOrQueueForActiveInstance(() => {\n          optionsRef.current.onError?.(error)\n        })\n      },\n      ...(initialOptions.tools !== undefined && {\n        tools: initialOptions.tools,\n      }),\n      ...(initialOptions.interrupts !== undefined && {\n        interrupts: initialOptions.interrupts,\n      }),\n      onCustomEvent: (eventType, data, context) => {\n        runOrQueueForActiveInstance(() => {\n          optionsRef.current.onCustomEvent?.(eventType, data, context)\n        })\n      },\n      ...(options.streamProcessor !== undefined && {\n        streamProcessor: options.streamProcessor,\n      }),\n      onMessagesChange: (newMessages: Array<UIMessage<TTools>>) => {\n        runOrQueueForActiveInstance(() => {\n          setMessages(newMessages)\n          const currentInstance = getActiveInstance()\n          if (!currentInstance) return\n          setHasOlderMessages(currentInstance.getHasOlderMessages())\n        })\n      },\n      onLoadingChange: (newIsLoading: boolean) => {\n        runOrQueueForActiveInstance(() => {\n          const currentInstance = getActiveInstance()\n          if (!currentInstance) return\n          setIsLoading(newIsLoading)\n          syncResumeState(currentInstance)\n        })\n      },\n      onErrorChange: (newError: Error | undefined) => {\n        runOrQueueForActiveInstance(() => {\n          setError(newError)\n        })\n      },\n      onStatusChange: (status: ChatClientState) => {\n        runOrQueueForActiveInstance(() => {\n          setStatus(status)\n        })\n      },\n      onSubscriptionChange: (nextIsSubscribed: boolean) => {\n        runOrQueueForActiveInstance(() => {\n          setIsSubscribed(nextIsSubscribed)\n        })\n      },\n      onConnectionStatusChange: (nextStatus: ConnectionStatus) => {\n        runOrQueueForActiveInstance(() => {\n          setConnectionStatus(nextStatus)\n        })\n      },\n      onSessionGeneratingChange: (isGenerating: boolean) => {\n        runOrQueueForActiveInstance(() => {\n          setSessionGenerating(isGenerating)\n        })\n      },\n      ...(optionsRef.current.queue !== undefined && {\n        queue: optionsRef.current.queue,\n      }),\n      onQueueChange: (nextQueue: Array<QueuedMessage>) => {\n        runOrQueueForActiveInstance(() => {\n          setQueue(nextQueue)\n        })\n      },\n      onRunIdChange: (nextRunId) => {\n        runOrQueueForActiveInstance(() => {\n          setRunId(nextRunId)\n        })\n      },\n      onResumeStateChange: (_nextResumeState, nextPendingInterrupts) => {\n        runOrQueueForActiveInstance(() => {\n          setInterruptState((current) => ({\n            ...current,\n            interrupts: nextPendingInterrupts,\n            pendingInterrupts: nextPendingInterrupts,\n          }))\n        })\n      },\n      onInterruptStateChange: (nextInterruptState, context) => {\n        runOrQueueForActiveInstance(() => {\n          setInterruptState(nextInterruptState)\n          optionsRef.current.onInterruptStateChange?.(\n            nextInterruptState,\n            context,\n          )\n        })\n      },\n    })\n    instanceHolder.current = instance\n    return { client: instance, initialization: initializationState }\n  }, [clientId, syncResumeState])\n\n  useEffect(() => {\n    activeClientRef.current = client\n    try {\n      // Keep initialization closed while draining so callbacks published by a\n      // queued callback are appended and delivered in the same commit.\n      while (initialization.callbacks.length > 0) {\n        if (activeClientRef.current !== client) {\n          initialization.callbacks.length = 0\n          break\n        }\n        initialization.callbacks.shift()?.()\n      }\n    } finally {\n      // A throw from a queued user callback must not leave the queue closed.\n      initialization.ready = true\n    }\n  }, [client, initialization])\n\n  useEffect(() => {\n    const clientMessages = client.getMessages()\n    if (clientMessages !== messagesRef.current) {\n      setMessages(clientMessages)\n    }\n    setHasOlderMessages(client.getHasOlderMessages())\n  }, [client])\n\n  // Sync each wire-payload slot in its own effect so an unrelated option\n  // changing doesn't re-run the others. `updateOptions` declares strict-optional\n  // fields and rejects explicit `undefined` under EOPT, so guard the optional\n  // slots before passing them.\n  useEffect(() => {\n    client.updateOptions({ body: options.body })\n  }, [client, options.body])\n\n  useEffect(() => {\n    if (options.forwardedProps !== undefined) {\n      client.updateOptions({ forwardedProps: options.forwardedProps })\n    }\n  }, [client, options.forwardedProps])\n\n  useEffect(() => {\n    if (options.tools !== undefined) {\n      client.updateOptions({ tools: options.tools })\n    }\n  }, [client, options.tools])\n\n  useEffect(() => {\n    client.updateOptions({ context: options.context })\n  }, [client, options.context])\n\n  useEffect(() => {\n    if (options.queue !== undefined) {\n      client.updateOptions({ queue: options.queue })\n    }\n  }, [client, options.queue])\n\n  useEffect(() => {\n    if (options.live) {\n      client.subscribe()\n      subscribedRef.current = true\n    } else if (subscribedRef.current) {\n      // Only tear down a subscription we actually started. Calling\n      // `unsubscribe()` on initial mount (when `live` was never enabled) would\n      // abort an in-flight delivery resume — `resumeInFlightRun` is kicked off\n      // in the client constructor, and `unsubscribe()` cancels the shared\n      // in-flight stream — so a mid-stream reload would drop its rejoin before\n      // it delivers a single chunk. This is exactly why a reload froze instead\n      // of continuing.\n      client.unsubscribe()\n      subscribedRef.current = false\n    }\n  }, [client, options.live])\n\n  // ONLY THE VIEW ON SCREEN HOLDS A STREAM.\n  //\n  // A page can own many chats — 40 sandboxes, 40 conversations — and a browser\n  // allows only ~6 connections per origin. One long-lived stream per chat reaches\n  // that ceiling after a handful of views, and every request after it QUEUES:\n  // measured, an in-page fetch took over two minutes while the same request from\n  // outside the browser took 17ms. So the connection follows the view.\n  //\n  // Immediate, not deferred: the deferred teardown below can be skipped when the\n  // same client remounts, which is right for disposal but useless for a\n  // connection. `attach` is idempotent and `detach` keeps the transcript and the\n  // resume pointer, so a Strict Mode remount is just detach-then-attach and the\n  // run is picked straight back up from the durable log.\n  useEffect(() => {\n    client.attach()\n    return () => {\n      client.detach()\n    }\n  }, [client])\n\n  useEffect(() => {\n    if (cleanupDisposalRef.current?.client === client) {\n      clearTimeout(cleanupDisposalRef.current.timeout)\n      cleanupDisposalRef.current = null\n    }\n    if (cleanupInvalidationRef.current) {\n      clearTimeout(cleanupInvalidationRef.current)\n      cleanupInvalidationRef.current = null\n    }\n    client.mountDevtools()\n    // Delivery-durability resume is transparent: the resumable SSE connection\n    // adapter re-attaches via the browser's native Last-Event-ID on reconnect.\n    // We only seed interrupt (state) resume from the client here.\n    syncResumeState(client)\n\n    return () => {\n      cleanupInvalidationRef.current = setTimeout(() => {\n        if (activeClientRef.current === client) {\n          activeClientRef.current = null\n        }\n        cleanupInvalidationRef.current = null\n      }, 0)\n      // Soft cleanup only: do NOT stop/unsubscribe here. React Strict Mode\n      // remounts fire this cleanup then re-attach the same client one tick\n      // later; calling `stop()` would abort a constructor rejoin\n      // (`resumeInFlightRun`) and can wipe the durable resume pointer before\n      // the first chunk. Real teardown lives in the deferred dispose path\n      // below, which only runs when the client is not remounted.\n      // Subscribe/unsubscribe on `options.live` is still owned by the\n      // dedicated effect above for live toggles.\n      const disposal = {\n        client,\n        timeout: setTimeout(() => {\n          if (optionsRef.current.live) {\n            client.unsubscribe()\n          } else {\n            client.stop()\n          }\n          client.dispose()\n          if (cleanupDisposalRef.current === disposal) {\n            cleanupDisposalRef.current = null\n          }\n        }, 0),\n      }\n      cleanupDisposalRef.current = disposal\n    }\n  }, [client, syncResumeState])\n\n  const sendMessage = useCallback(\n    async (\n      content: string | MultimodalContent,\n      sendOptions?: SendMessageOptions,\n    ) => {\n      try {\n        await client.sendMessage(content, undefined, sendOptions)\n      } finally {\n        syncResumeState(client)\n      }\n    },\n    [client, syncResumeState],\n  )\n\n  const cancelQueued = useCallback(\n    (id: string) => {\n      client.cancelQueued(id)\n    },\n    [client, syncResumeState],\n  )\n\n  const append = useCallback(\n    async (message: ModelMessage | UIMessage) => {\n      try {\n        await client.append(message)\n      } finally {\n        syncResumeState(client)\n      }\n    },\n    [client, syncResumeState],\n  )\n\n  const reload = useCallback(async () => {\n    try {\n      await client.reload()\n    } finally {\n      syncResumeState(client)\n    }\n  }, [client, syncResumeState])\n\n  const loadOlderMessages = useCallback(async () => {\n    await client.loadOlderMessages()\n    setHasOlderMessages(client.getHasOlderMessages())\n  }, [client])\n\n  const stop = useCallback(() => {\n    client.stop()\n  }, [client])\n\n  const clear = useCallback(() => {\n    client.clear()\n    syncResumeState(client)\n  }, [client, syncResumeState])\n\n  const setMessagesManually = useCallback(\n    (newMessages: Array<UIMessage<TTools>>) => {\n      client.setMessagesManually(newMessages)\n    },\n    [client],\n  )\n\n  const addToolResult = useCallback(\n    async (result: {\n      toolCallId: string\n      tool: string\n      output: any\n      state?: 'output-available' | 'output-error'\n      errorText?: string\n    }) => {\n      await client.addToolResult(result)\n    },\n    [client],\n  )\n\n  const addToolApprovalResponse = useCallback(\n    async (response: { id: string; approved: boolean }) => {\n      await client.addToolApprovalResponse(response)\n      syncResumeState(client)\n    },\n    [client, syncResumeState],\n  )\n\n  const resumeInterrupts = useCallback(\n    async (resumeItems: Array<RunAgentResumeItem>, state?: ChatResumeState) => {\n      const result = await client.resumeInterrupts(\n        resumeItems,\n        state ?? undefined,\n      )\n      syncResumeState(client)\n      return result\n    },\n    [client, syncResumeState],\n  )\n\n  const resolveInterrupts = useCallback(\n    (\n      resolution:\n        | boolean\n        | ((\n            interrupt: ResolvableChatInterrupt<TTools, TInterrupts>,\n          ) => undefined),\n    ) => {\n      if (typeof resolution === 'boolean') {\n        client.resolveInterrupts(resolution)\n      } else {\n        client.resolveInterrupts(resolution)\n      }\n    },\n    [client],\n  )\n\n  const cancelInterrupts = useCallback(() => {\n    client.cancelInterrupts()\n  }, [client])\n\n  const retryInterrupts = useCallback(() => {\n    client.retryInterrupts()\n  }, [client])\n\n  const resumeInterruptsUnsafe = useCallback(\n    (resumeItems: Array<RunAgentResumeItem>, state?: ChatResumeState) =>\n      client.resumeInterruptsUnsafe(resumeItems, state),\n    [client],\n  )\n\n  // The \"active\" structured-output part is the one on the assistant message\n  // that follows the latest user message. No such message exists between\n  // sendMessage() and the first chunk, so partial/final naturally read as\n  // cleared. Historical parts on earlier assistant messages remain available\n  // via `messages` directly.\n  //\n  // When there is NO user message yet (e.g. `initialMessages` contains only\n  // a stale assistant turn or a system prompt) we deliberately return null\n  // rather than scanning historical assistants — otherwise a `final` from a\n  // previous session would leak into the hook value on first render.\n  const renderedMessages = client.getMessages()\n\n  const activeStructuredPart = useMemo<StructuredOutputPart | null>(() => {\n    let lastUserIndex = -1\n    for (let i = renderedMessages.length - 1; i >= 0; i--) {\n      if (renderedMessages[i]?.role === 'user') {\n        lastUserIndex = i\n        break\n      }\n    }\n    if (lastUserIndex === -1) return null\n    for (let i = renderedMessages.length - 1; i > lastUserIndex; i--) {\n      const m = renderedMessages[i]\n      if (m?.role !== 'assistant') continue\n      const part = m.parts.find(\n        (p): p is StructuredOutputPart => p.type === 'structured-output',\n      )\n      if (part) return part\n    }\n    return null\n  }, [renderedMessages])\n\n  const partial = useMemo<Partial>(() => {\n    if (!activeStructuredPart) return {} as Partial\n    const v = activeStructuredPart.partial ?? activeStructuredPart.data\n    return (v ?? {}) as Partial\n  }, [activeStructuredPart])\n\n  const final = useMemo<Final | null>(() => {\n    if (!activeStructuredPart || activeStructuredPart.status !== 'complete') {\n      return null\n    }\n    return activeStructuredPart.data as Final\n  }, [activeStructuredPart])\n\n  const subagents = useMemo(() => client.getSubagents(), [renderedMessages])\n\n  // The runtime shape unconditionally exposes partial/final; the public\n  // return type hides them when no outputSchema was supplied. TS can't\n  // structurally narrow across that conditional, so the `as` is the seam.\n  // oxlint-disable-next-line eslint-js/no-restricted-syntax -- hook return shape diverges from generic UseChatReturn<TTools, TSchema> due to conditional type on TSchema; TS can't structurally narrow\n  return {\n    messages: renderedMessages,\n    subagents,\n    sendMessage,\n    append,\n    reload,\n    stop,\n    isLoading,\n    hasOlderMessages,\n    loadOlderMessages,\n    error,\n    status,\n    isSubscribed,\n    connectionStatus,\n    sessionGenerating,\n    setMessages: setMessagesManually,\n    clear,\n    addToolResult,\n    addToolApprovalResponse,\n    queue,\n    cancelQueued,\n    runId,\n    interrupts: interruptState.interrupts,\n    pendingInterrupts: interruptState.pendingInterrupts,\n    interruptErrors: interruptState.interruptErrors,\n    resuming: interruptState.resuming,\n    resolveInterrupts,\n    cancelInterrupts,\n    retryInterrupts,\n    resumeInterruptsUnsafe,\n    resumeInterrupts,\n    partial,\n    final,\n  } as unknown as UseChatReturn<TTools, TSchema, TInterrupts, TSubagents>\n}\n"],"mappings":";;;;AAgCA,IAAM,mBAAmB,OAAO,OAAO,CAAC,CAAC;AACzC,IAAM,yBAAyB,OAAO,OAAO,CAAC,CAAC;AAE/C,SAAgB,QAUd,SACyD;CAIzD,MAAM,SAAS,MAAM;CACrB,MAAM,WAAW,QAAQ,YAAY;CAErC,MAAM,CAAC,UAAU,eAAe,SAC9B,QAAQ,mBAAmB,CAAC,CAC9B;CACA,MAAM,CAAC,WAAW,gBAAgB,SAAS,KAAK;CAChD,MAAM,CAAC,kBAAkB,uBAAuB,SAAS,KAAK;CAC9D,MAAM,CAAC,OAAO,YAAY,SAA4B,KAAA,CAAS;CAC/D,MAAM,CAAC,QAAQ,aAAa,SAA0B,OAAO;CAC7D,MAAM,CAAC,cAAc,mBAAmB,SAAS,KAAK;CACtD,MAAM,CAAC,kBAAkB,uBACvB,SAA2B,cAAc;CAC3C,MAAM,CAAC,mBAAmB,wBAAwB,SAAS,KAAK;CAChE,MAAM,CAAC,OAAO,YAAY,SAA+B,CAAC,CAAC;CAC3D,MAAM,CAAC,OAAO,YAAY,SAAwB,IAAI;CACtD,MAAM,CAAC,gBAAgB,qBAAqB,gBAEnC;EACP,YAAY;EACZ,mBAAmB;EACnB,iBAAiB;EACjB,UAAU;CACZ,EAAE;CAMF,MAAM,cAAc,OAClB,QAAQ,mBAAmB,CAAC,CAC9B;CACA,MAAM,kBAAkB,OAAO,IAAI;CACnC,MAAM,gBAAgB,OAAO,KAAK;CAClC,MAAM,kBAAkB,OAA0B,IAAI;CACtD,MAAM,yBAAyB,OAC7B,IACF;CACA,MAAM,qBAAqB,OAGjB,IAAI;CAGd,YAAY,UAAU;CAGtB,MAAM,aACJ,OACE,OACF;CACF,WAAW,UAAU;CAErB,MAAM,kBAAkB,aAAa,WAA8B;EACjE,IAAI,CAAC,QAAQ;EACb,SAAS,OAAO,gBAAgB,CAAC;EACjC,kBAAkB,OAAO,kBAAkB,CAAC;CAC9C,GAAG,CAAC,CAAC;CAGL,MAAM,EAAE,QAAQ,mBAAmB,cAAc;EAC/C,MAAM,gBAAgB,QAAQ,mBAAmB,CAAC;EAClD,gBAAgB,UAAU;EAM1B,MAAM,iBAAiB,WAAW;EAClC,MAAM,YAAY,eAAe,aAC7B,EAAE,YAAY,eAAe,WAAW,IACxC,EAAE,SAAS,eAAe,QAAQ;EAEtC,MAAM,iBAEF,EAAE,SAAS,KAAA,EAAU;EACzB,MAAM,0BAA0B;GAC9B,MAAM,kBAAkB,eAAe;GACvC,IAAI,CAAC,mBAAmB,gBAAgB,YAAY,iBAClD;GAEF,OAAO;EACT;EAKA,MAAM,sBAAsB;GAC1B,OAAO;GACP,WAAW,CAAC;EACd;EACA,MAAM,+BAA+B,aAAyB;GAC5D,IAAI,CAAC,oBAAoB,OAAO;IAC9B,oBAAoB,UAAU,KAAK,QAAQ;IAC3C;GACF;GACA,MAAM,kBAAkB,eAAe;GACvC,IAAI,CAAC,mBAAmB,gBAAgB,YAAY,iBAClD;GACF,SAAS;EACX;EACA,MAAM,WAAW,IAAI,WAA0C;GAC7D,uBAAuB;GACvB,GAAG;GACH,iBAAiB;GACjB,GAAI,OAAO,eAAe,aAAa,YACvC,eAAe,gBAAgB,OAC3B;IACE,aAAa;IACb,UAAU,eAAe;IACzB,GAAI,eAAe,YAAY,KAAA,KAAa,EAC1C,SAAS,eAAe,QAC1B;GACF,IACA,OAAO,eAAe,aAAa,YACjC,eAAe,cACf;IACE,aAAa,eAAe;IAC5B,UAAU,eAAe;GAC3B,IACA,EACE,GAAI,eAAe,aAAa,KAAA,KAAa,EAC3C,UAAU,eAAe,SAC3B,EACF;GACN,GAAI,eAAe,SAAS,KAAA,KAAa,EAAE,MAAM,eAAe,KAAK;GACrE,GAAI,eAAe,mBAAmB,KAAA,KAAa,EACjD,gBAAgB,eAAe,eACjC;GACA,GAAI,eAAe,SAAS,KAAA,KAAa,EAAE,MAAM,eAAe,KAAK;GACrE,oBAAoB,WAAW,QAAQ,eAAe;GACtD,GAAI,eAAe,0BAA0B,KAAA,KAAa,EACxD,uBAAuB,eAAe,sBACxC;GACA,GAAI,eAAe,YAAY,KAAA,KAAa,EAC1C,SAAS,eAAe,QAC1B;GACA,UAAU;IACR,GAAG,eAAe;IAClB,WAAW;IACX,UAAU;IACV,YAAY,eAAe,eAAe,eAAe;GAC3D;GACA,aAAa,aAAa;IAExB,IAAI,CAAC,kBAAkB,GAAG;IAC1B,OAAO,WAAW,QAAQ,aAAa,QAAQ;GACjD;GACA,UAAU,UAAuB;IAC/B,kCAAkC;KAChC,WAAW,QAAQ,UAAU,KAAK;IACpC,CAAC;GACH;GACA,WAAW,YAA+B;IACxC,kCAAkC;KAChC,WAAW,QAAQ,WAAW,OAAO;IACvC,CAAC;GACH;GACA,UAAU,UAAiB;IACzB,kCAAkC;KAChC,WAAW,QAAQ,UAAU,KAAK;IACpC,CAAC;GACH;GACA,GAAI,eAAe,UAAU,KAAA,KAAa,EACxC,OAAO,eAAe,MACxB;GACA,GAAI,eAAe,eAAe,KAAA,KAAa,EAC7C,YAAY,eAAe,WAC7B;GACA,gBAAgB,WAAW,MAAM,YAAY;IAC3C,kCAAkC;KAChC,WAAW,QAAQ,gBAAgB,WAAW,MAAM,OAAO;IAC7D,CAAC;GACH;GACA,GAAI,QAAQ,oBAAoB,KAAA,KAAa,EAC3C,iBAAiB,QAAQ,gBAC3B;GACA,mBAAmB,gBAA0C;IAC3D,kCAAkC;KAChC,YAAY,WAAW;KACvB,MAAM,kBAAkB,kBAAkB;KAC1C,IAAI,CAAC,iBAAiB;KACtB,oBAAoB,gBAAgB,oBAAoB,CAAC;IAC3D,CAAC;GACH;GACA,kBAAkB,iBAA0B;IAC1C,kCAAkC;KAChC,MAAM,kBAAkB,kBAAkB;KAC1C,IAAI,CAAC,iBAAiB;KACtB,aAAa,YAAY;KACzB,gBAAgB,eAAe;IACjC,CAAC;GACH;GACA,gBAAgB,aAAgC;IAC9C,kCAAkC;KAChC,SAAS,QAAQ;IACnB,CAAC;GACH;GACA,iBAAiB,WAA4B;IAC3C,kCAAkC;KAChC,UAAU,MAAM;IAClB,CAAC;GACH;GACA,uBAAuB,qBAA8B;IACnD,kCAAkC;KAChC,gBAAgB,gBAAgB;IAClC,CAAC;GACH;GACA,2BAA2B,eAAiC;IAC1D,kCAAkC;KAChC,oBAAoB,UAAU;IAChC,CAAC;GACH;GACA,4BAA4B,iBAA0B;IACpD,kCAAkC;KAChC,qBAAqB,YAAY;IACnC,CAAC;GACH;GACA,GAAI,WAAW,QAAQ,UAAU,KAAA,KAAa,EAC5C,OAAO,WAAW,QAAQ,MAC5B;GACA,gBAAgB,cAAoC;IAClD,kCAAkC;KAChC,SAAS,SAAS;IACpB,CAAC;GACH;GACA,gBAAgB,cAAc;IAC5B,kCAAkC;KAChC,SAAS,SAAS;IACpB,CAAC;GACH;GACA,sBAAsB,kBAAkB,0BAA0B;IAChE,kCAAkC;KAChC,mBAAmB,aAAa;MAC9B,GAAG;MACH,YAAY;MACZ,mBAAmB;KACrB,EAAE;IACJ,CAAC;GACH;GACA,yBAAyB,oBAAoB,YAAY;IACvD,kCAAkC;KAChC,kBAAkB,kBAAkB;KACpC,WAAW,QAAQ,yBACjB,oBACA,OACF;IACF,CAAC;GACH;EACF,CAAC;EACD,eAAe,UAAU;EACzB,OAAO;GAAE,QAAQ;GAAU,gBAAgB;EAAoB;CACjE,GAAG,CAAC,UAAU,eAAe,CAAC;CAE9B,gBAAgB;EACd,gBAAgB,UAAU;EAC1B,IAAI;GAGF,OAAO,eAAe,UAAU,SAAS,GAAG;IAC1C,IAAI,gBAAgB,YAAY,QAAQ;KACtC,eAAe,UAAU,SAAS;KAClC;IACF;IACA,eAAe,UAAU,MAAM,CAAC,GAAG;GACrC;EACF,UAAU;GAER,eAAe,QAAQ;EACzB;CACF,GAAG,CAAC,QAAQ,cAAc,CAAC;CAE3B,gBAAgB;EACd,MAAM,iBAAiB,OAAO,YAAY;EAC1C,IAAI,mBAAmB,YAAY,SACjC,YAAY,cAAc;EAE5B,oBAAoB,OAAO,oBAAoB,CAAC;CAClD,GAAG,CAAC,MAAM,CAAC;CAMX,gBAAgB;EACd,OAAO,cAAc,EAAE,MAAM,QAAQ,KAAK,CAAC;CAC7C,GAAG,CAAC,QAAQ,QAAQ,IAAI,CAAC;CAEzB,gBAAgB;EACd,IAAI,QAAQ,mBAAmB,KAAA,GAC7B,OAAO,cAAc,EAAE,gBAAgB,QAAQ,eAAe,CAAC;CAEnE,GAAG,CAAC,QAAQ,QAAQ,cAAc,CAAC;CAEnC,gBAAgB;EACd,IAAI,QAAQ,UAAU,KAAA,GACpB,OAAO,cAAc,EAAE,OAAO,QAAQ,MAAM,CAAC;CAEjD,GAAG,CAAC,QAAQ,QAAQ,KAAK,CAAC;CAE1B,gBAAgB;EACd,OAAO,cAAc,EAAE,SAAS,QAAQ,QAAQ,CAAC;CACnD,GAAG,CAAC,QAAQ,QAAQ,OAAO,CAAC;CAE5B,gBAAgB;EACd,IAAI,QAAQ,UAAU,KAAA,GACpB,OAAO,cAAc,EAAE,OAAO,QAAQ,MAAM,CAAC;CAEjD,GAAG,CAAC,QAAQ,QAAQ,KAAK,CAAC;CAE1B,gBAAgB;EACd,IAAI,QAAQ,MAAM;GAChB,OAAO,UAAU;GACjB,cAAc,UAAU;EAC1B,OAAO,IAAI,cAAc,SAAS;GAQhC,OAAO,YAAY;GACnB,cAAc,UAAU;EAC1B;CACF,GAAG,CAAC,QAAQ,QAAQ,IAAI,CAAC;CAezB,gBAAgB;EACd,OAAO,OAAO;EACd,aAAa;GACX,OAAO,OAAO;EAChB;CACF,GAAG,CAAC,MAAM,CAAC;CAEX,gBAAgB;EACd,IAAI,mBAAmB,SAAS,WAAW,QAAQ;GACjD,aAAa,mBAAmB,QAAQ,OAAO;GAC/C,mBAAmB,UAAU;EAC/B;EACA,IAAI,uBAAuB,SAAS;GAClC,aAAa,uBAAuB,OAAO;GAC3C,uBAAuB,UAAU;EACnC;EACA,OAAO,cAAc;EAIrB,gBAAgB,MAAM;EAEtB,aAAa;GACX,uBAAuB,UAAU,iBAAiB;IAChD,IAAI,gBAAgB,YAAY,QAC9B,gBAAgB,UAAU;IAE5B,uBAAuB,UAAU;GACnC,GAAG,CAAC;GASJ,MAAM,WAAW;IACf;IACA,SAAS,iBAAiB;KACxB,IAAI,WAAW,QAAQ,MACrB,OAAO,YAAY;UAEnB,OAAO,KAAK;KAEd,OAAO,QAAQ;KACf,IAAI,mBAAmB,YAAY,UACjC,mBAAmB,UAAU;IAEjC,GAAG,CAAC;GACN;GACA,mBAAmB,UAAU;EAC/B;CACF,GAAG,CAAC,QAAQ,eAAe,CAAC;CAE5B,MAAM,cAAc,YAClB,OACE,SACA,gBACG;EACH,IAAI;GACF,MAAM,OAAO,YAAY,SAAS,KAAA,GAAW,WAAW;EAC1D,UAAU;GACR,gBAAgB,MAAM;EACxB;CACF,GACA,CAAC,QAAQ,eAAe,CAC1B;CAEA,MAAM,eAAe,aAClB,OAAe;EACd,OAAO,aAAa,EAAE;CACxB,GACA,CAAC,QAAQ,eAAe,CAC1B;CAEA,MAAM,SAAS,YACb,OAAO,YAAsC;EAC3C,IAAI;GACF,MAAM,OAAO,OAAO,OAAO;EAC7B,UAAU;GACR,gBAAgB,MAAM;EACxB;CACF,GACA,CAAC,QAAQ,eAAe,CAC1B;CAEA,MAAM,SAAS,YAAY,YAAY;EACrC,IAAI;GACF,MAAM,OAAO,OAAO;EACtB,UAAU;GACR,gBAAgB,MAAM;EACxB;CACF,GAAG,CAAC,QAAQ,eAAe,CAAC;CAE5B,MAAM,oBAAoB,YAAY,YAAY;EAChD,MAAM,OAAO,kBAAkB;EAC/B,oBAAoB,OAAO,oBAAoB,CAAC;CAClD,GAAG,CAAC,MAAM,CAAC;CAEX,MAAM,OAAO,kBAAkB;EAC7B,OAAO,KAAK;CACd,GAAG,CAAC,MAAM,CAAC;CAEX,MAAM,QAAQ,kBAAkB;EAC9B,OAAO,MAAM;EACb,gBAAgB,MAAM;CACxB,GAAG,CAAC,QAAQ,eAAe,CAAC;CAE5B,MAAM,sBAAsB,aACzB,gBAA0C;EACzC,OAAO,oBAAoB,WAAW;CACxC,GACA,CAAC,MAAM,CACT;CAEA,MAAM,gBAAgB,YACpB,OAAO,WAMD;EACJ,MAAM,OAAO,cAAc,MAAM;CACnC,GACA,CAAC,MAAM,CACT;CAEA,MAAM,0BAA0B,YAC9B,OAAO,aAAgD;EACrD,MAAM,OAAO,wBAAwB,QAAQ;EAC7C,gBAAgB,MAAM;CACxB,GACA,CAAC,QAAQ,eAAe,CAC1B;CAEA,MAAM,mBAAmB,YACvB,OAAO,aAAwC,UAA4B;EACzE,MAAM,SAAS,MAAM,OAAO,iBAC1B,aACA,SAAS,KAAA,CACX;EACA,gBAAgB,MAAM;EACtB,OAAO;CACT,GACA,CAAC,QAAQ,eAAe,CAC1B;CAEA,MAAM,oBAAoB,aAEtB,eAKG;EACH,IAAI,OAAO,eAAe,WACxB,OAAO,kBAAkB,UAAU;OAEnC,OAAO,kBAAkB,UAAU;CAEvC,GACA,CAAC,MAAM,CACT;CAEA,MAAM,mBAAmB,kBAAkB;EACzC,OAAO,iBAAiB;CAC1B,GAAG,CAAC,MAAM,CAAC;CAEX,MAAM,kBAAkB,kBAAkB;EACxC,OAAO,gBAAgB;CACzB,GAAG,CAAC,MAAM,CAAC;CAEX,MAAM,yBAAyB,aAC5B,aAAwC,UACvC,OAAO,uBAAuB,aAAa,KAAK,GAClD,CAAC,MAAM,CACT;CAYA,MAAM,mBAAmB,OAAO,YAAY;CAE5C,MAAM,uBAAuB,cAA2C;EACtE,IAAI,gBAAgB;EACpB,KAAK,IAAI,IAAI,iBAAiB,SAAS,GAAG,KAAK,GAAG,KAChD,IAAI,iBAAiB,EAAE,EAAE,SAAS,QAAQ;GACxC,gBAAgB;GAChB;EACF;EAEF,IAAI,kBAAkB,IAAI,OAAO;EACjC,KAAK,IAAI,IAAI,iBAAiB,SAAS,GAAG,IAAI,eAAe,KAAK;GAChE,MAAM,IAAI,iBAAiB;GAC3B,IAAI,GAAG,SAAS,aAAa;GAC7B,MAAM,OAAO,EAAE,MAAM,MAClB,MAAiC,EAAE,SAAS,mBAC/C;GACA,IAAI,MAAM,OAAO;EACnB;EACA,OAAO;CACT,GAAG,CAAC,gBAAgB,CAAC;CAErB,MAAM,UAAU,cAAuB;EACrC,IAAI,CAAC,sBAAsB,OAAO,CAAC;EAEnC,OADU,qBAAqB,WAAW,qBAAqB,QAClD,CAAC;CAChB,GAAG,CAAC,oBAAoB,CAAC;CAEzB,MAAM,QAAQ,cAA4B;EACxC,IAAI,CAAC,wBAAwB,qBAAqB,WAAW,YAC3D,OAAO;EAET,OAAO,qBAAqB;CAC9B,GAAG,CAAC,oBAAoB,CAAC;CAQzB,OAAO;EACL,UAAU;EACV,WARgB,cAAc,OAAO,aAAa,GAAG,CAAC,gBAAgB,CAQtE;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA,aAAa;EACb;EACA;EACA;EACA;EACA;EACA;EACA,YAAY,eAAe;EAC3B,mBAAmB,eAAe;EAClC,iBAAiB,eAAe;EAChC,UAAU,eAAe;EACzB;EACA;EACA;EACA;EACA;EACA;EACA;CACF;AACF"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tanstack/ai-react",
-  "version": "0.27.3",
+  "version": "0.28.0",
   "description": "React hooks for TanStack AI streaming chat, realtime voice, structured outputs, and media generation.",
   "author": "Tanner Linsley",
   "license": "MIT",
@@ -58,14 +58,14 @@
   ],
   "dependencies": {
     "@tanstack/markdown": "^0.0.13",
-    "@tanstack/ai-client": "^0.33.2"
+    "@tanstack/ai-client": "^0.34.0"
   },
   "peerDependencies": {
     "@mcp-ui/client": "^7",
     "@types/react": ">=18.0.0",
     "react": ">=18.0.0",
     "react-dom": ">=18.0.0",
-    "@tanstack/ai": "^0.58.0"
+    "@tanstack/ai": "^0.59.0"
   },
   "peerDependenciesMeta": {
     "@mcp-ui/client": {
@@ -85,7 +85,7 @@
     "react": "^19.2.3",
     "react-dom": "^19.2.3",
     "vite": "^8.2.1",
-    "@tanstack/ai": "0.58.0"
+    "@tanstack/ai": "0.59.0"
   },
   "scripts": {
     "clean": "premove ./build ./dist",

package/src/chat-ui/create-ui.tsx

@@ -1,4 +1,4 @@
-import { createContext, memo, useContext } from 'react'
+import { createContext, memo, useContext, useMemo } from 'react'
 import type { ComponentProps, ComponentType, Context, ReactNode } from 'react'
 import {
   automaticPartsForMessage,
@@ -8,8 +8,10 @@ import {
   selectMessageUI,
 } from '@tanstack/ai-client/ui'
 import type {
+  ChatUIApprovalToolName,
   ChatUIData,
   ChatUIHasNamedInterrupts,
+  ChatUIHasNamedSubagents,
   ChatUIHasNamedTools,
   ChatUIInterrupt,
   ChatUIInterruptName,
@@ -22,6 +24,9 @@ import type {
   ChatUISchemaOf,
   ChatUISelectedPart,
   ChatUISelectedPartOf,
+  ChatUISubagentName,
+  ChatUISubagentOf,
+  ChatUISubagentOptions,
   ChatUIToolApproval,
   ChatUIToolName,
   ChatUIToolsOf,
@@ -29,6 +34,8 @@ import type {
 import type {
   MessagePart,
   QueuedMessage,
+  SubagentHandle,
+  SubagentHandles,
   ToolCallPart,
   ToolResultPart,
   UIMessage,
@@ -56,6 +63,7 @@ export type LayoutProps<
   Messages: ComponentType
   Interrupts: ComponentType
   Queue: ComponentType
+  Subagents: ComponentType
   readonly __ui?: TOptions
 } & (TInput extends ComponentType<any> ? { Input: ComponentType } : {})
 
@@ -77,6 +85,38 @@ export type PartProps<TOptions, TKey extends ChatUIPartKey = ChatUIPartKey> = {
   part: ChatUIPartOf<TOptions, TKey>
 }
 
+/**
+ * Widgets for one subagent card. Each entry replaces the enclosing entry of
+ * the same key: the root map at top level, the outer card's map when nested.
+ * It applies to this card and the children nested in it. An unset key keeps
+ * the enclosing entry. Tool names and tool props come from that agent's
+ * `tools`.
+ */
+export type SubagentPartsProps<
+  TOptions,
+  TName extends ChatUISubagentName<TOptions> = ChatUISubagentName<TOptions>,
+> = {
+  partsComponents?: ChatUIPartsComponents<
+    ChatUISubagentOptions<TOptions, TName>
+  >
+  toolsComponents?: {
+    [K in ChatUIToolName<
+      ChatUISubagentOptions<TOptions, TName>
+    >]?: ComponentType<ToolProps<ChatUISubagentOptions<TOptions, TName>, K>>
+  }
+}
+
+export type SubagentProps<
+  TOptions,
+  TName extends ChatUISubagentName<TOptions> = ChatUISubagentName<TOptions>,
+> = {
+  subagent: [ChatUISubagentOf<TOptions, TName>] extends [never]
+    ? SubagentHandle & { name: TName }
+    : SubagentHandles<ReadonlyArray<ChatUISubagentOf<TOptions, TName>>>
+  Parts: ComponentType<SubagentPartsProps<TOptions, TName>>
+  readonly __ui?: TOptions
+}
+
 export type ToolProps<
   TOptions,
   TName extends ChatUIToolName<TOptions> = ChatUIToolName<TOptions>,
@@ -108,7 +148,7 @@ type GenericInterruptComponents<TOptions> =
       }
 
 type ToolApprovalMap<TOptions> = {
-  [K in ChatUIToolName<TOptions>]?: ComponentType<
+  [K in ChatUIApprovalToolName<TOptions>]?: ComponentType<
     InterruptProps<TOptions, K & ChatUIInterruptName<TOptions>>
   >
 }
@@ -161,6 +201,21 @@ export type ChatUIComponents<
           tools?: ToolApprovalMap<TOptions>
           generic?: GenericInterruptComponents<TOptions>
         }
+      }) &
+  (ChatUIHasNamedSubagents<TOptions> extends true
+    ? {
+        subagentsComponents: {
+          [K in ChatUISubagentName<TOptions>]: ComponentType<
+            SubagentProps<TOptions, K>
+          >
+        }
+      }
+    : {
+        subagentsComponents?: {
+          [K in ChatUISubagentName<TOptions>]?: ComponentType<
+            SubagentProps<TOptions, K>
+          >
+        }
       })
 
 /** Scoped contexts, for widgets in other files or nested chat trees. */
@@ -191,7 +246,7 @@ type PartMixins<TOptions> = {
 type InterruptMixins<TOptions> = {
   [K in ChatUINamedInterruptId<TOptions>]: BoundWidget
 } & {
-  [K in ChatUIToolName<TOptions>]?: BoundWidget
+  [K in ChatUIApprovalToolName<TOptions>]?: BoundWidget
 } & {
   fallback?: BoundWidget
   Render: BoundWidget
@@ -227,6 +282,32 @@ function queueItemEqual(
   )
 }
 
+// The handle is one live object that the client updates in place, so a row
+// gets its changing card fields as separate props. Messages are not one of
+// them: `SubagentMessages` reads them from the chat context, so new child
+// text shows without a row render.
+type SubagentRowProps = {
+  handle: SubagentHandle
+  status: SubagentHandle['status']
+  error: SubagentHandle['error']
+}
+
+function subagentListItemEqual(prev: SubagentRowProps, next: SubagentRowProps) {
+  return (
+    prev.handle.id === next.handle.id &&
+    prev.handle.stop === next.handle.stop &&
+    prev.status === next.status &&
+    prev.error === next.error
+  )
+}
+
+function subagentMessagesEqual(
+  prev: { messages: SubagentHandle['messages'] },
+  next: { messages: SubagentHandle['messages'] },
+) {
+  return prev.messages === next.messages
+}
+
 function selectedPartPropsEqual(
   prev: { selected: ChatUISelectedPart },
   next: { selected: ChatUISelectedPart },
@@ -254,6 +335,18 @@ function isSelectedPart(
   return 'key' in value && 'part' in value
 }
 
+/** `base` with each set entry of `overrides`. An unset entry keeps `base`. */
+function withWidgets(
+  base: Record<string, ComponentType<any> | undefined>,
+  overrides: Record<string, ComponentType<any> | undefined> | undefined,
+) {
+  const out = { ...base }
+  for (const [key, component] of Object.entries(overrides ?? {})) {
+    if (component) out[key] = component
+  }
+  return out
+}
+
 function bindMap(
   map: Record<string, ComponentType<any> | undefined> | undefined,
   bind: (component: ComponentType<any>) => BoundWidget,
@@ -288,12 +381,14 @@ export function createChatUI<
     partsComponents: parts,
     toolsComponents: tools,
     interruptsComponents: interrupts,
+    subagentsComponents: subagentComponents,
   } = config as ChatUIFactoryConfig<TOptions, TInput> & {
     toolsComponents?: Record<string, ComponentType<any> | undefined>
     interruptsComponents?: {
       tools?: Record<string, ComponentType<any> | undefined>
       generic?: Record<string, ComponentType<any> | undefined>
     }
+    subagentsComponents?: Record<string, ComponentType<any> | undefined>
   }
   const {
     layout: Layout,
@@ -445,6 +540,7 @@ export function createChatUI<
     Messages: Messages as ComponentType,
     Interrupts: Interrupts as ComponentType,
     Queue: Queue as ComponentType,
+    Subagents: Subagents as ComponentType,
     Input: (InputComponent ?? MissingInput) as ComponentType,
   }
 
@@ -525,6 +621,19 @@ export function createChatUI<
     inlineToolNames: ReadonlyArray<string>
   }
   const MessageRenderContext = createContext<MessageRenderValue | null>(null)
+  const SubagentRenderContext = createContext<{
+    handle: SubagentHandle
+    /** A `Subagents` row. Its messages come from the live chat state. */
+    listed?: true
+  } | null>(null)
+  // The widgets that dispatch uses. A subagent card's `Parts` can replace
+  // entries for its subtree. Outside a card these are the factory maps.
+  type WidgetMap = Record<string, ComponentType<any> | undefined>
+  const rootWidgets: { parts: WidgetMap; tools: WidgetMap } = {
+    parts: parts as WidgetMap,
+    tools: tools ?? {},
+  }
+  const WidgetsContext = createContext(rootWidgets)
 
   function Parts() {
     const scope = useContext(MessageRenderContext)
@@ -588,7 +697,7 @@ export function createChatUI<
     interrupts,
     inlineToolNames: names,
   }: {
-    message: ChatUIMessages<TOptions>[number]
+    message: UIMessage
     interrupts: ReadonlyArray<ChatUIInterrupt>
     inlineToolNames: ReadonlyArray<string>
   }) {
@@ -610,9 +719,10 @@ export function createChatUI<
   }: {
     selected: ChatUISelectedPart
   }) {
+    const widgets = useContext(WidgetsContext)
     if (selected.key === 'toolCall') {
       const name = selected.part.name
-      const Tool = tools?.[name as ChatUIToolName<TOptions>] as
+      const Tool = widgets.tools[name] as
         | ComponentType<ToolProps<TOptions>>
         | undefined
       if (!Tool) {
@@ -628,9 +738,30 @@ export function createChatUI<
       )
     }
 
-    const PartComponent = (parts[selected.key] ?? parts.fallback) as
-      | ComponentType<PartProps<TOptions>>
-      | undefined
+    if (selected.key === 'subagent' && selected.part.type === 'subagent') {
+      const name = selected.part.subagent.name
+      const Subagent = subagentComponents?.[name] as
+        | ComponentType<SubagentProps<TOptions>>
+        | undefined
+      if (!Subagent) {
+        throw new Error(`[tanstack-ai-ui] Missing subagentsComponents.${name}`)
+      }
+      return (
+        <SubagentRenderContext.Provider
+          value={{ handle: selected.part.subagent }}
+        >
+          <Subagent
+            Parts={SubagentMessages}
+            subagent={
+              selected.part.subagent as SubagentProps<TOptions>['subagent']
+            }
+          />
+        </SubagentRenderContext.Provider>
+      )
+    }
+
+    const PartComponent = (widgets.parts[selected.key] ??
+      widgets.parts.fallback) as ComponentType<PartProps<TOptions>> | undefined
     if (!PartComponent) {
       warn(
         `part:${selected.key}`,
@@ -746,6 +877,115 @@ export function createChatUI<
     )
   })
 
+  function readSubagentMessages() {
+    const scoped = useContext(SubagentRenderContext)
+    const chat = useChatContext()
+    if (scoped?.listed) {
+      const live = chat.subagents.find((item) => item.id === scoped.handle.id)
+      return (live ?? scoped.handle).messages
+    }
+    if (scoped) return scoped.handle.messages
+    const selected = useContext(PartContext)
+    if (selected?.key === 'subagent' && selected.part.type === 'subagent') {
+      return selected.part.subagent.messages
+    }
+    throw new Error(
+      '`Parts` must be rendered by a subagent component or `Subagents` item.',
+    )
+  }
+
+  const SubagentMessagesBody = memo(function SubagentMessagesBody({
+    messages,
+  }: {
+    messages: SubagentHandle['messages']
+  }) {
+    const chat = useChatContext()
+    const interrupts = readInterrupts(chat)
+    return (
+      <>
+        {messages.map((message) => (
+          <AutomaticParts
+            key={message.id}
+            inlineToolNames={inlineToolNames}
+            interrupts={interrupts}
+            message={message}
+          />
+        ))}
+      </>
+    )
+  }, subagentMessagesEqual)
+
+  function SubagentMessages({
+    partsComponents,
+    toolsComponents,
+  }: SubagentPartsProps<TOptions>) {
+    const messages = readSubagentMessages()
+    const outer = useContext(WidgetsContext)
+    const widgets = useMemo(
+      () =>
+        partsComponents || toolsComponents
+          ? {
+              parts: withWidgets(outer.parts, partsComponents),
+              tools: withWidgets(outer.tools, toolsComponents),
+            }
+          : outer,
+      [outer, partsComponents, toolsComponents],
+    )
+    // ponytail: root `<Interrupts />` still lists an approval for a tool call
+    // inside a card, so it shows in both places. Hiding it needs `selectChatUI`
+    // to collect tool-call ids from nested subagent messages (its `callIds` is
+    // top-level only) plus the card tool names in `inlineToolNames`.
+    return (
+      <WidgetsContext.Provider value={widgets}>
+        <SubagentMessagesBody messages={messages} />
+      </WidgetsContext.Provider>
+    )
+  }
+
+  const SubagentListItem = memo(function SubagentListItem({
+    handle,
+  }: SubagentRowProps) {
+    const Subagent = subagentComponents?.[handle.name] as
+      | ComponentType<SubagentProps<TOptions>>
+      | undefined
+    if (!Subagent) {
+      throw new Error(
+        `[tanstack-ai-ui] Missing subagentsComponents.${handle.name}`,
+      )
+    }
+    return (
+      <SubagentRenderContext.Provider value={{ handle, listed: true }}>
+        <Subagent
+          Parts={SubagentMessages}
+          subagent={handle as SubagentProps<TOptions>['subagent']}
+        />
+      </SubagentRenderContext.Provider>
+    )
+  }, subagentListItemEqual)
+
+  function Subagents({
+    children,
+  }: {
+    children?: (subagents: Array<SubagentHandle>) => ReactNode
+  } = {}) {
+    const chat = useChatContext()
+    const live = chat.subagents
+    if (children) return <>{children(live)}</>
+    if (live.length === 0) return null
+    return (
+      <>
+        {live.map((handle) => (
+          <SubagentListItem
+            key={handle.id}
+            handle={handle}
+            status={handle.status}
+            error={handle.error}
+          />
+        ))}
+      </>
+    )
+  }
+
   return {
     Chat,
     Provider,
@@ -755,6 +995,8 @@ export function createChatUI<
     Interrupts,
     Interrupt,
     Queue,
+    Subagents,
+    SubagentMessages,
     useChatContext,
     Input: InputComponent,
   }

package/src/create-chat-hook.ts

@@ -12,6 +12,7 @@ type ChatHookOverrides<
   TSchema extends SchemaInput | undefined,
   TContext,
   TInterrupts extends ReadonlyArray<InterruptDefinition<any, any, any, any>>,
+  TSubagents extends ReadonlyArray<{ name: string }> | undefined,
 > = {
   threadId?: string
   live?: boolean
@@ -21,7 +22,8 @@ type ChatHookOverrides<
     TTools,
     TSchema,
     TContext,
-    TInterrupts
+    TInterrupts,
+    TSubagents
   >['initialMessages']
 }
 
@@ -53,9 +55,17 @@ export function createChatHook<
   const TInterrupts extends ReadonlyArray<
     InterruptDefinition<any, any, any, any>
   > = readonly [],
->(options: UseChatOptions<TTools, TSchema, TContext, TInterrupts>) {
+  const TSubagents extends ReadonlyArray<{ name: string }> | undefined =
+    undefined,
+>(options: UseChatOptions<TTools, TSchema, TContext, TInterrupts, TSubagents>) {
   function useChat(
-    overrides?: ChatHookOverrides<TTools, TSchema, TContext, TInterrupts>,
+    overrides?: ChatHookOverrides<
+      TTools,
+      TSchema,
+      TContext,
+      TInterrupts,
+      TSubagents
+    >,
   ) {
     if (!overrides) {
       return useUnboundChat(options)