@tambo-ai/react 0.43.1 → 0.44.1

package/esm/hooks/use-thread-input.js.map

@@ -1 +1 @@
-{"version":3,"file":"use-thread-input.js","sourceRoot":"","sources":["../../src/hooks/use-thread-input.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AACpC,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AACpE,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAEvD;;;;GAIG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG;IAClC,kDAAkD;IAClD,KAAK,EAAE,yBAAyB;IAChC,0CAA0C;IAC1C,OAAO,EAAE,6CAA6C;IACtD,qDAAqD;IACrD,MAAM,EAAE,gCAAgC;IACxC,yCAAyC;IACzC,UAAU,EAAE,wBAAwB;CAC5B,CAAC;AAkCX;;;GAGG;AACH,MAAM,UAAU,mBAAmB,CAAC,UAAmB;IACrD,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,aAAa,EAAE,iBAAiB,EAAE,GAC5D,cAAc,EAAE,CAAC;IAEnB,MAAM,MAAM,GAAG,WAAW,CACxB,KAAK,EAAE,EACL,UAAU,EAAE,gBAAgB,EAC5B,cAAc,EACd,eAAe,EACf,iBAAiB,MAMf,EAAE,EAAE,EAAE;QACR,MAAM,UAAU,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;QAC7C,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACxB,MAAM,IAAI,gBAAgB,CACxB,0BAA0B,UAAU,CAAC,KAAK,IAAI,oBAAoB,CAAC,UAAU,EAAE,EAC/E,EAAE,KAAK,EAAE,UAAU,CAAC,KAAK,EAAE,CAC5B,CAAC;QACJ,CAAC;QAED,MAAM,iBAAiB,CAAC,UAAU,CAAC,cAAc,EAAE;YACjD,QAAQ,EAAE,MAAM,CAAC,EAAE;YACnB,UAAU,EAAE,gBAAgB,IAAI,UAAU,IAAI,SAAS;YACvD,cAAc,EAAE,cAAc;YAC9B,eAAe,EAAE,eAAe;YAChC,iBAAiB,EAAE,iBAAiB;SACrC,CAAC,CAAC;QACH,aAAa,CAAC,EAAE,CAAC,CAAC;IACpB,CAAC,EACD,CAAC,UAAU,EAAE,iBAAiB,EAAE,MAAM,CAAC,EAAE,EAAE,UAAU,EAAE,aAAa,CAAC,CACtE,CAAC;IACF,MAAM,EACJ,WAAW,EAAE,WAAW,EACxB,MAAM,EAAE,aAAa,EACrB,GAAG,aAAa,EACjB,GAAG,gBAAgB,CAAC;QACnB,UAAU,EAAE,MAAM;KACnB,CAAC,CAAC;IAEH,OAAO;QACL,GAAG,aAAa;QAChB,KAAK,EAAE,UAAU;QACjB,QAAQ,EAAE,aAAa;QACvB,MAAM,EAAE,WAAW;KACF,CAAC;AACtB,CAAC","sourcesContent":["import { UseMutationResult } from \"@tanstack/react-query\";\nimport { useCallback } from \"react\";\nimport { ThreadInputError } from \"../model/thread-input-error\";\nimport { validateInput } from \"../model/validate-input\";\nimport { useTamboThread } from \"../providers/tambo-thread-provider\";\nimport { useTamboMutation } from \"./react-query-hooks\";\n\n/**\n * Error messages for various input-related error scenarios\n * These messages are used to provide user-friendly error feedback\n * @readonly\n */\nexport const INPUT_ERROR_MESSAGES = {\n  /** Error when attempting to submit empty input */\n  EMPTY: \"Message cannot be empty\",\n  /** Error when network connection fails */\n  NETWORK: \"Network error. Please check your connection\",\n  /** Error when server fails to process the request */\n  SERVER: \"Server error. Please try again\",\n  /** Error when input format is invalid */\n  VALIDATION: \"Invalid message format\",\n} as const;\n\n/**\n * Interface for the thread input hook return value\n * Provides all necessary functions and state for managing thread input\n */\ninterface UseThreadInputInternal {\n  /** Current value of the input field */\n  value: string;\n  /**\n   * Function to update the input value\n   * @param value - New value for the input field\n   */\n  setValue: (value: string) => void;\n  /**\n   * Function to submit the current input value\n   * Validates input, handles errors, and cleans up state after submission\n   * @throws {ThreadInputError} If submission fails\n   * @returns Promise that resolves when submission is complete\n   */\n  submit: (options?: {\n    contextKey?: string;\n    streamResponse?: boolean;\n    forceToolChoice?: string;\n    additionalContext?: Record<string, any>;\n  }) => Promise<void>;\n}\nexport type UseThreadInput = UseThreadInputInternal &\n  UseMutationResult<\n    void,\n    Error,\n    { contextKey?: string; streamResponse?: boolean; forceToolChoice?: string }\n  >;\n\n/**\n * Hook for managing thread message input state and submission\n * @returns Interface for managing thread input state and submission\n */\nexport function useTamboThreadInput(contextKey?: string): UseThreadInput {\n  const { thread, inputValue, setInputValue, sendThreadMessage } =\n    useTamboThread();\n\n  const submit = useCallback(\n    async ({\n      contextKey: submitContextKey,\n      streamResponse,\n      forceToolChoice,\n      additionalContext,\n    }: {\n      contextKey?: string;\n      streamResponse?: boolean;\n      forceToolChoice?: string;\n      additionalContext?: Record<string, any>;\n    } = {}) => {\n      const validation = validateInput(inputValue);\n      if (!validation.isValid) {\n        throw new ThreadInputError(\n          `Cannot submit message: ${validation.error ?? INPUT_ERROR_MESSAGES.VALIDATION}`,\n          { cause: validation.error },\n        );\n      }\n\n      await sendThreadMessage(validation.sanitizedInput, {\n        threadId: thread.id,\n        contextKey: submitContextKey ?? contextKey ?? undefined,\n        streamResponse: streamResponse,\n        forceToolChoice: forceToolChoice,\n        additionalContext: additionalContext,\n      });\n      setInputValue(\"\");\n    },\n    [inputValue, sendThreadMessage, thread.id, contextKey, setInputValue],\n  );\n  const {\n    mutateAsync: submitAsync,\n    mutate: _unusedSubmit,\n    ...mutationState\n  } = useTamboMutation({\n    mutationFn: submit,\n  });\n\n  return {\n    ...mutationState,\n    value: inputValue,\n    setValue: setInputValue,\n    submit: submitAsync,\n  } as UseThreadInput;\n}\n"]}
+{"version":3,"file":"use-thread-input.js","sourceRoot":"","sources":["../../src/hooks/use-thread-input.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAC9C,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,oCAAoC,CAAC;AACpE,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAEvD;;;;GAIG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG;IAClC,kDAAkD;IAClD,KAAK,EAAE,yBAAyB;IAChC,0CAA0C;IAC1C,OAAO,EAAE,6CAA6C;IACtD,qDAAqD;IACrD,MAAM,EAAE,gCAAgC;IACxC,yCAAyC;IACzC,UAAU,EAAE,wBAAwB;CAC5B,CAAC;AAkCX;;;GAGG;AACH,MAAM,UAAU,mBAAmB,CAAC,UAAmB;IACrD,MAAM,EAAE,MAAM,EAAE,iBAAiB,EAAE,GAAG,cAAc,EAAE,CAAC;IACvD,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;IAEjD,MAAM,MAAM,GAAG,WAAW,CACxB,KAAK,EAAE,EACL,UAAU,EAAE,gBAAgB,EAC5B,cAAc,EACd,eAAe,EACf,iBAAiB,MAMf,EAAE,EAAE,EAAE;QACR,MAAM,UAAU,GAAG,aAAa,CAAC,UAAU,CAAC,CAAC;QAC7C,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC;YACxB,MAAM,IAAI,gBAAgB,CACxB,0BAA0B,UAAU,CAAC,KAAK,IAAI,oBAAoB,CAAC,UAAU,EAAE,EAC/E,EAAE,KAAK,EAAE,UAAU,CAAC,KAAK,EAAE,CAC5B,CAAC;QACJ,CAAC;QAED,MAAM,iBAAiB,CAAC,UAAU,CAAC,cAAc,EAAE;YACjD,QAAQ,EAAE,MAAM,CAAC,EAAE;YACnB,UAAU,EAAE,gBAAgB,IAAI,UAAU,IAAI,SAAS;YACvD,cAAc,EAAE,cAAc;YAC9B,eAAe,EAAE,eAAe;YAChC,iBAAiB,EAAE,iBAAiB;SACrC,CAAC,CAAC;QACH,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC,oBAAoB;IACzC,CAAC,EACD,CAAC,UAAU,EAAE,iBAAiB,EAAE,MAAM,CAAC,EAAE,EAAE,UAAU,EAAE,aAAa,CAAC,CACtE,CAAC;IACF,MAAM,EACJ,WAAW,EAAE,WAAW,EACxB,MAAM,EAAE,aAAa,EACrB,GAAG,aAAa,EACjB,GAAG,gBAAgB,CAAC;QACnB,UAAU,EAAE,MAAM;KACnB,CAAC,CAAC;IAEH,OAAO;QACL,GAAG,aAAa;QAChB,KAAK,EAAE,UAAU;QACjB,QAAQ,EAAE,aAAa;QACvB,MAAM,EAAE,WAAW;KACF,CAAC;AACtB,CAAC","sourcesContent":["import { UseMutationResult } from \"@tanstack/react-query\";\nimport { useCallback, useState } from \"react\";\nimport { ThreadInputError } from \"../model/thread-input-error\";\nimport { validateInput } from \"../model/validate-input\";\nimport { useTamboThread } from \"../providers/tambo-thread-provider\";\nimport { useTamboMutation } from \"./react-query-hooks\";\n\n/**\n * Error messages for various input-related error scenarios\n * These messages are used to provide user-friendly error feedback\n * @readonly\n */\nexport const INPUT_ERROR_MESSAGES = {\n  /** Error when attempting to submit empty input */\n  EMPTY: \"Message cannot be empty\",\n  /** Error when network connection fails */\n  NETWORK: \"Network error. Please check your connection\",\n  /** Error when server fails to process the request */\n  SERVER: \"Server error. Please try again\",\n  /** Error when input format is invalid */\n  VALIDATION: \"Invalid message format\",\n} as const;\n\n/**\n * Interface for the thread input hook return value\n * Provides all necessary functions and state for managing thread input\n */\ninterface UseThreadInputInternal {\n  /** Current value of the input field */\n  value: string;\n  /**\n   * Function to update the input value\n   * @param value - New value for the input field\n   */\n  setValue: (value: string) => void;\n  /**\n   * Function to submit the current input value\n   * Validates input, handles errors, and cleans up state after submission\n   * @throws {ThreadInputError} If submission fails\n   * @returns Promise that resolves when submission is complete\n   */\n  submit: (options?: {\n    contextKey?: string;\n    streamResponse?: boolean;\n    forceToolChoice?: string;\n    additionalContext?: Record<string, any>;\n  }) => Promise<void>;\n}\nexport type UseThreadInput = UseThreadInputInternal &\n  UseMutationResult<\n    void,\n    Error,\n    { contextKey?: string; streamResponse?: boolean; forceToolChoice?: string }\n  >;\n\n/**\n * Hook for managing thread message input state and submission\n * @returns Interface for managing thread input state and submission\n */\nexport function useTamboThreadInput(contextKey?: string): UseThreadInput {\n  const { thread, sendThreadMessage } = useTamboThread();\n  const [inputValue, setInputValue] = useState(\"\");\n\n  const submit = useCallback(\n    async ({\n      contextKey: submitContextKey,\n      streamResponse,\n      forceToolChoice,\n      additionalContext,\n    }: {\n      contextKey?: string;\n      streamResponse?: boolean;\n      forceToolChoice?: string;\n      additionalContext?: Record<string, any>;\n    } = {}) => {\n      const validation = validateInput(inputValue);\n      if (!validation.isValid) {\n        throw new ThreadInputError(\n          `Cannot submit message: ${validation.error ?? INPUT_ERROR_MESSAGES.VALIDATION}`,\n          { cause: validation.error },\n        );\n      }\n\n      await sendThreadMessage(validation.sanitizedInput, {\n        threadId: thread.id,\n        contextKey: submitContextKey ?? contextKey ?? undefined,\n        streamResponse: streamResponse,\n        forceToolChoice: forceToolChoice,\n        additionalContext: additionalContext,\n      });\n      setInputValue(\"\"); // Clear local state\n    },\n    [inputValue, sendThreadMessage, thread.id, contextKey, setInputValue],\n  );\n  const {\n    mutateAsync: submitAsync,\n    mutate: _unusedSubmit,\n    ...mutationState\n  } = useTamboMutation({\n    mutationFn: submit,\n  });\n\n  return {\n    ...mutationState,\n    value: inputValue,\n    setValue: setInputValue,\n    submit: submitAsync,\n  } as UseThreadInput;\n}\n"]}

package/esm/index.d.ts

@@ -1,11 +1,11 @@
 /** Exports for the library. Only publically available exports are re-exported here. Anything not exported here is not supported and may change or break at any time. */
 export { useTamboComponentState } from "./hooks/use-component-state";
-export { TamboMessageProvider, useTamboCurrentMessage, useTamboMessageContext, } from "./hooks/use-current-message";
+export { TamboMessageProvider, useTamboCurrentMessage, } from "./hooks/use-current-message";
 export { useTamboStreamingProps } from "./hooks/use-streaming-props";
 export * from "./hooks/use-suggestions";
 export { useTamboStreamStatus, type PropStatus, type StreamStatus, } from "./hooks/use-tambo-stream-status";
 export { useTamboThreadInput } from "./hooks/use-thread-input";
-export { TamboClientProvider, TamboComponentProvider, TamboContextHelpersProvider, TamboPropStreamProvider, TamboProvider, TamboStubProvider, TamboThreadProvider, useTambo, useTamboClient, useTamboContextHelpers, useTamboStream, useTamboThread, type TamboComponent, type TamboContextHelpersContextProps, type TamboContextHelpersProviderProps, type TamboRegistryContext, type TamboStubProviderProps, } from "./providers";
+export { TamboClientProvider, TamboComponentProvider, TamboContextHelpersProvider, TamboPropStreamProvider, TamboProvider, TamboStubProvider, TamboThreadProvider, useTambo, useTamboClient, useTamboContextHelpers, useTamboGenerationStage, useTamboStream, useTamboThread, type TamboComponent, type TamboContextHelpersContextProps, type TamboContextHelpersProviderProps, type TamboRegistryContext, type TamboStubProviderProps, } from "./providers";
 export type { APIError, RateLimitError, TamboAIError, } from "@tambo-ai/typescript-sdk";
 export type { Suggestion, SuggestionGenerateParams, SuggestionGenerateResponse, SuggestionListResponse, } from "@tambo-ai/typescript-sdk/resources/beta/threads/suggestions";
 export { useTamboThreadList } from "./hooks/use-tambo-threads";

package/esm/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,wKAAwK;AAExK,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,OAAO,EACL,oBAAoB,EACpB,sBAAsB,EACtB,sBAAsB,GACvB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,cAAc,yBAAyB,CAAC;AACxC,OAAO,EACL,oBAAoB,EACpB,KAAK,UAAU,EACf,KAAK,YAAY,GAClB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAG/D,OAAO,EACL,mBAAmB,EACnB,sBAAsB,EACtB,2BAA2B,EAC3B,uBAAuB,EACvB,aAAa,EACb,iBAAiB,EACjB,mBAAmB,EACnB,QAAQ,EACR,cAAc,EACd,sBAAsB,EACtB,cAAc,EACd,cAAc,EACd,KAAK,cAAc,EACnB,KAAK,+BAA+B,EACpC,KAAK,gCAAgC,EACrC,KAAK,oBAAoB,EACzB,KAAK,sBAAsB,GAC5B,MAAM,aAAa,CAAC;AAGrB,YAAY,EACV,QAAQ,EACR,cAAc,EACd,YAAY,GACb,MAAM,0BAA0B,CAAC;AAClC,YAAY,EACV,UAAU,EACV,wBAAwB,EACxB,0BAA0B,EAC1B,sBAAsB,GACvB,MAAM,6DAA6D,CAAC;AACrE,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EACL,KAAK,4BAA4B,EACjC,KAAK,iBAAiB,EACtB,KAAK,aAAa,EAClB,KAAK,mBAAmB,EACxB,KAAK,SAAS,GACf,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,eAAe,EACf,KAAK,kBAAkB,GACxB,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EAAE,KAAK,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAExD,YAAY,EACV,0BAA0B,IAAI,qBAAqB,EACnD,wBAAwB,GACzB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,qBAAqB,IAAI,gBAAgB,EACzC,KAAK,kBAAkB,EACvB,KAAK,0BAA0B,GAChC,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,oBAAoB,EAAE,MAAM,yCAAyC,CAAC;AAG/E,OAAO,EACL,wBAAwB,EACxB,wBAAwB,GACzB,MAAM,mBAAmB,CAAC;AAC3B,YAAY,EACV,iBAAiB,EACjB,eAAe,EACf,cAAc,GACf,MAAM,mBAAmB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,wKAAwK;AAExK,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,OAAO,EACL,oBAAoB,EACpB,sBAAsB,GACvB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,cAAc,yBAAyB,CAAC;AACxC,OAAO,EACL,oBAAoB,EACpB,KAAK,UAAU,EACf,KAAK,YAAY,GAClB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAG/D,OAAO,EACL,mBAAmB,EACnB,sBAAsB,EACtB,2BAA2B,EAC3B,uBAAuB,EACvB,aAAa,EACb,iBAAiB,EACjB,mBAAmB,EACnB,QAAQ,EACR,cAAc,EACd,sBAAsB,EACtB,uBAAuB,EACvB,cAAc,EACd,cAAc,EACd,KAAK,cAAc,EACnB,KAAK,+BAA+B,EACpC,KAAK,gCAAgC,EACrC,KAAK,oBAAoB,EACzB,KAAK,sBAAsB,GAC5B,MAAM,aAAa,CAAC;AAGrB,YAAY,EACV,QAAQ,EACR,cAAc,EACd,YAAY,GACb,MAAM,0BAA0B,CAAC;AAClC,YAAY,EACV,UAAU,EACV,wBAAwB,EACxB,0BAA0B,EAC1B,sBAAsB,GACvB,MAAM,6DAA6D,CAAC;AACrE,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAC/D,OAAO,EACL,KAAK,4BAA4B,EACjC,KAAK,iBAAiB,EACtB,KAAK,aAAa,EAClB,KAAK,mBAAmB,EACxB,KAAK,SAAS,GACf,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,eAAe,EACf,KAAK,kBAAkB,GACxB,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EAAE,KAAK,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAExD,YAAY,EACV,0BAA0B,IAAI,qBAAqB,EACnD,wBAAwB,GACzB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,qBAAqB,IAAI,gBAAgB,EACzC,KAAK,kBAAkB,EACvB,KAAK,0BAA0B,GAChC,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,oBAAoB,EAAE,MAAM,yCAAyC,CAAC;AAG/E,OAAO,EACL,wBAAwB,EACxB,wBAAwB,GACzB,MAAM,mBAAmB,CAAC;AAC3B,YAAY,EACV,iBAAiB,EACjB,eAAe,EACf,cAAc,GACf,MAAM,mBAAmB,CAAC"}

package/esm/index.js

@@ -1,12 +1,12 @@
 /** Exports for the library. Only publically available exports are re-exported here. Anything not exported here is not supported and may change or break at any time. */
 export { useTamboComponentState } from "./hooks/use-component-state";
-export { TamboMessageProvider, useTamboCurrentMessage, useTamboMessageContext, } from "./hooks/use-current-message";
+export { TamboMessageProvider, useTamboCurrentMessage, } from "./hooks/use-current-message";
 export { useTamboStreamingProps } from "./hooks/use-streaming-props";
 export * from "./hooks/use-suggestions";
 export { useTamboStreamStatus, } from "./hooks/use-tambo-stream-status";
 export { useTamboThreadInput } from "./hooks/use-thread-input";
 // Re-export provider components
-export { TamboClientProvider, TamboComponentProvider, TamboContextHelpersProvider, TamboPropStreamProvider, TamboProvider, TamboStubProvider, TamboThreadProvider, useTambo, useTamboClient, useTamboContextHelpers, useTamboStream, useTamboThread, } from "./providers";
+export { TamboClientProvider, TamboComponentProvider, TamboContextHelpersProvider, TamboPropStreamProvider, TamboProvider, TamboStubProvider, TamboThreadProvider, useTambo, useTamboClient, useTamboContextHelpers, useTamboGenerationStage, useTamboStream, useTamboThread, } from "./providers";
 export { useTamboThreadList } from "./hooks/use-tambo-threads";
 export { GenerationStage, } from "./model/generate-component-response";
 export { withTamboInteractable as withInteractable, } from "./providers/hoc/with-tambo-interactable";

package/esm/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,wKAAwK;AAExK,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,OAAO,EACL,oBAAoB,EACpB,sBAAsB,EACtB,sBAAsB,GACvB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,cAAc,yBAAyB,CAAC;AACxC,OAAO,EACL,oBAAoB,GAGrB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAE/D,gCAAgC;AAChC,OAAO,EACL,mBAAmB,EACnB,sBAAsB,EACtB,2BAA2B,EAC3B,uBAAuB,EACvB,aAAa,EACb,iBAAiB,EACjB,mBAAmB,EACnB,QAAQ,EACR,cAAc,EACd,sBAAsB,EACtB,cAAc,EACd,cAAc,GAMf,MAAM,aAAa,CAAC;AAcrB,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAQ/D,OAAO,EACL,eAAe,GAEhB,MAAM,qCAAqC,CAAC;AAO7C,OAAO,EACL,qBAAqB,IAAI,gBAAgB,GAG1C,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,oBAAoB,EAAE,MAAM,yCAAyC,CAAC;AAE/E,0BAA0B;AAC1B,OAAO,EACL,wBAAwB,EACxB,wBAAwB,GACzB,MAAM,mBAAmB,CAAC","sourcesContent":["/** Exports for the library. Only publically available exports are re-exported here. Anything not exported here is not supported and may change or break at any time. */\n\nexport { useTamboComponentState } from \"./hooks/use-component-state\";\nexport {\n  TamboMessageProvider,\n  useTamboCurrentMessage,\n  useTamboMessageContext,\n} from \"./hooks/use-current-message\";\nexport { useTamboStreamingProps } from \"./hooks/use-streaming-props\";\nexport * from \"./hooks/use-suggestions\";\nexport {\n  useTamboStreamStatus,\n  type PropStatus,\n  type StreamStatus,\n} from \"./hooks/use-tambo-stream-status\";\nexport { useTamboThreadInput } from \"./hooks/use-thread-input\";\n\n// Re-export provider components\nexport {\n  TamboClientProvider,\n  TamboComponentProvider,\n  TamboContextHelpersProvider,\n  TamboPropStreamProvider,\n  TamboProvider,\n  TamboStubProvider,\n  TamboThreadProvider,\n  useTambo,\n  useTamboClient,\n  useTamboContextHelpers,\n  useTamboStream,\n  useTamboThread,\n  type TamboComponent,\n  type TamboContextHelpersContextProps,\n  type TamboContextHelpersProviderProps,\n  type TamboRegistryContext,\n  type TamboStubProviderProps,\n} from \"./providers\";\n\n// Re-export types from Tambo Node SDK\nexport type {\n  APIError,\n  RateLimitError,\n  TamboAIError,\n} from \"@tambo-ai/typescript-sdk\";\nexport type {\n  Suggestion,\n  SuggestionGenerateParams,\n  SuggestionGenerateResponse,\n  SuggestionListResponse,\n} from \"@tambo-ai/typescript-sdk/resources/beta/threads/suggestions\";\nexport { useTamboThreadList } from \"./hooks/use-tambo-threads\";\nexport {\n  type ComponentContextToolMetadata,\n  type ComponentRegistry,\n  type ParameterSpec,\n  type RegisteredComponent,\n  type TamboTool,\n} from \"./model/component-metadata\";\nexport {\n  GenerationStage,\n  type TamboThreadMessage,\n} from \"./model/generate-component-response\";\nexport { type TamboThread } from \"./model/tambo-thread\";\n\nexport type {\n  TamboInteractableComponent as InteractableComponent,\n  TamboInteractableContext,\n} from \"./model/tambo-interactable\";\nexport {\n  withTamboInteractable as withInteractable,\n  type InteractableConfig,\n  type WithTamboInteractableProps,\n} from \"./providers/hoc/with-tambo-interactable\";\nexport { useTamboInteractable } from \"./providers/tambo-interactable-provider\";\n\n// Context helpers exports\nexport {\n  currentPageContextHelper,\n  currentTimeContextHelper,\n} from \"./context-helpers\";\nexport type {\n  AdditionalContext,\n  ContextHelperFn,\n  ContextHelpers,\n} from \"./context-helpers\";\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,wKAAwK;AAExK,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,OAAO,EACL,oBAAoB,EACpB,sBAAsB,GACvB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,sBAAsB,EAAE,MAAM,6BAA6B,CAAC;AACrE,cAAc,yBAAyB,CAAC;AACxC,OAAO,EACL,oBAAoB,GAGrB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAE/D,gCAAgC;AAChC,OAAO,EACL,mBAAmB,EACnB,sBAAsB,EACtB,2BAA2B,EAC3B,uBAAuB,EACvB,aAAa,EACb,iBAAiB,EACjB,mBAAmB,EACnB,QAAQ,EACR,cAAc,EACd,sBAAsB,EACtB,uBAAuB,EACvB,cAAc,EACd,cAAc,GAMf,MAAM,aAAa,CAAC;AAcrB,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AAQ/D,OAAO,EACL,eAAe,GAEhB,MAAM,qCAAqC,CAAC;AAO7C,OAAO,EACL,qBAAqB,IAAI,gBAAgB,GAG1C,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,oBAAoB,EAAE,MAAM,yCAAyC,CAAC;AAE/E,0BAA0B;AAC1B,OAAO,EACL,wBAAwB,EACxB,wBAAwB,GACzB,MAAM,mBAAmB,CAAC","sourcesContent":["/** Exports for the library. Only publically available exports are re-exported here. Anything not exported here is not supported and may change or break at any time. */\n\nexport { useTamboComponentState } from \"./hooks/use-component-state\";\nexport {\n  TamboMessageProvider,\n  useTamboCurrentMessage,\n} from \"./hooks/use-current-message\";\nexport { useTamboStreamingProps } from \"./hooks/use-streaming-props\";\nexport * from \"./hooks/use-suggestions\";\nexport {\n  useTamboStreamStatus,\n  type PropStatus,\n  type StreamStatus,\n} from \"./hooks/use-tambo-stream-status\";\nexport { useTamboThreadInput } from \"./hooks/use-thread-input\";\n\n// Re-export provider components\nexport {\n  TamboClientProvider,\n  TamboComponentProvider,\n  TamboContextHelpersProvider,\n  TamboPropStreamProvider,\n  TamboProvider,\n  TamboStubProvider,\n  TamboThreadProvider,\n  useTambo,\n  useTamboClient,\n  useTamboContextHelpers,\n  useTamboGenerationStage,\n  useTamboStream,\n  useTamboThread,\n  type TamboComponent,\n  type TamboContextHelpersContextProps,\n  type TamboContextHelpersProviderProps,\n  type TamboRegistryContext,\n  type TamboStubProviderProps,\n} from \"./providers\";\n\n// Re-export types from Tambo Node SDK\nexport type {\n  APIError,\n  RateLimitError,\n  TamboAIError,\n} from \"@tambo-ai/typescript-sdk\";\nexport type {\n  Suggestion,\n  SuggestionGenerateParams,\n  SuggestionGenerateResponse,\n  SuggestionListResponse,\n} from \"@tambo-ai/typescript-sdk/resources/beta/threads/suggestions\";\nexport { useTamboThreadList } from \"./hooks/use-tambo-threads\";\nexport {\n  type ComponentContextToolMetadata,\n  type ComponentRegistry,\n  type ParameterSpec,\n  type RegisteredComponent,\n  type TamboTool,\n} from \"./model/component-metadata\";\nexport {\n  GenerationStage,\n  type TamboThreadMessage,\n} from \"./model/generate-component-response\";\nexport { type TamboThread } from \"./model/tambo-thread\";\n\nexport type {\n  TamboInteractableComponent as InteractableComponent,\n  TamboInteractableContext,\n} from \"./model/tambo-interactable\";\nexport {\n  withTamboInteractable as withInteractable,\n  type InteractableConfig,\n  type WithTamboInteractableProps,\n} from \"./providers/hoc/with-tambo-interactable\";\nexport { useTamboInteractable } from \"./providers/tambo-interactable-provider\";\n\n// Context helpers exports\nexport {\n  currentPageContextHelper,\n  currentTimeContextHelper,\n} from \"./context-helpers\";\nexport type {\n  AdditionalContext,\n  ContextHelperFn,\n  ContextHelpers,\n} from \"./context-helpers\";\n"]}

package/esm/providers/__tests__/tambo-prop-stream-provider.test.js

@@ -2,11 +2,12 @@ import { render, screen } from "@testing-library/react";
 import React from "react";
 import { useTamboCurrentMessage } from "../../hooks/use-current-message";
 import { GenerationStage, } from "../../model/generate-component-response";
-import { useTamboThread, } from "../../providers/tambo-thread-provider";
+import { useTamboGenerationStage, useTamboThread, } from "../../providers/tambo-thread-provider";
 import { TamboPropStreamProvider, useTamboStream, } from "../tambo-prop-stream-provider";
 // Mock the required providers
 jest.mock("../../providers/tambo-thread-provider", () => ({
     useTamboThread: jest.fn(),
+    useTamboGenerationStage: jest.fn(),
 }));
 jest.mock("../../hooks/use-current-message", () => ({
     useTamboCurrentMessage: jest.fn(),
@@ -31,6 +32,32 @@ const createMockMessage = (overrides = {}) => ({
     threadId: "test-thread",
     ...overrides,
 });
+// Helper function to create mock CombinedTamboThreadContextProps
+const createMockThreadContext = (overrides = {}) => ({
+    // TamboThreadContextProps properties
+    thread: {
+        id: "test-thread",
+        projectId: "test-project",
+        messages: [],
+        name: "Test Thread",
+        createdAt: new Date().toISOString(),
+        updatedAt: new Date().toISOString(),
+    },
+    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(),
+    // GenerationStageContextProps properties
+    generationStage: GenerationStage.IDLE,
+    generationStatusMessage: "",
+    isIdle: true,
+    ...overrides,
+});
 // Helper component to test hook usage
 const TestHookComponent = ({ testKey = "default", }) => {
     const { streamStatus, getStatusForKey } = useTamboStream();
@@ -44,8 +71,11 @@ describe("TamboPropStreamProvider", () => {
         // Restore window for client-side tests
         global.window = originalWindow;
         // Default mock implementations
-        jest.mocked(useTamboThread).mockReturnValue({
+        jest.mocked(useTamboThread).mockReturnValue(createMockThreadContext());
+        jest.mocked(useTamboGenerationStage).mockReturnValue({
             generationStage: GenerationStage.IDLE,
+            generationStatusMessage: "",
+            isIdle: true,
         });
         jest.mocked(useTamboCurrentMessage).mockReturnValue({
             id: "test-message",
@@ -71,8 +101,13 @@ describe("TamboPropStreamProvider", () => {
     describe("Compound Components", () => {
         describe("Streaming Component", () => {
             it("should render streaming when isPending is true", () => {
-                jest.mocked(useTamboThread).mockReturnValue({
+                jest.mocked(useTamboThread).mockReturnValue(createMockThreadContext({
                     generationStage: GenerationStage.IDLE,
+                }));
+                jest.mocked(useTamboGenerationStage).mockReturnValue({
+                    generationStage: GenerationStage.IDLE,
+                    generationStatusMessage: "",
+                    isIdle: true,
                 });
                 jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
                     component: createMockComponent({ title: "" }),
@@ -83,8 +118,13 @@ describe("TamboPropStreamProvider", () => {
                 expect(screen.getByTestId("loading")).toBeInTheDocument();
             });
             it("should render streaming when isStreaming is true", () => {
-                jest.mocked(useTamboThread).mockReturnValue({
+                jest.mocked(useTamboThread).mockReturnValue(createMockThreadContext({
                     generationStage: GenerationStage.STREAMING_RESPONSE,
+                }));
+                jest.mocked(useTamboGenerationStage).mockReturnValue({
+                    generationStage: GenerationStage.STREAMING_RESPONSE,
+                    generationStatusMessage: "",
+                    isIdle: false,
                 });
                 jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
                     component: createMockComponent({ title: "Partial" }),
@@ -97,8 +137,13 @@ describe("TamboPropStreamProvider", () => {
         });
         describe("Success Component", () => {
             it("should not render success when isSuccess is false", () => {
-                jest.mocked(useTamboThread).mockReturnValue({
+                jest.mocked(useTamboThread).mockReturnValue(createMockThreadContext({
+                    generationStage: GenerationStage.IDLE,
+                }));
+                jest.mocked(useTamboGenerationStage).mockReturnValue({
                     generationStage: GenerationStage.IDLE,
+                    generationStatusMessage: "",
+                    isIdle: true,
                 });
                 jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
                     component: createMockComponent({ title: "" }),
@@ -111,8 +156,13 @@ describe("TamboPropStreamProvider", () => {
         });
         describe("Pending Component", () => {
             it("should render pending when no active status", () => {
-                jest.mocked(useTamboThread).mockReturnValue({
+                jest.mocked(useTamboThread).mockReturnValue(createMockThreadContext({
+                    generationStage: GenerationStage.IDLE,
+                }));
+                jest.mocked(useTamboGenerationStage).mockReturnValue({
                     generationStage: GenerationStage.IDLE,
+                    generationStatusMessage: "",
+                    isIdle: true,
                 });
                 jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
                     component: createMockComponent({ title: "" }),
@@ -123,8 +173,13 @@ describe("TamboPropStreamProvider", () => {
                 expect(screen.getByTestId("empty")).toBeInTheDocument();
             });
             it("should not render pending when isPending is true", () => {
-                jest.mocked(useTamboThread).mockReturnValue({
+                jest.mocked(useTamboThread).mockReturnValue(createMockThreadContext({
+                    generationStage: GenerationStage.STREAMING_RESPONSE,
+                }));
+                jest.mocked(useTamboGenerationStage).mockReturnValue({
                     generationStage: GenerationStage.STREAMING_RESPONSE,
+                    generationStatusMessage: "",
+                    isIdle: false,
                 });
                 jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
                     component: createMockComponent({ title: "Partial" }),
@@ -138,8 +193,13 @@ describe("TamboPropStreamProvider", () => {
     });
     describe("Key-based Status", () => {
         it("should provide status for keys not in propStatus", () => {
-            jest.mocked(useTamboThread).mockReturnValue({
+            jest.mocked(useTamboThread).mockReturnValue(createMockThreadContext({
+                generationStage: GenerationStage.COMPLETE,
+            }));
+            jest.mocked(useTamboGenerationStage).mockReturnValue({
                 generationStage: GenerationStage.COMPLETE,
+                generationStatusMessage: "",
+                isIdle: false,
             });
             jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
                 component: createMockComponent({ name: "John" }),
@@ -152,8 +212,13 @@ describe("TamboPropStreamProvider", () => {
     });
     describe("Compound Components with Keys", () => {
         it("should render loading for specific key when pending", () => {
-            jest.mocked(useTamboThread).mockReturnValue({
+            jest.mocked(useTamboThread).mockReturnValue(createMockThreadContext({
                 generationStage: GenerationStage.STREAMING_RESPONSE,
+            }));
+            jest.mocked(useTamboGenerationStage).mockReturnValue({
+                generationStage: GenerationStage.STREAMING_RESPONSE,
+                generationStatusMessage: "",
+                isIdle: false,
             });
             jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
                 component: createMockComponent({
@@ -176,8 +241,13 @@ describe("TamboPropStreamProvider", () => {
     });
     describe("Styling", () => {
         it("should apply className to loading component", () => {
-            jest.mocked(useTamboThread).mockReturnValue({
+            jest.mocked(useTamboThread).mockReturnValue(createMockThreadContext({
+                generationStage: GenerationStage.STREAMING_RESPONSE,
+            }));
+            jest.mocked(useTamboGenerationStage).mockReturnValue({
                 generationStage: GenerationStage.STREAMING_RESPONSE,
+                generationStatusMessage: "",
+                isIdle: false,
             });
             jest.mocked(useTamboCurrentMessage).mockReturnValue(createMockMessage({
                 component: createMockComponent({ title: "Partial" }),

package/esm/providers/__tests__/tambo-prop-stream-provider.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-prop-stream-provider.test.js","sourceRoot":"","sources":["../../../src/providers/__tests__/tambo-prop-stream-provider.test.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,sBAAsB,EAAE,MAAM,iCAAiC,CAAC;AACzE,OAAO,EACL,eAAe,GAEhB,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAEL,cAAc,GACf,MAAM,uCAAuC,CAAC;AAC/C,OAAO,EACL,uBAAuB,EACvB,cAAc,GACf,MAAM,+BAA+B,CAAC;AAEvC,8BAA8B;AAC9B,IAAI,CAAC,IAAI,CAAC,uCAAuC,EAAE,GAAG,EAAE,CAAC,CAAC;IACxD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,iCAAiC,EAAE,GAAG,EAAE,CAAC,CAAC;IAClD,sBAAsB,EAAE,IAAI,CAAC,EAAE,EAAE;CAClC,CAAC,CAAC,CAAC;AAEJ,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,sCAAsC;AACtC,MAAM,iBAAiB,GAAmC,CAAC,EACzD,OAAO,GAAG,SAAS,GACpB,EAAE,EAAE;IACH,MAAM,EAAE,YAAY,EAAE,eAAe,EAAE,GAAG,cAAc,EAAE,CAAC;IAC3D,MAAM,MAAM,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;IAExC,OAAO,CACL;QACE,4CAAiB,eAAe,IAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAO;QACrE,4CAAiB,YAAY,IAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAO,CACxD,CACP,CAAC;AACJ,CAAC,CAAC;AAEF,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,UAAU,CAAC,GAAG,EAAE;QACd,uCAAuC;QACvC,MAAM,CAAC,MAAM,GAAG,cAAc,CAAC;QAE/B,+BAA+B;QAC/B,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC;YAC1C,eAAe,EAAE,eAAe,CAAC,IAAI;SACX,CAAC,CAAC;QAE9B,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CAAC;YAClD,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,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,uCAAuC;YACvC,MAAM,aAAa,GAAG,OAAO,CAAC,KAAK,CAAC;YACpC,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;YAE1B,MAAM,CAAC,GAAG,EAAE;gBACV,MAAM,CAAC,oBAAC,iBAAiB,OAAG,CAAC,CAAC;YAChC,CAAC,CAAC,CAAC,OAAO,CACR,8DAA8D,CAC/D,CAAC;YAEF,OAAO,CAAC,KAAK,GAAG,aAAa,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;YACnC,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;gBACxD,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC;oBAC1C,eAAe,EAAE,eAAe,CAAC,IAAI;iBACX,CAAC,CAAC;gBAE9B,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;oBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;iBAC9C,CAAC,CACH,CAAC;gBAEF,MAAM,CACJ,oBAAC,uBAAuB;oBACtB,oBAAC,uBAAuB,CAAC,SAAS;wBAChC,4CAAiB,SAAS,iBAAiB,CACT,CACZ,CAC3B,CAAC;gBAEF,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC5D,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;gBAC1D,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC;oBAC1C,eAAe,EAAE,eAAe,CAAC,kBAAkB;iBACzB,CAAC,CAAC;gBAE9B,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;oBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;iBACrD,CAAC,CACH,CAAC;gBAEF,MAAM,CACJ,oBAAC,uBAAuB;oBACtB,oBAAC,uBAAuB,CAAC,SAAS;wBAChC,4CAAiB,SAAS,iBAAiB,CACT,CACZ,CAC3B,CAAC;gBAEF,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC5D,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;YACjC,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;gBAC3D,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC;oBAC1C,eAAe,EAAE,eAAe,CAAC,IAAI;iBACX,CAAC,CAAC;gBAE9B,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;oBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;iBAC9C,CAAC,CACH,CAAC;gBAEF,MAAM,CACJ,oBAAC,uBAAuB;oBACtB,oBAAC,uBAAuB,CAAC,OAAO;wBAC9B,4CAAiB,UAAU,gBAAgB,CACX,CACV,CAC3B,CAAC;gBAEF,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;YACnE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;YACjC,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;gBACrD,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC;oBAC1C,eAAe,EAAE,eAAe,CAAC,IAAI;iBACX,CAAC,CAAC;gBAE9B,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;oBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;iBAC9C,CAAC,CACH,CAAC;gBAEF,MAAM,CACJ,oBAAC,uBAAuB;oBACtB,oBAAC,uBAAuB,CAAC,OAAO;wBAC9B,4CAAiB,OAAO,aAAa,CACL,CACV,CAC3B,CAAC;gBAEF,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC1D,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;gBAC1D,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC;oBAC1C,eAAe,EAAE,eAAe,CAAC,kBAAkB;iBACzB,CAAC,CAAC;gBAE9B,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;oBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;iBACrD,CAAC,CACH,CAAC;gBAEF,MAAM,CACJ,oBAAC,uBAAuB;oBACtB,oBAAC,uBAAuB,CAAC,OAAO;wBAC9B,4CAAiB,OAAO,aAAa,CACL,CACV,CAC3B,CAAC;gBAEF,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;YAChE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,eAAe,CAAC,QAAQ;aACf,CAAC,CAAC;YAE9B,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;aACjD,CAAC,CACH,CAAC;YAEF,MAAM,CACJ,oBAAC,uBAAuB;gBACtB,oBAAC,iBAAiB,IAAC,OAAO,EAAC,aAAa,GAAG,CACnB,CAC3B,CAAC;YAEF,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAC1B,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,WAAY,CAC9C,CAAC;YACF,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,eAAe,CAAC,kBAAkB;aACzB,CAAC,CAAC;YAE9B,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC;oBAC7B,IAAI,EAAE,SAAS;oBACf,GAAG,EAAE,EAAE;iBACR,CAAC;aACH,CAAC,CACH,CAAC;YAEF,MAAM,CACJ,oBAAC,uBAAuB;gBACtB,oBAAC,uBAAuB,CAAC,SAAS,IAAC,SAAS,EAAC,MAAM;oBACjD,4CAAiB,cAAc,sBAAsB,CACnB;gBACpC,oBAAC,uBAAuB,CAAC,SAAS,IAAC,SAAS,EAAC,KAAK;oBAChD,4CAAiB,aAAa,qBAAqB,CACjB;gBACpC,oBAAC,uBAAuB,CAAC,OAAO,IAAC,SAAS,EAAC,MAAM;oBAC/C,4CAAiB,eAAe,qBAAqB,CACrB,CACV,CAC3B,CAAC;YAEF,gEAAgE;YAChE,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;QACxE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC;gBAC1C,eAAe,EAAE,eAAe,CAAC,kBAAkB;aACzB,CAAC,CAAC;YAE9B,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;aACrD,CAAC,CACH,CAAC;YAEF,MAAM,CACJ,oBAAC,uBAAuB;gBACtB,oBAAC,uBAAuB,CAAC,SAAS,IAAC,SAAS,EAAC,eAAe;oBAC1D,4CAAiB,SAAS,iBAAiB,CACT,CACZ,CAC3B,CAAC;YAEF,MAAM,cAAc,GAAG,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,aAAa,CAAC;YACnE,MAAM,CAAC,cAAc,CAAC,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { render, screen } from \"@testing-library/react\";\nimport React from \"react\";\nimport { useTamboCurrentMessage } from \"../../hooks/use-current-message\";\nimport {\n  GenerationStage,\n  TamboThreadMessage,\n} from \"../../model/generate-component-response\";\nimport {\n  TamboThreadContextProps,\n  useTamboThread,\n} from \"../../providers/tambo-thread-provider\";\nimport {\n  TamboPropStreamProvider,\n  useTamboStream,\n} from \"../tambo-prop-stream-provider\";\n\n// Mock the required providers\njest.mock(\"../../providers/tambo-thread-provider\", () => ({\n  useTamboThread: jest.fn(),\n}));\n\njest.mock(\"../../hooks/use-current-message\", () => ({\n  useTamboCurrentMessage: jest.fn(),\n}));\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// Helper component to test hook usage\nconst TestHookComponent: React.FC<{ testKey?: string }> = ({\n  testKey = \"default\",\n}) => {\n  const { streamStatus, getStatusForKey } = useTamboStream();\n  const status = getStatusForKey(testKey);\n\n  return (\n    <div>\n      <div data-testid=\"stream-status\">{JSON.stringify(streamStatus)}</div>\n      <div data-testid=\"key-status\">{JSON.stringify(status)}</div>\n    </div>\n  );\n};\n\ndescribe(\"TamboPropStreamProvider\", () => {\n  beforeEach(() => {\n    // Restore window for client-side tests\n    global.window = originalWindow;\n\n    // Default mock implementations\n    jest.mocked(useTamboThread).mockReturnValue({\n      generationStage: GenerationStage.IDLE,\n    } as TamboThreadContextProps);\n\n    jest.mocked(useTamboCurrentMessage).mockReturnValue({\n      id: \"test-message\",\n      component: {\n        props: {},\n      },\n    } as TamboThreadMessage);\n  });\n\n  afterEach(() => {\n    jest.clearAllMocks();\n  });\n\n  describe(\"Hook Error Handling\", () => {\n    it(\"should throw error when useTamboStream is used outside provider\", () => {\n      // Suppress console.error for this test\n      const originalError = console.error;\n      console.error = jest.fn();\n\n      expect(() => {\n        render(<TestHookComponent />);\n      }).toThrow(\n        \"useTamboStream must be used within a TamboPropStreamProvider\",\n      );\n\n      console.error = originalError;\n    });\n  });\n\n  describe(\"Compound Components\", () => {\n    describe(\"Streaming Component\", () => {\n      it(\"should render streaming when isPending is true\", () => {\n        jest.mocked(useTamboThread).mockReturnValue({\n          generationStage: GenerationStage.IDLE,\n        } as TamboThreadContextProps);\n\n        jest.mocked(useTamboCurrentMessage).mockReturnValue(\n          createMockMessage({\n            component: createMockComponent({ title: \"\" }),\n          }),\n        );\n\n        render(\n          <TamboPropStreamProvider>\n            <TamboPropStreamProvider.Streaming>\n              <div data-testid=\"loading\">Loading...</div>\n            </TamboPropStreamProvider.Streaming>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.getByTestId(\"loading\")).toBeInTheDocument();\n      });\n\n      it(\"should render streaming when isStreaming is true\", () => {\n        jest.mocked(useTamboThread).mockReturnValue({\n          generationStage: GenerationStage.STREAMING_RESPONSE,\n        } as TamboThreadContextProps);\n\n        jest.mocked(useTamboCurrentMessage).mockReturnValue(\n          createMockMessage({\n            component: createMockComponent({ title: \"Partial\" }),\n          }),\n        );\n\n        render(\n          <TamboPropStreamProvider>\n            <TamboPropStreamProvider.Streaming>\n              <div data-testid=\"loading\">Loading...</div>\n            </TamboPropStreamProvider.Streaming>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.getByTestId(\"loading\")).toBeInTheDocument();\n      });\n    });\n\n    describe(\"Success Component\", () => {\n      it(\"should not render success when isSuccess is false\", () => {\n        jest.mocked(useTamboThread).mockReturnValue({\n          generationStage: GenerationStage.IDLE,\n        } as TamboThreadContextProps);\n\n        jest.mocked(useTamboCurrentMessage).mockReturnValue(\n          createMockMessage({\n            component: createMockComponent({ title: \"\" }),\n          }),\n        );\n\n        render(\n          <TamboPropStreamProvider>\n            <TamboPropStreamProvider.Success>\n              <div data-testid=\"complete\">Complete!</div>\n            </TamboPropStreamProvider.Success>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.queryByTestId(\"complete\")).not.toBeInTheDocument();\n      });\n    });\n\n    describe(\"Pending Component\", () => {\n      it(\"should render pending when no active status\", () => {\n        jest.mocked(useTamboThread).mockReturnValue({\n          generationStage: GenerationStage.IDLE,\n        } as TamboThreadContextProps);\n\n        jest.mocked(useTamboCurrentMessage).mockReturnValue(\n          createMockMessage({\n            component: createMockComponent({ title: \"\" }),\n          }),\n        );\n\n        render(\n          <TamboPropStreamProvider>\n            <TamboPropStreamProvider.Pending>\n              <div data-testid=\"empty\">Empty!</div>\n            </TamboPropStreamProvider.Pending>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.getByTestId(\"empty\")).toBeInTheDocument();\n      });\n\n      it(\"should not render pending when isPending is true\", () => {\n        jest.mocked(useTamboThread).mockReturnValue({\n          generationStage: GenerationStage.STREAMING_RESPONSE,\n        } as TamboThreadContextProps);\n\n        jest.mocked(useTamboCurrentMessage).mockReturnValue(\n          createMockMessage({\n            component: createMockComponent({ title: \"Partial\" }),\n          }),\n        );\n\n        render(\n          <TamboPropStreamProvider>\n            <TamboPropStreamProvider.Pending>\n              <div data-testid=\"empty\">Empty!</div>\n            </TamboPropStreamProvider.Pending>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.queryByTestId(\"empty\")).not.toBeInTheDocument();\n      });\n    });\n  });\n\n  describe(\"Key-based Status\", () => {\n    it(\"should provide status for keys not in propStatus\", () => {\n      jest.mocked(useTamboThread).mockReturnValue({\n        generationStage: GenerationStage.COMPLETE,\n      } as TamboThreadContextProps);\n\n      jest.mocked(useTamboCurrentMessage).mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({ name: \"John\" }),\n        }),\n      );\n\n      render(\n        <TamboPropStreamProvider>\n          <TestHookComponent testKey=\"nonexistent\" />\n        </TamboPropStreamProvider>,\n      );\n\n      const keyStatus = JSON.parse(\n        screen.getByTestId(\"key-status\").textContent!,\n      );\n      expect(keyStatus.isPending).toBe(true);\n    });\n  });\n\n  describe(\"Compound Components with Keys\", () => {\n    it(\"should render loading for specific key when pending\", () => {\n      jest.mocked(useTamboThread).mockReturnValue({\n        generationStage: GenerationStage.STREAMING_RESPONSE,\n      } as TamboThreadContextProps);\n\n      jest.mocked(useTamboCurrentMessage).mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({\n            name: \"Partial\",\n            age: 25,\n          }),\n        }),\n      );\n\n      render(\n        <TamboPropStreamProvider>\n          <TamboPropStreamProvider.Streaming streamKey=\"name\">\n            <div data-testid=\"name-loading\">Name Loading...</div>\n          </TamboPropStreamProvider.Streaming>\n          <TamboPropStreamProvider.Streaming streamKey=\"age\">\n            <div data-testid=\"age-loading\">Age Loading...</div>\n          </TamboPropStreamProvider.Streaming>\n          <TamboPropStreamProvider.Success streamKey=\"name\">\n            <div data-testid=\"name-complete\">Name Complete!</div>\n          </TamboPropStreamProvider.Success>\n        </TamboPropStreamProvider>,\n      );\n\n      // Both props should be loading since they're in streaming stage\n      expect(screen.getByTestId(\"name-loading\")).toBeInTheDocument();\n      expect(screen.getByTestId(\"age-loading\")).toBeInTheDocument();\n      expect(screen.queryByTestId(\"name-complete\")).not.toBeInTheDocument();\n    });\n  });\n\n  describe(\"Styling\", () => {\n    it(\"should apply className to loading component\", () => {\n      jest.mocked(useTamboThread).mockReturnValue({\n        generationStage: GenerationStage.STREAMING_RESPONSE,\n      } as TamboThreadContextProps);\n\n      jest.mocked(useTamboCurrentMessage).mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({ title: \"Partial\" }),\n        }),\n      );\n\n      render(\n        <TamboPropStreamProvider>\n          <TamboPropStreamProvider.Streaming className=\"loading-class\">\n            <div data-testid=\"loading\">Loading...</div>\n          </TamboPropStreamProvider.Streaming>\n        </TamboPropStreamProvider>,\n      );\n\n      const loadingElement = screen.getByTestId(\"loading\").parentElement;\n      expect(loadingElement).toHaveClass(\"loading-class\");\n    });\n  });\n});\n"]}
+{"version":3,"file":"tambo-prop-stream-provider.test.js","sourceRoot":"","sources":["../../../src/providers/__tests__/tambo-prop-stream-provider.test.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,sBAAsB,EAAE,MAAM,iCAAiC,CAAC;AACzE,OAAO,EACL,eAAe,GAEhB,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAEL,uBAAuB,EACvB,cAAc,GACf,MAAM,uCAAuC,CAAC;AAC/C,OAAO,EACL,uBAAuB,EACvB,cAAc,GACf,MAAM,+BAA+B,CAAC;AAEvC,8BAA8B;AAC9B,IAAI,CAAC,IAAI,CAAC,uCAAuC,EAAE,GAAG,EAAE,CAAC,CAAC;IACxD,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,uBAAuB,EAAE,IAAI,CAAC,EAAE,EAAE;CACnC,CAAC,CAAC,CAAC;AAEJ,IAAI,CAAC,IAAI,CAAC,iCAAiC,EAAE,GAAG,EAAE,CAAC,CAAC;IAClD,sBAAsB,EAAE,IAAI,CAAC,EAAE,EAAE;CAClC,CAAC,CAAC,CAAC;AAEJ,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,iEAAiE;AACjE,MAAM,uBAAuB,GAAG,CAC9B,YAAsD,EAAE,EACvB,EAAE,CAAC,CAAC;IACrC,qCAAqC;IACrC,MAAM,EAAE;QACN,EAAE,EAAE,aAAa;QACjB,SAAS,EAAE,cAAc;QACzB,QAAQ,EAAE,EAAE;QACZ,IAAI,EAAE,aAAa;QACnB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACpC;IACD,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC9B,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;IACzB,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC3B,kBAAkB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC7B,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC3B,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC9B,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;IACjB,SAAS,EAAE,KAAK;IAChB,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAE;IAC5B,yCAAyC;IACzC,eAAe,EAAE,eAAe,CAAC,IAAI;IACrC,uBAAuB,EAAE,EAAE;IAC3B,MAAM,EAAE,IAAI;IACZ,GAAG,SAAS;CACb,CAAC,CAAC;AAEH,sCAAsC;AACtC,MAAM,iBAAiB,GAAmC,CAAC,EACzD,OAAO,GAAG,SAAS,GACpB,EAAE,EAAE;IACH,MAAM,EAAE,YAAY,EAAE,eAAe,EAAE,GAAG,cAAc,EAAE,CAAC;IAC3D,MAAM,MAAM,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;IAExC,OAAO,CACL;QACE,4CAAiB,eAAe,IAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAO;QACrE,4CAAiB,YAAY,IAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAO,CACxD,CACP,CAAC;AACJ,CAAC,CAAC;AAEF,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,UAAU,CAAC,GAAG,EAAE;QACd,uCAAuC;QACvC,MAAM,CAAC,MAAM,GAAG,cAAc,CAAC;QAE/B,+BAA+B;QAC/B,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,uBAAuB,EAAE,CAAC,CAAC;QAEvE,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;YACnD,eAAe,EAAE,eAAe,CAAC,IAAI;YACrC,uBAAuB,EAAE,EAAE;YAC3B,MAAM,EAAE,IAAI;SACb,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CAAC;YAClD,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,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,uCAAuC;YACvC,MAAM,aAAa,GAAG,OAAO,CAAC,KAAK,CAAC;YACpC,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;YAE1B,MAAM,CAAC,GAAG,EAAE;gBACV,MAAM,CAAC,oBAAC,iBAAiB,OAAG,CAAC,CAAC;YAChC,CAAC,CAAC,CAAC,OAAO,CACR,8DAA8D,CAC/D,CAAC;YAEF,OAAO,CAAC,KAAK,GAAG,aAAa,CAAC;QAChC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;YACnC,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;gBACxD,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;oBACtB,eAAe,EAAE,eAAe,CAAC,IAAI;iBACtC,CAAC,CACH,CAAC;gBAEF,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;oBACnD,eAAe,EAAE,eAAe,CAAC,IAAI;oBACrC,uBAAuB,EAAE,EAAE;oBAC3B,MAAM,EAAE,IAAI;iBACb,CAAC,CAAC;gBAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;oBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;iBAC9C,CAAC,CACH,CAAC;gBAEF,MAAM,CACJ,oBAAC,uBAAuB;oBACtB,oBAAC,uBAAuB,CAAC,SAAS;wBAChC,4CAAiB,SAAS,iBAAiB,CACT,CACZ,CAC3B,CAAC;gBAEF,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC5D,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;gBAC1D,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;oBACtB,eAAe,EAAE,eAAe,CAAC,kBAAkB;iBACpD,CAAC,CACH,CAAC;gBAEF,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;oBACnD,eAAe,EAAE,eAAe,CAAC,kBAAkB;oBACnD,uBAAuB,EAAE,EAAE;oBAC3B,MAAM,EAAE,KAAK;iBACd,CAAC,CAAC;gBAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;oBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;iBACrD,CAAC,CACH,CAAC;gBAEF,MAAM,CACJ,oBAAC,uBAAuB;oBACtB,oBAAC,uBAAuB,CAAC,SAAS;wBAChC,4CAAiB,SAAS,iBAAiB,CACT,CACZ,CAC3B,CAAC;gBAEF,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC5D,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;YACjC,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;gBAC3D,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;oBACtB,eAAe,EAAE,eAAe,CAAC,IAAI;iBACtC,CAAC,CACH,CAAC;gBAEF,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;oBACnD,eAAe,EAAE,eAAe,CAAC,IAAI;oBACrC,uBAAuB,EAAE,EAAE;oBAC3B,MAAM,EAAE,IAAI;iBACb,CAAC,CAAC;gBAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;oBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;iBAC9C,CAAC,CACH,CAAC;gBAEF,MAAM,CACJ,oBAAC,uBAAuB;oBACtB,oBAAC,uBAAuB,CAAC,OAAO;wBAC9B,4CAAiB,UAAU,gBAAgB,CACX,CACV,CAC3B,CAAC;gBAEF,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;YACnE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;YACjC,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;gBACrD,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;oBACtB,eAAe,EAAE,eAAe,CAAC,IAAI;iBACtC,CAAC,CACH,CAAC;gBAEF,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;oBACnD,eAAe,EAAE,eAAe,CAAC,IAAI;oBACrC,uBAAuB,EAAE,EAAE;oBAC3B,MAAM,EAAE,IAAI;iBACb,CAAC,CAAC;gBAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;oBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;iBAC9C,CAAC,CACH,CAAC;gBAEF,MAAM,CACJ,oBAAC,uBAAuB;oBACtB,oBAAC,uBAAuB,CAAC,OAAO;wBAC9B,4CAAiB,OAAO,aAAa,CACL,CACV,CAC3B,CAAC;gBAEF,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC1D,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;gBAC1D,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;oBACtB,eAAe,EAAE,eAAe,CAAC,kBAAkB;iBACpD,CAAC,CACH,CAAC;gBAEF,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;oBACnD,eAAe,EAAE,eAAe,CAAC,kBAAkB;oBACnD,uBAAuB,EAAE,EAAE;oBAC3B,MAAM,EAAE,KAAK;iBACd,CAAC,CAAC;gBAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;oBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;iBACrD,CAAC,CACH,CAAC;gBAEF,MAAM,CACJ,oBAAC,uBAAuB;oBACtB,oBAAC,uBAAuB,CAAC,OAAO;wBAC9B,4CAAiB,OAAO,aAAa,CACL,CACV,CAC3B,CAAC;gBAEF,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;YAChE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;gBACtB,eAAe,EAAE,eAAe,CAAC,QAAQ;aAC1C,CAAC,CACH,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;gBACnD,eAAe,EAAE,eAAe,CAAC,QAAQ;gBACzC,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;aACjD,CAAC,CACH,CAAC;YAEF,MAAM,CACJ,oBAAC,uBAAuB;gBACtB,oBAAC,iBAAiB,IAAC,OAAO,EAAC,aAAa,GAAG,CACnB,CAC3B,CAAC;YAEF,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAC1B,MAAM,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,WAAY,CAC9C,CAAC;YACF,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;gBACtB,eAAe,EAAE,eAAe,CAAC,kBAAkB;aACpD,CAAC,CACH,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;gBACnD,eAAe,EAAE,eAAe,CAAC,kBAAkB;gBACnD,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC;oBAC7B,IAAI,EAAE,SAAS;oBACf,GAAG,EAAE,EAAE;iBACR,CAAC;aACH,CAAC,CACH,CAAC;YAEF,MAAM,CACJ,oBAAC,uBAAuB;gBACtB,oBAAC,uBAAuB,CAAC,SAAS,IAAC,SAAS,EAAC,MAAM;oBACjD,4CAAiB,cAAc,sBAAsB,CACnB;gBACpC,oBAAC,uBAAuB,CAAC,SAAS,IAAC,SAAS,EAAC,KAAK;oBAChD,4CAAiB,aAAa,qBAAqB,CACjB;gBACpC,oBAAC,uBAAuB,CAAC,OAAO,IAAC,SAAS,EAAC,MAAM;oBAC/C,4CAAiB,eAAe,qBAAqB,CACrB,CACV,CAC3B,CAAC;YAEF,gEAAgE;YAChE,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;QACxE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CACzC,uBAAuB,CAAC;gBACtB,eAAe,EAAE,eAAe,CAAC,kBAAkB;aACpD,CAAC,CACH,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,eAAe,CAAC;gBACnD,eAAe,EAAE,eAAe,CAAC,kBAAkB;gBACnD,uBAAuB,EAAE,EAAE;gBAC3B,MAAM,EAAE,KAAK;aACd,CAAC,CAAC;YAEH,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,eAAe,CACjD,iBAAiB,CAAC;gBAChB,SAAS,EAAE,mBAAmB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;aACrD,CAAC,CACH,CAAC;YAEF,MAAM,CACJ,oBAAC,uBAAuB;gBACtB,oBAAC,uBAAuB,CAAC,SAAS,IAAC,SAAS,EAAC,eAAe;oBAC1D,4CAAiB,SAAS,iBAAiB,CACT,CACZ,CAC3B,CAAC;YAEF,MAAM,cAAc,GAAG,MAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,aAAa,CAAC;YACnE,MAAM,CAAC,cAAc,CAAC,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { render, screen } from \"@testing-library/react\";\nimport React from \"react\";\nimport { useTamboCurrentMessage } from \"../../hooks/use-current-message\";\nimport {\n  GenerationStage,\n  TamboThreadMessage,\n} from \"../../model/generate-component-response\";\nimport {\n  CombinedTamboThreadContextProps,\n  useTamboGenerationStage,\n  useTamboThread,\n} from \"../../providers/tambo-thread-provider\";\nimport {\n  TamboPropStreamProvider,\n  useTamboStream,\n} from \"../tambo-prop-stream-provider\";\n\n// Mock the required providers\njest.mock(\"../../providers/tambo-thread-provider\", () => ({\n  useTamboThread: jest.fn(),\n  useTamboGenerationStage: jest.fn(),\n}));\n\njest.mock(\"../../hooks/use-current-message\", () => ({\n  useTamboCurrentMessage: jest.fn(),\n}));\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// Helper function to create mock CombinedTamboThreadContextProps\nconst createMockThreadContext = (\n  overrides: Partial<CombinedTamboThreadContextProps> = {},\n): CombinedTamboThreadContextProps => ({\n  // TamboThreadContextProps properties\n  thread: {\n    id: \"test-thread\",\n    projectId: \"test-project\",\n    messages: [],\n    name: \"Test Thread\",\n    createdAt: new Date().toISOString(),\n    updatedAt: new Date().toISOString(),\n  },\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  // GenerationStageContextProps properties\n  generationStage: GenerationStage.IDLE,\n  generationStatusMessage: \"\",\n  isIdle: true,\n  ...overrides,\n});\n\n// Helper component to test hook usage\nconst TestHookComponent: React.FC<{ testKey?: string }> = ({\n  testKey = \"default\",\n}) => {\n  const { streamStatus, getStatusForKey } = useTamboStream();\n  const status = getStatusForKey(testKey);\n\n  return (\n    <div>\n      <div data-testid=\"stream-status\">{JSON.stringify(streamStatus)}</div>\n      <div data-testid=\"key-status\">{JSON.stringify(status)}</div>\n    </div>\n  );\n};\n\ndescribe(\"TamboPropStreamProvider\", () => {\n  beforeEach(() => {\n    // Restore window for client-side tests\n    global.window = originalWindow;\n\n    // Default mock implementations\n    jest.mocked(useTamboThread).mockReturnValue(createMockThreadContext());\n\n    jest.mocked(useTamboGenerationStage).mockReturnValue({\n      generationStage: GenerationStage.IDLE,\n      generationStatusMessage: \"\",\n      isIdle: true,\n    });\n\n    jest.mocked(useTamboCurrentMessage).mockReturnValue({\n      id: \"test-message\",\n      component: {\n        props: {},\n      },\n    } as TamboThreadMessage);\n  });\n\n  afterEach(() => {\n    jest.clearAllMocks();\n  });\n\n  describe(\"Hook Error Handling\", () => {\n    it(\"should throw error when useTamboStream is used outside provider\", () => {\n      // Suppress console.error for this test\n      const originalError = console.error;\n      console.error = jest.fn();\n\n      expect(() => {\n        render(<TestHookComponent />);\n      }).toThrow(\n        \"useTamboStream must be used within a TamboPropStreamProvider\",\n      );\n\n      console.error = originalError;\n    });\n  });\n\n  describe(\"Compound Components\", () => {\n    describe(\"Streaming Component\", () => {\n      it(\"should render streaming when isPending is true\", () => {\n        jest.mocked(useTamboThread).mockReturnValue(\n          createMockThreadContext({\n            generationStage: GenerationStage.IDLE,\n          }),\n        );\n\n        jest.mocked(useTamboGenerationStage).mockReturnValue({\n          generationStage: GenerationStage.IDLE,\n          generationStatusMessage: \"\",\n          isIdle: true,\n        });\n\n        jest.mocked(useTamboCurrentMessage).mockReturnValue(\n          createMockMessage({\n            component: createMockComponent({ title: \"\" }),\n          }),\n        );\n\n        render(\n          <TamboPropStreamProvider>\n            <TamboPropStreamProvider.Streaming>\n              <div data-testid=\"loading\">Loading...</div>\n            </TamboPropStreamProvider.Streaming>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.getByTestId(\"loading\")).toBeInTheDocument();\n      });\n\n      it(\"should render streaming when isStreaming is true\", () => {\n        jest.mocked(useTamboThread).mockReturnValue(\n          createMockThreadContext({\n            generationStage: GenerationStage.STREAMING_RESPONSE,\n          }),\n        );\n\n        jest.mocked(useTamboGenerationStage).mockReturnValue({\n          generationStage: GenerationStage.STREAMING_RESPONSE,\n          generationStatusMessage: \"\",\n          isIdle: false,\n        });\n\n        jest.mocked(useTamboCurrentMessage).mockReturnValue(\n          createMockMessage({\n            component: createMockComponent({ title: \"Partial\" }),\n          }),\n        );\n\n        render(\n          <TamboPropStreamProvider>\n            <TamboPropStreamProvider.Streaming>\n              <div data-testid=\"loading\">Loading...</div>\n            </TamboPropStreamProvider.Streaming>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.getByTestId(\"loading\")).toBeInTheDocument();\n      });\n    });\n\n    describe(\"Success Component\", () => {\n      it(\"should not render success when isSuccess is false\", () => {\n        jest.mocked(useTamboThread).mockReturnValue(\n          createMockThreadContext({\n            generationStage: GenerationStage.IDLE,\n          }),\n        );\n\n        jest.mocked(useTamboGenerationStage).mockReturnValue({\n          generationStage: GenerationStage.IDLE,\n          generationStatusMessage: \"\",\n          isIdle: true,\n        });\n\n        jest.mocked(useTamboCurrentMessage).mockReturnValue(\n          createMockMessage({\n            component: createMockComponent({ title: \"\" }),\n          }),\n        );\n\n        render(\n          <TamboPropStreamProvider>\n            <TamboPropStreamProvider.Success>\n              <div data-testid=\"complete\">Complete!</div>\n            </TamboPropStreamProvider.Success>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.queryByTestId(\"complete\")).not.toBeInTheDocument();\n      });\n    });\n\n    describe(\"Pending Component\", () => {\n      it(\"should render pending when no active status\", () => {\n        jest.mocked(useTamboThread).mockReturnValue(\n          createMockThreadContext({\n            generationStage: GenerationStage.IDLE,\n          }),\n        );\n\n        jest.mocked(useTamboGenerationStage).mockReturnValue({\n          generationStage: GenerationStage.IDLE,\n          generationStatusMessage: \"\",\n          isIdle: true,\n        });\n\n        jest.mocked(useTamboCurrentMessage).mockReturnValue(\n          createMockMessage({\n            component: createMockComponent({ title: \"\" }),\n          }),\n        );\n\n        render(\n          <TamboPropStreamProvider>\n            <TamboPropStreamProvider.Pending>\n              <div data-testid=\"empty\">Empty!</div>\n            </TamboPropStreamProvider.Pending>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.getByTestId(\"empty\")).toBeInTheDocument();\n      });\n\n      it(\"should not render pending when isPending is true\", () => {\n        jest.mocked(useTamboThread).mockReturnValue(\n          createMockThreadContext({\n            generationStage: GenerationStage.STREAMING_RESPONSE,\n          }),\n        );\n\n        jest.mocked(useTamboGenerationStage).mockReturnValue({\n          generationStage: GenerationStage.STREAMING_RESPONSE,\n          generationStatusMessage: \"\",\n          isIdle: false,\n        });\n\n        jest.mocked(useTamboCurrentMessage).mockReturnValue(\n          createMockMessage({\n            component: createMockComponent({ title: \"Partial\" }),\n          }),\n        );\n\n        render(\n          <TamboPropStreamProvider>\n            <TamboPropStreamProvider.Pending>\n              <div data-testid=\"empty\">Empty!</div>\n            </TamboPropStreamProvider.Pending>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.queryByTestId(\"empty\")).not.toBeInTheDocument();\n      });\n    });\n  });\n\n  describe(\"Key-based Status\", () => {\n    it(\"should provide status for keys not in propStatus\", () => {\n      jest.mocked(useTamboThread).mockReturnValue(\n        createMockThreadContext({\n          generationStage: GenerationStage.COMPLETE,\n        }),\n      );\n\n      jest.mocked(useTamboGenerationStage).mockReturnValue({\n        generationStage: GenerationStage.COMPLETE,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n\n      jest.mocked(useTamboCurrentMessage).mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({ name: \"John\" }),\n        }),\n      );\n\n      render(\n        <TamboPropStreamProvider>\n          <TestHookComponent testKey=\"nonexistent\" />\n        </TamboPropStreamProvider>,\n      );\n\n      const keyStatus = JSON.parse(\n        screen.getByTestId(\"key-status\").textContent!,\n      );\n      expect(keyStatus.isPending).toBe(true);\n    });\n  });\n\n  describe(\"Compound Components with Keys\", () => {\n    it(\"should render loading for specific key when pending\", () => {\n      jest.mocked(useTamboThread).mockReturnValue(\n        createMockThreadContext({\n          generationStage: GenerationStage.STREAMING_RESPONSE,\n        }),\n      );\n\n      jest.mocked(useTamboGenerationStage).mockReturnValue({\n        generationStage: GenerationStage.STREAMING_RESPONSE,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n\n      jest.mocked(useTamboCurrentMessage).mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({\n            name: \"Partial\",\n            age: 25,\n          }),\n        }),\n      );\n\n      render(\n        <TamboPropStreamProvider>\n          <TamboPropStreamProvider.Streaming streamKey=\"name\">\n            <div data-testid=\"name-loading\">Name Loading...</div>\n          </TamboPropStreamProvider.Streaming>\n          <TamboPropStreamProvider.Streaming streamKey=\"age\">\n            <div data-testid=\"age-loading\">Age Loading...</div>\n          </TamboPropStreamProvider.Streaming>\n          <TamboPropStreamProvider.Success streamKey=\"name\">\n            <div data-testid=\"name-complete\">Name Complete!</div>\n          </TamboPropStreamProvider.Success>\n        </TamboPropStreamProvider>,\n      );\n\n      // Both props should be loading since they're in streaming stage\n      expect(screen.getByTestId(\"name-loading\")).toBeInTheDocument();\n      expect(screen.getByTestId(\"age-loading\")).toBeInTheDocument();\n      expect(screen.queryByTestId(\"name-complete\")).not.toBeInTheDocument();\n    });\n  });\n\n  describe(\"Styling\", () => {\n    it(\"should apply className to loading component\", () => {\n      jest.mocked(useTamboThread).mockReturnValue(\n        createMockThreadContext({\n          generationStage: GenerationStage.STREAMING_RESPONSE,\n        }),\n      );\n\n      jest.mocked(useTamboGenerationStage).mockReturnValue({\n        generationStage: GenerationStage.STREAMING_RESPONSE,\n        generationStatusMessage: \"\",\n        isIdle: false,\n      });\n\n      jest.mocked(useTamboCurrentMessage).mockReturnValue(\n        createMockMessage({\n          component: createMockComponent({ title: \"Partial\" }),\n        }),\n      );\n\n      render(\n        <TamboPropStreamProvider>\n          <TamboPropStreamProvider.Streaming className=\"loading-class\">\n            <div data-testid=\"loading\">Loading...</div>\n          </TamboPropStreamProvider.Streaming>\n        </TamboPropStreamProvider>,\n      );\n\n      const loadingElement = screen.getByTestId(\"loading\").parentElement;\n      expect(loadingElement).toHaveClass(\"loading-class\");\n    });\n  });\n});\n"]}

package/esm/providers/__tests__/tambo-thread-provider.test.js

@@ -226,13 +226,6 @@ describe("TamboThreadProvider", () => {
         });
         expect(result.current.generationStage).toBe(GenerationStage.COMPLETE);
     });
-    it("should handle input value changes", () => {
-        const { result } = renderHook(() => useTamboThread(), { wrapper });
-        act(() => {
-            result.current.setInputValue("New input");
-        });
-        expect(result.current.inputValue).toBe("New input");
-    });
     it("should handle streaming responses", async () => {
         const mockStreamResponse = {
             responseMessageDto: {
@@ -621,5 +614,56 @@ describe("TamboThreadProvider", () => {
             expect(mockThreadsApi.advanceById).not.toHaveBeenCalled();
         });
     });
+    describe("error handling", () => {
+        it("should set generation stage to ERROR when non-streaming sendThreadMessage fails", async () => {
+            const testError = new Error("API call failed");
+            // Mock advanceById to throw an error
+            jest.mocked(mockThreadsApi.advanceById).mockRejectedValue(testError);
+            const { result } = renderHook(() => useTamboThread(), { wrapper });
+            // Expect the error to be thrown
+            await act(async () => {
+                await result.current.switchCurrentThread("test-thread-1");
+                await expect(result.current.sendThreadMessage("Hello", {
+                    threadId: "test-thread-1",
+                    streamResponse: false,
+                })).rejects.toThrow("API call failed");
+            });
+            // Verify generation stage is set to ERROR
+            expect(result.current.generationStage).toBe(GenerationStage.ERROR);
+        });
+        it("should set generation stage to ERROR when streaming sendThreadMessage fails", async () => {
+            const testError = new Error("Streaming API call failed");
+            // Mock advanceStream to throw an error
+            jest.mocked(advanceStream).mockRejectedValue(testError);
+            const { result } = renderHook(() => useTamboThread(), { wrapper });
+            // Expect the error to be thrown
+            await act(async () => {
+                await result.current.switchCurrentThread("test-thread-1");
+                await expect(result.current.sendThreadMessage("Hello", {
+                    threadId: "test-thread-1",
+                    streamResponse: true,
+                })).rejects.toThrow("Streaming API call failed");
+            });
+            // Verify generation stage is set to ERROR
+            expect(result.current.generationStage).toBe(GenerationStage.ERROR);
+        });
+        it("should set generation stage to ERROR when advance API call fails for placeholder thread", async () => {
+            const testError = new Error("Advance API call failed");
+            // Mock advance to throw an error
+            jest.mocked(mockThreadsApi.advance).mockRejectedValue(testError);
+            const { result } = renderHook(() => useTamboThread(), { wrapper });
+            // Start with placeholder thread (which is the default state)
+            expect(result.current.thread.id).toBe("placeholder");
+            // Expect the error to be thrown
+            await act(async () => {
+                await expect(result.current.sendThreadMessage("Hello", {
+                    threadId: "placeholder",
+                    streamResponse: false,
+                })).rejects.toThrow("Advance API call failed");
+            });
+            // Verify generation stage is set to ERROR
+            expect(result.current.generationStage).toBe(GenerationStage.ERROR);
+        });
+    });
 });
 //# sourceMappingURL=tambo-thread-provider.test.js.map