@tambo-ai/react 0.71.0 → 0.73.0

package/dist/v1/hooks/use-tambo-v1-component-state.d.ts

@@ -0,0 +1,44 @@
+/**
+ * Return type for useTamboV1ComponentState hook.
+ * Similar to useState but with additional metadata.
+ */
+export type UseTamboV1ComponentStateReturn<S> = [
+    currentState: S,
+    setState: (newState: S | ((prev: S) => S)) => void,
+    meta: {
+        isPending: boolean;
+        error: Error | null;
+        flush: () => void;
+    }
+];
+/**
+ * Hook for managing component state with bidirectional server sync.
+ *
+ * This hook acts like useState but automatically syncs state changes
+ * to the Tambo backend. Server-side state updates are also reflected
+ * in the component.
+ *
+ * Must be used within a component rendered via the component renderer.
+ * @param keyName - The unique key to identify this state value within the component's state
+ * @param initialValue - Initial value for the state (used if no server state exists)
+ * @param debounceTime - Debounce time in milliseconds (default: 500ms)
+ * @returns Tuple of [currentState, setState, meta]
+ * @example
+ * ```tsx
+ * function Counter() {
+ *   const [count, setCount, { isPending }] = useTamboV1ComponentState('count', 0);
+ *
+ *   return (
+ *     <div>
+ *       <span>{count}</span>
+ *       <button onClick={() => setCount(c => c + 1)} disabled={isPending}>
+ *         Increment
+ *       </button>
+ *     </div>
+ *   );
+ * }
+ * ```
+ */
+export declare function useTamboV1ComponentState<S = undefined>(keyName: string, initialValue?: S, debounceTime?: number): UseTamboV1ComponentStateReturn<S | undefined>;
+export declare function useTamboV1ComponentState<S>(keyName: string, initialValue: S, debounceTime?: number): UseTamboV1ComponentStateReturn<S>;
+//# sourceMappingURL=use-tambo-v1-component-state.d.ts.map

package/dist/v1/hooks/use-tambo-v1-component-state.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-tambo-v1-component-state.d.ts","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1-component-state.ts"],"names":[],"mappings":"AAoBA;;;GAGG;AACH,MAAM,MAAM,8BAA8B,CAAC,CAAC,IAAI;IAC9C,YAAY,EAAE,CAAC;IACf,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,IAAI;IAClD,IAAI,EAAE;QACJ,SAAS,EAAE,OAAO,CAAC;QACnB,KAAK,EAAE,KAAK,GAAG,IAAI,CAAC;QACpB,KAAK,EAAE,MAAM,IAAI,CAAC;KACnB;CACF,CAAC;AAgCF;;;;;;;;;;;;;;;;;;;;;;;;;;;GA2BG;AACH,wBAAgB,wBAAwB,CAAC,CAAC,GAAG,SAAS,EACpD,OAAO,EAAE,MAAM,EACf,YAAY,CAAC,EAAE,CAAC,EAChB,YAAY,CAAC,EAAE,MAAM,GACpB,8BAA8B,CAAC,CAAC,GAAG,SAAS,CAAC,CAAC;AACjD,wBAAgB,wBAAwB,CAAC,CAAC,EACxC,OAAO,EAAE,MAAM,EACf,YAAY,EAAE,CAAC,EACf,YAAY,CAAC,EAAE,MAAM,GACpB,8BAA8B,CAAC,CAAC,CAAC,CAAC"}

package/dist/v1/hooks/use-tambo-v1-component-state.js

@@ -0,0 +1,134 @@
+"use strict";
+"use client";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useTamboV1ComponentState = useTamboV1ComponentState;
+/**
+ * useTamboV1ComponentState - Component State Hook for v1 API
+ *
+ * Provides bidirectional state synchronization between React components
+ * and the Tambo backend. State changes are debounced before syncing to
+ * the server, and server state updates are reflected in the component.
+ *
+ * Must be used within a component rendered via the component renderer.
+ */
+const react_1 = require("react");
+const use_debounce_1 = require("use-debounce");
+const fast_equals_1 = require("fast-equals");
+const tambo_client_provider_1 = require("../../providers/tambo-client-provider");
+const component_renderer_1 = require("../utils/component-renderer");
+const tambo_v1_stream_context_1 = require("../providers/tambo-v1-stream-context");
+/**
+ * Find a component content block by ID in a specific thread.
+ * Only searches the specified thread to prevent cross-thread data access
+ * and improve performance (O(m*k) instead of O(n*m*k)).
+ * @param streamState - The current stream state
+ * @param threadId - The thread ID to search in
+ * @param componentId - The component ID to find
+ * @returns The component content block, or undefined if not found
+ */
+function findComponentContent(streamState, threadId, componentId) {
+    // Only search the specified thread (not all threads)
+    const threadState = streamState.threadMap[threadId];
+    if (!threadState) {
+        return undefined;
+    }
+    for (const message of threadState.thread.messages) {
+        for (const content of message.content) {
+            if (content.type === "component" && content.id === componentId) {
+                return content;
+            }
+        }
+    }
+    return undefined;
+}
+function useTamboV1ComponentState(keyName, initialValue, debounceTime = 500) {
+    const client = (0, tambo_client_provider_1.useTamboClient)();
+    const { componentId, threadId } = (0, component_renderer_1.useV1ComponentContent)();
+    const streamState = (0, tambo_v1_stream_context_1.useStreamState)();
+    // Find the component content to get server state (only search current thread)
+    const componentContent = findComponentContent(streamState, threadId, componentId);
+    const serverState = componentContent?.state;
+    const serverValue = serverState?.[keyName];
+    // Local state - initialized from server state or initial value
+    const [localState, setLocalState] = (0, react_1.useState)(() => serverValue ?? initialValue);
+    // Track pending state and errors
+    const [isPending, setIsPending] = (0, react_1.useState)(false);
+    const [error, setError] = (0, react_1.useState)(null);
+    // Track the last value we sent to avoid overwriting with stale server state
+    const lastSentValueRef = (0, react_1.useRef)(undefined);
+    // Track whether there's a pending local change that hasn't synced yet
+    const hasPendingLocalChangeRef = (0, react_1.useRef)(false);
+    // Track in-flight sync requests to avoid stale completions clearing pending state
+    const syncSeqRef = (0, react_1.useRef)(0);
+    // Debounced function to sync state to server
+    const syncToServer = (0, use_debounce_1.useDebouncedCallback)(async (newState) => {
+        const seq = ++syncSeqRef.current;
+        setIsPending(true);
+        setError(null);
+        lastSentValueRef.current = newState;
+        try {
+            await client.threads.state.updateState(componentId, {
+                threadId,
+                state: { [keyName]: newState },
+            });
+            // Clear pending flag after successful sync
+            hasPendingLocalChangeRef.current = false;
+        }
+        catch (err) {
+            // Clear pending flag on error to allow server reconciliation
+            hasPendingLocalChangeRef.current = false;
+            const syncError = err instanceof Error ? err : new Error(String(err));
+            setError(syncError);
+            console.error(`[useTamboV1ComponentState] Failed to sync state for ${componentId}:`, syncError);
+        }
+        finally {
+            // Only clear isPending if this is the most recent request
+            if (seq === syncSeqRef.current) {
+                setIsPending(false);
+            }
+        }
+    }, debounceTime);
+    // setState function that updates local state and triggers debounced sync
+    const setState = (0, react_1.useCallback)((newState) => {
+        setLocalState((prev) => {
+            const nextState = typeof newState === "function"
+                ? newState(prev)
+                : newState;
+            // Mark that we have a pending local change
+            hasPendingLocalChangeRef.current = true;
+            // Trigger debounced sync to server
+            void syncToServer(nextState);
+            return nextState;
+        });
+    }, [syncToServer]);
+    // Sync from server state when it changes (e.g., from streaming events)
+    (0, react_1.useEffect)(() => {
+        if (serverValue === undefined) {
+            return;
+        }
+        // Don't overwrite local changes that haven't synced yet
+        if (hasPendingLocalChangeRef.current) {
+            return;
+        }
+        // Only sync if the server value is different from what we last sent
+        // This prevents overwriting local state with stale server values
+        if (lastSentValueRef.current !== undefined &&
+            (0, fast_equals_1.deepEqual)(serverValue, lastSentValueRef.current)) {
+            return;
+        }
+        // Use functional update to avoid localState in deps
+        setLocalState((prev) => (0, fast_equals_1.deepEqual)(serverValue, prev) ? prev : serverValue);
+    }, [serverValue]);
+    // Flush pending updates on unmount
+    (0, react_1.useEffect)(() => {
+        return () => {
+            void syncToServer.flush();
+        };
+    }, [syncToServer]);
+    // Flush function for immediate sync
+    const flush = (0, react_1.useCallback)(() => {
+        void syncToServer.flush();
+    }, [syncToServer]);
+    return [localState, setState, { isPending, error, flush }];
+}
+//# sourceMappingURL=use-tambo-v1-component-state.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"use-tambo-v1-component-state.js","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1-component-state.ts"],"names":[],"mappings":";AAAA,YAAY,CAAC;;AAsGb,4DAiIC;AArOD;;;;;;;;GAQG;AAEH,iCAAiE;AACjE,+CAAoD;AACpD,6CAAwC;AACxC,iFAAuE;AACvE,oEAAoE;AACpE,kFAAsE;AAiBtE;;;;;;;;GAQG;AACH,SAAS,oBAAoB,CAC3B,WAA8C,EAC9C,QAAgB,EAChB,WAAmB;IAEnB,qDAAqD;IACrD,MAAM,WAAW,GAAG,WAAW,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IACpD,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,KAAK,MAAM,OAAO,IAAI,WAAW,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QAClD,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACtC,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,EAAE,KAAK,WAAW,EAAE,CAAC;gBAC/D,OAAO,OAAO,CAAC;YACjB,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAwCD,SAAgB,wBAAwB,CACtC,OAAe,EACf,YAAgB,EAChB,YAAY,GAAG,GAAG;IAElB,MAAM,MAAM,GAAG,IAAA,sCAAc,GAAE,CAAC;IAChC,MAAM,EAAE,WAAW,EAAE,QAAQ,EAAE,GAAG,IAAA,0CAAqB,GAAE,CAAC;IAC1D,MAAM,WAAW,GAAG,IAAA,wCAAc,GAAE,CAAC;IAErC,8EAA8E;IAC9E,MAAM,gBAAgB,GAAG,oBAAoB,CAC3C,WAAW,EACX,QAAQ,EACR,WAAW,CACZ,CAAC;IACF,MAAM,WAAW,GAAG,gBAAgB,EAAE,KAEzB,CAAC;IACd,MAAM,WAAW,GAAG,WAAW,EAAE,CAAC,OAAO,CAAkB,CAAC;IAE5D,+DAA+D;IAC/D,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,GAAG,IAAA,gBAAQ,EAC1C,GAAG,EAAE,CAAC,WAAW,IAAK,YAAkB,CACzC,CAAC;IAEF,iCAAiC;IACjC,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,IAAA,gBAAQ,EAAC,KAAK,CAAC,CAAC;IAClD,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,IAAA,gBAAQ,EAAe,IAAI,CAAC,CAAC;IAEvD,4EAA4E;IAC5E,MAAM,gBAAgB,GAAG,IAAA,cAAM,EAAgB,SAAS,CAAC,CAAC;IAE1D,sEAAsE;IACtE,MAAM,wBAAwB,GAAG,IAAA,cAAM,EAAC,KAAK,CAAC,CAAC;IAE/C,kFAAkF;IAClF,MAAM,UAAU,GAAG,IAAA,cAAM,EAAC,CAAC,CAAC,CAAC;IAE7B,6CAA6C;IAC7C,MAAM,YAAY,GAAG,IAAA,mCAAoB,EAAC,KAAK,EAAE,QAAW,EAAE,EAAE;QAC9D,MAAM,GAAG,GAAG,EAAE,UAAU,CAAC,OAAO,CAAC;QACjC,YAAY,CAAC,IAAI,CAAC,CAAC;QACnB,QAAQ,CAAC,IAAI,CAAC,CAAC;QACf,gBAAgB,CAAC,OAAO,GAAG,QAAQ,CAAC;QAEpC,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,WAAW,EAAE;gBAClD,QAAQ;gBACR,KAAK,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,QAAQ,EAAE;aAC/B,CAAC,CAAC;YACH,2CAA2C;YAC3C,wBAAwB,CAAC,OAAO,GAAG,KAAK,CAAC;QAC3C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,6DAA6D;YAC7D,wBAAwB,CAAC,OAAO,GAAG,KAAK,CAAC;YACzC,MAAM,SAAS,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;YACtE,QAAQ,CAAC,SAAS,CAAC,CAAC;YACpB,OAAO,CAAC,KAAK,CACX,uDAAuD,WAAW,GAAG,EACrE,SAAS,CACV,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,0DAA0D;YAC1D,IAAI,GAAG,KAAK,UAAU,CAAC,OAAO,EAAE,CAAC;gBAC/B,YAAY,CAAC,KAAK,CAAC,CAAC;YACtB,CAAC;QACH,CAAC;IACH,CAAC,EAAE,YAAY,CAAC,CAAC;IAEjB,yEAAyE;IACzE,MAAM,QAAQ,GAAG,IAAA,mBAAW,EAC1B,CAAC,QAA8B,EAAE,EAAE;QACjC,aAAa,CAAC,CAAC,IAAI,EAAE,EAAE;YACrB,MAAM,SAAS,GACb,OAAO,QAAQ,KAAK,UAAU;gBAC5B,CAAC,CAAE,QAA2B,CAAC,IAAI,CAAC;gBACpC,CAAC,CAAC,QAAQ,CAAC;YAEf,2CAA2C;YAC3C,wBAAwB,CAAC,OAAO,GAAG,IAAI,CAAC;YAExC,mCAAmC;YACnC,KAAK,YAAY,CAAC,SAAS,CAAC,CAAC;YAE7B,OAAO,SAAS,CAAC;QACnB,CAAC,CAAC,CAAC;IACL,CAAC,EACD,CAAC,YAAY,CAAC,CACf,CAAC;IAEF,uEAAuE;IACvE,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;YAC9B,OAAO;QACT,CAAC;QAED,wDAAwD;QACxD,IAAI,wBAAwB,CAAC,OAAO,EAAE,CAAC;YACrC,OAAO;QACT,CAAC;QAED,oEAAoE;QACpE,iEAAiE;QACjE,IACE,gBAAgB,CAAC,OAAO,KAAK,SAAS;YACtC,IAAA,uBAAS,EAAC,WAAW,EAAE,gBAAgB,CAAC,OAAO,CAAC,EAChD,CAAC;YACD,OAAO;QACT,CAAC;QAED,oDAAoD;QACpD,aAAa,CAAC,CAAC,IAAI,EAAE,EAAE,CACrB,IAAA,uBAAS,EAAC,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,WAAW,CAClD,CAAC;IACJ,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC;IAElB,mCAAmC;IACnC,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,OAAO,GAAG,EAAE;YACV,KAAK,YAAY,CAAC,KAAK,EAAE,CAAC;QAC5B,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;IAEnB,oCAAoC;IACpC,MAAM,KAAK,GAAG,IAAA,mBAAW,EAAC,GAAG,EAAE;QAC7B,KAAK,YAAY,CAAC,KAAK,EAAE,CAAC;IAC5B,CAAC,EAAE,CAAC,YAAY,CAAC,CAAC,CAAC;IAEnB,OAAO,CAAC,UAAU,EAAE,QAAQ,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;AAC7D,CAAC","sourcesContent":["\"use client\";\n\n/**\n * useTamboV1ComponentState - Component State Hook for v1 API\n *\n * Provides bidirectional state synchronization between React components\n * and the Tambo backend. State changes are debounced before syncing to\n * the server, and server state updates are reflected in the component.\n *\n * Must be used within a component rendered via the component renderer.\n */\n\nimport { useCallback, useEffect, useState, useRef } from \"react\";\nimport { useDebouncedCallback } from \"use-debounce\";\nimport { deepEqual } from \"fast-equals\";\nimport { useTamboClient } from \"../../providers/tambo-client-provider\";\nimport { useV1ComponentContent } from \"../utils/component-renderer\";\nimport { useStreamState } from \"../providers/tambo-v1-stream-context\";\nimport type { V1ComponentContent } from \"../types/message\";\n\n/**\n * Return type for useTamboV1ComponentState hook.\n * Similar to useState but with additional metadata.\n */\nexport type UseTamboV1ComponentStateReturn<S> = [\n  currentState: S,\n  setState: (newState: S | ((prev: S) => S)) => void,\n  meta: {\n    isPending: boolean;\n    error: Error | null;\n    flush: () => void;\n  },\n];\n\n/**\n * Find a component content block by ID in a specific thread.\n * Only searches the specified thread to prevent cross-thread data access\n * and improve performance (O(m*k) instead of O(n*m*k)).\n * @param streamState - The current stream state\n * @param threadId - The thread ID to search in\n * @param componentId - The component ID to find\n * @returns The component content block, or undefined if not found\n */\nfunction findComponentContent(\n  streamState: ReturnType<typeof useStreamState>,\n  threadId: string,\n  componentId: string,\n): V1ComponentContent | undefined {\n  // Only search the specified thread (not all threads)\n  const threadState = streamState.threadMap[threadId];\n  if (!threadState) {\n    return undefined;\n  }\n\n  for (const message of threadState.thread.messages) {\n    for (const content of message.content) {\n      if (content.type === \"component\" && content.id === componentId) {\n        return content;\n      }\n    }\n  }\n  return undefined;\n}\n\n/**\n * Hook for managing component state with bidirectional server sync.\n *\n * This hook acts like useState but automatically syncs state changes\n * to the Tambo backend. Server-side state updates are also reflected\n * in the component.\n *\n * Must be used within a component rendered via the component renderer.\n * @param keyName - The unique key to identify this state value within the component's state\n * @param initialValue - Initial value for the state (used if no server state exists)\n * @param debounceTime - Debounce time in milliseconds (default: 500ms)\n * @returns Tuple of [currentState, setState, meta]\n * @example\n * ```tsx\n * function Counter() {\n *   const [count, setCount, { isPending }] = useTamboV1ComponentState('count', 0);\n *\n *   return (\n *     <div>\n *       <span>{count}</span>\n *       <button onClick={() => setCount(c => c + 1)} disabled={isPending}>\n *         Increment\n *       </button>\n *     </div>\n *   );\n * }\n * ```\n */\nexport function useTamboV1ComponentState<S = undefined>(\n  keyName: string,\n  initialValue?: S,\n  debounceTime?: number,\n): UseTamboV1ComponentStateReturn<S | undefined>;\nexport function useTamboV1ComponentState<S>(\n  keyName: string,\n  initialValue: S,\n  debounceTime?: number,\n): UseTamboV1ComponentStateReturn<S>;\nexport function useTamboV1ComponentState<S>(\n  keyName: string,\n  initialValue?: S,\n  debounceTime = 500,\n): UseTamboV1ComponentStateReturn<S> {\n  const client = useTamboClient();\n  const { componentId, threadId } = useV1ComponentContent();\n  const streamState = useStreamState();\n\n  // Find the component content to get server state (only search current thread)\n  const componentContent = findComponentContent(\n    streamState,\n    threadId,\n    componentId,\n  );\n  const serverState = componentContent?.state as\n    | Record<string, unknown>\n    | undefined;\n  const serverValue = serverState?.[keyName] as S | undefined;\n\n  // Local state - initialized from server state or initial value\n  const [localState, setLocalState] = useState<S>(\n    () => serverValue ?? (initialValue as S),\n  );\n\n  // Track pending state and errors\n  const [isPending, setIsPending] = useState(false);\n  const [error, setError] = useState<Error | null>(null);\n\n  // Track the last value we sent to avoid overwriting with stale server state\n  const lastSentValueRef = useRef<S | undefined>(undefined);\n\n  // Track whether there's a pending local change that hasn't synced yet\n  const hasPendingLocalChangeRef = useRef(false);\n\n  // Track in-flight sync requests to avoid stale completions clearing pending state\n  const syncSeqRef = useRef(0);\n\n  // Debounced function to sync state to server\n  const syncToServer = useDebouncedCallback(async (newState: S) => {\n    const seq = ++syncSeqRef.current;\n    setIsPending(true);\n    setError(null);\n    lastSentValueRef.current = newState;\n\n    try {\n      await client.threads.state.updateState(componentId, {\n        threadId,\n        state: { [keyName]: newState },\n      });\n      // Clear pending flag after successful sync\n      hasPendingLocalChangeRef.current = false;\n    } catch (err) {\n      // Clear pending flag on error to allow server reconciliation\n      hasPendingLocalChangeRef.current = false;\n      const syncError = err instanceof Error ? err : new Error(String(err));\n      setError(syncError);\n      console.error(\n        `[useTamboV1ComponentState] Failed to sync state for ${componentId}:`,\n        syncError,\n      );\n    } finally {\n      // Only clear isPending if this is the most recent request\n      if (seq === syncSeqRef.current) {\n        setIsPending(false);\n      }\n    }\n  }, debounceTime);\n\n  // setState function that updates local state and triggers debounced sync\n  const setState = useCallback(\n    (newState: S | ((prev: S) => S)) => {\n      setLocalState((prev) => {\n        const nextState =\n          typeof newState === \"function\"\n            ? (newState as (prev: S) => S)(prev)\n            : newState;\n\n        // Mark that we have a pending local change\n        hasPendingLocalChangeRef.current = true;\n\n        // Trigger debounced sync to server\n        void syncToServer(nextState);\n\n        return nextState;\n      });\n    },\n    [syncToServer],\n  );\n\n  // Sync from server state when it changes (e.g., from streaming events)\n  useEffect(() => {\n    if (serverValue === undefined) {\n      return;\n    }\n\n    // Don't overwrite local changes that haven't synced yet\n    if (hasPendingLocalChangeRef.current) {\n      return;\n    }\n\n    // Only sync if the server value is different from what we last sent\n    // This prevents overwriting local state with stale server values\n    if (\n      lastSentValueRef.current !== undefined &&\n      deepEqual(serverValue, lastSentValueRef.current)\n    ) {\n      return;\n    }\n\n    // Use functional update to avoid localState in deps\n    setLocalState((prev) =>\n      deepEqual(serverValue, prev) ? prev : serverValue,\n    );\n  }, [serverValue]);\n\n  // Flush pending updates on unmount\n  useEffect(() => {\n    return () => {\n      void syncToServer.flush();\n    };\n  }, [syncToServer]);\n\n  // Flush function for immediate sync\n  const flush = useCallback(() => {\n    void syncToServer.flush();\n  }, [syncToServer]);\n\n  return [localState, setState, { isPending, error, flush }];\n}\n"]}

package/dist/v1/hooks/use-tambo-v1-component-state.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=use-tambo-v1-component-state.test.d.ts.map

package/dist/v1/hooks/use-tambo-v1-component-state.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-tambo-v1-component-state.test.d.ts","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1-component-state.test.tsx"],"names":[],"mappings":""}

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

@@ -0,0 +1,292 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const react_1 = require("@testing-library/react");
+const use_tambo_v1_component_state_1 = require("./use-tambo-v1-component-state");
+// Mock the required modules
+jest.mock("../../providers/tambo-client-provider", () => ({
+    useTamboClient: jest.fn(),
+}));
+jest.mock("../providers/tambo-v1-stream-context", () => ({
+    useStreamState: jest.fn(),
+}));
+jest.mock("../utils/component-renderer", () => ({
+    useV1ComponentContent: jest.fn(),
+}));
+// Create a mock debounced function with flush method
+const createMockDebouncedFunction = (fn) => {
+    const debouncedFn = jest.fn((...args) => fn(...args));
+    debouncedFn.flush = jest.fn();
+    debouncedFn.cancel = jest.fn();
+    debouncedFn.isPending = jest.fn(() => false);
+    return debouncedFn;
+};
+// Mock use-debounce
+jest.mock("use-debounce", () => ({
+    useDebouncedCallback: jest.fn(),
+}));
+// Import the mocked modules
+const tambo_client_provider_1 = require("../../providers/tambo-client-provider");
+const tambo_v1_stream_context_1 = require("../providers/tambo-v1-stream-context");
+const component_renderer_1 = require("../utils/component-renderer");
+const use_debounce_1 = require("use-debounce");
+describe("useTamboV1ComponentState", () => {
+    const mockUpdateState = jest.fn();
+    const mockComponentId = "comp_test123";
+    const mockThreadId = "thread_abc";
+    const mockMessageId = "msg_xyz";
+    // Helper to create mock stream state
+    const createMockStreamState = (componentState) => ({
+        threadMap: {
+            [mockThreadId]: {
+                thread: {
+                    id: mockThreadId,
+                    messages: [
+                        {
+                            id: mockMessageId,
+                            role: "assistant",
+                            content: [
+                                {
+                                    type: "component",
+                                    id: mockComponentId,
+                                    name: "TestComponent",
+                                    props: {},
+                                    state: componentState,
+                                    streamingState: "done",
+                                },
+                            ],
+                            createdAt: new Date().toISOString(),
+                        },
+                    ],
+                    status: "idle",
+                    createdAt: new Date().toISOString(),
+                    updatedAt: new Date().toISOString(),
+                },
+                streaming: { status: "idle" },
+                accumulatingToolArgs: new Map(),
+            },
+        },
+        currentThreadId: mockThreadId,
+    });
+    beforeEach(() => {
+        jest.clearAllMocks();
+        // Setup default mocks
+        jest.mocked(tambo_client_provider_1.useTamboClient).mockReturnValue({
+            threads: {
+                state: {
+                    updateState: mockUpdateState,
+                },
+            },
+        });
+        jest.mocked(component_renderer_1.useV1ComponentContent).mockReturnValue({
+            componentId: mockComponentId,
+            threadId: mockThreadId,
+            messageId: mockMessageId,
+            componentName: "TestComponent",
+        });
+        jest.mocked(tambo_v1_stream_context_1.useStreamState).mockReturnValue(createMockStreamState());
+        // Setup default mock for useDebouncedCallback
+        jest
+            .mocked(use_debounce_1.useDebouncedCallback)
+            .mockImplementation((fn) => createMockDebouncedFunction(fn));
+    });
+    afterEach(() => {
+        jest.restoreAllMocks();
+    });
+    describe("Initial State Management", () => {
+        it("should initialize with initialValue when no server state exists", () => {
+            const initialValue = "test-initial";
+            jest.mocked(tambo_v1_stream_context_1.useStreamState).mockReturnValue(createMockStreamState({}));
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_component_state_1.useTamboV1ComponentState)("testKey", initialValue));
+            expect(result.current[0]).toBe(initialValue);
+        });
+        it("should use server state over initialValue", () => {
+            const initialValue = "initial";
+            const serverValue = "server-value";
+            jest
+                .mocked(tambo_v1_stream_context_1.useStreamState)
+                .mockReturnValue(createMockStreamState({ testKey: serverValue }));
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_component_state_1.useTamboV1ComponentState)("testKey", initialValue));
+            expect(result.current[0]).toBe(serverValue);
+        });
+        it("should handle undefined initialValue gracefully", () => {
+            jest.mocked(tambo_v1_stream_context_1.useStreamState).mockReturnValue(createMockStreamState({}));
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_component_state_1.useTamboV1ComponentState)("testKey"));
+            expect(result.current[0]).toBeUndefined();
+        });
+        it("should handle different data types correctly", () => {
+            const testCases = [
+                { value: "string" },
+                { value: 42 },
+                { value: true },
+                { value: { name: "test" } },
+                { value: [1, 2, 3] },
+            ];
+            testCases.forEach(({ value }) => {
+                jest
+                    .mocked(tambo_v1_stream_context_1.useStreamState)
+                    .mockReturnValue(createMockStreamState({ testKey: value }));
+                const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_component_state_1.useTamboV1ComponentState)("testKey", value));
+                expect(result.current[0]).toEqual(value);
+            });
+        });
+    });
+    describe("State Updates", () => {
+        it("should update local state immediately when setState is called", () => {
+            const initialValue = "initial";
+            jest
+                .mocked(tambo_v1_stream_context_1.useStreamState)
+                .mockReturnValue(createMockStreamState({ testKey: initialValue }));
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_component_state_1.useTamboV1ComponentState)("testKey", initialValue));
+            const newValue = "updated";
+            (0, react_1.act)(() => {
+                result.current[1](newValue);
+            });
+            expect(result.current[0]).toBe(newValue);
+        });
+        it("should support functional updates", () => {
+            const initialValue = 5;
+            jest
+                .mocked(tambo_v1_stream_context_1.useStreamState)
+                .mockReturnValue(createMockStreamState({ counter: initialValue }));
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_component_state_1.useTamboV1ComponentState)("counter", initialValue));
+            (0, react_1.act)(() => {
+                result.current[1]((prev) => (prev ?? 0) + 1);
+            });
+            expect(result.current[0]).toBe(6);
+        });
+        it("should trigger debounced API call when setState is called", () => {
+            const initialValue = "initial";
+            jest
+                .mocked(tambo_v1_stream_context_1.useStreamState)
+                .mockReturnValue(createMockStreamState({ testKey: initialValue }));
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_component_state_1.useTamboV1ComponentState)("testKey", initialValue));
+            const newValue = "updated";
+            (0, react_1.act)(() => {
+                result.current[1](newValue);
+            });
+            // The debounced function should be called
+            expect(mockUpdateState).toHaveBeenCalledWith(mockComponentId, {
+                threadId: mockThreadId,
+                state: { testKey: newValue },
+            });
+        });
+        it("should work with complex objects", () => {
+            const initialValue = { name: "test", items: [1, 2, 3] };
+            jest
+                .mocked(tambo_v1_stream_context_1.useStreamState)
+                .mockReturnValue(createMockStreamState({ data: initialValue }));
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_component_state_1.useTamboV1ComponentState)("data", initialValue));
+            const newValue = { name: "updated", items: [4, 5, 6] };
+            (0, react_1.act)(() => {
+                result.current[1](newValue);
+            });
+            expect(result.current[0]).toEqual(newValue);
+        });
+    });
+    describe("Metadata", () => {
+        it("should return isPending and error in meta", () => {
+            jest.mocked(tambo_v1_stream_context_1.useStreamState).mockReturnValue(createMockStreamState({}));
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_component_state_1.useTamboV1ComponentState)("testKey", "initial"));
+            expect(result.current[2]).toEqual({
+                isPending: false,
+                error: null,
+                flush: expect.any(Function),
+            });
+        });
+        it("should provide a flush function", () => {
+            const mockFlush = jest.fn();
+            const mockDebouncedFn = createMockDebouncedFunction(jest.fn());
+            mockDebouncedFn.flush = mockFlush;
+            jest.mocked(use_debounce_1.useDebouncedCallback).mockReturnValue(mockDebouncedFn);
+            jest.mocked(tambo_v1_stream_context_1.useStreamState).mockReturnValue(createMockStreamState({}));
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_component_state_1.useTamboV1ComponentState)("testKey", "initial"));
+            (0, react_1.act)(() => {
+                result.current[2].flush();
+            });
+            expect(mockFlush).toHaveBeenCalled();
+        });
+    });
+    describe("Debouncing Behavior", () => {
+        it("should use default debounce time of 500ms", () => {
+            jest.mocked(tambo_v1_stream_context_1.useStreamState).mockReturnValue(createMockStreamState({}));
+            (0, react_1.renderHook)(() => (0, use_tambo_v1_component_state_1.useTamboV1ComponentState)("testKey", "initial"));
+            expect(use_debounce_1.useDebouncedCallback).toHaveBeenCalledWith(expect.any(Function), 500);
+        });
+        it("should use custom debounce time when provided", () => {
+            jest.mocked(tambo_v1_stream_context_1.useStreamState).mockReturnValue(createMockStreamState({}));
+            const customDebounceTime = 1000;
+            (0, react_1.renderHook)(() => (0, use_tambo_v1_component_state_1.useTamboV1ComponentState)("testKey", "initial", customDebounceTime));
+            expect(use_debounce_1.useDebouncedCallback).toHaveBeenCalledWith(expect.any(Function), customDebounceTime);
+        });
+        it("should flush debounced callback on unmount", () => {
+            const mockFlush = jest.fn();
+            const mockDebouncedFn = createMockDebouncedFunction(jest.fn());
+            mockDebouncedFn.flush = mockFlush;
+            jest.mocked(use_debounce_1.useDebouncedCallback).mockReturnValue(mockDebouncedFn);
+            jest.mocked(tambo_v1_stream_context_1.useStreamState).mockReturnValue(createMockStreamState({}));
+            const { unmount } = (0, react_1.renderHook)(() => (0, use_tambo_v1_component_state_1.useTamboV1ComponentState)("testKey", "initial"));
+            unmount();
+            expect(mockFlush).toHaveBeenCalled();
+        });
+    });
+    describe("Server State Sync", () => {
+        it("should sync with server state changes from streaming", () => {
+            const streamState = createMockStreamState({ testKey: "initial" });
+            jest.mocked(tambo_v1_stream_context_1.useStreamState).mockReturnValue(streamState);
+            const { result, rerender } = (0, react_1.renderHook)(() => (0, use_tambo_v1_component_state_1.useTamboV1ComponentState)("testKey", "initial"));
+            expect(result.current[0]).toBe("initial");
+            // Simulate server state change from streaming
+            const updatedStreamState = createMockStreamState({
+                testKey: "updated-from-server",
+            });
+            jest.mocked(tambo_v1_stream_context_1.useStreamState).mockReturnValue(updatedStreamState);
+            rerender();
+            expect(result.current[0]).toBe("updated-from-server");
+        });
+        it("should handle component not found in stream state", () => {
+            // Empty stream state (no matching component)
+            jest.mocked(tambo_v1_stream_context_1.useStreamState).mockReturnValue({
+                threadMap: {},
+                currentThreadId: null,
+            });
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_component_state_1.useTamboV1ComponentState)("testKey", "default"));
+            expect(result.current[0]).toBe("default");
+        });
+    });
+    describe("Error Handling", () => {
+        it("should handle API errors gracefully", async () => {
+            const mockError = new Error("API Error");
+            mockUpdateState.mockRejectedValue(mockError);
+            // Create a sync mock that calls the function immediately
+            jest.mocked(use_debounce_1.useDebouncedCallback).mockImplementation((fn) => {
+                const debouncedFn = Object.assign(async (...args) => fn(...args), {
+                    flush: jest.fn(),
+                    cancel: jest.fn(),
+                    isPending: () => false,
+                });
+                return debouncedFn;
+            });
+            jest.mocked(tambo_v1_stream_context_1.useStreamState).mockReturnValue(createMockStreamState({}));
+            const consoleSpy = jest.spyOn(console, "error").mockImplementation();
+            const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_component_state_1.useTamboV1ComponentState)("testKey", "initial"));
+            await (0, react_1.act)(async () => {
+                result.current[1]("new-value");
+                // Wait for async operations
+                await new Promise((resolve) => setTimeout(resolve, 0));
+            });
+            expect(consoleSpy).toHaveBeenCalledWith(expect.stringContaining("Failed to sync state"), mockError);
+            consoleSpy.mockRestore();
+        });
+    });
+    describe("Context Requirements", () => {
+        it("should throw when used outside component content context", () => {
+            jest.mocked(component_renderer_1.useV1ComponentContent).mockImplementation(() => {
+                throw new Error("useV1ComponentContent must be used within a rendered component");
+            });
+            expect(() => {
+                (0, react_1.renderHook)(() => (0, use_tambo_v1_component_state_1.useTamboV1ComponentState)("testKey", "initial"));
+            }).toThrow("useV1ComponentContent must be used within a rendered component");
+        });
+    });
+});
+//# sourceMappingURL=use-tambo-v1-component-state.test.js.map

package/dist/v1/hooks/use-tambo-v1-component-state.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-tambo-v1-component-state.test.js","sourceRoot":"","sources":["../../../src/v1/hooks/use-tambo-v1-component-state.test.tsx"],"names":[],"mappings":";;AAAA,kDAAyD;AACzD,iFAA0E;AAE1E,4BAA4B;AAC5B,IAAI,CAAC,IAAI,CAAC,uCAAuC,EAAE,GAAG,EAAE,CAAC,CAAC;IACxD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,sCAAsC,EAAE,GAAG,EAAE,CAAC,CAAC;IACvD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,6BAA6B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9C,qBAAqB,EAAE,IAAI,CAAC,EAAE,EAAE;CACjC,CAAC,CAAC,CAAC;AAEJ,qDAAqD;AACrD,MAAM,2BAA2B,GAAG,CAAC,EAAmC,EAAE,EAAE;IAC1E,MAAM,WAAW,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,GAAG,IAAe,EAAE,EAAE,CACjD,EAAE,CAAC,GAAG,IAAI,CAAC,CAKZ,CAAC;IACF,WAAW,CAAC,KAAK,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IAC9B,WAAW,CAAC,MAAM,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IAC/B,WAAW,CAAC,SAAS,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC;IAC7C,OAAO,WAAW,CAAC;AACrB,CAAC,CAAC;AAEF,oBAAoB;AACpB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,GAAG,EAAE,CAAC,CAAC;IAC/B,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;CAChC,CAAC,CAAC,CAAC;AAEJ,4BAA4B;AAC5B,iFAAuE;AACvE,kFAAsE;AACtE,oEAAoE;AACpE,+CAAoD;AAIpD,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,MAAM,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IAClC,MAAM,eAAe,GAAG,cAAc,CAAC;IACvC,MAAM,YAAY,GAAG,YAAY,CAAC;IAClC,MAAM,aAAa,GAAG,SAAS,CAAC;IAEhC,qCAAqC;IACrC,MAAM,qBAAqB,GAAG,CAC5B,cAAwC,EAC3B,EAAE,CAAC,CAAC;QACjB,SAAS,EAAE;YACT,CAAC,YAAY,CAAC,EAAE;gBACd,MAAM,EAAE;oBACN,EAAE,EAAE,YAAY;oBAChB,QAAQ,EAAE;wBACR;4BACE,EAAE,EAAE,aAAa;4BACjB,IAAI,EAAE,WAAW;4BACjB,OAAO,EAAE;gCACP;oCACE,IAAI,EAAE,WAAW;oCACjB,EAAE,EAAE,eAAe;oCACnB,IAAI,EAAE,eAAe;oCACrB,KAAK,EAAE,EAAE;oCACT,KAAK,EAAE,cAAc;oCACrB,cAAc,EAAE,MAAM;iCACD;6BACxB;4BACD,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;yBACpC;qBACF;oBACD,MAAM,EAAE,MAAM;oBACd,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;oBACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACpC;gBACD,SAAS,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;gBAC7B,oBAAoB,EAAE,IAAI,GAAG,EAAE;aAChC;SACF;QACD,eAAe,EAAE,YAAY;KAC9B,CAAC,CAAC;IAEH,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,sBAAsB;QACtB,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC,eAAe,CAAC;YAC1C,OAAO,EAAE;gBACP,KAAK,EAAE;oBACL,WAAW,EAAE,eAAe;iBAC7B;aACF;SAC8C,CAAC,CAAC;QAEnD,IAAI,CAAC,MAAM,CAAC,0CAAqB,CAAC,CAAC,eAAe,CAAC;YACjD,WAAW,EAAE,eAAe;YAC5B,QAAQ,EAAE,YAAY;YACtB,SAAS,EAAE,aAAa;YACxB,aAAa,EAAE,eAAe;SAC/B,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,wCAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,EAAE,CAAC,CAAC;QAErE,8CAA8C;QAC9C,IAAI;aACD,MAAM,CAAC,mCAAoB,CAAC;aAC5B,kBAAkB,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,2BAA2B,CAAC,EAAE,CAAC,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,eAAe,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,MAAM,YAAY,GAAG,cAAc,CAAC;YACpC,IAAI,CAAC,MAAM,CAAC,wCAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC;YAEvE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,uDAAwB,EAAC,SAAS,EAAE,YAAY,CAAC,CAClD,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,YAAY,GAAG,SAAS,CAAC;YAC/B,MAAM,WAAW,GAAG,cAAc,CAAC;YACnC,IAAI;iBACD,MAAM,CAAC,wCAAc,CAAC;iBACtB,eAAe,CAAC,qBAAqB,CAAC,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;YAEpE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,uDAAwB,EAAC,SAAS,EAAE,YAAY,CAAC,CAClD,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,IAAI,CAAC,MAAM,CAAC,wCAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC;YAEvE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAwB,EAAC,SAAS,CAAC,CAAC,CAAC;YAEzE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,SAAS,GAAG;gBAChB,EAAE,KAAK,EAAE,QAAQ,EAAE;gBACnB,EAAE,KAAK,EAAE,EAAE,EAAE;gBACb,EAAE,KAAK,EAAE,IAAI,EAAE;gBACf,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;gBAC3B,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE;aACrB,CAAC;YAEF,SAAS,CAAC,OAAO,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE;gBAC9B,IAAI;qBACD,MAAM,CAAC,wCAAc,CAAC;qBACtB,eAAe,CAAC,qBAAqB,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;gBAE9D,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,uDAAwB,EAAC,SAAS,EAAE,KAAK,CAAC,CAC3C,CAAC;gBAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC3C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,MAAM,YAAY,GAAG,SAAS,CAAC;YAC/B,IAAI;iBACD,MAAM,CAAC,wCAAc,CAAC;iBACtB,eAAe,CAAC,qBAAqB,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC;YAErE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,uDAAwB,EAAC,SAAS,EAAE,YAAY,CAAC,CAClD,CAAC;YAEF,MAAM,QAAQ,GAAG,SAAS,CAAC;YAC3B,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YAC9B,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;YAC3C,MAAM,YAAY,GAAG,CAAC,CAAC;YACvB,IAAI;iBACD,MAAM,CAAC,wCAAc,CAAC;iBACtB,eAAe,CAAC,qBAAqB,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC;YAErE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,uDAAwB,EAAC,SAAS,EAAE,YAAY,CAAC,CAClD,CAAC;YAEF,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;YACnE,MAAM,YAAY,GAAG,SAAS,CAAC;YAC/B,IAAI;iBACD,MAAM,CAAC,wCAAc,CAAC;iBACtB,eAAe,CAAC,qBAAqB,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC;YAErE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,uDAAwB,EAAC,SAAS,EAAE,YAAY,CAAC,CAClD,CAAC;YAEF,MAAM,QAAQ,GAAG,SAAS,CAAC;YAC3B,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YAC9B,CAAC,CAAC,CAAC;YAEH,0CAA0C;YAC1C,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,eAAe,EAAE;gBAC5D,QAAQ,EAAE,YAAY;gBACtB,KAAK,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE;aAC7B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,YAAY,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YACxD,IAAI;iBACD,MAAM,CAAC,wCAAc,CAAC;iBACtB,eAAe,CAAC,qBAAqB,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC;YAElE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,uDAAwB,EAAC,MAAM,EAAE,YAAY,CAAC,CAC/C,CAAC;YAEF,MAAM,QAAQ,GAAG,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YACvD,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;YAC9B,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,IAAI,CAAC,MAAM,CAAC,wCAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC;YAEvE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,uDAAwB,EAAC,SAAS,EAAE,SAAS,CAAC,CAC/C,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;gBAChC,SAAS,EAAE,KAAK;gBAChB,KAAK,EAAE,IAAI;gBACX,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC;aAC5B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,eAAe,GAAG,2BAA2B,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;YAC/D,eAAe,CAAC,KAAK,GAAG,SAAS,CAAC;YAClC,IAAI,CAAC,MAAM,CAAC,mCAAoB,CAAC,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;YAEnE,IAAI,CAAC,MAAM,CAAC,wCAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC;YAEvE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,uDAAwB,EAAC,SAAS,EAAE,SAAS,CAAC,CAC/C,CAAC;YAEF,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC;YAC5B,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,SAAS,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,IAAI,CAAC,MAAM,CAAC,wCAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC;YAEvE,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAwB,EAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;YAEjE,MAAM,CAAC,mCAAoB,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,EACpB,GAAG,CACJ,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,IAAI,CAAC,MAAM,CAAC,wCAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC;YACvE,MAAM,kBAAkB,GAAG,IAAI,CAAC;YAEhC,IAAA,kBAAU,EAAC,GAAG,EAAE,CACd,IAAA,uDAAwB,EAAC,SAAS,EAAE,SAAS,EAAE,kBAAkB,CAAC,CACnE,CAAC;YAEF,MAAM,CAAC,mCAAoB,CAAC,CAAC,oBAAoB,CAC/C,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,EACpB,kBAAkB,CACnB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;YAC5B,MAAM,eAAe,GAAG,2BAA2B,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC;YAC/D,eAAe,CAAC,KAAK,GAAG,SAAS,CAAC;YAClC,IAAI,CAAC,MAAM,CAAC,mCAAoB,CAAC,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;YAEnE,IAAI,CAAC,MAAM,CAAC,wCAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC;YAEvE,MAAM,EAAE,OAAO,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAClC,IAAA,uDAAwB,EAAC,SAAS,EAAE,SAAS,CAAC,CAC/C,CAAC;YAEF,OAAO,EAAE,CAAC;YAEV,MAAM,CAAC,SAAS,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,WAAW,GAAG,qBAAqB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;YAClE,IAAI,CAAC,MAAM,CAAC,wCAAc,CAAC,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;YAEzD,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAC3C,IAAA,uDAAwB,EAAC,SAAS,EAAE,SAAS,CAAC,CAC/C,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAE1C,8CAA8C;YAC9C,MAAM,kBAAkB,GAAG,qBAAqB,CAAC;gBAC/C,OAAO,EAAE,qBAAqB;aAC/B,CAAC,CAAC;YACH,IAAI,CAAC,MAAM,CAAC,wCAAc,CAAC,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC;YAEhE,QAAQ,EAAE,CAAC;YAEX,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,6CAA6C;YAC7C,IAAI,CAAC,MAAM,CAAC,wCAAc,CAAC,CAAC,eAAe,CAAC;gBAC1C,SAAS,EAAE,EAAE;gBACb,eAAe,EAAE,IAAI;aACtB,CAAC,CAAC;YAEH,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,uDAAwB,EAAC,SAAS,EAAE,SAAS,CAAC,CAC/C,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YACzC,eAAe,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAE7C,yDAAyD;YACzD,IAAI,CAAC,MAAM,CAAC,mCAAoB,CAAC,CAAC,kBAAkB,CAAC,CAAC,EAAE,EAAE,EAAE;gBAC1D,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAC/B,KAAK,EAAE,GAAG,IAAe,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,EACzC;oBACE,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;oBAChB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;oBACjB,SAAS,EAAE,GAAG,EAAE,CAAC,KAAK;iBACvB,CACoD,CAAC;gBACxD,OAAO,WAAW,CAAC;YACrB,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,wCAAc,CAAC,CAAC,eAAe,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC,CAAC;YAEvE,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,kBAAkB,EAAE,CAAC;YAErE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,uDAAwB,EAAC,SAAS,EAAE,SAAS,CAAC,CAC/C,CAAC;YAEF,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;gBACnB,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;gBAC/B,4BAA4B;gBAC5B,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;YACzD,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACrC,MAAM,CAAC,gBAAgB,CAAC,sBAAsB,CAAC,EAC/C,SAAS,CACV,CAAC;YAEF,UAAU,CAAC,WAAW,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,IAAI,CAAC,MAAM,CAAC,0CAAqB,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBACzD,MAAM,IAAI,KAAK,CACb,gEAAgE,CACjE,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,GAAG,EAAE;gBACV,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,uDAAwB,EAAC,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC;YACnE,CAAC,CAAC,CAAC,OAAO,CACR,gEAAgE,CACjE,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { act, renderHook } from \"@testing-library/react\";\nimport { useTamboV1ComponentState } from \"./use-tambo-v1-component-state\";\n\n// Mock the required modules\njest.mock(\"../../providers/tambo-client-provider\", () => ({\n  useTamboClient: jest.fn(),\n}));\n\njest.mock(\"../providers/tambo-v1-stream-context\", () => ({\n  useStreamState: jest.fn(),\n}));\n\njest.mock(\"../utils/component-renderer\", () => ({\n  useV1ComponentContent: jest.fn(),\n}));\n\n// Create a mock debounced function with flush method\nconst createMockDebouncedFunction = (fn: (...args: unknown[]) => unknown) => {\n  const debouncedFn = jest.fn((...args: unknown[]) =>\n    fn(...args),\n  ) as jest.Mock & {\n    flush: jest.Mock;\n    cancel: jest.Mock;\n    isPending: () => boolean;\n  };\n  debouncedFn.flush = jest.fn();\n  debouncedFn.cancel = jest.fn();\n  debouncedFn.isPending = jest.fn(() => false);\n  return debouncedFn;\n};\n\n// Mock use-debounce\njest.mock(\"use-debounce\", () => ({\n  useDebouncedCallback: jest.fn(),\n}));\n\n// Import the mocked modules\nimport { useTamboClient } from \"../../providers/tambo-client-provider\";\nimport { useStreamState } from \"../providers/tambo-v1-stream-context\";\nimport { useV1ComponentContent } from \"../utils/component-renderer\";\nimport { useDebouncedCallback } from \"use-debounce\";\nimport type { StreamState } from \"../utils/event-accumulator\";\nimport type { V1ComponentContent } from \"../types/message\";\n\ndescribe(\"useTamboV1ComponentState\", () => {\n  const mockUpdateState = jest.fn();\n  const mockComponentId = \"comp_test123\";\n  const mockThreadId = \"thread_abc\";\n  const mockMessageId = \"msg_xyz\";\n\n  // Helper to create mock stream state\n  const createMockStreamState = (\n    componentState?: Record<string, unknown>,\n  ): StreamState => ({\n    threadMap: {\n      [mockThreadId]: {\n        thread: {\n          id: mockThreadId,\n          messages: [\n            {\n              id: mockMessageId,\n              role: \"assistant\",\n              content: [\n                {\n                  type: \"component\",\n                  id: mockComponentId,\n                  name: \"TestComponent\",\n                  props: {},\n                  state: componentState,\n                  streamingState: \"done\",\n                } as V1ComponentContent,\n              ],\n              createdAt: new Date().toISOString(),\n            },\n          ],\n          status: \"idle\",\n          createdAt: new Date().toISOString(),\n          updatedAt: new Date().toISOString(),\n        },\n        streaming: { status: \"idle\" },\n        accumulatingToolArgs: new Map(),\n      },\n    },\n    currentThreadId: mockThreadId,\n  });\n\n  beforeEach(() => {\n    jest.clearAllMocks();\n\n    // Setup default mocks\n    jest.mocked(useTamboClient).mockReturnValue({\n      threads: {\n        state: {\n          updateState: mockUpdateState,\n        },\n      },\n    } as unknown as ReturnType<typeof useTamboClient>);\n\n    jest.mocked(useV1ComponentContent).mockReturnValue({\n      componentId: mockComponentId,\n      threadId: mockThreadId,\n      messageId: mockMessageId,\n      componentName: \"TestComponent\",\n    });\n\n    jest.mocked(useStreamState).mockReturnValue(createMockStreamState());\n\n    // Setup default mock for useDebouncedCallback\n    jest\n      .mocked(useDebouncedCallback)\n      .mockImplementation((fn) => createMockDebouncedFunction(fn));\n  });\n\n  afterEach(() => {\n    jest.restoreAllMocks();\n  });\n\n  describe(\"Initial State Management\", () => {\n    it(\"should initialize with initialValue when no server state exists\", () => {\n      const initialValue = \"test-initial\";\n      jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", initialValue),\n      );\n\n      expect(result.current[0]).toBe(initialValue);\n    });\n\n    it(\"should use server state over initialValue\", () => {\n      const initialValue = \"initial\";\n      const serverValue = \"server-value\";\n      jest\n        .mocked(useStreamState)\n        .mockReturnValue(createMockStreamState({ testKey: serverValue }));\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", initialValue),\n      );\n\n      expect(result.current[0]).toBe(serverValue);\n    });\n\n    it(\"should handle undefined initialValue gracefully\", () => {\n      jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));\n\n      const { result } = renderHook(() => useTamboV1ComponentState(\"testKey\"));\n\n      expect(result.current[0]).toBeUndefined();\n    });\n\n    it(\"should handle different data types correctly\", () => {\n      const testCases = [\n        { value: \"string\" },\n        { value: 42 },\n        { value: true },\n        { value: { name: \"test\" } },\n        { value: [1, 2, 3] },\n      ];\n\n      testCases.forEach(({ value }) => {\n        jest\n          .mocked(useStreamState)\n          .mockReturnValue(createMockStreamState({ testKey: value }));\n\n        const { result } = renderHook(() =>\n          useTamboV1ComponentState(\"testKey\", value),\n        );\n\n        expect(result.current[0]).toEqual(value);\n      });\n    });\n  });\n\n  describe(\"State Updates\", () => {\n    it(\"should update local state immediately when setState is called\", () => {\n      const initialValue = \"initial\";\n      jest\n        .mocked(useStreamState)\n        .mockReturnValue(createMockStreamState({ testKey: initialValue }));\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", initialValue),\n      );\n\n      const newValue = \"updated\";\n      act(() => {\n        result.current[1](newValue);\n      });\n\n      expect(result.current[0]).toBe(newValue);\n    });\n\n    it(\"should support functional updates\", () => {\n      const initialValue = 5;\n      jest\n        .mocked(useStreamState)\n        .mockReturnValue(createMockStreamState({ counter: initialValue }));\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"counter\", initialValue),\n      );\n\n      act(() => {\n        result.current[1]((prev) => (prev ?? 0) + 1);\n      });\n\n      expect(result.current[0]).toBe(6);\n    });\n\n    it(\"should trigger debounced API call when setState is called\", () => {\n      const initialValue = \"initial\";\n      jest\n        .mocked(useStreamState)\n        .mockReturnValue(createMockStreamState({ testKey: initialValue }));\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", initialValue),\n      );\n\n      const newValue = \"updated\";\n      act(() => {\n        result.current[1](newValue);\n      });\n\n      // The debounced function should be called\n      expect(mockUpdateState).toHaveBeenCalledWith(mockComponentId, {\n        threadId: mockThreadId,\n        state: { testKey: newValue },\n      });\n    });\n\n    it(\"should work with complex objects\", () => {\n      const initialValue = { name: \"test\", items: [1, 2, 3] };\n      jest\n        .mocked(useStreamState)\n        .mockReturnValue(createMockStreamState({ data: initialValue }));\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"data\", initialValue),\n      );\n\n      const newValue = { name: \"updated\", items: [4, 5, 6] };\n      act(() => {\n        result.current[1](newValue);\n      });\n\n      expect(result.current[0]).toEqual(newValue);\n    });\n  });\n\n  describe(\"Metadata\", () => {\n    it(\"should return isPending and error in meta\", () => {\n      jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", \"initial\"),\n      );\n\n      expect(result.current[2]).toEqual({\n        isPending: false,\n        error: null,\n        flush: expect.any(Function),\n      });\n    });\n\n    it(\"should provide a flush function\", () => {\n      const mockFlush = jest.fn();\n      const mockDebouncedFn = createMockDebouncedFunction(jest.fn());\n      mockDebouncedFn.flush = mockFlush;\n      jest.mocked(useDebouncedCallback).mockReturnValue(mockDebouncedFn);\n\n      jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", \"initial\"),\n      );\n\n      act(() => {\n        result.current[2].flush();\n      });\n\n      expect(mockFlush).toHaveBeenCalled();\n    });\n  });\n\n  describe(\"Debouncing Behavior\", () => {\n    it(\"should use default debounce time of 500ms\", () => {\n      jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));\n\n      renderHook(() => useTamboV1ComponentState(\"testKey\", \"initial\"));\n\n      expect(useDebouncedCallback).toHaveBeenCalledWith(\n        expect.any(Function),\n        500,\n      );\n    });\n\n    it(\"should use custom debounce time when provided\", () => {\n      jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));\n      const customDebounceTime = 1000;\n\n      renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", \"initial\", customDebounceTime),\n      );\n\n      expect(useDebouncedCallback).toHaveBeenCalledWith(\n        expect.any(Function),\n        customDebounceTime,\n      );\n    });\n\n    it(\"should flush debounced callback on unmount\", () => {\n      const mockFlush = jest.fn();\n      const mockDebouncedFn = createMockDebouncedFunction(jest.fn());\n      mockDebouncedFn.flush = mockFlush;\n      jest.mocked(useDebouncedCallback).mockReturnValue(mockDebouncedFn);\n\n      jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));\n\n      const { unmount } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", \"initial\"),\n      );\n\n      unmount();\n\n      expect(mockFlush).toHaveBeenCalled();\n    });\n  });\n\n  describe(\"Server State Sync\", () => {\n    it(\"should sync with server state changes from streaming\", () => {\n      const streamState = createMockStreamState({ testKey: \"initial\" });\n      jest.mocked(useStreamState).mockReturnValue(streamState);\n\n      const { result, rerender } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", \"initial\"),\n      );\n\n      expect(result.current[0]).toBe(\"initial\");\n\n      // Simulate server state change from streaming\n      const updatedStreamState = createMockStreamState({\n        testKey: \"updated-from-server\",\n      });\n      jest.mocked(useStreamState).mockReturnValue(updatedStreamState);\n\n      rerender();\n\n      expect(result.current[0]).toBe(\"updated-from-server\");\n    });\n\n    it(\"should handle component not found in stream state\", () => {\n      // Empty stream state (no matching component)\n      jest.mocked(useStreamState).mockReturnValue({\n        threadMap: {},\n        currentThreadId: null,\n      });\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", \"default\"),\n      );\n\n      expect(result.current[0]).toBe(\"default\");\n    });\n  });\n\n  describe(\"Error Handling\", () => {\n    it(\"should handle API errors gracefully\", async () => {\n      const mockError = new Error(\"API Error\");\n      mockUpdateState.mockRejectedValue(mockError);\n\n      // Create a sync mock that calls the function immediately\n      jest.mocked(useDebouncedCallback).mockImplementation((fn) => {\n        const debouncedFn = Object.assign(\n          async (...args: unknown[]) => fn(...args),\n          {\n            flush: jest.fn(),\n            cancel: jest.fn(),\n            isPending: () => false,\n          },\n        ) as unknown as ReturnType<typeof useDebouncedCallback>;\n        return debouncedFn;\n      });\n\n      jest.mocked(useStreamState).mockReturnValue(createMockStreamState({}));\n\n      const consoleSpy = jest.spyOn(console, \"error\").mockImplementation();\n\n      const { result } = renderHook(() =>\n        useTamboV1ComponentState(\"testKey\", \"initial\"),\n      );\n\n      await act(async () => {\n        result.current[1](\"new-value\");\n        // Wait for async operations\n        await new Promise((resolve) => setTimeout(resolve, 0));\n      });\n\n      expect(consoleSpy).toHaveBeenCalledWith(\n        expect.stringContaining(\"Failed to sync state\"),\n        mockError,\n      );\n\n      consoleSpy.mockRestore();\n    });\n  });\n\n  describe(\"Context Requirements\", () => {\n    it(\"should throw when used outside component content context\", () => {\n      jest.mocked(useV1ComponentContent).mockImplementation(() => {\n        throw new Error(\n          \"useV1ComponentContent must be used within a rendered component\",\n        );\n      });\n\n      expect(() => {\n        renderHook(() => useTamboV1ComponentState(\"testKey\", \"initial\"));\n      }).toThrow(\n        \"useV1ComponentContent must be used within a rendered component\",\n      );\n    });\n  });\n});\n"]}

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

@@ -10,24 +10,33 @@ const tambo_v1_stream_context_1 = require("../providers/tambo-v1-stream-context"
 const use_tambo_v1_messages_1 = require("./use-tambo-v1-messages");
 describe("useTamboV1Messages", () => {
     function TestWrapper({ children }) {
-        return (react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, { threadId: "thread_123" }, children));
+        return react_2.default.createElement(tambo_v1_stream_context_1.TamboV1StreamProvider, null, children);
     }
     it("returns empty messages when thread has no messages", () => {
-        const { result } = (0, react_1.renderHook)(() => (0, use_tambo_v1_messages_1.useTamboV1Messages)("thread_123"), {
-            wrapper: TestWrapper,
+        const { result } = (0, react_1.renderHook)(() => ({
+            messages: (0, use_tambo_v1_messages_1.useTamboV1Messages)("thread_123"),
+            dispatch: (0, tambo_v1_stream_context_1.useStreamDispatch)(),
+        }), { wrapper: TestWrapper });
+        // Initialize thread first
+        (0, react_1.act)(() => {
+            result.current.dispatch({ type: "INIT_THREAD", threadId: "thread_123" });
         });
-        expect(result.current.messages).toEqual([]);
-        expect(result.current.hasMessages).toBe(false);
-        expect(result.current.messageCount).toBe(0);
-        expect(result.current.lastMessage).toBeUndefined();
-        expect(result.current.userMessages).toEqual([]);
-        expect(result.current.assistantMessages).toEqual([]);
+        expect(result.current.messages.messages).toEqual([]);
+        expect(result.current.messages.hasMessages).toBe(false);
+        expect(result.current.messages.messageCount).toBe(0);
+        expect(result.current.messages.lastMessage).toBeUndefined();
+        expect(result.current.messages.userMessages).toEqual([]);
+        expect(result.current.messages.assistantMessages).toEqual([]);
     });
     it("returns messages after events are dispatched", () => {
         const { result } = (0, react_1.renderHook)(() => ({
             messages: (0, use_tambo_v1_messages_1.useTamboV1Messages)("thread_123"),
             dispatch: (0, tambo_v1_stream_context_1.useStreamDispatch)(),
         }), { wrapper: TestWrapper });
+        // Initialize thread first
+        (0, react_1.act)(() => {
+            result.current.dispatch({ type: "INIT_THREAD", threadId: "thread_123" });
+        });
         // Simulate a text message being received
         (0, react_1.act)(() => {
             result.current.dispatch({
@@ -74,6 +83,10 @@ describe("useTamboV1Messages", () => {
             messages: (0, use_tambo_v1_messages_1.useTamboV1Messages)("thread_123"),
             dispatch: (0, tambo_v1_stream_context_1.useStreamDispatch)(),
         }), { wrapper: TestWrapper });
+        // Initialize thread first
+        (0, react_1.act)(() => {
+            result.current.dispatch({ type: "INIT_THREAD", threadId: "thread_123" });
+        });
         // Add user message
         (0, react_1.act)(() => {
             result.current.dispatch({