@tambo-ai/react 0.48.0 → 0.49.0

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

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

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

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

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

@@ -0,0 +1,224 @@
+"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 zod_1 = require("zod");
+const tambo_registry_provider_1 = require("../tambo-registry-provider");
+// Shared tool registry for all tests
+const createMockTools = () => [
+    {
+        name: "test-tool-1",
+        description: "First test tool",
+        tool: jest.fn().mockResolvedValue("test-tool-1-result"),
+        toolSchema: zod_1.z
+            .function()
+            .args(zod_1.z.string().describe("input parameter"))
+            .returns(zod_1.z.string()),
+    },
+    {
+        name: "test-tool-2",
+        description: "Second test tool",
+        tool: jest.fn().mockResolvedValue("test-tool-2-result"),
+        toolSchema: zod_1.z
+            .function()
+            .args(zod_1.z.number().describe("number parameter"))
+            .returns(zod_1.z.string()),
+    },
+];
+const createMockComponents = (tools) => [
+    {
+        name: "TestComponent",
+        component: () => react_2.default.createElement("div", null, "TestComponent"),
+        description: "Test component",
+        propsSchema: zod_1.z.object({
+            test: zod_1.z.string(),
+        }),
+        associatedTools: tools,
+    },
+];
+describe("TamboRegistryProvider", () => {
+    const mockTools = createMockTools();
+    const mockComponents = createMockComponents(mockTools);
+    beforeEach(() => {
+        jest.clearAllMocks();
+    });
+    describe("Component and tool registration", () => {
+        it("should register components and tools correctly", () => {
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockComponents }, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            expect(result.current.componentList).toHaveProperty("TestComponent");
+            expect(result.current.toolRegistry).toHaveProperty("test-tool-1");
+            expect(result.current.toolRegistry).toHaveProperty("test-tool-2");
+            expect(result.current.componentToolAssociations).toHaveProperty("TestComponent");
+            expect(result.current.componentToolAssociations.TestComponent).toEqual([
+                "test-tool-1",
+                "test-tool-2",
+            ]);
+        });
+        it("should provide onCallUnregisteredTool in context", () => {
+            const mockonCallUnregisteredTool = jest
+                .fn()
+                .mockResolvedValue("test-result");
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockComponents, onCallUnregisteredTool: mockonCallUnregisteredTool }, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            // The onCallUnregisteredTool should be available in the context
+            expect(result.current.onCallUnregisteredTool).toBeDefined();
+            expect(typeof result.current.onCallUnregisteredTool).toBe("function");
+        });
+        it("should handle tool registration and association", () => {
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            // Register a new tool
+            (0, react_1.act)(() => {
+                result.current.registerTool(mockTools[0]);
+            });
+            expect(result.current.toolRegistry).toHaveProperty("test-tool-1");
+            // Register a new component
+            (0, react_1.act)(() => {
+                result.current.registerComponent(mockComponents[0]);
+            });
+            expect(result.current.componentList).toHaveProperty("TestComponent");
+            expect(result.current.componentToolAssociations).toHaveProperty("TestComponent");
+        });
+        it("should handle multiple tool registration", () => {
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            // Register multiple tools
+            (0, react_1.act)(() => {
+                result.current.registerTools(mockTools);
+            });
+            expect(result.current.toolRegistry).toHaveProperty("test-tool-1");
+            expect(result.current.toolRegistry).toHaveProperty("test-tool-2");
+            expect(Object.keys(result.current.toolRegistry)).toHaveLength(2);
+        });
+        it("should handle tool association with components", () => {
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockComponents }, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            // Add a new tool association
+            (0, react_1.act)(() => {
+                const newTool = {
+                    name: "new-tool",
+                    description: "New tool",
+                    tool: jest.fn().mockResolvedValue("new-tool-result"),
+                    toolSchema: zod_1.z
+                        .function()
+                        .args(zod_1.z.string().describe("input"))
+                        .returns(zod_1.z.string()),
+                };
+                result.current.addToolAssociation("TestComponent", newTool);
+            });
+            expect(result.current.componentToolAssociations.TestComponent).toContain("new-tool");
+        });
+        it("should throw error when adding tool association to non-existent component", () => {
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            const newTool = {
+                name: "new-tool",
+                description: "New tool",
+                tool: jest.fn().mockResolvedValue("new-tool-result"),
+                toolSchema: zod_1.z
+                    .function()
+                    .args(zod_1.z.string().describe("input"))
+                    .returns(zod_1.z.string()),
+            };
+            expect(() => {
+                (0, react_1.act)(() => {
+                    result.current.addToolAssociation("NonExistentComponent", newTool);
+                });
+            }).toThrow("Component NonExistentComponent not found in registry");
+        });
+        it("should validate tool schemas and throw error for invalid schemas", () => {
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            const invalidTool = {
+                name: "invalid-tool",
+                description: "Invalid tool",
+                tool: jest.fn().mockResolvedValue("result"),
+                toolSchema: zod_1.z.record(zod_1.z.string()), // This should cause validation to fail
+            };
+            // This should throw during registration due to invalid schema
+            expect(() => {
+                (0, react_1.act)(() => {
+                    result.current.registerTool(invalidTool);
+                });
+            }).toThrow('z.record() is not supported in toolSchema of tool "invalid-tool"');
+        });
+        it("should validate component schemas and throw error for invalid schemas", () => {
+            const invalidComponent = {
+                name: "InvalidComponent",
+                component: () => react_2.default.createElement("div", null, "Invalid"),
+                description: "Invalid component",
+                propsSchema: zod_1.z.record(zod_1.z.string()), // This should cause validation to fail
+            };
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            // This should throw during registration due to invalid schema
+            expect(() => {
+                (0, react_1.act)(() => {
+                    result.current.registerComponent(invalidComponent);
+                });
+            }).toThrow('z.record() is not supported in propsSchema of component "InvalidComponent"');
+        });
+        it("should throw error when component has both propsSchema and propsDefinition", () => {
+            const invalidComponent = {
+                name: "InvalidComponent",
+                component: () => react_2.default.createElement("div", null, "Invalid"),
+                description: "Invalid component",
+                propsSchema: zod_1.z.object({ test: zod_1.z.string() }),
+                propsDefinition: { test: "string" }, // Both defined - should throw
+            };
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            expect(() => {
+                (0, react_1.act)(() => {
+                    result.current.registerComponent(invalidComponent);
+                });
+            }).toThrow("Component InvalidComponent cannot have both propsSchema and propsDefinition defined");
+        });
+        it("should throw error when component has neither propsSchema nor propsDefinition", () => {
+            const invalidComponent = {
+                name: "InvalidComponent",
+                component: () => react_2.default.createElement("div", null, "Invalid"),
+                description: "Invalid component",
+                // Neither propsSchema nor propsDefinition defined - should throw
+            };
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            expect(() => {
+                (0, react_1.act)(() => {
+                    result.current.registerComponent(invalidComponent);
+                });
+            }).toThrow("Component InvalidComponent must have either propsSchema (recommended) or propsDefinition defined");
+        });
+    });
+    describe("Tool call handling", () => {
+        it("should provide onCallUnregisteredTool callback for handling unknown tools", async () => {
+            const mockonCallUnregisteredTool = jest
+                .fn()
+                .mockResolvedValue("unknown-tool-result");
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockComponents, onCallUnregisteredTool: mockonCallUnregisteredTool }, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            // Verify the callback is available
+            expect(result.current.onCallUnregisteredTool).toBeDefined();
+            expect(typeof result.current.onCallUnregisteredTool).toBe("function");
+            // Simulate calling the unknown tool handler
+            const toolName = "unknown-tool";
+            const args = [{ parameterName: "input", parameterValue: "test-input" }];
+            await (0, react_1.act)(async () => {
+                if (result.current.onCallUnregisteredTool) {
+                    await result.current.onCallUnregisteredTool(toolName, args);
+                }
+            });
+            expect(mockonCallUnregisteredTool).toHaveBeenCalledWith(toolName, args);
+        });
+        it("should handle onCallUnregisteredTool being undefined", () => {
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockComponents }, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            expect(result.current.onCallUnregisteredTool).toBeUndefined();
+        });
+    });
+});
+//# sourceMappingURL=tambo-registry-provider.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tambo-registry-provider.test.js","sourceRoot":"","sources":["../../../src/providers/__tests__/tambo-registry-provider.test.tsx"],"names":[],"mappings":";;;;;AAAA,kDAAyD;AACzD,kDAA0B;AAC1B,6BAAwB;AAExB,wEAGoC;AAEpC,qCAAqC;AACrC,MAAM,eAAe,GAAG,GAAgB,EAAE,CAAC;IACzC;QACE,IAAI,EAAE,aAAa;QACnB,WAAW,EAAE,iBAAiB;QAC9B,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,oBAAoB,CAAC;QACvD,UAAU,EAAE,OAAC;aACV,QAAQ,EAAE;aACV,IAAI,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;aAC5C,OAAO,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC;KACvB;IACD;QACE,IAAI,EAAE,aAAa;QACnB,WAAW,EAAE,kBAAkB;QAC/B,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,oBAAoB,CAAC;QACvD,UAAU,EAAE,OAAC;aACV,QAAQ,EAAE;aACV,IAAI,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC;aAC7C,OAAO,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC;KACvB;CACF,CAAC;AAEF,MAAM,oBAAoB,GAAG,CAAC,KAAkB,EAAoB,EAAE,CAAC;IACrE;QACE,IAAI,EAAE,eAAe;QACrB,SAAS,EAAE,GAAG,EAAE,CAAC,2DAAwB;QACzC,WAAW,EAAE,gBAAgB;QAC7B,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;YACpB,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;SACjB,CAAC;QACF,eAAe,EAAE,KAAK;KACvB;CACF,CAAC;AAEF,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;IACrC,MAAM,SAAS,GAAG,eAAe,EAAE,CAAC;IACpC,MAAM,cAAc,GAAG,oBAAoB,CAAC,SAAS,CAAC,CAAC;IAEvD,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC/C,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,IAAC,UAAU,EAAE,cAAc,IAC9C,QAAQ,CACa,CACzB,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC,cAAc,CAC7D,eAAe,CAChB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,yBAAyB,CAAC,aAAa,CAAC,CAAC,OAAO,CAAC;gBACrE,aAAa;gBACb,aAAa;aACd,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,0BAA0B,GAAG,IAAI;iBACpC,EAAE,EAAE;iBACJ,iBAAiB,CAAC,aAAa,CAAC,CAAC;YAEpC,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,IACpB,UAAU,EAAE,cAAc,EAC1B,sBAAsB,EAAE,0BAA0B,IAEjD,QAAQ,CACa,CACzB,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,gEAAgE;YAChE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC,WAAW,EAAE,CAAC;YAC5D,MAAM,CAAC,OAAO,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,QAAE,QAAQ,CAAyB,CAC1D,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,sBAAsB;YACtB,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;YAC5C,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YAElE,2BAA2B;YAC3B,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;YACtD,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAC,cAAc,CAC7D,eAAe,CAChB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,QAAE,QAAQ,CAAyB,CAC1D,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,0BAA0B;YAC1B,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;YAC1C,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,cAAc,CAAC,aAAa,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,IAAC,UAAU,EAAE,cAAc,IAC9C,QAAQ,CACa,CACzB,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,6BAA6B;YAC7B,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,OAAO,GAAc;oBACzB,IAAI,EAAE,UAAU;oBAChB,WAAW,EAAE,UAAU;oBACvB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,iBAAiB,CAAC;oBACpD,UAAU,EAAE,OAAC;yBACV,QAAQ,EAAE;yBACV,IAAI,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;yBAClC,OAAO,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC;iBACvB,CAAC;gBACF,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;YAC9D,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,yBAAyB,CAAC,aAAa,CAAC,CAAC,SAAS,CACtE,UAAU,CACX,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2EAA2E,EAAE,GAAG,EAAE;YACnF,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,QAAE,QAAQ,CAAyB,CAC1D,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,MAAM,OAAO,GAAc;gBACzB,IAAI,EAAE,UAAU;gBAChB,WAAW,EAAE,UAAU;gBACvB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,iBAAiB,CAAC;gBACpD,UAAU,EAAE,OAAC;qBACV,QAAQ,EAAE;qBACV,IAAI,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;qBAClC,OAAO,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC;aACvB,CAAC;YAEF,MAAM,CAAC,GAAG,EAAE;gBACV,IAAA,WAAG,EAAC,GAAG,EAAE;oBACP,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,sBAAsB,EAAE,OAAO,CAAC,CAAC;gBACrE,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,OAAO,CAAC,sDAAsD,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;YAC1E,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,QAAE,QAAQ,CAAyB,CAC1D,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,MAAM,WAAW,GAAc;gBAC7B,IAAI,EAAE,cAAc;gBACpB,WAAW,EAAE,cAAc;gBAC3B,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC;gBAC3C,UAAU,EAAE,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,EAAE,uCAAuC;aAC1E,CAAC;YAEF,8DAA8D;YAC9D,MAAM,CAAC,GAAG,EAAE;gBACV,IAAA,WAAG,EAAC,GAAG,EAAE;oBACP,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC;gBAC3C,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,OAAO,CACR,kEAAkE,CACnE,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;YAC/E,MAAM,gBAAgB,GAAmB;gBACvC,IAAI,EAAE,kBAAkB;gBACxB,SAAS,EAAE,GAAG,EAAE,CAAC,qDAAkB;gBACnC,WAAW,EAAE,mBAAmB;gBAChC,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,EAAE,uCAAuC;aAC3E,CAAC;YAEF,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,QAAE,QAAQ,CAAyB,CAC1D,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,8DAA8D;YAC9D,MAAM,CAAC,GAAG,EAAE;gBACV,IAAA,WAAG,EAAC,GAAG,EAAE;oBACP,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;gBACrD,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,OAAO,CACR,4EAA4E,CAC7E,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4EAA4E,EAAE,GAAG,EAAE;YACpF,MAAM,gBAAgB,GAAmB;gBACvC,IAAI,EAAE,kBAAkB;gBACxB,SAAS,EAAE,GAAG,EAAE,CAAC,qDAAkB;gBACnC,WAAW,EAAE,mBAAmB;gBAChC,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,EAAE,CAAC;gBAC3C,eAAe,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,8BAA8B;aACpE,CAAC;YAEF,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,QAAE,QAAQ,CAAyB,CAC1D,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,MAAM,CAAC,GAAG,EAAE;gBACV,IAAA,WAAG,EAAC,GAAG,EAAE;oBACP,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;gBACrD,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,OAAO,CACR,qFAAqF,CACtF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+EAA+E,EAAE,GAAG,EAAE;YACvF,MAAM,gBAAgB,GAAmB;gBACvC,IAAI,EAAE,kBAAkB;gBACxB,SAAS,EAAE,GAAG,EAAE,CAAC,qDAAkB;gBACnC,WAAW,EAAE,mBAAmB;gBAChC,iEAAiE;aAClE,CAAC;YAEF,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,QAAE,QAAQ,CAAyB,CAC1D,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,MAAM,CAAC,GAAG,EAAE;gBACV,IAAA,WAAG,EAAC,GAAG,EAAE;oBACP,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;gBACrD,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,OAAO,CACR,kGAAkG,CACnG,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;YACzF,MAAM,0BAA0B,GAAG,IAAI;iBACpC,EAAE,EAAE;iBACJ,iBAAiB,CAAC,qBAAqB,CAAC,CAAC;YAE5C,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,IACpB,UAAU,EAAE,cAAc,EAC1B,sBAAsB,EAAE,0BAA0B,IAEjD,QAAQ,CACa,CACzB,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,mCAAmC;YACnC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC,WAAW,EAAE,CAAC;YAC5D,MAAM,CAAC,OAAO,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAEtE,4CAA4C;YAC5C,MAAM,QAAQ,GAAG,cAAc,CAAC;YAChC,MAAM,IAAI,GAAG,CAAC,EAAE,aAAa,EAAE,OAAO,EAAE,cAAc,EAAE,YAAY,EAAE,CAAC,CAAC;YAExE,MAAM,IAAA,WAAG,EAAC,KAAK,IAAI,EAAE;gBACnB,IAAI,MAAM,CAAC,OAAO,CAAC,sBAAsB,EAAE,CAAC;oBAC1C,MAAM,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;gBAC9D,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,0BAA0B,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,IAAC,UAAU,EAAE,cAAc,IAC9C,QAAQ,CACa,CACzB,CAAC;YAEF,MAAM,EAAE,MAAM,EAAE,GAAG,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAErE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC,aAAa,EAAE,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { act, renderHook } from \"@testing-library/react\";\nimport React from \"react\";\nimport { z } from \"zod\";\nimport { TamboComponent, TamboTool } from \"../../model/component-metadata\";\nimport {\n  TamboRegistryProvider,\n  useTamboRegistry,\n} from \"../tambo-registry-provider\";\n\n// Shared tool registry for all tests\nconst createMockTools = (): TamboTool[] => [\n  {\n    name: \"test-tool-1\",\n    description: \"First test tool\",\n    tool: jest.fn().mockResolvedValue(\"test-tool-1-result\"),\n    toolSchema: z\n      .function()\n      .args(z.string().describe(\"input parameter\"))\n      .returns(z.string()),\n  },\n  {\n    name: \"test-tool-2\",\n    description: \"Second test tool\",\n    tool: jest.fn().mockResolvedValue(\"test-tool-2-result\"),\n    toolSchema: z\n      .function()\n      .args(z.number().describe(\"number parameter\"))\n      .returns(z.string()),\n  },\n];\n\nconst createMockComponents = (tools: TamboTool[]): TamboComponent[] => [\n  {\n    name: \"TestComponent\",\n    component: () => <div>TestComponent</div>,\n    description: \"Test component\",\n    propsSchema: z.object({\n      test: z.string(),\n    }),\n    associatedTools: tools,\n  },\n];\n\ndescribe(\"TamboRegistryProvider\", () => {\n  const mockTools = createMockTools();\n  const mockComponents = createMockComponents(mockTools);\n\n  beforeEach(() => {\n    jest.clearAllMocks();\n  });\n\n  describe(\"Component and tool registration\", () => {\n    it(\"should register components and tools correctly\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider components={mockComponents}>\n          {children}\n        </TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      expect(result.current.componentList).toHaveProperty(\"TestComponent\");\n      expect(result.current.toolRegistry).toHaveProperty(\"test-tool-1\");\n      expect(result.current.toolRegistry).toHaveProperty(\"test-tool-2\");\n      expect(result.current.componentToolAssociations).toHaveProperty(\n        \"TestComponent\",\n      );\n      expect(result.current.componentToolAssociations.TestComponent).toEqual([\n        \"test-tool-1\",\n        \"test-tool-2\",\n      ]);\n    });\n\n    it(\"should provide onCallUnregisteredTool in context\", () => {\n      const mockonCallUnregisteredTool = jest\n        .fn()\n        .mockResolvedValue(\"test-result\");\n\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider\n          components={mockComponents}\n          onCallUnregisteredTool={mockonCallUnregisteredTool}\n        >\n          {children}\n        </TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      // The onCallUnregisteredTool should be available in the context\n      expect(result.current.onCallUnregisteredTool).toBeDefined();\n      expect(typeof result.current.onCallUnregisteredTool).toBe(\"function\");\n    });\n\n    it(\"should handle tool registration and association\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      // Register a new tool\n      act(() => {\n        result.current.registerTool(mockTools[0]);\n      });\n\n      expect(result.current.toolRegistry).toHaveProperty(\"test-tool-1\");\n\n      // Register a new component\n      act(() => {\n        result.current.registerComponent(mockComponents[0]);\n      });\n\n      expect(result.current.componentList).toHaveProperty(\"TestComponent\");\n      expect(result.current.componentToolAssociations).toHaveProperty(\n        \"TestComponent\",\n      );\n    });\n\n    it(\"should handle multiple tool registration\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      // Register multiple tools\n      act(() => {\n        result.current.registerTools(mockTools);\n      });\n\n      expect(result.current.toolRegistry).toHaveProperty(\"test-tool-1\");\n      expect(result.current.toolRegistry).toHaveProperty(\"test-tool-2\");\n      expect(Object.keys(result.current.toolRegistry)).toHaveLength(2);\n    });\n\n    it(\"should handle tool association with components\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider components={mockComponents}>\n          {children}\n        </TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      // Add a new tool association\n      act(() => {\n        const newTool: TamboTool = {\n          name: \"new-tool\",\n          description: \"New tool\",\n          tool: jest.fn().mockResolvedValue(\"new-tool-result\"),\n          toolSchema: z\n            .function()\n            .args(z.string().describe(\"input\"))\n            .returns(z.string()),\n        };\n        result.current.addToolAssociation(\"TestComponent\", newTool);\n      });\n\n      expect(result.current.componentToolAssociations.TestComponent).toContain(\n        \"new-tool\",\n      );\n    });\n\n    it(\"should throw error when adding tool association to non-existent component\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      const newTool: TamboTool = {\n        name: \"new-tool\",\n        description: \"New tool\",\n        tool: jest.fn().mockResolvedValue(\"new-tool-result\"),\n        toolSchema: z\n          .function()\n          .args(z.string().describe(\"input\"))\n          .returns(z.string()),\n      };\n\n      expect(() => {\n        act(() => {\n          result.current.addToolAssociation(\"NonExistentComponent\", newTool);\n        });\n      }).toThrow(\"Component NonExistentComponent not found in registry\");\n    });\n\n    it(\"should validate tool schemas and throw error for invalid schemas\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      const invalidTool: TamboTool = {\n        name: \"invalid-tool\",\n        description: \"Invalid tool\",\n        tool: jest.fn().mockResolvedValue(\"result\"),\n        toolSchema: z.record(z.string()), // This should cause validation to fail\n      };\n\n      // This should throw during registration due to invalid schema\n      expect(() => {\n        act(() => {\n          result.current.registerTool(invalidTool);\n        });\n      }).toThrow(\n        'z.record() is not supported in toolSchema of tool \"invalid-tool\"',\n      );\n    });\n\n    it(\"should validate component schemas and throw error for invalid schemas\", () => {\n      const invalidComponent: TamboComponent = {\n        name: \"InvalidComponent\",\n        component: () => <div>Invalid</div>,\n        description: \"Invalid component\",\n        propsSchema: z.record(z.string()), // This should cause validation to fail\n      };\n\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      // This should throw during registration due to invalid schema\n      expect(() => {\n        act(() => {\n          result.current.registerComponent(invalidComponent);\n        });\n      }).toThrow(\n        'z.record() is not supported in propsSchema of component \"InvalidComponent\"',\n      );\n    });\n\n    it(\"should throw error when component has both propsSchema and propsDefinition\", () => {\n      const invalidComponent: TamboComponent = {\n        name: \"InvalidComponent\",\n        component: () => <div>Invalid</div>,\n        description: \"Invalid component\",\n        propsSchema: z.object({ test: z.string() }),\n        propsDefinition: { test: \"string\" }, // Both defined - should throw\n      };\n\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      expect(() => {\n        act(() => {\n          result.current.registerComponent(invalidComponent);\n        });\n      }).toThrow(\n        \"Component InvalidComponent cannot have both propsSchema and propsDefinition defined\",\n      );\n    });\n\n    it(\"should throw error when component has neither propsSchema nor propsDefinition\", () => {\n      const invalidComponent: TamboComponent = {\n        name: \"InvalidComponent\",\n        component: () => <div>Invalid</div>,\n        description: \"Invalid component\",\n        // Neither propsSchema nor propsDefinition defined - should throw\n      };\n\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      expect(() => {\n        act(() => {\n          result.current.registerComponent(invalidComponent);\n        });\n      }).toThrow(\n        \"Component InvalidComponent must have either propsSchema (recommended) or propsDefinition defined\",\n      );\n    });\n  });\n\n  describe(\"Tool call handling\", () => {\n    it(\"should provide onCallUnregisteredTool callback for handling unknown tools\", async () => {\n      const mockonCallUnregisteredTool = jest\n        .fn()\n        .mockResolvedValue(\"unknown-tool-result\");\n\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider\n          components={mockComponents}\n          onCallUnregisteredTool={mockonCallUnregisteredTool}\n        >\n          {children}\n        </TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      // Verify the callback is available\n      expect(result.current.onCallUnregisteredTool).toBeDefined();\n      expect(typeof result.current.onCallUnregisteredTool).toBe(\"function\");\n\n      // Simulate calling the unknown tool handler\n      const toolName = \"unknown-tool\";\n      const args = [{ parameterName: \"input\", parameterValue: \"test-input\" }];\n\n      await act(async () => {\n        if (result.current.onCallUnregisteredTool) {\n          await result.current.onCallUnregisteredTool(toolName, args);\n        }\n      });\n\n      expect(mockonCallUnregisteredTool).toHaveBeenCalledWith(toolName, args);\n    });\n\n    it(\"should handle onCallUnregisteredTool being undefined\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider components={mockComponents}>\n          {children}\n        </TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      expect(result.current.onCallUnregisteredTool).toBeUndefined();\n    });\n  });\n});\n"]}

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

@@ -304,6 +304,105 @@ describe("TamboThreadProvider", () => {
         expect(result.current.generationStage).toBe(generate_component_response_1.GenerationStage.COMPLETE);
         expect(mockRegistry[0]?.associatedTools?.[0]?.tool).toHaveBeenCalledWith("test");
     });
+    it("should handle unregistered tool calls with onCallUnregisteredTool", async () => {
+        const mockOnCallUnregisteredTool = jest
+            .fn()
+            .mockResolvedValue("unregistered-tool-result");
+        const wrapperWithUnregisteredTool = ({ children, }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry, onCallUnregisteredTool: mockOnCallUnregisteredTool },
+            react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
+                    currentTimeContextHelper: () => null,
+                    currentPageContextHelper: () => null,
+                } },
+                react_2.default.createElement(tambo_thread_provider_1.TamboThreadProvider, { streaming: false }, children))));
+        const mockUnregisteredToolCallResponse = {
+            responseMessageDto: {
+                id: "unregistered-tool-call-1",
+                content: [{ type: "text", text: "Unregistered tool response" }],
+                role: "tool",
+                threadId: "test-thread-1",
+                toolCallRequest: {
+                    toolName: "unregistered-tool",
+                    parameters: [
+                        { parameterName: "input", parameterValue: "test-input" },
+                    ],
+                },
+                componentState: {},
+                createdAt: new Date().toISOString(),
+            },
+            generationStage: generate_component_response_1.GenerationStage.COMPLETE,
+            mcpAccessToken: "test-mcp-access-token",
+        };
+        jest
+            .mocked(mockThreadsApi.advanceById)
+            .mockResolvedValueOnce(mockUnregisteredToolCallResponse)
+            .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,
+            mcpAccessToken: "test-mcp-access-token",
+        });
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), {
+            wrapper: wrapperWithUnregisteredTool,
+        });
+        await (0, react_1.act)(async () => {
+            await result.current.sendThreadMessage("Use unregistered tool", {
+                threadId: "test-thread-1",
+                streamResponse: false,
+            });
+        });
+        expect(result.current.generationStage).toBe(generate_component_response_1.GenerationStage.COMPLETE);
+        expect(mockOnCallUnregisteredTool).toHaveBeenCalledWith("unregistered-tool", [{ parameterName: "input", parameterValue: "test-input" }]);
+    });
+    it("should handle unregistered tool calls without onCallUnregisteredTool", async () => {
+        const mockUnregisteredToolCallResponse = {
+            responseMessageDto: {
+                id: "unregistered-tool-call-1",
+                content: [{ type: "text", text: "Unregistered tool response" }],
+                role: "tool",
+                threadId: "test-thread-1",
+                toolCallRequest: {
+                    toolName: "unregistered-tool",
+                    parameters: [
+                        { parameterName: "input", parameterValue: "test-input" },
+                    ],
+                },
+                componentState: {},
+                createdAt: new Date().toISOString(),
+            },
+            generationStage: generate_component_response_1.GenerationStage.COMPLETE,
+            mcpAccessToken: "test-mcp-access-token",
+        };
+        jest
+            .mocked(mockThreadsApi.advanceById)
+            .mockResolvedValueOnce(mockUnregisteredToolCallResponse)
+            .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,
+            mcpAccessToken: "test-mcp-access-token",
+        });
+        const { result } = (0, react_1.renderHook)(() => (0, tambo_thread_provider_1.useTamboThread)(), { wrapper });
+        await (0, react_1.act)(async () => {
+            await result.current.sendThreadMessage("Use unregistered tool", {
+                threadId: "test-thread-1",
+                streamResponse: false,
+            });
+        });
+        expect(result.current.generationStage).toBe(generate_component_response_1.GenerationStage.COMPLETE);
+        // Should not throw an error, but the tool call should fail gracefully
+    });
     describe("streaming behavior", () => {
         it("should call advanceStream when streamResponse=true", async () => {
             // Use wrapper with streaming=true to show that explicit streamResponse=true works