@tambo-ai/react 0.65.1 → 0.65.3

package/dist/hooks/use-streaming-props.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-streaming-props.js","sourceRoot":"","sources":["../../src/hooks/use-streaming-props.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;AAwBb,wDA2BC;AAlDD,iCAAkC;AAElC;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,SAAgB,sBAAsB,CACpC,YAA2B,EAC3B,QAA4B,EAC5B,cAA0B;IAE1B,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,YAAY,EAAE,CAAC;YACjB,IAAI,YAAY,GAAG,KAAK,CAAC;YACzB,MAAM,OAAO,GAAe,EAAE,CAAC;YAE/B,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;gBACtD,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC;oBACvD,YAAY,GAAG,IAAI,CAAC;oBACpB,OAAO,CAAC,GAAc,CAAC,GAAG,KAAmB,CAAC;gBAChD,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,IAAI,YAAY,EAAE,CAAC;gBACjB,QAAQ,CAAC;oBACP,GAAG,YAAY;oBACf,GAAG,OAAO;iBACX,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,sCAAsC;QACtC,uDAAuD;IACzD,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;AACzC,CAAC","sourcesContent":["\"use client\";\nimport { useEffect } from \"react\";\n\n/**\n * A helper hook that automatically updates Tambo component state when specified props change.\n *\n * This hook streamlines the common pattern of updating component state when receiving new\n * streamed values from Tambo, eliminating the need to write repetitive useEffect code.\n * @param currentState - The current state object from useTamboComponentState\n * @param setState - The setState function from useTamboComponentState\n * @param streamingProps - An object mapping state keys to prop values that should update the state\n * @example\n * ```tsx\n * // Instead of writing a complex useEffect:\n * const [emailState, setEmailState] = useTamboComponentState(\"email\", initialState);\n *\n * // Simply use:\n * useTamboStreamingProps(emailState, setEmailState, {\n *   subject: aiGeneratedSubject,\n *   body: aiGeneratedBody,\n *   usersEmail: usersEmail\n * });\n * ```\n */\nexport function useTamboStreamingProps<T extends Record<string, any>>(\n  currentState: T | undefined,\n  setState: (state: T) => void,\n  streamingProps: Partial<T>,\n) {\n  useEffect(() => {\n    if (currentState) {\n      let shouldUpdate = false;\n      const updates: Partial<T> = {};\n\n      Object.entries(streamingProps).forEach(([key, value]) => {\n        if (value !== undefined && value !== currentState[key]) {\n          shouldUpdate = true;\n          updates[key as keyof T] = value as T[keyof T];\n        }\n      });\n\n      if (shouldUpdate) {\n        setState({\n          ...currentState,\n          ...updates,\n        });\n      }\n    }\n    // Only run when streamingProps change\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n  }, [...Object.values(streamingProps)]);\n}\n"]}
+{"version":3,"file":"use-streaming-props.js","sourceRoot":"","sources":["../../src/hooks/use-streaming-props.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;AAab,wDA2BC;AAtCD,iCAAkC;AAElC;;;;;;;;GAQG;AACH,SAAgB,sBAAsB,CACpC,YAA2B,EAC3B,QAA4B,EAC5B,cAA0B;IAE1B,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,YAAY,EAAE,CAAC;YACjB,IAAI,YAAY,GAAG,KAAK,CAAC;YACzB,MAAM,OAAO,GAAe,EAAE,CAAC;YAE/B,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;gBACtD,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC;oBACvD,YAAY,GAAG,IAAI,CAAC;oBACpB,OAAO,CAAC,GAAc,CAAC,GAAG,KAAmB,CAAC;gBAChD,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,IAAI,YAAY,EAAE,CAAC;gBACjB,QAAQ,CAAC;oBACP,GAAG,YAAY;oBACf,GAAG,OAAO;iBACX,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,sCAAsC;QACtC,uDAAuD;IACzD,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC;AACzC,CAAC","sourcesContent":["\"use client\";\n\nimport { useEffect } from \"react\";\n\n/**\n * Low-level helper that merges streamed props into state.\n * @deprecated Use `useTamboComponentState` with `setFromProp` instead.\n * This hook will be removed in 1.0.0.\n * @see {@link https://docs.tambo.co/concepts/streaming/streaming-props}\n * @param currentState - Current state object\n * @param setState - State setter function\n * @param streamingProps - Props to merge into state when they change\n */\nexport function useTamboStreamingProps<T extends Record<string, any>>(\n  currentState: T | undefined,\n  setState: (state: T) => void,\n  streamingProps: Partial<T>,\n) {\n  useEffect(() => {\n    if (currentState) {\n      let shouldUpdate = false;\n      const updates: Partial<T> = {};\n\n      Object.entries(streamingProps).forEach(([key, value]) => {\n        if (value !== undefined && value !== currentState[key]) {\n          shouldUpdate = true;\n          updates[key as keyof T] = value as T[keyof T];\n        }\n      });\n\n      if (shouldUpdate) {\n        setState({\n          ...currentState,\n          ...updates,\n        });\n      }\n    }\n    // Only run when streamingProps change\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n  }, [...Object.values(streamingProps)]);\n}\n"]}

package/dist/hooks/use-suggestions.d.ts

@@ -33,7 +33,7 @@ export interface useTamboSuggestionsResultInternal {
     /** Result and network state for generating suggestions */
     generateResult: UseTamboMutationResult<TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined, Error, AbortController>;
     /** The full suggestions query object from React Query */
-    suggestionsResult: UseTamboQueryResult<TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined, Error>;
+    suggestionsResult: UseTamboQueryResult<TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined>;
 }
 type useTamboSuggestionsResult = CombinedMutationResult<any, Error> & useTamboSuggestionsResultInternal;
 /**

package/dist/hooks/use-suggestions.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"use-suggestions.d.ts","sourceRoot":"","sources":["../../src/hooks/use-suggestions.ts"],"names":[],"mappings":"AACA,OAAO,OAAO,MAAM,0BAA0B,CAAC;AAY/C,OAAO,EACL,sBAAsB,EAEvB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EACL,sBAAsB,EACtB,mBAAmB,EAGpB,MAAM,qBAAqB,CAAC;AAE7B;;GAEG;AACH,MAAM,WAAW,0BAA0B;IACzC,kEAAkE;IAClE,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED;;GAEG;AACH,MAAM,WAAW,iCAAiC;IAChD,4EAA4E;IAC5E,WAAW,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;IAC/C,8CAA8C;IAC9C,oBAAoB,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC;;;;OAIG;IACH,MAAM,EAAE,CAAC,aAAa,EAAE;QACtB,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;QAC5C,YAAY,CAAC,EAAE,OAAO,CAAC;KACxB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAEpB,0DAA0D;IAC1D,YAAY,EAAE,sBAAsB,CAClC,IAAI,EACJ,KAAK,EACL;QAAE,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;QAAC,YAAY,CAAC,EAAE,OAAO,CAAA;KAAE,CACxE,CAAC;IAEF,0DAA0D;IAC1D,cAAc,EAAE,sBAAsB,CACpC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,0BAA0B,GAAG,SAAS,EACvE,KAAK,EACL,eAAe,CAChB,CAAC;IAEF,yDAAyD;IACzD,iBAAiB,EAAE,mBAAmB,CACpC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,0BAA0B,GAAG,SAAS,EACvE,KAAK,CACN,CAAC;CACH;AAED,KAAK,yBAAyB,GAAG,sBAAsB,CAAC,GAAG,EAAE,KAAK,CAAC,GACjE,iCAAiC,CAAC;AAEpC;;;;GAIG;AACH,wBAAgB,mBAAmB,CACjC,OAAO,GAAE,0BAA+B,GACvC,yBAAyB,CAsI3B"}
+{"version":3,"file":"use-suggestions.d.ts","sourceRoot":"","sources":["../../src/hooks/use-suggestions.ts"],"names":[],"mappings":"AACA,OAAO,OAAO,MAAM,0BAA0B,CAAC;AAY/C,OAAO,EACL,sBAAsB,EAEvB,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EACL,sBAAsB,EACtB,mBAAmB,EAGpB,MAAM,qBAAqB,CAAC;AAE7B;;GAEG;AACH,MAAM,WAAW,0BAA0B;IACzC,kEAAkE;IAClE,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAED;;GAEG;AACH,MAAM,WAAW,iCAAiC;IAChD,4EAA4E;IAC5E,WAAW,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC;IAC/C,8CAA8C;IAC9C,oBAAoB,EAAE,MAAM,GAAG,IAAI,CAAC;IACpC;;;;OAIG;IACH,MAAM,EAAE,CAAC,aAAa,EAAE;QACtB,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;QAC5C,YAAY,CAAC,EAAE,OAAO,CAAC;KACxB,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAEpB,0DAA0D;IAC1D,YAAY,EAAE,sBAAsB,CAClC,IAAI,EACJ,KAAK,EACL;QAAE,UAAU,EAAE,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;QAAC,YAAY,CAAC,EAAE,OAAO,CAAA;KAAE,CACxE,CAAC;IAEF,0DAA0D;IAC1D,cAAc,EAAE,sBAAsB,CACpC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,0BAA0B,GAAG,SAAS,EACvE,KAAK,EACL,eAAe,CAChB,CAAC;IAEF,yDAAyD;IACzD,iBAAiB,EAAE,mBAAmB,CACpC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,0BAA0B,GAAG,SAAS,CACxE,CAAC;CACH;AAED,KAAK,yBAAyB,GAAG,sBAAsB,CAAC,GAAG,EAAE,KAAK,CAAC,GACjE,iCAAiC,CAAC;AAEpC;;;;GAIG;AACH,wBAAgB,mBAAmB,CACjC,OAAO,GAAE,0BAA+B,GACvC,yBAAyB,CAsI3B"}

package/dist/hooks/use-suggestions.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-suggestions.js","sourceRoot":"","sources":["../../src/hooks/use-suggestions.ts"],"names":[],"mappings":";AAAA,YAAY,CAAC;;AAgFb,kDAwIC;AAtND,iCAA4C;AAC5C,sFAAmE;AACnE,4DAAwD;AACxD,8EAAoE;AACpE,gEAAuD;AACvD,kFAAwE;AACxE,0FAGkD;AAClD,8EAAoE;AACpE,qDAG6B;AAC7B,+CAA0D;AAC1D,2DAK6B;AAoD7B;;;;GAIG;AACH,SAAgB,mBAAmB,CACjC,UAAsC,EAAE;IAExC,MAAM,EAAE,cAAc,GAAG,CAAC,EAAE,GAAG,OAAO,CAAC;IACvC,MAAM,EAAE,MAAM,EAAE,eAAe,EAAE,GAAG,IAAA,sCAAc,GAAE,CAAC;IACrD,MAAM,EAAE,iBAAiB,EAAE,GAAG,IAAA,yBAAQ,GAAE,CAAC;IACzC,MAAM,WAAW,GAAG,IAAA,sCAAc,GAAE,CAAC;IACrC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,yBAAyB,EAAE,GAC9D,IAAA,0CAAgB,GAAE,CAAC;IAErB,MAAM,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,GAAG,IAAA,gBAAQ,EAE9D,IAAI,CAAC,CAAC;IACR,MAAM,EAAE,QAAQ,EAAE,aAAa,EAAE,GAAG,IAAA,iDAAmB,GAAE,CAAC;IAE1D,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAClE,MAAM,iBAAiB,GAAG,aAAa,EAAE,IAAI,KAAK,WAAW,CAAC;IAC9D,MAAM,eAAe,GAAG,aAAa,EAAE,EAAE,CAAC;IAE1C,qDAAqD;IACrD,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,uBAAuB,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC;IAEtB,MAAM,yBAAyB,GAC7B,eAAe,IAAI,iBAAiB,IAAI,IAAA,yCAAW,EAAC,eAAe,CAAC,CAAC;IACvE,4EAA4E;IAC5E,MAAM,iBAAiB,GAAG,IAAA,iCAAa,EAAC;QACtC,2EAA2E;QAC3E,4DAA4D;QAC5D,QAAQ,EAAE;YACR,aAAa;YACb,yBAAyB,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI;SACnD;QACD,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,yBAAyB,EAAE,CAAC;gBAC/B,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,8CAA8C;YAC9C,MAAM,UAAU,GAAG,IAAA,iCAAsB,EACvC,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;YAEF,OAAO,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CACxD,eAAe,EACf;gBACE,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,cAAc;gBACd,mBAAmB,EAAE,UAAU;aAChC,CACF,CAAC;QACJ,CAAC;QACD,2DAA2D;QAC3D,OAAO,EAAE,OAAO,CAAC,eAAe,IAAI,iBAAiB,CAAC;QACtD,6CAA6C;QAC7C,oBAAoB,EAAE,KAAK;QAC3B,kBAAkB,EAAE,KAAK;QACzB,yBAAyB;QACzB,KAAK,EAAE,KAAK;KACb,CAAC,CAAC;IAEH,6BAA6B;IAC7B,MAAM,mBAAmB,GAAG,IAAA,oCAAgB,EAI1C;QACA,UAAU,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,YAAY,GAAG,KAAK,EAAE,EAAE,EAAE;YACzD,MAAM,UAAU,GAAG,IAAA,8BAAa,EAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;YAChE,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;gBACxB,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;oBACrB,MAAM,UAAU,CAAC,KAAK,CAAC;gBACzB,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,kDAAoB,CAAC,UAAU,CAAC,CAAC;YACnD,CAAC;YAED,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,iBAAiB,CAAC,UAAU,CAAC,cAAc,EAAE;oBACjD,QAAQ,EAAE,MAAM,CAAC,EAAE;iBACpB,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;YAC3C,CAAC;YACD,uBAAuB,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QACzC,CAAC;KACF,CAAC,CAAC;IAEH,gCAAgC;IAChC,MAAM,qBAAqB,GAAG,IAAA,oCAAgB,EAI5C;QACA,UAAU,EAAE,KAAK,EAAE,eAAgC,EAAE,EAAE;YACrD,IAAI,CAAC,yBAAyB,EAAE,CAAC;gBAC/B,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,8CAA8C;YAC9C,MAAM,UAAU,GAAG,IAAA,iCAAsB,EACvC,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;YACF,OAAO,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CACxD,eAAe,EACf;gBACE,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,cAAc;gBACd,mBAAmB,EAAE,UAAU;aAChC,EACD,EAAE,MAAM,EAAE,eAAe,CAAC,MAAM,EAAE,CACnC,CAAC;QACJ,CAAC;QACD,yBAAyB;QACzB,KAAK,EAAE,KAAK;KACb,CAAC,CAAC;IAEH,mEAAmE;IACnE,8DAA8D;IAC9D,MAAM,WAAW,GAAG,iBAAiB;QACnC,CAAC,CAAC,CAAC,iBAAiB,CAAC,IAAI,IAAI,qBAAqB,CAAC,IAAI,IAAI,EAAE,CAAC;QAC9D,CAAC,CAAC,EAAE,CAAC;IAEP,OAAO;QACL,WAAW;QACX,MAAM,EAAE,mBAAmB,CAAC,WAAW;QACvC,oBAAoB;QACpB,YAAY,EAAE,mBAAmB;QACjC,cAAc,EAAE,qBAAqB;QACrC,iBAAiB;QACjB,GAAG,IAAA,oCAAsB,EAAC,mBAAmB,EAAE,qBAAqB,CAAC;KACtE,CAAC;AACJ,CAAC","sourcesContent":["\"use client\";\nimport TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { useEffect, useState } from \"react\";\nimport { isIdleStage } from \"../model/generate-component-response\";\nimport { validateInput } from \"../model/validate-input\";\nimport { useTamboClient } from \"../providers/tambo-client-provider\";\nimport { useTambo } from \"../providers/tambo-provider\";\nimport { useTamboRegistry } from \"../providers/tambo-registry-provider\";\nimport {\n  INPUT_ERROR_MESSAGES,\n  useTamboThreadInput,\n} from \"../providers/tambo-thread-input-provider\";\nimport { useTamboThread } from \"../providers/tambo-thread-provider\";\nimport {\n  CombinedMutationResult,\n  combineMutationResults,\n} from \"../util/query-utils\";\nimport { getAvailableComponents } from \"../util/registry\";\nimport {\n  UseTamboMutationResult,\n  UseTamboQueryResult,\n  useTamboMutation,\n  useTamboQuery,\n} from \"./react-query-hooks\";\n\n/**\n * Configuration options for the useTamboSuggestions hook\n */\nexport interface useTamboSuggestionsOptions {\n  /** Maximum number of suggestions to generate (1-10, default 3) */\n  maxSuggestions?: number;\n}\n\n/**\n * Return value interface for useTamboSuggestions hook\n */\nexport interface useTamboSuggestionsResultInternal {\n  /** List of available suggestions (also available in generateResult.data) */\n  suggestions: TamboAI.Beta.Threads.Suggestion[];\n  /** ID of the currently selected suggestion */\n  selectedSuggestionId: string | null;\n  /**\n   * Accept and apply a suggestion (also available in acceptResult.mutateAsync)\n   * @param suggestion - The suggestion to accept\n   * @param shouldSubmit - Whether to automatically submit after accepting (default: false)\n   */\n  accept: (acceptOptions: {\n    suggestion: TamboAI.Beta.Threads.Suggestion;\n    shouldSubmit?: boolean;\n  }) => Promise<void>;\n\n  /** Result and network state for accepting a suggestion */\n  acceptResult: UseTamboMutationResult<\n    void,\n    Error,\n    { suggestion: TamboAI.Beta.Threads.Suggestion; shouldSubmit?: boolean }\n  >;\n\n  /** Result and network state for generating suggestions */\n  generateResult: UseTamboMutationResult<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error,\n    AbortController\n  >;\n\n  /** The full suggestions query object from React Query */\n  suggestionsResult: UseTamboQueryResult<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error\n  >;\n}\n\ntype useTamboSuggestionsResult = CombinedMutationResult<any, Error> &\n  useTamboSuggestionsResultInternal;\n\n/**\n * Hook for managing Tambo AI suggestions in a thread\n * @param options - Configuration options for suggestion generation\n * @returns Object containing suggestions state and control functions\n */\nexport function useTamboSuggestions(\n  options: useTamboSuggestionsOptions = {},\n): useTamboSuggestionsResult {\n  const { maxSuggestions = 3 } = options;\n  const { thread, generationStage } = useTamboThread();\n  const { sendThreadMessage } = useTambo();\n  const tamboClient = useTamboClient();\n  const { componentList, toolRegistry, componentToolAssociations } =\n    useTamboRegistry();\n\n  const [selectedSuggestionId, setSelectedSuggestionId] = useState<\n    string | null\n  >(null);\n  const { setValue: setInputValue } = useTamboThreadInput();\n\n  const latestMessage = thread.messages[thread.messages.length - 1];\n  const isLatestFromTambo = latestMessage?.role === \"assistant\";\n  const latestMessageId = latestMessage?.id;\n\n  // Reset selected suggestion when the message changes\n  useEffect(() => {\n    setSelectedSuggestionId(null);\n  }, [latestMessageId]);\n\n  const shouldGenerateSuggestions =\n    latestMessageId && isLatestFromTambo && isIdleStage(generationStage);\n  // Use React Query to fetch suggestions when a new hydra message is received\n  const suggestionsResult = useTamboQuery({\n    // Make sure the query key changes when the message changes, so that we are\n    // always generating suggestions when there is a new message\n    queryKey: [\n      \"suggestions\",\n      shouldGenerateSuggestions ? latestMessageId : null,\n    ],\n    queryFn: async () => {\n      if (!shouldGenerateSuggestions) {\n        return [];\n      }\n\n      // Get registered components from the registry\n      const components = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n\n      return await tamboClient.beta.threads.suggestions.generate(\n        latestMessageId,\n        {\n          id: thread.id,\n          maxSuggestions,\n          availableComponents: components,\n        },\n      );\n    },\n    // Only run the query if we have a valid message from hydra\n    enabled: Boolean(latestMessageId && isLatestFromTambo),\n    // Don't refetch on window focus or reconnect\n    refetchOnWindowFocus: false,\n    refetchOnReconnect: false,\n    // Don't retry on failure\n    retry: false,\n  });\n\n  // Accept suggestion mutation\n  const acceptMutationState = useTamboMutation<\n    void,\n    Error,\n    { suggestion: TamboAI.Beta.Threads.Suggestion; shouldSubmit?: boolean }\n  >({\n    mutationFn: async ({ suggestion, shouldSubmit = false }) => {\n      const validation = validateInput(suggestion.detailedSuggestion);\n      if (!validation.isValid) {\n        if (validation.error) {\n          throw validation.error;\n        }\n        throw new Error(INPUT_ERROR_MESSAGES.VALIDATION);\n      }\n\n      if (shouldSubmit) {\n        await sendThreadMessage(validation.sanitizedInput, {\n          threadId: thread.id,\n        });\n      } else {\n        setInputValue(validation.sanitizedInput);\n      }\n      setSelectedSuggestionId(suggestion.id);\n    },\n  });\n\n  // Generate suggestions mutation\n  const generateMutationState = useTamboMutation<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error,\n    AbortController\n  >({\n    mutationFn: async (abortController: AbortController) => {\n      if (!shouldGenerateSuggestions) {\n        return undefined;\n      }\n\n      // Get registered components from the registry\n      const components = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n      return await tamboClient.beta.threads.suggestions.generate(\n        latestMessageId,\n        {\n          id: thread.id,\n          maxSuggestions,\n          availableComponents: components,\n        },\n        { signal: abortController.signal },\n      );\n    },\n    // Don't retry on failure\n    retry: false,\n  });\n\n  // Use the query data if available, otherwise use the mutation data\n  // Only return suggestions if the latest message is from hydra\n  const suggestions = isLatestFromTambo\n    ? (suggestionsResult.data ?? generateMutationState.data ?? [])\n    : [];\n\n  return {\n    suggestions,\n    accept: acceptMutationState.mutateAsync,\n    selectedSuggestionId,\n    acceptResult: acceptMutationState,\n    generateResult: generateMutationState,\n    suggestionsResult,\n    ...combineMutationResults(acceptMutationState, generateMutationState),\n  };\n}\n"]}
+{"version":3,"file":"use-suggestions.js","sourceRoot":"","sources":["../../src/hooks/use-suggestions.ts"],"names":[],"mappings":";AAAA,YAAY,CAAC;;AA+Eb,kDAwIC;AArND,iCAA4C;AAC5C,sFAAmE;AACnE,4DAAwD;AACxD,8EAAoE;AACpE,gEAAuD;AACvD,kFAAwE;AACxE,0FAGkD;AAClD,8EAAoE;AACpE,qDAG6B;AAC7B,+CAA0D;AAC1D,2DAK6B;AAmD7B;;;;GAIG;AACH,SAAgB,mBAAmB,CACjC,UAAsC,EAAE;IAExC,MAAM,EAAE,cAAc,GAAG,CAAC,EAAE,GAAG,OAAO,CAAC;IACvC,MAAM,EAAE,MAAM,EAAE,eAAe,EAAE,GAAG,IAAA,sCAAc,GAAE,CAAC;IACrD,MAAM,EAAE,iBAAiB,EAAE,GAAG,IAAA,yBAAQ,GAAE,CAAC;IACzC,MAAM,WAAW,GAAG,IAAA,sCAAc,GAAE,CAAC;IACrC,MAAM,EAAE,aAAa,EAAE,YAAY,EAAE,yBAAyB,EAAE,GAC9D,IAAA,0CAAgB,GAAE,CAAC;IAErB,MAAM,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,GAAG,IAAA,gBAAQ,EAE9D,IAAI,CAAC,CAAC;IACR,MAAM,EAAE,QAAQ,EAAE,aAAa,EAAE,GAAG,IAAA,iDAAmB,GAAE,CAAC;IAE1D,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAClE,MAAM,iBAAiB,GAAG,aAAa,EAAE,IAAI,KAAK,WAAW,CAAC;IAC9D,MAAM,eAAe,GAAG,aAAa,EAAE,EAAE,CAAC;IAE1C,qDAAqD;IACrD,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,uBAAuB,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC;IAEtB,MAAM,yBAAyB,GAC7B,eAAe,IAAI,iBAAiB,IAAI,IAAA,yCAAW,EAAC,eAAe,CAAC,CAAC;IACvE,4EAA4E;IAC5E,MAAM,iBAAiB,GAAG,IAAA,iCAAa,EAAC;QACtC,2EAA2E;QAC3E,4DAA4D;QAC5D,QAAQ,EAAE;YACR,aAAa;YACb,yBAAyB,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI;SACnD;QACD,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,yBAAyB,EAAE,CAAC;gBAC/B,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,8CAA8C;YAC9C,MAAM,UAAU,GAAG,IAAA,iCAAsB,EACvC,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;YAEF,OAAO,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CACxD,eAAe,EACf;gBACE,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,cAAc;gBACd,mBAAmB,EAAE,UAAU;aAChC,CACF,CAAC;QACJ,CAAC;QACD,2DAA2D;QAC3D,OAAO,EAAE,OAAO,CAAC,eAAe,IAAI,iBAAiB,CAAC;QACtD,6CAA6C;QAC7C,oBAAoB,EAAE,KAAK;QAC3B,kBAAkB,EAAE,KAAK;QACzB,yBAAyB;QACzB,KAAK,EAAE,KAAK;KACb,CAAC,CAAC;IAEH,6BAA6B;IAC7B,MAAM,mBAAmB,GAAG,IAAA,oCAAgB,EAI1C;QACA,UAAU,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,YAAY,GAAG,KAAK,EAAE,EAAE,EAAE;YACzD,MAAM,UAAU,GAAG,IAAA,8BAAa,EAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC;YAChE,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;gBACxB,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;oBACrB,MAAM,UAAU,CAAC,KAAK,CAAC;gBACzB,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,kDAAoB,CAAC,UAAU,CAAC,CAAC;YACnD,CAAC;YAED,IAAI,YAAY,EAAE,CAAC;gBACjB,MAAM,iBAAiB,CAAC,UAAU,CAAC,cAAc,EAAE;oBACjD,QAAQ,EAAE,MAAM,CAAC,EAAE;iBACpB,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,aAAa,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;YAC3C,CAAC;YACD,uBAAuB,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QACzC,CAAC;KACF,CAAC,CAAC;IAEH,gCAAgC;IAChC,MAAM,qBAAqB,GAAG,IAAA,oCAAgB,EAI5C;QACA,UAAU,EAAE,KAAK,EAAE,eAAgC,EAAE,EAAE;YACrD,IAAI,CAAC,yBAAyB,EAAE,CAAC;gBAC/B,OAAO,SAAS,CAAC;YACnB,CAAC;YAED,8CAA8C;YAC9C,MAAM,UAAU,GAAG,IAAA,iCAAsB,EACvC,aAAa,EACb,YAAY,EACZ,yBAAyB,CAC1B,CAAC;YACF,OAAO,MAAM,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,QAAQ,CACxD,eAAe,EACf;gBACE,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,cAAc;gBACd,mBAAmB,EAAE,UAAU;aAChC,EACD,EAAE,MAAM,EAAE,eAAe,CAAC,MAAM,EAAE,CACnC,CAAC;QACJ,CAAC;QACD,yBAAyB;QACzB,KAAK,EAAE,KAAK;KACb,CAAC,CAAC;IAEH,mEAAmE;IACnE,8DAA8D;IAC9D,MAAM,WAAW,GAAG,iBAAiB;QACnC,CAAC,CAAC,CAAC,iBAAiB,CAAC,IAAI,IAAI,qBAAqB,CAAC,IAAI,IAAI,EAAE,CAAC;QAC9D,CAAC,CAAC,EAAE,CAAC;IAEP,OAAO;QACL,WAAW;QACX,MAAM,EAAE,mBAAmB,CAAC,WAAW;QACvC,oBAAoB;QACpB,YAAY,EAAE,mBAAmB;QACjC,cAAc,EAAE,qBAAqB;QACrC,iBAAiB;QACjB,GAAG,IAAA,oCAAsB,EAAC,mBAAmB,EAAE,qBAAqB,CAAC;KACtE,CAAC;AACJ,CAAC","sourcesContent":["\"use client\";\nimport TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { useEffect, useState } from \"react\";\nimport { isIdleStage } from \"../model/generate-component-response\";\nimport { validateInput } from \"../model/validate-input\";\nimport { useTamboClient } from \"../providers/tambo-client-provider\";\nimport { useTambo } from \"../providers/tambo-provider\";\nimport { useTamboRegistry } from \"../providers/tambo-registry-provider\";\nimport {\n  INPUT_ERROR_MESSAGES,\n  useTamboThreadInput,\n} from \"../providers/tambo-thread-input-provider\";\nimport { useTamboThread } from \"../providers/tambo-thread-provider\";\nimport {\n  CombinedMutationResult,\n  combineMutationResults,\n} from \"../util/query-utils\";\nimport { getAvailableComponents } from \"../util/registry\";\nimport {\n  UseTamboMutationResult,\n  UseTamboQueryResult,\n  useTamboMutation,\n  useTamboQuery,\n} from \"./react-query-hooks\";\n\n/**\n * Configuration options for the useTamboSuggestions hook\n */\nexport interface useTamboSuggestionsOptions {\n  /** Maximum number of suggestions to generate (1-10, default 3) */\n  maxSuggestions?: number;\n}\n\n/**\n * Return value interface for useTamboSuggestions hook\n */\nexport interface useTamboSuggestionsResultInternal {\n  /** List of available suggestions (also available in generateResult.data) */\n  suggestions: TamboAI.Beta.Threads.Suggestion[];\n  /** ID of the currently selected suggestion */\n  selectedSuggestionId: string | null;\n  /**\n   * Accept and apply a suggestion (also available in acceptResult.mutateAsync)\n   * @param suggestion - The suggestion to accept\n   * @param shouldSubmit - Whether to automatically submit after accepting (default: false)\n   */\n  accept: (acceptOptions: {\n    suggestion: TamboAI.Beta.Threads.Suggestion;\n    shouldSubmit?: boolean;\n  }) => Promise<void>;\n\n  /** Result and network state for accepting a suggestion */\n  acceptResult: UseTamboMutationResult<\n    void,\n    Error,\n    { suggestion: TamboAI.Beta.Threads.Suggestion; shouldSubmit?: boolean }\n  >;\n\n  /** Result and network state for generating suggestions */\n  generateResult: UseTamboMutationResult<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error,\n    AbortController\n  >;\n\n  /** The full suggestions query object from React Query */\n  suggestionsResult: UseTamboQueryResult<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined\n  >;\n}\n\ntype useTamboSuggestionsResult = CombinedMutationResult<any, Error> &\n  useTamboSuggestionsResultInternal;\n\n/**\n * Hook for managing Tambo AI suggestions in a thread\n * @param options - Configuration options for suggestion generation\n * @returns Object containing suggestions state and control functions\n */\nexport function useTamboSuggestions(\n  options: useTamboSuggestionsOptions = {},\n): useTamboSuggestionsResult {\n  const { maxSuggestions = 3 } = options;\n  const { thread, generationStage } = useTamboThread();\n  const { sendThreadMessage } = useTambo();\n  const tamboClient = useTamboClient();\n  const { componentList, toolRegistry, componentToolAssociations } =\n    useTamboRegistry();\n\n  const [selectedSuggestionId, setSelectedSuggestionId] = useState<\n    string | null\n  >(null);\n  const { setValue: setInputValue } = useTamboThreadInput();\n\n  const latestMessage = thread.messages[thread.messages.length - 1];\n  const isLatestFromTambo = latestMessage?.role === \"assistant\";\n  const latestMessageId = latestMessage?.id;\n\n  // Reset selected suggestion when the message changes\n  useEffect(() => {\n    setSelectedSuggestionId(null);\n  }, [latestMessageId]);\n\n  const shouldGenerateSuggestions =\n    latestMessageId && isLatestFromTambo && isIdleStage(generationStage);\n  // Use React Query to fetch suggestions when a new hydra message is received\n  const suggestionsResult = useTamboQuery({\n    // Make sure the query key changes when the message changes, so that we are\n    // always generating suggestions when there is a new message\n    queryKey: [\n      \"suggestions\",\n      shouldGenerateSuggestions ? latestMessageId : null,\n    ],\n    queryFn: async () => {\n      if (!shouldGenerateSuggestions) {\n        return [];\n      }\n\n      // Get registered components from the registry\n      const components = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n\n      return await tamboClient.beta.threads.suggestions.generate(\n        latestMessageId,\n        {\n          id: thread.id,\n          maxSuggestions,\n          availableComponents: components,\n        },\n      );\n    },\n    // Only run the query if we have a valid message from hydra\n    enabled: Boolean(latestMessageId && isLatestFromTambo),\n    // Don't refetch on window focus or reconnect\n    refetchOnWindowFocus: false,\n    refetchOnReconnect: false,\n    // Don't retry on failure\n    retry: false,\n  });\n\n  // Accept suggestion mutation\n  const acceptMutationState = useTamboMutation<\n    void,\n    Error,\n    { suggestion: TamboAI.Beta.Threads.Suggestion; shouldSubmit?: boolean }\n  >({\n    mutationFn: async ({ suggestion, shouldSubmit = false }) => {\n      const validation = validateInput(suggestion.detailedSuggestion);\n      if (!validation.isValid) {\n        if (validation.error) {\n          throw validation.error;\n        }\n        throw new Error(INPUT_ERROR_MESSAGES.VALIDATION);\n      }\n\n      if (shouldSubmit) {\n        await sendThreadMessage(validation.sanitizedInput, {\n          threadId: thread.id,\n        });\n      } else {\n        setInputValue(validation.sanitizedInput);\n      }\n      setSelectedSuggestionId(suggestion.id);\n    },\n  });\n\n  // Generate suggestions mutation\n  const generateMutationState = useTamboMutation<\n    TamboAI.Beta.Threads.Suggestions.SuggestionGenerateResponse | undefined,\n    Error,\n    AbortController\n  >({\n    mutationFn: async (abortController: AbortController) => {\n      if (!shouldGenerateSuggestions) {\n        return undefined;\n      }\n\n      // Get registered components from the registry\n      const components = getAvailableComponents(\n        componentList,\n        toolRegistry,\n        componentToolAssociations,\n      );\n      return await tamboClient.beta.threads.suggestions.generate(\n        latestMessageId,\n        {\n          id: thread.id,\n          maxSuggestions,\n          availableComponents: components,\n        },\n        { signal: abortController.signal },\n      );\n    },\n    // Don't retry on failure\n    retry: false,\n  });\n\n  // Use the query data if available, otherwise use the mutation data\n  // Only return suggestions if the latest message is from hydra\n  const suggestions = isLatestFromTambo\n    ? (suggestionsResult.data ?? generateMutationState.data ?? [])\n    : [];\n\n  return {\n    suggestions,\n    accept: acceptMutationState.mutateAsync,\n    selectedSuggestionId,\n    acceptResult: acceptMutationState,\n    generateResult: generateMutationState,\n    suggestionsResult,\n    ...combineMutationResults(acceptMutationState, generateMutationState),\n  };\n}\n"]}

package/dist/hooks/use-suggestions.test.d.ts.map

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

package/dist/hooks/{__tests__/use-suggestions.test.js → use-suggestions.test.js}

@@ -2,27 +2,27 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 const react_query_1 = require("@tanstack/react-query");
 const react_1 = require("@testing-library/react");
-const generate_component_response_1 = require("../../model/generate-component-response");
-const tambo_client_provider_1 = require("../../providers/tambo-client-provider");
-const tambo_provider_1 = require("../../providers/tambo-provider");
-const tambo_thread_input_provider_1 = require("../../providers/tambo-thread-input-provider");
-const tambo_thread_provider_1 = require("../../providers/tambo-thread-provider");
-const react_query_hooks_1 = require("../react-query-hooks");
-const use_suggestions_1 = require("../use-suggestions");
+const generate_component_response_1 = require("../model/generate-component-response");
+const tambo_client_provider_1 = require("../providers/tambo-client-provider");
+const tambo_provider_1 = require("../providers/tambo-provider");
+const tambo_thread_input_provider_1 = require("../providers/tambo-thread-input-provider");
+const tambo_thread_provider_1 = require("../providers/tambo-thread-provider");
+const react_query_hooks_1 = require("./react-query-hooks");
+const use_suggestions_1 = require("./use-suggestions");
 // Mock the required providers
-jest.mock("../../providers/tambo-client-provider", () => ({
+jest.mock("../providers/tambo-client-provider", () => ({
     useTamboClient: jest.fn(),
     useTamboQueryClient: jest.fn(),
 }));
-jest.mock("../../providers/tambo-provider", () => ({ useTambo: jest.fn() }));
-jest.mock("../../providers/tambo-thread-provider", () => ({
+jest.mock("../providers/tambo-provider", () => ({ useTambo: jest.fn() }));
+jest.mock("../providers/tambo-thread-provider", () => ({
     useTamboThread: jest.fn(),
 }));
-jest.mock("../../providers/tambo-thread-input-provider", () => ({
+jest.mock("../providers/tambo-thread-input-provider", () => ({
     useTamboThreadInput: jest.fn(),
 }));
 // Mock the react-query-hooks
-jest.mock("../react-query-hooks", () => ({
+jest.mock("./react-query-hooks", () => ({
     useTamboQuery: jest.fn(),
     useTamboMutation: jest.fn(),
 }));
@@ -61,22 +61,29 @@ describe("useTamboSuggestions", () => {
         ...overrides,
     });
     // Helper function to create mock CombinedTamboThreadContextProps
-    const createMockThreadContext = (overrides = {}) => ({
-        thread: createMockThread(),
-        switchCurrentThread: jest.fn(),
-        startNewThread: jest.fn(),
-        updateThreadName: jest.fn(),
-        generateThreadName: jest.fn(),
-        addThreadMessage: jest.fn(),
-        updateThreadMessage: jest.fn(),
-        cancel: jest.fn(),
-        streaming: false,
-        sendThreadMessage: jest.fn(),
-        generationStage: generate_component_response_1.GenerationStage.IDLE,
-        generationStatusMessage: "",
-        isIdle: true,
-        ...overrides,
-    });
+    const createMockThreadContext = (overrides = {}) => {
+        const mockThread = createMockThread();
+        return {
+            thread: mockThread,
+            currentThreadId: mockThread.id,
+            currentThread: mockThread,
+            threadMap: { [mockThread.id]: mockThread },
+            setThreadMap: jest.fn(),
+            switchCurrentThread: jest.fn(),
+            startNewThread: jest.fn(),
+            updateThreadName: jest.fn(),
+            generateThreadName: jest.fn(),
+            addThreadMessage: jest.fn(),
+            updateThreadMessage: jest.fn(),
+            cancel: jest.fn(),
+            streaming: false,
+            sendThreadMessage: jest.fn(),
+            generationStage: generate_component_response_1.GenerationStage.IDLE,
+            generationStatusMessage: "",
+            isIdle: true,
+            ...overrides,
+        };
+    };
     beforeEach(() => {
         jest.mocked(tambo_client_provider_1.useTamboQueryClient).mockReturnValue(new react_query_1.QueryClient());
         jest.mocked(react_query_hooks_1.useTamboMutation).mockImplementation(({ mutationFn }) => ({
@@ -96,7 +103,7 @@ describe("useTamboSuggestions", () => {
             thread: createMockThread({
                 messages: [
                     createMockMessage({
-                        role: "hydra",
+                        role: "assistant",
                     }),
                 ],
             }),

package/dist/hooks/use-suggestions.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-suggestions.test.js","sourceRoot":"","sources":["../../src/hooks/use-suggestions.test.tsx"],"names":[],"mappings":";;AACA,uDAAoE;AACpE,kDAAyD;AACzD,sFAG8C;AAE9C,8EAG4C;AAC5C,gEAA0E;AAC1E,0FAGkD;AAClD,8EAG4C;AAE5C,2DAAsE;AACtE,uDAAwD;AAExD,8BAA8B;AAC9B,IAAI,CAAC,IAAI,CAAC,oCAAoC,EAAE,GAAG,EAAE,CAAC,CAAC;IACrD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;CAC/B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,6BAA6B,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;AAE1E,IAAI,CAAC,IAAI,CAAC,oCAAoC,EAAE,GAAG,EAAE,CAAC,CAAC;IACrD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,0CAA0C,EAAE,GAAG,EAAE,CAAC,CAAC;IAC3D,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;CAC/B,CAAC,CAAC,CAAC;AAEJ,6BAA6B;AAC7B,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE;IACxB,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;CAC5B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,MAAM,eAAe,GAAsC;QACzD;YACE,EAAE,EAAE,cAAc;YAClB,SAAS,EAAE,iBAAiB;YAC5B,KAAK,EAAE,mBAAmB;YAC1B,kBAAkB,EAAE,mBAAmB;SACxC;QACD;YACE,EAAE,EAAE,cAAc;YAClB,SAAS,EAAE,iBAAiB;YAC5B,KAAK,EAAE,mBAAmB;YAC1B,kBAAkB,EAAE,mBAAmB;SACxC;KACF,CAAC;IAEF,oDAAoD;IACpD,MAAM,iBAAiB,GAAG,CACxB,YAAyC,EAAE,EACvB,EAAE,CAAC,CAAC;QACxB,EAAE,EAAE,iBAAiB;QACrB,cAAc,EAAE,EAAE;QAClB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;QACjD,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,IAAI,EAAE,WAAW;QACjB,QAAQ,EAAE,gBAAgB;QAC1B,GAAG,SAAS;KACb,CAAC,CAAC;IAEH,6CAA6C;IAC7C,MAAM,gBAAgB,GAAG,CACvB,YAAkC,EAAE,EACvB,EAAE,CAAC,CAAC;QACjB,EAAE,EAAE,gBAAgB;QACpB,SAAS,EAAE,iBAAiB;QAC5B,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,QAAQ,EAAE,EAAE;QACZ,GAAG,SAAS;KACb,CAAC,CAAC;IAEH,iEAAiE;IACjE,MAAM,uBAAuB,GAAG,CAC9B,YAAsD,EAAE,EACvB,EAAE;QACnC,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;QACtC,OAAO;YACL,MAAM,EAAE,UAAU;YAClB,eAAe,EAAE,UAAU,CAAC,EAAE;YAC9B,aAAa,EAAE,UAAU;YACzB,SAAS,EAAE,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC,EAAE,UAAU,EAAE;YAC1C,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;YACvB,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC9B,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;YACzB,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC3B,kBAAkB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC7B,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC3B,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC9B,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;YACjB,SAAS,EAAE,KAAK;YAChB,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC5B,eAAe,EAAE,6CAAe,CAAC,IAAI;YACrC,uBAAuB,EAAE,EAAE;YAC3B,MAAM,EAAE,IAAI;YACZ,GAAG,SAAS;SACb,CAAC;IACJ,CAAC,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,MAAM,CAAC,2CAAmB,CAAC,CAAC,eAAe,CAAC,IAAI,yBAAW,EAAE,CAAC,CAAC;QACpE,IAAI,CAAC,MAAM,CAAC,oCAAgB,CAAC,CAAC,kBAAkB,CAC9C,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CACjB,CAAC;YACC,WAAW,EAAE,UAAU;YACvB,SAAS,EAAE,KAAK;YAChB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,IAAI;SACZ,CAAQ,CACZ,CAAC;QACF,qCAAqC;QACrC,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC,eAAe,CAAC;YAC1C,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,WAAW,EAAE,EAAE,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE;SAC5B,CAAC,CAAC;QACnC,IAAI,CAAC,MAAM,CAAC,yBAAQ,CAAC,CAAC,eAAe,CAAC;YACpC,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;SACe,CAAC,CAAC;QAC/C,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;YACtB,MAAM,EAAE,gBAAgB,CAAC;gBACvB,QAAQ,EAAE;oBACR,iBAAiB,CAAC;wBAChB,IAAI,EAAE,WAAW;qBAClB,CAAC;iBACH;aACF,CAAC;SACH,CAAC,CACH,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,iDAAmB,CAAC,CAAC,eAAe,CAAC;YAC/C,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;YACnB,KAAK,EAAE,EAAE;YACT,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;SACqC,CAAC,CAAC;QAC1D,yCAAyC;QACzC,IAAI,CAAC,MAAM,CAAC,iCAAa,CAAC,CAAC,eAAe,CAAC;YACzC,IAAI,EAAE,EAAE;YACR,SAAS,EAAE,KAAK;YAChB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,IAAI;SAC+C,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;QAC7E,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,QAAQ,EAAE,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;QAClE,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC,eAAe,CAAC;YAC1C,IAAI,EAAE;gBACJ,OAAO,EAAE;oBACP,WAAW,EAAE;wBACX,QAAQ,EAAE,YAAY;wBACtB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;qBAChB;iBACF;aACF;SAC8B,CAAC,CAAC;QACnC,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;YACtB,MAAM,EAAE,gBAAgB,CAAC;gBACvB,QAAQ,EAAE;oBACR,iBAAiB,CAAC;wBAChB,IAAI,EAAE,WAAW;qBAClB,CAAC;iBACH;aACF,CAAC;SACH,CAAC,CACH,CAAC;QAEF,uDAAuD;QACvD,IAAI,CAAC,MAAM,CAAC,iCAAa,CAAC,CAAC,eAAe,CAAC;YACzC,IAAI,EAAE,eAAe;YACrB,SAAS,EAAE,KAAK;YAChB,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,IAAI;SACwB,CAAC,CAAC;QAEvC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,6BAA6B;QAC7B,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,wEAAwE;QACxE,wEAAwE;QACxE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAC/B,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC,eAAe,CAAC;YAC1C,IAAI,EAAE;gBACJ,OAAO,EAAE;oBACP,WAAW,EAAE;wBACX,QAAQ,EAAE,YAAY;wBACtB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;qBAChB;iBACF;aACF;SAC8B,CAAC,CAAC;QACnC,8CAA8C;QAC9C,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;YACtB,MAAM,EAAE,gBAAgB,CAAC;gBACvB,QAAQ,EAAE;oBACR,iBAAiB,CAAC;wBAChB,IAAI,EAAE,MAAM;qBACb,CAAC;iBACH;aACF,CAAC;SACH,CAAC,CACH,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,6BAA6B;QAC7B,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAC/B,IAAI,CAAC,MAAM,CAAC,iDAAmB,CAAC,CAAC,eAAe,CAAC;YAC/C,QAAQ,EAAE,YAAY;YACtB,KAAK,EAAE,EAAE;YACT,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;SACqC,CAAC,CAAC;QAE1D,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;gBAC1B,UAAU,EAAE,eAAe,CAAC,CAAC,CAAC;gBAC9B,YAAY,EAAE,KAAK;aACpB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,mBAAmB,CAAC,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,qBAAqB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QACxC,IAAI,CAAC,MAAM,CAAC,yBAAQ,CAAC,CAAC,eAAe,CAAC;YACpC,iBAAiB,EAAE,qBAAqB;SACG,CAAC,CAAC;QAE/C,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;gBAC1B,UAAU,EAAE,eAAe,CAAC,CAAC,CAAC;gBAC9B,YAAY,EAAE,IAAI;aACnB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,qBAAqB,CAAC,CAAC,oBAAoB,CAAC,mBAAmB,EAAE;YACtE,QAAQ,EAAE,gBAAgB;SAC3B,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACpE,MAAM,iBAAiB,GAAG;YACxB,EAAE,EAAE,oBAAoB;YACxB,SAAS,EAAE,iBAAiB;YAC5B,KAAK,EAAE,oBAAoB;YAC3B,kBAAkB,EAAE,EAAE,EAAE,0CAA0C;SACnE,CAAC;QAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,qCAAmB,GAAE,CAAC,CAAC;QAE3D,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CACV,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC;gBACpB,UAAU,EAAE,iBAAiB;gBAC7B,YAAY,EAAE,KAAK;aACpB,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,QAAQ,EAAE,CAAC;IACzD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import type TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { QueryClient, UseQueryResult } from \"@tanstack/react-query\";\nimport { act, renderHook } from \"@testing-library/react\";\nimport {\n  GenerationStage,\n  TamboThreadMessage,\n} from \"../model/generate-component-response\";\nimport { TamboThread } from \"../model/tambo-thread\";\nimport {\n  useTamboClient,\n  useTamboQueryClient,\n} from \"../providers/tambo-client-provider\";\nimport { TamboContextProps, useTambo } from \"../providers/tambo-provider\";\nimport {\n  TamboThreadInputContextProps,\n  useTamboThreadInput,\n} from \"../providers/tambo-thread-input-provider\";\nimport {\n  CombinedTamboThreadContextProps,\n  useTamboThread,\n} from \"../providers/tambo-thread-provider\";\nimport { PartialTamboAI } from \"../testing/types\";\nimport { useTamboMutation, useTamboQuery } from \"./react-query-hooks\";\nimport { useTamboSuggestions } from \"./use-suggestions\";\n\n// Mock the required providers\njest.mock(\"../providers/tambo-client-provider\", () => ({\n  useTamboClient: jest.fn(),\n  useTamboQueryClient: jest.fn(),\n}));\n\njest.mock(\"../providers/tambo-provider\", () => ({ useTambo: jest.fn() }));\n\njest.mock(\"../providers/tambo-thread-provider\", () => ({\n  useTamboThread: jest.fn(),\n}));\n\njest.mock(\"../providers/tambo-thread-input-provider\", () => ({\n  useTamboThreadInput: jest.fn(),\n}));\n\n// Mock the react-query-hooks\njest.mock(\"./react-query-hooks\", () => ({\n  useTamboQuery: jest.fn(),\n  useTamboMutation: jest.fn(),\n}));\n\ndescribe(\"useTamboSuggestions\", () => {\n  const mockSuggestions: TamboAI.Beta.Threads.Suggestion[] = [\n    {\n      id: \"suggestion-1\",\n      messageId: \"test-message-id\",\n      title: \"Test Suggestion 1\",\n      detailedSuggestion: \"Test suggestion 1\",\n    },\n    {\n      id: \"suggestion-2\",\n      messageId: \"test-message-id\",\n      title: \"Test Suggestion 2\",\n      detailedSuggestion: \"Test suggestion 2\",\n    },\n  ];\n\n  // Helper function to create mock TamboThreadMessage\n  const createMockMessage = (\n    overrides: Partial<TamboThreadMessage> = {},\n  ): TamboThreadMessage => ({\n    id: \"test-message-id\",\n    componentState: {},\n    content: [{ type: \"text\", text: \"Test message\" }],\n    createdAt: new Date().toISOString(),\n    role: \"assistant\",\n    threadId: \"test-thread-id\",\n    ...overrides,\n  });\n\n  // Helper function to create mock TamboThread\n  const createMockThread = (\n    overrides: Partial<TamboThread> = {},\n  ): TamboThread => ({\n    id: \"test-thread-id\",\n    projectId: \"test-project-id\",\n    createdAt: new Date().toISOString(),\n    updatedAt: new Date().toISOString(),\n    messages: [],\n    ...overrides,\n  });\n\n  // Helper function to create mock CombinedTamboThreadContextProps\n  const createMockThreadContext = (\n    overrides: Partial<CombinedTamboThreadContextProps> = {},\n  ): CombinedTamboThreadContextProps => {\n    const mockThread = createMockThread();\n    return {\n      thread: mockThread,\n      currentThreadId: mockThread.id,\n      currentThread: mockThread,\n      threadMap: { [mockThread.id]: mockThread },\n      setThreadMap: jest.fn(),\n      switchCurrentThread: jest.fn(),\n      startNewThread: jest.fn(),\n      updateThreadName: jest.fn(),\n      generateThreadName: jest.fn(),\n      addThreadMessage: jest.fn(),\n      updateThreadMessage: jest.fn(),\n      cancel: jest.fn(),\n      streaming: false,\n      sendThreadMessage: jest.fn(),\n      generationStage: GenerationStage.IDLE,\n      generationStatusMessage: \"\",\n      isIdle: true,\n      ...overrides,\n    };\n  };\n\n  beforeEach(() => {\n    jest.mocked(useTamboQueryClient).mockReturnValue(new QueryClient());\n    jest.mocked(useTamboMutation).mockImplementation(\n      ({ mutationFn }) =>\n        ({\n          mutateAsync: mutationFn,\n          isLoading: false,\n          isError: false,\n          error: null,\n        }) as any,\n    );\n    // Setup default mock implementations\n    jest.mocked(useTamboClient).mockReturnValue({\n      beta: { threads: { suggestions: { generate: jest.fn() } } },\n    } satisfies PartialTamboAI as any);\n    jest.mocked(useTambo).mockReturnValue({\n      sendThreadMessage: jest.fn(),\n    } satisfies Partial<TamboContextProps> as any);\n    jest.mocked(useTamboThread).mockReturnValue(\n      createMockThreadContext({\n        thread: createMockThread({\n          messages: [\n            createMockMessage({\n              role: \"assistant\",\n            }),\n          ],\n        }),\n      }),\n    );\n    jest.mocked(useTamboThreadInput).mockReturnValue({\n      setValue: jest.fn(),\n      value: \"\",\n      submit: jest.fn(),\n    } satisfies Partial<TamboThreadInputContextProps> as any);\n    // Default query mock returns empty array\n    jest.mocked(useTamboQuery).mockReturnValue({\n      data: [],\n      isLoading: false,\n      isError: false,\n      error: null,\n    } satisfies Partial<UseQueryResult<unknown, unknown>> as any);\n  });\n\n  it(\"should initialize with empty suggestions and no selected suggestion\", () => {\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    expect(result.current.suggestions).toEqual([]);\n    expect(result.current.selectedSuggestionId).toBeNull();\n  });\n\n  it(\"should generate suggestions when latest message is from Tambo\", async () => {\n    const mockGenerate = jest.fn().mockResolvedValue(mockSuggestions);\n    jest.mocked(useTamboClient).mockReturnValue({\n      beta: {\n        threads: {\n          suggestions: {\n            generate: mockGenerate,\n            list: jest.fn(),\n          },\n        },\n      },\n    } satisfies PartialTamboAI as any);\n    jest.mocked(useTamboThread).mockReturnValue(\n      createMockThreadContext({\n        thread: createMockThread({\n          messages: [\n            createMockMessage({\n              role: \"assistant\",\n            }),\n          ],\n        }),\n      }),\n    );\n\n    // Mock the query result to return the mock suggestions\n    jest.mocked(useTamboQuery).mockReturnValue({\n      data: mockSuggestions,\n      isLoading: false,\n      isError: false,\n      error: null,\n    } as UseQueryResult<unknown, unknown>);\n\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    // Wait for the effect to run\n    await act(async () => {\n      await new Promise((resolve) => setTimeout(resolve, 0));\n    });\n\n    // Since we're mocking useTamboQuery to return the suggestions directly,\n    // the generate function won't be called, so we don't need to check that\n    expect(result.current.suggestions).toEqual(mockSuggestions);\n  });\n\n  it(\"should not generate suggestions when latest message is not from Tambo\", async () => {\n    const mockGenerate = jest.fn();\n    jest.mocked(useTamboClient).mockReturnValue({\n      beta: {\n        threads: {\n          suggestions: {\n            generate: mockGenerate,\n            list: jest.fn(),\n          },\n        },\n      },\n    } satisfies PartialTamboAI as any);\n    // Mock the thread to have a non-Tambo message\n    jest.mocked(useTamboThread).mockReturnValue(\n      createMockThreadContext({\n        thread: createMockThread({\n          messages: [\n            createMockMessage({\n              role: \"user\",\n            }),\n          ],\n        }),\n      }),\n    );\n\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    // Wait for the effect to run\n    await act(async () => {\n      await new Promise((resolve) => setTimeout(resolve, 0));\n    });\n\n    expect(mockGenerate).not.toHaveBeenCalled();\n    expect(result.current.suggestions).toEqual([]);\n  });\n\n  it(\"should accept a suggestion and update input value\", async () => {\n    const mockSetValue = jest.fn();\n    jest.mocked(useTamboThreadInput).mockReturnValue({\n      setValue: mockSetValue,\n      value: \"\",\n      submit: jest.fn(),\n    } satisfies Partial<TamboThreadInputContextProps> as any);\n\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    await act(async () => {\n      await result.current.accept({\n        suggestion: mockSuggestions[0],\n        shouldSubmit: false,\n      });\n    });\n\n    expect(mockSetValue).toHaveBeenCalledWith(\"Test suggestion 1\");\n    expect(result.current.selectedSuggestionId).toBe(\"suggestion-1\");\n  });\n\n  it(\"should accept a suggestion and submit it\", async () => {\n    const mockSendThreadMessage = jest.fn();\n    jest.mocked(useTambo).mockReturnValue({\n      sendThreadMessage: mockSendThreadMessage,\n    } satisfies Partial<TamboContextProps> as any);\n\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    await act(async () => {\n      await result.current.accept({\n        suggestion: mockSuggestions[0],\n        shouldSubmit: true,\n      });\n    });\n\n    expect(mockSendThreadMessage).toHaveBeenCalledWith(\"Test suggestion 1\", {\n      threadId: \"test-thread-id\",\n    });\n    expect(result.current.selectedSuggestionId).toBe(\"suggestion-1\");\n  });\n\n  it(\"should throw error when accepting invalid suggestion\", async () => {\n    const invalidSuggestion = {\n      id: \"invalid-suggestion\",\n      messageId: \"test-message-id\",\n      title: \"Invalid Suggestion\",\n      detailedSuggestion: \"\", // Empty suggestion should fail validation\n    };\n\n    const { result } = renderHook(() => useTamboSuggestions());\n\n    await act(async () => {\n      await expect(\n        result.current.accept({\n          suggestion: invalidSuggestion,\n          shouldSubmit: false,\n        }),\n      ).rejects.toThrow(\"Message cannot be empty\");\n    });\n\n    expect(result.current.selectedSuggestionId).toBeNull();\n  });\n});\n"]}

package/dist/hooks/use-tambo-stream-status.d.ts

@@ -57,14 +57,15 @@ export interface PropStatus {
     error?: Error;
 }
 /**
- * Hook that exposes per-prop and global streaming status for tambo-ai components.
- * Provides streaming readiness flags so consumers can show loaders, skeletons,
- * or errors while LLM-generated props stream in.
- * This hook tracks status for the specific component in the current message only.
- * Once a component's props complete streaming, they remain stable regardless of
- * other components being generated in the thread.
- * @template Props - The type of the component props being tracked (defaults to Record<string, any>)
- * @returns An object containing both global streamStatus and per-prop propStatus
+ * Track streaming status for Tambo component props.
+ *
+ * **Important**: Props update repeatedly during streaming and may be partial.
+ * Use `propStatus.<field>?.isSuccess` before treating a prop as complete.
+ *
+ * Pair with `useTamboComponentState` to disable inputs while streaming.
+ * @see {@link https://docs.tambo.co/concepts/streaming/streaming-best-practices}
+ * @template Props - Component props type
+ * @returns `streamStatus` (overall) and `propStatus` (per-prop) flags
  * @throws {Error} When used during SSR/SSG
  * @example
  * ```tsx

package/dist/hooks/use-tambo-stream-status.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"use-tambo-stream-status.d.ts","sourceRoot":"","sources":["../../src/hooks/use-tambo-stream-status.tsx"],"names":[],"mappings":"AAMA;;;;GAIG;AACH,MAAM,WAAW,YAAY;IAC3B;;;OAGG;IACH,SAAS,EAAE,OAAO,CAAC;IAEnB;;;OAGG;IACH,WAAW,EAAE,OAAO,CAAC;IAErB;;;OAGG;IACH,SAAS,EAAE,OAAO,CAAC;IAEnB;;;OAGG;IACH,OAAO,EAAE,OAAO,CAAC;IAEjB;;;OAGG;IACH,WAAW,CAAC,EAAE,KAAK,CAAC;CACrB;AAED;;;GAGG;AACH,MAAM,WAAW,UAAU;IACzB;;;OAGG;IACH,SAAS,EAAE,OAAO,CAAC;IAEnB;;;OAGG;IACH,WAAW,EAAE,OAAO,CAAC;IAErB;;;OAGG;IACH,SAAS,EAAE,OAAO,CAAC;IAEnB;;;OAGG;IACH,KAAK,CAAC,EAAE,KAAK,CAAC;CACf;AA8ND;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,wBAAgB,oBAAoB,CAClC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KACpD;IACH,YAAY,EAAE,YAAY,CAAC;IAC3B,UAAU,EAAE,MAAM,CAAC,MAAM,KAAK,EAAE,UAAU,CAAC,CAAC;CAC7C,CAmCA"}
+{"version":3,"file":"use-tambo-stream-status.d.ts","sourceRoot":"","sources":["../../src/hooks/use-tambo-stream-status.tsx"],"names":[],"mappings":"AAMA;;;;GAIG;AACH,MAAM,WAAW,YAAY;IAC3B;;;OAGG;IACH,SAAS,EAAE,OAAO,CAAC;IAEnB;;;OAGG;IACH,WAAW,EAAE,OAAO,CAAC;IAErB;;;OAGG;IACH,SAAS,EAAE,OAAO,CAAC;IAEnB;;;OAGG;IACH,OAAO,EAAE,OAAO,CAAC;IAEjB;;;OAGG;IACH,WAAW,CAAC,EAAE,KAAK,CAAC;CACrB;AAED;;;GAGG;AACH,MAAM,WAAW,UAAU;IACzB;;;OAGG;IACH,SAAS,EAAE,OAAO,CAAC;IAEnB;;;OAGG;IACH,WAAW,EAAE,OAAO,CAAC;IAErB;;;OAGG;IACH,SAAS,EAAE,OAAO,CAAC;IAEnB;;;OAGG;IACH,KAAK,CAAC,EAAE,KAAK,CAAC;CACf;AA8ND;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,wBAAgB,oBAAoB,CAClC,KAAK,SAAS,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KACpD;IACH,YAAY,EAAE,YAAY,CAAC;IAC3B,UAAU,EAAE,MAAM,CAAC,MAAM,KAAK,EAAE,UAAU,CAAC,CAAC;CAC7C,CAmCA"}

package/dist/hooks/use-tambo-stream-status.js

@@ -162,14 +162,15 @@ function deriveGlobalStreamStatus(generationStage, propStatus, hasComponent, gen
     };
 }
 /**
- * Hook that exposes per-prop and global streaming status for tambo-ai components.
- * Provides streaming readiness flags so consumers can show loaders, skeletons,
- * or errors while LLM-generated props stream in.
- * This hook tracks status for the specific component in the current message only.
- * Once a component's props complete streaming, they remain stable regardless of
- * other components being generated in the thread.
- * @template Props - The type of the component props being tracked (defaults to Record<string, any>)
- * @returns An object containing both global streamStatus and per-prop propStatus
+ * Track streaming status for Tambo component props.
+ *
+ * **Important**: Props update repeatedly during streaming and may be partial.
+ * Use `propStatus.<field>?.isSuccess` before treating a prop as complete.
+ *
+ * Pair with `useTamboComponentState` to disable inputs while streaming.
+ * @see {@link https://docs.tambo.co/concepts/streaming/streaming-best-practices}
+ * @template Props - Component props type
+ * @returns `streamStatus` (overall) and `propStatus` (per-prop) flags
  * @throws {Error} When used during SSR/SSG
  * @example
  * ```tsx

package/dist/hooks/use-tambo-stream-status.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-tambo-stream-status.js","sourceRoot":"","sources":["../../src/hooks/use-tambo-stream-status.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;AA+Tb,oDAwCC;AAtWD,iCAAqD;AACrD,sFAAuE;AACvE,8EAA6E;AAC7E,+DAA+D;AAqE/D;;;;GAIG;AACH,SAAS,gBAAgB;IACvB,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CACb,6DAA6D;YAC3D,4CAA4C;YAC5C,6DAA6D,CAChE,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,uBAAuB,CAC9B,KAAwB,EACxB,eAAgC,EAChC,SAAiB;IAEjB,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,IAAA,gBAAQ,EAU9C,EAAE,CAAC,CAAC;IAEN,mDAAmD;IACnD,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,0DAA0D;YAC1D,MAAM,iBAAiB,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAChD,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,KAAK,SAAS,CAC5D,CAAC;YACF,OAAO,iBAAiB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;IAEhB,oFAAoF;IACpF,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,MAAM,OAAO,GAAG,EAAE,GAAG,IAAI,EAAE,CAAC;YAC5B,IAAI,UAAU,GAAG,KAAK,CAAC;YAEvB,oDAAoD;YACpD,MAAM,gBAAgB,GAAa,EAAE,CAAC;YACtC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;gBAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI;oBAC3B,UAAU,EAAE,KAAK;oBACjB,UAAU,EAAE,KAAK;iBAClB,CAAC;gBAEF,+EAA+E;gBAC/E,MAAM,UAAU,GACd,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,EAAE,CAAC;gBACxD,MAAM,WAAW,GAAG,UAAU,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;gBAEtD,IAAI,WAAW,EAAE,CAAC;oBAChB,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAC7B,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,mEAAmE;YACnE,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;gBAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI;oBAC3B,UAAU,EAAE,KAAK;oBACjB,UAAU,EAAE,KAAK;iBAClB,CAAC;gBAEF,+EAA+E;gBAC/E,MAAM,UAAU,GACd,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,EAAE,CAAC;gBACxD,MAAM,WAAW,GAAG,UAAU,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;gBAEtD;;;;mBAIG;gBACH,MAAM,uBAAuB,GAAG,gBAAgB,CAAC,IAAI,CACnD,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,KAAK,GAAG,CACrC,CAAC;gBACF,MAAM,oBAAoB,GACxB,eAAe,KAAK,6CAAe,CAAC,QAAQ,CAAC;gBAC/C,MAAM,UAAU,GACd,OAAO,CAAC,UAAU;oBAClB,CAAC,uBAAuB,IAAI,oBAAoB,CAAC;oBACjD,CAAC,OAAO,CAAC,UAAU,CAAC;gBAEtB,8DAA8D;gBAC9D,IAAI,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;oBAC1D,2CAA2C;oBAC3C,OAAO;gBACT,CAAC;gBAED,IAAI,WAAW,IAAI,UAAU,EAAE,CAAC;oBAC9B,OAAO,CAAC,GAAG,CAAC,GAAG;wBACb,GAAG,OAAO;wBACV,UAAU,EAAE,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU;wBACnD,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU;wBAClD,SAAS;qBACV,CAAC;oBACF,UAAU,GAAG,IAAI,CAAC;gBACpB,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,OAAO,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,KAAK,EAAE,eAAe,EAAE,SAAS,CAAC,CAAC,CAAC;IAExC,mDAAmD;IACnD,OAAO,IAAA,eAAO,EAAC,GAAG,EAAE;QAClB,IAAI,CAAC,KAAK;YAAE,OAAO,EAAqC,CAAC;QAEzD,MAAM,MAAM,GAAG,EAAqC,CAAC;QAErD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YACjC,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,IAAI;gBACpC,UAAU,EAAE,KAAK;gBACjB,UAAU,EAAE,KAAK;gBACjB,SAAS,EAAE,EAAE;aACd,CAAC;YAEF,+DAA+D;YAC/D,MAAM,wBAAwB,GAC5B,QAAQ,CAAC,UAAU,IAAI,QAAQ,CAAC,SAAS,KAAK,SAAS,CAAC;YAE1D,sFAAsF;YACtF,MAAM,qBAAqB,GACzB,CAAC,wBAAwB;gBACzB,eAAe,KAAK,6CAAe,CAAC,kBAAkB,CAAC;YAEzD,MAAM,CAAC,GAAkB,CAAC,GAAG;gBAC3B,SAAS,EAAE,CAAC,QAAQ,CAAC,UAAU,IAAI,CAAC,wBAAwB;gBAC5D,WAAW,EACT,QAAQ,CAAC,UAAU;oBACnB,CAAC,wBAAwB;oBACzB,qBAAqB;gBACvB,SAAS,EAAE,wBAAwB;gBACnC,KAAK,EAAE,QAAQ,CAAC,KAAK;aACtB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAChB,CAAC,EAAE,CAAC,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,SAAS,CAAC,CAAC,CAAC;AACxD,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,wBAAwB,CAC/B,eAAgC,EAChC,UAA2C,EAC3C,YAAqB,EACrB,eAAuB;IAEvB,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,UAAU,CAAiB,CAAC;IAE/D,gGAAgG;IAChG,MAAM,kBAAkB,GACtB,YAAY,CAAC,MAAM,GAAG,CAAC,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IAEpE,+DAA+D;IAC/D,MAAM,qBAAqB,GACzB,CAAC,kBAAkB;QACnB,eAAe,KAAK,6CAAe,CAAC,kBAAkB,CAAC;IACzD,MAAM,iBAAiB,GAAG,eAAe,KAAK,6CAAe,CAAC,KAAK,CAAC;IAEpE,mDAAmD;IACnD,MAAM,UAAU,GACd,eAAe,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,KAAK,CAAC;IAE9D,OAAO;QACL,+EAA+E;QAC/E,SAAS,EACP,CAAC,YAAY;YACb,CAAC,CAAC,qBAAqB;gBACrB,CAAC,kBAAkB;gBACnB,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAE3C,iGAAiG;QACjG,WAAW,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC;QAEpD,oFAAoF;QACpF,SAAS,EAAE,kBAAkB;QAE7B,kDAAkD;QAClD,OAAO,EACL,iBAAiB;YACjB,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;YACjC,CAAC,CAAC,eAAe;QAEnB,WAAW,EAAE,UAAU;KACxB,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,SAAgB,oBAAoB;IAMlC,oDAAoD;IACpD,gBAAgB,EAAE,CAAC;IAEnB,MAAM,EAAE,eAAe,EAAE,GAAG,IAAA,+CAAuB,GAAE,CAAC;IACtD,MAAM,OAAO,GAAG,IAAA,4CAAsB,GAAE,CAAC;IAEzC,uDAAuD;IACvD,MAAM,cAAc,GAAI,OAAO,EAAE,SAAS,EAAE,KAAe,IAAK,EAAY,CAAC;IAE7E,sCAAsC;IACtC,MAAM,UAAU,GAAG,uBAAuB,CACxC,cAAc,EACd,eAAe,EACf,OAAO,EAAE,EAAE,IAAI,EAAE,CAClB,CAAC;IAEF,0EAA0E;IAC1E,MAAM,YAAY,GAAG,IAAA,eAAO,EAAC,GAAG,EAAE;QAChC,MAAM,eAAe,GAAG,OAAO,EAAE,KAAK;YACpC,CAAC,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;YAC1B,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,YAAY,GAAG,CAAC,CAAC,OAAO,EAAE,SAAS,CAAC;QAC1C,OAAO,wBAAwB,CAC7B,eAAe,EACf,UAAU,EACV,YAAY,EACZ,eAAe,CAChB,CAAC;IACJ,CAAC,EAAE,CAAC,eAAe,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;IAE3C,OAAO;QACL,YAAY;QACZ,UAAU;KACX,CAAC;AACJ,CAAC","sourcesContent":["\"use client\";\nimport { useEffect, useMemo, useState } from \"react\";\nimport { GenerationStage } from \"../model/generate-component-response\";\nimport { useTamboGenerationStage } from \"../providers/tambo-thread-provider\";\nimport { useTamboCurrentMessage } from \"./use-current-message\";\n\n/**\n * Global stream status flags for a specific component in a message.\n * Represents the aggregate state across all props for this component only.\n * Once a component completes, its status remains stable regardless of other generations.\n */\nexport interface StreamStatus {\n  /**\n   * Indicates no tokens have been received for any prop and generation is not active.\n   * Useful for showing initial loading states before any data arrives.\n   */\n  isPending: boolean;\n\n  /**\n   * Indicates active streaming - either generation is streaming OR at least one prop is still streaming.\n   * Use this to show loading animations or skeleton states during data transmission.\n   */\n  isStreaming: boolean;\n\n  /**\n   * Indicates successful completion - generation is complete AND every prop finished without error.\n   * Safe to render the final component when this is true.\n   */\n  isSuccess: boolean;\n\n  /**\n   * Indicates a fatal error occurred in any prop or the stream itself.\n   * Check streamError for details about what went wrong.\n   */\n  isError: boolean;\n\n  /**\n   * The first fatal error encountered during streaming (if any).\n   * Will be undefined if no errors occurred.\n   */\n  streamError?: Error;\n}\n\n/**\n * Streaming status flags for individual component props.\n * Tracks the state of each prop as it streams from the LLM.\n */\nexport interface PropStatus {\n  /**\n   * Indicates no tokens have been received for this specific prop yet.\n   * The prop value is still undefined, null, or empty string.\n   */\n  isPending: boolean;\n\n  /**\n   * Indicates at least one token has been received but streaming is not complete.\n   * The prop has partial content that may still be updating.\n   */\n  isStreaming: boolean;\n\n  /**\n   * Indicates this prop has finished streaming successfully.\n   * The prop value is complete and stable.\n   */\n  isSuccess: boolean;\n\n  /**\n   * The error that occurred during streaming (if any).\n   * Will be undefined if no error occurred for this prop.\n   */\n  error?: Error;\n}\n\n/**\n * SSR Guard - throws during server-side rendering.\n * Ensures the hook is only used in browser contexts.\n * @throws {Error} When called during server-side rendering\n */\nfunction assertClientSide() {\n  if (typeof window === \"undefined\") {\n    throw new Error(\n      \"useTamboStreamStatus can only be used in browser contexts. \" +\n        \"This hook is not compatible with SSR/SSG. \" +\n        \"Consider wrapping it in useEffect or using dynamic imports.\",\n    );\n  }\n}\n\n/**\n * Track streaming status for individual props by monitoring their values.\n * Monitors when props receive their first token and when they complete streaming.\n * Maintains stable state per message - once props complete for a message, they stay complete.\n * @template Props - The type of the component props being tracked\n * @param props - The current component props object\n * @param generationStage - The current generation stage from the LLM\n * @param messageId - The ID of the current message to track component-specific state\n * @returns A record mapping each prop key to its PropStatus\n */\nfunction usePropsStreamingStatus<Props extends Record<string, any>>(\n  props: Props | undefined,\n  generationStage: GenerationStage,\n  messageId: string,\n): Record<keyof Props, PropStatus> {\n  const [propTracking, setPropTracking] = useState<\n    Record<\n      string,\n      {\n        hasStarted: boolean;\n        isComplete: boolean;\n        error?: Error;\n        messageId: string;\n      }\n    >\n  >({});\n\n  /** Reset tracking only when the message changes */\n  useEffect(() => {\n    setPropTracking((prev) => {\n      // If we have tracking data for a different message, reset\n      const hasOldMessageData = Object.values(prev).some(\n        (track) => track.messageId && track.messageId !== messageId,\n      );\n      return hasOldMessageData ? {} : prev;\n    });\n  }, [messageId]);\n\n  /** Track when props start streaming (receive first token) and when they complete */\n  useEffect(() => {\n    if (!props) return;\n\n    setPropTracking((prev) => {\n      const updated = { ...prev };\n      let hasChanges = false;\n\n      // First pass: identify which props are starting now\n      const propsStartingNow: string[] = [];\n      Object.entries(props).forEach(([key, value]) => {\n        const current = prev[key] || {\n          hasStarted: false,\n          isComplete: false,\n        };\n\n        /** A prop starts streaming when it has a non-empty value for the first time */\n        const hasContent =\n          value !== undefined && value !== null && value !== \"\";\n        const justStarted = hasContent && !current.hasStarted;\n\n        if (justStarted) {\n          propsStartingNow.push(key);\n        }\n      });\n\n      // Second pass: update tracking and mark previous props as complete\n      Object.entries(props).forEach(([key, value]) => {\n        const current = prev[key] || {\n          hasStarted: false,\n          isComplete: false,\n        };\n\n        /** A prop starts streaming when it has a non-empty value for the first time */\n        const hasContent =\n          value !== undefined && value !== null && value !== \"\";\n        const justStarted = hasContent && !current.hasStarted;\n\n        /**\n         * A prop is complete when it has started and either:\n         * 1. A following prop has started, OR\n         * 2. Generation is complete (for the final prop)\n         */\n        const hasFollowingPropStarted = propsStartingNow.some(\n          (startingKey) => startingKey !== key,\n        );\n        const isGenerationComplete =\n          generationStage === GenerationStage.COMPLETE;\n        const isComplete =\n          current.hasStarted &&\n          (hasFollowingPropStarted || isGenerationComplete) &&\n          !current.isComplete;\n\n        // Once a prop is complete for this message, it stays complete\n        if (current.isComplete && current.messageId === messageId) {\n          // Skip - already complete for this message\n          return;\n        }\n\n        if (justStarted || isComplete) {\n          updated[key] = {\n            ...current,\n            hasStarted: justStarted ? true : current.hasStarted,\n            isComplete: isComplete ? true : current.isComplete,\n            messageId,\n          };\n          hasChanges = true;\n        }\n      });\n\n      return hasChanges ? updated : prev;\n    });\n  }, [props, generationStage, messageId]);\n\n  /** Convert tracking state to PropStatus objects */\n  return useMemo(() => {\n    if (!props) return {} as Record<keyof Props, PropStatus>;\n\n    const result = {} as Record<keyof Props, PropStatus>;\n\n    Object.keys(props).forEach((key) => {\n      const tracking = propTracking[key] || {\n        hasStarted: false,\n        isComplete: false,\n        messageId: \"\",\n      };\n\n      // If this prop is complete for this message, it stays complete\n      const isCompleteForThisMessage =\n        tracking.isComplete && tracking.messageId === messageId;\n\n      // Only consider generation stage if this prop isn't already complete for this message\n      const isGenerationStreaming =\n        !isCompleteForThisMessage &&\n        generationStage === GenerationStage.STREAMING_RESPONSE;\n\n      result[key as keyof Props] = {\n        isPending: !tracking.hasStarted && !isCompleteForThisMessage,\n        isStreaming:\n          tracking.hasStarted &&\n          !isCompleteForThisMessage &&\n          isGenerationStreaming,\n        isSuccess: isCompleteForThisMessage,\n        error: tracking.error,\n      };\n    });\n\n    return result;\n  }, [props, propTracking, generationStage, messageId]);\n}\n\n/**\n * Derives global StreamStatus from generation stage and individual prop statuses.\n * Aggregates individual prop states into a unified stream status.\n * @template Props - The type of the component props\n * @param generationStage - The current generation stage from the LLM\n * @param propStatus - Status record for each individual prop\n * @param hasComponent - Whether a component exists in the current message\n * @param generationError - Any error from the generation process itself\n * @returns The aggregated StreamStatus for the entire component\n */\nfunction deriveGlobalStreamStatus<Props extends Record<string, any>>(\n  generationStage: GenerationStage,\n  propStatus: Record<keyof Props, PropStatus>,\n  hasComponent: boolean,\n  generationError?: Error,\n): StreamStatus {\n  const propStatuses = Object.values(propStatus) as PropStatus[];\n\n  // If all props are already successful, the component is complete regardless of generation stage\n  const allPropsSuccessful =\n    propStatuses.length > 0 && propStatuses.every((p) => p.isSuccess);\n\n  // Only consider generation stage if not all props are complete\n  const isGenerationStreaming =\n    !allPropsSuccessful &&\n    generationStage === GenerationStage.STREAMING_RESPONSE;\n  const isGenerationError = generationStage === GenerationStage.ERROR;\n\n  /** Find first error from generation or any prop */\n  const firstError =\n    generationError ?? propStatuses.find((p) => p.error)?.error;\n\n  return {\n    /** isPending: no component yet OR (has component but no props have started) */\n    isPending:\n      !hasComponent ||\n      (!isGenerationStreaming &&\n        !allPropsSuccessful &&\n        propStatuses.every((p) => p.isPending)),\n\n    /** isStreaming: any prop is streaming (generation stage doesn't matter if props are complete) */\n    isStreaming: propStatuses.some((p) => p.isStreaming),\n\n    /** isSuccess: all props successful (component is stable once all props complete) */\n    isSuccess: allPropsSuccessful,\n\n    /** isError: generation error OR any prop error */\n    isError:\n      isGenerationError ||\n      propStatuses.some((p) => p.error) ||\n      !!generationError,\n\n    streamError: firstError,\n  };\n}\n\n/**\n * Hook that exposes per-prop and global streaming status for tambo-ai components.\n * Provides streaming readiness flags so consumers can show loaders, skeletons,\n * or errors while LLM-generated props stream in.\n * This hook tracks status for the specific component in the current message only.\n * Once a component's props complete streaming, they remain stable regardless of\n * other components being generated in the thread.\n * @template Props - The type of the component props being tracked (defaults to Record<string, any>)\n * @returns An object containing both global streamStatus and per-prop propStatus\n * @throws {Error} When used during SSR/SSG\n * @example\n * ```tsx\n * // Wait for entire stream\n * const { streamStatus } = useTamboStreamStatus();\n * if (!streamStatus.isSuccess) return <Spinner />;\n * return <Card {...props} />;\n * ```\n * @example\n * ```tsx\n * // Highlight in-flight props\n * const { propStatus } = useTamboStreamStatus<Props>();\n * <h2 className={propStatus.title.isStreaming ? \"animate-pulse\" : \"\"}>\n *   {title}\n * </h2>\n * ```\n */\nexport function useTamboStreamStatus<\n  Props extends Record<string, any> = Record<string, any>,\n>(): {\n  streamStatus: StreamStatus;\n  propStatus: Record<keyof Props, PropStatus>;\n} {\n  /** SSR Guard - ensure client-side only execution */\n  assertClientSide();\n\n  const { generationStage } = useTamboGenerationStage();\n  const message = useTamboCurrentMessage();\n\n  /** Get the current component props from the message */\n  const componentProps = (message?.component?.props as Props) || ({} as Props);\n\n  /** Track per-prop streaming status */\n  const propStatus = usePropsStreamingStatus(\n    componentProps,\n    generationStage,\n    message?.id ?? \"\",\n  );\n\n  /** Derive global stream status from prop statuses and generation stage */\n  const streamStatus = useMemo(() => {\n    const generationError = message?.error\n      ? new Error(message.error)\n      : undefined;\n    const hasComponent = !!message?.component;\n    return deriveGlobalStreamStatus(\n      generationStage,\n      propStatus,\n      hasComponent,\n      generationError,\n    );\n  }, [generationStage, propStatus, message]);\n\n  return {\n    streamStatus,\n    propStatus,\n  };\n}\n"]}
+{"version":3,"file":"use-tambo-stream-status.js","sourceRoot":"","sources":["../../src/hooks/use-tambo-stream-status.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;AAgUb,oDAwCC;AAvWD,iCAAqD;AACrD,sFAAuE;AACvE,8EAA6E;AAC7E,+DAA+D;AAqE/D;;;;GAIG;AACH,SAAS,gBAAgB;IACvB,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CACb,6DAA6D;YAC3D,4CAA4C;YAC5C,6DAA6D,CAChE,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,uBAAuB,CAC9B,KAAwB,EACxB,eAAgC,EAChC,SAAiB;IAEjB,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,IAAA,gBAAQ,EAU9C,EAAE,CAAC,CAAC;IAEN,mDAAmD;IACnD,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,0DAA0D;YAC1D,MAAM,iBAAiB,GAAG,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAChD,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,KAAK,SAAS,CAC5D,CAAC;YACF,OAAO,iBAAiB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;QACvC,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;IAEhB,oFAAoF;IACpF,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,MAAM,OAAO,GAAG,EAAE,GAAG,IAAI,EAAE,CAAC;YAC5B,IAAI,UAAU,GAAG,KAAK,CAAC;YAEvB,oDAAoD;YACpD,MAAM,gBAAgB,GAAa,EAAE,CAAC;YACtC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;gBAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI;oBAC3B,UAAU,EAAE,KAAK;oBACjB,UAAU,EAAE,KAAK;iBAClB,CAAC;gBAEF,+EAA+E;gBAC/E,MAAM,UAAU,GACd,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,EAAE,CAAC;gBACxD,MAAM,WAAW,GAAG,UAAU,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;gBAEtD,IAAI,WAAW,EAAE,CAAC;oBAChB,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBAC7B,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,mEAAmE;YACnE,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;gBAC7C,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI;oBAC3B,UAAU,EAAE,KAAK;oBACjB,UAAU,EAAE,KAAK;iBAClB,CAAC;gBAEF,+EAA+E;gBAC/E,MAAM,UAAU,GACd,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,EAAE,CAAC;gBACxD,MAAM,WAAW,GAAG,UAAU,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;gBAEtD;;;;mBAIG;gBACH,MAAM,uBAAuB,GAAG,gBAAgB,CAAC,IAAI,CACnD,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,KAAK,GAAG,CACrC,CAAC;gBACF,MAAM,oBAAoB,GACxB,eAAe,KAAK,6CAAe,CAAC,QAAQ,CAAC;gBAC/C,MAAM,UAAU,GACd,OAAO,CAAC,UAAU;oBAClB,CAAC,uBAAuB,IAAI,oBAAoB,CAAC;oBACjD,CAAC,OAAO,CAAC,UAAU,CAAC;gBAEtB,8DAA8D;gBAC9D,IAAI,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,SAAS,KAAK,SAAS,EAAE,CAAC;oBAC1D,2CAA2C;oBAC3C,OAAO;gBACT,CAAC;gBAED,IAAI,WAAW,IAAI,UAAU,EAAE,CAAC;oBAC9B,OAAO,CAAC,GAAG,CAAC,GAAG;wBACb,GAAG,OAAO;wBACV,UAAU,EAAE,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU;wBACnD,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU;wBAClD,SAAS;qBACV,CAAC;oBACF,UAAU,GAAG,IAAI,CAAC;gBACpB,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,OAAO,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;QACrC,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,KAAK,EAAE,eAAe,EAAE,SAAS,CAAC,CAAC,CAAC;IAExC,mDAAmD;IACnD,OAAO,IAAA,eAAO,EAAC,GAAG,EAAE;QAClB,IAAI,CAAC,KAAK;YAAE,OAAO,EAAqC,CAAC;QAEzD,MAAM,MAAM,GAAG,EAAqC,CAAC;QAErD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YACjC,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,IAAI;gBACpC,UAAU,EAAE,KAAK;gBACjB,UAAU,EAAE,KAAK;gBACjB,SAAS,EAAE,EAAE;aACd,CAAC;YAEF,+DAA+D;YAC/D,MAAM,wBAAwB,GAC5B,QAAQ,CAAC,UAAU,IAAI,QAAQ,CAAC,SAAS,KAAK,SAAS,CAAC;YAE1D,sFAAsF;YACtF,MAAM,qBAAqB,GACzB,CAAC,wBAAwB;gBACzB,eAAe,KAAK,6CAAe,CAAC,kBAAkB,CAAC;YAEzD,MAAM,CAAC,GAAkB,CAAC,GAAG;gBAC3B,SAAS,EAAE,CAAC,QAAQ,CAAC,UAAU,IAAI,CAAC,wBAAwB;gBAC5D,WAAW,EACT,QAAQ,CAAC,UAAU;oBACnB,CAAC,wBAAwB;oBACzB,qBAAqB;gBACvB,SAAS,EAAE,wBAAwB;gBACnC,KAAK,EAAE,QAAQ,CAAC,KAAK;aACtB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAChB,CAAC,EAAE,CAAC,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,SAAS,CAAC,CAAC,CAAC;AACxD,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,wBAAwB,CAC/B,eAAgC,EAChC,UAA2C,EAC3C,YAAqB,EACrB,eAAuB;IAEvB,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IAE/C,gGAAgG;IAChG,MAAM,kBAAkB,GACtB,YAAY,CAAC,MAAM,GAAG,CAAC,IAAI,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IAEpE,+DAA+D;IAC/D,MAAM,qBAAqB,GACzB,CAAC,kBAAkB;QACnB,eAAe,KAAK,6CAAe,CAAC,kBAAkB,CAAC;IACzD,MAAM,iBAAiB,GAAG,eAAe,KAAK,6CAAe,CAAC,KAAK,CAAC;IAEpE,mDAAmD;IACnD,MAAM,UAAU,GACd,eAAe,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,KAAK,CAAC;IAE9D,OAAO;QACL,+EAA+E;QAC/E,SAAS,EACP,CAAC,YAAY;YACb,CAAC,CAAC,qBAAqB;gBACrB,CAAC,kBAAkB;gBACnB,YAAY,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAE3C,iGAAiG;QACjG,WAAW,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC;QAEpD,oFAAoF;QACpF,SAAS,EAAE,kBAAkB;QAE7B,kDAAkD;QAClD,OAAO,EACL,iBAAiB;YACjB,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;YACjC,CAAC,CAAC,eAAe;QAEnB,WAAW,EAAE,UAAU;KACxB,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,SAAgB,oBAAoB;IAMlC,oDAAoD;IACpD,gBAAgB,EAAE,CAAC;IAEnB,MAAM,EAAE,eAAe,EAAE,GAAG,IAAA,+CAAuB,GAAE,CAAC;IACtD,MAAM,OAAO,GAAG,IAAA,4CAAsB,GAAE,CAAC;IAEzC,uDAAuD;IACvD,MAAM,cAAc,GAAI,OAAO,EAAE,SAAS,EAAE,KAAe,IAAK,EAAY,CAAC;IAE7E,sCAAsC;IACtC,MAAM,UAAU,GAAG,uBAAuB,CACxC,cAAc,EACd,eAAe,EACf,OAAO,EAAE,EAAE,IAAI,EAAE,CAClB,CAAC;IAEF,0EAA0E;IAC1E,MAAM,YAAY,GAAG,IAAA,eAAO,EAAC,GAAG,EAAE;QAChC,MAAM,eAAe,GAAG,OAAO,EAAE,KAAK;YACpC,CAAC,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;YAC1B,CAAC,CAAC,SAAS,CAAC;QACd,MAAM,YAAY,GAAG,CAAC,CAAC,OAAO,EAAE,SAAS,CAAC;QAC1C,OAAO,wBAAwB,CAC7B,eAAe,EACf,UAAU,EACV,YAAY,EACZ,eAAe,CAChB,CAAC;IACJ,CAAC,EAAE,CAAC,eAAe,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;IAE3C,OAAO;QACL,YAAY;QACZ,UAAU;KACX,CAAC;AACJ,CAAC","sourcesContent":["\"use client\";\nimport { useEffect, useMemo, useState } from \"react\";\nimport { GenerationStage } from \"../model/generate-component-response\";\nimport { useTamboGenerationStage } from \"../providers/tambo-thread-provider\";\nimport { useTamboCurrentMessage } from \"./use-current-message\";\n\n/**\n * Global stream status flags for a specific component in a message.\n * Represents the aggregate state across all props for this component only.\n * Once a component completes, its status remains stable regardless of other generations.\n */\nexport interface StreamStatus {\n  /**\n   * Indicates no tokens have been received for any prop and generation is not active.\n   * Useful for showing initial loading states before any data arrives.\n   */\n  isPending: boolean;\n\n  /**\n   * Indicates active streaming - either generation is streaming OR at least one prop is still streaming.\n   * Use this to show loading animations or skeleton states during data transmission.\n   */\n  isStreaming: boolean;\n\n  /**\n   * Indicates successful completion - generation is complete AND every prop finished without error.\n   * Safe to render the final component when this is true.\n   */\n  isSuccess: boolean;\n\n  /**\n   * Indicates a fatal error occurred in any prop or the stream itself.\n   * Check streamError for details about what went wrong.\n   */\n  isError: boolean;\n\n  /**\n   * The first fatal error encountered during streaming (if any).\n   * Will be undefined if no errors occurred.\n   */\n  streamError?: Error;\n}\n\n/**\n * Streaming status flags for individual component props.\n * Tracks the state of each prop as it streams from the LLM.\n */\nexport interface PropStatus {\n  /**\n   * Indicates no tokens have been received for this specific prop yet.\n   * The prop value is still undefined, null, or empty string.\n   */\n  isPending: boolean;\n\n  /**\n   * Indicates at least one token has been received but streaming is not complete.\n   * The prop has partial content that may still be updating.\n   */\n  isStreaming: boolean;\n\n  /**\n   * Indicates this prop has finished streaming successfully.\n   * The prop value is complete and stable.\n   */\n  isSuccess: boolean;\n\n  /**\n   * The error that occurred during streaming (if any).\n   * Will be undefined if no error occurred for this prop.\n   */\n  error?: Error;\n}\n\n/**\n * SSR Guard - throws during server-side rendering.\n * Ensures the hook is only used in browser contexts.\n * @throws {Error} When called during server-side rendering\n */\nfunction assertClientSide() {\n  if (typeof window === \"undefined\") {\n    throw new Error(\n      \"useTamboStreamStatus can only be used in browser contexts. \" +\n        \"This hook is not compatible with SSR/SSG. \" +\n        \"Consider wrapping it in useEffect or using dynamic imports.\",\n    );\n  }\n}\n\n/**\n * Track streaming status for individual props by monitoring their values.\n * Monitors when props receive their first token and when they complete streaming.\n * Maintains stable state per message - once props complete for a message, they stay complete.\n * @template Props - The type of the component props being tracked\n * @param props - The current component props object\n * @param generationStage - The current generation stage from the LLM\n * @param messageId - The ID of the current message to track component-specific state\n * @returns A record mapping each prop key to its PropStatus\n */\nfunction usePropsStreamingStatus<Props extends Record<string, any>>(\n  props: Props | undefined,\n  generationStage: GenerationStage,\n  messageId: string,\n): Record<keyof Props, PropStatus> {\n  const [propTracking, setPropTracking] = useState<\n    Record<\n      string,\n      {\n        hasStarted: boolean;\n        isComplete: boolean;\n        error?: Error;\n        messageId: string;\n      }\n    >\n  >({});\n\n  /** Reset tracking only when the message changes */\n  useEffect(() => {\n    setPropTracking((prev) => {\n      // If we have tracking data for a different message, reset\n      const hasOldMessageData = Object.values(prev).some(\n        (track) => track.messageId && track.messageId !== messageId,\n      );\n      return hasOldMessageData ? {} : prev;\n    });\n  }, [messageId]);\n\n  /** Track when props start streaming (receive first token) and when they complete */\n  useEffect(() => {\n    if (!props) return;\n\n    setPropTracking((prev) => {\n      const updated = { ...prev };\n      let hasChanges = false;\n\n      // First pass: identify which props are starting now\n      const propsStartingNow: string[] = [];\n      Object.entries(props).forEach(([key, value]) => {\n        const current = prev[key] || {\n          hasStarted: false,\n          isComplete: false,\n        };\n\n        /** A prop starts streaming when it has a non-empty value for the first time */\n        const hasContent =\n          value !== undefined && value !== null && value !== \"\";\n        const justStarted = hasContent && !current.hasStarted;\n\n        if (justStarted) {\n          propsStartingNow.push(key);\n        }\n      });\n\n      // Second pass: update tracking and mark previous props as complete\n      Object.entries(props).forEach(([key, value]) => {\n        const current = prev[key] || {\n          hasStarted: false,\n          isComplete: false,\n        };\n\n        /** A prop starts streaming when it has a non-empty value for the first time */\n        const hasContent =\n          value !== undefined && value !== null && value !== \"\";\n        const justStarted = hasContent && !current.hasStarted;\n\n        /**\n         * A prop is complete when it has started and either:\n         * 1. A following prop has started, OR\n         * 2. Generation is complete (for the final prop)\n         */\n        const hasFollowingPropStarted = propsStartingNow.some(\n          (startingKey) => startingKey !== key,\n        );\n        const isGenerationComplete =\n          generationStage === GenerationStage.COMPLETE;\n        const isComplete =\n          current.hasStarted &&\n          (hasFollowingPropStarted || isGenerationComplete) &&\n          !current.isComplete;\n\n        // Once a prop is complete for this message, it stays complete\n        if (current.isComplete && current.messageId === messageId) {\n          // Skip - already complete for this message\n          return;\n        }\n\n        if (justStarted || isComplete) {\n          updated[key] = {\n            ...current,\n            hasStarted: justStarted ? true : current.hasStarted,\n            isComplete: isComplete ? true : current.isComplete,\n            messageId,\n          };\n          hasChanges = true;\n        }\n      });\n\n      return hasChanges ? updated : prev;\n    });\n  }, [props, generationStage, messageId]);\n\n  /** Convert tracking state to PropStatus objects */\n  return useMemo(() => {\n    if (!props) return {} as Record<keyof Props, PropStatus>;\n\n    const result = {} as Record<keyof Props, PropStatus>;\n\n    Object.keys(props).forEach((key) => {\n      const tracking = propTracking[key] || {\n        hasStarted: false,\n        isComplete: false,\n        messageId: \"\",\n      };\n\n      // If this prop is complete for this message, it stays complete\n      const isCompleteForThisMessage =\n        tracking.isComplete && tracking.messageId === messageId;\n\n      // Only consider generation stage if this prop isn't already complete for this message\n      const isGenerationStreaming =\n        !isCompleteForThisMessage &&\n        generationStage === GenerationStage.STREAMING_RESPONSE;\n\n      result[key as keyof Props] = {\n        isPending: !tracking.hasStarted && !isCompleteForThisMessage,\n        isStreaming:\n          tracking.hasStarted &&\n          !isCompleteForThisMessage &&\n          isGenerationStreaming,\n        isSuccess: isCompleteForThisMessage,\n        error: tracking.error,\n      };\n    });\n\n    return result;\n  }, [props, propTracking, generationStage, messageId]);\n}\n\n/**\n * Derives global StreamStatus from generation stage and individual prop statuses.\n * Aggregates individual prop states into a unified stream status.\n * @template Props - The type of the component props\n * @param generationStage - The current generation stage from the LLM\n * @param propStatus - Status record for each individual prop\n * @param hasComponent - Whether a component exists in the current message\n * @param generationError - Any error from the generation process itself\n * @returns The aggregated StreamStatus for the entire component\n */\nfunction deriveGlobalStreamStatus<Props extends Record<string, any>>(\n  generationStage: GenerationStage,\n  propStatus: Record<keyof Props, PropStatus>,\n  hasComponent: boolean,\n  generationError?: Error,\n): StreamStatus {\n  const propStatuses = Object.values(propStatus);\n\n  // If all props are already successful, the component is complete regardless of generation stage\n  const allPropsSuccessful =\n    propStatuses.length > 0 && propStatuses.every((p) => p.isSuccess);\n\n  // Only consider generation stage if not all props are complete\n  const isGenerationStreaming =\n    !allPropsSuccessful &&\n    generationStage === GenerationStage.STREAMING_RESPONSE;\n  const isGenerationError = generationStage === GenerationStage.ERROR;\n\n  /** Find first error from generation or any prop */\n  const firstError =\n    generationError ?? propStatuses.find((p) => p.error)?.error;\n\n  return {\n    /** isPending: no component yet OR (has component but no props have started) */\n    isPending:\n      !hasComponent ||\n      (!isGenerationStreaming &&\n        !allPropsSuccessful &&\n        propStatuses.every((p) => p.isPending)),\n\n    /** isStreaming: any prop is streaming (generation stage doesn't matter if props are complete) */\n    isStreaming: propStatuses.some((p) => p.isStreaming),\n\n    /** isSuccess: all props successful (component is stable once all props complete) */\n    isSuccess: allPropsSuccessful,\n\n    /** isError: generation error OR any prop error */\n    isError:\n      isGenerationError ||\n      propStatuses.some((p) => p.error) ||\n      !!generationError,\n\n    streamError: firstError,\n  };\n}\n\n/**\n * Track streaming status for Tambo component props.\n *\n * **Important**: Props update repeatedly during streaming and may be partial.\n * Use `propStatus.<field>?.isSuccess` before treating a prop as complete.\n *\n * Pair with `useTamboComponentState` to disable inputs while streaming.\n * @see {@link https://docs.tambo.co/concepts/streaming/streaming-best-practices}\n * @template Props - Component props type\n * @returns `streamStatus` (overall) and `propStatus` (per-prop) flags\n * @throws {Error} When used during SSR/SSG\n * @example\n * ```tsx\n * // Wait for entire stream\n * const { streamStatus } = useTamboStreamStatus();\n * if (!streamStatus.isSuccess) return <Spinner />;\n * return <Card {...props} />;\n * ```\n * @example\n * ```tsx\n * // Highlight in-flight props\n * const { propStatus } = useTamboStreamStatus<Props>();\n * <h2 className={propStatus.title.isStreaming ? \"animate-pulse\" : \"\"}>\n *   {title}\n * </h2>\n * ```\n */\nexport function useTamboStreamStatus<\n  Props extends Record<string, any> = Record<string, any>,\n>(): {\n  streamStatus: StreamStatus;\n  propStatus: Record<keyof Props, PropStatus>;\n} {\n  /** SSR Guard - ensure client-side only execution */\n  assertClientSide();\n\n  const { generationStage } = useTamboGenerationStage();\n  const message = useTamboCurrentMessage();\n\n  /** Get the current component props from the message */\n  const componentProps = (message?.component?.props as Props) || ({} as Props);\n\n  /** Track per-prop streaming status */\n  const propStatus = usePropsStreamingStatus(\n    componentProps,\n    generationStage,\n    message?.id ?? \"\",\n  );\n\n  /** Derive global stream status from prop statuses and generation stage */\n  const streamStatus = useMemo(() => {\n    const generationError = message?.error\n      ? new Error(message.error)\n      : undefined;\n    const hasComponent = !!message?.component;\n    return deriveGlobalStreamStatus(\n      generationStage,\n      propStatus,\n      hasComponent,\n      generationError,\n    );\n  }, [generationStage, propStatus, message]);\n\n  return {\n    streamStatus,\n    propStatus,\n  };\n}\n"]}

package/dist/hooks/use-tambo-stream-status.test.d.ts.map

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

package/dist/hooks/{__tests__/use-tambo-stream-status.test.js → use-tambo-stream-status.test.js}

@@ -1,18 +1,18 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 const react_1 = require("@testing-library/react");
-const generate_component_response_1 = require("../../model/generate-component-response");
-const use_tambo_stream_status_1 = require("../use-tambo-stream-status");
+const generate_component_response_1 = require("../model/generate-component-response");
+const use_tambo_stream_status_1 = require("./use-tambo-stream-status");
 // Mock the required providers
-jest.mock("../../providers/tambo-thread-provider", () => ({
+jest.mock("../providers/tambo-thread-provider", () => ({
     useTamboGenerationStage: jest.fn(),
 }));
-jest.mock("../use-current-message", () => ({
+jest.mock("./use-current-message", () => ({
     useTamboCurrentMessage: jest.fn(),
 }));
 // Import the mocked functions
-const tambo_thread_provider_1 = require("../../providers/tambo-thread-provider");
-const use_current_message_1 = require("../use-current-message");
+const tambo_thread_provider_1 = require("../providers/tambo-thread-provider");
+const use_current_message_1 = require("./use-current-message");
 // Mock window for SSR tests
 const originalWindow = global.window;
 // Helper function to create mock ComponentDecisionV2

package/dist/hooks/use-tambo-stream-status.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-tambo-stream-status.test.js","sourceRoot":"","sources":["../../src/hooks/use-tambo-stream-status.test.tsx"],"names":[],"mappings":";;AAAA,kDAAoD;AACpD,sFAG8C;AAC9C,uEAAiE;AAEjE,8BAA8B;AAC9B,IAAI,CAAC,IAAI,CAAC,oCAAoC,EAAE,GAAG,EAAE,CAAC,CAAC;IACrD,uBAAuB,EAAE,IAAI,CAAC,EAAE,EAAE;CACnC,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC;IACxC,sBAAsB,EAAE,IAAI,CAAC,EAAE,EAAE;CAClC,CAAC,CAAC,CAAC;AAEJ,8BAA8B;AAC9B,8EAA6E;AAC7E,+DAA+D;AAE/D,4BAA4B;AAC5B,MAAM,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC;AAErC,qDAAqD;AACrD,MAAM,mBAAmB,GAAG,CAAC,QAAiC,EAAE,EAAO,EAAE,CAAC,CAAC;IACzE,aAAa,EAAE,eAAe;IAC9B,cAAc,EAAE,EAAE;IAClB,OAAO,EAAE,qBAAqB;IAC9B,KAAK;IACL,SAAS,EAAE,gBAAgB;CAC5B,CAAC,CAAC;AAEH,oDAAoD;AACpD,MAAM,iBAAiB,GAAG,CACxB,YAAyC,EAAE,EACvB,EAAE,CAAC,CAAC;IACxB,EAAE,EAAE,cAAc;IAClB,cAAc,EAAE,EAAE;IAClB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;IACjD,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;IACnC,IAAI,EAAE,WAAW;IACjB,QAAQ,EAAE,aAAa;IACvB,GAAG,SAAS;CACb,CAAC,CAAC;AAEH,2BAA2B;AAC3B,MAAM,2BAA2B,GAAG,IAAI,CAAC,MAAM,CAAC,+CAAuB,CAAC,CAAC;AACzE,MAAM,0BAA0B,GAAG,IAAI,CAAC,MAAM,CAAC,4CAAsB,CAAC,CAAC;AAEvE,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,UAAU,CAAC,GAAG,EAAE;QACd,uCAAuC;QACvC,MAAM,CAAC,MAAM,GAAG,cAAc,CAAC;QAE/B,+BAA+B;QAC/B,2BAA2B,CAAC,eAAe,CAAC;YAC1C,eAAe,EAAE,6CAAe,CAAC,IAAI;YACrC,uBAAuB,EAAE,EAAE;YAC3B,MAAM,EAAE,IAAI;SACb,CAAC,CAAC;QAEH,0BAA0B,CAAC,eAAe,CAAC;YACzC,EAAE,EAAE,cAAc;YAClB,SAAS,EAAE;gBACT,KAAK,EAAE,EAAE;aACV;SACoB,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,2BAA2B,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,6CAAe,CAAC,IAAI;gBACrC,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,IAAI;aACb,CAAC,CAAC;YAEH,0BAA0B,CAAC,eAAe,CACxC,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;aACxD,CAAC,CACH,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,GAAmC,CACxD,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC;gBAC1C,SAAS,EAAE,IAAI;gBACf,WAAW,EAAE,KAAK;gBAClB,SAAS,EAAE,KAAK;gBAChB,OAAO,EAAE,KAAK;gBACd,WAAW,EAAE,SAAS;aACvB,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC;gBAC9C,SAAS,EAAE,IAAI;gBACf,WAAW,EAAE,KAAK;gBAClB,SAAS,EAAE,KAAK;gBAChB,KAAK,EAAE,SAAS;aACjB,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC;gBAC7C,SAAS,EAAE,IAAI;gBACf,WAAW,EAAE,KAAK;gBAClB,SAAS,EAAE,KAAK;gBAChB,KAAK,EAAE,SAAS;aACjB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,uGAAuG,EAAE,GAAG,EAAE;YAC/G,2BAA2B,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,6CAAe,CAAC,kBAAkB;gBACnD,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YAEH,0BAA0B,CAAC,eAAe,CACxC,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;aACxD,CAAC,CACH,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,GAAmC,CACxD,CAAC;YAEF,+EAA+E;YAC/E,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAE1D,gFAAgF;YAChF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iFAAiF,EAAE,GAAG,EAAE;YACzF,2BAA2B,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,6CAAe,CAAC,kBAAkB;gBACnD,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YAEH,0BAA0B,CAAC,eAAe,CACxC,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;aAC7D,CAAC,CACH,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,GAAmC,CACxD,CAAC;YAEF,sDAAsD;YACtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAE9D,4DAA4D;YAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAE5D,oEAAoE;YACpE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;YAC1E,+BAA+B;YAC/B,2BAA2B,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,6CAAe,CAAC,IAAI;gBACrC,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,IAAI;aACb,CAAC,CAAC;YAEH,0BAA0B,CAAC,eAAe,CACxC,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;aACxD,CAAC,CACH,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAC3C,IAAA,8CAAoB,GAAmC,CACxD,CAAC;YAEF,mCAAmC;YACnC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAE1D,+DAA+D;YAC/D,2BAA2B,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,6CAAe,CAAC,kBAAkB;gBACnD,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YAEH,0BAA0B,CAAC,eAAe,CACxC,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC;oBAC7B,KAAK,EAAE,aAAa;oBACpB,IAAI,EAAE,cAAc;iBACrB,CAAC;aACH,CAAC,CACH,CAAC;YAEF,QAAQ,EAAE,CAAC;YAEX,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAE1D,uCAAuC;YACvC,2BAA2B,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,6CAAe,CAAC,QAAQ;gBACzC,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YAEH,QAAQ,EAAE,CAAC;YAEX,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,2BAA2B,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,6CAAe,CAAC,KAAK;gBACtC,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,mBAAmB,CAAC;YACzC,0BAA0B,CAAC,eAAe,CACxC,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;gBACvD,KAAK,EAAE,YAAY;aACpB,CAAC,CACH,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,GAAmC,CACxD,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,IAAI,CAC3D,YAAY,CACb,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,kFAAkF,EAAE,GAAG,EAAE;YAC1F,2BAA2B,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,6CAAe,CAAC,IAAI;gBACrC,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,IAAI;aACb,CAAC,CAAC;YAEH,0BAA0B,CAAC,eAAe,CACxC,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;aACpE,CAAC,CACH,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,GAIhB,CACL,CAAC;YAEF,mDAAmD;YACnD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kFAAkF,EAAE,GAAG,EAAE;YAC1F,2BAA2B,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,6CAAe,CAAC,QAAQ;gBACzC,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YAEH,2BAA2B;YAC3B,0BAA0B,CAAC,eAAe,CACxC,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC;oBAC7B,KAAK,EAAE,gBAAgB;oBACvB,IAAI,EAAE,oBAAoB;iBAC3B,CAAC;aACH,CAAC,CACH,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,GAAmC,CACxD,CAAC;YAEF,6DAA6D;YAC7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,mHAAmH;QAClL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kFAAkF,EAAE,GAAG,EAAE;YAC1F,4CAA4C;YAC5C,2BAA2B,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,6CAAe,CAAC,kBAAkB;gBACnD,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YAEH,0BAA0B,CAAC,eAAe,CACxC,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC;oBAC7B,KAAK,EAAE,EAAE;oBACT,IAAI,EAAE,EAAE;iBACT,CAAC;aACH,CAAC,CACH,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAC3C,IAAA,8CAAoB,GAAmC,CACxD,CAAC;YAEF,sCAAsC;YACtC,0BAA0B,CAAC,eAAe,CACxC,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC;oBAC7B,KAAK,EAAE,gBAAgB;oBACvB,IAAI,EAAE,EAAE;iBACT,CAAC;aACH,CAAC,CACH,CAAC;YACF,QAAQ,EAAE,CAAC;YAEX,qCAAqC;YACrC,0BAA0B,CAAC,eAAe,CACxC,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC;oBAC7B,KAAK,EAAE,gBAAgB;oBACvB,IAAI,EAAE,eAAe;iBACtB,CAAC;aACH,CAAC,CACH,CAAC;YACF,QAAQ,EAAE,CAAC;YAEX,8BAA8B;YAC9B,2BAA2B,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,6CAAe,CAAC,QAAQ;gBACzC,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YACH,QAAQ,EAAE,CAAC;YAEX,sCAAsC;YACtC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;YAC9E,2BAA2B,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,6CAAe,CAAC,QAAQ;gBACzC,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YAEH,0BAA0B,CAAC,eAAe,CACxC,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC;oBAC7B,KAAK,EAAE,YAAY;oBACnB,IAAI,EAAE,WAAW;iBAClB,CAAC;gBACF,KAAK,EAAE,sBAAsB;aAC9B,CAAC,CACH,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,8CAAoB,GAAmC,CACxD,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,IAAI,CAC3D,sBAAsB,CACvB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;QAC3B,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;YAOpE,0BAA0B,CAAC,eAAe,CACxC,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC;oBAC7B,KAAK,EAAE,MAAM;oBACb,WAAW,EAAE,WAAW;oBACxB,KAAK,EAAE,EAAE;iBACV,CAAC;aACH,CAAC,CACH,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,8CAAoB,GAAa,CAAC,CAAC;YAEvE,+CAA+C;YAC/C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;YACtD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;YAEtD,sDAAsD;YACtD,mEAAmE;QACrE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,0BAA0B,CAAC,eAAe,CACxC,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC;aACzD,CAAC,CACH,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,8CAAoB,GAAE,CAAC,CAAC;YAE5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,WAAW,EAAE,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;QAClD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,0BAA0B,CAAC,eAAe,CAAC;gBACzC,EAAE,EAAE,cAAc;gBAClB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,EAAE;gBACX,cAAc,EAAE,EAAE;gBAClB,SAAS,EAAE,EAAE;gBACb,QAAQ,EAAE,EAAE;aACS,CAAC,CAAC;YAEzB,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,8CAAoB,GAAE,CAAC,CAAC;YAE5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,0BAA0B,CAAC,eAAe,CAAC;gBACzC,EAAE,EAAE,cAAc;gBAClB,6BAA6B;aACR,CAAC,CAAC;YAEzB,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,8CAAoB,GAAE,CAAC,CAAC;YAE5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,6BAA6B;YAC7B,2BAA2B,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,6CAAe,CAAC,kBAAkB;gBACnD,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YACH,0BAA0B,CAAC,eAAe,CACxC,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;aAC9C,CAAC,CACH,CAAC;YACF,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAC3C,IAAA,8CAAoB,GAAqB,CAC1C,CAAC;YACF,8BAA8B;YAC9B,0BAA0B,CAAC,eAAe,CACxC,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;aACzD,CAAC,CACH,CAAC;YACF,QAAQ,EAAE,CAAC;YACX,sBAAsB;YACtB,2BAA2B,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,6CAAe,CAAC,QAAQ;gBACzC,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YACH,QAAQ,EAAE,CAAC;YACX,iCAAiC;YACjC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAE7D,sEAAsE;YACtE,2BAA2B,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,6CAAe,CAAC,kBAAkB;gBACnD,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YACH,0BAA0B,CAAC,eAAe,CACxC,iBAAiB,CAAC;gBAChB,EAAE,EAAE,gBAAgB,EAAE,wCAAwC;gBAC9D,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;aAC9C,CAAC,CACH,CAAC;YACF,QAAQ,EAAE,CAAC;YACX,0BAA0B;YAC1B,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { renderHook } from \"@testing-library/react\";\nimport {\n  GenerationStage,\n  TamboThreadMessage,\n} from \"../model/generate-component-response\";\nimport { useTamboStreamStatus } from \"./use-tambo-stream-status\";\n\n// Mock the required providers\njest.mock(\"../providers/tambo-thread-provider\", () => ({\n  useTamboGenerationStage: jest.fn(),\n}));\n\njest.mock(\"./use-current-message\", () => ({\n  useTamboCurrentMessage: jest.fn(),\n}));\n\n// Import the mocked functions\nimport { useTamboGenerationStage } from \"../providers/tambo-thread-provider\";\nimport { useTamboCurrentMessage } from \"./use-current-message\";\n\n// Mock window for SSR tests\nconst originalWindow = global.window;\n\n// Helper function to create mock ComponentDecisionV2\nconst createMockComponent = (props: Record<string, unknown> = {}): any => ({\n  componentName: \"TestComponent\",\n  componentState: {},\n  message: \"Component generated\",\n  props,\n  reasoning: \"Test reasoning\",\n});\n\n// Helper function to create mock TamboThreadMessage\nconst createMockMessage = (\n  overrides: Partial<TamboThreadMessage> = {},\n): TamboThreadMessage => ({\n  id: \"test-message\",\n  componentState: {},\n  content: [{ type: \"text\", text: \"test content\" }],\n  createdAt: new Date().toISOString(),\n  role: \"assistant\",\n  threadId: \"test-thread\",\n  ...overrides,\n});\n\n// Get the mocked functions\nconst mockUseTamboGenerationStage = jest.mocked(useTamboGenerationStage);\nconst mockUseTamboCurrentMessage = jest.mocked(useTamboCurrentMessage);\n\ndescribe(\"useTamboStreamStatus\", () => {\n  beforeEach(() => {\n    // Restore window for client-side tests\n    global.window = originalWindow;\n\n    // Default mock implementations\n    mockUseTamboGenerationStage.mockReturnValue({\n      generationStage: GenerationStage.IDLE,\n      generationStatusMessage: \"\",\n      isIdle: true,\n    });\n\n    mockUseTamboCurrentMessage.mockReturnValue({\n      id: \"test-message\",\n      component: {\n        props: {},\n      },\n    } as TamboThreadMessage);\n  });\n\n  afterEach(() => {\n    jest.clearAllMocks();\n  });\n\n  describe(\"Initial State\", () => {\n    it(\"should start with all flags as pending when idle and no props\", () => {\n      mockUseTamboGenerationStage.mockReturnValue({\n        generationStage: GenerationStage.IDLE,\n        generationStatusMessage: \"\",\n        isIdle: true,\n      });\n\n      mockUseTamboCurrentMessage.mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({ title: \"\", body: \"\" }),\n        }),\n      );\n\n      const { result } = renderHook(() =>\n        useTamboStreamStatus<{ title: string; body: string }>(),\n      );\n\n      expect(result.current.streamStatus).toEqual({\n        isPending: true,\n        isStreaming: false,\n        isSuccess: false,\n        isError: false,\n        streamError: undefined,\n      });\n\n      expect(result.current.propStatus.title).toEqual({\n        isPending: true,\n        isStreaming: false,\n        isSuccess: false,\n        error: undefined,\n      });\n\n      expect(result.current.propStatus.body).toEqual({\n        isPending: true,\n        isStreaming: false,\n        isSuccess: false,\n        error: undefined,\n      });\n    });\n  });\n\n  describe(\"Generation vs Props Streaming\", () => {\n    it(\"should show generation streaming but props still pending when STREAMING_RESPONSE with no prop content\", () => {\n      mockUseTamboGenerationStage.mockReturnValue({\n        generationStage: GenerationStage.STREAMING_RESPONSE,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n\n      mockUseTamboCurrentMessage.mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({ title: \"\", body: \"\" }),\n        }),\n      );\n\n      const { result } = renderHook(() =>\n        useTamboStreamStatus<{ title: string; body: string }>(),\n      );\n\n      // Global streaming should be false because no props are actually streaming yet\n      expect(result.current.streamStatus.isStreaming).toBe(false);\n      expect(result.current.streamStatus.isPending).toBe(false);\n\n      // Individual props should be pending because they haven't started streaming yet\n      expect(result.current.propStatus.title.isPending).toBe(true);\n      expect(result.current.propStatus.title.isStreaming).toBe(false);\n      expect(result.current.propStatus.body.isPending).toBe(true);\n      expect(result.current.propStatus.body.isStreaming).toBe(false);\n    });\n\n    it(\"should show prop streaming when props receive content during STREAMING_RESPONSE\", () => {\n      mockUseTamboGenerationStage.mockReturnValue({\n        generationStage: GenerationStage.STREAMING_RESPONSE,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n\n      mockUseTamboCurrentMessage.mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({ title: \"Hello\", body: \"\" }),\n        }),\n      );\n\n      const { result } = renderHook(() =>\n        useTamboStreamStatus<{ title: string; body: string }>(),\n      );\n\n      // Title prop should be streaming since it has content\n      expect(result.current.propStatus.title.isStreaming).toBe(true);\n      expect(result.current.propStatus.title.isPending).toBe(false);\n\n      // Body prop should still be pending since it has no content\n      expect(result.current.propStatus.body.isStreaming).toBe(false);\n      expect(result.current.propStatus.body.isPending).toBe(true);\n\n      // Global should be streaming because at least one prop is streaming\n      expect(result.current.streamStatus.isStreaming).toBe(true);\n    });\n  });\n\n  describe(\"Boolean Lifecycle\", () => {\n    it(\"should transition through Init -> Streaming -> Success lifecycle\", () => {\n      // Start with IDLE (Init phase)\n      mockUseTamboGenerationStage.mockReturnValue({\n        generationStage: GenerationStage.IDLE,\n        generationStatusMessage: \"\",\n        isIdle: true,\n      });\n\n      mockUseTamboCurrentMessage.mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({ title: \"\", body: \"\" }),\n        }),\n      );\n\n      const { result, rerender } = renderHook(() =>\n        useTamboStreamStatus<{ title: string; body: string }>(),\n      );\n\n      // Phase 1: Init - isPending = true\n      expect(result.current.streamStatus.isPending).toBe(true);\n      expect(result.current.streamStatus.isStreaming).toBe(false);\n      expect(result.current.streamStatus.isSuccess).toBe(false);\n\n      // Phase 2: Streaming - move to STREAMING_RESPONSE with content\n      mockUseTamboGenerationStage.mockReturnValue({\n        generationStage: GenerationStage.STREAMING_RESPONSE,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n\n      mockUseTamboCurrentMessage.mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({\n            title: \"Hello World\",\n            body: \"Some content\",\n          }),\n        }),\n      );\n\n      rerender();\n\n      expect(result.current.streamStatus.isPending).toBe(false);\n      expect(result.current.streamStatus.isStreaming).toBe(true);\n      expect(result.current.streamStatus.isSuccess).toBe(false);\n\n      // Phase 3: Complete - move to COMPLETE\n      mockUseTamboGenerationStage.mockReturnValue({\n        generationStage: GenerationStage.COMPLETE,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n\n      rerender();\n\n      expect(result.current.streamStatus.isPending).toBe(false);\n      expect(result.current.streamStatus.isStreaming).toBe(false);\n      expect(result.current.streamStatus.isSuccess).toBe(true);\n    });\n\n    it(\"should handle error state correctly\", () => {\n      mockUseTamboGenerationStage.mockReturnValue({\n        generationStage: GenerationStage.ERROR,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n\n      const errorMessage = \"Generation failed\";\n      mockUseTamboCurrentMessage.mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({ title: \"\", body: \"\" }),\n          error: errorMessage,\n        }),\n      );\n\n      const { result } = renderHook(() =>\n        useTamboStreamStatus<{ title: string; body: string }>(),\n      );\n\n      expect(result.current.streamStatus.isPending).toBe(true);\n      expect(result.current.streamStatus.isStreaming).toBe(false);\n      expect(result.current.streamStatus.isSuccess).toBe(false);\n      expect(result.current.streamStatus.isError).toBe(true);\n      expect(result.current.streamStatus.streamError?.message).toBe(\n        errorMessage,\n      );\n    });\n  });\n\n  describe(\"Derivation Rules\", () => {\n    it(\"should derive isPending correctly (no generation activity AND all props pending)\", () => {\n      mockUseTamboGenerationStage.mockReturnValue({\n        generationStage: GenerationStage.IDLE,\n        generationStatusMessage: \"\",\n        isIdle: true,\n      });\n\n      mockUseTamboCurrentMessage.mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({ title: \"\", body: \"\", footer: \"\" }),\n        }),\n      );\n\n      const { result } = renderHook(() =>\n        useTamboStreamStatus<{\n          title: string;\n          body: string;\n          footer: string;\n        }>(),\n      );\n\n      // All props are pending and no generation activity\n      expect(result.current.streamStatus.isPending).toBe(true);\n      expect(result.current.propStatus.title.isPending).toBe(true);\n      expect(result.current.propStatus.body.isPending).toBe(true);\n      expect(result.current.propStatus.footer.isPending).toBe(true);\n    });\n\n    it(\"should derive isStreaming correctly (generation streaming OR any prop streaming)\", () => {\n      mockUseTamboGenerationStage.mockReturnValue({\n        generationStage: GenerationStage.COMPLETE,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n\n      // One prop still streaming\n      mockUseTamboCurrentMessage.mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({\n            title: \"Complete Title\",\n            body: \"Still streaming...\",\n          }),\n        }),\n      );\n\n      const { result } = renderHook(() =>\n        useTamboStreamStatus<{ title: string; body: string }>(),\n      );\n\n      // Should be streaming because at least one prop is streaming\n      expect(result.current.streamStatus.isStreaming).toBe(false); // Note: this will be false in our implementation because props are considered complete when generation is COMPLETE\n    });\n\n    it(\"should derive isSuccess correctly (generation complete AND all props successful)\", () => {\n      // Step 1: Start with streaming, props empty\n      mockUseTamboGenerationStage.mockReturnValue({\n        generationStage: GenerationStage.STREAMING_RESPONSE,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n\n      mockUseTamboCurrentMessage.mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({\n            title: \"\",\n            body: \"\",\n          }),\n        }),\n      );\n\n      const { result, rerender } = renderHook(() =>\n        useTamboStreamStatus<{ title: string; body: string }>(),\n      );\n\n      // Step 2: Simulate streaming in title\n      mockUseTamboCurrentMessage.mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({\n            title: \"Complete Title\",\n            body: \"\",\n          }),\n        }),\n      );\n      rerender();\n\n      // Step 3: Simulate streaming in body\n      mockUseTamboCurrentMessage.mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({\n            title: \"Complete Title\",\n            body: \"Complete Body\",\n          }),\n        }),\n      );\n      rerender();\n\n      // Step 4: Generation complete\n      mockUseTamboGenerationStage.mockReturnValue({\n        generationStage: GenerationStage.COMPLETE,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n      rerender();\n\n      // Now both props should be successful\n      expect(result.current.propStatus.title.isSuccess).toBe(true);\n      expect(result.current.propStatus.body.isSuccess).toBe(true);\n      expect(result.current.streamStatus.isSuccess).toBe(true);\n    });\n\n    it(\"should derive isError correctly (generation error OR any prop error)\", () => {\n      mockUseTamboGenerationStage.mockReturnValue({\n        generationStage: GenerationStage.COMPLETE,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n\n      mockUseTamboCurrentMessage.mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({\n            title: \"Good Title\",\n            body: \"Good Body\",\n          }),\n          error: \"Something went wrong\",\n        }),\n      );\n\n      const { result } = renderHook(() =>\n        useTamboStreamStatus<{ title: string; body: string }>(),\n      );\n\n      expect(result.current.streamStatus.isError).toBe(true);\n      expect(result.current.streamStatus.streamError?.message).toBe(\n        \"Something went wrong\",\n      );\n    });\n  });\n\n  describe(\"Type Safety\", () => {\n    it(\"should provide strongly typed prop status based on generic\", () => {\n      interface TestProps {\n        title: string;\n        description: string;\n        count: number;\n      }\n\n      mockUseTamboCurrentMessage.mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({\n            title: \"Test\",\n            description: \"Test desc\",\n            count: 42,\n          }),\n        }),\n      );\n\n      const { result } = renderHook(() => useTamboStreamStatus<TestProps>());\n\n      // TypeScript should infer these keys correctly\n      expect(result.current.propStatus.title).toBeDefined();\n      expect(result.current.propStatus.description).toBeDefined();\n      expect(result.current.propStatus.count).toBeDefined();\n\n      // This would cause a TypeScript error if uncommented:\n      // expect(result.current.propStatus.nonExistentProp).toBeDefined();\n    });\n\n    it(\"should work without generic type parameter\", () => {\n      mockUseTamboCurrentMessage.mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({ dynamicProp: \"value\" }),\n        }),\n      );\n\n      const { result } = renderHook(() => useTamboStreamStatus());\n\n      expect(result.current.streamStatus).toBeDefined();\n      expect(result.current.propStatus).toBeDefined();\n    });\n  });\n\n  describe(\"Edge Cases\", () => {\n    it(\"should handle missing message gracefully\", () => {\n      mockUseTamboCurrentMessage.mockReturnValue({\n        id: \"test-message\",\n        role: \"user\",\n        content: [],\n        componentState: {},\n        createdAt: \"\",\n        threadId: \"\",\n      } as TamboThreadMessage);\n\n      const { result } = renderHook(() => useTamboStreamStatus());\n\n      expect(result.current.streamStatus.isPending).toBe(true);\n      expect(result.current.propStatus).toEqual({});\n    });\n\n    it(\"should handle missing component props gracefully\", () => {\n      mockUseTamboCurrentMessage.mockReturnValue({\n        id: \"test-message\",\n        // Missing component property\n      } as TamboThreadMessage);\n\n      const { result } = renderHook(() => useTamboStreamStatus());\n\n      expect(result.current.streamStatus.isPending).toBe(true);\n      expect(result.current.propStatus).toEqual({});\n    });\n\n    it(\"should reset prop tracking when generation restarts\", () => {\n      // Step 1: Complete a message\n      mockUseTamboGenerationStage.mockReturnValue({\n        generationStage: GenerationStage.STREAMING_RESPONSE,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n      mockUseTamboCurrentMessage.mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({ title: \"\" }),\n        }),\n      );\n      const { result, rerender } = renderHook(() =>\n        useTamboStreamStatus<{ title: string }>(),\n      );\n      // Simulate streaming in title\n      mockUseTamboCurrentMessage.mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({ title: \"First Title\" }),\n        }),\n      );\n      rerender();\n      // Complete generation\n      mockUseTamboGenerationStage.mockReturnValue({\n        generationStage: GenerationStage.COMPLETE,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n      rerender();\n      // Should be successful initially\n      expect(result.current.propStatus.title.isSuccess).toBe(true);\n\n      // Step 2: Start new generation with a new message ID to trigger reset\n      mockUseTamboGenerationStage.mockReturnValue({\n        generationStage: GenerationStage.CHOOSING_COMPONENT,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n      mockUseTamboCurrentMessage.mockReturnValue(\n        createMockMessage({\n          id: \"new-message-id\", // Different message ID to trigger reset\n          component: createMockComponent({ title: \"\" }),\n        }),\n      );\n      rerender();\n      // Should reset to pending\n      expect(result.current.propStatus.title.isPending).toBe(true);\n      expect(result.current.propStatus.title.isSuccess).toBe(false);\n    });\n  });\n});\n"]}

package/dist/hooks/use-tambo-threads.test.d.ts.map

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

package/dist/hooks/{__tests__/use-tambo-threads.test.js → use-tambo-threads.test.js}

@@ -2,9 +2,9 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 const react_query_1 = require("@tanstack/react-query");
 const react_1 = require("@testing-library/react");
-const tambo_client_provider_1 = require("../../providers/tambo-client-provider");
-const use_tambo_threads_1 = require("../use-tambo-threads");
-jest.mock("../../providers/tambo-client-provider", () => ({
+const tambo_client_provider_1 = require("../providers/tambo-client-provider");
+const use_tambo_threads_1 = require("./use-tambo-threads");
+jest.mock("../providers/tambo-client-provider", () => ({
     useTamboClient: jest.fn(),
     useTamboQueryClient: jest.fn(),
 }));

package/dist/hooks/use-tambo-threads.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-tambo-threads.test.js","sourceRoot":"","sources":["../../src/hooks/use-tambo-threads.test.tsx"],"names":[],"mappings":";;AACA,uDAAoD;AACpD,kDAA6D;AAE7D,8EAG4C;AAE5C,2DAAyD;AAEzD,IAAI,CAAC,IAAI,CAAC,oCAAoC,EAAE,GAAG,EAAE,CAAC,CAAC;IACrD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;CAC/B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;IAClC,MAAM,WAAW,GAAG;QAClB,EAAE,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE;QACrC,EAAE,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE;KACtC,CAAC;IAEF,MAAM,YAAY,GAAG;QACnB,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;QACrB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;QACnB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;KAC6B,CAAC;IAEjD,MAAM,cAAc,GAAG;QACrB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;QACf,QAAQ,EAAE;YACR,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YACf,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;YACjB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;YACjB,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;SAChC;QACD,WAAW,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YACf,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;SACpB;QACD,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;QACnB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;QACjB,OAAO,EAAE,IAAI,CAAC,EAAE,EAAE;QAClB,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;KAC2B,CAAC;IAEpD,MAAM,QAAQ,GAAG;QACf,QAAQ,EAAE,YAAY;QACtB,OAAO,EAAE,cAAc;QACvB,QAAQ,EAAE;YACR,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;SACpB;KAC+B,CAAC;IAEnC,MAAM,WAAW,GAAG;QAClB,MAAM,EAAE,EAAE;QACV,IAAI,EAAE,QAAQ;KACgC,CAAC;IAEjD,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,MAAM,CAAC,2CAAmB,CAAC,CAAC,eAAe,CAAC,IAAI,yBAAW,EAAE,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wEAAwE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC;QAC/C,UAAU,CAAC,eAAe,CAAC;YACzB,GAAG,WAAW;YACd,IAAI,EAAE;gBACJ,GAAG,QAAQ;gBACX,QAAQ,EAAE;oBACR,GAAG,YAAY;oBACf,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,iBAAiB,EAAE,CAAC;iBACnE;gBACD,OAAO,EAAE;oBACP,GAAG,cAAc;oBACjB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,WAAW,CAAC;iBAC/C;aACF;SAC8B,CAAC,CAAC;QAEnC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,sCAAkB,GAAE,CAAC,CAAC;QAE1D,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;QAC1D,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC;QAC/C,UAAU,CAAC,eAAe,CAAC;YACzB,GAAG,WAAW;YACd,IAAI,EAAE;gBACJ,GAAG,QAAQ;gBACX,OAAO,EAAE;oBACP,GAAG,cAAc;oBACjB,IAAI,EAAE,QAAQ;iBACf;aACF;SAC8B,CAAC,CAAC;QAEnC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,sCAAkB,EAAC,EAAE,SAAS,EAAE,gBAAgB,EAAE,CAAC,CACpD,CAAC;QAEF,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,gBAAgB,EAAE,EAAE,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;QAC1D,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC;QAC/C,UAAU,CAAC,eAAe,CAAC;YACzB,GAAG,WAAW;YACd,IAAI,EAAE;gBACJ,GAAG,QAAQ;gBACX,QAAQ,EAAE;oBACR,GAAG,YAAY;oBACf,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,iBAAiB,EAAE,CAAC;iBACnE;gBACD,OAAO,EAAE;oBACP,GAAG,cAAc;oBACjB,IAAI,EAAE,QAAQ;iBACf;aACF;SAC8B,CAAC,CAAC;QAEnC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,sCAAkB,EAAC,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC,CACnD,CAAC;QAEF,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,iBAAiB,EAAE;YACvD,UAAU,EAAE,cAAc;SAC3B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QAC3C,IAAI,cAAoC,CAAC;QACzC,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACtC,cAAc,GAAG,OAAO,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC;QAC/C,UAAU,CAAC,eAAe,CAAC;YACzB,GAAG,WAAW;YACd,IAAI,EAAE;gBACJ,GAAG,QAAQ;gBACX,QAAQ,EAAE;oBACR,GAAG,YAAY;oBACf,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,iBAAiB,EAAE,CAAC;iBACnE;gBACD,OAAO,EAAE;oBACP,GAAG,cAAc;oBACjB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,OAAO,CAAC;iBACzC;aACF;SAC8B,CAAC,CAAC;QAEnC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,sCAAkB,EAAC,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CACzC,CAAC;QAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KA6B5C,CAAC,CAAC;QAEH,cAAe,CAAC,WAAW,CAAC,CAAC;QAC7B,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QACvD,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC;QAC/C,UAAU,CAAC,eAAe,CAAC;YACzB,GAAG,WAAW;YACd,IAAI,EAAE;gBACJ,GAAG,QAAQ;gBACX,QAAQ,EAAE;oBACR,GAAG,YAAY;oBACf,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,iBAAiB,EAAE,CAAC;iBACnE;gBACD,OAAO,EAAE;oBACP,GAAG,cAAc;oBACjB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;wBAC5C,6CAA6C;wBAC7C,MAAM,SAAS,CAAC;oBAClB,CAAC,CAAC;iBACH;aACF;SAC8B,CAAC,CAAC;QAEnC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CACjC,IAAA,sCAAkB,EAAC,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CACzC,CAAC;QAEF,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,MAAM,CAAC,OAAO,CAAC;YACrD,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC9B,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC3B,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { QueryClient } from \"@tanstack/react-query\";\nimport { renderHook, waitFor } from \"@testing-library/react\";\nimport { DeepPartial } from \"ts-essentials\";\nimport {\n  useTamboClient,\n  useTamboQueryClient,\n} from \"../providers/tambo-client-provider\";\nimport { PartialTamboAI } from \"../testing/types\";\nimport { useTamboThreadList } from \"./use-tambo-threads\";\n\njest.mock(\"../providers/tambo-client-provider\", () => ({\n  useTamboClient: jest.fn(),\n  useTamboQueryClient: jest.fn(),\n}));\n\ndescribe(\"useTamboThreadList\", () => {\n  const mockThreads = [\n    { id: \"thread-1\", title: \"Thread 1\" },\n    { id: \"thread-2\", title: \"Thread 2\" },\n  ];\n\n  const mockProjects = {\n    getCurrent: jest.fn(),\n    retrieve: jest.fn(),\n    delete: jest.fn(),\n  } satisfies Partial<TamboAI[\"beta\"][\"projects\"]>;\n\n  const mockThreadsApi = {\n    list: jest.fn(),\n    messages: {\n      list: jest.fn(),\n      create: jest.fn(),\n      delete: jest.fn(),\n      updateComponentState: jest.fn(),\n    },\n    suggestions: {\n      list: jest.fn(),\n      generate: jest.fn(),\n    },\n    create: jest.fn(),\n    retrieve: jest.fn(),\n    update: jest.fn(),\n    delete: jest.fn(),\n    advance: jest.fn(),\n    advanceByID: jest.fn(),\n  } satisfies DeepPartial<TamboAI[\"beta\"][\"threads\"]>;\n\n  const mockBeta = {\n    projects: mockProjects,\n    threads: mockThreadsApi,\n    registry: {\n      retrieve: jest.fn(),\n    },\n  } satisfies PartialTamboAI[\"beta\"];\n\n  const mockTamboAI = {\n    apiKey: \"\",\n    beta: mockBeta,\n  } satisfies PartialTamboAI as unknown as TamboAI;\n\n  beforeEach(() => {\n    jest.mocked(useTamboQueryClient).mockReturnValue(new QueryClient());\n  });\n\n  it(\"should fetch threads for current project when no projectId is provided\", async () => {\n    const mockClient = jest.mocked(useTamboClient);\n    mockClient.mockReturnValue({\n      ...mockTamboAI,\n      beta: {\n        ...mockBeta,\n        projects: {\n          ...mockProjects,\n          getCurrent: jest.fn().mockResolvedValue({ id: \"current-project\" }),\n        },\n        threads: {\n          ...mockThreadsApi,\n          list: jest.fn().mockResolvedValue(mockThreads),\n        },\n      },\n    } satisfies PartialTamboAI as any);\n\n    const { result } = renderHook(() => useTamboThreadList());\n\n    await waitFor(() => {\n      expect(result.current.data).toEqual(mockThreads);\n    });\n  });\n\n  it(\"should fetch threads for specified projectId\", async () => {\n    const mockList = jest.fn().mockResolvedValue(mockThreads);\n    const mockClient = jest.mocked(useTamboClient);\n    mockClient.mockReturnValue({\n      ...mockTamboAI,\n      beta: {\n        ...mockBeta,\n        threads: {\n          ...mockThreadsApi,\n          list: mockList,\n        },\n      },\n    } satisfies PartialTamboAI as any);\n\n    const { result } = renderHook(() =>\n      useTamboThreadList({ projectId: \"custom-project\" }),\n    );\n\n    await waitFor(() => {\n      expect(result.current.data).toEqual(mockThreads);\n    });\n\n    expect(mockList).toHaveBeenCalledWith(\"custom-project\", {});\n  });\n\n  it(\"should fetch threads with contextKey when provided\", async () => {\n    const mockList = jest.fn().mockResolvedValue(mockThreads);\n    const mockClient = jest.mocked(useTamboClient);\n    mockClient.mockReturnValue({\n      ...mockTamboAI,\n      beta: {\n        ...mockBeta,\n        projects: {\n          ...mockProjects,\n          getCurrent: jest.fn().mockResolvedValue({ id: \"current-project\" }),\n        },\n        threads: {\n          ...mockThreadsApi,\n          list: mockList,\n        },\n      },\n    } satisfies PartialTamboAI as any);\n\n    const { result } = renderHook(() =>\n      useTamboThreadList({ contextKey: \"test-context\" }),\n    );\n\n    await waitFor(() => {\n      expect(result.current.data).toEqual(mockThreads);\n    });\n\n    expect(mockList).toHaveBeenCalledWith(\"current-project\", {\n      contextKey: \"test-context\",\n    });\n  });\n\n  it(\"should handle loading state\", async () => {\n    let resolvePromise: (value: any) => void;\n    const promise = new Promise((resolve) => {\n      resolvePromise = resolve;\n    });\n\n    const mockClient = jest.mocked(useTamboClient);\n    mockClient.mockReturnValue({\n      ...mockTamboAI,\n      beta: {\n        ...mockBeta,\n        projects: {\n          ...mockProjects,\n          getCurrent: jest.fn().mockResolvedValue({ id: \"current-project\" }),\n        },\n        threads: {\n          ...mockThreadsApi,\n          list: jest.fn().mockReturnValue(promise),\n        },\n      },\n    } satisfies PartialTamboAI as any);\n\n    const { result } = renderHook(() =>\n      useTamboThreadList({}, { retry: false }),\n    );\n\n    expect(result.current).toMatchInlineSnapshot(`\n      {\n        \"data\": null,\n        \"dataUpdatedAt\": 0,\n        \"error\": null,\n        \"errorUpdateCount\": 0,\n        \"errorUpdatedAt\": 0,\n        \"failureCount\": 0,\n        \"failureReason\": null,\n        \"fetchStatus\": \"fetching\",\n        \"isEnabled\": true,\n        \"isError\": false,\n        \"isFetched\": false,\n        \"isFetchedAfterMount\": false,\n        \"isFetching\": true,\n        \"isInitialLoading\": true,\n        \"isLoading\": true,\n        \"isLoadingError\": false,\n        \"isPaused\": false,\n        \"isPending\": true,\n        \"isPlaceholderData\": false,\n        \"isRefetchError\": false,\n        \"isRefetching\": false,\n        \"isStale\": true,\n        \"isSuccess\": false,\n        \"promise\": Promise {},\n        \"refetch\": [Function],\n        \"status\": \"pending\",\n      }\n    `);\n\n    resolvePromise!(mockThreads);\n    await waitFor(() => {\n      expect(result.current.isLoading).toBe(false);\n    });\n  });\n\n  it(\"should handle error state\", async () => {\n    const mockError = new Error(\"Failed to fetch threads\");\n    const mockClient = jest.mocked(useTamboClient);\n    mockClient.mockReturnValue({\n      ...mockTamboAI,\n      beta: {\n        ...mockBeta,\n        projects: {\n          ...mockProjects,\n          getCurrent: jest.fn().mockResolvedValue({ id: \"current-project\" }),\n        },\n        threads: {\n          ...mockThreadsApi,\n          list: jest.fn().mockImplementation(async () => {\n            // console.log(\"Mocking error\", mockCount++);\n            throw mockError;\n          }),\n        },\n      },\n    } satisfies PartialTamboAI as any);\n\n    const { result } = renderHook(() =>\n      useTamboThreadList({}, { retry: false }),\n    );\n\n    await waitFor(() => {\n      const { isLoading, error, isError } = result.current;\n      expect(isLoading).toBe(false);\n      expect(isError).toBe(true);\n      expect(error).toBe(mockError);\n    });\n  });\n});\n"]}

package/dist/mcp/elicitation.d.ts

@@ -1,14 +1,22 @@
-import type { ElicitRequest } from "@modelcontextprotocol/sdk/types.js";
+import type { ElicitRequestFormParams, PrimitiveSchemaDefinition } from "@modelcontextprotocol/sdk/spec.types.js";
 import { MCPElicitationHandler } from "./mcp-client";
+/**
+ * Schema type for elicitation request fields
+ */
+export type ElicitationRequestedSchema = ElicitRequestFormParams["requestedSchema"];
 /**
  * Elicitation request from MCP server
  */
 export interface TamboElicitationRequest {
     message: string;
-    requestedSchema: ElicitRequest["params"]["requestedSchema"];
+    requestedSchema: ElicitationRequestedSchema;
     /** AbortSignal that fires when the server cancels the request (e.g., timeout) */
     signal?: AbortSignal;
 }
+/**
+ * Re-export PrimitiveSchemaDefinition for consumers that need to work with schema fields
+ */
+export type { PrimitiveSchemaDefinition };
 /**
  * Elicitation response to be sent back
  */

package/dist/mcp/elicitation.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"elicitation.d.ts","sourceRoot":"","sources":["../../src/mcp/elicitation.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAGV,aAAa,EAEd,MAAM,oCAAoC,CAAC;AAE5C,OAAO,EAAE,qBAAqB,EAAE,MAAM,cAAc,CAAC;AAErD;;GAEG;AACH,MAAM,WAAW,uBAAuB;IACtC,OAAO,EAAE,MAAM,CAAC;IAChB,eAAe,EAAE,aAAa,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,CAAC;IAC5D,iFAAiF;IACjF,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAED;;GAEG;AACH,MAAM,WAAW,wBAAwB;IACvC,MAAM,EAAE,QAAQ,GAAG,SAAS,GAAG,QAAQ,CAAC;IACxC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClC,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;CACtB;AAED;;;GAGG;AACH,MAAM,WAAW,uBAAuB;IACtC,0DAA0D;IAC1D,WAAW,EAAE,uBAAuB,GAAG,IAAI,CAAC;IAC5C,sFAAsF;IACtF,kBAAkB,EAAE,CAAC,CAAC,QAAQ,EAAE,wBAAwB,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC;CAC3E;AAED;;;;;;GAMG;AACH,wBAAgB,cAAc;;;oCAId,wBAAwB,KAAK,IAAI;+FAAjC,wBAAwB,KAAK,IAAI;;EAqDhD"}
+{"version":3,"file":"elicitation.d.ts","sourceRoot":"","sources":["../../src/mcp/elicitation.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EACV,uBAAuB,EACvB,yBAAyB,EAC1B,MAAM,yCAAyC,CAAC;AAQjD,OAAO,EAAE,qBAAqB,EAAE,MAAM,cAAc,CAAC;AAErD;;GAEG;AACH,MAAM,MAAM,0BAA0B,GACpC,uBAAuB,CAAC,iBAAiB,CAAC,CAAC;AAE7C;;GAEG;AACH,MAAM,WAAW,uBAAuB;IACtC,OAAO,EAAE,MAAM,CAAC;IAChB,eAAe,EAAE,0BAA0B,CAAC;IAC5C,iFAAiF;IACjF,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAED;;GAEG;AACH,YAAY,EAAE,yBAAyB,EAAE,CAAC;AAE1C;;GAEG;AACH,MAAM,WAAW,wBAAwB;IACvC,MAAM,EAAE,QAAQ,GAAG,SAAS,GAAG,QAAQ,CAAC;IACxC,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAClC,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;CACtB;AAED;;;GAGG;AACH,MAAM,WAAW,uBAAuB;IACtC,0DAA0D;IAC1D,WAAW,EAAE,uBAAuB,GAAG,IAAI,CAAC;IAC5C,sFAAsF;IACtF,kBAAkB,EAAE,CAAC,CAAC,QAAQ,EAAE,wBAAwB,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC;CAC3E;AAoBD;;;;;;GAMG;AACH,wBAAgB,cAAc;;;oCAId,wBAAwB,KAAK,IAAI;+FAAjC,wBAAwB,KAAK,IAAI;;EA0DhD"}