mcp-use 1.3.4-canary.2 → 1.4.0

package/dist/{chunk-6EHM3S3M.js → chunk-35A6O3YH.js}

@@ -1578,18 +1578,35 @@ var MCPAgent = class {
         logger.info(
           `\u{1F50C} Found ${Object.keys(this.sessions).length} existing sessions`
         );
-        if (Object.keys(this.sessions).length === 0) {
+        const nonCodeModeSessions = Object.keys(this.sessions).filter(
+          (name) => name !== "code_mode"
+        );
+        if (nonCodeModeSessions.length === 0) {
           logger.info("\u{1F504} No active sessions found, creating new ones...");
           this.sessions = await this.client.createAllSessions();
           logger.info(
             `\u2705 Created ${Object.keys(this.sessions).length} new sessions`
           );
         }
-        this._tools = await LangChainAdapter.createTools(this.client);
+        if (this.client.codeMode) {
+          const codeModeSession = this.sessions["code_mode"];
+          if (codeModeSession) {
+            this._tools = await this.adapter.createToolsFromConnectors([
+              codeModeSession.connector
+            ]);
+            logger.info(`\u{1F6E0}\uFE0F Created ${this._tools.length} code mode tools`);
+          } else {
+            throw new Error(
+              "Code mode enabled but code_mode session not found"
+            );
+          }
+        } else {
+          this._tools = await LangChainAdapter.createTools(this.client);
+          logger.info(
+            `\u{1F6E0}\uFE0F Created ${this._tools.length} LangChain tools from client`
+          );
+        }
         this._tools.push(...this.additionalTools);
-        logger.info(
-          `\u{1F6E0}\uFE0F Created ${this._tools.length} LangChain tools from client`
-        );
       } else {
         logger.info(
           `\u{1F517} Connecting to ${this.connectors.length} direct connectors...`
@@ -2154,6 +2171,8 @@ var MCPAgent = class {
           tags: this.getTags(),
           // Set trace name for LangChain/Langfuse
           runName: this.metadata.trace_name || "mcp-use-agent",
+          // Set recursion limit to 3x maxSteps to account for model calls + tool executions
+          recursionLimit: this.maxSteps * 3,
           // Pass sessionId for Langfuse if present in metadata
           ...this.metadata.session_id && {
             sessionId: this.metadata.session_id
@@ -2367,8 +2386,8 @@ var MCPAgent = class {
       this._agentExecutor = null;
       this._tools = [];
       if (this.client) {
-        logger.info("\u{1F504} Closing sessions through client");
-        await this.client.closeAllSessions();
+        logger.info("\u{1F504} Closing client and cleaning up resources");
+        await this.client.close();
         this.sessions = {};
       } else {
         for (const connector of this.connectors) {
@@ -2384,6 +2403,26 @@ var MCPAgent = class {
       logger.info("\u{1F44B} Agent closed successfully");
     }
   }
+  /**
+   * Yields with pretty-printed output for code mode.
+   * This method formats and displays tool executions in a user-friendly way with syntax highlighting.
+   */
+  async *prettyStreamEvents(query, maxSteps, manageConnector = true, externalHistory, outputSchema) {
+    const { prettyStreamEvents: prettyStream } = await import("./display-LIYVTGEU.js");
+    const finalResponse = "";
+    for await (const _ of prettyStream(
+      this.streamEvents(
+        query,
+        maxSteps,
+        manageConnector,
+        externalHistory,
+        outputSchema
+      )
+    )) {
+      yield;
+    }
+    return finalResponse;
+  }
   /**
    * Yields LangChain StreamEvent objects from the underlying streamEvents() method.
    * This provides token-level streaming and fine-grained event updates.
@@ -2411,11 +2450,11 @@ var MCPAgent = class {
         throw new Error("MCP agent failed to initialize");
       }
       this.maxSteps = maxSteps ?? this.maxSteps;
-      const display_query = query.length > 50 ? `${query.slice(0, 50).replace(/\n/g, " ")}...` : query.replace(/\n/g, " ");
+      const display_query = typeof query === "string" && query.length > 50 ? `${query.slice(0, 50).replace(/\n/g, " ")}...` : typeof query === "string" ? query.replace(/\n/g, " ") : String(query);
       logger.info(`\u{1F4AC} Received query for streamEvents: '${display_query}'`);
       if (this.memoryEnabled) {
-        logger.info(`\u{1F504} Adding user message to history: ${query}`);
-        this.addToHistory(new HumanMessage(query));
+        logger.info(`\u{1F504} Adding user message to history: ${display_query}`);
+        this.addToHistory(new HumanMessage({ content: query }));
       }
       const historyToUse = externalHistory ?? this.conversationHistory;
       const langchainHistory = [];
@@ -2443,6 +2482,8 @@ var MCPAgent = class {
           tags: this.getTags(),
           // Set trace name for LangChain/Langfuse
           runName: this.metadata.trace_name || "mcp-use-agent",
+          // Set recursion limit to 3x maxSteps to account for model calls + tool executions
+          recursionLimit: this.maxSteps * 3,
           // Pass sessionId for Langfuse if present in metadata
           ...this.metadata.session_id && {
             sessionId: this.metadata.session_id
@@ -2542,6 +2583,7 @@ var MCPAgent = class {
       } else if (this.memoryEnabled && finalResponse) {
         this.addToHistory(new AIMessage(finalResponse));
       }
+      console.log("\n\n");
       logger.info(`\u{1F389} StreamEvents complete - ${eventCount} events emitted`);
       success = true;
     } catch (e) {

package/dist/chunk-DSBKVAWD.js

@@ -0,0 +1,263 @@
+import {
+  logger
+} from "./chunk-34R6SIER.js";
+import {
+  __name
+} from "./chunk-3GQAWCBQ.js";
+
+// src/connectors/base.ts
+var BaseConnector = class {
+  static {
+    __name(this, "BaseConnector");
+  }
+  client = null;
+  connectionManager = null;
+  toolsCache = null;
+  capabilitiesCache = null;
+  serverInfoCache = 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();
+    this.capabilitiesCache = capabilities;
+    const serverInfo = this.client.getServerVersion();
+    this.serverInfoCache = serverInfo || null;
+    const listToolsRes = await this.client.listTools(
+      void 0,
+      defaultRequestOptions
+    );
+    this.toolsCache = listToolsRes.tools ?? [];
+    logger.debug(`Fetched ${this.toolsCache.length} tools from server`);
+    logger.debug("Server capabilities:", capabilities);
+    logger.debug("Server info:", serverInfo);
+    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;
+  }
+  /** Expose cached server capabilities. */
+  get serverCapabilities() {
+    return this.capabilitiesCache;
+  }
+  /** Expose cached server info. */
+  get serverInfo() {
+    return this.serverInfoCache;
+  }
+  /** 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");
+    }
+    if (!this.capabilitiesCache?.resources) {
+      logger.debug("Server does not advertise resources capability, skipping");
+      return { resources: [] };
+    }
+    try {
+      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 };
+    } catch (err) {
+      if (err.code === -32601) {
+        logger.debug("Server advertised resources but method not found");
+        return { resources: [] };
+      }
+      throw err;
+    }
+  }
+  /**
+   * 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 res;
+  }
+  /**
+   * 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");
+    }
+    if (!this.capabilitiesCache?.prompts) {
+      logger.debug("Server does not advertise prompts capability, skipping");
+      return { prompts: [] };
+    }
+    try {
+      logger.debug("Listing prompts");
+      return await this.client.listPrompts();
+    } catch (err) {
+      if (err.code === -32601) {
+        logger.debug("Server advertised prompts but method not found");
+        return { prompts: [] };
+      }
+      throw err;
+    }
+  }
+  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)`);
+    }
+  }
+};
+
+export {
+  BaseConnector
+};

package/dist/{chunk-TIUSJAAE.js → chunk-GRLCLVAK.js}

@@ -1,3 +1,6 @@
+import {
+  BaseConnector
+} from "./chunk-DSBKVAWD.js";
 import {
   logger
 } from "./chunk-34R6SIER.js";
@@ -33,259 +36,6 @@ var MCPSession = class {
   }
 };
 
-// src/connectors/base.ts
-var BaseConnector = class {
-  static {
-    __name(this, "BaseConnector");
-  }
-  client = null;
-  connectionManager = null;
-  toolsCache = null;
-  capabilitiesCache = null;
-  serverInfoCache = 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();
-    this.capabilitiesCache = capabilities;
-    const serverInfo = this.client.getServerVersion();
-    this.serverInfoCache = serverInfo || null;
-    const listToolsRes = await this.client.listTools(
-      void 0,
-      defaultRequestOptions
-    );
-    this.toolsCache = listToolsRes.tools ?? [];
-    logger.debug(`Fetched ${this.toolsCache.length} tools from server`);
-    logger.debug("Server capabilities:", capabilities);
-    logger.debug("Server info:", serverInfo);
-    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;
-  }
-  /** Expose cached server capabilities. */
-  get serverCapabilities() {
-    return this.capabilitiesCache;
-  }
-  /** Expose cached server info. */
-  get serverInfo() {
-    return this.serverInfoCache;
-  }
-  /** 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");
-    }
-    if (!this.capabilitiesCache?.resources) {
-      logger.debug("Server does not advertise resources capability, skipping");
-      return { resources: [] };
-    }
-    try {
-      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 };
-    } catch (err) {
-      if (err.code === -32601) {
-        logger.debug("Server advertised resources but method not found");
-        return { resources: [] };
-      }
-      throw err;
-    }
-  }
-  /**
-   * 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 res;
-  }
-  /**
-   * 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");
-    }
-    if (!this.capabilitiesCache?.prompts) {
-      logger.debug("Server does not advertise prompts capability, skipping");
-      return { prompts: [] };
-    }
-    try {
-      logger.debug("Listing prompts");
-      return await this.client.listPrompts();
-    } catch (err) {
-      if (err.code === -32601) {
-        logger.debug("Server advertised prompts but method not found");
-        return { prompts: [] };
-      }
-      throw err;
-    }
-  }
-  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";
@@ -1414,7 +1164,6 @@ export {
   MCPSession,
   BaseMCPClient,
   ConnectionManager,
-  BaseConnector,
   HttpConnector,
   WebSocketConnector,
   BrowserOAuthClientProvider,

package/dist/{chunk-QKPVHYJU.js → chunk-RE7EYFDV.js}

@@ -1,7 +1,7 @@
 import {
   BrowserMCPClient,
   BrowserOAuthClientProvider
-} from "./chunk-TIUSJAAE.js";
+} from "./chunk-GRLCLVAK.js";
 import {
   __name
 } from "./chunk-3GQAWCBQ.js";