@tambo-ai/react 0.65.3 → 0.66.1

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

@@ -5,36 +5,33 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 const react_1 = require("@testing-library/react");
 const react_2 = __importDefault(require("react"));
-const v3_1 = require("zod/v3");
+const zodInternalAlias_1 = require("zodInternalAlias");
+const registry_1 = require("../util/registry");
 const tambo_registry_provider_1 = require("./tambo-registry-provider");
 // Shared tool registry for all tests
 const createMockTools = () => [
-    {
+    (0, registry_1.defineTool)({
         name: "test-tool-1",
         description: "First test tool",
         tool: jest.fn().mockResolvedValue("test-tool-1-result"),
-        toolSchema: v3_1.z
-            .function()
-            .args(v3_1.z.string().describe("input parameter"))
-            .returns(v3_1.z.string()),
-    },
-    {
+        inputSchema: zodInternalAlias_1.z.tuple([zodInternalAlias_1.z.string().describe("input parameter")]),
+        outputSchema: zodInternalAlias_1.z.string(),
+    }),
+    (0, registry_1.defineTool)({
         name: "test-tool-2",
         description: "Second test tool",
         tool: jest.fn().mockResolvedValue("test-tool-2-result"),
-        toolSchema: v3_1.z
-            .function()
-            .args(v3_1.z.number().describe("number parameter"))
-            .returns(v3_1.z.string()),
-    },
+        inputSchema: zodInternalAlias_1.z.tuple([zodInternalAlias_1.z.number().describe("number parameter")]),
+        outputSchema: zodInternalAlias_1.z.string(),
+    }),
 ];
 const createMockComponents = (tools) => [
     {
         name: "TestComponent",
         component: () => react_2.default.createElement("div", null, "TestComponent"),
         description: "Test component",
-        propsSchema: v3_1.z.object({
-            test: v3_1.z.string(),
+        propsSchema: zodInternalAlias_1.z.object({
+            test: zodInternalAlias_1.z.string(),
         }),
         associatedTools: tools,
     },
@@ -102,10 +99,8 @@ describe("TamboRegistryProvider", () => {
                 name: "new-tool",
                 description: "New tool",
                 tool: jest.fn().mockResolvedValue("new-tool-result"),
-                toolSchema: v3_1.z
-                    .function()
-                    .args(v3_1.z.string().describe("input"))
-                    .returns(v3_1.z.string()),
+                inputSchema: zodInternalAlias_1.z.tuple([zodInternalAlias_1.z.string().describe("input")]),
+                outputSchema: zodInternalAlias_1.z.string(),
             };
             // First register the tool
             (0, react_1.act)(() => {
@@ -124,10 +119,8 @@ describe("TamboRegistryProvider", () => {
                 name: "new-tool",
                 description: "New tool",
                 tool: jest.fn().mockResolvedValue("new-tool-result"),
-                toolSchema: v3_1.z
-                    .function()
-                    .args(v3_1.z.string().describe("input"))
-                    .returns(v3_1.z.string()),
+                inputSchema: zodInternalAlias_1.z.tuple([zodInternalAlias_1.z.string().describe("input")]),
+                outputSchema: zodInternalAlias_1.z.string(),
             };
             expect(() => {
                 (0, react_1.act)(() => {
@@ -142,21 +135,22 @@ describe("TamboRegistryProvider", () => {
                 name: "invalid-tool",
                 description: "Invalid tool",
                 tool: jest.fn().mockResolvedValue("result"),
-                toolSchema: v3_1.z.record(v3_1.z.string()), // This should cause validation to fail
+                inputSchema: zodInternalAlias_1.z.record(zodInternalAlias_1.z.string(), zodInternalAlias_1.z.string()), // This should cause validation to fail
+                outputSchema: zodInternalAlias_1.z.string(),
             };
             // 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"');
+            }).toThrow('Record types (objects with dynamic keys) are not supported in inputSchema of tool "invalid-tool". Found at path "(root)". Replace it with an object using explicit keys.');
         });
         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: v3_1.z.record(v3_1.z.string()), // This should cause validation to fail
+                propsSchema: zodInternalAlias_1.z.record(zodInternalAlias_1.z.string(), zodInternalAlias_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 });
@@ -165,14 +159,14 @@ describe("TamboRegistryProvider", () => {
                 (0, react_1.act)(() => {
                     result.current.registerComponent(invalidComponent);
                 });
-            }).toThrow('z.record() is not supported in propsSchema of component "InvalidComponent"');
+            }).toThrow('Record types (objects with dynamic keys) are 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: v3_1.z.object({ test: v3_1.z.string() }),
+                propsSchema: zodInternalAlias_1.z.object({ test: zodInternalAlias_1.z.string() }),
                 propsDefinition: { test: "string" }, // Both defined - should throw
             };
             const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null, children));
@@ -205,7 +199,8 @@ describe("TamboRegistryProvider", () => {
                 name: "invalid tool name", // Contains spaces
                 description: "Tool with spaces in name",
                 tool: jest.fn().mockResolvedValue("result"),
-                toolSchema: v3_1.z.function().args(v3_1.z.string()).returns(v3_1.z.string()),
+                inputSchema: zodInternalAlias_1.z.tuple([zodInternalAlias_1.z.string()]),
+                outputSchema: zodInternalAlias_1.z.string(),
             };
             expect(() => {
                 (0, react_1.act)(() => {
@@ -218,7 +213,7 @@ describe("TamboRegistryProvider", () => {
                 name: "Invalid Component Name", // Contains spaces
                 component: () => react_2.default.createElement("div", null, "Invalid"),
                 description: "Component with spaces in name",
-                propsSchema: v3_1.z.object({ test: v3_1.z.string() }),
+                propsSchema: zodInternalAlias_1.z.object({ test: zodInternalAlias_1.z.string() }),
             };
             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 });
@@ -255,5 +250,132 @@ describe("TamboRegistryProvider", () => {
             expect(result.current.onCallUnregisteredTool).toBeUndefined();
         });
     });
+    describe("Resource registration", () => {
+        it("should register static resources via props", () => {
+            const staticResources = [
+                {
+                    uri: "file:///local/doc.txt",
+                    name: "Local Document",
+                    mimeType: "text/plain",
+                },
+                {
+                    uri: "file:///local/image.png",
+                    name: "Local Image",
+                    mimeType: "image/png",
+                },
+            ];
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { resources: staticResources }, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            expect(result.current.resources).toHaveLength(2);
+            expect(result.current.resources[0].uri).toBe("file:///local/doc.txt");
+            expect(result.current.resources[1].uri).toBe("file:///local/image.png");
+        });
+        it("should register resources dynamically via registerResource", () => {
+            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 newResource = {
+                uri: "file:///dynamic/doc.txt",
+                name: "Dynamic Document",
+                mimeType: "text/plain",
+            };
+            (0, react_1.act)(() => {
+                result.current.registerResource(newResource);
+            });
+            expect(result.current.resources).toHaveLength(1);
+            expect(result.current.resources[0].uri).toBe("file:///dynamic/doc.txt");
+        });
+        it("should register multiple resources via registerResources", () => {
+            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 newResources = [
+                {
+                    uri: "file:///batch/doc1.txt",
+                    name: "Batch Document 1",
+                    mimeType: "text/plain",
+                },
+                {
+                    uri: "file:///batch/doc2.txt",
+                    name: "Batch Document 2",
+                    mimeType: "text/plain",
+                },
+            ];
+            (0, react_1.act)(() => {
+                result.current.registerResources(newResources);
+            });
+            expect(result.current.resources).toHaveLength(2);
+            expect(result.current.resources[0].uri).toBe("file:///batch/doc1.txt");
+            expect(result.current.resources[1].uri).toBe("file:///batch/doc2.txt");
+        });
+        it("should register resource source via props", () => {
+            const mockListResources = jest.fn().mockResolvedValue([]);
+            const mockGetResource = jest.fn().mockResolvedValue({
+                contents: [{ uri: "", mimeType: "text/plain", text: "" }],
+            });
+            const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { listResources: mockListResources, getResource: mockGetResource }, children));
+            const { result } = (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            expect(result.current.resourceSource).not.toBeNull();
+            expect(result.current.resourceSource?.listResources).toBe(mockListResources);
+            expect(result.current.resourceSource?.getResource).toBe(mockGetResource);
+        });
+        it("should register resource source via registerResourceSource", () => {
+            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 mockListResources = jest.fn().mockResolvedValue([]);
+            const mockGetResource = jest.fn().mockResolvedValue({
+                contents: [{ uri: "", mimeType: "text/plain", text: "" }],
+            });
+            (0, react_1.act)(() => {
+                result.current.registerResourceSource({
+                    listResources: mockListResources,
+                    getResource: mockGetResource,
+                });
+            });
+            expect(result.current.resourceSource).not.toBeNull();
+            expect(result.current.resourceSource?.listResources).toBe(mockListResources);
+            expect(result.current.resourceSource?.getResource).toBe(mockGetResource);
+        });
+        it("should throw error when only listResources is provided", () => {
+            const mockListResources = jest.fn().mockResolvedValue([]);
+            expect(() => {
+                const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { listResources: mockListResources }, children));
+                (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            }).toThrow("Both listResources and getResource must be provided together");
+        });
+        it("should throw error when only getResource is provided", () => {
+            const mockGetResource = jest.fn().mockResolvedValue({
+                contents: [{ uri: "", mimeType: "text/plain", text: "" }],
+            });
+            expect(() => {
+                const wrapper = ({ children }) => (react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { getResource: mockGetResource }, children));
+                (0, react_1.renderHook)(() => (0, tambo_registry_provider_1.useTamboRegistry)(), { wrapper });
+            }).toThrow("Both listResources and getResource must be provided together");
+        });
+        it("should validate resource has required uri field", () => {
+            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 invalidResource = {
+                name: "Document without URI",
+                mimeType: "text/plain",
+            };
+            expect(() => {
+                (0, react_1.act)(() => {
+                    result.current.registerResource(invalidResource);
+                });
+            }).toThrow("Resource must have a 'uri' field");
+        });
+        it("should validate resource has required name field", () => {
+            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 invalidResource = {
+                uri: "file:///doc.txt",
+                mimeType: "text/plain",
+            };
+            expect(() => {
+                (0, react_1.act)(() => {
+                    result.current.registerResource(invalidResource);
+                });
+            }).toThrow("Resource with URI 'file:///doc.txt' must have a 'name' field");
+        });
+    });
 });
 //# sourceMappingURL=tambo-registry-provider.test.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"tambo-registry-provider.test.js","sourceRoot":"","sources":["../../src/providers/tambo-registry-provider.test.tsx"],"names":[],"mappings":";;;;;AAAA,kDAAyD;AACzD,kDAA0B;AAC1B,+BAA2B;AAE3B,uEAGmC;AAEnC,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,MAAC;aACV,QAAQ,EAAE;aACV,IAAI,CAAC,MAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC;aAC5C,OAAO,CAAC,MAAC,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,MAAC;aACV,QAAQ,EAAE;aACV,IAAI,CAAC,MAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC;aAC7C,OAAO,CAAC,MAAC,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,MAAC,CAAC,MAAM,CAAC;YACpB,IAAI,EAAE,MAAC,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,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,MAAC;qBACV,QAAQ,EAAE;qBACV,IAAI,CAAC,MAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;qBAClC,OAAO,CAAC,MAAC,CAAC,MAAM,EAAE,CAAC;aACvB,CAAC;YAEF,0BAA0B;YAC1B,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YACvC,CAAC,CAAC,CAAC;YAEH,2BAA2B;YAC3B,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,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,MAAC;qBACV,QAAQ,EAAE;qBACV,IAAI,CAAC,MAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;qBAClC,OAAO,CAAC,MAAC,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,MAAC,CAAC,MAAM,CAAC,MAAC,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,MAAC,CAAC,MAAM,CAAC,MAAC,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,MAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,MAAC,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;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,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,mBAAmB,EAAE,kBAAkB;gBAC7C,WAAW,EAAE,0BAA0B;gBACvC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC;gBAC3C,UAAU,EAAE,MAAC,CAAC,QAAQ,EAAE,CAAC,IAAI,CAAC,MAAC,CAAC,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,MAAC,CAAC,MAAM,EAAE,CAAC;aAC9D,CAAC;YAEF,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,wFAAwF,CACzF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,gBAAgB,GAAmB;gBACvC,IAAI,EAAE,wBAAwB,EAAE,kBAAkB;gBAClD,SAAS,EAAE,GAAG,EAAE,CAAC,qDAAkB;gBACnC,WAAW,EAAE,+BAA+B;gBAC5C,WAAW,EAAE,MAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,MAAC,CAAC,MAAM,EAAE,EAAE,CAAC;aAC5C,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/v3\";\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      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      // First register the tool\n      act(() => {\n        result.current.registerTool(newTool);\n      });\n\n      // Then add the association\n      act(() => {\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    it(\"should throw error when tool name contains spaces\", () => {\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 name\", // Contains spaces\n        description: \"Tool with spaces in name\",\n        tool: jest.fn().mockResolvedValue(\"result\"),\n        toolSchema: z.function().args(z.string()).returns(z.string()),\n      };\n\n      expect(() => {\n        act(() => {\n          result.current.registerTool(invalidTool);\n        });\n      }).toThrow(\n        'tool \"invalid tool name\" must only contain letters, numbers, underscores, and hyphens.',\n      );\n    });\n\n    it(\"should throw error when component name contains spaces\", () => {\n      const invalidComponent: TamboComponent = {\n        name: \"Invalid Component Name\", // Contains spaces\n        component: () => <div>Invalid</div>,\n        description: \"Component with spaces in name\",\n        propsSchema: z.object({ test: z.string() }),\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 \"Invalid Component Name\" must only contain letters, numbers, underscores, and hyphens.',\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"]}
+{"version":3,"file":"tambo-registry-provider.test.js","sourceRoot":"","sources":["../../src/providers/tambo-registry-provider.test.tsx"],"names":[],"mappings":";;;;;AAAA,kDAAyD;AACzD,kDAA0B;AAC1B,uDAAqC;AAErC,+CAA8C;AAC9C,uEAGmC;AAEnC,qCAAqC;AACrC,MAAM,eAAe,GAAG,GAAgB,EAAE,CAAC;IACzC,IAAA,qBAAU,EAAC;QACT,IAAI,EAAE,aAAa;QACnB,WAAW,EAAE,iBAAiB;QAC9B,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,oBAAoB,CAAC;QACvD,WAAW,EAAE,oBAAC,CAAC,KAAK,CAAC,CAAC,oBAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC,CAAC;QAC9D,YAAY,EAAE,oBAAC,CAAC,MAAM,EAAE;KACzB,CAAC;IACF,IAAA,qBAAU,EAAC;QACT,IAAI,EAAE,aAAa;QACnB,WAAW,EAAE,kBAAkB;QAC/B,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,oBAAoB,CAAC;QACvD,WAAW,EAAE,oBAAC,CAAC,KAAK,CAAC,CAAC,oBAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC,CAAC;QAC/D,YAAY,EAAE,oBAAC,CAAC,MAAM,EAAE;KACzB,CAAC;CACH,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,oBAAC,CAAC,MAAM,CAAC;YACpB,IAAI,EAAE,oBAAC,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,MAAM,OAAO,GAAc;gBACzB,IAAI,EAAE,UAAU;gBAChB,WAAW,EAAE,UAAU;gBACvB,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,iBAAiB,CAAC;gBACpD,WAAW,EAAE,oBAAC,CAAC,KAAK,CAAC,CAAC,oBAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;gBACpD,YAAY,EAAE,oBAAC,CAAC,MAAM,EAAE;aACzB,CAAC;YAEF,0BAA0B;YAC1B,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;YACvC,CAAC,CAAC,CAAC;YAEH,2BAA2B;YAC3B,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,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,WAAW,EAAE,oBAAC,CAAC,KAAK,CAAC,CAAC,oBAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;gBACpD,YAAY,EAAE,oBAAC,CAAC,MAAM,EAAE;aACzB,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,WAAW,EAAE,oBAAC,CAAC,MAAM,CAAC,oBAAC,CAAC,MAAM,EAAE,EAAE,oBAAC,CAAC,MAAM,EAAE,CAAC,EAAE,uCAAuC;gBACtF,YAAY,EAAE,oBAAC,CAAC,MAAM,EAAE;aACzB,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,0KAA0K,CAC3K,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,oBAAC,CAAC,MAAM,CAAC,oBAAC,CAAC,MAAM,EAAE,EAAE,oBAAC,CAAC,MAAM,EAAE,CAAC,EAAE,uCAAuC;aACvF,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,2GAA2G,CAC5G,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,oBAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,oBAAC,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;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,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,mBAAmB,EAAE,kBAAkB;gBAC7C,WAAW,EAAE,0BAA0B;gBACvC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,QAAQ,CAAC;gBAC3C,WAAW,EAAE,oBAAC,CAAC,KAAK,CAAC,CAAC,oBAAC,CAAC,MAAM,EAAE,CAAC,CAAC;gBAClC,YAAY,EAAE,oBAAC,CAAC,MAAM,EAAE;aACzB,CAAC;YAEF,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,wFAAwF,CACzF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,gBAAgB,GAAmB;gBACvC,IAAI,EAAE,wBAAwB,EAAE,kBAAkB;gBAClD,SAAS,EAAE,GAAG,EAAE,CAAC,qDAAkB;gBACnC,WAAW,EAAE,+BAA+B;gBAC5C,WAAW,EAAE,oBAAC,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,oBAAC,CAAC,MAAM,EAAE,EAAE,CAAC;aAC5C,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;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,eAAe,GAAG;gBACtB;oBACE,GAAG,EAAE,uBAAuB;oBAC5B,IAAI,EAAE,gBAAgB;oBACtB,QAAQ,EAAE,YAAY;iBACvB;gBACD;oBACE,GAAG,EAAE,yBAAyB;oBAC9B,IAAI,EAAE,aAAa;oBACnB,QAAQ,EAAE,WAAW;iBACtB;aACF,CAAC;YAEF,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,IAAC,SAAS,EAAE,eAAe,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,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;YACtE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;YACpE,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,GAAG;gBAClB,GAAG,EAAE,yBAAyB;gBAC9B,IAAI,EAAE,kBAAkB;gBACxB,QAAQ,EAAE,YAAY;aACvB,CAAC;YAEF,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,WAAW,CAAC,CAAC;YAC/C,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,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,YAAY,GAAG;gBACnB;oBACE,GAAG,EAAE,wBAAwB;oBAC7B,IAAI,EAAE,kBAAkB;oBACxB,QAAQ,EAAE,YAAY;iBACvB;gBACD;oBACE,GAAG,EAAE,wBAAwB;oBAC7B,IAAI,EAAE,kBAAkB;oBACxB,QAAQ,EAAE,YAAY;iBACvB;aACF,CAAC;YAEF,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;YACjD,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YACvE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,iBAAiB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAC1D,MAAM,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAClD,QAAQ,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;aAC1D,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,IACpB,aAAa,EAAE,iBAAiB,EAChC,WAAW,EAAE,eAAe,IAE3B,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,cAAc,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC,IAAI,CACvD,iBAAiB,CAClB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC3E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;YACpE,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,iBAAiB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAC1D,MAAM,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAClD,QAAQ,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;aAC1D,CAAC,CAAC;YAEH,IAAA,WAAG,EAAC,GAAG,EAAE;gBACP,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC;oBACpC,aAAa,EAAE,iBAAiB;oBAChC,WAAW,EAAE,eAAe;iBAC7B,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,aAAa,CAAC,CAAC,IAAI,CACvD,iBAAiB,CAClB,CAAC;YACF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC3E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,iBAAiB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAE1D,MAAM,CAAC,GAAG,EAAE;gBACV,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,IAAC,aAAa,EAAE,iBAAiB,IACpD,QAAQ,CACa,CACzB,CAAC;gBACF,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC,OAAO,CACR,8DAA8D,CAC/D,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;gBAClD,QAAQ,EAAE,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,QAAQ,EAAE,YAAY,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;aAC1D,CAAC,CAAC;YAEH,MAAM,CAAC,GAAG,EAAE;gBACV,MAAM,OAAO,GAAG,CAAC,EAAE,QAAQ,EAAiC,EAAE,EAAE,CAAC,CAC/D,8BAAC,+CAAqB,IAAC,WAAW,EAAE,eAAe,IAChD,QAAQ,CACa,CACzB,CAAC;gBACF,IAAA,kBAAU,EAAC,GAAG,EAAE,CAAC,IAAA,0CAAgB,GAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YACpD,CAAC,CAAC,CAAC,OAAO,CACR,8DAA8D,CAC/D,CAAC;QACJ,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,MAAM,eAAe,GAAG;gBACtB,IAAI,EAAE,sBAAsB;gBAC5B,QAAQ,EAAE,YAAY;aAChB,CAAC;YAET,MAAM,CAAC,GAAG,EAAE;gBACV,IAAA,WAAG,EAAC,GAAG,EAAE;oBACP,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC;gBACnD,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,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,eAAe,GAAG;gBACtB,GAAG,EAAE,iBAAiB;gBACtB,QAAQ,EAAE,YAAY;aAChB,CAAC;YAET,MAAM,CAAC,GAAG,EAAE;gBACV,IAAA,WAAG,EAAC,GAAG,EAAE;oBACP,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC;gBACnD,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC,OAAO,CACR,8DAA8D,CAC/D,CAAC;QACJ,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 \"zodInternalAlias\";\nimport { TamboComponent, TamboTool } from \"../model/component-metadata\";\nimport { defineTool } from \"../util/registry\";\nimport {\n  TamboRegistryProvider,\n  useTamboRegistry,\n} from \"./tambo-registry-provider\";\n\n// Shared tool registry for all tests\nconst createMockTools = (): TamboTool[] => [\n  defineTool({\n    name: \"test-tool-1\",\n    description: \"First test tool\",\n    tool: jest.fn().mockResolvedValue(\"test-tool-1-result\"),\n    inputSchema: z.tuple([z.string().describe(\"input parameter\")]),\n    outputSchema: z.string(),\n  }),\n  defineTool({\n    name: \"test-tool-2\",\n    description: \"Second test tool\",\n    tool: jest.fn().mockResolvedValue(\"test-tool-2-result\"),\n    inputSchema: z.tuple([z.number().describe(\"number parameter\")]),\n    outputSchema: 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      const newTool: TamboTool = {\n        name: \"new-tool\",\n        description: \"New tool\",\n        tool: jest.fn().mockResolvedValue(\"new-tool-result\"),\n        inputSchema: z.tuple([z.string().describe(\"input\")]),\n        outputSchema: z.string(),\n      };\n\n      // First register the tool\n      act(() => {\n        result.current.registerTool(newTool);\n      });\n\n      // Then add the association\n      act(() => {\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        inputSchema: z.tuple([z.string().describe(\"input\")]),\n        outputSchema: 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        inputSchema: z.record(z.string(), z.string()), // This should cause validation to fail\n        outputSchema: z.string(),\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        'Record types (objects with dynamic keys) are not supported in inputSchema of tool \"invalid-tool\". Found at path \"(root)\". Replace it with an object using explicit keys.',\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(), 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        'Record types (objects with dynamic keys) are 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    it(\"should throw error when tool name contains spaces\", () => {\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 name\", // Contains spaces\n        description: \"Tool with spaces in name\",\n        tool: jest.fn().mockResolvedValue(\"result\"),\n        inputSchema: z.tuple([z.string()]),\n        outputSchema: z.string(),\n      };\n\n      expect(() => {\n        act(() => {\n          result.current.registerTool(invalidTool);\n        });\n      }).toThrow(\n        'tool \"invalid tool name\" must only contain letters, numbers, underscores, and hyphens.',\n      );\n    });\n\n    it(\"should throw error when component name contains spaces\", () => {\n      const invalidComponent: TamboComponent = {\n        name: \"Invalid Component Name\", // Contains spaces\n        component: () => <div>Invalid</div>,\n        description: \"Component with spaces in name\",\n        propsSchema: z.object({ test: z.string() }),\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 \"Invalid Component Name\" must only contain letters, numbers, underscores, and hyphens.',\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  describe(\"Resource registration\", () => {\n    it(\"should register static resources via props\", () => {\n      const staticResources = [\n        {\n          uri: \"file:///local/doc.txt\",\n          name: \"Local Document\",\n          mimeType: \"text/plain\",\n        },\n        {\n          uri: \"file:///local/image.png\",\n          name: \"Local Image\",\n          mimeType: \"image/png\",\n        },\n      ];\n\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider resources={staticResources}>\n          {children}\n        </TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      expect(result.current.resources).toHaveLength(2);\n      expect(result.current.resources[0].uri).toBe(\"file:///local/doc.txt\");\n      expect(result.current.resources[1].uri).toBe(\"file:///local/image.png\");\n    });\n\n    it(\"should register resources dynamically via registerResource\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      const newResource = {\n        uri: \"file:///dynamic/doc.txt\",\n        name: \"Dynamic Document\",\n        mimeType: \"text/plain\",\n      };\n\n      act(() => {\n        result.current.registerResource(newResource);\n      });\n\n      expect(result.current.resources).toHaveLength(1);\n      expect(result.current.resources[0].uri).toBe(\"file:///dynamic/doc.txt\");\n    });\n\n    it(\"should register multiple resources via registerResources\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      const newResources = [\n        {\n          uri: \"file:///batch/doc1.txt\",\n          name: \"Batch Document 1\",\n          mimeType: \"text/plain\",\n        },\n        {\n          uri: \"file:///batch/doc2.txt\",\n          name: \"Batch Document 2\",\n          mimeType: \"text/plain\",\n        },\n      ];\n\n      act(() => {\n        result.current.registerResources(newResources);\n      });\n\n      expect(result.current.resources).toHaveLength(2);\n      expect(result.current.resources[0].uri).toBe(\"file:///batch/doc1.txt\");\n      expect(result.current.resources[1].uri).toBe(\"file:///batch/doc2.txt\");\n    });\n\n    it(\"should register resource source via props\", () => {\n      const mockListResources = jest.fn().mockResolvedValue([]);\n      const mockGetResource = jest.fn().mockResolvedValue({\n        contents: [{ uri: \"\", mimeType: \"text/plain\", text: \"\" }],\n      });\n\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider\n          listResources={mockListResources}\n          getResource={mockGetResource}\n        >\n          {children}\n        </TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      expect(result.current.resourceSource).not.toBeNull();\n      expect(result.current.resourceSource?.listResources).toBe(\n        mockListResources,\n      );\n      expect(result.current.resourceSource?.getResource).toBe(mockGetResource);\n    });\n\n    it(\"should register resource source via registerResourceSource\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      const mockListResources = jest.fn().mockResolvedValue([]);\n      const mockGetResource = jest.fn().mockResolvedValue({\n        contents: [{ uri: \"\", mimeType: \"text/plain\", text: \"\" }],\n      });\n\n      act(() => {\n        result.current.registerResourceSource({\n          listResources: mockListResources,\n          getResource: mockGetResource,\n        });\n      });\n\n      expect(result.current.resourceSource).not.toBeNull();\n      expect(result.current.resourceSource?.listResources).toBe(\n        mockListResources,\n      );\n      expect(result.current.resourceSource?.getResource).toBe(mockGetResource);\n    });\n\n    it(\"should throw error when only listResources is provided\", () => {\n      const mockListResources = jest.fn().mockResolvedValue([]);\n\n      expect(() => {\n        const wrapper = ({ children }: { children: React.ReactNode }) => (\n          <TamboRegistryProvider listResources={mockListResources}>\n            {children}\n          </TamboRegistryProvider>\n        );\n        renderHook(() => useTamboRegistry(), { wrapper });\n      }).toThrow(\n        \"Both listResources and getResource must be provided together\",\n      );\n    });\n\n    it(\"should throw error when only getResource is provided\", () => {\n      const mockGetResource = jest.fn().mockResolvedValue({\n        contents: [{ uri: \"\", mimeType: \"text/plain\", text: \"\" }],\n      });\n\n      expect(() => {\n        const wrapper = ({ children }: { children: React.ReactNode }) => (\n          <TamboRegistryProvider getResource={mockGetResource}>\n            {children}\n          </TamboRegistryProvider>\n        );\n        renderHook(() => useTamboRegistry(), { wrapper });\n      }).toThrow(\n        \"Both listResources and getResource must be provided together\",\n      );\n    });\n\n    it(\"should validate resource has required uri field\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      const invalidResource = {\n        name: \"Document without URI\",\n        mimeType: \"text/plain\",\n      } as any;\n\n      expect(() => {\n        act(() => {\n          result.current.registerResource(invalidResource);\n        });\n      }).toThrow(\"Resource must have a 'uri' field\");\n    });\n\n    it(\"should validate resource has required name field\", () => {\n      const wrapper = ({ children }: { children: React.ReactNode }) => (\n        <TamboRegistryProvider>{children}</TamboRegistryProvider>\n      );\n\n      const { result } = renderHook(() => useTamboRegistry(), { wrapper });\n\n      const invalidResource = {\n        uri: \"file:///doc.txt\",\n        mimeType: \"text/plain\",\n      } as any;\n\n      expect(() => {\n        act(() => {\n          result.current.registerResource(invalidResource);\n        });\n      }).toThrow(\n        \"Resource with URI 'file:///doc.txt' must have a 'name' field\",\n      );\n    });\n  });\n});\n"]}

package/dist/providers/tambo-registry-schema-compat.test.d.ts

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

package/dist/providers/tambo-registry-schema-compat.test.d.ts.map

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