@tambo-ai/react 1.1.0 → 1.2.1

package/dist/v1/hooks/use-tambo-v1.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-tambo-v1.js","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1.ts"],"names":[],"mappings":";AAAA,YAAY,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAiNb,4BAqOC;AApbD;;;;;GAKG;AAEH,sCAAwC;AAExC,+CAMe;AACf,iFAG+C;AAC/C,sEAAgE;AAChE,uEAA8D;AAE9D,qFAGiD;AACjD,+EAAwE;AACxE,kFAK8C;AAQ9C,kEAGoC;AA+JpC;;;GAGG;AACH,SAAgB,QAAQ;IACtB,MAAM,MAAM,GAAG,IAAA,sCAAc,GAAE,CAAC;IAChC,MAAM,WAAW,GAAG,IAAA,2CAAmB,GAAE,CAAC;IAC1C,MAAM,EAAE,OAAO,EAAE,GAAG,IAAA,kCAAc,GAAE,CAAC;IACrC,MAAM,WAAW,GAAG,IAAA,wCAAc,GAAE,CAAC;IACrC,MAAM,QAAQ,GAAG,IAAA,2CAAiB,GAAE,CAAC;IACrC,MAAM,QAAQ,GAAG,IAAA,kBAAU,EAAC,8CAAoB,CAAC,CAAC;IAClD,MAAM,gBAAgB,GAAG,IAAA,6CAAmB,GAAE,CAAC;IAC/C,MAAM,SAAS,GAAG,IAAA,2CAAiB,GAAE,CAAC;IAEtC,8EAA8E;IAC9E,4CAA4C;IAC5C,MAAM,iBAAiB,GAAG,IAAA,cAAM,EAAmC,IAAI,GAAG,EAAE,CAAC,CAAC;IAE9E,0CAA0C;IAC1C,MAAM,WAAW,GAAG,WAAW,CAAC,SAAS,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;IAEvE,8EAA8E;IAC9E,0CAA0C;IAC1C,MAAM,YAAY,GAAG,IAAA,cAAM,EAAC,WAAW,CAAC,SAAS,CAAC,CAAC;IACnD,YAAY,CAAC,OAAO,GAAG,WAAW,CAAC,SAAS,CAAC;IAE7C,wCAAwC;IACxC,MAAM,SAAS,GAAG,IAAA,mBAAW,EAAC,KAAK,IAAI,EAAE;QACvC,MAAM,KAAK,GAAG,WAAW,EAAE,SAAS,CAAC,KAAK,CAAC;QAC3C,MAAM,QAAQ,GAAG,WAAW,CAAC,eAAe,CAAC;QAE7C,4DAA4D;QAC5D,IAAI,CAAC,KAAK,IAAI,IAAA,yCAAqB,EAAC,QAAQ,CAAC,EAAE,CAAC;YAC9C,OAAO;QACT,CAAC;QAED,yDAAyD;QACzD,QAAQ,CAAC;YACP,IAAI,EAAE,OAAO;YACb,QAAQ;YACR,KAAK,EAAE;gBACL,IAAI,EAAE,gBAAS,CAAC,SAAS;gBACzB,OAAO,EAAE,eAAe;gBACxB,IAAI,EAAE,WAAW;gBACjB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB;SACF,CAAC,CAAC;QAEH,6BAA6B;QAC7B,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;QACjE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,yDAAyD;YACzD,OAAO,CAAC,IAAI,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;QACzD,CAAC;IACH,CAAC,EAAE;QACD,MAAM;QACN,OAAO;QACP,WAAW,CAAC,eAAe;QAC3B,WAAW,EAAE,SAAS,CAAC,KAAK;QAC5B,QAAQ;KACT,CAAC,CAAC;IAEH,yBAAyB;IACzB,MAAM,gBAAgB,GAAG,IAAA,mBAAW,EAClC,KAAK,EAAE,QAAgB,EAAE,IAAY,EAAE,EAAE;QACvC,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAEzD,IAAI,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YACnC,QAAQ,CAAC;gBACP,IAAI,EAAE,oBAAoB;gBAC1B,QAAQ;gBACR,IAAI,EAAE,IAAI;aACX,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC;YACH,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChB,WAAW,CAAC,iBAAiB,CAAC;oBAC5B,QAAQ,EAAE,CAAC,YAAY,EAAE,MAAM,CAAC;iBACjC,CAAC;gBACF,WAAW,CAAC,iBAAiB,CAAC;oBAC5B,QAAQ,EAAE,CAAC,YAAY,EAAE,QAAQ,CAAC;iBACnC,CAAC;aACH,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,IAAI,CACV,8DAA8D,EAC9D,KAAK,CACN,CAAC;QACJ,CAAC;IACH,CAAC,EACD,CAAC,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,CAAC,CACzC,CAAC;IAEF,8DAA8D;IAC9D,OAAO,IAAA,eAAO,EAAC,GAAG,EAAE;QAClB,MAAM,MAAM,GAAG,WAAW,CAAC;QAC3B,MAAM,WAAW,GAAG,MAAM,EAAE,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC;QAClD,MAAM,cAAc,GAAmB,MAAM,EAAE,SAAS,IAAI;YAC1D,MAAM,EAAE,MAAe;SACxB,CAAC;QAEF,gFAAgF;QAChF,wFAAwF;QACxF,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAU,CAAC;QAC3C,KAAK,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;YAC9B,KAAK,MAAM,OAAO,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;gBAClC,IAAI,OAAO,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;oBACnC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBAC1C,CAAC;YACH,CAAC;QACH,CAAC;QAED,kEAAkE;QAClE,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,OAAO,EAAsB,EAAE;YAC/D,MAAM,kBAAkB,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,EAAW,EAAE;gBAClE,mDAAmD;gBACnD,IAAI,OAAO,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAChC,MAAM,YAAY,GAAG,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;oBACtD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC;oBAElC,yCAAyC;oBACzC,MAAM,iBAAiB,GAA0B;wBAC/C,oBAAoB,EAClB,OAAO,KAAK,CAAC,oBAAoB,KAAK,QAAQ;4BAC5C,CAAC,CAAC,KAAK,CAAC,oBAAoB;4BAC5B,CAAC,CAAC,SAAS;wBACf,8BAA8B,EAC5B,OAAO,KAAK,CAAC,8BAA8B,KAAK,QAAQ;4BACtD,CAAC,CAAC,KAAK,CAAC,8BAA8B;4BACtC,CAAC,CAAC,SAAS;qBAChB,CAAC;oBAEF,mDAAmD;oBACnD,MAAM,aAAa,GAAG,YAAY;wBAChC,CAAC,CAAC,CAAC,iBAAiB,CAAC,8BAA8B;4BACjD,UAAU,OAAO,CAAC,IAAI,EAAE,CAAC;wBAC3B,CAAC,CAAC,CAAC,iBAAiB,CAAC,oBAAoB;4BACvC,WAAW,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;oBAE/B,oFAAoF;oBACpF,MAAM,UAAU,GAA4B,EAAE,CAAC;oBAC/C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;wBACjD,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;4BAC/B,UAAU,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;wBAC1B,CAAC;oBACH,CAAC;oBAED,MAAM,SAAS,GAAwB;wBACrC,GAAG,OAAO;wBACV,KAAK,EAAE,UAAU;wBACjB,YAAY;wBACZ,aAAa;wBACb,iBAAiB;qBAClB,CAAC;oBACF,OAAO,SAAS,CAAC;gBACnB,CAAC;gBAED,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBACjC,OAAO,OAAO,CAAC;gBACjB,CAAC;gBAED,MAAM,gBAAgB,GAAG,OAAO,CAAC;gBACjC,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;gBAC/D,MAAM,KAAK,GAAG,iBAAiB,CAAC,OAAO,CAAC;gBACxC,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;gBAE9C,iDAAiD;gBACjD,IAAI,MAAM,EAAE,SAAS,KAAK,SAAS,EAAE,CAAC;oBACpC,OAAO;wBACL,GAAG,gBAAgB;wBACnB,iBAAiB,EAAE,MAAM,CAAC,OAAO;qBAClC,CAAC;gBACJ,CAAC;gBAED,6BAA6B;gBAC7B,MAAM,OAAO,GAAG,eAAK,CAAC,aAAa,CAAC,yCAAiB,EAAE;oBACrD,GAAG,EAAE,gBAAgB,CAAC,EAAE;oBACxB,OAAO,EAAE,gBAAgB;oBACzB,QAAQ,EAAE,WAAW,CAAC,eAAe;oBACrC,SAAS,EAAE,OAAO,CAAC,EAAE;iBACtB,CAAC,CAAC;gBAEH,eAAe;gBACf,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;gBAEvD,OAAO;oBACL,GAAG,gBAAgB;oBACnB,iBAAiB,EAAE,OAAO;iBAC3B,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,OAAO;gBACL,GAAG,OAAO;gBACV,OAAO,EAAE,kBAAkB;aAC5B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,OAAO;YACL,MAAM;YACN,MAAM;YACN,QAAQ;YACR,cAAc;YACd,WAAW,EAAE,cAAc,CAAC,MAAM,KAAK,WAAW;YAClD,SAAS,EAAE,cAAc,CAAC,MAAM,KAAK,SAAS;YAC9C,MAAM,EAAE,cAAc,CAAC,MAAM,KAAK,MAAM;YACxC,iBAAiB,EAAE,QAAQ,CAAC,iBAAiB;YAC7C,YAAY,EAAE,QAAQ,CAAC,YAAY;YACnC,aAAa,EAAE,QAAQ,CAAC,aAAa;YACrC,aAAa,EAAE,QAAQ,CAAC,aAAa;YACrC,YAAY,EAAE,QAAQ,CAAC,YAAY;YACnC,eAAe,EAAE,WAAW,CAAC,eAAe;YAC5C,UAAU,EAAE,gBAAgB,CAAC,UAAU;YACvC,YAAY,EAAE,gBAAgB,CAAC,YAAY;YAC3C,cAAc,EAAE,gBAAgB,CAAC,cAAc;YAC/C,QAAQ;YACR,SAAS;YACT,SAAS;YACT,YAAY,EAAE,SAAS,CAAC,MAAM,KAAK,YAAY;YAC/C,gBAAgB;SACjB,CAAC;IACJ,CAAC,EAAE;QACD,SAAS;QACT,gBAAgB;QAChB,MAAM;QACN,WAAW;QACX,QAAQ;QACR,WAAW,CAAC,eAAe;QAC3B,gBAAgB;QAChB,QAAQ;QACR,SAAS;KACV,CAAC,CAAC;AACL,CAAC","sourcesContent":["\"use client\";\n\n/**\n * useTambo - Main Hook\n *\n * Combines all contexts into a single hook for convenient access\n * to thread state, streaming status, registry, and client.\n */\n\nimport { EventType } from \"@ag-ui/core\";\nimport type TamboAI from \"@tambo-ai/typescript-sdk\";\nimport React, {\n  useCallback,\n  useContext,\n  useMemo,\n  useRef,\n  type ReactElement,\n} from \"react\";\nimport {\n  useTamboClient,\n  useTamboQueryClient,\n} from \"../../providers/tambo-client-provider\";\nimport { useTamboConfig } from \"../providers/tambo-v1-provider\";\nimport { useTamboAuthState } from \"./use-tambo-v1-auth-state\";\nimport type { TamboAuthState } from \"../types/auth\";\nimport {\n  TamboRegistryContext,\n  type TamboRegistryContext as TamboRegistryContextType,\n} from \"../../providers/tambo-registry-provider\";\nimport { ComponentRenderer } from \"../components/v1-component-renderer\";\nimport {\n  useStreamDispatch,\n  useStreamState,\n  useThreadManagement,\n  type ThreadManagement,\n} from \"../providers/tambo-v1-stream-context\";\nimport type {\n  Content,\n  TamboToolDisplayProps,\n  TamboThreadMessage,\n  TamboToolUseContent,\n} from \"../types/message\";\nimport type { StreamingState } from \"../types/thread\";\nimport {\n  isPlaceholderThreadId,\n  type ThreadState,\n} from \"../utils/event-accumulator\";\n\n/**\n * Return type for useTambo hook\n */\nexport interface UseTamboReturn {\n  /**\n   * The Tambo API client instance\n   */\n  client: TamboAI;\n\n  /**\n   * Current thread state for the given threadId, or undefined if not loaded\n   */\n  thread: ThreadState | undefined;\n\n  /**\n   * Messages in the current thread\n   */\n  messages: TamboThreadMessage[];\n\n  /**\n   * Current streaming state\n   */\n  streamingState: StreamingState;\n\n  /**\n   * Whether the thread is currently streaming a response\n   */\n  isStreaming: boolean;\n\n  /**\n   * Whether the thread is waiting for the AI to start responding\n   */\n  isWaiting: boolean;\n\n  /**\n   * Whether the thread is idle (not streaming or waiting)\n   */\n  isIdle: boolean;\n\n  /**\n   * Register a component with the registry\n   */\n  registerComponent: TamboRegistryContextType[\"registerComponent\"];\n\n  /**\n   * Register a tool with the registry\n   */\n  registerTool: TamboRegistryContextType[\"registerTool\"];\n\n  /**\n   * Register multiple tools with the registry\n   */\n  registerTools: TamboRegistryContextType[\"registerTools\"];\n\n  /**\n   * The component registry (Map of name -> component definition)\n   */\n  componentList: TamboRegistryContextType[\"componentList\"];\n\n  /**\n   * The tool registry (Map of name -> tool definition)\n   */\n  toolRegistry: TamboRegistryContextType[\"toolRegistry\"];\n\n  /**\n   * Current thread ID (always available - uses \"placeholder\" for new threads)\n   */\n  currentThreadId: string;\n\n  /**\n   * Initialize a new thread in the stream context\n   */\n  initThread: ThreadManagement[\"initThread\"];\n\n  /**\n   * Switch the current active thread\n   */\n  switchThread: ThreadManagement[\"switchThread\"];\n\n  /**\n   * Start a new thread (generates a temporary ID)\n   */\n  startNewThread: ThreadManagement[\"startNewThread\"];\n\n  /**\n   * Dispatch function for stream events (advanced usage)\n   */\n  dispatch: ReturnType<typeof useStreamDispatch>;\n\n  /**\n   * Cancel the current run on this thread.\n   * Optimistically updates local state and sends cancellation request to the API.\n   * No-op if there's no active run or thread is a placeholder.\n   */\n  cancelRun: () => Promise<void>;\n\n  /**\n   * Current authentication state.\n   * Use this to show auth-related UI or conditionally render features.\n   */\n  authState: TamboAuthState;\n\n  /**\n   * Shorthand for `authState.status === \"identified\"`.\n   * When true, the SDK is ready to make API calls.\n   */\n  isIdentified: boolean;\n\n  /**\n   * Update a thread's name.\n   * Useful for implementing manual thread renaming UI in history sidebars.\n   * Cache invalidation is best-effort; failures will be logged and won't reject.\n   * @param threadId - ID of the thread to rename\n   * @param name - New name for the thread\n   * @returns Promise that resolves when the update completes\n   */\n  updateThreadName: (threadId: string, name: string) => Promise<void>;\n}\n\n/**\n * Main hook for accessing Tambo functionality.\n *\n * Combines thread state, streaming status, registry, and client\n * into a single convenient hook.\n *\n * Messages returned include renderedComponent on component content blocks,\n * allowing direct rendering via {content.renderedComponent}.\n * @param threadId - Optional thread ID to get state for\n * @returns Combined context with thread state, messages, and utilities\n * @example\n * ```tsx\n * function ChatInterface() {\n *   const {\n *     thread,\n *     messages,\n *     isStreaming,\n *     registerComponent,\n *   } = useTambo('thread_123');\n *\n *   return (\n *     <div>\n *       {messages.map(msg => <Message key={msg.id} message={msg} />)}\n *       {isStreaming && <LoadingIndicator />}\n *     </div>\n *   );\n * }\n * ```\n */\n/**\n * Cache entry for a rendered component wrapper.\n * Stores the element and the props JSON used to create it.\n */\ninterface ComponentCacheEntry {\n  element: ReactElement;\n  propsJson: string;\n}\n\n/**\n *\n * @returns The combined Tambo context\n */\nexport function useTambo(): UseTamboReturn {\n  const client = useTamboClient();\n  const queryClient = useTamboQueryClient();\n  const { userKey } = useTamboConfig();\n  const streamState = useStreamState();\n  const dispatch = useStreamDispatch();\n  const registry = useContext(TamboRegistryContext);\n  const threadManagement = useThreadManagement();\n  const authState = useTamboAuthState();\n\n  // Cache for rendered component wrappers - maintains stable element references\n  // across renders when props haven't changed\n  const componentCacheRef = useRef<Map<string, ComponentCacheEntry>>(new Map());\n\n  // Get thread state for the current thread\n  const threadState = streamState.threadMap[streamState.currentThreadId];\n\n  // Keep a live snapshot of the threadMap for callbacks without forcing them to\n  // re-create on every stream state update.\n  const threadMapRef = useRef(streamState.threadMap);\n  threadMapRef.current = streamState.threadMap;\n\n  // Cancel the current run on this thread\n  const cancelRun = useCallback(async () => {\n    const runId = threadState?.streaming.runId;\n    const threadId = streamState.currentThreadId;\n\n    // No-op if there's no active run or thread is a placeholder\n    if (!runId || isPlaceholderThreadId(threadId)) {\n      return;\n    }\n\n    // Optimistically update local state with RUN_ERROR event\n    dispatch({\n      type: \"EVENT\",\n      threadId,\n      event: {\n        type: EventType.RUN_ERROR,\n        message: \"Run cancelled\",\n        code: \"CANCELLED\",\n        timestamp: Date.now(),\n      },\n    });\n\n    // Call API to cancel the run\n    try {\n      await client.threads.runs.delete(runId, { threadId, userKey });\n    } catch (error) {\n      // Log but don't rethrow - local state is already updated\n      console.warn(\"Failed to cancel run on server:\", error);\n    }\n  }, [\n    client,\n    userKey,\n    streamState.currentThreadId,\n    threadState?.streaming.runId,\n    dispatch,\n  ]);\n\n  // Update a thread's name\n  const updateThreadName = useCallback(\n    async (threadId: string, name: string) => {\n      await client.threads.update(threadId, { name, userKey });\n\n      if (threadMapRef.current[threadId]) {\n        dispatch({\n          type: \"UPDATE_THREAD_NAME\",\n          threadId,\n          name: name,\n        });\n      }\n\n      try {\n        await Promise.all([\n          queryClient.invalidateQueries({\n            queryKey: [\"v1-threads\", \"list\"],\n          }),\n          queryClient.invalidateQueries({\n            queryKey: [\"v1-threads\", threadId],\n          }),\n        ]);\n      } catch (error) {\n        console.warn(\n          \"[useTambo] Failed to invalidate thread queries after rename:\",\n          error,\n        );\n      }\n    },\n    [client, userKey, dispatch, queryClient],\n  );\n\n  // Memoize the return object to prevent unnecessary re-renders\n  return useMemo(() => {\n    const thread = threadState;\n    const rawMessages = thread?.thread.messages ?? [];\n    const streamingState: StreamingState = thread?.streaming ?? {\n      status: \"idle\" as const,\n    };\n\n    // Build a set of tool_use IDs that have completed (have a matching tool_result)\n    // We need to look across all messages since tool_result might be in a different message\n    const completedToolIds = new Set<string>();\n    for (const msg of rawMessages) {\n      for (const content of msg.content) {\n        if (content.type === \"tool_result\") {\n          completedToolIds.add(content.toolUseId);\n        }\n      }\n    }\n\n    // Transform messages to add computed properties to content blocks\n    const messages = rawMessages.map((message): TamboThreadMessage => {\n      const transformedContent = message.content.map((content): Content => {\n        // Transform tool_use content to add computed state\n        if (content.type === \"tool_use\") {\n          const hasCompleted = completedToolIds.has(content.id);\n          const input = content.input ?? {};\n\n          // Extract Tambo display props from input\n          const tamboDisplayProps: TamboToolDisplayProps = {\n            _tambo_statusMessage:\n              typeof input._tambo_statusMessage === \"string\"\n                ? input._tambo_statusMessage\n                : undefined,\n            _tambo_completionStatusMessage:\n              typeof input._tambo_completionStatusMessage === \"string\"\n                ? input._tambo_completionStatusMessage\n                : undefined,\n          };\n\n          // Compute status message based on completion state\n          const statusMessage = hasCompleted\n            ? (tamboDisplayProps._tambo_completionStatusMessage ??\n              `Called ${content.name}`)\n            : (tamboDisplayProps._tambo_statusMessage ??\n              `Calling ${content.name}`);\n\n          // Filter out _tambo_* properties from input - consumers only see actual tool params\n          const cleanInput: Record<string, unknown> = {};\n          for (const [key, value] of Object.entries(input)) {\n            if (!key.startsWith(\"_tambo_\")) {\n              cleanInput[key] = value;\n            }\n          }\n\n          const v1Content: TamboToolUseContent = {\n            ...content,\n            input: cleanInput,\n            hasCompleted,\n            statusMessage,\n            tamboDisplayProps,\n          };\n          return v1Content;\n        }\n\n        if (content.type !== \"component\") {\n          return content;\n        }\n\n        const componentContent = content;\n        const propsJson = JSON.stringify(componentContent.props ?? {});\n        const cache = componentCacheRef.current;\n        const cached = cache.get(componentContent.id);\n\n        // Return cached element if props haven't changed\n        if (cached?.propsJson === propsJson) {\n          return {\n            ...componentContent,\n            renderedComponent: cached.element,\n          };\n        }\n\n        // Create new wrapper element\n        const element = React.createElement(ComponentRenderer, {\n          key: componentContent.id,\n          content: componentContent,\n          threadId: streamState.currentThreadId,\n          messageId: message.id,\n        });\n\n        // Update cache\n        cache.set(componentContent.id, { element, propsJson });\n\n        return {\n          ...componentContent,\n          renderedComponent: element,\n        };\n      });\n\n      return {\n        ...message,\n        content: transformedContent,\n      };\n    });\n\n    return {\n      client,\n      thread,\n      messages,\n      streamingState,\n      isStreaming: streamingState.status === \"streaming\",\n      isWaiting: streamingState.status === \"waiting\",\n      isIdle: streamingState.status === \"idle\",\n      registerComponent: registry.registerComponent,\n      registerTool: registry.registerTool,\n      registerTools: registry.registerTools,\n      componentList: registry.componentList,\n      toolRegistry: registry.toolRegistry,\n      currentThreadId: streamState.currentThreadId,\n      initThread: threadManagement.initThread,\n      switchThread: threadManagement.switchThread,\n      startNewThread: threadManagement.startNewThread,\n      dispatch,\n      cancelRun,\n      authState,\n      isIdentified: authState.status === \"identified\",\n      updateThreadName,\n    };\n  }, [\n    cancelRun,\n    updateThreadName,\n    client,\n    threadState,\n    registry,\n    streamState.currentThreadId,\n    threadManagement,\n    dispatch,\n    authState,\n  ]);\n}\n"]}
+{"version":3,"file":"use-tambo-v1.js","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1.ts"],"names":[],"mappings":";AAAA,YAAY,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AA+Mb,4BAqOC;AAlbD;;;;;GAKG;AAEH,sCAAwC;AAExC,+CAMe;AACf,iFAG+C;AAC/C,sEAAgE;AAChE,uEAA8D;AAE9D,qFAGiD;AACjD,+EAAwE;AACxE,kFAK8C;AAQ9C,6CAA2E;AAgK3E;;;GAGG;AACH,SAAgB,QAAQ;IACtB,MAAM,MAAM,GAAG,IAAA,sCAAc,GAAE,CAAC;IAChC,MAAM,WAAW,GAAG,IAAA,2CAAmB,GAAE,CAAC;IAC1C,MAAM,EAAE,OAAO,EAAE,GAAG,IAAA,kCAAc,GAAE,CAAC;IACrC,MAAM,WAAW,GAAG,IAAA,wCAAc,GAAE,CAAC;IACrC,MAAM,QAAQ,GAAG,IAAA,2CAAiB,GAAE,CAAC;IACrC,MAAM,QAAQ,GAAG,IAAA,kBAAU,EAAC,8CAAoB,CAAC,CAAC;IAClD,MAAM,gBAAgB,GAAG,IAAA,6CAAmB,GAAE,CAAC;IAC/C,MAAM,SAAS,GAAG,IAAA,2CAAiB,GAAE,CAAC;IAEtC,8EAA8E;IAC9E,4CAA4C;IAC5C,MAAM,iBAAiB,GAAG,IAAA,cAAM,EAAmC,IAAI,GAAG,EAAE,CAAC,CAAC;IAE9E,0CAA0C;IAC1C,MAAM,WAAW,GAAG,WAAW,CAAC,SAAS,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;IAEvE,8EAA8E;IAC9E,0CAA0C;IAC1C,MAAM,YAAY,GAAG,IAAA,cAAM,EAAC,WAAW,CAAC,SAAS,CAAC,CAAC;IACnD,YAAY,CAAC,OAAO,GAAG,WAAW,CAAC,SAAS,CAAC;IAE7C,wCAAwC;IACxC,MAAM,SAAS,GAAG,IAAA,mBAAW,EAAC,KAAK,IAAI,EAAE;QACvC,MAAM,KAAK,GAAG,WAAW,EAAE,SAAS,CAAC,KAAK,CAAC;QAC3C,MAAM,QAAQ,GAAG,WAAW,CAAC,eAAe,CAAC;QAE7C,4DAA4D;QAC5D,IAAI,CAAC,KAAK,IAAI,IAAA,8BAAqB,EAAC,QAAQ,CAAC,EAAE,CAAC;YAC9C,OAAO;QACT,CAAC;QAED,yDAAyD;QACzD,QAAQ,CAAC;YACP,IAAI,EAAE,OAAO;YACb,QAAQ;YACR,KAAK,EAAE;gBACL,IAAI,EAAE,gBAAS,CAAC,SAAS;gBACzB,OAAO,EAAE,eAAe;gBACxB,IAAI,EAAE,WAAW;gBACjB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB;SACF,CAAC,CAAC;QAEH,6BAA6B;QAC7B,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;QACjE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,yDAAyD;YACzD,OAAO,CAAC,IAAI,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;QACzD,CAAC;IACH,CAAC,EAAE;QACD,MAAM;QACN,OAAO;QACP,WAAW,CAAC,eAAe;QAC3B,WAAW,EAAE,SAAS,CAAC,KAAK;QAC5B,QAAQ;KACT,CAAC,CAAC;IAEH,yBAAyB;IACzB,MAAM,gBAAgB,GAAG,IAAA,mBAAW,EAClC,KAAK,EAAE,QAAgB,EAAE,IAAY,EAAE,EAAE;QACvC,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAEzD,IAAI,YAAY,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YACnC,QAAQ,CAAC;gBACP,IAAI,EAAE,oBAAoB;gBAC1B,QAAQ;gBACR,IAAI,EAAE,IAAI;aACX,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC;YACH,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChB,WAAW,CAAC,iBAAiB,CAAC;oBAC5B,QAAQ,EAAE,CAAC,YAAY,EAAE,MAAM,CAAC;iBACjC,CAAC;gBACF,WAAW,CAAC,iBAAiB,CAAC;oBAC5B,QAAQ,EAAE,CAAC,YAAY,EAAE,QAAQ,CAAC;iBACnC,CAAC;aACH,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,IAAI,CACV,8DAA8D,EAC9D,KAAK,CACN,CAAC;QACJ,CAAC;IACH,CAAC,EACD,CAAC,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,WAAW,CAAC,CACzC,CAAC;IAEF,8DAA8D;IAC9D,OAAO,IAAA,eAAO,EAAC,GAAG,EAAE;QAClB,MAAM,MAAM,GAAG,WAAW,CAAC;QAC3B,MAAM,WAAW,GAAG,MAAM,EAAE,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC;QAClD,MAAM,cAAc,GAAmB,MAAM,EAAE,SAAS,IAAI;YAC1D,MAAM,EAAE,MAAe;SACxB,CAAC;QAEF,gFAAgF;QAChF,wFAAwF;QACxF,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAU,CAAC;QAC3C,KAAK,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;YAC9B,KAAK,MAAM,OAAO,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;gBAClC,IAAI,OAAO,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;oBACnC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBAC1C,CAAC;YACH,CAAC;QACH,CAAC;QAED,kEAAkE;QAClE,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,OAAO,EAA2B,EAAE;YACpE,MAAM,kBAAkB,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,EAAW,EAAE;gBAClE,mDAAmD;gBACnD,IAAI,OAAO,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAChC,MAAM,YAAY,GAAG,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;oBACtD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC;oBAElC,yCAAyC;oBACzC,MAAM,iBAAiB,GAA0B;wBAC/C,oBAAoB,EAClB,OAAO,KAAK,CAAC,oBAAoB,KAAK,QAAQ;4BAC5C,CAAC,CAAC,KAAK,CAAC,oBAAoB;4BAC5B,CAAC,CAAC,SAAS;wBACf,8BAA8B,EAC5B,OAAO,KAAK,CAAC,8BAA8B,KAAK,QAAQ;4BACtD,CAAC,CAAC,KAAK,CAAC,8BAA8B;4BACtC,CAAC,CAAC,SAAS;qBAChB,CAAC;oBAEF,mDAAmD;oBACnD,MAAM,aAAa,GAAG,YAAY;wBAChC,CAAC,CAAC,CAAC,iBAAiB,CAAC,8BAA8B;4BACjD,UAAU,OAAO,CAAC,IAAI,EAAE,CAAC;wBAC3B,CAAC,CAAC,CAAC,iBAAiB,CAAC,oBAAoB;4BACvC,WAAW,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;oBAE/B,oFAAoF;oBACpF,MAAM,UAAU,GAA4B,EAAE,CAAC;oBAC/C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;wBACjD,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;4BAC/B,UAAU,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;wBAC1B,CAAC;oBACH,CAAC;oBAED,MAAM,SAAS,GAAwB;wBACrC,GAAG,OAAO;wBACV,KAAK,EAAE,UAAU;wBACjB,YAAY;wBACZ,aAAa;wBACb,iBAAiB;qBAClB,CAAC;oBACF,OAAO,SAAS,CAAC;gBACnB,CAAC;gBAED,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBACjC,OAAO,OAAO,CAAC;gBACjB,CAAC;gBAED,MAAM,gBAAgB,GAAG,OAAO,CAAC;gBACjC,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;gBAC/D,MAAM,KAAK,GAAG,iBAAiB,CAAC,OAAO,CAAC;gBACxC,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;gBAE9C,iDAAiD;gBACjD,IAAI,MAAM,EAAE,SAAS,KAAK,SAAS,EAAE,CAAC;oBACpC,OAAO;wBACL,GAAG,gBAAgB;wBACnB,iBAAiB,EAAE,MAAM,CAAC,OAAO;qBAClC,CAAC;gBACJ,CAAC;gBAED,6BAA6B;gBAC7B,MAAM,OAAO,GAAG,eAAK,CAAC,aAAa,CAAC,yCAAiB,EAAE;oBACrD,GAAG,EAAE,gBAAgB,CAAC,EAAE;oBACxB,OAAO,EAAE,gBAAgB;oBACzB,QAAQ,EAAE,WAAW,CAAC,eAAe;oBACrC,SAAS,EAAE,OAAO,CAAC,EAAE;iBACtB,CAAC,CAAC;gBAEH,eAAe;gBACf,KAAK,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;gBAEvD,OAAO;oBACL,GAAG,gBAAgB;oBACnB,iBAAiB,EAAE,OAAO;iBAC3B,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,OAAO;gBACL,GAAG,OAAO;gBACV,OAAO,EAAE,kBAAkB;aAC5B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,OAAO;YACL,MAAM;YACN,MAAM;YACN,QAAQ;YACR,cAAc;YACd,WAAW,EAAE,cAAc,CAAC,MAAM,KAAK,WAAW;YAClD,SAAS,EAAE,cAAc,CAAC,MAAM,KAAK,SAAS;YAC9C,MAAM,EAAE,cAAc,CAAC,MAAM,KAAK,MAAM;YACxC,iBAAiB,EAAE,QAAQ,CAAC,iBAAiB;YAC7C,YAAY,EAAE,QAAQ,CAAC,YAAY;YACnC,aAAa,EAAE,QAAQ,CAAC,aAAa;YACrC,aAAa,EAAE,QAAQ,CAAC,aAAa;YACrC,YAAY,EAAE,QAAQ,CAAC,YAAY;YACnC,eAAe,EAAE,WAAW,CAAC,eAAe;YAC5C,UAAU,EAAE,gBAAgB,CAAC,UAAU;YACvC,YAAY,EAAE,gBAAgB,CAAC,YAAY;YAC3C,cAAc,EAAE,gBAAgB,CAAC,cAAc;YAC/C,QAAQ;YACR,SAAS;YACT,SAAS;YACT,YAAY,EAAE,SAAS,CAAC,MAAM,KAAK,YAAY;YAC/C,gBAAgB;SACjB,CAAC;IACJ,CAAC,EAAE;QACD,SAAS;QACT,gBAAgB;QAChB,MAAM;QACN,WAAW;QACX,QAAQ;QACR,WAAW,CAAC,eAAe;QAC3B,gBAAgB;QAChB,QAAQ;QACR,SAAS;KACV,CAAC,CAAC;AACL,CAAC","sourcesContent":["\"use client\";\n\n/**\n * useTambo - Main Hook\n *\n * Combines all contexts into a single hook for convenient access\n * to thread state, streaming status, registry, and client.\n */\n\nimport { EventType } from \"@ag-ui/core\";\nimport type TamboAI from \"@tambo-ai/typescript-sdk\";\nimport React, {\n  useCallback,\n  useContext,\n  useMemo,\n  useRef,\n  type ReactElement,\n} from \"react\";\nimport {\n  useTamboClient,\n  useTamboQueryClient,\n} from \"../../providers/tambo-client-provider\";\nimport { useTamboConfig } from \"../providers/tambo-v1-provider\";\nimport { useTamboAuthState } from \"./use-tambo-v1-auth-state\";\nimport type { TamboAuthState } from \"@tambo-ai/client\";\nimport {\n  TamboRegistryContext,\n  type TamboRegistryContext as TamboRegistryContextType,\n} from \"../../providers/tambo-registry-provider\";\nimport { ComponentRenderer } from \"../components/v1-component-renderer\";\nimport {\n  useStreamDispatch,\n  useStreamState,\n  useThreadManagement,\n  type ThreadManagement,\n} from \"../providers/tambo-v1-stream-context\";\nimport type {\n  Content,\n  ReactTamboThreadMessage,\n  TamboToolDisplayProps,\n  TamboToolUseContent,\n} from \"../types/message\";\nimport type { StreamingState } from \"@tambo-ai/client\";\nimport { isPlaceholderThreadId, type ThreadState } from \"@tambo-ai/client\";\n\n/**\n * Return type for useTambo hook\n */\nexport interface UseTamboReturn {\n  /**\n   * The Tambo API client instance\n   */\n  client: TamboAI;\n\n  /**\n   * Current thread state for the given threadId, or undefined if not loaded\n   */\n  thread: ThreadState | undefined;\n\n  /**\n   * Messages in the current thread.\n   * Component content blocks include `renderedComponent` for direct rendering.\n   */\n  messages: ReactTamboThreadMessage[];\n\n  /**\n   * Current streaming state\n   */\n  streamingState: StreamingState;\n\n  /**\n   * Whether the thread is currently streaming a response\n   */\n  isStreaming: boolean;\n\n  /**\n   * Whether the thread is waiting for the AI to start responding\n   */\n  isWaiting: boolean;\n\n  /**\n   * Whether the thread is idle (not streaming or waiting)\n   */\n  isIdle: boolean;\n\n  /**\n   * Register a component with the registry\n   */\n  registerComponent: TamboRegistryContextType[\"registerComponent\"];\n\n  /**\n   * Register a tool with the registry\n   */\n  registerTool: TamboRegistryContextType[\"registerTool\"];\n\n  /**\n   * Register multiple tools with the registry\n   */\n  registerTools: TamboRegistryContextType[\"registerTools\"];\n\n  /**\n   * The component registry (Map of name -> component definition)\n   */\n  componentList: TamboRegistryContextType[\"componentList\"];\n\n  /**\n   * The tool registry (Map of name -> tool definition)\n   */\n  toolRegistry: TamboRegistryContextType[\"toolRegistry\"];\n\n  /**\n   * Current thread ID (always available - uses \"placeholder\" for new threads)\n   */\n  currentThreadId: string;\n\n  /**\n   * Initialize a new thread in the stream context\n   */\n  initThread: ThreadManagement[\"initThread\"];\n\n  /**\n   * Switch the current active thread\n   */\n  switchThread: ThreadManagement[\"switchThread\"];\n\n  /**\n   * Start a new thread (generates a temporary ID)\n   */\n  startNewThread: ThreadManagement[\"startNewThread\"];\n\n  /**\n   * Dispatch function for stream events (advanced usage)\n   */\n  dispatch: ReturnType<typeof useStreamDispatch>;\n\n  /**\n   * Cancel the current run on this thread.\n   * Optimistically updates local state and sends cancellation request to the API.\n   * No-op if there's no active run or thread is a placeholder.\n   */\n  cancelRun: () => Promise<void>;\n\n  /**\n   * Current authentication state.\n   * Use this to show auth-related UI or conditionally render features.\n   */\n  authState: TamboAuthState;\n\n  /**\n   * Shorthand for `authState.status === \"identified\"`.\n   * When true, the SDK is ready to make API calls.\n   */\n  isIdentified: boolean;\n\n  /**\n   * Update a thread's name.\n   * Useful for implementing manual thread renaming UI in history sidebars.\n   * Cache invalidation is best-effort; failures will be logged and won't reject.\n   * @param threadId - ID of the thread to rename\n   * @param name - New name for the thread\n   * @returns Promise that resolves when the update completes\n   */\n  updateThreadName: (threadId: string, name: string) => Promise<void>;\n}\n\n/**\n * Main hook for accessing Tambo functionality.\n *\n * Combines thread state, streaming status, registry, and client\n * into a single convenient hook.\n *\n * Messages returned include renderedComponent on component content blocks,\n * allowing direct rendering via {content.renderedComponent}.\n * @param threadId - Optional thread ID to get state for\n * @returns Combined context with thread state, messages, and utilities\n * @example\n * ```tsx\n * function ChatInterface() {\n *   const {\n *     thread,\n *     messages,\n *     isStreaming,\n *     registerComponent,\n *   } = useTambo('thread_123');\n *\n *   return (\n *     <div>\n *       {messages.map(msg => <Message key={msg.id} message={msg} />)}\n *       {isStreaming && <LoadingIndicator />}\n *     </div>\n *   );\n * }\n * ```\n */\n/**\n * Cache entry for a rendered component wrapper.\n * Stores the element and the props JSON used to create it.\n */\ninterface ComponentCacheEntry {\n  element: ReactElement;\n  propsJson: string;\n}\n\n/**\n *\n * @returns The combined Tambo context\n */\nexport function useTambo(): UseTamboReturn {\n  const client = useTamboClient();\n  const queryClient = useTamboQueryClient();\n  const { userKey } = useTamboConfig();\n  const streamState = useStreamState();\n  const dispatch = useStreamDispatch();\n  const registry = useContext(TamboRegistryContext);\n  const threadManagement = useThreadManagement();\n  const authState = useTamboAuthState();\n\n  // Cache for rendered component wrappers - maintains stable element references\n  // across renders when props haven't changed\n  const componentCacheRef = useRef<Map<string, ComponentCacheEntry>>(new Map());\n\n  // Get thread state for the current thread\n  const threadState = streamState.threadMap[streamState.currentThreadId];\n\n  // Keep a live snapshot of the threadMap for callbacks without forcing them to\n  // re-create on every stream state update.\n  const threadMapRef = useRef(streamState.threadMap);\n  threadMapRef.current = streamState.threadMap;\n\n  // Cancel the current run on this thread\n  const cancelRun = useCallback(async () => {\n    const runId = threadState?.streaming.runId;\n    const threadId = streamState.currentThreadId;\n\n    // No-op if there's no active run or thread is a placeholder\n    if (!runId || isPlaceholderThreadId(threadId)) {\n      return;\n    }\n\n    // Optimistically update local state with RUN_ERROR event\n    dispatch({\n      type: \"EVENT\",\n      threadId,\n      event: {\n        type: EventType.RUN_ERROR,\n        message: \"Run cancelled\",\n        code: \"CANCELLED\",\n        timestamp: Date.now(),\n      },\n    });\n\n    // Call API to cancel the run\n    try {\n      await client.threads.runs.delete(runId, { threadId, userKey });\n    } catch (error) {\n      // Log but don't rethrow - local state is already updated\n      console.warn(\"Failed to cancel run on server:\", error);\n    }\n  }, [\n    client,\n    userKey,\n    streamState.currentThreadId,\n    threadState?.streaming.runId,\n    dispatch,\n  ]);\n\n  // Update a thread's name\n  const updateThreadName = useCallback(\n    async (threadId: string, name: string) => {\n      await client.threads.update(threadId, { name, userKey });\n\n      if (threadMapRef.current[threadId]) {\n        dispatch({\n          type: \"UPDATE_THREAD_NAME\",\n          threadId,\n          name: name,\n        });\n      }\n\n      try {\n        await Promise.all([\n          queryClient.invalidateQueries({\n            queryKey: [\"v1-threads\", \"list\"],\n          }),\n          queryClient.invalidateQueries({\n            queryKey: [\"v1-threads\", threadId],\n          }),\n        ]);\n      } catch (error) {\n        console.warn(\n          \"[useTambo] Failed to invalidate thread queries after rename:\",\n          error,\n        );\n      }\n    },\n    [client, userKey, dispatch, queryClient],\n  );\n\n  // Memoize the return object to prevent unnecessary re-renders\n  return useMemo(() => {\n    const thread = threadState;\n    const rawMessages = thread?.thread.messages ?? [];\n    const streamingState: StreamingState = thread?.streaming ?? {\n      status: \"idle\" as const,\n    };\n\n    // Build a set of tool_use IDs that have completed (have a matching tool_result)\n    // We need to look across all messages since tool_result might be in a different message\n    const completedToolIds = new Set<string>();\n    for (const msg of rawMessages) {\n      for (const content of msg.content) {\n        if (content.type === \"tool_result\") {\n          completedToolIds.add(content.toolUseId);\n        }\n      }\n    }\n\n    // Transform messages to add computed properties to content blocks\n    const messages = rawMessages.map((message): ReactTamboThreadMessage => {\n      const transformedContent = message.content.map((content): Content => {\n        // Transform tool_use content to add computed state\n        if (content.type === \"tool_use\") {\n          const hasCompleted = completedToolIds.has(content.id);\n          const input = content.input ?? {};\n\n          // Extract Tambo display props from input\n          const tamboDisplayProps: TamboToolDisplayProps = {\n            _tambo_statusMessage:\n              typeof input._tambo_statusMessage === \"string\"\n                ? input._tambo_statusMessage\n                : undefined,\n            _tambo_completionStatusMessage:\n              typeof input._tambo_completionStatusMessage === \"string\"\n                ? input._tambo_completionStatusMessage\n                : undefined,\n          };\n\n          // Compute status message based on completion state\n          const statusMessage = hasCompleted\n            ? (tamboDisplayProps._tambo_completionStatusMessage ??\n              `Called ${content.name}`)\n            : (tamboDisplayProps._tambo_statusMessage ??\n              `Calling ${content.name}`);\n\n          // Filter out _tambo_* properties from input - consumers only see actual tool params\n          const cleanInput: Record<string, unknown> = {};\n          for (const [key, value] of Object.entries(input)) {\n            if (!key.startsWith(\"_tambo_\")) {\n              cleanInput[key] = value;\n            }\n          }\n\n          const v1Content: TamboToolUseContent = {\n            ...content,\n            input: cleanInput,\n            hasCompleted,\n            statusMessage,\n            tamboDisplayProps,\n          };\n          return v1Content;\n        }\n\n        if (content.type !== \"component\") {\n          return content;\n        }\n\n        const componentContent = content;\n        const propsJson = JSON.stringify(componentContent.props ?? {});\n        const cache = componentCacheRef.current;\n        const cached = cache.get(componentContent.id);\n\n        // Return cached element if props haven't changed\n        if (cached?.propsJson === propsJson) {\n          return {\n            ...componentContent,\n            renderedComponent: cached.element,\n          };\n        }\n\n        // Create new wrapper element\n        const element = React.createElement(ComponentRenderer, {\n          key: componentContent.id,\n          content: componentContent,\n          threadId: streamState.currentThreadId,\n          messageId: message.id,\n        });\n\n        // Update cache\n        cache.set(componentContent.id, { element, propsJson });\n\n        return {\n          ...componentContent,\n          renderedComponent: element,\n        };\n      });\n\n      return {\n        ...message,\n        content: transformedContent,\n      };\n    });\n\n    return {\n      client,\n      thread,\n      messages,\n      streamingState,\n      isStreaming: streamingState.status === \"streaming\",\n      isWaiting: streamingState.status === \"waiting\",\n      isIdle: streamingState.status === \"idle\",\n      registerComponent: registry.registerComponent,\n      registerTool: registry.registerTool,\n      registerTools: registry.registerTools,\n      componentList: registry.componentList,\n      toolRegistry: registry.toolRegistry,\n      currentThreadId: streamState.currentThreadId,\n      initThread: threadManagement.initThread,\n      switchThread: threadManagement.switchThread,\n      startNewThread: threadManagement.startNewThread,\n      dispatch,\n      cancelRun,\n      authState,\n      isIdentified: authState.status === \"identified\",\n      updateThreadName,\n    };\n  }, [\n    cancelRun,\n    updateThreadName,\n    client,\n    threadState,\n    registry,\n    streamState.currentThreadId,\n    threadManagement,\n    dispatch,\n    authState,\n  ]);\n}\n"]}

package/dist/v1/hooks/use-tambo-v1.test.js

@@ -39,7 +39,7 @@ const react_2 = __importStar(require("react"));
 const tambo_client_provider_1 = require("../../providers/tambo-client-provider");
 const tambo_registry_provider_1 = require("../../providers/tambo-registry-provider");
 const tambo_v1_stream_context_1 = require("../providers/tambo-v1-stream-context");
-const event_accumulator_1 = require("../utils/event-accumulator");
+const client_1 = require("@tambo-ai/client");
 const use_tambo_v1_1 = require("./use-tambo-v1");
 jest.mock("../../providers/tambo-client-provider", () => ({
     useTamboClient: jest.fn(),
@@ -82,6 +82,7 @@ describe("useTambo", () => {
         registerComponent: jest.fn(),
         registerTool: jest.fn(),
         registerTools: jest.fn(),
+        unregisterTools: jest.fn(),
         addToolAssociation: jest.fn(),
         registerMcpServer: jest.fn(),
         registerMcpServers: jest.fn(),
@@ -105,7 +106,7 @@ describe("useTambo", () => {
     // Wrapper that uses the real reducer so state updates properly
     function createWrapperWithRealReducer(initialState, stateRef) {
         return function WrapperWithRealReducer({ children, }) {
-            const [state, dispatch] = (0, react_2.useReducer)(event_accumulator_1.streamReducer, initialState);
+            const [state, dispatch] = (0, react_2.useReducer)(client_1.streamReducer, initialState);
             if (stateRef) {
                 stateRef.current = state;
             }
@@ -275,7 +276,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "idle" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_123",
@@ -313,7 +314,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "idle" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_123",
@@ -353,7 +354,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "idle" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_123",
@@ -390,7 +391,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "idle" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_123",
@@ -430,7 +431,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "idle" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_123",
@@ -472,7 +473,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "idle" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_123",
@@ -510,7 +511,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "idle" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "placeholder",
@@ -553,7 +554,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "idle" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_123",
@@ -607,7 +608,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "idle" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_123",
@@ -650,7 +651,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "idle" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_123",
@@ -706,7 +707,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "idle" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_123",
@@ -751,7 +752,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "idle" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_123",
@@ -793,7 +794,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "idle" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_123",
@@ -834,7 +835,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "idle" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_123",
@@ -893,7 +894,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "idle" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_123",
@@ -925,7 +926,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "streaming", runId: "run_456" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_123",
@@ -960,7 +961,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "idle" }, // No runId
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_123",
@@ -993,7 +994,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "streaming", runId: "run_123" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "placeholder",
@@ -1033,7 +1034,7 @@ describe("useTambo", () => {
                             lastRunCancelled: false,
                         },
                         streaming: { status: "streaming", runId: "run_456" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_123",
@@ -1071,7 +1072,7 @@ describe("useTambo", () => {
                             name: "Old Name",
                         },
                         streaming: { status: "idle" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_456",
@@ -1148,7 +1149,7 @@ describe("useTambo", () => {
                             name: "Old Name",
                         },
                         streaming: { status: "idle" },
-                        accumulatingToolArgs: new Map(),
+                        accumulatingToolArgs: {},
                     },
                 },
                 currentThreadId: "thread_456",