@tambo-ai/react 0.37.2 → 0.38.0

package/dist/index.d.ts

@@ -2,8 +2,8 @@
 export { useTamboComponentState } from "./hooks/use-component-state";
 export { TamboMessageProvider, useTamboCurrentMessage, useTamboMessageContext, } from "./hooks/use-current-message";
 export { useTamboStreamingProps } from "./hooks/use-streaming-props";
-export { useTamboStreamStatus, type StreamStatus, type PropStatus, } from "./hooks/use-tambo-stream-status";
 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, TamboProvider, TamboStubProvider, TamboThreadProvider, useTambo, useTamboClient, useTamboThread, type TamboComponent, type TamboRegistryContext, type TamboStubProviderProps, } from "./providers";
 export type { APIError, RateLimitError, TamboAIError, } from "@tambo-ai/typescript-sdk";
@@ -12,4 +12,7 @@ export { useTamboThreadList } from "./hooks/use-tambo-threads";
 export { type ComponentContextToolMetadata, type ComponentRegistry, type ParameterSpec, type RegisteredComponent, type TamboTool, } from "./model/component-metadata";
 export { GenerationStage, type TamboThreadMessage, } from "./model/generate-component-response";
 export { type TamboThread } from "./model/tambo-thread";
+export type { TamboInteractableComponent as InteractableComponent, TamboInteractableContext, } from "./model/tambo-interactable";
+export { withTamboInteractable as withInteractable, type InteractableConfig, type WithTamboInteractableProps, } from "./providers/hoc/with-tambo-interactable";
+export { useTamboInteractable } from "./providers/tambo-interactable-provider";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,wKAAwK;AACxK,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,OAAO,EACL,oBAAoB,EACpB,KAAK,YAAY,EACjB,KAAK,UAAU,GAChB,MAAM,iCAAiC,CAAC;AACzC,cAAc,yBAAyB,CAAC;AACxC,OAAO,EAAE,mBAAmB,EAAE,MAAM,0BAA0B,CAAC;AAG/D,OAAO,EACL,mBAAmB,EACnB,sBAAsB,EACtB,aAAa,EACb,iBAAiB,EACjB,mBAAmB,EACnB,QAAQ,EACR,cAAc,EACd,cAAc,EACd,KAAK,cAAc,EACnB,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"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,wKAAwK;AACxK,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,aAAa,EACb,iBAAiB,EACjB,mBAAmB,EACnB,QAAQ,EACR,cAAc,EACd,cAAc,EACd,KAAK,cAAc,EACnB,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"}

package/dist/index.js

@@ -14,7 +14,7 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.GenerationStage = exports.useTamboThreadList = exports.useTamboThread = exports.useTamboClient = exports.useTambo = exports.TamboThreadProvider = exports.TamboStubProvider = exports.TamboProvider = exports.TamboComponentProvider = exports.TamboClientProvider = exports.useTamboThreadInput = exports.useTamboStreamStatus = exports.useTamboStreamingProps = exports.useTamboMessageContext = exports.useTamboCurrentMessage = exports.TamboMessageProvider = exports.useTamboComponentState = void 0;
+exports.useTamboInteractable = exports.withInteractable = exports.GenerationStage = exports.useTamboThreadList = exports.useTamboThread = exports.useTamboClient = exports.useTambo = exports.TamboThreadProvider = exports.TamboStubProvider = exports.TamboProvider = exports.TamboComponentProvider = exports.TamboClientProvider = exports.useTamboThreadInput = exports.useTamboStreamStatus = exports.useTamboStreamingProps = exports.useTamboMessageContext = exports.useTamboCurrentMessage = exports.TamboMessageProvider = exports.useTamboComponentState = void 0;
 /** 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. */
 var use_component_state_1 = require("./hooks/use-component-state");
 Object.defineProperty(exports, "useTamboComponentState", { enumerable: true, get: function () { return use_component_state_1.useTamboComponentState; } });
@@ -24,9 +24,9 @@ Object.defineProperty(exports, "useTamboCurrentMessage", { enumerable: true, get
 Object.defineProperty(exports, "useTamboMessageContext", { enumerable: true, get: function () { return use_current_message_1.useTamboMessageContext; } });
 var use_streaming_props_1 = require("./hooks/use-streaming-props");
 Object.defineProperty(exports, "useTamboStreamingProps", { enumerable: true, get: function () { return use_streaming_props_1.useTamboStreamingProps; } });
+__exportStar(require("./hooks/use-suggestions"), exports);
 var use_tambo_stream_status_1 = require("./hooks/use-tambo-stream-status");
 Object.defineProperty(exports, "useTamboStreamStatus", { enumerable: true, get: function () { return use_tambo_stream_status_1.useTamboStreamStatus; } });
-__exportStar(require("./hooks/use-suggestions"), exports);
 var use_thread_input_1 = require("./hooks/use-thread-input");
 Object.defineProperty(exports, "useTamboThreadInput", { enumerable: true, get: function () { return use_thread_input_1.useTamboThreadInput; } });
 // Re-export provider components
@@ -43,4 +43,8 @@ var use_tambo_threads_1 = require("./hooks/use-tambo-threads");
 Object.defineProperty(exports, "useTamboThreadList", { enumerable: true, get: function () { return use_tambo_threads_1.useTamboThreadList; } });
 var generate_component_response_1 = require("./model/generate-component-response");
 Object.defineProperty(exports, "GenerationStage", { enumerable: true, get: function () { return generate_component_response_1.GenerationStage; } });
+var with_tambo_interactable_1 = require("./providers/hoc/with-tambo-interactable");
+Object.defineProperty(exports, "withInteractable", { enumerable: true, get: function () { return with_tambo_interactable_1.withTamboInteractable; } });
+var tambo_interactable_provider_1 = require("./providers/tambo-interactable-provider");
+Object.defineProperty(exports, "useTamboInteractable", { enumerable: true, get: function () { return tambo_interactable_provider_1.useTamboInteractable; } });
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAAA,wKAAwK;AACxK,mEAAqE;AAA5D,6HAAA,sBAAsB,OAAA;AAC/B,mEAIqC;AAHnC,2HAAA,oBAAoB,OAAA;AACpB,6HAAA,sBAAsB,OAAA;AACtB,6HAAA,sBAAsB,OAAA;AAExB,mEAAqE;AAA5D,6HAAA,sBAAsB,OAAA;AAC/B,2EAIyC;AAHvC,+HAAA,oBAAoB,OAAA;AAItB,0DAAwC;AACxC,6DAA+D;AAAtD,uHAAA,mBAAmB,OAAA;AAE5B,gCAAgC;AAChC,yCAYqB;AAXnB,gHAAA,mBAAmB,OAAA;AACnB,mHAAA,sBAAsB,OAAA;AACtB,0GAAA,aAAa,OAAA;AACb,8GAAA,iBAAiB,OAAA;AACjB,gHAAA,mBAAmB,OAAA;AACnB,qGAAA,QAAQ,OAAA;AACR,2GAAA,cAAc,OAAA;AACd,2GAAA,cAAc,OAAA;AAkBhB,+DAA+D;AAAtD,uHAAA,kBAAkB,OAAA;AAQ3B,mFAG6C;AAF3C,8HAAA,eAAe,OAAA","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. */\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 {\n  useTamboStreamStatus,\n  type StreamStatus,\n  type PropStatus,\n} from \"./hooks/use-tambo-stream-status\";\nexport * from \"./hooks/use-suggestions\";\nexport { useTamboThreadInput } from \"./hooks/use-thread-input\";\n\n// Re-export provider components\nexport {\n  TamboClientProvider,\n  TamboComponentProvider,\n  TamboProvider,\n  TamboStubProvider,\n  TamboThreadProvider,\n  useTambo,\n  useTamboClient,\n  useTamboThread,\n  type TamboComponent,\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"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAAA,wKAAwK;AACxK,mEAAqE;AAA5D,6HAAA,sBAAsB,OAAA;AAC/B,mEAIqC;AAHnC,2HAAA,oBAAoB,OAAA;AACpB,6HAAA,sBAAsB,OAAA;AACtB,6HAAA,sBAAsB,OAAA;AAExB,mEAAqE;AAA5D,6HAAA,sBAAsB,OAAA;AAC/B,0DAAwC;AACxC,2EAIyC;AAHvC,+HAAA,oBAAoB,OAAA;AAItB,6DAA+D;AAAtD,uHAAA,mBAAmB,OAAA;AAE5B,gCAAgC;AAChC,yCAYqB;AAXnB,gHAAA,mBAAmB,OAAA;AACnB,mHAAA,sBAAsB,OAAA;AACtB,0GAAA,aAAa,OAAA;AACb,8GAAA,iBAAiB,OAAA;AACjB,gHAAA,mBAAmB,OAAA;AACnB,qGAAA,QAAQ,OAAA;AACR,2GAAA,cAAc,OAAA;AACd,2GAAA,cAAc,OAAA;AAkBhB,+DAA+D;AAAtD,uHAAA,kBAAkB,OAAA;AAQ3B,mFAG6C;AAF3C,8HAAA,eAAe,OAAA;AASjB,mFAIiD;AAH/C,2HAAA,qBAAqB,OAAoB;AAI3C,uFAA+E;AAAtE,mIAAA,oBAAoB,OAAA","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. */\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  TamboProvider,\n  TamboStubProvider,\n  TamboThreadProvider,\n  useTambo,\n  useTamboClient,\n  useTamboThread,\n  type TamboComponent,\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"]}

package/dist/model/tambo-interactable.d.ts

@@ -0,0 +1,24 @@
+import { TamboComponent } from "./component-metadata";
+export interface TamboInteractableComponent extends TamboComponent {
+    /** Unique identifier for this component instance */
+    id: string;
+    /** Current props for the component */
+    props: Record<string, any>;
+}
+export interface TamboInteractableContext {
+    /** List of all interactable components */
+    interactableComponents: TamboInteractableComponent[];
+    /** Add a new interactable component */
+    addInteractableComponent: (component: Omit<TamboInteractableComponent, "id" | "createdAt">) => string;
+    /** Remove an interactable component by ID */
+    removeInteractableComponent: (id: string) => void;
+    /** Update an interactable component's props */
+    updateInteractableComponentProps: (id: string, newProps: Record<string, any>) => void;
+    /** Get an interactable component by ID */
+    getInteractableComponent: (id: string) => TamboInteractableComponent | undefined;
+    /** Get all interactable components by component name */
+    getInteractableComponentsByName: (componentName: string) => TamboInteractableComponent[];
+    /** Clear all interactable components */
+    clearAllInteractableComponents: () => void;
+}
+//# sourceMappingURL=tambo-interactable.d.ts.map

package/dist/model/tambo-interactable.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-interactable.d.ts","sourceRoot":"","sources":["../../src/model/tambo-interactable.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEtD,MAAM,WAAW,0BAA2B,SAAQ,cAAc;IAChE,oDAAoD;IACpD,EAAE,EAAE,MAAM,CAAC;IACX,sCAAsC;IACtC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC5B;AAED,MAAM,WAAW,wBAAwB;IACvC,0CAA0C;IAC1C,sBAAsB,EAAE,0BAA0B,EAAE,CAAC;IACrD,uCAAuC;IACvC,wBAAwB,EAAE,CACxB,SAAS,EAAE,IAAI,CAAC,0BAA0B,EAAE,IAAI,GAAG,WAAW,CAAC,KAC5D,MAAM,CAAC;IACZ,6CAA6C;IAC7C,2BAA2B,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,IAAI,CAAC;IAClD,+CAA+C;IAC/C,gCAAgC,EAAE,CAChC,EAAE,EAAE,MAAM,EACV,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAC1B,IAAI,CAAC;IACV,0CAA0C;IAC1C,wBAAwB,EAAE,CACxB,EAAE,EAAE,MAAM,KACP,0BAA0B,GAAG,SAAS,CAAC;IAC5C,wDAAwD;IACxD,+BAA+B,EAAE,CAC/B,aAAa,EAAE,MAAM,KAClB,0BAA0B,EAAE,CAAC;IAClC,wCAAwC;IACxC,8BAA8B,EAAE,MAAM,IAAI,CAAC;CAC5C"}

package/dist/model/tambo-interactable.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=tambo-interactable.js.map

package/dist/model/tambo-interactable.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-interactable.js","sourceRoot":"","sources":["../../src/model/tambo-interactable.ts"],"names":[],"mappings":"","sourcesContent":["import { TamboComponent } from \"./component-metadata\";\n\nexport interface TamboInteractableComponent extends TamboComponent {\n  /** Unique identifier for this component instance */\n  id: string;\n  /** Current props for the component */\n  props: Record<string, any>;\n}\n\nexport interface TamboInteractableContext {\n  /** List of all interactable components */\n  interactableComponents: TamboInteractableComponent[];\n  /** Add a new interactable component */\n  addInteractableComponent: (\n    component: Omit<TamboInteractableComponent, \"id\" | \"createdAt\">,\n  ) => string;\n  /** Remove an interactable component by ID */\n  removeInteractableComponent: (id: string) => void;\n  /** Update an interactable component's props */\n  updateInteractableComponentProps: (\n    id: string,\n    newProps: Record<string, any>,\n  ) => void;\n  /** Get an interactable component by ID */\n  getInteractableComponent: (\n    id: string,\n  ) => TamboInteractableComponent | undefined;\n  /** Get all interactable components by component name */\n  getInteractableComponentsByName: (\n    componentName: string,\n  ) => TamboInteractableComponent[];\n  /** Clear all interactable components */\n  clearAllInteractableComponents: () => void;\n}\n"]}

package/dist/providers/__tests__/tambo-prop-stream-provider.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=tambo-prop-stream-provider.test.d.ts.map

package/dist/providers/__tests__/tambo-prop-stream-provider.test.d.ts.map

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

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

@@ -0,0 +1,278 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const react_1 = require("@testing-library/react");
+const react_2 = __importDefault(require("react"));
+const tambo_prop_stream_provider_1 = require("../tambo-prop-stream-provider");
+// Helper component to test hook usage
+const TestHookComponent = ({ testKey = "default", }) => {
+    const { data, streamStatus, getStatusForKey } = (0, tambo_prop_stream_provider_1.useTamboStream)();
+    const status = getStatusForKey(testKey);
+    return (react_2.default.createElement("div", null,
+        react_2.default.createElement("div", { "data-testid": "data" }, JSON.stringify(data)),
+        react_2.default.createElement("div", { "data-testid": "stream-status" }, JSON.stringify(streamStatus)),
+        react_2.default.createElement("div", { "data-testid": "key-status" }, JSON.stringify(status))));
+};
+describe("TamboPropStreamProvider", () => {
+    describe("Hook Error Handling", () => {
+        it("should throw error when useTamboStream is used outside provider", () => {
+            // Suppress console.error for this test
+            const originalError = console.error;
+            console.error = jest.fn();
+            expect(() => {
+                (0, react_1.render)(react_2.default.createElement(TestHookComponent, null));
+            }).toThrow("useTamboStream must be used within a TamboPropStreamProvider");
+            console.error = originalError;
+        });
+    });
+    describe("Basic Functionality", () => {
+        it("should provide data and stream status through context", () => {
+            const testData = { message: "Hello World" };
+            const testStreamStatus = {
+                isPending: false,
+                isStreaming: false,
+                isSuccess: true,
+                isError: false,
+                streamError: undefined,
+            };
+            (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: testData, streamStatus: testStreamStatus },
+                react_2.default.createElement(TestHookComponent, null)));
+            expect(react_1.screen.getByTestId("data")).toHaveTextContent(JSON.stringify(testData));
+            expect(react_1.screen.getByTestId("stream-status")).toHaveTextContent(JSON.stringify(testStreamStatus));
+        });
+        it("should use default stream status when none provided", () => {
+            const testData = { message: "Hello World" };
+            const expectedDefaultStatus = {
+                isPending: false,
+                isStreaming: false,
+                isSuccess: true,
+                isError: false,
+                streamError: undefined,
+            };
+            (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: testData },
+                react_2.default.createElement(TestHookComponent, null)));
+            expect(react_1.screen.getByTestId("stream-status")).toHaveTextContent(JSON.stringify(expectedDefaultStatus));
+        });
+    });
+    describe("Compound Components", () => {
+        describe("Loading Component", () => {
+            it("should render loading when isPending is true", () => {
+                const streamStatus = {
+                    isPending: true,
+                    isStreaming: false,
+                    isSuccess: false,
+                    isError: false,
+                };
+                (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: null, streamStatus: streamStatus },
+                    react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider.Loading, null,
+                        react_2.default.createElement("div", { "data-testid": "loading" }, "Loading..."))));
+                expect(react_1.screen.getByTestId("loading")).toBeInTheDocument();
+            });
+            it("should render loading when isStreaming is true", () => {
+                const streamStatus = {
+                    isPending: false,
+                    isStreaming: true,
+                    isSuccess: false,
+                    isError: false,
+                };
+                (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: null, streamStatus: streamStatus },
+                    react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider.Loading, null,
+                        react_2.default.createElement("div", { "data-testid": "loading" }, "Loading..."))));
+                expect(react_1.screen.getByTestId("loading")).toBeInTheDocument();
+            });
+            it("should not render loading when not pending or streaming", () => {
+                const streamStatus = {
+                    isPending: false,
+                    isStreaming: false,
+                    isSuccess: true,
+                    isError: false,
+                };
+                (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: "test", streamStatus: streamStatus },
+                    react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider.Loading, null,
+                        react_2.default.createElement("div", { "data-testid": "loading" }, "Loading..."))));
+                expect(react_1.screen.queryByTestId("loading")).not.toBeInTheDocument();
+            });
+        });
+        describe("Complete Component", () => {
+            it("should render complete when isSuccess is true and data exists", () => {
+                const streamStatus = {
+                    isPending: false,
+                    isStreaming: false,
+                    isSuccess: true,
+                    isError: false,
+                };
+                (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: "test data", streamStatus: streamStatus },
+                    react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider.Complete, null,
+                        react_2.default.createElement("div", { "data-testid": "complete" }, "Complete!"))));
+                expect(react_1.screen.getByTestId("complete")).toBeInTheDocument();
+            });
+            it("should not render complete when data is null", () => {
+                const streamStatus = {
+                    isPending: false,
+                    isStreaming: false,
+                    isSuccess: true,
+                    isError: false,
+                };
+                (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: null, streamStatus: streamStatus },
+                    react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider.Complete, null,
+                        react_2.default.createElement("div", { "data-testid": "complete" }, "Complete!"))));
+                expect(react_1.screen.queryByTestId("complete")).not.toBeInTheDocument();
+            });
+            it("should not render complete when data is undefined", () => {
+                const streamStatus = {
+                    isPending: false,
+                    isStreaming: false,
+                    isSuccess: true,
+                    isError: false,
+                };
+                (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: undefined, streamStatus: streamStatus },
+                    react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider.Complete, null,
+                        react_2.default.createElement("div", { "data-testid": "complete" }, "Complete!"))));
+                expect(react_1.screen.queryByTestId("complete")).not.toBeInTheDocument();
+            });
+        });
+        describe("Empty Component", () => {
+            it("should render empty when no data and not loading/streaming/success/error", () => {
+                const streamStatus = {
+                    isPending: false,
+                    isStreaming: false,
+                    isSuccess: false,
+                    isError: false,
+                };
+                (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: null, streamStatus: streamStatus },
+                    react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider.Empty, null,
+                        react_2.default.createElement("div", { "data-testid": "empty" }, "No data"))));
+                expect(react_1.screen.getByTestId("empty")).toBeInTheDocument();
+            });
+            it("should render empty when data is empty string", () => {
+                const streamStatus = {
+                    isPending: false,
+                    isStreaming: false,
+                    isSuccess: false,
+                    isError: false,
+                };
+                (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: "", streamStatus: streamStatus },
+                    react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider.Empty, null,
+                        react_2.default.createElement("div", { "data-testid": "empty" }, "No data"))));
+                expect(react_1.screen.getByTestId("empty")).toBeInTheDocument();
+            });
+            it("should not render empty when loading", () => {
+                const streamStatus = {
+                    isPending: true,
+                    isStreaming: false,
+                    isSuccess: false,
+                    isError: false,
+                };
+                (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: null, streamStatus: streamStatus },
+                    react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider.Empty, null,
+                        react_2.default.createElement("div", { "data-testid": "empty" }, "No data"))));
+                expect(react_1.screen.queryByTestId("empty")).not.toBeInTheDocument();
+            });
+            it("should not render empty when successful", () => {
+                const streamStatus = {
+                    isPending: false,
+                    isStreaming: false,
+                    isSuccess: true,
+                    isError: false,
+                };
+                (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: null, streamStatus: streamStatus },
+                    react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider.Empty, null,
+                        react_2.default.createElement("div", { "data-testid": "empty" }, "No data"))));
+                expect(react_1.screen.queryByTestId("empty")).not.toBeInTheDocument();
+            });
+        });
+    });
+    describe("Per-Key Status Tracking", () => {
+        it("should track status for object keys", () => {
+            const testData = { name: "John", age: null };
+            const streamStatus = {
+                isPending: false,
+                isStreaming: false,
+                isSuccess: true,
+                isError: false,
+            };
+            (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: testData, streamStatus: streamStatus },
+                react_2.default.createElement(TestHookComponent, { testKey: "name" })));
+            const keyStatus = JSON.parse(react_1.screen.getByTestId("key-status").textContent ?? "{}");
+            expect(keyStatus.isSuccess).toBe(true); // name has data
+        });
+        it("should show loading for keys without data", () => {
+            const testData = { name: "John", age: null };
+            const streamStatus = {
+                isPending: false,
+                isStreaming: false,
+                isSuccess: true,
+                isError: false,
+            };
+            (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: testData, streamStatus: streamStatus },
+                react_2.default.createElement(TestHookComponent, { testKey: "age" })));
+            const keyStatus = JSON.parse(react_1.screen.getByTestId("key-status").textContent ?? "{}");
+            expect(keyStatus.isPending).toBe(true); // age has no data
+        });
+        it("should handle per-key loading states", () => {
+            (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: { name: "John", age: null } },
+                react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider.Loading, { streamKey: "name" },
+                    react_2.default.createElement("div", { "data-testid": "name-loading" }, "Name loading...")),
+                react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider.Loading, { streamKey: "age" },
+                    react_2.default.createElement("div", { "data-testid": "age-loading" }, "Age loading...")),
+                react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider.Complete, { streamKey: "name" },
+                    react_2.default.createElement("div", { "data-testid": "name-complete" }, "Name: John"))));
+            // Name should be complete (has data)
+            expect(react_1.screen.getByTestId("name-complete")).toBeInTheDocument();
+            expect(react_1.screen.queryByTestId("name-loading")).not.toBeInTheDocument();
+            // Age should be loading (no data)
+            expect(react_1.screen.getByTestId("age-loading")).toBeInTheDocument();
+        });
+    });
+    describe("Edge Cases", () => {
+        it("should handle null data gracefully", () => {
+            (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: null },
+                react_2.default.createElement(TestHookComponent, null)));
+            expect(react_1.screen.getByTestId("data")).toHaveTextContent("null");
+        });
+        it("should handle undefined data gracefully", () => {
+            (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: undefined },
+                react_2.default.createElement(TestHookComponent, null)));
+            expect(react_1.screen.getByTestId("data")).toHaveTextContent("");
+        });
+        it("should handle array data", () => {
+            const arrayData = ["item1", "item2"];
+            (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: arrayData },
+                react_2.default.createElement(TestHookComponent, null)));
+            expect(react_1.screen.getByTestId("data")).toHaveTextContent(JSON.stringify(arrayData));
+        });
+        it("should handle primitive data types", () => {
+            (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: "string data" },
+                react_2.default.createElement(TestHookComponent, null)));
+            expect(react_1.screen.getByTestId("data")).toHaveTextContent('"string data"');
+        });
+        it("should fallback to default status for unknown keys", () => {
+            const testData = { name: "John" };
+            (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: testData },
+                react_2.default.createElement(TestHookComponent, { testKey: "unknown-key" })));
+            const keyStatus = JSON.parse(react_1.screen.getByTestId("key-status").textContent ?? "{}");
+            // Should fallback to default status
+            expect(keyStatus.isSuccess).toBe(true);
+        });
+    });
+    describe("Component Attributes", () => {
+        it("should add correct data attributes to components", () => {
+            const streamStatus = {
+                isPending: true,
+                isStreaming: false,
+                isSuccess: false,
+                isError: false,
+            };
+            (0, react_1.render)(react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider, { data: null, streamStatus: streamStatus },
+                react_2.default.createElement(tambo_prop_stream_provider_1.TamboPropStreamProvider.Loading, { className: "loading-class" },
+                    react_2.default.createElement("div", null, "Loading..."))));
+            const loadingElement = react_1.screen.getByText("Loading...").parentElement;
+            expect(loadingElement).toHaveAttribute("data-stream-key", "default");
+            expect(loadingElement).toHaveAttribute("data-stream-state", "loading");
+            expect(loadingElement).toHaveClass("loading-class");
+        });
+    });
+});
+//# sourceMappingURL=tambo-prop-stream-provider.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-prop-stream-provider.test.js","sourceRoot":"","sources":["../../../src/providers/__tests__/tambo-prop-stream-provider.test.tsx"],"names":[],"mappings":";;;;;AAAA,kDAAwD;AACxD,kDAA0B;AAE1B,8EAGuC;AAEvC,sCAAsC;AACtC,MAAM,iBAAiB,GAAmC,CAAC,EACzD,OAAO,GAAG,SAAS,GACpB,EAAE,EAAE;IACH,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,eAAe,EAAE,GAAG,IAAA,2CAAc,GAAE,CAAC;IACjE,MAAM,MAAM,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;IAExC,OAAO,CACL;QACE,sDAAiB,MAAM,IAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAO;QACpD,sDAAiB,eAAe,IAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAO;QACrE,sDAAiB,YAAY,IAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAO,CACxD,CACP,CAAC;AACJ,CAAC,CAAC;AAEF,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;IACvC,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,IAAA,cAAM,EAAC,8BAAC,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,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,MAAM,QAAQ,GAAG,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;YAC5C,MAAM,gBAAgB,GAAiB;gBACrC,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,KAAK;gBAClB,SAAS,EAAE,IAAI;gBACf,OAAO,EAAE,KAAK;gBACd,WAAW,EAAE,SAAS;aACvB,CAAC;YAEF,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IACtB,IAAI,EAAE,QAAQ,EACd,YAAY,EAAE,gBAAgB;gBAE9B,8BAAC,iBAAiB,OAAG,CACG,CAC3B,CAAC;YAEF,MAAM,CAAC,cAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,iBAAiB,CAClD,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,CACzB,CAAC;YACF,MAAM,CAAC,cAAM,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC,CAAC,iBAAiB,CAC3D,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,CACjC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,QAAQ,GAAG,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC;YAC5C,MAAM,qBAAqB,GAAG;gBAC5B,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,KAAK;gBAClB,SAAS,EAAE,IAAI;gBACf,OAAO,EAAE,KAAK;gBACd,WAAW,EAAE,SAAS;aACvB,CAAC;YAEF,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAE,QAAQ;gBACrC,8BAAC,iBAAiB,OAAG,CACG,CAC3B,CAAC;YAEF,MAAM,CAAC,cAAM,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC,CAAC,iBAAiB,CAC3D,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAAC,CACtC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;YACjC,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;gBACtD,MAAM,YAAY,GAAiB;oBACjC,SAAS,EAAE,IAAI;oBACf,WAAW,EAAE,KAAK;oBAClB,SAAS,EAAE,KAAK;oBAChB,OAAO,EAAE,KAAK;iBACf,CAAC;gBAEF,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY;oBAC7D,8BAAC,oDAAuB,CAAC,OAAO;wBAC9B,sDAAiB,SAAS,iBAAiB,CACX,CACV,CAC3B,CAAC;gBAEF,MAAM,CAAC,cAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC5D,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;gBACxD,MAAM,YAAY,GAAiB;oBACjC,SAAS,EAAE,KAAK;oBAChB,WAAW,EAAE,IAAI;oBACjB,SAAS,EAAE,KAAK;oBAChB,OAAO,EAAE,KAAK;iBACf,CAAC;gBAEF,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY;oBAC7D,8BAAC,oDAAuB,CAAC,OAAO;wBAC9B,sDAAiB,SAAS,iBAAiB,CACX,CACV,CAC3B,CAAC;gBAEF,MAAM,CAAC,cAAM,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC5D,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;gBACjE,MAAM,YAAY,GAAiB;oBACjC,SAAS,EAAE,KAAK;oBAChB,WAAW,EAAE,KAAK;oBAClB,SAAS,EAAE,IAAI;oBACf,OAAO,EAAE,KAAK;iBACf,CAAC;gBAEF,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAC,MAAM,EAAC,YAAY,EAAE,YAAY;oBAC7D,8BAAC,oDAAuB,CAAC,OAAO;wBAC9B,sDAAiB,SAAS,iBAAiB,CACX,CACV,CAC3B,CAAC;gBAEF,MAAM,CAAC,cAAM,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;YAClE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;YAClC,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;gBACvE,MAAM,YAAY,GAAiB;oBACjC,SAAS,EAAE,KAAK;oBAChB,WAAW,EAAE,KAAK;oBAClB,SAAS,EAAE,IAAI;oBACf,OAAO,EAAE,KAAK;iBACf,CAAC;gBAEF,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAC,WAAW,EAAC,YAAY,EAAE,YAAY;oBAClE,8BAAC,oDAAuB,CAAC,QAAQ;wBAC/B,sDAAiB,UAAU,gBAAgB,CACV,CACX,CAC3B,CAAC;gBAEF,MAAM,CAAC,cAAM,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC7D,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;gBACtD,MAAM,YAAY,GAAiB;oBACjC,SAAS,EAAE,KAAK;oBAChB,WAAW,EAAE,KAAK;oBAClB,SAAS,EAAE,IAAI;oBACf,OAAO,EAAE,KAAK;iBACf,CAAC;gBAEF,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY;oBAC7D,8BAAC,oDAAuB,CAAC,QAAQ;wBAC/B,sDAAiB,UAAU,gBAAgB,CACV,CACX,CAC3B,CAAC;gBAEF,MAAM,CAAC,cAAM,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;YACnE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;gBAC3D,MAAM,YAAY,GAAiB;oBACjC,SAAS,EAAE,KAAK;oBAChB,WAAW,EAAE,KAAK;oBAClB,SAAS,EAAE,IAAI;oBACf,OAAO,EAAE,KAAK;iBACf,CAAC;gBAEF,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAE,SAAS,EAAE,YAAY,EAAE,YAAY;oBAClE,8BAAC,oDAAuB,CAAC,QAAQ;wBAC/B,sDAAiB,UAAU,gBAAgB,CACV,CACX,CAC3B,CAAC;gBAEF,MAAM,CAAC,cAAM,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;YACnE,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;YAC/B,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;gBAClF,MAAM,YAAY,GAAiB;oBACjC,SAAS,EAAE,KAAK;oBAChB,WAAW,EAAE,KAAK;oBAClB,SAAS,EAAE,KAAK;oBAChB,OAAO,EAAE,KAAK;iBACf,CAAC;gBAEF,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY;oBAC7D,8BAAC,oDAAuB,CAAC,KAAK;wBAC5B,sDAAiB,OAAO,cAAc,CACR,CACR,CAC3B,CAAC;gBAEF,MAAM,CAAC,cAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC1D,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;gBACvD,MAAM,YAAY,GAAiB;oBACjC,SAAS,EAAE,KAAK;oBAChB,WAAW,EAAE,KAAK;oBAClB,SAAS,EAAE,KAAK;oBAChB,OAAO,EAAE,KAAK;iBACf,CAAC;gBAEF,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAC,EAAE,EAAC,YAAY,EAAE,YAAY;oBACzD,8BAAC,oDAAuB,CAAC,KAAK;wBAC5B,sDAAiB,OAAO,cAAc,CACR,CACR,CAC3B,CAAC;gBAEF,MAAM,CAAC,cAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAC1D,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;gBAC9C,MAAM,YAAY,GAAiB;oBACjC,SAAS,EAAE,IAAI;oBACf,WAAW,EAAE,KAAK;oBAClB,SAAS,EAAE,KAAK;oBAChB,OAAO,EAAE,KAAK;iBACf,CAAC;gBAEF,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY;oBAC7D,8BAAC,oDAAuB,CAAC,KAAK;wBAC5B,sDAAiB,OAAO,cAAc,CACR,CACR,CAC3B,CAAC;gBAEF,MAAM,CAAC,cAAM,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;YAChE,CAAC,CAAC,CAAC;YAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;gBACjD,MAAM,YAAY,GAAiB;oBACjC,SAAS,EAAE,KAAK;oBAChB,WAAW,EAAE,KAAK;oBAClB,SAAS,EAAE,IAAI;oBACf,OAAO,EAAE,KAAK;iBACf,CAAC;gBAEF,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY;oBAC7D,8BAAC,oDAAuB,CAAC,KAAK;wBAC5B,sDAAiB,OAAO,cAAc,CACR,CACR,CAC3B,CAAC;gBAEF,MAAM,CAAC,cAAM,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,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,QAAQ,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;YAC7C,MAAM,YAAY,GAAiB;gBACjC,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,KAAK;gBAClB,SAAS,EAAE,IAAI;gBACf,OAAO,EAAE,KAAK;aACf,CAAC;YAEF,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAE,QAAQ,EAAE,YAAY,EAAE,YAAY;gBACjE,8BAAC,iBAAiB,IAAC,OAAO,EAAC,MAAM,GAAG,CACZ,CAC3B,CAAC;YAEF,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAC1B,cAAM,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,WAAW,IAAI,IAAI,CACrD,CAAC;YACF,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,gBAAgB;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,QAAQ,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,CAAC;YAC7C,MAAM,YAAY,GAAiB;gBACjC,SAAS,EAAE,KAAK;gBAChB,WAAW,EAAE,KAAK;gBAClB,SAAS,EAAE,IAAI;gBACf,OAAO,EAAE,KAAK;aACf,CAAC;YAEF,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAE,QAAQ,EAAE,YAAY,EAAE,YAAY;gBACjE,8BAAC,iBAAiB,IAAC,OAAO,EAAC,KAAK,GAAG,CACX,CAC3B,CAAC;YAEF,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAC1B,cAAM,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,WAAW,IAAI,IAAI,CACrD,CAAC;YACF,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,kBAAkB;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE;gBACxD,8BAAC,oDAAuB,CAAC,OAAO,IAAC,SAAS,EAAC,MAAM;oBAC/C,sDAAiB,cAAc,sBAAsB,CACrB;gBAClC,8BAAC,oDAAuB,CAAC,OAAO,IAAC,SAAS,EAAC,KAAK;oBAC9C,sDAAiB,aAAa,qBAAqB,CACnB;gBAClC,8BAAC,oDAAuB,CAAC,QAAQ,IAAC,SAAS,EAAC,MAAM;oBAChD,sDAAiB,eAAe,iBAAiB,CAChB,CACX,CAC3B,CAAC;YAEF,qCAAqC;YACrC,MAAM,CAAC,cAAM,CAAC,WAAW,CAAC,eAAe,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;YAChE,MAAM,CAAC,cAAM,CAAC,aAAa,CAAC,cAAc,CAAC,CAAC,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC;YAErE,kCAAkC;YAClC,MAAM,CAAC,cAAM,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,CAAC,iBAAiB,EAAE,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAE,IAAI;gBACjC,8BAAC,iBAAiB,OAAG,CACG,CAC3B,CAAC;YAEF,MAAM,CAAC,cAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAE,SAAS;gBACtC,8BAAC,iBAAiB,OAAG,CACG,CAC3B,CAAC;YAEF,MAAM,CAAC,cAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAClC,MAAM,SAAS,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACrC,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAE,SAAS;gBACtC,8BAAC,iBAAiB,OAAG,CACG,CAC3B,CAAC;YAEF,MAAM,CAAC,cAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,iBAAiB,CAClD,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAC1B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAC,aAAa;gBACzC,8BAAC,iBAAiB,OAAG,CACG,CAC3B,CAAC;YAEF,MAAM,CAAC,cAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,iBAAiB,CAAC,eAAe,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,QAAQ,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YAElC,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAE,QAAQ;gBACrC,8BAAC,iBAAiB,IAAC,OAAO,EAAC,aAAa,GAAG,CACnB,CAC3B,CAAC;YAEF,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAC1B,cAAM,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,WAAW,IAAI,IAAI,CACrD,CAAC;YACF,oCAAoC;YACpC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,YAAY,GAAiB;gBACjC,SAAS,EAAE,IAAI;gBACf,WAAW,EAAE,KAAK;gBAClB,SAAS,EAAE,KAAK;gBAChB,OAAO,EAAE,KAAK;aACf,CAAC;YAEF,IAAA,cAAM,EACJ,8BAAC,oDAAuB,IAAC,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY;gBAC7D,8BAAC,oDAAuB,CAAC,OAAO,IAAC,SAAS,EAAC,eAAe;oBACxD,wDAAqB,CACW,CACV,CAC3B,CAAC;YAEF,MAAM,cAAc,GAAG,cAAM,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,aAAa,CAAC;YACpE,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,iBAAiB,EAAE,SAAS,CAAC,CAAC;YACrE,MAAM,CAAC,cAAc,CAAC,CAAC,eAAe,CAAC,mBAAmB,EAAE,SAAS,CAAC,CAAC;YACvE,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 { StreamStatus } from \"../../hooks/use-tambo-stream-status\";\nimport {\n  TamboPropStreamProvider,\n  useTamboStream,\n} from \"../tambo-prop-stream-provider\";\n\n// Helper component to test hook usage\nconst TestHookComponent: React.FC<{ testKey?: string }> = ({\n  testKey = \"default\",\n}) => {\n  const { data, streamStatus, getStatusForKey } = useTamboStream();\n  const status = getStatusForKey(testKey);\n\n  return (\n    <div>\n      <div data-testid=\"data\">{JSON.stringify(data)}</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  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(\"Basic Functionality\", () => {\n    it(\"should provide data and stream status through context\", () => {\n      const testData = { message: \"Hello World\" };\n      const testStreamStatus: StreamStatus = {\n        isPending: false,\n        isStreaming: false,\n        isSuccess: true,\n        isError: false,\n        streamError: undefined,\n      };\n\n      render(\n        <TamboPropStreamProvider\n          data={testData}\n          streamStatus={testStreamStatus}\n        >\n          <TestHookComponent />\n        </TamboPropStreamProvider>,\n      );\n\n      expect(screen.getByTestId(\"data\")).toHaveTextContent(\n        JSON.stringify(testData),\n      );\n      expect(screen.getByTestId(\"stream-status\")).toHaveTextContent(\n        JSON.stringify(testStreamStatus),\n      );\n    });\n\n    it(\"should use default stream status when none provided\", () => {\n      const testData = { message: \"Hello World\" };\n      const expectedDefaultStatus = {\n        isPending: false,\n        isStreaming: false,\n        isSuccess: true,\n        isError: false,\n        streamError: undefined,\n      };\n\n      render(\n        <TamboPropStreamProvider data={testData}>\n          <TestHookComponent />\n        </TamboPropStreamProvider>,\n      );\n\n      expect(screen.getByTestId(\"stream-status\")).toHaveTextContent(\n        JSON.stringify(expectedDefaultStatus),\n      );\n    });\n  });\n\n  describe(\"Compound Components\", () => {\n    describe(\"Loading Component\", () => {\n      it(\"should render loading when isPending is true\", () => {\n        const streamStatus: StreamStatus = {\n          isPending: true,\n          isStreaming: false,\n          isSuccess: false,\n          isError: false,\n        };\n\n        render(\n          <TamboPropStreamProvider data={null} streamStatus={streamStatus}>\n            <TamboPropStreamProvider.Loading>\n              <div data-testid=\"loading\">Loading...</div>\n            </TamboPropStreamProvider.Loading>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.getByTestId(\"loading\")).toBeInTheDocument();\n      });\n\n      it(\"should render loading when isStreaming is true\", () => {\n        const streamStatus: StreamStatus = {\n          isPending: false,\n          isStreaming: true,\n          isSuccess: false,\n          isError: false,\n        };\n\n        render(\n          <TamboPropStreamProvider data={null} streamStatus={streamStatus}>\n            <TamboPropStreamProvider.Loading>\n              <div data-testid=\"loading\">Loading...</div>\n            </TamboPropStreamProvider.Loading>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.getByTestId(\"loading\")).toBeInTheDocument();\n      });\n\n      it(\"should not render loading when not pending or streaming\", () => {\n        const streamStatus: StreamStatus = {\n          isPending: false,\n          isStreaming: false,\n          isSuccess: true,\n          isError: false,\n        };\n\n        render(\n          <TamboPropStreamProvider data=\"test\" streamStatus={streamStatus}>\n            <TamboPropStreamProvider.Loading>\n              <div data-testid=\"loading\">Loading...</div>\n            </TamboPropStreamProvider.Loading>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.queryByTestId(\"loading\")).not.toBeInTheDocument();\n      });\n    });\n\n    describe(\"Complete Component\", () => {\n      it(\"should render complete when isSuccess is true and data exists\", () => {\n        const streamStatus: StreamStatus = {\n          isPending: false,\n          isStreaming: false,\n          isSuccess: true,\n          isError: false,\n        };\n\n        render(\n          <TamboPropStreamProvider data=\"test data\" streamStatus={streamStatus}>\n            <TamboPropStreamProvider.Complete>\n              <div data-testid=\"complete\">Complete!</div>\n            </TamboPropStreamProvider.Complete>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.getByTestId(\"complete\")).toBeInTheDocument();\n      });\n\n      it(\"should not render complete when data is null\", () => {\n        const streamStatus: StreamStatus = {\n          isPending: false,\n          isStreaming: false,\n          isSuccess: true,\n          isError: false,\n        };\n\n        render(\n          <TamboPropStreamProvider data={null} streamStatus={streamStatus}>\n            <TamboPropStreamProvider.Complete>\n              <div data-testid=\"complete\">Complete!</div>\n            </TamboPropStreamProvider.Complete>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.queryByTestId(\"complete\")).not.toBeInTheDocument();\n      });\n\n      it(\"should not render complete when data is undefined\", () => {\n        const streamStatus: StreamStatus = {\n          isPending: false,\n          isStreaming: false,\n          isSuccess: true,\n          isError: false,\n        };\n\n        render(\n          <TamboPropStreamProvider data={undefined} streamStatus={streamStatus}>\n            <TamboPropStreamProvider.Complete>\n              <div data-testid=\"complete\">Complete!</div>\n            </TamboPropStreamProvider.Complete>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.queryByTestId(\"complete\")).not.toBeInTheDocument();\n      });\n    });\n\n    describe(\"Empty Component\", () => {\n      it(\"should render empty when no data and not loading/streaming/success/error\", () => {\n        const streamStatus: StreamStatus = {\n          isPending: false,\n          isStreaming: false,\n          isSuccess: false,\n          isError: false,\n        };\n\n        render(\n          <TamboPropStreamProvider data={null} streamStatus={streamStatus}>\n            <TamboPropStreamProvider.Empty>\n              <div data-testid=\"empty\">No data</div>\n            </TamboPropStreamProvider.Empty>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.getByTestId(\"empty\")).toBeInTheDocument();\n      });\n\n      it(\"should render empty when data is empty string\", () => {\n        const streamStatus: StreamStatus = {\n          isPending: false,\n          isStreaming: false,\n          isSuccess: false,\n          isError: false,\n        };\n\n        render(\n          <TamboPropStreamProvider data=\"\" streamStatus={streamStatus}>\n            <TamboPropStreamProvider.Empty>\n              <div data-testid=\"empty\">No data</div>\n            </TamboPropStreamProvider.Empty>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.getByTestId(\"empty\")).toBeInTheDocument();\n      });\n\n      it(\"should not render empty when loading\", () => {\n        const streamStatus: StreamStatus = {\n          isPending: true,\n          isStreaming: false,\n          isSuccess: false,\n          isError: false,\n        };\n\n        render(\n          <TamboPropStreamProvider data={null} streamStatus={streamStatus}>\n            <TamboPropStreamProvider.Empty>\n              <div data-testid=\"empty\">No data</div>\n            </TamboPropStreamProvider.Empty>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.queryByTestId(\"empty\")).not.toBeInTheDocument();\n      });\n\n      it(\"should not render empty when successful\", () => {\n        const streamStatus: StreamStatus = {\n          isPending: false,\n          isStreaming: false,\n          isSuccess: true,\n          isError: false,\n        };\n\n        render(\n          <TamboPropStreamProvider data={null} streamStatus={streamStatus}>\n            <TamboPropStreamProvider.Empty>\n              <div data-testid=\"empty\">No data</div>\n            </TamboPropStreamProvider.Empty>\n          </TamboPropStreamProvider>,\n        );\n\n        expect(screen.queryByTestId(\"empty\")).not.toBeInTheDocument();\n      });\n    });\n  });\n\n  describe(\"Per-Key Status Tracking\", () => {\n    it(\"should track status for object keys\", () => {\n      const testData = { name: \"John\", age: null };\n      const streamStatus: StreamStatus = {\n        isPending: false,\n        isStreaming: false,\n        isSuccess: true,\n        isError: false,\n      };\n\n      render(\n        <TamboPropStreamProvider data={testData} streamStatus={streamStatus}>\n          <TestHookComponent testKey=\"name\" />\n        </TamboPropStreamProvider>,\n      );\n\n      const keyStatus = JSON.parse(\n        screen.getByTestId(\"key-status\").textContent ?? \"{}\",\n      );\n      expect(keyStatus.isSuccess).toBe(true); // name has data\n    });\n\n    it(\"should show loading for keys without data\", () => {\n      const testData = { name: \"John\", age: null };\n      const streamStatus: StreamStatus = {\n        isPending: false,\n        isStreaming: false,\n        isSuccess: true,\n        isError: false,\n      };\n\n      render(\n        <TamboPropStreamProvider data={testData} streamStatus={streamStatus}>\n          <TestHookComponent testKey=\"age\" />\n        </TamboPropStreamProvider>,\n      );\n\n      const keyStatus = JSON.parse(\n        screen.getByTestId(\"key-status\").textContent ?? \"{}\",\n      );\n      expect(keyStatus.isPending).toBe(true); // age has no data\n    });\n\n    it(\"should handle per-key loading states\", () => {\n      render(\n        <TamboPropStreamProvider data={{ name: \"John\", age: null }}>\n          <TamboPropStreamProvider.Loading streamKey=\"name\">\n            <div data-testid=\"name-loading\">Name loading...</div>\n          </TamboPropStreamProvider.Loading>\n          <TamboPropStreamProvider.Loading streamKey=\"age\">\n            <div data-testid=\"age-loading\">Age loading...</div>\n          </TamboPropStreamProvider.Loading>\n          <TamboPropStreamProvider.Complete streamKey=\"name\">\n            <div data-testid=\"name-complete\">Name: John</div>\n          </TamboPropStreamProvider.Complete>\n        </TamboPropStreamProvider>,\n      );\n\n      // Name should be complete (has data)\n      expect(screen.getByTestId(\"name-complete\")).toBeInTheDocument();\n      expect(screen.queryByTestId(\"name-loading\")).not.toBeInTheDocument();\n\n      // Age should be loading (no data)\n      expect(screen.getByTestId(\"age-loading\")).toBeInTheDocument();\n    });\n  });\n\n  describe(\"Edge Cases\", () => {\n    it(\"should handle null data gracefully\", () => {\n      render(\n        <TamboPropStreamProvider data={null}>\n          <TestHookComponent />\n        </TamboPropStreamProvider>,\n      );\n\n      expect(screen.getByTestId(\"data\")).toHaveTextContent(\"null\");\n    });\n\n    it(\"should handle undefined data gracefully\", () => {\n      render(\n        <TamboPropStreamProvider data={undefined}>\n          <TestHookComponent />\n        </TamboPropStreamProvider>,\n      );\n\n      expect(screen.getByTestId(\"data\")).toHaveTextContent(\"\");\n    });\n\n    it(\"should handle array data\", () => {\n      const arrayData = [\"item1\", \"item2\"];\n      render(\n        <TamboPropStreamProvider data={arrayData}>\n          <TestHookComponent />\n        </TamboPropStreamProvider>,\n      );\n\n      expect(screen.getByTestId(\"data\")).toHaveTextContent(\n        JSON.stringify(arrayData),\n      );\n    });\n\n    it(\"should handle primitive data types\", () => {\n      render(\n        <TamboPropStreamProvider data=\"string data\">\n          <TestHookComponent />\n        </TamboPropStreamProvider>,\n      );\n\n      expect(screen.getByTestId(\"data\")).toHaveTextContent('\"string data\"');\n    });\n\n    it(\"should fallback to default status for unknown keys\", () => {\n      const testData = { name: \"John\" };\n\n      render(\n        <TamboPropStreamProvider data={testData}>\n          <TestHookComponent testKey=\"unknown-key\" />\n        </TamboPropStreamProvider>,\n      );\n\n      const keyStatus = JSON.parse(\n        screen.getByTestId(\"key-status\").textContent ?? \"{}\",\n      );\n      // Should fallback to default status\n      expect(keyStatus.isSuccess).toBe(true);\n    });\n  });\n\n  describe(\"Component Attributes\", () => {\n    it(\"should add correct data attributes to components\", () => {\n      const streamStatus: StreamStatus = {\n        isPending: true,\n        isStreaming: false,\n        isSuccess: false,\n        isError: false,\n      };\n\n      render(\n        <TamboPropStreamProvider data={null} streamStatus={streamStatus}>\n          <TamboPropStreamProvider.Loading className=\"loading-class\">\n            <div>Loading...</div>\n          </TamboPropStreamProvider.Loading>\n        </TamboPropStreamProvider>,\n      );\n\n      const loadingElement = screen.getByText(\"Loading...\").parentElement;\n      expect(loadingElement).toHaveAttribute(\"data-stream-key\", \"default\");\n      expect(loadingElement).toHaveAttribute(\"data-stream-state\", \"loading\");\n      expect(loadingElement).toHaveClass(\"loading-class\");\n    });\n  });\n});\n"]}

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

@@ -204,6 +204,7 @@ describe("TamboThreadProvider", () => {
             availableComponents: (0, tools_1.serializeRegistry)(mockRegistry),
             contextKey: undefined,
             clientTools: [],
+            toolCallCounts: {},
         });
         expect(result.current.generationStage).toBe(generate_component_response_1.GenerationStage.COMPLETE);
     });
@@ -329,6 +330,7 @@ describe("TamboThreadProvider", () => {
                 contextKey: undefined,
                 clientTools: [],
                 forceToolChoice: undefined,
+                toolCallCounts: {},
             }, "test-thread-1");
             // Should not call advance or advanceById
             expect(mockThreadsApi.advance).not.toHaveBeenCalled();
@@ -356,6 +358,7 @@ describe("TamboThreadProvider", () => {
                 contextKey: undefined,
                 clientTools: [],
                 forceToolChoice: undefined,
+                toolCallCounts: {},
             });
             // Should not call advance or advanceStream
             expect(mockThreadsApi.advance).not.toHaveBeenCalled();
@@ -383,6 +386,7 @@ describe("TamboThreadProvider", () => {
                 contextKey: undefined,
                 clientTools: [],
                 forceToolChoice: undefined,
+                toolCallCounts: {},
             });
             // Should not call advance or advanceStream
             expect(mockThreadsApi.advance).not.toHaveBeenCalled();
@@ -429,6 +433,7 @@ describe("TamboThreadProvider", () => {
                 contextKey: undefined,
                 clientTools: [],
                 forceToolChoice: undefined,
+                toolCallCounts: {},
             }, "test-thread-1");
             // Should not call advance or advanceById
             expect(mockThreadsApi.advance).not.toHaveBeenCalled();
@@ -458,6 +463,7 @@ describe("TamboThreadProvider", () => {
                 contextKey: undefined,
                 clientTools: [],
                 forceToolChoice: undefined,
+                toolCallCounts: {},
             });
             // Should not call advanceById or advanceStream
             expect(mockThreadsApi.advanceById).not.toHaveBeenCalled();
@@ -506,6 +512,7 @@ describe("TamboThreadProvider", () => {
                 contextKey: undefined,
                 clientTools: [],
                 forceToolChoice: undefined,
+                toolCallCounts: {},
             }, undefined);
             // Should not call advance or advanceById
             expect(mockThreadsApi.advance).not.toHaveBeenCalled();