@ai-sdk/mcp 0.0.18 → 0.0.20

package/CHANGELOG.md

@@ -1,5 +1,30 @@
 # @ai-sdk/mcp
 
+## 0.0.20
+
+### Patch Changes
+
+- 1673f7b: fix (mcp): handle SSE messages without explicit event fields
+
+## 0.0.19
+
+### Patch Changes
+
+- d872a7a: fix(mcp): lock first sse endpoint received via event
+- f4cd468: fix(mcp): prevent prototype-named tools from bypassing the `schemas` allowlist
+
+  When using `client.tools({ schemas })` to expose only an explicitly allowed
+  subset of an MCP server's tools, the allowlist check used the `in` operator,
+  which also matches inherited `Object.prototype` properties. A server-advertised
+  tool named `constructor`, `toString`, `__proto__`, etc. would pass the check
+  even though the developer never defined it in `schemas`, and was then exposed to
+  the model and executable. The check now uses `Object.hasOwn`, so only
+  explicitly defined tools are returned.
+
+- Updated dependencies [9f67efe]
+- Updated dependencies [eea9166]
+  - @ai-sdk/provider-utils@3.0.26
+
 ## 0.0.18
 
 ### Patch Changes

package/dist/index.js

@@ -1041,6 +1041,9 @@ async function authInternal(provider, {
 }
 
 // src/tool/mcp-sse-transport.ts
+function isMessageEvent(event) {
+  return event === void 0 || event === "message";
+}
 var SseMCPTransport = class {
   constructor({
     url,
@@ -1118,7 +1121,7 @@ var SseMCPTransport = class {
           const stream = response.body.pipeThrough(new TextDecoderStream()).pipeThrough(new import_provider_utils3.EventSourceParserStream());
           const reader = stream.getReader();
           const processEvents = async () => {
-            var _a4, _b4, _c2;
+            var _a4, _b4, _c2, _d2, _e2;
             try {
               while (true) {
                 const { done, value } = await reader.read();
@@ -1133,24 +1136,32 @@ var SseMCPTransport = class {
                 }
                 const { event, data } = value;
                 if (event === "endpoint") {
-                  this.endpoint = new URL(data, this.url);
-                  if (this.endpoint.origin !== this.url.origin) {
+                  if (this.endpoint) {
+                    continue;
+                  }
+                  const endpoint = new URL(data, this.url);
+                  if (endpoint.origin !== this.url.origin) {
+                    this.connected = false;
+                    this.endpoint = void 0;
+                    (_a4 = this.sseConnection) == null ? void 0 : _a4.close();
+                    (_b4 = this.abortController) == null ? void 0 : _b4.abort();
                     throw new MCPClientError({
-                      message: `MCP SSE Transport Error: Endpoint origin does not match connection origin: ${this.endpoint.origin}`
+                      message: `MCP SSE Transport Error: Endpoint origin does not match connection origin: ${endpoint.origin}`
                     });
                   }
+                  this.endpoint = endpoint;
                   this.connected = true;
                   resolve();
-                } else if (event === "message") {
+                } else if (isMessageEvent(event)) {
                   try {
                     const message = await parseJSONRPCMessage(data);
-                    (_a4 = this.onmessage) == null ? void 0 : _a4.call(this, message);
+                    (_c2 = this.onmessage) == null ? void 0 : _c2.call(this, message);
                   } catch (error) {
                     const e = new MCPClientError({
                       message: "MCP SSE Transport Error: Failed to parse message",
                       cause: error
                     });
-                    (_b4 = this.onerror) == null ? void 0 : _b4.call(this, e);
+                    (_d2 = this.onerror) == null ? void 0 : _d2.call(this, e);
                   }
                 }
               }
@@ -1158,7 +1169,7 @@ var SseMCPTransport = class {
               if (error instanceof Error && error.name === "AbortError") {
                 return;
               }
-              (_c2 = this.onerror) == null ? void 0 : _c2.call(this, error);
+              (_e2 = this.onerror) == null ? void 0 : _e2.call(this, error);
               reject(error);
             }
           };
@@ -1180,6 +1191,7 @@ var SseMCPTransport = class {
   async close() {
     var _a3, _b3, _c;
     this.connected = false;
+    this.endpoint = void 0;
     (_a3 = this.sseConnection) == null ? void 0 : _a3.close();
     (_b3 = this.abortController) == null ? void 0 : _b3.abort();
     (_c = this.onclose) == null ? void 0 : _c.call(this);
@@ -1241,6 +1253,9 @@ var SseMCPTransport = class {
 
 // src/tool/mcp-http-transport.ts
 var import_provider_utils4 = require("@ai-sdk/provider-utils");
+function isMessageEvent2(event) {
+  return event === void 0 || event === "message";
+}
 var HttpMCPTransport = class {
   constructor({
     url,
@@ -1383,7 +1398,7 @@ var HttpMCPTransport = class {
                 const { done, value } = await reader.read();
                 if (done) return;
                 const { event, data } = value;
-                if (event === "message") {
+                if (isMessageEvent2(event)) {
                   try {
                     const msg = await parseJSONRPCMessage(data);
                     (_a4 = this.onmessage) == null ? void 0 : _a4.call(this, msg);
@@ -1508,7 +1523,7 @@ var HttpMCPTransport = class {
             if (id) {
               this.lastInboundEventId = id;
             }
-            if (event === "message") {
+            if (isMessageEvent2(event)) {
               try {
                 const msg = await parseJSONRPCMessage(data);
                 (_a4 = this.onmessage) == null ? void 0 : _a4.call(this, msg);
@@ -1863,7 +1878,7 @@ var DefaultMCPClient = class {
     try {
       const listToolsResult = await this.listTools();
       for (const { name: name3, description, inputSchema } of listToolsResult.tools) {
-        if (schemas !== "automatic" && !(name3 in schemas)) {
+        if (schemas !== "automatic" && !Object.prototype.hasOwnProperty.call(schemas, name3)) {
           continue;
         }
         const self = this;