@tambo-ai/react 0.65.2 → 0.66.0

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

@@ -6,14 +6,14 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const typescript_sdk_1 = require("@tambo-ai/typescript-sdk");
 const react_1 = require("@testing-library/react");
 const react_2 = __importDefault(require("react"));
-const zod_1 = require("zod");
-const generate_component_response_1 = require("../../model/generate-component-response");
-const tools_1 = require("../../testing/tools");
-const tambo_client_provider_1 = require("../tambo-client-provider");
-const tambo_context_helpers_provider_1 = require("../tambo-context-helpers-provider");
-const tambo_mcp_token_provider_1 = require("../tambo-mcp-token-provider");
-const tambo_registry_provider_1 = require("../tambo-registry-provider");
-const tambo_thread_provider_1 = require("../tambo-thread-provider");
+const zodInternalAlias_1 = require("zodInternalAlias");
+const generate_component_response_1 = require("../model/generate-component-response");
+const tools_1 = require("../testing/tools");
+const tambo_client_provider_1 = require("./tambo-client-provider");
+const tambo_context_helpers_provider_1 = require("./tambo-context-helpers-provider");
+const tambo_mcp_token_provider_1 = require("./tambo-mcp-token-provider");
+const tambo_registry_provider_1 = require("./tambo-registry-provider");
+const tambo_thread_provider_1 = require("./tambo-thread-provider");
 // Mock crypto.randomUUID
 Object.defineProperty(global, "crypto", {
     value: {
@@ -21,7 +21,7 @@ Object.defineProperty(global, "crypto", {
     },
 });
 // Mock the required providers
-jest.mock("../tambo-client-provider", () => {
+jest.mock("./tambo-client-provider", () => {
     return {
         useTamboClient: jest.fn(),
         useTamboQueryClient: jest.fn(),
@@ -32,8 +32,8 @@ jest.mock("@tambo-ai/typescript-sdk", () => ({
     advanceStream: jest.fn(),
 }));
 // Mock the getCustomContext
-jest.mock("../../util/registry", () => ({
-    ...jest.requireActual("../../util/registry"),
+jest.mock("../util/registry", () => ({
+    ...jest.requireActual("../util/registry"),
     getCustomContext: () => ({
         message: "additional instructions",
     }),
@@ -99,18 +99,16 @@ describe("TamboThreadProvider", () => {
             name: "TestOnly",
             component: () => react_2.default.createElement("div", null, "TestOnly"),
             description: "TestOnly",
-            propsSchema: zod_1.z.object({
-                test: zod_1.z.string(),
+            propsSchema: zodInternalAlias_1.z.object({
+                test: zodInternalAlias_1.z.string(),
             }),
             associatedTools: [
                 {
                     name: "test-tool",
                     tool: jest.fn().mockResolvedValue("test-tool"),
                     description: "test-tool",
-                    toolSchema: zod_1.z
-                        .function()
-                        .args(zod_1.z.string().describe("test-param-description"))
-                        .returns(zod_1.z.string()),
+                    inputSchema: zodInternalAlias_1.z.tuple([zodInternalAlias_1.z.string().describe("test-param-description")]),
+                    outputSchema: zodInternalAlias_1.z.string(),
                 },
             ],
         },
@@ -123,8 +121,6 @@ describe("TamboThreadProvider", () => {
                 client,
                 queryClient,
                 isUpdatingToken: false,
-                mcpAccessToken: null,
-                setMcpAccessToken: () => { },
             } },
             react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
                 react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
@@ -342,7 +338,10 @@ describe("TamboThreadProvider", () => {
             });
         });
         expect(result.current.generationStage).toBe(generate_component_response_1.GenerationStage.COMPLETE);
-        expect(mockRegistry[0]?.associatedTools?.[0]?.tool).toHaveBeenCalledWith("test");
+        // New inputSchema interface: tool receives single object arg
+        expect(mockRegistry[0]?.associatedTools?.[0]?.tool).toHaveBeenCalledWith({
+            test: "test",
+        });
     });
     it("should handle unregistered tool calls with onCallUnregisteredTool", async () => {
         const mockOnCallUnregisteredTool = jest
@@ -355,8 +354,6 @@ describe("TamboThreadProvider", () => {
                     client,
                     queryClient,
                     isUpdatingToken: false,
-                    mcpAccessToken: null,
-                    setMcpAccessToken: () => { },
                 } },
                 react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry, onCallUnregisteredTool: mockOnCallUnregisteredTool },
                     react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
@@ -465,8 +462,6 @@ describe("TamboThreadProvider", () => {
                         client,
                         queryClient,
                         isUpdatingToken: false,
-                        mcpAccessToken: null,
-                        setMcpAccessToken: () => { },
                     } },
                     react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
                         react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
@@ -538,8 +533,6 @@ describe("TamboThreadProvider", () => {
                         client,
                         queryClient,
                         isUpdatingToken: false,
-                        mcpAccessToken: null,
-                        setMcpAccessToken: () => { },
                     } },
                     react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
                         react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
@@ -592,8 +585,6 @@ describe("TamboThreadProvider", () => {
                         client,
                         queryClient,
                         isUpdatingToken: false,
-                        mcpAccessToken: null,
-                        setMcpAccessToken: () => { },
                     } },
                     react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
                         react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
@@ -646,8 +637,6 @@ describe("TamboThreadProvider", () => {
                         client,
                         queryClient,
                         isUpdatingToken: false,
-                        mcpAccessToken: null,
-                        setMcpAccessToken: () => { },
                     } },
                     react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
                         react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
@@ -719,8 +708,6 @@ describe("TamboThreadProvider", () => {
                         client,
                         queryClient,
                         isUpdatingToken: false,
-                        mcpAccessToken: null,
-                        setMcpAccessToken: () => { },
                     } },
                     react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
                         react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
@@ -775,8 +762,6 @@ describe("TamboThreadProvider", () => {
                         client,
                         queryClient,
                         isUpdatingToken: false,
-                        mcpAccessToken: null,
-                        setMcpAccessToken: () => { },
                     } },
                     react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
                         react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
@@ -978,16 +963,14 @@ describe("TamboThreadProvider", () => {
                     name: "TestComponent",
                     component: () => react_2.default.createElement("div", null, "Test"),
                     description: "Test",
-                    propsSchema: zod_1.z.object({ test: zod_1.z.string() }),
+                    propsSchema: zodInternalAlias_1.z.object({ test: zodInternalAlias_1.z.string() }),
                     associatedTools: [
                         {
                             name: "custom-tool",
                             tool: jest.fn().mockResolvedValue({ data: "tool result" }),
                             description: "Tool with custom transform",
-                            toolSchema: zod_1.z
-                                .function()
-                                .args(zod_1.z.string())
-                                .returns(zod_1.z.object({ data: zod_1.z.string() })),
+                            inputSchema: zodInternalAlias_1.z.tuple([zodInternalAlias_1.z.string()]),
+                            outputSchema: zodInternalAlias_1.z.object({ data: zodInternalAlias_1.z.string() }),
                             transformToContent: mockTransformToContent,
                         },
                     ],
@@ -1000,8 +983,6 @@ describe("TamboThreadProvider", () => {
                         client,
                         queryClient,
                         isUpdatingToken: false,
-                        mcpAccessToken: null,
-                        setMcpAccessToken: () => { },
                     } },
                     react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: customToolRegistry },
                         react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
@@ -1051,8 +1032,8 @@ describe("TamboThreadProvider", () => {
                     streamResponse: false,
                 });
             });
-            // Verify the tool was called
-            expect(customToolRegistry[0]?.associatedTools?.[0]?.tool).toHaveBeenCalledWith("test");
+            // Verify the tool was called with single object arg (new inputSchema interface)
+            expect(customToolRegistry[0]?.associatedTools?.[0]?.tool).toHaveBeenCalledWith({ input: "test" });
             // Verify transformToContent was called with the tool result
             expect(mockTransformToContent).toHaveBeenCalledWith({
                 data: "tool result",
@@ -1083,16 +1064,14 @@ describe("TamboThreadProvider", () => {
                     name: "TestComponent",
                     component: () => react_2.default.createElement("div", null, "Test"),
                     description: "Test",
-                    propsSchema: zod_1.z.object({ test: zod_1.z.string() }),
+                    propsSchema: zodInternalAlias_1.z.object({ test: zodInternalAlias_1.z.string() }),
                     associatedTools: [
                         {
                             name: "async-tool",
                             tool: jest.fn().mockResolvedValue({ data: "async tool result" }),
                             description: "Tool with async transform",
-                            toolSchema: zod_1.z
-                                .function()
-                                .args(zod_1.z.string())
-                                .returns(zod_1.z.object({ data: zod_1.z.string() })),
+                            inputSchema: zodInternalAlias_1.z.tuple([zodInternalAlias_1.z.string()]),
+                            outputSchema: zodInternalAlias_1.z.object({ data: zodInternalAlias_1.z.string() }),
                             transformToContent: mockTransformToContent,
                         },
                     ],
@@ -1105,8 +1084,6 @@ describe("TamboThreadProvider", () => {
                         client,
                         queryClient,
                         isUpdatingToken: false,
-                        mcpAccessToken: null,
-                        setMcpAccessToken: () => { },
                     } },
                     react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: customToolRegistry },
                         react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
@@ -1169,8 +1146,8 @@ describe("TamboThreadProvider", () => {
                     streamResponse: true,
                 });
             });
-            // Verify the tool was called
-            expect(customToolRegistry[0]?.associatedTools?.[0]?.tool).toHaveBeenCalledWith("async-test");
+            // Verify the tool was called with single object arg (new inputSchema interface)
+            expect(customToolRegistry[0]?.associatedTools?.[0]?.tool).toHaveBeenCalledWith({ input: "async-test" });
             // Verify transformToContent was called
             expect(mockTransformToContent).toHaveBeenCalledWith({
                 data: "async tool result",
@@ -1191,7 +1168,7 @@ describe("TamboThreadProvider", () => {
                     name: "TestComponent",
                     component: () => react_2.default.createElement("div", null, "Test"),
                     description: "Test",
-                    propsSchema: zod_1.z.object({ test: zod_1.z.string() }),
+                    propsSchema: zodInternalAlias_1.z.object({ test: zodInternalAlias_1.z.string() }),
                     associatedTools: [
                         {
                             name: "no-transform-tool",
@@ -1199,13 +1176,11 @@ describe("TamboThreadProvider", () => {
                                 .fn()
                                 .mockResolvedValue({ complex: "data", nested: { value: 42 } }),
                             description: "Tool without custom transform",
-                            toolSchema: zod_1.z
-                                .function()
-                                .args(zod_1.z.string())
-                                .returns(zod_1.z.object({
-                                complex: zod_1.z.string(),
-                                nested: zod_1.z.object({ value: zod_1.z.number() }),
-                            })),
+                            inputSchema: zodInternalAlias_1.z.tuple([zodInternalAlias_1.z.string()]),
+                            outputSchema: zodInternalAlias_1.z.object({
+                                complex: zodInternalAlias_1.z.string(),
+                                nested: zodInternalAlias_1.z.object({ value: zodInternalAlias_1.z.number() }),
+                            }),
                             // No transformToContent provided
                         },
                     ],
@@ -1218,8 +1193,6 @@ describe("TamboThreadProvider", () => {
                         client,
                         queryClient,
                         isUpdatingToken: false,
-                        mcpAccessToken: null,
-                        setMcpAccessToken: () => { },
                     } },
                     react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: toolWithoutTransform },
                         react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
@@ -1269,8 +1242,8 @@ describe("TamboThreadProvider", () => {
                     streamResponse: false,
                 });
             });
-            // Verify the tool was called
-            expect(toolWithoutTransform[0]?.associatedTools?.[0]?.tool).toHaveBeenCalledWith("test");
+            // Verify the tool was called with single object arg (new inputSchema interface)
+            expect(toolWithoutTransform[0]?.associatedTools?.[0]?.tool).toHaveBeenCalledWith({ input: "test" });
             // Verify the second advance call used stringified content
             expect(mockThreadsApi.advanceByID).toHaveBeenLastCalledWith("test-thread-1", expect.objectContaining({
                 messageToAppend: expect.objectContaining({
@@ -1296,7 +1269,7 @@ describe("TamboThreadProvider", () => {
                     name: "TestComponent",
                     component: () => react_2.default.createElement("div", null, "Test"),
                     description: "Test",
-                    propsSchema: zod_1.z.object({ test: zod_1.z.string() }),
+                    propsSchema: zodInternalAlias_1.z.object({ test: zodInternalAlias_1.z.string() }),
                     associatedTools: [
                         {
                             name: "error-tool",
@@ -1304,7 +1277,8 @@ describe("TamboThreadProvider", () => {
                                 .fn()
                                 .mockRejectedValue(new Error("Tool execution failed")),
                             description: "Tool that errors",
-                            toolSchema: zod_1.z.function().args(zod_1.z.string()).returns(zod_1.z.string()),
+                            inputSchema: zodInternalAlias_1.z.tuple([zodInternalAlias_1.z.string()]),
+                            outputSchema: zodInternalAlias_1.z.string(),
                             transformToContent: mockTransformToContent,
                         },
                     ],
@@ -1317,8 +1291,6 @@ describe("TamboThreadProvider", () => {
                         client,
                         queryClient,
                         isUpdatingToken: false,
-                        mcpAccessToken: null,
-                        setMcpAccessToken: () => { },
                     } },
                     react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: toolWithTransform },
                         react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
@@ -1368,8 +1340,8 @@ describe("TamboThreadProvider", () => {
                     streamResponse: false,
                 });
             });
-            // Verify the tool was called
-            expect(toolWithTransform[0]?.associatedTools?.[0]?.tool).toHaveBeenCalledWith("test");
+            // Verify the tool was called with single object arg (new inputSchema interface)
+            expect(toolWithTransform[0]?.associatedTools?.[0]?.tool).toHaveBeenCalledWith({ input: "test" });
             // Verify transformToContent was NOT called for error responses
             expect(mockTransformToContent).not.toHaveBeenCalled();
             // Verify the second advance call used text content with the error message
@@ -1395,8 +1367,6 @@ describe("TamboThreadProvider", () => {
                         client,
                         queryClient,
                         isUpdatingToken: false,
-                        mcpAccessToken: null,
-                        setMcpAccessToken: () => { },
                     } },
                     react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
                         react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
@@ -1451,8 +1421,6 @@ describe("TamboThreadProvider", () => {
                         client,
                         queryClient,
                         isUpdatingToken: false,
-                        mcpAccessToken: null,
-                        setMcpAccessToken: () => { },
                     } },
                     react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
                         react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
@@ -1503,8 +1471,6 @@ describe("TamboThreadProvider", () => {
                         client,
                         queryClient,
                         isUpdatingToken: false,
-                        mcpAccessToken: null,
-                        setMcpAccessToken: () => { },
                     } },
                     react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
                         react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {
@@ -1560,8 +1526,6 @@ describe("TamboThreadProvider", () => {
                         client,
                         queryClient,
                         isUpdatingToken: false,
-                        mcpAccessToken: null,
-                        setMcpAccessToken: () => { },
                     } },
                     react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, { components: mockRegistry },
                         react_2.default.createElement(tambo_context_helpers_provider_1.TamboContextHelpersProvider, { contextHelpers: {