@tambo-ai/react 0.20.2 → 0.20.3

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

@@ -1,3 +1,5 @@
+import type TamboAI from "@tambo-ai/typescript-sdk";
+import { UseQueryOptions } from "@tanstack/react-query";
 interface UseTamboThreadListConfig {
     /**
      * The projectId to get the threads for. If not provided, the current project
@@ -33,7 +35,7 @@ interface UseTamboThreadListConfig {
  * @param config.contextKey - The context key to get the threads for
  * @returns The threads for the specified project and optional context key
  */
-export declare function useTamboThreadList({ projectId, contextKey, }?: UseTamboThreadListConfig): import("@tanstack/query-core").QueryObserverRefetchErrorResult<import("@tambo-ai/typescript-sdk/resources/beta").ThreadsOffsetAndLimit | null, Error> | import("@tanstack/query-core").QueryObserverSuccessResult<import("@tambo-ai/typescript-sdk/resources/beta").ThreadsOffsetAndLimit | null, Error> | import("@tanstack/query-core").QueryObserverLoadingErrorResult<import("@tambo-ai/typescript-sdk/resources/beta").ThreadsOffsetAndLimit | null, Error> | import("@tanstack/query-core").QueryObserverPendingResult<import("@tambo-ai/typescript-sdk/resources/beta").ThreadsOffsetAndLimit | null, Error> | import("@tanstack/query-core").QueryObserverPlaceholderResult<import("@tambo-ai/typescript-sdk/resources/beta").ThreadsOffsetAndLimit | null, Error> | {
+export declare function useTamboThreadList({ projectId, contextKey }?: UseTamboThreadListConfig, options?: Partial<UseQueryOptions<TamboAI.Beta.Threads.ThreadsOffsetAndLimit | null>>): import("@tanstack/react-query").QueryObserverRefetchErrorResult<TamboAI.Beta.Threads.ThreadsOffsetAndLimit | null, Error> | import("@tanstack/react-query").QueryObserverSuccessResult<TamboAI.Beta.Threads.ThreadsOffsetAndLimit | null, Error> | import("@tanstack/react-query").QueryObserverLoadingErrorResult<TamboAI.Beta.Threads.ThreadsOffsetAndLimit | null, Error> | import("@tanstack/react-query").QueryObserverPendingResult<TamboAI.Beta.Threads.ThreadsOffsetAndLimit | null, Error> | import("@tanstack/react-query").QueryObserverPlaceholderResult<TamboAI.Beta.Threads.ThreadsOffsetAndLimit | null, Error> | {
     error: Error;
     isError: true;
     isPending: false;
@@ -55,8 +57,8 @@ export declare function useTamboThreadList({ projectId, contextKey, }?: UseTambo
     isPaused: boolean;
     isRefetching: boolean;
     isStale: boolean;
-    refetch: (options?: import("@tanstack/query-core").RefetchOptions) => Promise<import("@tanstack/query-core").QueryObserverResult<string, Error>>;
-    fetchStatus: import("@tanstack/query-core").FetchStatus;
+    refetch: (options?: import("@tanstack/react-query").RefetchOptions) => Promise<import("@tanstack/react-query").QueryObserverResult<string, Error>>;
+    fetchStatus: import("@tanstack/react-query").FetchStatus;
     promise: Promise<string>;
     data: null;
 } | {
@@ -81,8 +83,8 @@ export declare function useTamboThreadList({ projectId, contextKey, }?: UseTambo
     isPaused: boolean;
     isRefetching: boolean;
     isStale: boolean;
-    refetch: (options?: import("@tanstack/query-core").RefetchOptions) => Promise<import("@tanstack/query-core").QueryObserverResult<string, Error>>;
-    fetchStatus: import("@tanstack/query-core").FetchStatus;
+    refetch: (options?: import("@tanstack/react-query").RefetchOptions) => Promise<import("@tanstack/react-query").QueryObserverResult<string, Error>>;
+    fetchStatus: import("@tanstack/react-query").FetchStatus;
     promise: Promise<string>;
     data: null;
 } | {
@@ -107,8 +109,8 @@ export declare function useTamboThreadList({ projectId, contextKey, }?: UseTambo
     isPaused: boolean;
     isRefetching: boolean;
     isStale: boolean;
-    refetch: (options?: import("@tanstack/query-core").RefetchOptions) => Promise<import("@tanstack/query-core").QueryObserverResult<string, Error>>;
-    fetchStatus: import("@tanstack/query-core").FetchStatus;
+    refetch: (options?: import("@tanstack/react-query").RefetchOptions) => Promise<import("@tanstack/react-query").QueryObserverResult<string, Error>>;
+    fetchStatus: import("@tanstack/react-query").FetchStatus;
     promise: Promise<string>;
     data: null;
 } | {
@@ -133,8 +135,8 @@ export declare function useTamboThreadList({ projectId, contextKey, }?: UseTambo
     isPaused: boolean;
     isRefetching: boolean;
     isStale: boolean;
-    refetch: (options?: import("@tanstack/query-core").RefetchOptions) => Promise<import("@tanstack/query-core").QueryObserverResult<string, Error>>;
-    fetchStatus: import("@tanstack/query-core").FetchStatus;
+    refetch: (options?: import("@tanstack/react-query").RefetchOptions) => Promise<import("@tanstack/react-query").QueryObserverResult<string, Error>>;
+    fetchStatus: import("@tanstack/react-query").FetchStatus;
     promise: Promise<string>;
     data: null;
 } | {
@@ -159,8 +161,8 @@ export declare function useTamboThreadList({ projectId, contextKey, }?: UseTambo
     isPaused: boolean;
     isRefetching: boolean;
     isStale: boolean;
-    refetch: (options?: import("@tanstack/query-core").RefetchOptions) => Promise<import("@tanstack/query-core").QueryObserverResult<string, Error>>;
-    fetchStatus: import("@tanstack/query-core").FetchStatus;
+    refetch: (options?: import("@tanstack/react-query").RefetchOptions) => Promise<import("@tanstack/react-query").QueryObserverResult<string, Error>>;
+    fetchStatus: import("@tanstack/react-query").FetchStatus;
     promise: Promise<string>;
     data: null;
 };

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

@@ -1 +1 @@
-{"version":3,"file":"use-tambo-threads.d.ts","sourceRoot":"","sources":["../../src/hooks/use-tambo-threads.ts"],"names":[],"mappings":"AAGA,UAAU,wBAAwB;IAChC;;;OAGG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;;OAGG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,UAAU,wBAAwB;IAChC;;;OAGG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;;OAGG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,kBAAkB,CAAC,EACjC,SAAS,EACT,UAAU,GACX,GAAE,wBAA6B;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAyB/B"}
+{"version":3,"file":"use-tambo-threads.d.ts","sourceRoot":"","sources":["../../src/hooks/use-tambo-threads.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,OAAO,MAAM,0BAA0B,CAAC;AACpD,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAIxD,UAAU,wBAAwB;IAChC;;;OAGG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;;OAGG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,UAAU,wBAAwB;IAChC;;;OAGG;IACH,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;;OAGG;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,kBAAkB,CAChC,EAAE,SAAS,EAAE,UAAU,EAAE,GAAE,wBAA6B,EACxD,OAAO,GAAE,OAAO,CACd,eAAe,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,qBAAqB,GAAG,IAAI,CAAC,CAC9D;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA4BP"}

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

@@ -14,9 +14,10 @@ const react_query_hooks_1 = require("./react-query-hooks");
  * @param config.contextKey - The context key to get the threads for
  * @returns The threads for the specified project and optional context key
  */
-function useTamboThreadList({ projectId, contextKey, } = {}) {
+function useTamboThreadList({ projectId, contextKey } = {}, options = {}) {
     const client = (0, tambo_client_provider_1.useTamboClient)();
     const { data: queriedProjectId, ...projectIdState } = (0, react_query_hooks_1.useTamboQuery)({
+        ...options,
         queryKey: ["projectId"],
         queryFn: async () => {
             return (await client.beta.projects.getCurrent()).id;
@@ -24,6 +25,7 @@ function useTamboThreadList({ projectId, contextKey, } = {}) {
     });
     const currentProjectId = projectId ?? queriedProjectId;
     const threadState = (0, react_query_hooks_1.useTamboQuery)({
+        ...options,
         enabled: !!currentProjectId,
         queryKey: ["threads", currentProjectId, contextKey],
         queryFn: async () => {

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

@@ -1 +1 @@
-{"version":3,"file":"use-tambo-threads.js","sourceRoot":"","sources":["../../src/hooks/use-tambo-threads.ts"],"names":[],"mappings":";;AAwCA,gDA4BC;AApED,8EAAoE;AACpE,2DAAoD;AA4BpD;;;;;;;;;;GAUG;AACH,SAAgB,kBAAkB,CAAC,EACjC,SAAS,EACT,UAAU,MACkB,EAAE;IAC9B,MAAM,MAAM,GAAG,IAAA,sCAAc,GAAE,CAAC;IAChC,MAAM,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,cAAc,EAAE,GAAG,IAAA,iCAAa,EAAC;QAClE,QAAQ,EAAE,CAAC,WAAW,CAAC;QACvB,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,OAAO,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,EAAE,CAAC;QACtD,CAAC;KACF,CAAC,CAAC;IACH,MAAM,gBAAgB,GAAG,SAAS,IAAI,gBAAgB,CAAC;IAEvD,MAAM,WAAW,GAAG,IAAA,iCAAa,EAAC;QAChC,OAAO,EAAE,CAAC,CAAC,gBAAgB;QAC3B,QAAQ,EAAE,CAAC,SAAS,EAAE,gBAAgB,EAAE,UAAU,CAAC;QACnD,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACtB,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAAE;gBAClE,UAAU;aACX,CAAC,CAAC;YACH,OAAO,UAAU,CAAC;QACpB,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,gBAAgB,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,cAAc,EAAE,CAAC;AAC5E,CAAC","sourcesContent":["import { useTamboClient } from \"../providers/tambo-client-provider\";\nimport { useTamboQuery } from \"./react-query-hooks\";\n\ninterface UseTamboThreadListConfig {\n  /**\n   * The projectId to get the threads for. If not provided, the current project\n   * will be used.\n   */\n  projectId?: string;\n  /**\n   * The context key to get the threads for. If not provided, all threads for\n   * the project will be returned.\n   */\n  contextKey?: string;\n}\n\ninterface UseTamboThreadListConfig {\n  /**\n   * The projectId to get the threads for. If not provided, the current project\n   * will be used.\n   */\n  projectId?: string;\n  /**\n   * The context key to get the threads for. If not provided, all threads for\n   * the project will be returned.\n   */\n  contextKey?: string;\n}\n\n/**\n * Get all the threads for the specified project.\n *\n * If contextKey is empty, then all threads for the project will be returned.\n * If contextKey is not empty, then only the threads for the specified context\n * key will be returned.\n * @param config - The config for the useTamboThreadList hook\n * @param config.projectId - The projectId to get the threads for\n * @param config.contextKey - The context key to get the threads for\n * @returns The threads for the specified project and optional context key\n */\nexport function useTamboThreadList({\n  projectId,\n  contextKey,\n}: UseTamboThreadListConfig = {}) {\n  const client = useTamboClient();\n  const { data: queriedProjectId, ...projectIdState } = useTamboQuery({\n    queryKey: [\"projectId\"],\n    queryFn: async () => {\n      return (await client.beta.projects.getCurrent()).id;\n    },\n  });\n  const currentProjectId = projectId ?? queriedProjectId;\n\n  const threadState = useTamboQuery({\n    enabled: !!currentProjectId,\n    queryKey: [\"threads\", currentProjectId, contextKey],\n    queryFn: async () => {\n      if (!currentProjectId) {\n        return null;\n      }\n      const threadIter = await client.beta.threads.list(currentProjectId, {\n        contextKey,\n      });\n      return threadIter;\n    },\n  });\n\n  return currentProjectId ? threadState : { data: null, ...projectIdState };\n}\n"]}
+{"version":3,"file":"use-tambo-threads.js","sourceRoot":"","sources":["../../src/hooks/use-tambo-threads.ts"],"names":[],"mappings":";;AA0CA,gDAgCC;AAxED,8EAAoE;AACpE,2DAAoD;AA4BpD;;;;;;;;;;GAUG;AACH,SAAgB,kBAAkB,CAChC,EAAE,SAAS,EAAE,UAAU,KAA+B,EAAE,EACxD,UAEI,EAAE;IAEN,MAAM,MAAM,GAAG,IAAA,sCAAc,GAAE,CAAC;IAChC,MAAM,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,cAAc,EAAE,GAAG,IAAA,iCAAa,EAAC;QAClE,GAAI,OAA8C;QAClD,QAAQ,EAAE,CAAC,WAAW,CAAC;QACvB,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,OAAO,CAAC,MAAM,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,EAAE,CAAC;QACtD,CAAC;KACF,CAAC,CAAC;IACH,MAAM,gBAAgB,GAAG,SAAS,IAAI,gBAAgB,CAAC;IAEvD,MAAM,WAAW,GAAG,IAAA,iCAAa,EAAC;QAChC,GAAG,OAAO;QACV,OAAO,EAAE,CAAC,CAAC,gBAAgB;QAC3B,QAAQ,EAAE,CAAC,SAAS,EAAE,gBAAgB,EAAE,UAAU,CAAC;QACnD,OAAO,EAAE,KAAK,IAAI,EAAE;YAClB,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACtB,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,gBAAgB,EAAE;gBAClE,UAAU;aACX,CAAC,CAAC;YACH,OAAO,UAAU,CAAC;QACpB,CAAC;KACF,CAAC,CAAC;IAEH,OAAO,gBAAgB,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,cAAc,EAAE,CAAC;AAC5E,CAAC","sourcesContent":["import type TamboAI from \"@tambo-ai/typescript-sdk\";\nimport { UseQueryOptions } from \"@tanstack/react-query\";\nimport { useTamboClient } from \"../providers/tambo-client-provider\";\nimport { useTamboQuery } from \"./react-query-hooks\";\n\ninterface UseTamboThreadListConfig {\n  /**\n   * The projectId to get the threads for. If not provided, the current project\n   * will be used.\n   */\n  projectId?: string;\n  /**\n   * The context key to get the threads for. If not provided, all threads for\n   * the project will be returned.\n   */\n  contextKey?: string;\n}\n\ninterface UseTamboThreadListConfig {\n  /**\n   * The projectId to get the threads for. If not provided, the current project\n   * will be used.\n   */\n  projectId?: string;\n  /**\n   * The context key to get the threads for. If not provided, all threads for\n   * the project will be returned.\n   */\n  contextKey?: string;\n}\n\n/**\n * Get all the threads for the specified project.\n *\n * If contextKey is empty, then all threads for the project will be returned.\n * If contextKey is not empty, then only the threads for the specified context\n * key will be returned.\n * @param config - The config for the useTamboThreadList hook\n * @param config.projectId - The projectId to get the threads for\n * @param config.contextKey - The context key to get the threads for\n * @returns The threads for the specified project and optional context key\n */\nexport function useTamboThreadList(\n  { projectId, contextKey }: UseTamboThreadListConfig = {},\n  options: Partial<\n    UseQueryOptions<TamboAI.Beta.Threads.ThreadsOffsetAndLimit | null>\n  > = {},\n) {\n  const client = useTamboClient();\n  const { data: queriedProjectId, ...projectIdState } = useTamboQuery({\n    ...(options as unknown as UseQueryOptions<string>),\n    queryKey: [\"projectId\"],\n    queryFn: async () => {\n      return (await client.beta.projects.getCurrent()).id;\n    },\n  });\n  const currentProjectId = projectId ?? queriedProjectId;\n\n  const threadState = useTamboQuery({\n    ...options,\n    enabled: !!currentProjectId,\n    queryKey: [\"threads\", currentProjectId, contextKey],\n    queryFn: async () => {\n      if (!currentProjectId) {\n        return null;\n      }\n      const threadIter = await client.beta.threads.list(currentProjectId, {\n        contextKey,\n      });\n      return threadIter;\n    },\n  });\n\n  return currentProjectId ? threadState : { data: null, ...projectIdState };\n}\n"]}

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

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

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

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

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

@@ -0,0 +1,279 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const typescript_sdk_1 = require("@tambo-ai/typescript-sdk");
+const react_1 = require("@testing-library/react");
+const react_2 = __importDefault(require("react"));
+const zod_1 = require("zod");
+const generate_component_response_1 = require("../../model/generate-component-response");
+const tools_1 = require("../../testing/tools");
+const tambo_client_provider_1 = require("../tambo-client-provider");
+const tambo_registry_provider_1 = require("../tambo-registry-provider");
+const tambo_thread_provider_1 = require("../tambo-thread-provider");
+// Mock crypto.randomUUID
+Object.defineProperty(global, "crypto", {
+    value: {
+        randomUUID: jest.fn().mockReturnValue("test-uuid"),
+    },
+});
+// Mock the required providers
+jest.mock("../tambo-client-provider", () => ({
+    useTamboClient: jest.fn(),
+}));
+jest.mock("@tambo-ai/typescript-sdk", () => ({
+    advanceStream: jest.fn(),
+}));
+// Test utilities
+const createMockMessage = (overrides = {}) => ({
+    id: "test-message-1",
+    content: [{ type: "text", text: "Hello" }],
+    role: "user",
+    threadId: "test-thread-1",
+    createdAt: new Date().toISOString(),
+    componentState: {},
+    ...overrides,
+});
+const createMockThread = (overrides = {}) => ({
+    id: "test-thread-1",
+    messages: [],
+    createdAt: "2024-01-01T00:00:00Z",
+    projectId: "test-project",
+    updatedAt: "2024-01-01T00:00:00Z",
+    metadata: {},
+    ...overrides,
+});
+const createMockAdvanceResponse = (overrides = {}) => ({
+    responseMessageDto: {
+        id: "test-uuid",
+        content: [{ type: "text", text: "Default response" }],
+        role: "assistant",
+        threadId: "test-thread-1",
+        component: undefined,
+        componentState: {},
+        createdAt: new Date().toISOString(),
+    },
+    generationStage: generate_component_response_1.GenerationStage.COMPLETE,
+    ...overrides,
+});
+describe("TamboThreadProvider", () => {
+    const mockThread = createMockThread();
+    const mockThreadsApi = {
+        messages: {
+            create: jest.fn(),
+        },
+        retrieve: jest.fn(),
+        advanceById: jest.fn(),
+    };
+    const mockBeta = {
+        threads: mockThreadsApi,
+    };
+    const mockTamboAI = {
+        apiKey: "",
+        components: {},
+        beta: mockBeta,
+    };
+    const mockRegistry = [
+        {
+            name: "TestOnly",
+            component: () => react_2.default.createElement("div", null, "TestOnly"),
+            description: "TestOnly",
+            propsSchema: zod_1.z.object({
+                test: zod_1.z.string(),
+            }),
+            associatedTools: [
+                {
+                    name: "test-tool",
+                    tool: jest.fn().mockResolvedValue("test-tool"),
+                    description: "test-tool",
+                    toolSchema: zod_1.z
+                        .function()
+                        .args(zod_1.z.string().describe("test-param-description"))
+                        .returns(zod_1.z.string()),
+                },
+            ],
+        },
+    ];
+    const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
+        react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, null, children)));
+    beforeEach(() => {
+        jest.mocked(mockThreadsApi.retrieve).mockResolvedValue(mockThread);
+        jest
+            .mocked(mockThreadsApi.messages.create)
+            .mockResolvedValue(createMockMessage());
+        jest
+            .mocked(mockThreadsApi.advanceById)
+            .mockResolvedValue(createMockAdvanceResponse());
+        jest.mocked(tambo_client_provider_1.useTamboClient).mockReturnValue(mockTamboAI);
+    });
+    it("should initialize with placeholder thread", () => {
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        expect(result.current.thread.id).toBe("placeholder");
+        expect(result.current.isIdle).toBe(true);
+        expect(result.current.generationStage).toBe(generate_component_response_1.GenerationStage.IDLE);
+    });
+    it("should switch to a new thread", async () => {
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        await (0, react_1.act)(async () => {
+            await result.current.switchCurrentThread("test-thread-1");
+        });
+        expect(mockThreadsApi.retrieve).toHaveBeenCalledWith("test-thread-1");
+        expect(result.current.thread.id).toBe("test-thread-1");
+    });
+    it("should start a new thread", async () => {
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        await (0, react_1.act)(async () => {
+            result.current.startNewThread();
+        });
+        expect(result.current.thread.id).toBe("placeholder");
+        expect(result.current.isIdle).toBe(true);
+    });
+    it("should add a message to the thread", async () => {
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        const testMessage = {
+            id: "test-message-1",
+            content: [{ type: "text", text: "Hello" }],
+            role: "user",
+            threadId: "test-thread-1",
+            createdAt: new Date().toISOString(),
+            componentState: {},
+        };
+        await (0, react_1.act)(async () => {
+            await result.current.addThreadMessage(testMessage, true);
+        });
+        expect(mockThreadsApi.messages.create).toHaveBeenCalledWith("test-thread-1", {
+            content: testMessage.content,
+            role: testMessage.role,
+        });
+    });
+    it("should update a message in the thread", async () => {
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        const testMessage = {
+            id: "test-message-1",
+            content: [{ type: "text", text: "Updated message" }],
+            role: "user",
+            threadId: "test-thread-1",
+            createdAt: new Date().toISOString(),
+            componentState: {},
+        };
+        await (0, react_1.act)(async () => {
+            await result.current.updateThreadMessage("test-message-1", testMessage, true);
+        });
+        expect(mockThreadsApi.messages.create).toHaveBeenCalledWith("test-thread-1", {
+            content: testMessage.content,
+            role: testMessage.role,
+        });
+    });
+    it("should send a message and update thread state", async () => {
+        const mockAdvanceResponse = {
+            responseMessageDto: {
+                id: "response-1",
+                content: [{ type: "text", text: "Response" }],
+                role: "assistant",
+                threadId: "test-thread-1",
+                component: undefined,
+                componentState: {},
+                createdAt: new Date().toISOString(),
+            },
+            generationStage: generate_component_response_1.GenerationStage.COMPLETE,
+        };
+        jest
+            .mocked(mockThreadsApi.advanceById)
+            .mockResolvedValue(mockAdvanceResponse);
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        await (0, react_1.act)(async () => {
+            await result.current.sendThreadMessage("Hello", {
+                threadId: "test-thread-1",
+                streamResponse: false,
+            });
+        });
+        expect(mockThreadsApi.advanceById).toHaveBeenCalledWith("test-thread-1", {
+            messageToAppend: {
+                content: [{ type: "text", text: "Hello" }],
+                role: "user",
+            },
+            availableComponents: (0, tools_1.serializeRegistry)(mockRegistry),
+            contextKey: undefined,
+        });
+        expect(result.current.generationStage).toBe(generate_component_response_1.GenerationStage.COMPLETE);
+    });
+    it("should handle input value changes", () => {
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        (0, react_1.act)(() => {
+            result.current.setInputValue("New input");
+        });
+        expect(result.current.inputValue).toBe("New input");
+    });
+    it("should handle streaming responses", async () => {
+        const mockStreamResponse = {
+            responseMessageDto: {
+                id: "stream-1",
+                content: [{ type: "text", text: "Streaming response" }],
+                role: "assistant",
+                threadId: "test-thread-1",
+                component: undefined,
+                componentState: {},
+                createdAt: new Date().toISOString(),
+            },
+            generationStage: generate_component_response_1.GenerationStage.COMPLETE,
+        };
+        // Create an async iterator mock
+        const mockAsyncIterator = {
+            [Symbol.asyncIterator]: async function* () {
+                yield mockStreamResponse;
+            },
+        };
+        // Mock advanceStream to return our async iterator
+        jest.mocked(typescript_sdk_1.advanceStream).mockResolvedValue(mockAsyncIterator);
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        await (0, react_1.act)(async () => {
+            await result.current.sendThreadMessage("Hello", {
+                threadId: "test-thread-1",
+                streamResponse: true,
+            });
+        });
+        expect(result.current.generationStage).toBe(generate_component_response_1.GenerationStage.IDLE);
+    });
+    it("should handle tool calls during message processing", async () => {
+        const mockToolCallResponse = {
+            responseMessageDto: {
+                id: "tool-call-1",
+                content: [{ type: "text", text: "Tool response" }],
+                role: "tool",
+                threadId: "test-thread-1",
+                toolCallRequest: {
+                    toolName: "test-tool",
+                    parameters: [{ parameterName: "test", parameterValue: "test" }],
+                },
+                componentState: {},
+                createdAt: new Date().toISOString(),
+            },
+            generationStage: generate_component_response_1.GenerationStage.COMPLETE,
+        };
+        jest
+            .mocked(mockThreadsApi.advanceById)
+            .mockResolvedValueOnce(mockToolCallResponse)
+            .mockResolvedValueOnce({
+            responseMessageDto: {
+                id: "advance-response2",
+                content: [{ type: "text", text: "response 2" }],
+                role: "user",
+                threadId: "test-thread-1",
+                componentState: {},
+                createdAt: new Date().toISOString(),
+            },
+            generationStage: generate_component_response_1.GenerationStage.COMPLETE,
+        });
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        await (0, react_1.act)(async () => {
+            await result.current.sendThreadMessage("Use tool", {
+                threadId: "test-thread-1",
+                streamResponse: false,
+            });
+        });
+        expect(result.current.generationStage).toBe(generate_component_response_1.GenerationStage.COMPLETE);
+        expect(mockRegistry[0]?.associatedTools?.[0]?.tool).toHaveBeenCalledWith("test");
+    });
+});
+//# sourceMappingURL=tambo-thread-provider.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-thread-provider.test.js","sourceRoot":"","sources":["../../../src/providers/__tests__/tambo-thread-provider.test.tsx"],"names":[],"mappings":";;;;;AAAA,6DAAkE;AAClE,kDAAyD;AACzD,kDAA0B;AAE1B,6BAAwB;AAExB,yFAGiD;AACjD,+CAAwD;AACxD,oEAA0D;AAC1D,wEAAmE;AACnE,oEAA+E;AAI/E,yBAAyB;AACzB,MAAM,CAAC,cAAc,CAAC,MAAM,EAAE,QAAQ,EAAE;IACtC,KAAK,EAAE;QACL,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,WAAW,CAAC;KACnD;CACF,CAAC,CAAC;AAEH,8BAA8B;AAC9B,IAAI,CAAC,IAAI,CAAC,0BAA0B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC3C,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE;CAC1B,CAAC,CAAC,CAAC;AACJ,IAAI,CAAC,IAAI,CAAC,0BAA0B,EAAE,GAAG,EAAE,CAAC,CAAC;IAC3C,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE;CACzB,CAAC,CAAC,CAAC;AAEJ,iBAAiB;AACjB,MAAM,iBAAiB,GAAG,CACxB,YAAyC,EAAE,EACvB,EAAE,CAAC,CAAC;IACxB,EAAE,EAAE,gBAAgB;IACpB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;IAC1C,IAAI,EAAE,MAAM;IACZ,QAAQ,EAAE,eAAe;IACzB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;IACnC,cAAc,EAAE,EAAE;IAClB,GAAG,SAAS;CACb,CAAC,CAAC;AAEH,MAAM,gBAAgB,GAAG,CACvB,YAAkD,EAAE,EACpD,EAAE,CAAC,CAAC;IACJ,EAAE,EAAE,eAAe;IACnB,QAAQ,EAAE,EAAE;IACZ,SAAS,EAAE,sBAAsB;IACjC,SAAS,EAAE,cAAc;IACzB,SAAS,EAAE,sBAAsB;IACjC,QAAQ,EAAE,EAAE;IACZ,GAAG,SAAS;CACb,CAAC,CAAC;AAEH,MAAM,yBAAyB,GAAG,CAChC,YAAiE,EAAE,EACvB,EAAE,CAAC,CAAC;IAChD,kBAAkB,EAAE;QAClB,EAAE,EAAE,WAAW;QACf,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC;QAC9D,IAAI,EAAE,WAAW;QACjB,QAAQ,EAAE,eAAe;QACzB,SAAS,EAAE,SAAS;QACpB,cAAc,EAAE,EAAE;QAClB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACpC;IACD,eAAe,EAAE,6CAAe,CAAC,QAAQ;IACzC,GAAG,SAAS;CACb,CAAC,CAAC;AAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,MAAM,UAAU,GAAG,gBAAgB,EAAE,CAAC;IAEtC,MAAM,cAAc,GAAG;QACrB,QAAQ,EAAE;YACR,MAAM,EAAE,IAAI,CAAC,EAAE,EAAE;SAClB;QACD,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;QACnB,WAAW,EAAE,IAAI,CAAC,EAAE,EAAE;KAGY,CAAC;IAErC,MAAM,QAAQ,GAAG;QACf,OAAO,EAAE,cAAc;KACS,CAAC;IAEnC,MAAM,WAAW,GAAG;QAClB,MAAM,EAAE,EAAE;QACV,UAAU,EAAE,EAAE;QACd,IAAI,EAAE,QAAQ;KACgC,CAAC;IAEjD,MAAM,YAAY,GAAqB;QACrC;YACE,IAAI,EAAE,UAAU;YAChB,SAAS,EAAE,GAAG,EAAE,CAAC,sDAAmB;YACpC,WAAW,EAAE,UAAU;YACvB,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;gBACpB,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;aACjB,CAAC;YACF,eAAe,EAAE;gBACf;oBACE,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,WAAW,CAAC;oBAC9C,WAAW,EAAE,WAAW;oBACxB,UAAU,EAAE,OAAC;yBACV,QAAQ,EAAE;yBACV,IAAI,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC,CAAC;yBACnD,OAAO,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC;iBACvB;aACF;SACF;KACF,CAAC;IAEF,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,IAAC,UAAU,EAAE,YAAY;QAC7C,8BAAC,2CAAmB,QAAE,QAAQ,CAAuB,CAC/B,CACzB,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;QACnE,IAAI;aACD,MAAM,CAAC,cAAc,CAAC,QAAQ,CAAC,MAAM,CAAC;aACtC,iBAAiB,CAAC,iBAAiB,EAAE,CAAC,CAAC;QAC1C,IAAI;aACD,MAAM,CAAC,cAAc,CAAC,WAAW,CAAC;aAClC,iBAAiB,CAAC,yBAAyB,EAAE,CAAC,CAAC;QAClD,IAAI,CAAC,MAAM,CAAC,sCAAc,CAAC,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,sCAAc,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAEnE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,6CAAe,CAAC,IAAI,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,sCAAc,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAEnE,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,OAAO,CAAC,mBAAmB,CAAC,eAAe,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,eAAe,CAAC,CAAC;QACtE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,sCAAc,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAEnE,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,sCAAc,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QACnE,MAAM,WAAW,GAAuB;YACtC,EAAE,EAAE,gBAAgB;YACpB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;YAC1C,IAAI,EAAE,MAAM;YACZ,QAAQ,EAAE,eAAe;YACzB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,cAAc,EAAE,EAAE;SACnB,CAAC;QAEF,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,cAAc,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,oBAAoB,CACzD,eAAe,EACf;YACE,OAAO,EAAE,WAAW,CAAC,OAAO;YAC5B,IAAI,EAAE,WAAW,CAAC,IAAI;SACvB,CACF,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,sCAAc,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QACnE,MAAM,WAAW,GAAuB;YACtC,EAAE,EAAE,gBAAgB;YACpB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC;YACpD,IAAI,EAAE,MAAM;YACZ,QAAQ,EAAE,eAAe;YACzB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,cAAc,EAAE,EAAE;SACnB,CAAC;QAEF,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,OAAO,CAAC,mBAAmB,CACtC,gBAAgB,EAChB,WAAW,EACX,IAAI,CACL,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,cAAc,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,oBAAoB,CACzD,eAAe,EACf;YACE,OAAO,EAAE,WAAW,CAAC,OAAO;YAC5B,IAAI,EAAE,WAAW,CAAC,IAAI;SACvB,CACF,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,mBAAmB,GAA+C;YACtE,kBAAkB,EAAE;gBAClB,EAAE,EAAE,YAAY;gBAChB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;gBAC7C,IAAI,EAAE,WAAW;gBACjB,QAAQ,EAAE,eAAe;gBACzB,SAAS,EAAE,SAAS;gBACpB,cAAc,EAAE,EAAE;gBAClB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC;YACD,eAAe,EAAE,6CAAe,CAAC,QAAQ;SAC1C,CAAC;QAEF,IAAI;aACD,MAAM,CAAC,cAAc,CAAC,WAAW,CAAC;aAClC,iBAAiB,CAAC,mBAAmB,CAAC,CAAC;QAE1C,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,sCAAc,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAEnE,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,OAAO,EAAE;gBAC9C,QAAQ,EAAE,eAAe;gBACzB,cAAc,EAAE,KAAK;aACtB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,oBAAoB,CAAC,eAAe,EAAE;YACvE,eAAe,EAAE;gBACf,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;gBAC1C,IAAI,EAAE,MAAM;aACb;YACD,mBAAmB,EAAE,IAAA,yBAAiB,EAAC,YAAY,CAAC;YACpD,UAAU,EAAE,SAAS;SACtB,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,6CAAe,CAAC,QAAQ,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,sCAAc,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAEnE,IAAA,WAAG,EAAC,GAAG,EAAE;YACP,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,kBAAkB,GAA+C;YACrE,kBAAkB,EAAE;gBAClB,EAAE,EAAE,UAAU;gBACd,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC;gBACvD,IAAI,EAAE,WAAW;gBACjB,QAAQ,EAAE,eAAe;gBACzB,SAAS,EAAE,SAAS;gBACpB,cAAc,EAAE,EAAE;gBAClB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC;YACD,eAAe,EAAE,6CAAe,CAAC,QAAQ;SAC1C,CAAC;QAEF,gCAAgC;QAChC,MAAM,iBAAiB,GAAG;YACxB,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,KAAK,SAAS,CAAC;gBACrC,MAAM,kBAAkB,CAAC;YAC3B,CAAC;SACF,CAAC;QAEF,kDAAkD;QAClD,IAAI,CAAC,MAAM,CAAC,8BAAa,CAAC,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QAEhE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,sCAAc,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAEnE,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,OAAO,EAAE;gBAC9C,QAAQ,EAAE,eAAe;gBACzB,cAAc,EAAE,IAAI;aACrB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,6CAAe,CAAC,IAAI,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,oBAAoB,GAA+C;YACvE,kBAAkB,EAAE;gBAClB,EAAE,EAAE,aAAa;gBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;gBAClD,IAAI,EAAE,MAAM;gBACZ,QAAQ,EAAE,eAAe;gBACzB,eAAe,EAAE;oBACf,QAAQ,EAAE,WAAW;oBACrB,UAAU,EAAE,CAAC,EAAE,aAAa,EAAE,MAAM,EAAE,cAAc,EAAE,MAAM,EAAE,CAAC;iBAChE;gBACD,cAAc,EAAE,EAAE;gBAClB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC;YACD,eAAe,EAAE,6CAAe,CAAC,QAAQ;SAC1C,CAAC;QAEF,IAAI;aACD,MAAM,CAAC,cAAc,CAAC,WAAW,CAAC;aAClC,qBAAqB,CAAC,oBAAoB,CAAC;aAC3C,qBAAqB,CAAC;YACrB,kBAAkB,EAAE;gBAClB,EAAE,EAAE,mBAAmB;gBACvB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;gBAC/C,IAAI,EAAE,MAAM;gBACZ,QAAQ,EAAE,eAAe;gBACzB,cAAc,EAAE,EAAE;gBAClB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACpC;YACD,eAAe,EAAE,6CAAe,CAAC,QAAQ;SAC1C,CAAC,CAAC;QAEL,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,sCAAc,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAEnE,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,UAAU,EAAE;gBACjD,QAAQ,EAAE,eAAe;gBACzB,cAAc,EAAE,KAAK;aACtB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,6CAAe,CAAC,QAAQ,CAAC,CAAC;QACtE,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,oBAAoB,CACtE,MAAM,CACP,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import TamboAI, { advanceStream } from \"@tambo-ai/typescript-sdk\";\nimport { act, renderHook } from \"@testing-library/react\";\nimport React from \"react\";\nimport { DeepPartial } from \"ts-essentials\";\nimport { z } from \"zod\";\nimport { TamboComponent } from \"../../model/component-metadata\";\nimport {\n  GenerationStage,\n  TamboThreadMessage,\n} from \"../../model/generate-component-response\";\nimport { serializeRegistry } from \"../../testing/tools\";\nimport { useTamboClient } from \"../tambo-client-provider\";\nimport { TamboRegistryProvider } from \"../tambo-registry-provider\";\nimport { TamboThreadProvider, useTamboThread } from \"../tambo-thread-provider\";\n\ntype PartialTamboAI = DeepPartial<TamboAI>;\n\n// Mock crypto.randomUUID\nObject.defineProperty(global, \"crypto\", {\n  value: {\n    randomUUID: jest.fn().mockReturnValue(\"test-uuid\"),\n  },\n});\n\n// Mock the required providers\njest.mock(\"../tambo-client-provider\", () => ({\n  useTamboClient: jest.fn(),\n}));\njest.mock(\"@tambo-ai/typescript-sdk\", () => ({\n  advanceStream: jest.fn(),\n}));\n\n// Test utilities\nconst createMockMessage = (\n  overrides: Partial<TamboThreadMessage> = {},\n): TamboThreadMessage => ({\n  id: \"test-message-1\",\n  content: [{ type: \"text\", text: \"Hello\" }],\n  role: \"user\",\n  threadId: \"test-thread-1\",\n  createdAt: new Date().toISOString(),\n  componentState: {},\n  ...overrides,\n});\n\nconst createMockThread = (\n  overrides: Partial<TamboAI.Beta.Threads.Thread> = {},\n) => ({\n  id: \"test-thread-1\",\n  messages: [],\n  createdAt: \"2024-01-01T00:00:00Z\",\n  projectId: \"test-project\",\n  updatedAt: \"2024-01-01T00:00:00Z\",\n  metadata: {},\n  ...overrides,\n});\n\nconst createMockAdvanceResponse = (\n  overrides: Partial<TamboAI.Beta.Threads.ThreadAdvanceResponse> = {},\n): TamboAI.Beta.Threads.ThreadAdvanceResponse => ({\n  responseMessageDto: {\n    id: \"test-uuid\",\n    content: [{ type: \"text\" as const, text: \"Default response\" }],\n    role: \"assistant\",\n    threadId: \"test-thread-1\",\n    component: undefined,\n    componentState: {},\n    createdAt: new Date().toISOString(),\n  },\n  generationStage: GenerationStage.COMPLETE,\n  ...overrides,\n});\n\ndescribe(\"TamboThreadProvider\", () => {\n  const mockThread = createMockThread();\n\n  const mockThreadsApi = {\n    messages: {\n      create: jest.fn(),\n    },\n    retrieve: jest.fn(),\n    advanceById: jest.fn(),\n  } satisfies DeepPartial<\n    TamboAI[\"beta\"][\"threads\"]\n  > as unknown as TamboAI.Beta.Threads;\n\n  const mockBeta = {\n    threads: mockThreadsApi,\n  } satisfies PartialTamboAI[\"beta\"];\n\n  const mockTamboAI = {\n    apiKey: \"\",\n    components: {},\n    beta: mockBeta,\n  } satisfies PartialTamboAI as unknown as TamboAI;\n\n  const mockRegistry: TamboComponent[] = [\n    {\n      name: \"TestOnly\",\n      component: () => <div>TestOnly</div>,\n      description: \"TestOnly\",\n      propsSchema: z.object({\n        test: z.string(),\n      }),\n      associatedTools: [\n        {\n          name: \"test-tool\",\n          tool: jest.fn().mockResolvedValue(\"test-tool\"),\n          description: \"test-tool\",\n          toolSchema: z\n            .function()\n            .args(z.string().describe(\"test-param-description\"))\n            .returns(z.string()),\n        },\n      ],\n    },\n  ];\n\n  const wrapper = ({ children }: { children: React.ReactNode }) => (\n    <TamboRegistryProvider components={mockRegistry}>\n      <TamboThreadProvider>{children}</TamboThreadProvider>\n    </TamboRegistryProvider>\n  );\n\n  beforeEach(() => {\n    jest.mocked(mockThreadsApi.retrieve).mockResolvedValue(mockThread);\n    jest\n      .mocked(mockThreadsApi.messages.create)\n      .mockResolvedValue(createMockMessage());\n    jest\n      .mocked(mockThreadsApi.advanceById)\n      .mockResolvedValue(createMockAdvanceResponse());\n    jest.mocked(useTamboClient).mockReturnValue(mockTamboAI);\n  });\n\n  it(\"should initialize with placeholder thread\", () => {\n    const { result } = renderHook(() => useTamboThread(), { wrapper });\n\n    expect(result.current.thread.id).toBe(\"placeholder\");\n    expect(result.current.isIdle).toBe(true);\n    expect(result.current.generationStage).toBe(GenerationStage.IDLE);\n  });\n\n  it(\"should switch to a new thread\", async () => {\n    const { result } = renderHook(() => useTamboThread(), { wrapper });\n\n    await act(async () => {\n      await result.current.switchCurrentThread(\"test-thread-1\");\n    });\n\n    expect(mockThreadsApi.retrieve).toHaveBeenCalledWith(\"test-thread-1\");\n    expect(result.current.thread.id).toBe(\"test-thread-1\");\n  });\n\n  it(\"should start a new thread\", async () => {\n    const { result } = renderHook(() => useTamboThread(), { wrapper });\n\n    await act(async () => {\n      result.current.startNewThread();\n    });\n\n    expect(result.current.thread.id).toBe(\"placeholder\");\n    expect(result.current.isIdle).toBe(true);\n  });\n\n  it(\"should add a message to the thread\", async () => {\n    const { result } = renderHook(() => useTamboThread(), { wrapper });\n    const testMessage: TamboThreadMessage = {\n      id: \"test-message-1\",\n      content: [{ type: \"text\", text: \"Hello\" }],\n      role: \"user\",\n      threadId: \"test-thread-1\",\n      createdAt: new Date().toISOString(),\n      componentState: {},\n    };\n\n    await act(async () => {\n      await result.current.addThreadMessage(testMessage, true);\n    });\n\n    expect(mockThreadsApi.messages.create).toHaveBeenCalledWith(\n      \"test-thread-1\",\n      {\n        content: testMessage.content,\n        role: testMessage.role,\n      },\n    );\n  });\n\n  it(\"should update a message in the thread\", async () => {\n    const { result } = renderHook(() => useTamboThread(), { wrapper });\n    const testMessage: TamboThreadMessage = {\n      id: \"test-message-1\",\n      content: [{ type: \"text\", text: \"Updated message\" }],\n      role: \"user\",\n      threadId: \"test-thread-1\",\n      createdAt: new Date().toISOString(),\n      componentState: {},\n    };\n\n    await act(async () => {\n      await result.current.updateThreadMessage(\n        \"test-message-1\",\n        testMessage,\n        true,\n      );\n    });\n\n    expect(mockThreadsApi.messages.create).toHaveBeenCalledWith(\n      \"test-thread-1\",\n      {\n        content: testMessage.content,\n        role: testMessage.role,\n      },\n    );\n  });\n\n  it(\"should send a message and update thread state\", async () => {\n    const mockAdvanceResponse: TamboAI.Beta.Threads.ThreadAdvanceResponse = {\n      responseMessageDto: {\n        id: \"response-1\",\n        content: [{ type: \"text\", text: \"Response\" }],\n        role: \"assistant\",\n        threadId: \"test-thread-1\",\n        component: undefined,\n        componentState: {},\n        createdAt: new Date().toISOString(),\n      },\n      generationStage: GenerationStage.COMPLETE,\n    };\n\n    jest\n      .mocked(mockThreadsApi.advanceById)\n      .mockResolvedValue(mockAdvanceResponse);\n\n    const { result } = renderHook(() => useTamboThread(), { wrapper });\n\n    await act(async () => {\n      await result.current.sendThreadMessage(\"Hello\", {\n        threadId: \"test-thread-1\",\n        streamResponse: false,\n      });\n    });\n\n    expect(mockThreadsApi.advanceById).toHaveBeenCalledWith(\"test-thread-1\", {\n      messageToAppend: {\n        content: [{ type: \"text\", text: \"Hello\" }],\n        role: \"user\",\n      },\n      availableComponents: serializeRegistry(mockRegistry),\n      contextKey: undefined,\n    });\n    expect(result.current.generationStage).toBe(GenerationStage.COMPLETE);\n  });\n\n  it(\"should handle input value changes\", () => {\n    const { result } = renderHook(() => useTamboThread(), { wrapper });\n\n    act(() => {\n      result.current.setInputValue(\"New input\");\n    });\n\n    expect(result.current.inputValue).toBe(\"New input\");\n  });\n\n  it(\"should handle streaming responses\", async () => {\n    const mockStreamResponse: TamboAI.Beta.Threads.ThreadAdvanceResponse = {\n      responseMessageDto: {\n        id: \"stream-1\",\n        content: [{ type: \"text\", text: \"Streaming response\" }],\n        role: \"assistant\",\n        threadId: \"test-thread-1\",\n        component: undefined,\n        componentState: {},\n        createdAt: new Date().toISOString(),\n      },\n      generationStage: GenerationStage.COMPLETE,\n    };\n\n    // Create an async iterator mock\n    const mockAsyncIterator = {\n      [Symbol.asyncIterator]: async function* () {\n        yield mockStreamResponse;\n      },\n    };\n\n    // Mock advanceStream to return our async iterator\n    jest.mocked(advanceStream).mockResolvedValue(mockAsyncIterator);\n\n    const { result } = renderHook(() => useTamboThread(), { wrapper });\n\n    await act(async () => {\n      await result.current.sendThreadMessage(\"Hello\", {\n        threadId: \"test-thread-1\",\n        streamResponse: true,\n      });\n    });\n\n    expect(result.current.generationStage).toBe(GenerationStage.IDLE);\n  });\n\n  it(\"should handle tool calls during message processing\", async () => {\n    const mockToolCallResponse: TamboAI.Beta.Threads.ThreadAdvanceResponse = {\n      responseMessageDto: {\n        id: \"tool-call-1\",\n        content: [{ type: \"text\", text: \"Tool response\" }],\n        role: \"tool\",\n        threadId: \"test-thread-1\",\n        toolCallRequest: {\n          toolName: \"test-tool\",\n          parameters: [{ parameterName: \"test\", parameterValue: \"test\" }],\n        },\n        componentState: {},\n        createdAt: new Date().toISOString(),\n      },\n      generationStage: GenerationStage.COMPLETE,\n    };\n\n    jest\n      .mocked(mockThreadsApi.advanceById)\n      .mockResolvedValueOnce(mockToolCallResponse)\n      .mockResolvedValueOnce({\n        responseMessageDto: {\n          id: \"advance-response2\",\n          content: [{ type: \"text\", text: \"response 2\" }],\n          role: \"user\",\n          threadId: \"test-thread-1\",\n          componentState: {},\n          createdAt: new Date().toISOString(),\n        },\n        generationStage: GenerationStage.COMPLETE,\n      });\n\n    const { result } = renderHook(() => useTamboThread(), { wrapper });\n\n    await act(async () => {\n      await result.current.sendThreadMessage(\"Use tool\", {\n        threadId: \"test-thread-1\",\n        streamResponse: false,\n      });\n    });\n    expect(result.current.generationStage).toBe(GenerationStage.COMPLETE);\n    expect(mockRegistry[0]?.associatedTools?.[0]?.tool).toHaveBeenCalledWith(\n      \"test\",\n    );\n  });\n});\n"]}

package/dist/providers/tambo-provider.d.ts

@@ -15,7 +15,7 @@ import { TamboThreadContextProps } from "./tambo-thread-provider";
  * @returns The TamboProvider component
  */
 export declare const TamboProvider: React.FC<PropsWithChildren<TamboClientProviderProps & TamboRegistryProviderProps>>;
-type TamboContextProps = TamboClientContextProps & TamboThreadContextProps & TamboComponentContextProps;
+export type TamboContextProps = TamboClientContextProps & TamboThreadContextProps & TamboComponentContextProps;
 export declare const TamboContext: React.Context<TamboContextProps>;
 /**
  * The useTambo hook provides access to the Tambo API. This is the primary entrypoint
@@ -25,5 +25,4 @@ export declare const TamboContext: React.Context<TamboContextProps>;
  * @returns The Tambo API
  */
 export declare const useTambo: () => TamboContextProps;
-export {};
 //# sourceMappingURL=tambo-provider.d.ts.map

package/dist/providers/tambo-provider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAAE,iBAAiB,EAA6B,MAAM,OAAO,CAAC;AAC5E,OAAO,EACL,uBAAuB,EAEvB,wBAAwB,EAGzB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,0BAA0B,EAG3B,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAEL,0BAA0B,EAC3B,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,uBAAuB,EAGxB,MAAM,yBAAyB,CAAC;AAEjC;;;;;;;;;;GAUG;AACH,eAAO,MAAM,aAAa,EAAE,KAAK,CAAC,EAAE,CAClC,iBAAiB,CAAC,wBAAwB,GAAG,0BAA0B,CAAC,CAsBzE,CAAC;AACF,KAAK,iBAAiB,GAAG,uBAAuB,GAC9C,uBAAuB,GACvB,0BAA0B,CAAC;AAE7B,eAAO,MAAM,YAAY,kCAExB,CAAC;AA6BF;;;;;;GAMG;AACH,eAAO,MAAM,QAAQ,yBAEpB,CAAC"}
+{"version":3,"file":"tambo-provider.d.ts","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAAE,iBAAiB,EAA6B,MAAM,OAAO,CAAC;AAC5E,OAAO,EACL,uBAAuB,EAEvB,wBAAwB,EAGzB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EACL,0BAA0B,EAG3B,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAEL,0BAA0B,EAC3B,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACL,uBAAuB,EAGxB,MAAM,yBAAyB,CAAC;AAEjC;;;;;;;;;;GAUG;AACH,eAAO,MAAM,aAAa,EAAE,KAAK,CAAC,EAAE,CAClC,iBAAiB,CAAC,wBAAwB,GAAG,0BAA0B,CAAC,CAsBzE,CAAC;AACF,MAAM,MAAM,iBAAiB,GAAG,uBAAuB,GACrD,uBAAuB,GACvB,0BAA0B,CAAC;AAE7B,eAAO,MAAM,YAAY,kCAExB,CAAC;AA6BF;;;;;;GAMG;AACH,eAAO,MAAM,QAAQ,yBAEpB,CAAC"}

package/dist/providers/tambo-provider.js.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AACb,+CAA4E;AAC5E,mEAMiC;AACjC,yEAIoC;AACpC,uEAGmC;AACnC,mEAIiC;AAEjC;;;;;;;;;;GAUG;AACI,MAAM,aAAa,GAEtB,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,EAAE,EAAE;IAC9D,iCAAiC;IACjC,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;QAClC,OAAO,CAAC,KAAK,CAAC,6CAA6C,CAAC,CAAC;IAC/D,CAAC;IAED,OAAO,CACL,8BAAC,2CAAmB,IAClB,QAAQ,EAAE,QAAQ,EAClB,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,WAAW;QAExB,8BAAC,+CAAqB,IAAC,UAAU,EAAE,UAAU;YAC3C,8BAAC,2CAAmB;gBAClB,8BAAC,iDAAsB;oBACrB,8BAAC,sBAAsB,QAAE,QAAQ,CAA0B,CACpC,CACL,CACA,CACJ,CACvB,CAAC;AACJ,CAAC,CAAC;AAvBW,QAAA,aAAa,iBAuBxB;AAKW,QAAA,YAAY,GAAG,IAAA,qBAAa,EACvC,EAAuB,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,sBAAsB,GAAgC,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;IAC3E,MAAM,OAAO,GAAG,IAAA,sCAAc,GAAE,CAAC;IACjC,MAAM,MAAM,GAAG,IAAA,sCAAc,GAAE,CAAC;IAChC,MAAM,WAAW,GAAG,IAAA,2CAAmB,GAAE,CAAC;IAC1C,MAAM,iBAAiB,GAAG,IAAA,4CAAiB,GAAE,CAAC;IAE9C,OAAO,CACL,8BAAC,oBAAY,CAAC,QAAQ,IACpB,KAAK,EAAE;YACL,MAAM;YACN,WAAW;YACX,GAAG,iBAAiB;YACpB,GAAG,OAAO;SACX,IAEA,QAAQ,CACa,CACzB,CAAC;AACJ,CAAC,CAAC;AAEF;;;;;;GAMG;AACI,MAAM,QAAQ,GAAG,GAAG,EAAE;IAC3B,OAAO,IAAA,kBAAU,EAAC,oBAAY,CAAC,CAAC;AAClC,CAAC,CAAC;AAFW,QAAA,QAAQ,YAEnB","sourcesContent":["\"use client\";\nimport React, { PropsWithChildren, createContext, useContext } from \"react\";\nimport {\n  TamboClientContextProps,\n  TamboClientProvider,\n  TamboClientProviderProps,\n  useTamboClient,\n  useTamboQueryClient,\n} from \"./tambo-client-provider\";\nimport {\n  TamboComponentContextProps,\n  TamboComponentProvider,\n  useTamboComponent,\n} from \"./tambo-component-provider\";\nimport {\n  TamboRegistryProvider,\n  TamboRegistryProviderProps,\n} from \"./tambo-registry-provider\";\nimport {\n  TamboThreadContextProps,\n  TamboThreadProvider,\n  useTamboThread,\n} from \"./tambo-thread-provider\";\n\n/**\n * The TamboProvider gives full access to the whole Tambo API. This includes the\n * TamboAI client, the component registry, and the current thread context.\n * @param props - The props for the TamboProvider\n * @param props.children - The children to wrap\n * @param props.tamboUrl - The URL of the Tambo API\n * @param props.apiKey - The API key for the Tambo API\n * @param props.components - The components to register\n * @param props.environment - The environment to use for the Tambo API\n * @returns The TamboProvider component\n */\nexport const TamboProvider: React.FC<\n  PropsWithChildren<TamboClientProviderProps & TamboRegistryProviderProps>\n> = ({ children, tamboUrl, apiKey, components, environment }) => {\n  // Should only be used in browser\n  if (typeof window === \"undefined\") {\n    console.error(\"TamboProvider must be used within a browser\");\n  }\n\n  return (\n    <TamboClientProvider\n      tamboUrl={tamboUrl}\n      apiKey={apiKey}\n      environment={environment}\n    >\n      <TamboRegistryProvider components={components}>\n        <TamboThreadProvider>\n          <TamboComponentProvider>\n            <TamboCompositeProvider>{children}</TamboCompositeProvider>\n          </TamboComponentProvider>\n        </TamboThreadProvider>\n      </TamboRegistryProvider>\n    </TamboClientProvider>\n  );\n};\ntype TamboContextProps = TamboClientContextProps &\n  TamboThreadContextProps &\n  TamboComponentContextProps;\n\nexport const TamboContext = createContext<TamboContextProps>(\n  {} as TamboContextProps,\n);\n\n/**\n * TamboCompositeProvider is a provider that combines the TamboClient,\n * TamboThread, and TamboComponent providers\n * @param props - The props for the TamboCompositeProvider\n * @param props.children - The children to wrap\n * @returns The wrapped component\n */\nconst TamboCompositeProvider: React.FC<PropsWithChildren> = ({ children }) => {\n  const threads = useTamboThread();\n  const client = useTamboClient();\n  const queryClient = useTamboQueryClient();\n  const componentRegistry = useTamboComponent();\n\n  return (\n    <TamboContext.Provider\n      value={{\n        client,\n        queryClient,\n        ...componentRegistry,\n        ...threads,\n      }}\n    >\n      {children}\n    </TamboContext.Provider>\n  );\n};\n\n/**\n * The useTambo hook provides access to the Tambo API. This is the primary entrypoint\n * for the Tambo React SDK.\n *\n * This includes the TamboAI client, the component registry, and the current thread context.\n * @returns The Tambo API\n */\nexport const useTambo = () => {\n  return useContext(TamboContext);\n};\n"]}
+{"version":3,"file":"tambo-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-provider.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AACb,+CAA4E;AAC5E,mEAMiC;AACjC,yEAIoC;AACpC,uEAGmC;AACnC,mEAIiC;AAEjC;;;;;;;;;;GAUG;AACI,MAAM,aAAa,GAEtB,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,WAAW,EAAE,EAAE,EAAE;IAC9D,iCAAiC;IACjC,IAAI,OAAO,MAAM,KAAK,WAAW,EAAE,CAAC;QAClC,OAAO,CAAC,KAAK,CAAC,6CAA6C,CAAC,CAAC;IAC/D,CAAC;IAED,OAAO,CACL,8BAAC,2CAAmB,IAClB,QAAQ,EAAE,QAAQ,EAClB,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,WAAW;QAExB,8BAAC,+CAAqB,IAAC,UAAU,EAAE,UAAU;YAC3C,8BAAC,2CAAmB;gBAClB,8BAAC,iDAAsB;oBACrB,8BAAC,sBAAsB,QAAE,QAAQ,CAA0B,CACpC,CACL,CACA,CACJ,CACvB,CAAC;AACJ,CAAC,CAAC;AAvBW,QAAA,aAAa,iBAuBxB;AAKW,QAAA,YAAY,GAAG,IAAA,qBAAa,EACvC,EAAuB,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,sBAAsB,GAAgC,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE;IAC3E,MAAM,OAAO,GAAG,IAAA,sCAAc,GAAE,CAAC;IACjC,MAAM,MAAM,GAAG,IAAA,sCAAc,GAAE,CAAC;IAChC,MAAM,WAAW,GAAG,IAAA,2CAAmB,GAAE,CAAC;IAC1C,MAAM,iBAAiB,GAAG,IAAA,4CAAiB,GAAE,CAAC;IAE9C,OAAO,CACL,8BAAC,oBAAY,CAAC,QAAQ,IACpB,KAAK,EAAE;YACL,MAAM;YACN,WAAW;YACX,GAAG,iBAAiB;YACpB,GAAG,OAAO;SACX,IAEA,QAAQ,CACa,CACzB,CAAC;AACJ,CAAC,CAAC;AAEF;;;;;;GAMG;AACI,MAAM,QAAQ,GAAG,GAAG,EAAE;IAC3B,OAAO,IAAA,kBAAU,EAAC,oBAAY,CAAC,CAAC;AAClC,CAAC,CAAC;AAFW,QAAA,QAAQ,YAEnB","sourcesContent":["\"use client\";\nimport React, { PropsWithChildren, createContext, useContext } from \"react\";\nimport {\n  TamboClientContextProps,\n  TamboClientProvider,\n  TamboClientProviderProps,\n  useTamboClient,\n  useTamboQueryClient,\n} from \"./tambo-client-provider\";\nimport {\n  TamboComponentContextProps,\n  TamboComponentProvider,\n  useTamboComponent,\n} from \"./tambo-component-provider\";\nimport {\n  TamboRegistryProvider,\n  TamboRegistryProviderProps,\n} from \"./tambo-registry-provider\";\nimport {\n  TamboThreadContextProps,\n  TamboThreadProvider,\n  useTamboThread,\n} from \"./tambo-thread-provider\";\n\n/**\n * The TamboProvider gives full access to the whole Tambo API. This includes the\n * TamboAI client, the component registry, and the current thread context.\n * @param props - The props for the TamboProvider\n * @param props.children - The children to wrap\n * @param props.tamboUrl - The URL of the Tambo API\n * @param props.apiKey - The API key for the Tambo API\n * @param props.components - The components to register\n * @param props.environment - The environment to use for the Tambo API\n * @returns The TamboProvider component\n */\nexport const TamboProvider: React.FC<\n  PropsWithChildren<TamboClientProviderProps & TamboRegistryProviderProps>\n> = ({ children, tamboUrl, apiKey, components, environment }) => {\n  // Should only be used in browser\n  if (typeof window === \"undefined\") {\n    console.error(\"TamboProvider must be used within a browser\");\n  }\n\n  return (\n    <TamboClientProvider\n      tamboUrl={tamboUrl}\n      apiKey={apiKey}\n      environment={environment}\n    >\n      <TamboRegistryProvider components={components}>\n        <TamboThreadProvider>\n          <TamboComponentProvider>\n            <TamboCompositeProvider>{children}</TamboCompositeProvider>\n          </TamboComponentProvider>\n        </TamboThreadProvider>\n      </TamboRegistryProvider>\n    </TamboClientProvider>\n  );\n};\nexport type TamboContextProps = TamboClientContextProps &\n  TamboThreadContextProps &\n  TamboComponentContextProps;\n\nexport const TamboContext = createContext<TamboContextProps>(\n  {} as TamboContextProps,\n);\n\n/**\n * TamboCompositeProvider is a provider that combines the TamboClient,\n * TamboThread, and TamboComponent providers\n * @param props - The props for the TamboCompositeProvider\n * @param props.children - The children to wrap\n * @returns The wrapped component\n */\nconst TamboCompositeProvider: React.FC<PropsWithChildren> = ({ children }) => {\n  const threads = useTamboThread();\n  const client = useTamboClient();\n  const queryClient = useTamboQueryClient();\n  const componentRegistry = useTamboComponent();\n\n  return (\n    <TamboContext.Provider\n      value={{\n        client,\n        queryClient,\n        ...componentRegistry,\n        ...threads,\n      }}\n    >\n      {children}\n    </TamboContext.Provider>\n  );\n};\n\n/**\n * The useTambo hook provides access to the Tambo API. This is the primary entrypoint\n * for the Tambo React SDK.\n *\n * This includes the TamboAI client, the component registry, and the current thread context.\n * @returns The Tambo API\n */\nexport const useTambo = () => {\n  return useContext(TamboContext);\n};\n"]}

package/dist/providers/tambo-registry-provider.js

@@ -39,7 +39,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.useTamboRegistry = exports.TamboRegistryProvider = void 0;
 const react_1 = __importStar(require("react"));
-const zod_1 = __importStar(require("zod"));
+const zod_1 = require("zod");
 const zod_to_json_schema_1 = __importDefault(require("zod-to-json-schema"));
 const TamboRegistryContext = (0, react_1.createContext)({
     componentList: {},
@@ -166,9 +166,6 @@ function getSerializedProps(propsDefinition, propsSchema, name) {
         console.warn(`propsDefinition is deprecated. Use propsSchema instead.`);
         return propsDefinition;
     }
-    if (propsSchema instanceof zod_1.ZodSchema || propsSchema instanceof zod_1.default.ZodObject) {
-        return (0, zod_to_json_schema_1.default)(propsSchema);
-    }
     if (isZodSchema(propsSchema)) {
         try {
             return (0, zod_to_json_schema_1.default)(propsSchema);

package/dist/providers/tambo-registry-provider.js.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-registry-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-registry-provider.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AACb,+CAOe;AACf,2CAAmC;AACnC,4EAAiD;AAiBjD,MAAM,oBAAoB,GAAG,IAAA,qBAAa,EAAuB;IAC/D,aAAa,EAAE,EAAE;IACjB,YAAY,EAAE,EAAE;IAChB,yBAAyB,EAAE,EAAE;IAC7B;;OAEG;IACH,iBAAiB,EAAE,GAAG,EAAE,GAAE,CAAC;IAC3B;;OAEG;IACH,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;IACtB;;OAEG;IACH,aAAa,EAAE,GAAG,EAAE,GAAE,CAAC;IACvB;;OAEG;IACH,kBAAkB,EAAE,GAAG,EAAE,GAAE,CAAC;CAC7B,CAAC,CAAC;AAOH;;;;;;;GAOG;AACI,MAAM,qBAAqB,GAE9B,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,cAAc,EAAE,EAAE,EAAE;IAC/C,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,IAAA,gBAAQ,EAAoB,EAAE,CAAC,CAAC;IAC1E,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,IAAA,gBAAQ,EAC9C,EAAE,CACH,CAAC;IACF,MAAM,CAAC,yBAAyB,EAAE,4BAA4B,CAAC,GAAG,IAAA,gBAAQ,EAExE,EAAE,CAAC,CAAC;IAEN,MAAM,YAAY,GAAG,IAAA,mBAAW,EAC9B,CAAC,IAAe,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAC1C,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;gBACvC,OAAO,CAAC,IAAI,CAAC,oBAAoB,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAChD,CAAC;YACD,OAAO;gBACL,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI;aAClB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,aAAa,GAAG,IAAA,mBAAW,EAC/B,CAAC,KAAkB,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAC7C,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC,CAAC;IAC/D,CAAC,EACD,CAAC,YAAY,CAAC,CACf,CAAC;IAEF,MAAM,kBAAkB,GAAG,IAAA,mBAAW,EACpC,CAAC,aAAqB,EAAE,IAAe,EAAE,EAAE;QACzC,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,aAAa,aAAa,wBAAwB,CAAC,CAAC;QACtE,CAAC;QACD,4BAA4B,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACtC,GAAG,IAAI;YACP,CAAC,aAAa,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC;SAC7D,CAAC,CAAC,CAAC;IACN,CAAC,EACD,CAAC,aAAa,CAAC,CAChB,CAAC;IAEF,MAAM,iBAAiB,GAAG,IAAA,mBAAW,EACnC,CAAC,OAAuB,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAClD,MAAM,EACJ,IAAI,EACJ,WAAW,EACX,SAAS,EACT,WAAW,EACX,eAAe,EACf,gBAAgB,EAChB,eAAe,GAChB,GAAG,OAAO,CAAC;QAEZ,0DAA0D;QAC1D,IAAI,CAAC,WAAW,IAAI,CAAC,eAAe,EAAE,CAAC;YACrC,MAAM,IAAI,KAAK,CACb,aAAa,IAAI,wEAAwE,CAC1F,CAAC;QACJ,CAAC;QAED,sDAAsD;QACtD,IAAI,WAAW,IAAI,eAAe,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CACb,aAAa,IAAI,0GAA0G,CAC5H,CAAC;QACJ,CAAC;QAED,kDAAkD;QAClD,MAAM,KAAK,GAAG,kBAAkB,CAAC,eAAe,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;QAErE,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE;YACxB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;gBAClC,OAAO,CAAC,IAAI,CAAC,yBAAyB,IAAI,EAAE,CAAC,CAAC;YAChD,CAAC;YACD,OAAO;gBACL,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,EAAE;oBACN,SAAS;oBACT,gBAAgB;oBAChB,IAAI;oBACJ,WAAW;oBACX,KAAK;oBACL,YAAY,EAAE,EAAE;iBACjB;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,IAAI,eAAe,EAAE,CAAC;YACpB,aAAa,CAAC,eAAe,CAAC,CAAC;YAC/B,4BAA4B,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACtC,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;aACjD,CAAC,CAAC,CAAC;QACN,CAAC;IACH,CAAC,EACD,CAAC,aAAa,CAAC,CAChB,CAAC;IACF,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,cAAc,EAAE,CAAC;YACnB,cAAc,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;gBACnC,iBAAiB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YACtC,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,EAAE,CAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC,CAAC;IAExC,MAAM,KAAK,GAAG;QACZ,aAAa;QACb,YAAY;QACZ,yBAAyB;QACzB,iBAAiB;QACjB,YAAY;QACZ,aAAa;QACb,kBAAkB;KACnB,CAAC;IAEF,OAAO,CACL,8BAAC,oBAAoB,CAAC,QAAQ,IAAC,KAAK,EAAE,KAAK,IACxC,QAAQ,CACqB,CACjC,CAAC;AACJ,CAAC,CAAC;AA5HW,QAAA,qBAAqB,yBA4HhC;AAEF;;;;GAIG;AACI,MAAM,gBAAgB,GAAG,GAAG,EAAE;IACnC,OAAO,IAAA,kBAAU,EAAC,oBAAoB,CAAC,CAAC;AAC1C,CAAC,CAAC;AAFW,QAAA,gBAAgB,oBAE3B;AACF,SAAS,kBAAkB,CACzB,eAAoB,EACpB,WAAgB,EAChB,IAAY;IAEZ,IAAI,eAAe,EAAE,CAAC;QACpB,OAAO,CAAC,IAAI,CAAC,yDAAyD,CAAC,CAAC;QACxE,OAAO,eAAe,CAAC;IACzB,CAAC;IACD,IAAI,WAAW,YAAY,eAAS,IAAI,WAAW,YAAY,aAAC,CAAC,SAAS,EAAE,CAAC;QAC3E,OAAO,IAAA,4BAAe,EAAC,WAAW,CAAC,CAAC;IACtC,CAAC;IAED,IAAI,WAAW,CAAC,WAAW,CAAC,EAAE,CAAC;QAC7B,IAAI,CAAC;YACH,OAAO,IAAA,4BAAe,EAAC,WAAW,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CACX,oBAAoB,IAAI,+BAA+B,EACvD,KAAK,CACN,CAAC;QACJ,CAAC;IACH,CAAC;IACD,qFAAqF;IACrF,IAAI,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC;QAC9B,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,4BAA4B,IAAI,EAAE,CAAC,CAAC;AACtD,CAAC;AAED;;;;;GAKG;AACH,SAAS,YAAY,CAAC,WAAgB;IACpC,OAAO,CACL,WAAW;QACX,OAAO,WAAW,KAAK,QAAQ;QAC/B,WAAW,CAAC,IAAI,KAAK,QAAQ;QAC7B,WAAW,CAAC,UAAU,CACvB,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,SAAS,WAAW,CAAC,GAAY;IAC/B,IAAI,GAAG,YAAY,eAAS,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IACD,6CAA6C;IAC7C,OAAO,CACL,OAAO,GAAG,KAAK,QAAQ;QACvB,GAAG,KAAK,IAAI;QACZ,OAAQ,GAAW,CAAC,SAAS,KAAK,UAAU;QAC5C,OAAQ,GAAW,CAAC,IAAI,KAAK,QAAQ,CACtC,CAAC;AACJ,CAAC","sourcesContent":["\"use client\";\nimport React, {\n  createContext,\n  PropsWithChildren,\n  useCallback,\n  useContext,\n  useEffect,\n  useState,\n} from \"react\";\nimport z, { ZodSchema } from \"zod\";\nimport zodToJsonSchema from \"zod-to-json-schema\";\nimport {\n  ComponentRegistry,\n  TamboComponent,\n  TamboTool,\n} from \"../model/component-metadata\";\n\nexport interface TamboRegistryContext {\n  componentList: ComponentRegistry;\n  toolRegistry: Record<string, TamboTool>;\n  componentToolAssociations: Record<string, string[]>;\n  registerComponent: (options: TamboComponent) => void;\n  registerTool: (tool: TamboTool) => void;\n  registerTools: (tools: TamboTool[]) => void;\n  addToolAssociation: (componentName: string, tool: TamboTool) => void;\n}\n\nconst TamboRegistryContext = createContext<TamboRegistryContext>({\n  componentList: {},\n  toolRegistry: {},\n  componentToolAssociations: {},\n  /**\n   *\n   */\n  registerComponent: () => {},\n  /**\n   *\n   */\n  registerTool: () => {},\n  /**\n   *\n   */\n  registerTools: () => {},\n  /**\n   *\n   */\n  addToolAssociation: () => {},\n});\n\nexport interface TamboRegistryProviderProps {\n  /** The components to register */\n  components?: TamboComponent[];\n}\n\n/**\n * The TamboRegistryProvider is a React provider that provides a component\n * registry to the descendants of the provider.\n * @param props - The props for the TamboRegistryProvider\n * @param props.children - The children to wrap\n * @param props.components - The components to register\n * @returns The TamboRegistryProvider component\n */\nexport const TamboRegistryProvider: React.FC<\n  PropsWithChildren<TamboRegistryProviderProps>\n> = ({ children, components: userComponents }) => {\n  const [componentList, setComponentList] = useState<ComponentRegistry>({});\n  const [toolRegistry, setToolRegistry] = useState<Record<string, TamboTool>>(\n    {},\n  );\n  const [componentToolAssociations, setComponentToolAssociations] = useState<\n    Record<string, string[]>\n  >({});\n\n  const registerTool = useCallback(\n    (tool: TamboTool, warnOnOverwrite = true) => {\n      setToolRegistry((prev) => {\n        if (prev[tool.name] && warnOnOverwrite) {\n          console.warn(`Overwriting tool ${tool.name}`);\n        }\n        return {\n          ...prev,\n          [tool.name]: tool,\n        };\n      });\n    },\n    [],\n  );\n\n  const registerTools = useCallback(\n    (tools: TamboTool[], warnOnOverwrite = true) => {\n      tools.forEach((tool) => registerTool(tool, warnOnOverwrite));\n    },\n    [registerTool],\n  );\n\n  const addToolAssociation = useCallback(\n    (componentName: string, tool: TamboTool) => {\n      if (!componentList[componentName]) {\n        throw new Error(`Component ${componentName} not found in registry`);\n      }\n      setComponentToolAssociations((prev) => ({\n        ...prev,\n        [componentName]: [...(prev[componentName] || []), tool.name],\n      }));\n    },\n    [componentList],\n  );\n\n  const registerComponent = useCallback(\n    (options: TamboComponent, warnOnOverwrite = true) => {\n      const {\n        name,\n        description,\n        component,\n        propsSchema,\n        propsDefinition,\n        loadingComponent,\n        associatedTools,\n      } = options;\n\n      // Validate that at least one props definition is provided\n      if (!propsSchema && !propsDefinition) {\n        throw new Error(\n          `Component ${name} must have either propsSchema (recommended) or propsDefinition defined`,\n        );\n      }\n\n      // Validate that only one props definition is provided\n      if (propsSchema && propsDefinition) {\n        throw new Error(\n          `Component ${name} cannot have both propsSchema and propsDefinition defined. Use only one. We recommend using propsSchema.`,\n        );\n      }\n\n      // Convert propsSchema to JSON Schema if it exists\n      const props = getSerializedProps(propsDefinition, propsSchema, name);\n\n      setComponentList((prev) => {\n        if (prev[name] && warnOnOverwrite) {\n          console.warn(`overwriting component ${name}`);\n        }\n        return {\n          ...prev,\n          [name]: {\n            component,\n            loadingComponent,\n            name,\n            description,\n            props,\n            contextTools: [],\n          },\n        };\n      });\n      if (associatedTools) {\n        registerTools(associatedTools);\n        setComponentToolAssociations((prev) => ({\n          ...prev,\n          [name]: associatedTools.map((tool) => tool.name),\n        }));\n      }\n    },\n    [registerTools],\n  );\n  useEffect(() => {\n    if (userComponents) {\n      userComponents.forEach((component) => {\n        registerComponent(component, false);\n      });\n    }\n  }, [registerComponent, userComponents]);\n\n  const value = {\n    componentList,\n    toolRegistry,\n    componentToolAssociations,\n    registerComponent,\n    registerTool,\n    registerTools,\n    addToolAssociation,\n  };\n\n  return (\n    <TamboRegistryContext.Provider value={value}>\n      {children}\n    </TamboRegistryContext.Provider>\n  );\n};\n\n/**\n * The useTamboRegistry hook provides access to the component registry\n * to the descendants of the TamboRegistryProvider.\n * @returns The component registry\n */\nexport const useTamboRegistry = () => {\n  return useContext(TamboRegistryContext);\n};\nfunction getSerializedProps(\n  propsDefinition: any,\n  propsSchema: any,\n  name: string,\n) {\n  if (propsDefinition) {\n    console.warn(`propsDefinition is deprecated. Use propsSchema instead.`);\n    return propsDefinition;\n  }\n  if (propsSchema instanceof ZodSchema || propsSchema instanceof z.ZodObject) {\n    return zodToJsonSchema(propsSchema);\n  }\n\n  if (isZodSchema(propsSchema)) {\n    try {\n      return zodToJsonSchema(propsSchema);\n    } catch (error) {\n      console.error(\n        `Error converting ${name} props schema to JSON Schema:`,\n        error,\n      );\n    }\n  }\n  // try to roughly detect JSONSchema, should always be an object with a properties key\n  if (isJSONSchema(propsSchema)) {\n    return propsSchema;\n  }\n\n  throw new Error(`Invalid props schema for ${name}`);\n}\n\n/**\n * Checks if the propsSchema is a JSON Schema. This is a rough check, and the\n * server will provide the definitive check.\n * @param propsSchema - The props schema to check\n * @returns True if the props schema is a JSON Schema, false otherwise\n */\nfunction isJSONSchema(propsSchema: any) {\n  return (\n    propsSchema &&\n    typeof propsSchema === \"object\" &&\n    propsSchema.type === \"object\" &&\n    propsSchema.properties\n  );\n}\n\n/**\n * Since we require a certain zod version, we need to check if the object is a ZodSchema\n * @param obj - The object to check\n * @returns True if the object is a ZodSchema, false otherwise\n */\nfunction isZodSchema(obj: unknown): obj is ZodSchema {\n  if (obj instanceof ZodSchema) {\n    return true;\n  }\n  // try to detect if the object is a ZodSchema\n  return (\n    typeof obj === \"object\" &&\n    obj !== null &&\n    typeof (obj as any).safeParse === \"function\" &&\n    typeof (obj as any)._def === \"object\"\n  );\n}\n"]}
+{"version":3,"file":"tambo-registry-provider.js","sourceRoot":"","sources":["../../src/providers/tambo-registry-provider.tsx"],"names":[],"mappings":";AAAA,YAAY,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AACb,+CAOe;AACf,6BAAgC;AAChC,4EAAiD;AAiBjD,MAAM,oBAAoB,GAAG,IAAA,qBAAa,EAAuB;IAC/D,aAAa,EAAE,EAAE;IACjB,YAAY,EAAE,EAAE;IAChB,yBAAyB,EAAE,EAAE;IAC7B;;OAEG;IACH,iBAAiB,EAAE,GAAG,EAAE,GAAE,CAAC;IAC3B;;OAEG;IACH,YAAY,EAAE,GAAG,EAAE,GAAE,CAAC;IACtB;;OAEG;IACH,aAAa,EAAE,GAAG,EAAE,GAAE,CAAC;IACvB;;OAEG;IACH,kBAAkB,EAAE,GAAG,EAAE,GAAE,CAAC;CAC7B,CAAC,CAAC;AAOH;;;;;;;GAOG;AACI,MAAM,qBAAqB,GAE9B,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,cAAc,EAAE,EAAE,EAAE;IAC/C,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,IAAA,gBAAQ,EAAoB,EAAE,CAAC,CAAC;IAC1E,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,IAAA,gBAAQ,EAC9C,EAAE,CACH,CAAC;IACF,MAAM,CAAC,yBAAyB,EAAE,4BAA4B,CAAC,GAAG,IAAA,gBAAQ,EAExE,EAAE,CAAC,CAAC;IAEN,MAAM,YAAY,GAAG,IAAA,mBAAW,EAC9B,CAAC,IAAe,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAC1C,eAAe,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;gBACvC,OAAO,CAAC,IAAI,CAAC,oBAAoB,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;YAChD,CAAC;YACD,OAAO;gBACL,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI;aAClB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,EACD,EAAE,CACH,CAAC;IAEF,MAAM,aAAa,GAAG,IAAA,mBAAW,EAC/B,CAAC,KAAkB,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAC7C,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC,CAAC;IAC/D,CAAC,EACD,CAAC,YAAY,CAAC,CACf,CAAC;IAEF,MAAM,kBAAkB,GAAG,IAAA,mBAAW,EACpC,CAAC,aAAqB,EAAE,IAAe,EAAE,EAAE;QACzC,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,aAAa,aAAa,wBAAwB,CAAC,CAAC;QACtE,CAAC;QACD,4BAA4B,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACtC,GAAG,IAAI;YACP,CAAC,aAAa,CAAC,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC;SAC7D,CAAC,CAAC,CAAC;IACN,CAAC,EACD,CAAC,aAAa,CAAC,CAChB,CAAC;IAEF,MAAM,iBAAiB,GAAG,IAAA,mBAAW,EACnC,CAAC,OAAuB,EAAE,eAAe,GAAG,IAAI,EAAE,EAAE;QAClD,MAAM,EACJ,IAAI,EACJ,WAAW,EACX,SAAS,EACT,WAAW,EACX,eAAe,EACf,gBAAgB,EAChB,eAAe,GAChB,GAAG,OAAO,CAAC;QAEZ,0DAA0D;QAC1D,IAAI,CAAC,WAAW,IAAI,CAAC,eAAe,EAAE,CAAC;YACrC,MAAM,IAAI,KAAK,CACb,aAAa,IAAI,wEAAwE,CAC1F,CAAC;QACJ,CAAC;QAED,sDAAsD;QACtD,IAAI,WAAW,IAAI,eAAe,EAAE,CAAC;YACnC,MAAM,IAAI,KAAK,CACb,aAAa,IAAI,0GAA0G,CAC5H,CAAC;QACJ,CAAC;QAED,kDAAkD;QAClD,MAAM,KAAK,GAAG,kBAAkB,CAAC,eAAe,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;QAErE,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE;YACxB,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,eAAe,EAAE,CAAC;gBAClC,OAAO,CAAC,IAAI,CAAC,yBAAyB,IAAI,EAAE,CAAC,CAAC;YAChD,CAAC;YACD,OAAO;gBACL,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,EAAE;oBACN,SAAS;oBACT,gBAAgB;oBAChB,IAAI;oBACJ,WAAW;oBACX,KAAK;oBACL,YAAY,EAAE,EAAE;iBACjB;aACF,CAAC;QACJ,CAAC,CAAC,CAAC;QACH,IAAI,eAAe,EAAE,CAAC;YACpB,aAAa,CAAC,eAAe,CAAC,CAAC;YAC/B,4BAA4B,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACtC,GAAG,IAAI;gBACP,CAAC,IAAI,CAAC,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;aACjD,CAAC,CAAC,CAAC;QACN,CAAC;IACH,CAAC,EACD,CAAC,aAAa,CAAC,CAChB,CAAC;IACF,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,cAAc,EAAE,CAAC;YACnB,cAAc,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;gBACnC,iBAAiB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;YACtC,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC,EAAE,CAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC,CAAC;IAExC,MAAM,KAAK,GAAG;QACZ,aAAa;QACb,YAAY;QACZ,yBAAyB;QACzB,iBAAiB;QACjB,YAAY;QACZ,aAAa;QACb,kBAAkB;KACnB,CAAC;IAEF,OAAO,CACL,8BAAC,oBAAoB,CAAC,QAAQ,IAAC,KAAK,EAAE,KAAK,IACxC,QAAQ,CACqB,CACjC,CAAC;AACJ,CAAC,CAAC;AA5HW,QAAA,qBAAqB,yBA4HhC;AAEF;;;;GAIG;AACI,MAAM,gBAAgB,GAAG,GAAG,EAAE;IACnC,OAAO,IAAA,kBAAU,EAAC,oBAAoB,CAAC,CAAC;AAC1C,CAAC,CAAC;AAFW,QAAA,gBAAgB,oBAE3B;AACF,SAAS,kBAAkB,CACzB,eAAoB,EACpB,WAAgB,EAChB,IAAY;IAEZ,IAAI,eAAe,EAAE,CAAC;QACpB,OAAO,CAAC,IAAI,CAAC,yDAAyD,CAAC,CAAC;QACxE,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,IAAI,WAAW,CAAC,WAAW,CAAC,EAAE,CAAC;QAC7B,IAAI,CAAC;YACH,OAAO,IAAA,4BAAe,EAAC,WAAW,CAAC,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CACX,oBAAoB,IAAI,+BAA+B,EACvD,KAAK,CACN,CAAC;QACJ,CAAC;IACH,CAAC;IACD,qFAAqF;IACrF,IAAI,YAAY,CAAC,WAAW,CAAC,EAAE,CAAC;QAC9B,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,4BAA4B,IAAI,EAAE,CAAC,CAAC;AACtD,CAAC;AAED;;;;;GAKG;AACH,SAAS,YAAY,CAAC,WAAgB;IACpC,OAAO,CACL,WAAW;QACX,OAAO,WAAW,KAAK,QAAQ;QAC/B,WAAW,CAAC,IAAI,KAAK,QAAQ;QAC7B,WAAW,CAAC,UAAU,CACvB,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,SAAS,WAAW,CAAC,GAAY;IAC/B,IAAI,GAAG,YAAY,eAAS,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IACD,6CAA6C;IAC7C,OAAO,CACL,OAAO,GAAG,KAAK,QAAQ;QACvB,GAAG,KAAK,IAAI;QACZ,OAAQ,GAAW,CAAC,SAAS,KAAK,UAAU;QAC5C,OAAQ,GAAW,CAAC,IAAI,KAAK,QAAQ,CACtC,CAAC;AACJ,CAAC","sourcesContent":["\"use client\";\nimport React, {\n  createContext,\n  PropsWithChildren,\n  useCallback,\n  useContext,\n  useEffect,\n  useState,\n} from \"react\";\nimport { ZodSchema } from \"zod\";\nimport zodToJsonSchema from \"zod-to-json-schema\";\nimport {\n  ComponentRegistry,\n  TamboComponent,\n  TamboTool,\n} from \"../model/component-metadata\";\n\nexport interface TamboRegistryContext {\n  componentList: ComponentRegistry;\n  toolRegistry: Record<string, TamboTool>;\n  componentToolAssociations: Record<string, string[]>;\n  registerComponent: (options: TamboComponent) => void;\n  registerTool: (tool: TamboTool) => void;\n  registerTools: (tools: TamboTool[]) => void;\n  addToolAssociation: (componentName: string, tool: TamboTool) => void;\n}\n\nconst TamboRegistryContext = createContext<TamboRegistryContext>({\n  componentList: {},\n  toolRegistry: {},\n  componentToolAssociations: {},\n  /**\n   *\n   */\n  registerComponent: () => {},\n  /**\n   *\n   */\n  registerTool: () => {},\n  /**\n   *\n   */\n  registerTools: () => {},\n  /**\n   *\n   */\n  addToolAssociation: () => {},\n});\n\nexport interface TamboRegistryProviderProps {\n  /** The components to register */\n  components?: TamboComponent[];\n}\n\n/**\n * The TamboRegistryProvider is a React provider that provides a component\n * registry to the descendants of the provider.\n * @param props - The props for the TamboRegistryProvider\n * @param props.children - The children to wrap\n * @param props.components - The components to register\n * @returns The TamboRegistryProvider component\n */\nexport const TamboRegistryProvider: React.FC<\n  PropsWithChildren<TamboRegistryProviderProps>\n> = ({ children, components: userComponents }) => {\n  const [componentList, setComponentList] = useState<ComponentRegistry>({});\n  const [toolRegistry, setToolRegistry] = useState<Record<string, TamboTool>>(\n    {},\n  );\n  const [componentToolAssociations, setComponentToolAssociations] = useState<\n    Record<string, string[]>\n  >({});\n\n  const registerTool = useCallback(\n    (tool: TamboTool, warnOnOverwrite = true) => {\n      setToolRegistry((prev) => {\n        if (prev[tool.name] && warnOnOverwrite) {\n          console.warn(`Overwriting tool ${tool.name}`);\n        }\n        return {\n          ...prev,\n          [tool.name]: tool,\n        };\n      });\n    },\n    [],\n  );\n\n  const registerTools = useCallback(\n    (tools: TamboTool[], warnOnOverwrite = true) => {\n      tools.forEach((tool) => registerTool(tool, warnOnOverwrite));\n    },\n    [registerTool],\n  );\n\n  const addToolAssociation = useCallback(\n    (componentName: string, tool: TamboTool) => {\n      if (!componentList[componentName]) {\n        throw new Error(`Component ${componentName} not found in registry`);\n      }\n      setComponentToolAssociations((prev) => ({\n        ...prev,\n        [componentName]: [...(prev[componentName] || []), tool.name],\n      }));\n    },\n    [componentList],\n  );\n\n  const registerComponent = useCallback(\n    (options: TamboComponent, warnOnOverwrite = true) => {\n      const {\n        name,\n        description,\n        component,\n        propsSchema,\n        propsDefinition,\n        loadingComponent,\n        associatedTools,\n      } = options;\n\n      // Validate that at least one props definition is provided\n      if (!propsSchema && !propsDefinition) {\n        throw new Error(\n          `Component ${name} must have either propsSchema (recommended) or propsDefinition defined`,\n        );\n      }\n\n      // Validate that only one props definition is provided\n      if (propsSchema && propsDefinition) {\n        throw new Error(\n          `Component ${name} cannot have both propsSchema and propsDefinition defined. Use only one. We recommend using propsSchema.`,\n        );\n      }\n\n      // Convert propsSchema to JSON Schema if it exists\n      const props = getSerializedProps(propsDefinition, propsSchema, name);\n\n      setComponentList((prev) => {\n        if (prev[name] && warnOnOverwrite) {\n          console.warn(`overwriting component ${name}`);\n        }\n        return {\n          ...prev,\n          [name]: {\n            component,\n            loadingComponent,\n            name,\n            description,\n            props,\n            contextTools: [],\n          },\n        };\n      });\n      if (associatedTools) {\n        registerTools(associatedTools);\n        setComponentToolAssociations((prev) => ({\n          ...prev,\n          [name]: associatedTools.map((tool) => tool.name),\n        }));\n      }\n    },\n    [registerTools],\n  );\n  useEffect(() => {\n    if (userComponents) {\n      userComponents.forEach((component) => {\n        registerComponent(component, false);\n      });\n    }\n  }, [registerComponent, userComponents]);\n\n  const value = {\n    componentList,\n    toolRegistry,\n    componentToolAssociations,\n    registerComponent,\n    registerTool,\n    registerTools,\n    addToolAssociation,\n  };\n\n  return (\n    <TamboRegistryContext.Provider value={value}>\n      {children}\n    </TamboRegistryContext.Provider>\n  );\n};\n\n/**\n * The useTamboRegistry hook provides access to the component registry\n * to the descendants of the TamboRegistryProvider.\n * @returns The component registry\n */\nexport const useTamboRegistry = () => {\n  return useContext(TamboRegistryContext);\n};\nfunction getSerializedProps(\n  propsDefinition: any,\n  propsSchema: any,\n  name: string,\n) {\n  if (propsDefinition) {\n    console.warn(`propsDefinition is deprecated. Use propsSchema instead.`);\n    return propsDefinition;\n  }\n\n  if (isZodSchema(propsSchema)) {\n    try {\n      return zodToJsonSchema(propsSchema);\n    } catch (error) {\n      console.error(\n        `Error converting ${name} props schema to JSON Schema:`,\n        error,\n      );\n    }\n  }\n  // try to roughly detect JSONSchema, should always be an object with a properties key\n  if (isJSONSchema(propsSchema)) {\n    return propsSchema;\n  }\n\n  throw new Error(`Invalid props schema for ${name}`);\n}\n\n/**\n * Checks if the propsSchema is a JSON Schema. This is a rough check, and the\n * server will provide the definitive check.\n * @param propsSchema - The props schema to check\n * @returns True if the props schema is a JSON Schema, false otherwise\n */\nfunction isJSONSchema(propsSchema: any) {\n  return (\n    propsSchema &&\n    typeof propsSchema === \"object\" &&\n    propsSchema.type === \"object\" &&\n    propsSchema.properties\n  );\n}\n\n/**\n * Since we require a certain zod version, we need to check if the object is a ZodSchema\n * @param obj - The object to check\n * @returns True if the object is a ZodSchema, false otherwise\n */\nfunction isZodSchema(obj: unknown): obj is ZodSchema {\n  if (obj instanceof ZodSchema) {\n    return true;\n  }\n  // try to detect if the object is a ZodSchema\n  return (\n    typeof obj === \"object\" &&\n    obj !== null &&\n    typeof (obj as any).safeParse === \"function\" &&\n    typeof (obj as any)._def === \"object\"\n  );\n}\n"]}

package/dist/setupTests.d.ts

@@ -0,0 +1,3 @@
+import "@tambo-ai/typescript-sdk/shims/node";
+import "@testing-library/jest-dom";
+//# sourceMappingURL=setupTests.d.ts.map

package/dist/setupTests.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"setupTests.d.ts","sourceRoot":"","sources":["../src/setupTests.ts"],"names":[],"mappings":"AAAA,OAAO,qCAAqC,CAAC;AAC7C,OAAO,2BAA2B,CAAC"}

package/dist/setupTests.js

@@ -0,0 +1,8 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+require("@tambo-ai/typescript-sdk/shims/node");
+require("@testing-library/jest-dom");
+// Mock Date.now() to return a fixed timestamp for consistent testing
+const mockDate = new Date(2025, 0, 5, 12, 32, 58, 936);
+global.Date.now = jest.fn(() => mockDate.getTime());
+//# sourceMappingURL=setupTests.js.map

package/dist/setupTests.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"setupTests.js","sourceRoot":"","sources":["../src/setupTests.ts"],"names":[],"mappings":";;AAAA,+CAA6C;AAC7C,qCAAmC;AAEnC,qEAAqE;AACrE,MAAM,QAAQ,GAAG,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;AACvD,MAAM,CAAC,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC","sourcesContent":["import \"@tambo-ai/typescript-sdk/shims/node\";\nimport \"@testing-library/jest-dom\";\n\n// Mock Date.now() to return a fixed timestamp for consistent testing\nconst mockDate = new Date(2025, 0, 5, 12, 32, 58, 936);\nglobal.Date.now = jest.fn(() => mockDate.getTime());\n"]}

package/dist/testing/tools.d.ts

@@ -0,0 +1,24 @@
+import { TamboComponent } from "../providers";
+/**
+ * Serializes the registry for testing purposes
+ * @param mockRegistry - The registry to serialize
+ * @returns The serialized registry
+ */
+export declare function serializeRegistry(mockRegistry: TamboComponent[]): {
+    props: import("zod-to-json-schema").JsonSchema7Type & {
+        $schema?: string | undefined;
+        definitions?: {
+            [key: string]: import("zod-to-json-schema").JsonSchema7Type;
+        } | undefined;
+    };
+    contextTools: {
+        parameters: any;
+        name: string;
+        description: string;
+    }[] | undefined;
+    name: string;
+    description: string;
+    propsDefinition?: any;
+    loadingComponent?: import("react").ComponentType<any>;
+}[];
+//# sourceMappingURL=tools.d.ts.map