ai 4.1.61 → 4.1.62

package/dist/index.js

@@ -6673,79 +6673,11 @@ function tool(tool2) {
   return tool2;
 }
 
-// core/tool/mcp/create-child-process.ts
-async function createChildProcess(config, signal) {
-  var _a17, _b, _c;
-  const runtime = detectRuntime();
-  if (runtime !== "node") {
-    throw new MCPClientError({
-      message: "Attempted to use child_process module outside of Node.js environment"
-    });
-  }
-  let childProcess;
-  const nodePrefix = "node:";
-  try {
-    childProcess = await import(`${nodePrefix}child_process`);
-  } catch (error) {
-    try {
-      childProcess = require(`${nodePrefix}child_process`);
-    } catch (innerError) {
-      throw new MCPClientError({
-        message: "Failed to load child_process module dynamically",
-        cause: innerError
-      });
-    }
-  }
-  const { spawn } = childProcess;
-  return spawn(config.command, (_a17 = config.args) != null ? _a17 : [], {
-    env: (_b = config.env) != null ? _b : getDefaultEnvironment(),
-    stdio: ["pipe", "pipe", (_c = config.stderr) != null ? _c : "inherit"],
-    shell: false,
-    signal,
-    windowsHide: globalThis.process.platform === "win32" && isElectron(),
-    cwd: config.cwd
-  });
-}
-function detectRuntime() {
-  var _a17, _b;
-  if (typeof window !== "undefined") {
-    return "browser";
-  }
-  if (((_b = (_a17 = globalThis.process) == null ? void 0 : _a17.release) == null ? void 0 : _b.name) === "node") {
-    return "node";
-  }
-  return null;
-}
-function getDefaultEnvironment() {
-  const DEFAULT_INHERITED_ENV_VARS = globalThis.process.platform === "win32" ? [
-    "APPDATA",
-    "HOMEDRIVE",
-    "HOMEPATH",
-    "LOCALAPPDATA",
-    "PATH",
-    "PROCESSOR_ARCHITECTURE",
-    "SYSTEMDRIVE",
-    "SYSTEMROOT",
-    "TEMP",
-    "USERNAME",
-    "USERPROFILE"
-  ] : ["HOME", "LOGNAME", "PATH", "SHELL", "TERM", "USER"];
-  const env = {};
-  for (const key of DEFAULT_INHERITED_ENV_VARS) {
-    const value = globalThis.process.env[key];
-    if (value === void 0) {
-      continue;
-    }
-    if (value.startsWith("()")) {
-      continue;
-    }
-    env[key] = value;
-  }
-  return env;
-}
-function isElectron() {
-  return "type" in globalThis.process;
-}
+// core/tool/mcp/mcp-sse-transport.ts
+var import_stream = require("eventsource-parser/stream");
+
+// core/tool/mcp/json-rpc-message.ts
+var import_zod9 = require("zod");
 
 // core/tool/mcp/types.ts
 var import_zod8 = require("zod");
@@ -6754,7 +6686,6 @@ var SUPPORTED_PROTOCOL_VERSIONS = [
   LATEST_PROTOCOL_VERSION,
   "2024-10-07"
 ];
-var JSONRPC_VERSION = "2.0";
 var ClientOrServerImplementationSchema = import_zod8.z.object({
   name: import_zod8.z.string(),
   version: import_zod8.z.string()
@@ -6762,43 +6693,11 @@ var ClientOrServerImplementationSchema = import_zod8.z.object({
 var BaseParamsSchema = import_zod8.z.object({
   _meta: import_zod8.z.optional(import_zod8.z.object({}).passthrough())
 }).passthrough();
-var RequestSchema = import_zod8.z.object({
-  method: import_zod8.z.string(),
-  params: import_zod8.z.optional(BaseParamsSchema)
-});
 var ResultSchema = BaseParamsSchema;
-var NotificationSchema = import_zod8.z.object({
+var RequestSchema = import_zod8.z.object({
   method: import_zod8.z.string(),
   params: import_zod8.z.optional(BaseParamsSchema)
 });
-var RequestIdSchema = import_zod8.z.union([import_zod8.z.string(), import_zod8.z.number().int()]);
-var JSONRPCRequestSchema = import_zod8.z.object({
-  jsonrpc: import_zod8.z.literal(JSONRPC_VERSION),
-  id: RequestIdSchema
-}).merge(RequestSchema).strict();
-var JSONRPCResponseSchema = import_zod8.z.object({
-  jsonrpc: import_zod8.z.literal(JSONRPC_VERSION),
-  id: RequestIdSchema,
-  result: ResultSchema
-}).strict();
-var JSONRPCErrorSchema = import_zod8.z.object({
-  jsonrpc: import_zod8.z.literal(JSONRPC_VERSION),
-  id: RequestIdSchema,
-  error: import_zod8.z.object({
-    code: import_zod8.z.number().int(),
-    message: import_zod8.z.string(),
-    data: import_zod8.z.optional(import_zod8.z.unknown())
-  })
-}).strict();
-var JSONRPCNotificationSchema = import_zod8.z.object({
-  jsonrpc: import_zod8.z.literal(JSONRPC_VERSION)
-}).merge(NotificationSchema).strict();
-var JSONRPCMessageSchema = import_zod8.z.union([
-  JSONRPCRequestSchema,
-  JSONRPCNotificationSchema,
-  JSONRPCResponseSchema,
-  JSONRPCErrorSchema
-]);
 var ServerCapabilitiesSchema = import_zod8.z.object({
   experimental: import_zod8.z.optional(import_zod8.z.object({}).passthrough()),
   logging: import_zod8.z.optional(import_zod8.z.object({}).passthrough()),
@@ -6879,127 +6778,43 @@ var CallToolResultSchema = ResultSchema.extend({
   })
 );
 
-// core/tool/mcp/mcp-stdio-transport.ts
-var StdioClientTransport = class {
-  constructor(server) {
-    this.abortController = new AbortController();
-    this.readBuffer = new ReadBuffer();
-    this.serverParams = server;
-  }
-  async start() {
-    if (this.process) {
-      throw new MCPClientError({
-        message: "StdioClientTransport already started."
-      });
-    }
-    return new Promise(async (resolve, reject) => {
-      var _a17, _b, _c, _d;
-      try {
-        const process2 = await createChildProcess(
-          this.serverParams,
-          this.abortController.signal
-        );
-        this.process = process2;
-        this.process.on("error", (error) => {
-          var _a18, _b2;
-          if (error.name === "AbortError") {
-            (_a18 = this.onClose) == null ? void 0 : _a18.call(this);
-            return;
-          }
-          reject(error);
-          (_b2 = this.onError) == null ? void 0 : _b2.call(this, error);
-        });
-        this.process.on("spawn", () => {
-          resolve();
-        });
-        this.process.on("close", (_code) => {
-          var _a18;
-          this.process = void 0;
-          (_a18 = this.onClose) == null ? void 0 : _a18.call(this);
-        });
-        (_a17 = this.process.stdin) == null ? void 0 : _a17.on("error", (error) => {
-          var _a18;
-          (_a18 = this.onError) == null ? void 0 : _a18.call(this, error);
-        });
-        (_b = this.process.stdout) == null ? void 0 : _b.on("data", (chunk) => {
-          this.readBuffer.append(chunk);
-          this.processReadBuffer();
-        });
-        (_c = this.process.stdout) == null ? void 0 : _c.on("error", (error) => {
-          var _a18;
-          (_a18 = this.onError) == null ? void 0 : _a18.call(this, error);
-        });
-      } catch (error) {
-        reject(error);
-        (_d = this.onError) == null ? void 0 : _d.call(this, error);
-      }
-    });
-  }
-  processReadBuffer() {
-    var _a17, _b;
-    while (true) {
-      try {
-        const message = this.readBuffer.readMessage();
-        if (message === null) {
-          break;
-        }
-        (_a17 = this.onMessage) == null ? void 0 : _a17.call(this, message);
-      } catch (error) {
-        (_b = this.onError) == null ? void 0 : _b.call(this, error);
-      }
-    }
-  }
-  async close() {
-    this.abortController.abort();
-    this.process = void 0;
-    this.readBuffer.clear();
-  }
-  send(message) {
-    return new Promise((resolve) => {
-      var _a17;
-      if (!((_a17 = this.process) == null ? void 0 : _a17.stdin)) {
-        throw new MCPClientError({
-          message: "StdioClientTransport not connected"
-        });
-      }
-      const json = serializeMessage(message);
-      if (this.process.stdin.write(json)) {
-        resolve();
-      } else {
-        this.process.stdin.once("drain", resolve);
-      }
-    });
-  }
-};
-var ReadBuffer = class {
-  append(chunk) {
-    this.buffer = this.buffer ? Buffer.concat([this.buffer, chunk]) : chunk;
-  }
-  readMessage() {
-    if (!this.buffer)
-      return null;
-    const index = this.buffer.indexOf("\n");
-    if (index === -1) {
-      return null;
-    }
-    const line = this.buffer.toString("utf8", 0, index);
-    this.buffer = this.buffer.subarray(index + 1);
-    return deserializeMessage(line);
-  }
-  clear() {
-    this.buffer = void 0;
-  }
-};
-function serializeMessage(message) {
-  return JSON.stringify(message) + "\n";
-}
-function deserializeMessage(line) {
-  return JSONRPCMessageSchema.parse(JSON.parse(line));
-}
+// core/tool/mcp/json-rpc-message.ts
+var JSONRPC_VERSION = "2.0";
+var JSONRPCRequestSchema = import_zod9.z.object({
+  jsonrpc: import_zod9.z.literal(JSONRPC_VERSION),
+  id: import_zod9.z.union([import_zod9.z.string(), import_zod9.z.number().int()])
+}).merge(RequestSchema).strict();
+var JSONRPCResponseSchema = import_zod9.z.object({
+  jsonrpc: import_zod9.z.literal(JSONRPC_VERSION),
+  id: import_zod9.z.union([import_zod9.z.string(), import_zod9.z.number().int()]),
+  result: ResultSchema
+}).strict();
+var JSONRPCErrorSchema = import_zod9.z.object({
+  jsonrpc: import_zod9.z.literal(JSONRPC_VERSION),
+  id: import_zod9.z.union([import_zod9.z.string(), import_zod9.z.number().int()]),
+  error: import_zod9.z.object({
+    code: import_zod9.z.number().int(),
+    message: import_zod9.z.string(),
+    data: import_zod9.z.optional(import_zod9.z.unknown())
+  })
+}).strict();
+var JSONRPCNotificationSchema = import_zod9.z.object({
+  jsonrpc: import_zod9.z.literal(JSONRPC_VERSION)
+}).merge(
+  import_zod9.z.object({
+    method: import_zod9.z.string(),
+    params: import_zod9.z.optional(BaseParamsSchema)
+  })
+).strict();
+var JSONRPCMessageSchema = import_zod9.z.union([
+  JSONRPCRequestSchema,
+  JSONRPCNotificationSchema,
+  JSONRPCResponseSchema,
+  JSONRPCErrorSchema
+]);
 
 // core/tool/mcp/mcp-sse-transport.ts
-var import_stream = require("eventsource-parser/stream");
-var SSEClientTransport = class {
+var SseMCPTransport = class {
   constructor({ url }) {
     this.connected = false;
     this.url = new URL(url);
@@ -7023,7 +6838,7 @@ var SSEClientTransport = class {
             const error = new MCPClientError({
               message: `MCP SSE Transport Error: ${response.status} ${response.statusText}`
             });
-            (_b = this.onError) == null ? void 0 : _b.call(this, error);
+            (_b = this.onerror) == null ? void 0 : _b.call(this, error);
             return reject(error);
           }
           const stream = response.body.pipeThrough(new TextDecoderStream()).pipeThrough(new import_stream.EventSourceParserStream());
@@ -7057,13 +6872,13 @@ var SSEClientTransport = class {
                     const message = JSONRPCMessageSchema.parse(
                       JSON.parse(data)
                     );
-                    (_a18 = this.onMessage) == null ? void 0 : _a18.call(this, message);
+                    (_a18 = this.onmessage) == null ? void 0 : _a18.call(this, message);
                   } catch (error) {
                     const e = new MCPClientError({
                       message: "MCP SSE Transport Error: Failed to parse message",
                       cause: error
                     });
-                    (_b2 = this.onError) == null ? void 0 : _b2.call(this, e);
+                    (_b2 = this.onerror) == null ? void 0 : _b2.call(this, e);
                   }
                 }
               }
@@ -7071,7 +6886,7 @@ var SSEClientTransport = class {
               if (error instanceof Error && error.name === "AbortError") {
                 return;
               }
-              (_c2 = this.onError) == null ? void 0 : _c2.call(this, error);
+              (_c2 = this.onerror) == null ? void 0 : _c2.call(this, error);
               reject(error);
             }
           };
@@ -7083,7 +6898,7 @@ var SSEClientTransport = class {
           if (error instanceof Error && error.name === "AbortError") {
             return;
           }
-          (_c = this.onError) == null ? void 0 : _c.call(this, error);
+          (_c = this.onerror) == null ? void 0 : _c.call(this, error);
           reject(error);
         }
       };
@@ -7095,7 +6910,7 @@ var SSEClientTransport = class {
     this.connected = false;
     (_a17 = this.sseConnection) == null ? void 0 : _a17.close();
     (_b = this.abortController) == null ? void 0 : _b.abort();
-    (_c = this.onClose) == null ? void 0 : _c.call(this);
+    (_c = this.onclose) == null ? void 0 : _c.call(this);
   }
   async send(message) {
     var _a17, _b, _c;
@@ -7119,11 +6934,11 @@ var SSEClientTransport = class {
         const error = new MCPClientError({
           message: `MCP SSE Transport Error: POSTing to endpoint (HTTP ${response.status}): ${text2}`
         });
-        (_b = this.onError) == null ? void 0 : _b.call(this, error);
+        (_b = this.onerror) == null ? void 0 : _b.call(this, error);
         return;
       }
     } catch (error) {
-      (_c = this.onError) == null ? void 0 : _c.call(this, error);
+      (_c = this.onerror) == null ? void 0 : _c.call(this, error);
       return;
     }
   }
@@ -7131,7 +6946,15 @@ var SSEClientTransport = class {
 
 // core/tool/mcp/mcp-transport.ts
 function createMcpTransport(config) {
-  return config.type === "stdio" ? new StdioClientTransport(config) : new SSEClientTransport(config);
+  if (config.type !== "sse") {
+    throw new MCPClientError({
+      message: "Unsupported or invalid transport configuration. If you are using a custom transport, make sure it implements the MCPTransport interface."
+    });
+  }
+  return new SseMCPTransport(config);
+}
+function isCustomMcpTransport(transport) {
+  return "start" in transport && typeof transport.start === "function" && "send" in transport && typeof transport.send === "function" && "close" in transport && typeof transport.close === "function";
 }
 
 // core/tool/mcp/mcp-client.ts
@@ -7152,10 +6975,14 @@ var MCPClient = class {
     this.serverCapabilities = {};
     this.isClosed = true;
     this.onUncaughtError = onUncaughtError;
-    this.transport = createMcpTransport(transportConfig);
-    this.transport.onClose = () => this.onClose();
-    this.transport.onError = (error) => this.onError(error);
-    this.transport.onMessage = (message) => {
+    if (isCustomMcpTransport(transportConfig)) {
+      this.transport = transportConfig;
+    } else {
+      this.transport = createMcpTransport(transportConfig);
+    }
+    this.transport.onclose = () => this.onClose();
+    this.transport.onerror = (error) => this.onError(error);
+    this.transport.onmessage = (message) => {
       if ("method" in message) {
         this.onError(
           new MCPClientError({