mcp-use 1.2.1-canary.0 → 1.2.2-canary.0

package/dist/{chunk-MGUO7HXB.js → chunk-UVUM35MV.js}

@@ -1,6 +1,6 @@
 import {
   logger
-} from "./chunk-ZUEQQ6YK.js";
+} from "./chunk-VPPILX7B.js";
 import {
   __name
 } from "./chunk-SHUYVCID.js";
@@ -510,9 +510,9 @@ var ObservabilityManager = class {
       return;
     }
     try {
-      const { langfuseHandler: langfuseHandler2, langfuseInitPromise: langfuseInitPromise2 } = await import("./langfuse-6AJGHMAV.js");
+      const { langfuseHandler: langfuseHandler2, langfuseInitPromise: langfuseInitPromise2 } = await import("./langfuse-MO3AMDBE.js");
       if (this.agentId || this.metadata || this.metadataProvider || this.tagsProvider) {
-        const { initializeLangfuse } = await import("./langfuse-6AJGHMAV.js");
+        const { initializeLangfuse } = await import("./langfuse-MO3AMDBE.js");
         await initializeLangfuse(this.agentId, this.metadata, this.metadataProvider, this.tagsProvider);
         logger.debug(`ObservabilityManager: Reinitialized Langfuse with agent ID: ${this.agentId}, metadata: ${JSON.stringify(this.metadata)}`);
       } else {
@@ -2403,841 +2403,6 @@ ${formatPrompt}`);
   }
 };
 
-// src/session.ts
-var MCPSession = class {
-  static {
-    __name(this, "MCPSession");
-  }
-  connector;
-  autoConnect;
-  constructor(connector, autoConnect = true) {
-    this.connector = connector;
-    this.autoConnect = autoConnect;
-  }
-  async connect() {
-    await this.connector.connect();
-  }
-  async disconnect() {
-    await this.connector.disconnect();
-  }
-  async initialize() {
-    if (!this.isConnected && this.autoConnect) {
-      await this.connect();
-    }
-    await this.connector.initialize();
-  }
-  get isConnected() {
-    return this.connector && this.connector.isClientConnected;
-  }
-};
-
-// src/connectors/base.ts
-var BaseConnector = class {
-  static {
-    __name(this, "BaseConnector");
-  }
-  client = null;
-  connectionManager = null;
-  toolsCache = null;
-  connected = false;
-  opts;
-  constructor(opts = {}) {
-    this.opts = opts;
-  }
-  /** Disconnect and release resources. */
-  async disconnect() {
-    if (!this.connected) {
-      logger.debug("Not connected to MCP implementation");
-      return;
-    }
-    logger.debug("Disconnecting from MCP implementation");
-    await this.cleanupResources();
-    this.connected = false;
-    logger.debug("Disconnected from MCP implementation");
-  }
-  /** Check if the client is connected */
-  get isClientConnected() {
-    return this.client != null;
-  }
-  /**
-   * Initialise the MCP session **after** `connect()` has succeeded.
-   *
-   * In the SDK, `Client.connect(transport)` automatically performs the
-   * protocol‑level `initialize` handshake, so we only need to cache the list of
-   * tools and expose some server info.
-   */
-  async initialize(defaultRequestOptions = this.opts.defaultRequestOptions ?? {}) {
-    if (!this.client) {
-      throw new Error("MCP client is not connected");
-    }
-    logger.debug("Caching server capabilities & tools");
-    const capabilities = this.client.getServerCapabilities();
-    const listToolsRes = await this.client.listTools(void 0, defaultRequestOptions);
-    this.toolsCache = listToolsRes.tools ?? [];
-    logger.debug(`Fetched ${this.toolsCache.length} tools from server`);
-    return capabilities;
-  }
-  /** Lazily expose the cached tools list. */
-  get tools() {
-    if (!this.toolsCache) {
-      throw new Error("MCP client is not initialized; call initialize() first");
-    }
-    return this.toolsCache;
-  }
-  /** Call a tool on the server. */
-  async callTool(name, args, options) {
-    if (!this.client) {
-      throw new Error("MCP client is not connected");
-    }
-    logger.debug(`Calling tool '${name}' with args`, args);
-    const res = await this.client.callTool({ name, arguments: args }, void 0, options);
-    logger.debug(`Tool '${name}' returned`, res);
-    return res;
-  }
-  /**
-   * List resources from the server with optional pagination
-   * 
-   * @param cursor - Optional cursor for pagination
-   * @param options - Request options
-   * @returns Resource list with optional nextCursor for pagination
-   */
-  async listResources(cursor, options) {
-    if (!this.client) {
-      throw new Error("MCP client is not connected");
-    }
-    logger.debug("Listing resources", cursor ? `with cursor: ${cursor}` : "");
-    return await this.client.listResources({ cursor }, options);
-  }
-  /**
-   * List all resources from the server, automatically handling pagination
-   * 
-   * @param options - Request options
-   * @returns Complete list of all resources
-   */
-  async listAllResources(options) {
-    if (!this.client) {
-      throw new Error("MCP client is not connected");
-    }
-    logger.debug("Listing all resources (with auto-pagination)");
-    const allResources = [];
-    let cursor = void 0;
-    do {
-      const result = await this.client.listResources({ cursor }, options);
-      allResources.push(...result.resources || []);
-      cursor = result.nextCursor;
-    } while (cursor);
-    return { resources: allResources };
-  }
-  /**
-   * List resource templates from the server
-   * 
-   * @param options - Request options
-   * @returns List of available resource templates
-   */
-  async listResourceTemplates(options) {
-    if (!this.client) {
-      throw new Error("MCP client is not connected");
-    }
-    logger.debug("Listing resource templates");
-    return await this.client.listResourceTemplates(void 0, options);
-  }
-  /** Read a resource by URI. */
-  async readResource(uri, options) {
-    if (!this.client) {
-      throw new Error("MCP client is not connected");
-    }
-    logger.debug(`Reading resource ${uri}`);
-    const res = await this.client.readResource({ uri }, options);
-    return { content: res.content, mimeType: res.mimeType };
-  }
-  /**
-   * Subscribe to resource updates
-   * 
-   * @param uri - URI of the resource to subscribe to
-   * @param options - Request options
-   */
-  async subscribeToResource(uri, options) {
-    if (!this.client) {
-      throw new Error("MCP client is not connected");
-    }
-    logger.debug(`Subscribing to resource: ${uri}`);
-    return await this.client.subscribeResource({ uri }, options);
-  }
-  /**
-   * Unsubscribe from resource updates
-   * 
-   * @param uri - URI of the resource to unsubscribe from
-   * @param options - Request options
-   */
-  async unsubscribeFromResource(uri, options) {
-    if (!this.client) {
-      throw new Error("MCP client is not connected");
-    }
-    logger.debug(`Unsubscribing from resource: ${uri}`);
-    return await this.client.unsubscribeResource({ uri }, options);
-  }
-  async listPrompts() {
-    if (!this.client) {
-      throw new Error("MCP client is not connected");
-    }
-    logger.debug("Listing prompt");
-    return await this.client.listPrompts();
-  }
-  async getPrompt(name, args) {
-    if (!this.client) {
-      throw new Error("MCP client is not connected");
-    }
-    logger.debug(`Getting prompt ${name}`);
-    return await this.client.getPrompt({ name, arguments: args });
-  }
-  /** Send a raw request through the client. */
-  async request(method, params = null, options) {
-    if (!this.client) {
-      throw new Error("MCP client is not connected");
-    }
-    logger.debug(`Sending raw request '${method}' with params`, params);
-    return await this.client.request({ method, params: params ?? {} }, void 0, options);
-  }
-  /**
-   * Helper to tear down the client & connection manager safely.
-   */
-  async cleanupResources() {
-    const issues = [];
-    if (this.client) {
-      try {
-        if (typeof this.client.close === "function") {
-          await this.client.close();
-        }
-      } catch (e) {
-        const msg = `Error closing client: ${e}`;
-        logger.warn(msg);
-        issues.push(msg);
-      } finally {
-        this.client = null;
-      }
-    }
-    if (this.connectionManager) {
-      try {
-        await this.connectionManager.stop();
-      } catch (e) {
-        const msg = `Error stopping connection manager: ${e}`;
-        logger.warn(msg);
-        issues.push(msg);
-      } finally {
-        this.connectionManager = null;
-      }
-    }
-    this.toolsCache = null;
-    if (issues.length) {
-      logger.warn(`Resource cleanup finished with ${issues.length} issue(s)`);
-    }
-  }
-};
-
-// src/connectors/http.ts
-import { Client } from "@modelcontextprotocol/sdk/client/index.js";
-import { StreamableHTTPError } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
-
-// src/task_managers/sse.ts
-import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
-
-// src/task_managers/base.ts
-var ConnectionManager = class {
-  static {
-    __name(this, "ConnectionManager");
-  }
-  _readyPromise;
-  _readyResolver;
-  _donePromise;
-  _doneResolver;
-  _exception = null;
-  _connection = null;
-  _task = null;
-  _abortController = null;
-  constructor() {
-    this.reset();
-  }
-  /**
-   * Start the connection manager and establish a connection.
-   *
-   * @returns The established connection.
-   * @throws If the connection cannot be established.
-   */
-  async start() {
-    this.reset();
-    logger.debug(`Starting ${this.constructor.name}`);
-    this._task = this.connectionTask();
-    await this._readyPromise;
-    if (this._exception) {
-      throw this._exception;
-    }
-    if (this._connection === null) {
-      throw new Error("Connection was not established");
-    }
-    return this._connection;
-  }
-  /**
-   * Stop the connection manager and close the connection.
-   */
-  async stop() {
-    if (this._task && this._abortController) {
-      logger.debug(`Cancelling ${this.constructor.name} task`);
-      this._abortController.abort();
-      try {
-        await this._task;
-      } catch (e) {
-        if (e instanceof Error && e.name === "AbortError") {
-          logger.debug(`${this.constructor.name} task aborted successfully`);
-        } else {
-          logger.warn(`Error stopping ${this.constructor.name} task: ${e}`);
-        }
-      }
-    }
-    await this._donePromise;
-    logger.debug(`${this.constructor.name} task completed`);
-  }
-  /**
-   * Reset all internal state.
-   */
-  reset() {
-    this._readyPromise = new Promise((res) => this._readyResolver = res);
-    this._donePromise = new Promise((res) => this._doneResolver = res);
-    this._exception = null;
-    this._connection = null;
-    this._task = null;
-    this._abortController = new AbortController();
-  }
-  /**
-   * The background task responsible for establishing and maintaining the
-   * connection until it is cancelled.
-   */
-  async connectionTask() {
-    logger.debug(`Running ${this.constructor.name} task`);
-    try {
-      this._connection = await this.establishConnection();
-      logger.debug(`${this.constructor.name} connected successfully`);
-      this._readyResolver();
-      await this.waitForAbort();
-    } catch (err) {
-      this._exception = err;
-      logger.error(`Error in ${this.constructor.name} task: ${err}`);
-      this._readyResolver();
-    } finally {
-      if (this._connection !== null) {
-        try {
-          await this.closeConnection(this._connection);
-        } catch (closeErr) {
-          logger.warn(`Error closing connection in ${this.constructor.name}: ${closeErr}`);
-        }
-        this._connection = null;
-      }
-      this._doneResolver();
-    }
-  }
-  /**
-   * Helper that returns a promise which resolves when the abort signal fires.
-   */
-  async waitForAbort() {
-    return new Promise((_resolve, _reject) => {
-      if (!this._abortController) {
-        return;
-      }
-      const signal = this._abortController.signal;
-      if (signal.aborted) {
-        _resolve();
-        return;
-      }
-      const onAbort = /* @__PURE__ */ __name(() => {
-        signal.removeEventListener("abort", onAbort);
-        _resolve();
-      }, "onAbort");
-      signal.addEventListener("abort", onAbort);
-    });
-  }
-};
-
-// src/task_managers/sse.ts
-var SseConnectionManager = class extends ConnectionManager {
-  static {
-    __name(this, "SseConnectionManager");
-  }
-  url;
-  opts;
-  _transport = null;
-  /**
-   * Create an SSE connection manager.
-   *
-   * @param url  The SSE endpoint URL.
-   * @param opts Optional transport options (auth, headers, etc.).
-   */
-  constructor(url, opts) {
-    super();
-    this.url = typeof url === "string" ? new URL(url) : url;
-    this.opts = opts;
-  }
-  /**
-   * Spawn a new `SSEClientTransport` and start the connection.
-   */
-  async establishConnection() {
-    this._transport = new SSEClientTransport(this.url, this.opts);
-    logger.debug(`${this.constructor.name} connected successfully`);
-    return this._transport;
-  }
-  /**
-   * Close the underlying transport and clean up resources.
-   */
-  async closeConnection(_connection) {
-    if (this._transport) {
-      try {
-        await this._transport.close();
-      } catch (e) {
-        logger.warn(`Error closing SSE transport: ${e}`);
-      } finally {
-        this._transport = null;
-      }
-    }
-  }
-};
-
-// src/task_managers/streamable_http.ts
-import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
-var StreamableHttpConnectionManager = class extends ConnectionManager {
-  static {
-    __name(this, "StreamableHttpConnectionManager");
-  }
-  url;
-  opts;
-  _transport = null;
-  /**
-   * Create a Streamable HTTP connection manager.
-   *
-   * @param url  The HTTP endpoint URL.
-   * @param opts Optional transport options (auth, headers, etc.).
-   */
-  constructor(url, opts) {
-    super();
-    this.url = typeof url === "string" ? new URL(url) : url;
-    this.opts = opts;
-  }
-  /**
-   * Spawn a new `StreamableHTTPClientTransport` and return it.
-   * The Client.connect() method will handle starting the transport.
-   */
-  async establishConnection() {
-    this._transport = new StreamableHTTPClientTransport(this.url, this.opts);
-    logger.debug(`${this.constructor.name} created successfully`);
-    return this._transport;
-  }
-  /**
-   * Close the underlying transport and clean up resources.
-   */
-  async closeConnection(_connection) {
-    if (this._transport) {
-      try {
-        await this._transport.close();
-      } catch (e) {
-        logger.warn(`Error closing Streamable HTTP transport: ${e}`);
-      } finally {
-        this._transport = null;
-      }
-    }
-  }
-  /**
-   * Get the session ID from the transport if available.
-   */
-  get sessionId() {
-    return this._transport?.sessionId;
-  }
-};
-
-// src/connectors/http.ts
-var HttpConnector = class extends BaseConnector {
-  static {
-    __name(this, "HttpConnector");
-  }
-  baseUrl;
-  headers;
-  timeout;
-  sseReadTimeout;
-  clientInfo;
-  preferSse;
-  transportType = null;
-  constructor(baseUrl, opts = {}) {
-    super(opts);
-    this.baseUrl = baseUrl.replace(/\/$/, "");
-    this.headers = { ...opts.headers ?? {} };
-    if (opts.authToken) {
-      this.headers.Authorization = `Bearer ${opts.authToken}`;
-    }
-    this.timeout = opts.timeout ?? 3e4;
-    this.sseReadTimeout = opts.sseReadTimeout ?? 3e5;
-    this.clientInfo = opts.clientInfo ?? { name: "http-connector", version: "1.0.0" };
-    this.preferSse = opts.preferSse ?? false;
-  }
-  /** Establish connection to the MCP implementation via HTTP (streamable or SSE). */
-  async connect() {
-    if (this.connected) {
-      logger.debug("Already connected to MCP implementation");
-      return;
-    }
-    const baseUrl = this.baseUrl;
-    if (this.preferSse) {
-      logger.debug(`Connecting to MCP implementation via HTTP/SSE: ${baseUrl}`);
-      await this.connectWithSse(baseUrl);
-      return;
-    }
-    logger.debug(`Connecting to MCP implementation via HTTP: ${baseUrl}`);
-    try {
-      logger.info("\u{1F504} Attempting streamable HTTP transport...");
-      await this.connectWithStreamableHttp(baseUrl);
-      logger.info("\u2705 Successfully connected via streamable HTTP");
-    } catch (err) {
-      let fallbackReason = "Unknown error";
-      if (err instanceof StreamableHTTPError) {
-        if (err.code === 400 && err.message.includes("Missing session ID")) {
-          fallbackReason = "Server requires session ID (FastMCP compatibility) - using SSE transport";
-          logger.warn(`\u26A0\uFE0F  ${fallbackReason}`);
-        } else if (err.code === 404 || err.code === 405) {
-          fallbackReason = `Server returned ${err.code} - server likely doesn't support streamable HTTP`;
-          logger.debug(fallbackReason);
-        } else {
-          fallbackReason = `Server returned ${err.code}: ${err.message}`;
-          logger.debug(fallbackReason);
-        }
-      } else if (err instanceof Error) {
-        const errorStr = err.toString();
-        const errorMsg = err.message || "";
-        if (errorStr.includes("Missing session ID") || errorStr.includes("Bad Request: Missing session ID") || errorMsg.includes("FastMCP session ID error")) {
-          fallbackReason = "Server requires session ID (FastMCP compatibility) - using SSE transport";
-          logger.warn(`\u26A0\uFE0F  ${fallbackReason}`);
-        } else if (errorStr.includes("405 Method Not Allowed") || errorStr.includes("404 Not Found")) {
-          fallbackReason = "Server doesn't support streamable HTTP (405/404)";
-          logger.debug(fallbackReason);
-        } else {
-          fallbackReason = `Streamable HTTP failed: ${err.message}`;
-          logger.debug(fallbackReason);
-        }
-      }
-      logger.info("\u{1F504} Falling back to SSE transport...");
-      try {
-        await this.connectWithSse(baseUrl);
-      } catch (sseErr) {
-        logger.error(`Failed to connect with both transports:`);
-        logger.error(`  Streamable HTTP: ${fallbackReason}`);
-        logger.error(`  SSE: ${sseErr}`);
-        await this.cleanupResources();
-        throw new Error("Could not connect to server with any available transport");
-      }
-    }
-  }
-  async connectWithStreamableHttp(baseUrl) {
-    try {
-      this.connectionManager = new StreamableHttpConnectionManager(
-        baseUrl,
-        {
-          requestInit: {
-            headers: this.headers
-          },
-          // Pass through timeout and other options
-          reconnectionOptions: {
-            maxReconnectionDelay: 3e4,
-            initialReconnectionDelay: 1e3,
-            reconnectionDelayGrowFactor: 1.5,
-            maxRetries: 2
-          }
-        }
-      );
-      const transport = await this.connectionManager.start();
-      this.client = new Client(this.clientInfo, this.opts.clientOptions);
-      try {
-        await this.client.connect(transport);
-      } catch (connectErr) {
-        if (connectErr instanceof Error) {
-          const errMsg = connectErr.message || connectErr.toString();
-          if (errMsg.includes("Missing session ID") || errMsg.includes("Bad Request: Missing session ID")) {
-            const wrappedError = new Error(`FastMCP session ID error: ${errMsg}`);
-            wrappedError.cause = connectErr;
-            throw wrappedError;
-          }
-        }
-        throw connectErr;
-      }
-      this.connected = true;
-      this.transportType = "streamable-http";
-      logger.debug(`Successfully connected to MCP implementation via streamable HTTP: ${baseUrl}`);
-    } catch (err) {
-      await this.cleanupResources();
-      throw err;
-    }
-  }
-  async connectWithSse(baseUrl) {
-    try {
-      this.connectionManager = new SseConnectionManager(
-        baseUrl,
-        {
-          requestInit: {
-            headers: this.headers
-          }
-        }
-      );
-      const transport = await this.connectionManager.start();
-      this.client = new Client(this.clientInfo, this.opts.clientOptions);
-      await this.client.connect(transport);
-      this.connected = true;
-      this.transportType = "sse";
-      logger.debug(`Successfully connected to MCP implementation via HTTP/SSE: ${baseUrl}`);
-    } catch (err) {
-      await this.cleanupResources();
-      throw err;
-    }
-  }
-  get publicIdentifier() {
-    return {
-      type: "http",
-      url: this.baseUrl,
-      transport: this.transportType || "unknown"
-    };
-  }
-  /**
-   * Get the transport type being used (streamable-http or sse)
-   */
-  getTransportType() {
-    return this.transportType;
-  }
-};
-
-// src/connectors/websocket.ts
-import { v4 as uuidv42 } from "uuid";
-
-// src/task_managers/websocket.ts
-import WS from "ws";
-var WebSocketConnectionManager = class extends ConnectionManager {
-  static {
-    __name(this, "WebSocketConnectionManager");
-  }
-  url;
-  headers;
-  _ws = null;
-  /**
-   * @param url     The WebSocket URL to connect to.
-   * @param headers Optional headers to include in the connection handshake.
-   */
-  constructor(url, headers = {}) {
-    super();
-    this.url = url;
-    this.headers = headers;
-  }
-  /** Establish a WebSocket connection and wait until it is open. */
-  async establishConnection() {
-    logger.debug(`Connecting to WebSocket: ${this.url}`);
-    return new Promise((resolve, reject) => {
-      const ws = new WS(this.url, { headers: this.headers });
-      this._ws = ws;
-      const onOpen = /* @__PURE__ */ __name(() => {
-        cleanup();
-        logger.debug("WebSocket connected successfully");
-        resolve(ws);
-      }, "onOpen");
-      const onError = /* @__PURE__ */ __name((err) => {
-        cleanup();
-        logger.error(`Failed to connect to WebSocket: ${err}`);
-        reject(err);
-      }, "onError");
-      const cleanup = /* @__PURE__ */ __name(() => {
-        ws.off("open", onOpen);
-        ws.off("error", onError);
-      }, "cleanup");
-      ws.on("open", onOpen);
-      ws.on("error", onError);
-    });
-  }
-  /** Cleanly close the WebSocket connection. */
-  async closeConnection(connection) {
-    logger.debug("Closing WebSocket connection");
-    return new Promise((resolve) => {
-      const onClose = /* @__PURE__ */ __name(() => {
-        connection.off("close", onClose);
-        this._ws = null;
-        resolve();
-      }, "onClose");
-      if (connection.readyState === WS.CLOSED) {
-        onClose();
-        return;
-      }
-      connection.on("close", onClose);
-      try {
-        connection.close();
-      } catch (e) {
-        logger.warn(`Error closing WebSocket connection: ${e}`);
-        onClose();
-      }
-    });
-  }
-};
-
-// src/connectors/websocket.ts
-var WebSocketConnector = class extends BaseConnector {
-  static {
-    __name(this, "WebSocketConnector");
-  }
-  url;
-  headers;
-  connectionManager = null;
-  ws = null;
-  receiverTask = null;
-  pending = /* @__PURE__ */ new Map();
-  toolsCache = null;
-  constructor(url, opts = {}) {
-    super();
-    this.url = url;
-    this.headers = { ...opts.headers ?? {} };
-    if (opts.authToken)
-      this.headers.Authorization = `Bearer ${opts.authToken}`;
-  }
-  async connect() {
-    if (this.connected) {
-      logger.debug("Already connected to MCP implementation");
-      return;
-    }
-    logger.debug(`Connecting via WebSocket: ${this.url}`);
-    try {
-      this.connectionManager = new WebSocketConnectionManager(this.url, this.headers);
-      this.ws = await this.connectionManager.start();
-      this.receiverTask = this.receiveLoop();
-      this.connected = true;
-      logger.debug("WebSocket connected successfully");
-    } catch (e) {
-      logger.error(`Failed to connect: ${e}`);
-      await this.cleanupResources();
-      throw e;
-    }
-  }
-  async disconnect() {
-    if (!this.connected) {
-      logger.debug("Not connected to MCP implementation");
-      return;
-    }
-    logger.debug("Disconnecting \u2026");
-    await this.cleanupResources();
-    this.connected = false;
-  }
-  sendRequest(method, params = null) {
-    if (!this.ws)
-      throw new Error("WebSocket is not connected");
-    const id = uuidv42();
-    const payload = JSON.stringify({ id, method, params: params ?? {} });
-    return new Promise((resolve, reject) => {
-      this.pending.set(id, { resolve, reject });
-      this.ws.send(payload, (err) => {
-        if (err) {
-          this.pending.delete(id);
-          reject(err);
-        }
-      });
-    });
-  }
-  async receiveLoop() {
-    if (!this.ws)
-      return;
-    const socket = this.ws;
-    const onMessage = /* @__PURE__ */ __name((msg) => {
-      let data;
-      try {
-        data = JSON.parse(msg.data ?? msg);
-      } catch (e) {
-        logger.warn("Received non\u2011JSON frame", e);
-        return;
-      }
-      const id = data.id;
-      if (id && this.pending.has(id)) {
-        const { resolve, reject } = this.pending.get(id);
-        this.pending.delete(id);
-        if ("result" in data)
-          resolve(data.result);
-        else if ("error" in data)
-          reject(data.error);
-      } else {
-        logger.debug("Received unsolicited message", data);
-      }
-    }, "onMessage");
-    if (socket.addEventListener) {
-      socket.addEventListener("message", onMessage);
-    } else {
-      socket.on("message", onMessage);
-    }
-    return new Promise((resolve) => {
-      const onClose = /* @__PURE__ */ __name(() => {
-        if (socket.removeEventListener) {
-          socket.removeEventListener("message", onMessage);
-        } else {
-          socket.off("message", onMessage);
-        }
-        this.rejectAll(new Error("WebSocket closed"));
-        resolve();
-      }, "onClose");
-      if (socket.addEventListener) {
-        socket.addEventListener("close", onClose);
-      } else {
-        socket.on("close", onClose);
-      }
-    });
-  }
-  rejectAll(err) {
-    for (const { reject } of this.pending.values()) reject(err);
-    this.pending.clear();
-  }
-  async initialize() {
-    logger.debug("Initializing MCP session over WebSocket");
-    const result = await this.sendRequest("initialize");
-    const toolsList = await this.listTools();
-    this.toolsCache = toolsList.map((t) => t);
-    logger.debug(`Initialized with ${this.toolsCache.length} tools`);
-    return result;
-  }
-  async listTools() {
-    const res = await this.sendRequest("tools/list");
-    return res.tools ?? [];
-  }
-  async callTool(name, args) {
-    return await this.sendRequest("tools/call", { name, arguments: args });
-  }
-  async listResources() {
-    const resources = await this.sendRequest("resources/list");
-    return { resources: Array.isArray(resources) ? resources : [] };
-  }
-  async readResource(uri) {
-    const res = await this.sendRequest("resources/read", { uri });
-    return { content: res.content, mimeType: res.mimeType };
-  }
-  async request(method, params = null) {
-    return await this.sendRequest(method, params);
-  }
-  get tools() {
-    if (!this.toolsCache)
-      throw new Error("MCP client is not initialized");
-    return this.toolsCache;
-  }
-  async cleanupResources() {
-    if (this.receiverTask)
-      await this.receiverTask.catch(() => {
-      });
-    this.receiverTask = null;
-    this.rejectAll(new Error("WebSocket disconnected"));
-    if (this.connectionManager) {
-      await this.connectionManager.stop();
-      this.connectionManager = null;
-      this.ws = null;
-    }
-    this.toolsCache = null;
-  }
-  get publicIdentifier() {
-    return {
-      type: "websocket",
-      url: this.url
-    };
-  }
-};
-
 // src/agents/utils/ai_sdk.ts
 async function* streamEventsToAISDK(streamEvents) {
   for await (const event of streamEvents) {
@@ -3293,119 +2458,6 @@ async function* streamEventsToAISDKWithTools(streamEvents) {
 }
 __name(streamEventsToAISDKWithTools, "streamEventsToAISDKWithTools");
 
-// src/client/base.ts
-var BaseMCPClient = class {
-  static {
-    __name(this, "BaseMCPClient");
-  }
-  config = {};
-  sessions = {};
-  activeSessions = [];
-  constructor(config) {
-    if (config) {
-      this.config = config;
-    }
-  }
-  static fromDict(_cfg) {
-    throw new Error("fromDict must be implemented by concrete class");
-  }
-  addServer(name, serverConfig) {
-    this.config.mcpServers = this.config.mcpServers || {};
-    this.config.mcpServers[name] = serverConfig;
-  }
-  removeServer(name) {
-    if (this.config.mcpServers?.[name]) {
-      delete this.config.mcpServers[name];
-      this.activeSessions = this.activeSessions.filter((n) => n !== name);
-    }
-  }
-  getServerNames() {
-    return Object.keys(this.config.mcpServers ?? {});
-  }
-  getServerConfig(name) {
-    return this.config.mcpServers?.[name];
-  }
-  getConfig() {
-    return this.config ?? {};
-  }
-  async createSession(serverName, autoInitialize = true) {
-    const servers = this.config.mcpServers ?? {};
-    if (Object.keys(servers).length === 0) {
-      logger.warn("No MCP servers defined in config");
-    }
-    if (!servers[serverName]) {
-      throw new Error(`Server '${serverName}' not found in config`);
-    }
-    const connector = this.createConnectorFromConfig(servers[serverName]);
-    const session = new MCPSession(connector);
-    if (autoInitialize) {
-      await session.initialize();
-    }
-    this.sessions[serverName] = session;
-    if (!this.activeSessions.includes(serverName)) {
-      this.activeSessions.push(serverName);
-    }
-    return session;
-  }
-  async createAllSessions(autoInitialize = true) {
-    const servers = this.config.mcpServers ?? {};
-    if (Object.keys(servers).length === 0) {
-      logger.warn("No MCP servers defined in config");
-    }
-    for (const name of Object.keys(servers)) {
-      await this.createSession(name, autoInitialize);
-    }
-    return this.sessions;
-  }
-  getSession(serverName) {
-    const session = this.sessions[serverName];
-    if (!session) {
-      return null;
-    }
-    return session;
-  }
-  getAllActiveSessions() {
-    return Object.fromEntries(
-      this.activeSessions.map((n) => [n, this.sessions[n]])
-    );
-  }
-  async closeSession(serverName) {
-    const session = this.sessions[serverName];
-    if (!session) {
-      logger.warn(`No session exists for server ${serverName}, nothing to close`);
-      return;
-    }
-    try {
-      logger.debug(`Closing session for server ${serverName}`);
-      await session.disconnect();
-    } catch (e) {
-      logger.error(`Error closing session for server '${serverName}': ${e}`);
-    } finally {
-      delete this.sessions[serverName];
-      this.activeSessions = this.activeSessions.filter((n) => n !== serverName);
-    }
-  }
-  async closeAllSessions() {
-    const serverNames = Object.keys(this.sessions);
-    const errors = [];
-    for (const serverName of serverNames) {
-      try {
-        logger.debug(`Closing session for server ${serverName}`);
-        await this.closeSession(serverName);
-      } catch (e) {
-        const errorMsg = `Failed to close session for server '${serverName}': ${e}`;
-        logger.error(errorMsg);
-        errors.push(errorMsg);
-      }
-    }
-    if (errors.length) {
-      logger.error(`Encountered ${errors.length} errors while closing sessions`);
-    } else {
-      logger.debug("All sessions closed successfully");
-    }
-  }
-};
-
 export {
   BaseAdapter,
   LangChainAdapter,
@@ -3420,12 +2472,6 @@ export {
   setTelemetrySource,
   RemoteAgent,
   MCPAgent,
-  MCPSession,
-  BaseMCPClient,
-  ConnectionManager,
-  BaseConnector,
-  HttpConnector,
-  WebSocketConnector,
   streamEventsToAISDK,
   createReadableStreamFromGenerator,
   streamEventsToAISDKWithTools