@tambo-ai/react 0.53.2 → 0.54.1

package/dist/mcp/__tests__/tambo-mcp-provider.test.js

@@ -1,5 +1,42 @@
 "use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
 Object.defineProperty(exports, "__esModule", { value: true });
+const react_1 = require("@testing-library/react");
+const react_2 = __importStar(require("react"));
+const tambo_registry_provider_1 = require("../../providers/tambo-registry-provider");
+const mcp_client_1 = require("../mcp-client");
 const tambo_mcp_provider_1 = require("../tambo-mcp-provider");
 // Mock the MCP client to avoid ES module issues
 jest.mock("../mcp-client", () => ({
@@ -112,4 +149,82 @@ describe("extractErrorMessage", () => {
         });
     });
 });
+describe("TamboMcpProvider server list changes", () => {
+    beforeEach(() => {
+        // Mock registry so tool registration is a no-op
+        tambo_registry_provider_1.useTamboRegistry.mockReturnValue({
+            registerTool: jest.fn(),
+        });
+        // Ensure MCPClient.create exists and returns a fake client with listTools
+        mcp_client_1.MCPClient.create = jest
+            .fn()
+            .mockResolvedValue({ listTools: jest.fn().mockResolvedValue([]) });
+    });
+    const Capture = ({ onUpdate, }) => {
+        const servers = (0, tambo_mcp_provider_1.useTamboMcpServers)();
+        (0, react_2.useEffect)(() => {
+            onUpdate(servers);
+        }, [servers, onUpdate]);
+        return react_2.default.createElement("div", { "data-testid": "urls" }, servers.map((s) => s.url).join(","));
+    };
+    it("adds a new server when the list grows", async () => {
+        let latest = [];
+        const { rerender, getByTestId } = (0, react_1.render)(react_2.default.createElement(tambo_mcp_provider_1.TamboMcpProvider, { mcpServers: ["https://a.example"] },
+            react_2.default.createElement(Capture, { onUpdate: (s) => (latest = s) })));
+        // Wait for initial connection
+        await (0, react_1.waitFor)(() => {
+            expect(latest.length).toBe(1);
+        });
+        // Add new server
+        rerender(react_2.default.createElement(tambo_mcp_provider_1.TamboMcpProvider, { mcpServers: ["https://a.example", "https://b.example"] },
+            react_2.default.createElement(Capture, { onUpdate: (s) => (latest = s) })));
+        await (0, react_1.waitFor)(() => {
+            expect(latest.length).toBe(2);
+            const urls = getByTestId("urls").textContent || "";
+            expect(urls).toContain("https://a.example");
+            expect(urls).toContain("https://b.example");
+        });
+    });
+    it("removes a server when the list shrinks", async () => {
+        let latest = [];
+        const { rerender, getByTestId } = (0, react_1.render)(react_2.default.createElement(tambo_mcp_provider_1.TamboMcpProvider, { mcpServers: ["https://a.example", "https://b.example"] },
+            react_2.default.createElement(Capture, { onUpdate: (s) => (latest = s) })));
+        await (0, react_1.waitFor)(() => {
+            expect(latest.length).toBe(2);
+        });
+        // Remove one server
+        rerender(react_2.default.createElement(tambo_mcp_provider_1.TamboMcpProvider, { mcpServers: ["https://a.example"] },
+            react_2.default.createElement(Capture, { onUpdate: (s) => (latest = s) })));
+        await (0, react_1.waitFor)(() => {
+            expect(latest.length).toBe(1);
+            const urls = getByTestId("urls").textContent || "";
+            expect(urls).toContain("https://a.example");
+            expect(urls).not.toContain("https://b.example");
+        });
+    });
+    it("does not duplicate when a new copy of the same list is passed", async () => {
+        let latest = [];
+        const initial = [
+            { url: "https://a.example", transport: mcp_client_1.MCPTransport.SSE },
+            { url: "https://b.example", transport: mcp_client_1.MCPTransport.SSE },
+        ];
+        const { rerender } = (0, react_1.render)(react_2.default.createElement(tambo_mcp_provider_1.TamboMcpProvider, { mcpServers: initial },
+            react_2.default.createElement(Capture, { onUpdate: (s) => (latest = s) })));
+        await (0, react_1.waitFor)(() => {
+            expect(latest.length).toBe(2);
+        });
+        // Pass a new array instance with the same logical servers
+        const same = [
+            { url: "https://a.example", transport: mcp_client_1.MCPTransport.SSE },
+            { url: "https://b.example", transport: mcp_client_1.MCPTransport.SSE },
+        ];
+        rerender(react_2.default.createElement(tambo_mcp_provider_1.TamboMcpProvider, { mcpServers: same },
+            react_2.default.createElement(Capture, { onUpdate: (s) => (latest = s) })));
+        await (0, react_1.waitFor)(() => {
+            expect(latest.length).toBe(2);
+            const urls = latest.map((s) => s.url).sort();
+            expect(urls).toEqual(["https://a.example", "https://b.example"].sort());
+        });
+    });
+});
 //# sourceMappingURL=tambo-mcp-provider.test.js.map

package/dist/mcp/__tests__/tambo-mcp-provider.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"tambo-mcp-provider.test.js","sourceRoot":"","sources":["../../../src/mcp/__tests__/tambo-mcp-provider.test.tsx"],"names":[],"mappings":";;AAAA,8DAA4D;AAE5D,gDAAgD;AAChD,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;IAChC,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;IACpB,YAAY,EAAE;QACZ,GAAG,EAAE,KAAK;QACV,IAAI,EAAE,MAAM;KACb;CACF,CAAC,CAAC,CAAC;AAEJ,wDAAwD;AACxD,IAAI,CAAC,IAAI,CAAC,yCAAyC,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1D,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;CAC5B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,MAAM,OAAO,GAAG;gBACd,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAChC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,uBAAuB,EAAE;gBAC/C,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,8BAA8B,EAAE,EAAE,yBAAyB;aAClF,CAAC;YAEF,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;YACtE,MAAM,OAAO,GAAG,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC,CAAC;YAEjE,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;YAC7E,MAAM,OAAO,GAAG;gBACd,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,8BAA8B,EAAE;gBACtD,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,EAAE,kBAAkB,EAAE;aAC9C,CAAC;YAEF,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,OAAO,GAAU,EAAE,CAAC;YAE1B,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,OAAO,GAAG;gBACd,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,EAAE,gBAAgB,EAAE;gBAC3C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE;gBACrC,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,4BAA4B,EAAE;gBACpD,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE;gBACtC,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,WAAW,EAAE;aACvC,CAAC;YAEF,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,OAAO,GAAG;gBACd,IAAI;gBACJ,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE;gBACrC,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,wBAAwB;gBAC1C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,oBAAoB;gBAClD,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qBAAqB,EAAE;aAC9C,CAAC;YAEF,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;QAC1C,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,OAAO,GAAG,sBAAsB,CAAC;YAEvC,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,OAAO,GAAG,IAAI,CAAC;YAErB,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,OAAO,GAAG,SAAS,CAAC;YAE1B,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,OAAO,GAAG,EAAE,CAAC;YAEnB,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,OAAO,GAAG,KAAK,CAAC;YAEtB,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,OAAO,GAAG,EAAE,KAAK,EAAE,sBAAsB,EAAE,CAAC;YAElD,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,OAAO,GAAG;gBACd,KAAK,EAAE,sBAAsB;gBAC7B,IAAI,EAAE,GAAG;gBACT,OAAO,EAAE,EAAE,OAAO,EAAE,uBAAuB,EAAE;aAC9C,CAAC;YAEF,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACjB,2FAA2F,CAC5F,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,OAAO,GAAG,EAAE,CAAC;YAEnB,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { extractErrorMessage } from \"../tambo-mcp-provider\";\n\n// Mock the MCP client to avoid ES module issues\njest.mock(\"../mcp-client\", () => ({\n  MCPClient: jest.fn(),\n  MCPTransport: {\n    SSE: \"sse\",\n    HTTP: \"http\",\n  },\n}));\n\n// Mock the registry provider to avoid dependency issues\njest.mock(\"../../providers/tambo-registry-provider\", () => ({\n  useTamboRegistry: jest.fn(),\n}));\n\ndescribe(\"extractErrorMessage\", () => {\n  describe(\"Array content handling\", () => {\n    it(\"should extract text from array content with multiple text items\", () => {\n      const content = [\n        { type: \"text\", text: \"Error:\" },\n        { type: \"text\", text: \"Tool execution failed\" },\n        { type: \"image\", url: \"http://example.com/error.png\" }, // Should be filtered out\n      ];\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"Error: Tool execution failed\");\n    });\n\n    it(\"should extract text from array content with single text item\", () => {\n      const content = [{ type: \"text\", text: \"Simple error message\" }];\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"Simple error message\");\n    });\n\n    it(\"should return fallback message for array content with no text items\", () => {\n      const content = [\n        { type: \"image\", url: \"http://example.com/error.png\" },\n        { type: \"resource\", uri: \"file://error.log\" },\n      ];\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"Error occurred but no details provided\");\n    });\n\n    it(\"should return fallback message for empty array content\", () => {\n      const content: any[] = [];\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"Error occurred but no details provided\");\n    });\n\n    it(\"should handle array content with mixed types correctly\", () => {\n      const content = [\n        { type: \"resource\", uri: \"file://log.txt\" },\n        { type: \"text\", text: \"First error\" },\n        { type: \"image\", url: \"http://example.com/img.png\" },\n        { type: \"text\", text: \"Second error\" },\n        { type: \"unknown\", data: \"something\" },\n      ];\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"First error Second error\");\n    });\n\n    it(\"should handle array content with malformed items\", () => {\n      const content = [\n        null,\n        { type: \"text\", text: \"Valid error\" },\n        { type: \"text\" }, // Missing text property\n        { type: \"text\", text: null }, // Invalid text type\n        { type: \"text\", text: \"Another valid error\" },\n      ];\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"Valid error Another valid error\");\n    });\n  });\n\n  describe(\"Non-array content handling\", () => {\n    it(\"should return string content as-is\", () => {\n      const content = \"Direct error message\";\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"Direct error message\");\n    });\n\n    it(\"should handle null content\", () => {\n      const content = null;\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"Unknown error occurred\");\n    });\n\n    it(\"should handle undefined content\", () => {\n      const content = undefined;\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"Unknown error occurred\");\n    });\n\n    it(\"should handle number content\", () => {\n      const content = 42;\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"42\");\n    });\n\n    it(\"should handle boolean content\", () => {\n      const content = false;\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"false\");\n    });\n\n    it(\"should handle object content\", () => {\n      const content = { error: \"Something went wrong\" };\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe('{\"error\":\"Something went wrong\"}');\n    });\n\n    it(\"should handle complex object content\", () => {\n      const content = {\n        error: \"Something went wrong\",\n        code: 500,\n        details: { message: \"Internal server error\" },\n      };\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\n        '{\"error\":\"Something went wrong\",\"code\":500,\"details\":{\"message\":\"Internal server error\"}}',\n      );\n    });\n\n    it(\"should handle empty object content\", () => {\n      const content = {};\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"{}\");\n    });\n  });\n});\n"]}
+{"version":3,"file":"tambo-mcp-provider.test.js","sourceRoot":"","sources":["../../../src/mcp/__tests__/tambo-mcp-provider.test.tsx"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,kDAAyD;AACzD,+CAAyC;AACzC,qFAA2E;AAC3E,8CAAwD;AACxD,8DAK+B;AAE/B,gDAAgD;AAChD,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;IAChC,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;IACpB,YAAY,EAAE;QACZ,GAAG,EAAE,KAAK;QACV,IAAI,EAAE,MAAM;KACb;CACF,CAAC,CAAC,CAAC;AAEJ,wDAAwD;AACxD,IAAI,CAAC,IAAI,CAAC,yCAAyC,EAAE,GAAG,EAAE,CAAC,CAAC;IAC1D,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE;CAC5B,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,MAAM,OAAO,GAAG;gBACd,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAChC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,uBAAuB,EAAE;gBAC/C,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,8BAA8B,EAAE,EAAE,yBAAyB;aAClF,CAAC;YAEF,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;YACtE,MAAM,OAAO,GAAG,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC,CAAC;YAEjE,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;YAC7E,MAAM,OAAO,GAAG;gBACd,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,8BAA8B,EAAE;gBACtD,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,EAAE,kBAAkB,EAAE;aAC9C,CAAC;YAEF,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,OAAO,GAAU,EAAE,CAAC;YAE1B,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,OAAO,GAAG;gBACd,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,EAAE,gBAAgB,EAAE;gBAC3C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE;gBACrC,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,4BAA4B,EAAE;gBACpD,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE;gBACtC,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,WAAW,EAAE;aACvC,CAAC;YAEF,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,OAAO,GAAG;gBACd,IAAI;gBACJ,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE;gBACrC,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,wBAAwB;gBAC1C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,oBAAoB;gBAClD,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qBAAqB,EAAE;aAC9C,CAAC;YAEF,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,iCAAiC,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;QAC1C,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,OAAO,GAAG,sBAAsB,CAAC;YAEvC,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,OAAO,GAAG,IAAI,CAAC;YAErB,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;YACzC,MAAM,OAAO,GAAG,SAAS,CAAC;YAE1B,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,OAAO,GAAG,EAAE,CAAC;YAEnB,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,OAAO,GAAG,KAAK,CAAC;YAEtB,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,OAAO,GAAG,EAAE,KAAK,EAAE,sBAAsB,EAAE,CAAC;YAElD,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,OAAO,GAAG;gBACd,KAAK,EAAE,sBAAsB;gBAC7B,IAAI,EAAE,GAAG;gBACT,OAAO,EAAE,EAAE,OAAO,EAAE,uBAAuB,EAAE;aAC9C,CAAC;YAEF,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CACjB,2FAA2F,CAC5F,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;YAC5C,MAAM,OAAO,GAAG,EAAE,CAAC;YAEnB,MAAM,MAAM,GAAG,IAAA,wCAAmB,EAAC,OAAO,CAAC,CAAC;YAE5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,sCAAsC,EAAE,GAAG,EAAE;IACpD,UAAU,CAAC,GAAG,EAAE;QACd,gDAAgD;QAC/C,0CAAyC,CAAC,eAAe,CAAC;YACzD,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;SACxB,CAAC,CAAC;QAEH,0EAA0E;QACzE,sBAA4B,CAAC,MAAM,GAAG,IAAI;aACxC,EAAE,EAAE;aACJ,iBAAiB,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,MAAM,OAAO,GAA2D,CAAC,EACvE,QAAQ,GACT,EAAE,EAAE;QACH,MAAM,OAAO,GAAG,IAAA,uCAAkB,GAAE,CAAC;QACrC,IAAA,iBAAS,EAAC,GAAG,EAAE;YACb,QAAQ,CAAC,OAAO,CAAC,CAAC;QACpB,CAAC,EAAE,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC;QACxB,OAAO,sDAAiB,MAAM,IAAE,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAO,CAAC;IAC7E,CAAC,CAAC;IAEF,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,IAAI,MAAM,GAAgB,EAAE,CAAC;QAC7B,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,IAAA,cAAM,EACtC,8BAAC,qCAAgB,IAAC,UAAU,EAAE,CAAC,mBAAmB,CAAC;YACjD,8BAAC,OAAO,IAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,GAAI,CACzB,CACpB,CAAC;QAEF,8BAA8B;QAC9B,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,iBAAiB;QACjB,QAAQ,CACN,8BAAC,qCAAgB,IAAC,UAAU,EAAE,CAAC,mBAAmB,EAAE,mBAAmB,CAAC;YACtE,8BAAC,OAAO,IAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,GAAI,CACzB,CACpB,CAAC;QAEF,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,IAAI,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,WAAW,IAAI,EAAE,CAAC;YACnD,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;YAC5C,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,IAAI,MAAM,GAAgB,EAAE,CAAC;QAC7B,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,IAAA,cAAM,EACtC,8BAAC,qCAAgB,IAAC,UAAU,EAAE,CAAC,mBAAmB,EAAE,mBAAmB,CAAC;YACtE,8BAAC,OAAO,IAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,GAAI,CACzB,CACpB,CAAC;QAEF,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,oBAAoB;QACpB,QAAQ,CACN,8BAAC,qCAAgB,IAAC,UAAU,EAAE,CAAC,mBAAmB,CAAC;YACjD,8BAAC,OAAO,IAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,GAAI,CACzB,CACpB,CAAC;QAEF,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,IAAI,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,WAAW,IAAI,EAAE,CAAC;YACnD,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;YAC5C,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,IAAI,MAAM,GAAgB,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAG;YACd,EAAE,GAAG,EAAE,mBAAmB,EAAE,SAAS,EAAE,yBAAY,CAAC,GAAG,EAAE;YACzD,EAAE,GAAG,EAAE,mBAAmB,EAAE,SAAS,EAAE,yBAAY,CAAC,GAAG,EAAE;SAC1D,CAAC;QAEF,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,cAAM,EACzB,8BAAC,qCAAgB,IAAC,UAAU,EAAE,OAAO;YACnC,8BAAC,OAAO,IAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,GAAI,CACzB,CACpB,CAAC;QAEF,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,0DAA0D;QAC1D,MAAM,IAAI,GAAG;YACX,EAAE,GAAG,EAAE,mBAAmB,EAAE,SAAS,EAAE,yBAAY,CAAC,GAAG,EAAE;YACzD,EAAE,GAAG,EAAE,mBAAmB,EAAE,SAAS,EAAE,yBAAY,CAAC,GAAG,EAAE;SAC1D,CAAC;QACF,QAAQ,CACN,8BAAC,qCAAgB,IAAC,UAAU,EAAE,IAAI;YAChC,8BAAC,OAAO,IAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,GAAI,CACzB,CACpB,CAAC;QAEF,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;YAC7C,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,mBAAmB,EAAE,mBAAmB,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { render, waitFor } from \"@testing-library/react\";\nimport React, { useEffect } from \"react\";\nimport { useTamboRegistry } from \"../../providers/tambo-registry-provider\";\nimport { MCPClient, MCPTransport } from \"../mcp-client\";\nimport {\n  extractErrorMessage,\n  TamboMcpProvider,\n  useTamboMcpServers,\n  type McpServer,\n} from \"../tambo-mcp-provider\";\n\n// Mock the MCP client to avoid ES module issues\njest.mock(\"../mcp-client\", () => ({\n  MCPClient: jest.fn(),\n  MCPTransport: {\n    SSE: \"sse\",\n    HTTP: \"http\",\n  },\n}));\n\n// Mock the registry provider to avoid dependency issues\njest.mock(\"../../providers/tambo-registry-provider\", () => ({\n  useTamboRegistry: jest.fn(),\n}));\n\ndescribe(\"extractErrorMessage\", () => {\n  describe(\"Array content handling\", () => {\n    it(\"should extract text from array content with multiple text items\", () => {\n      const content = [\n        { type: \"text\", text: \"Error:\" },\n        { type: \"text\", text: \"Tool execution failed\" },\n        { type: \"image\", url: \"http://example.com/error.png\" }, // Should be filtered out\n      ];\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"Error: Tool execution failed\");\n    });\n\n    it(\"should extract text from array content with single text item\", () => {\n      const content = [{ type: \"text\", text: \"Simple error message\" }];\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"Simple error message\");\n    });\n\n    it(\"should return fallback message for array content with no text items\", () => {\n      const content = [\n        { type: \"image\", url: \"http://example.com/error.png\" },\n        { type: \"resource\", uri: \"file://error.log\" },\n      ];\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"Error occurred but no details provided\");\n    });\n\n    it(\"should return fallback message for empty array content\", () => {\n      const content: any[] = [];\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"Error occurred but no details provided\");\n    });\n\n    it(\"should handle array content with mixed types correctly\", () => {\n      const content = [\n        { type: \"resource\", uri: \"file://log.txt\" },\n        { type: \"text\", text: \"First error\" },\n        { type: \"image\", url: \"http://example.com/img.png\" },\n        { type: \"text\", text: \"Second error\" },\n        { type: \"unknown\", data: \"something\" },\n      ];\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"First error Second error\");\n    });\n\n    it(\"should handle array content with malformed items\", () => {\n      const content = [\n        null,\n        { type: \"text\", text: \"Valid error\" },\n        { type: \"text\" }, // Missing text property\n        { type: \"text\", text: null }, // Invalid text type\n        { type: \"text\", text: \"Another valid error\" },\n      ];\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"Valid error Another valid error\");\n    });\n  });\n\n  describe(\"Non-array content handling\", () => {\n    it(\"should return string content as-is\", () => {\n      const content = \"Direct error message\";\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"Direct error message\");\n    });\n\n    it(\"should handle null content\", () => {\n      const content = null;\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"Unknown error occurred\");\n    });\n\n    it(\"should handle undefined content\", () => {\n      const content = undefined;\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"Unknown error occurred\");\n    });\n\n    it(\"should handle number content\", () => {\n      const content = 42;\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"42\");\n    });\n\n    it(\"should handle boolean content\", () => {\n      const content = false;\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"false\");\n    });\n\n    it(\"should handle object content\", () => {\n      const content = { error: \"Something went wrong\" };\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe('{\"error\":\"Something went wrong\"}');\n    });\n\n    it(\"should handle complex object content\", () => {\n      const content = {\n        error: \"Something went wrong\",\n        code: 500,\n        details: { message: \"Internal server error\" },\n      };\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\n        '{\"error\":\"Something went wrong\",\"code\":500,\"details\":{\"message\":\"Internal server error\"}}',\n      );\n    });\n\n    it(\"should handle empty object content\", () => {\n      const content = {};\n\n      const result = extractErrorMessage(content);\n\n      expect(result).toBe(\"{}\");\n    });\n  });\n});\n\ndescribe(\"TamboMcpProvider server list changes\", () => {\n  beforeEach(() => {\n    // Mock registry so tool registration is a no-op\n    (useTamboRegistry as unknown as jest.Mock).mockReturnValue({\n      registerTool: jest.fn(),\n    });\n\n    // Ensure MCPClient.create exists and returns a fake client with listTools\n    (MCPClient as unknown as any).create = jest\n      .fn()\n      .mockResolvedValue({ listTools: jest.fn().mockResolvedValue([]) });\n  });\n\n  const Capture: React.FC<{ onUpdate: (servers: McpServer[]) => void }> = ({\n    onUpdate,\n  }) => {\n    const servers = useTamboMcpServers();\n    useEffect(() => {\n      onUpdate(servers);\n    }, [servers, onUpdate]);\n    return <div data-testid=\"urls\">{servers.map((s) => s.url).join(\",\")}</div>;\n  };\n\n  it(\"adds a new server when the list grows\", async () => {\n    let latest: McpServer[] = [];\n    const { rerender, getByTestId } = render(\n      <TamboMcpProvider mcpServers={[\"https://a.example\"]}>\n        <Capture onUpdate={(s) => (latest = s)} />\n      </TamboMcpProvider>,\n    );\n\n    // Wait for initial connection\n    await waitFor(() => {\n      expect(latest.length).toBe(1);\n    });\n\n    // Add new server\n    rerender(\n      <TamboMcpProvider mcpServers={[\"https://a.example\", \"https://b.example\"]}>\n        <Capture onUpdate={(s) => (latest = s)} />\n      </TamboMcpProvider>,\n    );\n\n    await waitFor(() => {\n      expect(latest.length).toBe(2);\n      const urls = getByTestId(\"urls\").textContent || \"\";\n      expect(urls).toContain(\"https://a.example\");\n      expect(urls).toContain(\"https://b.example\");\n    });\n  });\n\n  it(\"removes a server when the list shrinks\", async () => {\n    let latest: McpServer[] = [];\n    const { rerender, getByTestId } = render(\n      <TamboMcpProvider mcpServers={[\"https://a.example\", \"https://b.example\"]}>\n        <Capture onUpdate={(s) => (latest = s)} />\n      </TamboMcpProvider>,\n    );\n\n    await waitFor(() => {\n      expect(latest.length).toBe(2);\n    });\n\n    // Remove one server\n    rerender(\n      <TamboMcpProvider mcpServers={[\"https://a.example\"]}>\n        <Capture onUpdate={(s) => (latest = s)} />\n      </TamboMcpProvider>,\n    );\n\n    await waitFor(() => {\n      expect(latest.length).toBe(1);\n      const urls = getByTestId(\"urls\").textContent || \"\";\n      expect(urls).toContain(\"https://a.example\");\n      expect(urls).not.toContain(\"https://b.example\");\n    });\n  });\n\n  it(\"does not duplicate when a new copy of the same list is passed\", async () => {\n    let latest: McpServer[] = [];\n    const initial = [\n      { url: \"https://a.example\", transport: MCPTransport.SSE },\n      { url: \"https://b.example\", transport: MCPTransport.SSE },\n    ];\n\n    const { rerender } = render(\n      <TamboMcpProvider mcpServers={initial}>\n        <Capture onUpdate={(s) => (latest = s)} />\n      </TamboMcpProvider>,\n    );\n\n    await waitFor(() => {\n      expect(latest.length).toBe(2);\n    });\n\n    // Pass a new array instance with the same logical servers\n    const same = [\n      { url: \"https://a.example\", transport: MCPTransport.SSE },\n      { url: \"https://b.example\", transport: MCPTransport.SSE },\n    ];\n    rerender(\n      <TamboMcpProvider mcpServers={same}>\n        <Capture onUpdate={(s) => (latest = s)} />\n      </TamboMcpProvider>,\n    );\n\n    await waitFor(() => {\n      expect(latest.length).toBe(2);\n      const urls = latest.map((s) => s.url).sort();\n      expect(urls).toEqual([\"https://a.example\", \"https://b.example\"].sort());\n    });\n  });\n});\n"]}

package/dist/mcp/__tests__/use-mcp-servers.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=use-mcp-servers.test.d.ts.map

package/dist/mcp/__tests__/use-mcp-servers.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-mcp-servers.test.d.ts","sourceRoot":"","sources":["../../../src/mcp/__tests__/use-mcp-servers.test.tsx"],"names":[],"mappings":""}

package/dist/mcp/__tests__/use-mcp-servers.test.js

@@ -0,0 +1,118 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+const react_1 = require("@testing-library/react");
+const react_2 = __importStar(require("react"));
+const tambo_registry_provider_1 = require("../../providers/tambo-registry-provider");
+const tambo_mcp_provider_1 = require("../tambo-mcp-provider");
+// Mock the registry to provide a no-op registerTool
+// Do not mock the registry; use the real provider in render
+// Mock the MCP client; use a mutable implementation to avoid TDZ issues
+let createImpl = jest.fn();
+jest.mock("../mcp-client", () => ({
+    MCPClient: { create: (...args) => createImpl(...args) },
+    MCPTransport: { SSE: "sse", HTTP: "http" },
+}));
+// Import after mocks note: jest.mock calls are hoisted, so standard imports are fine
+describe("useTamboMcpServers + TamboMcpProvider", () => {
+    beforeEach(() => {
+        createImpl = jest.fn();
+    });
+    it("provides normalized MCP server entries to inner components", async () => {
+        const fakeClient = { listTools: jest.fn().mockResolvedValue([]) };
+        createImpl.mockResolvedValue(fakeClient);
+        const Inner = () => {
+            const servers = (0, tambo_mcp_provider_1.useTamboMcpServers)();
+            return (react_2.default.createElement("div", null,
+                react_2.default.createElement("div", { "data-testid": "count" }, servers.length),
+                react_2.default.createElement("div", { "data-testid": "urls" }, servers.map((s) => s.url).join(","))));
+        };
+        const { getByTestId } = (0, react_1.render)(react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null,
+            react_2.default.createElement(tambo_mcp_provider_1.TamboMcpProvider, { mcpServers: [{ url: "https://one.example" }, "https://two.example"] },
+                react_2.default.createElement(Inner, null))));
+        await (0, react_1.waitFor)(() => {
+            expect(getByTestId("count").textContent).toBe("2");
+            const urls = getByTestId("urls").textContent || "";
+            expect(urls).toContain("https://one.example");
+            expect(urls).toContain("https://two.example");
+        });
+    });
+    it("marks a successfully connected server with a client instance", async () => {
+        const fakeClient = { listTools: jest.fn().mockResolvedValue([]) };
+        createImpl.mockResolvedValue(fakeClient);
+        let latest = [];
+        const Capture = () => {
+            const servers = (0, tambo_mcp_provider_1.useTamboMcpServers)();
+            (0, react_2.useEffect)(() => {
+                latest = servers;
+            }, [servers]);
+            return null;
+        };
+        (0, react_1.render)(react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null,
+            react_2.default.createElement(tambo_mcp_provider_1.TamboMcpProvider, { mcpServers: [{ url: "https://ok.example" }] },
+                react_2.default.createElement(Capture, null))));
+        await (0, react_1.waitFor)(() => {
+            expect(latest.length).toBe(1);
+            expect("client" in latest[0]).toBe(true);
+            expect(latest[0].client).toBe(fakeClient);
+            // no connectionError on connected server
+            expect(latest[0].connectionError).toBeUndefined();
+        });
+    });
+    it("marks a failed server with a connectionError and no client", async () => {
+        const boom = new Error("boom");
+        createImpl.mockRejectedValue(boom);
+        let latest = [];
+        const Capture = () => {
+            const servers = (0, tambo_mcp_provider_1.useTamboMcpServers)();
+            (0, react_2.useEffect)(() => {
+                latest = servers;
+            }, [servers]);
+            return null;
+        };
+        (0, react_1.render)(react_2.default.createElement(tambo_registry_provider_1.TamboRegistryProvider, null,
+            react_2.default.createElement(tambo_mcp_provider_1.TamboMcpProvider, { mcpServers: [{ url: "https://fail.example" }] },
+                react_2.default.createElement(Capture, null))));
+        await (0, react_1.waitFor)(() => {
+            expect(latest.length).toBe(1);
+            expect("client" in latest[0]).toBe(false);
+            // @ts-expect-error narrowing at runtime
+            expect(latest[0].connectionError).toBeInstanceOf(Error);
+            // @ts-expect-error narrowing at runtime
+            expect(latest[0].connectionError?.message).toBe("boom");
+        });
+    });
+});
+//# sourceMappingURL=use-mcp-servers.test.js.map

package/dist/mcp/__tests__/use-mcp-servers.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"use-mcp-servers.test.js","sourceRoot":"","sources":["../../../src/mcp/__tests__/use-mcp-servers.test.tsx"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,kDAAyD;AACzD,+CAAyC;AACzC,qFAAgF;AAChF,8DAI+B;AAE/B,oDAAoD;AACpD,4DAA4D;AAE5D,wEAAwE;AACxE,IAAI,UAAU,GAAwB,IAAI,CAAC,EAAE,EAAE,CAAC;AAChD,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;IAChC,SAAS,EAAE,EAAE,MAAM,EAAE,CAAC,GAAG,IAAW,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,EAAE;IAC9D,YAAY,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE;CAC3C,CAAC,CAAC,CAAC;AAEJ,qFAAqF;AAErF,QAAQ,CAAC,uCAAuC,EAAE,GAAG,EAAE;IACrD,UAAU,CAAC,GAAG,EAAE;QACd,UAAU,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,UAAU,GAAG,EAAE,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,EAAS,CAAC;QACzE,UAAU,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;QAEzC,MAAM,KAAK,GAAa,GAAG,EAAE;YAC3B,MAAM,OAAO,GAAG,IAAA,uCAAkB,GAAE,CAAC;YACrC,OAAO,CACL;gBACE,sDAAiB,OAAO,IAAE,OAAO,CAAC,MAAM,CAAO;gBAC/C,sDAAiB,MAAM,IAAE,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAO,CAC/D,CACP,CAAC;QACJ,CAAC,CAAC;QAEF,MAAM,EAAE,WAAW,EAAE,GAAG,IAAA,cAAM,EAC5B,8BAAC,+CAAqB;YACpB,8BAAC,qCAAgB,IACf,UAAU,EAAE,CAAC,EAAE,GAAG,EAAE,qBAAqB,EAAE,EAAE,qBAAqB,CAAC;gBAEnE,8BAAC,KAAK,OAAG,CACQ,CACG,CACzB,CAAC;QAEF,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACnD,MAAM,IAAI,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC,WAAW,IAAI,EAAE,CAAC;YACnD,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;YAC9C,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,UAAU,GAAG,EAAE,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,EAAS,CAAC;QACzE,UAAU,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;QAEzC,IAAI,MAAM,GAAgB,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAa,GAAG,EAAE;YAC7B,MAAM,OAAO,GAAG,IAAA,uCAAkB,GAAE,CAAC;YACrC,IAAA,iBAAS,EAAC,GAAG,EAAE;gBACb,MAAM,GAAG,OAAO,CAAC;YACnB,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;YACd,OAAO,IAAI,CAAC;QACd,CAAC,CAAC;QAEF,IAAA,cAAM,EACJ,8BAAC,+CAAqB;YACpB,8BAAC,qCAAgB,IAAC,UAAU,EAAE,CAAC,EAAE,GAAG,EAAE,oBAAoB,EAAE,CAAC;gBAC3D,8BAAC,OAAO,OAAG,CACM,CACG,CACzB,CAAC;QAEF,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,CAAE,MAAM,CAAC,CAAC,CAAS,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACnD,yCAAyC;YACzC,MAAM,CAAE,MAAM,CAAC,CAAC,CAAS,CAAC,eAAe,CAAC,CAAC,aAAa,EAAE,CAAC;QAC7D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,IAAI,GAAG,IAAI,KAAK,CAAC,MAAM,CAAC,CAAC;QAC/B,UAAU,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAEnC,IAAI,MAAM,GAAgB,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAa,GAAG,EAAE;YAC7B,MAAM,OAAO,GAAG,IAAA,uCAAkB,GAAE,CAAC;YACrC,IAAA,iBAAS,EAAC,GAAG,EAAE;gBACb,MAAM,GAAG,OAAO,CAAC;YACnB,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;YACd,OAAO,IAAI,CAAC;QACd,CAAC,CAAC;QAEF,IAAA,cAAM,EACJ,8BAAC,+CAAqB;YACpB,8BAAC,qCAAgB,IAAC,UAAU,EAAE,CAAC,EAAE,GAAG,EAAE,sBAAsB,EAAE,CAAC;gBAC7D,8BAAC,OAAO,OAAG,CACM,CACG,CACzB,CAAC;QAEF,MAAM,IAAA,eAAO,EAAC,GAAG,EAAE;YACjB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1C,wCAAwC;YACxC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YACxD,wCAAwC;YACxC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC","sourcesContent":["import { render, waitFor } from \"@testing-library/react\";\nimport React, { useEffect } from \"react\";\nimport { TamboRegistryProvider } from \"../../providers/tambo-registry-provider\";\nimport {\n  TamboMcpProvider,\n  useTamboMcpServers,\n  type McpServer,\n} from \"../tambo-mcp-provider\";\n\n// Mock the registry to provide a no-op registerTool\n// Do not mock the registry; use the real provider in render\n\n// Mock the MCP client; use a mutable implementation to avoid TDZ issues\nlet createImpl: jest.Mock<any, any> = jest.fn();\njest.mock(\"../mcp-client\", () => ({\n  MCPClient: { create: (...args: any[]) => createImpl(...args) },\n  MCPTransport: { SSE: \"sse\", HTTP: \"http\" },\n}));\n\n// Import after mocks note: jest.mock calls are hoisted, so standard imports are fine\n\ndescribe(\"useTamboMcpServers + TamboMcpProvider\", () => {\n  beforeEach(() => {\n    createImpl = jest.fn();\n  });\n\n  it(\"provides normalized MCP server entries to inner components\", async () => {\n    const fakeClient = { listTools: jest.fn().mockResolvedValue([]) } as any;\n    createImpl.mockResolvedValue(fakeClient);\n\n    const Inner: React.FC = () => {\n      const servers = useTamboMcpServers();\n      return (\n        <div>\n          <div data-testid=\"count\">{servers.length}</div>\n          <div data-testid=\"urls\">{servers.map((s) => s.url).join(\",\")}</div>\n        </div>\n      );\n    };\n\n    const { getByTestId } = render(\n      <TamboRegistryProvider>\n        <TamboMcpProvider\n          mcpServers={[{ url: \"https://one.example\" }, \"https://two.example\"]}\n        >\n          <Inner />\n        </TamboMcpProvider>\n      </TamboRegistryProvider>,\n    );\n\n    await waitFor(() => {\n      expect(getByTestId(\"count\").textContent).toBe(\"2\");\n      const urls = getByTestId(\"urls\").textContent || \"\";\n      expect(urls).toContain(\"https://one.example\");\n      expect(urls).toContain(\"https://two.example\");\n    });\n  });\n\n  it(\"marks a successfully connected server with a client instance\", async () => {\n    const fakeClient = { listTools: jest.fn().mockResolvedValue([]) } as any;\n    createImpl.mockResolvedValue(fakeClient);\n\n    let latest: McpServer[] = [];\n    const Capture: React.FC = () => {\n      const servers = useTamboMcpServers();\n      useEffect(() => {\n        latest = servers;\n      }, [servers]);\n      return null;\n    };\n\n    render(\n      <TamboRegistryProvider>\n        <TamboMcpProvider mcpServers={[{ url: \"https://ok.example\" }]}>\n          <Capture />\n        </TamboMcpProvider>\n      </TamboRegistryProvider>,\n    );\n\n    await waitFor(() => {\n      expect(latest.length).toBe(1);\n      expect(\"client\" in latest[0]).toBe(true);\n      expect((latest[0] as any).client).toBe(fakeClient);\n      // no connectionError on connected server\n      expect((latest[0] as any).connectionError).toBeUndefined();\n    });\n  });\n\n  it(\"marks a failed server with a connectionError and no client\", async () => {\n    const boom = new Error(\"boom\");\n    createImpl.mockRejectedValue(boom);\n\n    let latest: McpServer[] = [];\n    const Capture: React.FC = () => {\n      const servers = useTamboMcpServers();\n      useEffect(() => {\n        latest = servers;\n      }, [servers]);\n      return null;\n    };\n\n    render(\n      <TamboRegistryProvider>\n        <TamboMcpProvider mcpServers={[{ url: \"https://fail.example\" }]}>\n          <Capture />\n        </TamboMcpProvider>\n      </TamboRegistryProvider>,\n    );\n\n    await waitFor(() => {\n      expect(latest.length).toBe(1);\n      expect(\"client\" in latest[0]).toBe(false);\n      // @ts-expect-error narrowing at runtime\n      expect(latest[0].connectionError).toBeInstanceOf(Error);\n      // @ts-expect-error narrowing at runtime\n      expect(latest[0].connectionError?.message).toBe(\"boom\");\n    });\n  });\n});\n"]}

package/dist/mcp/index.d.ts

@@ -1,3 +1,3 @@
 export { MCPTransport } from "./mcp-client";
-export { TamboMcpProvider, type McpServerInfo } from "./tambo-mcp-provider";
+export { TamboMcpProvider, useTamboMcpServers, type ConnectedMcpServer, type FailedMcpServer, type McpServer, type McpServerInfo, } from "./tambo-mcp-provider";
 //# sourceMappingURL=index.d.ts.map

package/dist/mcp/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/mcp/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAC5C,OAAO,EAAE,gBAAgB,EAAE,KAAK,aAAa,EAAE,MAAM,sBAAsB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/mcp/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAC5C,OAAO,EACL,gBAAgB,EAChB,kBAAkB,EAClB,KAAK,kBAAkB,EACvB,KAAK,eAAe,EACpB,KAAK,SAAS,EACd,KAAK,aAAa,GACnB,MAAM,sBAAsB,CAAC"}

package/dist/mcp/index.js

@@ -1,8 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.TamboMcpProvider = exports.MCPTransport = void 0;
+exports.useTamboMcpServers = exports.TamboMcpProvider = exports.MCPTransport = void 0;
 var mcp_client_1 = require("./mcp-client");
 Object.defineProperty(exports, "MCPTransport", { enumerable: true, get: function () { return mcp_client_1.MCPTransport; } });
 var tambo_mcp_provider_1 = require("./tambo-mcp-provider");
 Object.defineProperty(exports, "TamboMcpProvider", { enumerable: true, get: function () { return tambo_mcp_provider_1.TamboMcpProvider; } });
+Object.defineProperty(exports, "useTamboMcpServers", { enumerable: true, get: function () { return tambo_mcp_provider_1.useTamboMcpServers; } });
 //# sourceMappingURL=index.js.map

package/dist/mcp/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/mcp/index.ts"],"names":[],"mappings":";;;AAAA,2CAA4C;AAAnC,0GAAA,YAAY,OAAA;AACrB,2DAA4E;AAAnE,sHAAA,gBAAgB,OAAA","sourcesContent":["export { MCPTransport } from \"./mcp-client\";\nexport { TamboMcpProvider, type McpServerInfo } from \"./tambo-mcp-provider\";\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/mcp/index.ts"],"names":[],"mappings":";;;AAAA,2CAA4C;AAAnC,0GAAA,YAAY,OAAA;AACrB,2DAO8B;AAN5B,sHAAA,gBAAgB,OAAA;AAChB,wHAAA,kBAAkB,OAAA","sourcesContent":["export { MCPTransport } from \"./mcp-client\";\nexport {\n  TamboMcpProvider,\n  useTamboMcpServers,\n  type ConnectedMcpServer,\n  type FailedMcpServer,\n  type McpServer,\n  type McpServerInfo,\n} from \"./tambo-mcp-provider\";\n"]}

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

@@ -1,5 +1,5 @@
-import { FC } from "react";
-import { MCPTransport } from "./mcp-client";
+import React, { FC } from "react";
+import { MCPClient, MCPTransport } from "./mcp-client";
 /**
  * Extracts error message from MCP tool result content.
  * Handles both array and string content formats.
@@ -14,6 +14,14 @@ export interface McpServerInfo {
     transport?: MCPTransport;
     customHeaders?: Record<string, string>;
 }
+export interface ConnectedMcpServer extends McpServerInfo {
+    client: MCPClient;
+}
+export interface FailedMcpServer extends McpServerInfo {
+    client?: never;
+    connectionError: Error;
+}
+export type McpServer = ConnectedMcpServer | FailedMcpServer;
 /**
  * This provider is used to register tools from MCP servers.
  * @returns the wrapped children
@@ -22,4 +30,28 @@ export declare const TamboMcpProvider: FC<{
     mcpServers: (McpServerInfo | string)[];
     children: React.ReactNode;
 }>;
+/**
+ * Hook to access the actual MCP servers, as they are connected (or fail to
+ * connect).
+ *
+ * You can call methods on the MCP client that is included in the MCP server
+ * object.
+ *
+ * If the server fails to connect, the `client` property will be `undefined` and
+ * the `connectionError` property will be set.
+ *
+ * For example, to forcibly disconnect and reconnect all MCP servers:
+ *
+ * ```tsx
+ * const mcpServers = useTamboMcpServers();
+ * mcpServers.forEach((mcpServer) => {
+ *   mcpServer.client?.reconnect();
+ * });
+ * ```
+ *
+ * Note that the MCP servers are not guaranteed to be in the same order as the
+ * input array, because they are added as they are connected.
+ * @returns The MCP servers
+ */
+export declare const useTamboMcpServers: () => McpServer[];
 //# sourceMappingURL=tambo-mcp-provider.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"tambo-mcp-provider.d.ts","sourceRoot":"","sources":["../../src/mcp/tambo-mcp-provider.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAa,MAAM,OAAO,CAAC;AAGtC,OAAO,EAAa,YAAY,EAAE,MAAM,cAAc,CAAC;AAEvD;;;;;GAKG;AACH,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM,CAsB5D;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,YAAY,CAAC;IACzB,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACxC;AACD;;;GAGG;AACH,eAAO,MAAM,gBAAgB,EAAE,EAAE,CAAC;IAChC,UAAU,EAAE,CAAC,aAAa,GAAG,MAAM,CAAC,EAAE,CAAC;IACvC,QAAQ,EAAE,KAAK,CAAC,SAAS,CAAC;CAC3B,CAkEA,CAAC"}
+{"version":3,"file":"tambo-mcp-provider.d.ts","sourceRoot":"","sources":["../../src/mcp/tambo-mcp-provider.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,EAEZ,EAAE,EAIH,MAAM,OAAO,CAAC;AAGf,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAEvD;;;;;GAKG;AACH,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM,CAsB5D;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,GAAG,EAAE,MAAM,CAAC;IACZ,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,YAAY,CAAC;IACzB,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CACxC;AAED,MAAM,WAAW,kBAAmB,SAAQ,aAAa;IACvD,MAAM,EAAE,SAAS,CAAC;CACnB;AAED,MAAM,WAAW,eAAgB,SAAQ,aAAa;IACpD,MAAM,CAAC,EAAE,KAAK,CAAC;IACf,eAAe,EAAE,KAAK,CAAC;CACxB;AAED,MAAM,MAAM,SAAS,GAAG,kBAAkB,GAAG,eAAe,CAAC;AAG7D;;;GAGG;AACH,eAAO,MAAM,gBAAgB,EAAE,EAAE,CAAC;IAChC,UAAU,EAAE,CAAC,aAAa,GAAG,MAAM,CAAC,EAAE,CAAC;IACvC,QAAQ,EAAE,KAAK,CAAC,SAAS,CAAC;CAC3B,CAsIA,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,eAAO,MAAM,kBAAkB,mBAE9B,CAAC"}

package/dist/mcp/tambo-mcp-provider.js

@@ -1,8 +1,42 @@
 "use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.TamboMcpProvider = void 0;
+exports.useTamboMcpServers = exports.TamboMcpProvider = void 0;
 exports.extractErrorMessage = extractErrorMessage;
-const react_1 = require("react");
+const fast_equals_1 = require("fast-equals");
+const react_1 = __importStar(require("react"));
 const tambo_registry_provider_1 = require("../providers/tambo-registry-provider");
 const mcp_client_1 = require("./mcp-client");
 /**
@@ -28,28 +62,65 @@ function extractErrorMessage(content) {
     }
     return `${content}`;
 }
+const McpProviderContext = (0, react_1.createContext)([]);
 /**
  * This provider is used to register tools from MCP servers.
  * @returns the wrapped children
  */
 const TamboMcpProvider = ({ mcpServers, children }) => {
     const { registerTool } = (0, tambo_registry_provider_1.useTamboRegistry)();
+    const [connectedMcpServers, setConnectedMcpServers] = (0, react_1.useState)([]);
     (0, react_1.useEffect)(() => {
         if (!mcpServers) {
             return;
         }
-        async function registerMcpServers(mcpServers) {
+        async function registerMcpServers(mcpServerInfos) {
             // Maps tool names to the MCP client that registered them
             const mcpServerMap = new Map();
-            const serverToolLists = mcpServers.map(async (mcpServer) => {
-                const server = typeof mcpServer === "string"
-                    ? { url: mcpServer, transport: mcp_client_1.MCPTransport.SSE }
-                    : mcpServer;
-                const { url, transport = mcp_client_1.MCPTransport.SSE, customHeaders } = server;
-                const mcpClient = await mcp_client_1.MCPClient.create(url, transport, customHeaders);
-                const tools = await mcpClient.listTools();
+            setConnectedMcpServers((prev) => 
+            // remove any servers that are not in the new list
+            prev.filter((s) => mcpServerInfos.some((mcpServerInfo) => equalsMcpServer(s, mcpServerInfo))));
+            // initialize the MCP clients, converting McpServerInfo -> McpServer
+            const mcpServers = await Promise.allSettled(mcpServerInfos.map(async (mcpServerInfo) => {
+                try {
+                    const client = await mcp_client_1.MCPClient.create(mcpServerInfo.url, mcpServerInfo.transport, mcpServerInfo.customHeaders);
+                    const connectedMcpServer = {
+                        ...mcpServerInfo,
+                        client: client,
+                    };
+                    // note because the promises may resolve in any order, the resulting
+                    // array may not be in the same order as the input array
+                    setConnectedMcpServers((prev) => [
+                        // replace the server if it already exists
+                        ...prev.filter((s) => !equalsMcpServer(s, mcpServerInfo)),
+                        connectedMcpServer,
+                    ]);
+                    return connectedMcpServer;
+                }
+                catch (error) {
+                    const failedMcpServer = {
+                        ...mcpServerInfo,
+                        connectionError: error,
+                    };
+                    // note because the promises may resolve in any order, the resulting
+                    // array may not be in the same order as the input array
+                    setConnectedMcpServers((prev) => [
+                        // replace the server if it already exists
+                        ...prev.filter((s) => !equalsMcpServer(s, mcpServerInfo)),
+                        failedMcpServer,
+                    ]);
+                    return failedMcpServer;
+                }
+            }));
+            // note do not rely on the state
+            const connectedMcpServers = mcpServers
+                .filter((result) => result.status === "fulfilled")
+                .map((result) => result.value);
+            // Now create a map of tool name to MCP client
+            const serverToolLists = connectedMcpServers.map(async (mcpServer) => {
+                const tools = (await mcpServer.client?.listTools()) ?? [];
                 tools.forEach((tool) => {
-                    mcpServerMap.set(tool.name, mcpClient);
+                    mcpServerMap.set(tool.name, mcpServer);
                 });
                 return tools;
             });
@@ -74,7 +145,11 @@ const TamboMcpProvider = ({ mcpServers, children }) => {
                             // should never happen
                             throw new Error(`MCP server for tool ${tool.name} not found`);
                         }
-                        const result = await mcpServer.callTool(tool.name, args);
+                        if (!mcpServer.client) {
+                            // this can't actually happen because the tool can't be registered if the server is not connected
+                            throw new Error(`MCP server for tool ${tool.name} is not connected`);
+                        }
+                        const result = await mcpServer.client.callTool(tool.name, args);
                         if (result.isError) {
                             const errorMessage = extractErrorMessage(result.content);
                             throw new Error(errorMessage);
@@ -85,9 +160,45 @@ const TamboMcpProvider = ({ mcpServers, children }) => {
                 });
             });
         }
-        registerMcpServers(mcpServers);
+        // normalize the server infos
+        const mcpServerInfos = mcpServers.map((mcpServer) => typeof mcpServer === "string"
+            ? { url: mcpServer, transport: mcp_client_1.MCPTransport.SSE }
+            : mcpServer);
+        registerMcpServers(mcpServerInfos);
     }, [mcpServers, registerTool]);
-    return children;
+    return (react_1.default.createElement(McpProviderContext.Provider, { value: connectedMcpServers }, children));
 };
 exports.TamboMcpProvider = TamboMcpProvider;
+/**
+ * Hook to access the actual MCP servers, as they are connected (or fail to
+ * connect).
+ *
+ * You can call methods on the MCP client that is included in the MCP server
+ * object.
+ *
+ * If the server fails to connect, the `client` property will be `undefined` and
+ * the `connectionError` property will be set.
+ *
+ * For example, to forcibly disconnect and reconnect all MCP servers:
+ *
+ * ```tsx
+ * const mcpServers = useTamboMcpServers();
+ * mcpServers.forEach((mcpServer) => {
+ *   mcpServer.client?.reconnect();
+ * });
+ * ```
+ *
+ * Note that the MCP servers are not guaranteed to be in the same order as the
+ * input array, because they are added as they are connected.
+ * @returns The MCP servers
+ */
+const useTamboMcpServers = () => {
+    return (0, react_1.useContext)(McpProviderContext);
+};
+exports.useTamboMcpServers = useTamboMcpServers;
+function equalsMcpServer(s, mcpServerInfo) {
+    return (s.url === mcpServerInfo.url &&
+        s.transport === mcpServerInfo.transport &&
+        (0, fast_equals_1.deepEqual)(s.customHeaders, mcpServerInfo.customHeaders));
+}
 //# sourceMappingURL=tambo-mcp-provider.js.map