mcp-use 0.1.0 → 0.1.1

package/dist/examples/add_server_tool.js

@@ -9,6 +9,7 @@ import { config } from 'dotenv';
 import { MCPAgent, MCPClient } from '../index.js';
 import { LangChainAdapter } from '../src/adapters/langchain_adapter.js';
 import { ServerManager } from '../src/managers/server_manager.js';
+import { AddMCPServerFromConfigTool } from '../src/managers/tools/add_server_from_config.js';
 // Load environment variables from .env file
 config();
 async function main() {
@@ -16,19 +17,21 @@ async function main() {
     const client = new MCPClient();
     // The LLM to power the agent
     const llm = new ChatOpenAI({ model: 'gpt-4o', temperature: 0 });
+    const serverManager = new ServerManager(client, new LangChainAdapter());
+    serverManager.setManagementTools([new AddMCPServerFromConfigTool(serverManager)]);
     // Create the agent, enabling the ServerManager
     const agent = new MCPAgent({
         llm,
         client,
         maxSteps: 30,
-        useServerManager: true,
-        serverManagerFactory: client => new ServerManager(client, new LangChainAdapter()),
         autoInitialize: true,
+        useServerManager: true,
+        serverManagerFactory: () => serverManager,
     });
     // Define the server configuration that the agent will be asked to add.
     const serverConfigA = {
         command: 'npx',
-        args: ['@playwright/mcp@latest'],
+        args: ['@playwright/mcp@latest', '--headless'],
         env: {
             DISPLAY: ':1',
         },
@@ -54,8 +57,19 @@ async function main() {
     \`\`\`
     and give me a house in the location of the company mcp-use.
     `;
-    // Run the agent. It will first use the AddMCPServerTool, then the tools from the new server.
-    const result = await agent.run(query);
+    // Run the agent. We call `stream()` to get the async generator.
+    const stepIterator = agent.stream(query);
+    let result;
+    while (true) {
+        const { done, value } = await stepIterator.next();
+        if (done) {
+            result = value;
+            break;
+        }
+        // You can inspect the intermediate steps here.
+        console.log('--- Agent Step ---');
+        console.dir(value, { depth: 4 });
+    }
     console.log(`\n✅ Final Result:\n${result}`);
     // Clean up the session created by the agent
     await client.closeAllSessions();

package/dist/index.d.ts

@@ -9,5 +9,6 @@ import { Logger, logger } from './src/logging.js';
 import { MCPSession } from './src/session.js';
 export { BaseAdapter, LangChainAdapter } from './src/adapters/index.js';
 export { ServerManager } from './src/managers/server_manager.js';
+export * from './src/managers/tools/index.js';
 export { BaseConnector, HttpConnector, loadConfigFile, Logger, logger, MCPAgent, MCPClient, MCPSession, StdioConnector, WebSocketConnector };
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAA;AACpD,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAA;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAA;AAChD,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAA;AACxD,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAA;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAA;AAC1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,+BAA+B,CAAA;AAElE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAA;AACjD,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAA;AAE7C,OAAO,EAAE,WAAW,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAA;AACvE,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAA;AAEhE,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,2BAA2B,CAAA;AACpD,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAA;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAA;AAChD,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAA;AACxD,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAA;AACxD,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAA;AAC1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,+BAA+B,CAAA;AAElE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAA;AACjD,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAA;AAE7C,OAAO,EAAE,WAAW,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAA;AACvE,OAAO,EAAE,aAAa,EAAE,MAAM,kCAAkC,CAAA;AAChE,cAAc,+BAA+B,CAAA;AAE7C,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAA"}

package/dist/index.js

@@ -9,4 +9,5 @@ import { Logger, logger } from './src/logging.js';
 import { MCPSession } from './src/session.js';
 export { BaseAdapter, LangChainAdapter } from './src/adapters/index.js';
 export { ServerManager } from './src/managers/server_manager.js';
+export * from './src/managers/tools/index.js';
 export { BaseConnector, HttpConnector, loadConfigFile, Logger, logger, MCPAgent, MCPClient, MCPSession, StdioConnector, WebSocketConnector };

package/dist/src/agents/mcp_agent.d.ts

@@ -1,10 +1,12 @@
 import type { BaseLanguageModelInterface } from '@langchain/core/language_models/base';
 import type { BaseMessage } from '@langchain/core/messages';
+import type { StructuredToolInterface } from '@langchain/core/tools';
+import type { AgentStep } from 'langchain/agents';
 import type { MCPClient } from '../client.js';
 import type { BaseConnector } from '../connectors/base.js';
-import type { ServerManager } from '../managers/server_manager.js';
 import { SystemMessage } from '@langchain/core/messages';
 import { LangChainAdapter } from '../adapters/langchain_adapter.js';
+import { ServerManager } from '../managers/server_manager.js';
 export declare class MCPAgent {
     private llm;
     private client?;
@@ -13,6 +15,7 @@ export declare class MCPAgent {
     private autoInitialize;
     private memoryEnabled;
     private disallowedTools;
+    private additionalTools;
     private useServerManager;
     private verbose;
     private systemPrompt?;
@@ -37,6 +40,7 @@ export declare class MCPAgent {
         systemPromptTemplate?: string | null;
         additionalInstructions?: string | null;
         disallowedTools?: string[];
+        additionalTools?: StructuredToolInterface[];
         useServerManager?: boolean;
         verbose?: boolean;
         adapter?: LangChainAdapter;
@@ -52,7 +56,15 @@ export declare class MCPAgent {
     setSystemMessage(message: string): void;
     setDisallowedTools(disallowedTools: string[]): void;
     getDisallowedTools(): string[];
+    private _consumeAndReturn;
+    /**
+     * Runs the agent and returns a promise for the final result.
+     */
     run(query: string, maxSteps?: number, manageConnector?: boolean, externalHistory?: BaseMessage[]): Promise<string>;
+    /**
+     * Runs the agent and yields intermediate steps as an async generator.
+     */
+    stream(query: string, maxSteps?: number, manageConnector?: boolean, externalHistory?: BaseMessage[]): AsyncGenerator<AgentStep, string, void>;
     close(): Promise<void>;
 }
 //# sourceMappingURL=mcp_agent.d.ts.map

package/dist/src/agents/mcp_agent.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mcp_agent.d.ts","sourceRoot":"","sources":["../../../src/agents/mcp_agent.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,0BAA0B,EAAqB,MAAM,sCAAsC,CAAA;AACzG,OAAO,KAAK,EACV,WAAW,EACZ,MAAM,0BAA0B,CAAA;AAGjC,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAA;AAC7C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAA;AAC1D,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAA;AAElE,OAAO,EAGL,aAAa,EACd,MAAM,0BAA0B,CAAA;AAUjC,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAA;AAKnE,qBAAa,QAAQ;IACnB,OAAO,CAAC,GAAG,CAA4B;IACvC,OAAO,CAAC,MAAM,CAAC,CAAW;IAC1B,OAAO,CAAC,UAAU,CAAiB;IACnC,OAAO,CAAC,QAAQ,CAAQ;IACxB,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,eAAe,CAAU;IACjC,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,YAAY,CAAC,CAAe;IACpC,OAAO,CAAC,4BAA4B,CAAC,CAAe;IACpD,OAAO,CAAC,sBAAsB,CAAC,CAAe;IAE9C,OAAO,CAAC,WAAW,CAAQ;IAC3B,OAAO,CAAC,mBAAmB,CAAoB;IAC/C,OAAO,CAAC,aAAa,CAA6B;IAClD,OAAO,CAAC,QAAQ,CAAiC;IACjD,OAAO,CAAC,aAAa,CAA6B;IAClD,OAAO,CAAC,KAAK,CAAgC;IAC7C,OAAO,CAAC,OAAO,CAAkB;IACjC,OAAO,CAAC,aAAa,CAA6B;gBAEtC,OAAO,EAAE;QACnB,GAAG,EAAE,0BAA0B,CAAA;QAC/B,MAAM,CAAC,EAAE,SAAS,CAAA;QAClB,UAAU,CAAC,EAAE,aAAa,EAAE,CAAA;QAC5B,QAAQ,CAAC,EAAE,MAAM,CAAA;QACjB,cAAc,CAAC,EAAE,OAAO,CAAA;QACxB,aAAa,CAAC,EAAE,OAAO,CAAA;QACvB,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;QAC5B,oBAAoB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;QACpC,sBAAsB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;QACtC,eAAe,CAAC,EAAE,MAAM,EAAE,CAAA;QAC1B,gBAAgB,CAAC,EAAE,OAAO,CAAA;QAC1B,OAAO,CAAC,EAAE,OAAO,CAAA;QACjB,OAAO,CAAC,EAAE,gBAAgB,CAAA;QAC1B,oBAAoB,CAAC,EAAE,CAAC,MAAM,EAAE,SAAS,KAAK,aAAa,CAAA;KAC5D;IAkCY,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;YA8D1B,4BAA4B;IAuB1C,OAAO,CAAC,WAAW;IAyBZ,sBAAsB,IAAI,WAAW,EAAE;IAIvC,wBAAwB,IAAI,IAAI;IAIvC,OAAO,CAAC,YAAY;IAKb,gBAAgB,IAAI,aAAa,GAAG,IAAI;IAIxC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAavC,kBAAkB,CAAC,eAAe,EAAE,MAAM,EAAE,GAAG,IAAI;IAQnD,kBAAkB,IAAI,MAAM,EAAE;IAIxB,GAAG,CACd,KAAK,EAAE,MAAM,EACb,QAAQ,CAAC,EAAE,MAAM,EACjB,eAAe,UAAO,EACtB,eAAe,CAAC,EAAE,WAAW,EAAE,GAC9B,OAAO,CAAC,MAAM,CAAC;IAyJL,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAyBpC"}
+{"version":3,"file":"mcp_agent.d.ts","sourceRoot":"","sources":["../../../src/agents/mcp_agent.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,0BAA0B,EAAqB,MAAM,sCAAsC,CAAA;AACzG,OAAO,KAAK,EACV,WAAW,EACZ,MAAM,0BAA0B,CAAA;AACjC,OAAO,KAAK,EAAE,uBAAuB,EAAiB,MAAM,uBAAuB,CAAA;AACnF,OAAO,KAAK,EAAe,SAAS,EAAE,MAAM,kBAAkB,CAAA;AAC9D,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAA;AAC7C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAA;AAE1D,OAAO,EAGL,aAAa,EACd,MAAM,0BAA0B,CAAA;AAUjC,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAA;AAEnE,OAAO,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAA;AAI7D,qBAAa,QAAQ;IACnB,OAAO,CAAC,GAAG,CAA4B;IACvC,OAAO,CAAC,MAAM,CAAC,CAAW;IAC1B,OAAO,CAAC,UAAU,CAAiB;IACnC,OAAO,CAAC,QAAQ,CAAQ;IACxB,OAAO,CAAC,cAAc,CAAS;IAC/B,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,eAAe,CAAU;IACjC,OAAO,CAAC,eAAe,CAA2B;IAClD,OAAO,CAAC,gBAAgB,CAAS;IACjC,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,YAAY,CAAC,CAAe;IACpC,OAAO,CAAC,4BAA4B,CAAC,CAAe;IACpD,OAAO,CAAC,sBAAsB,CAAC,CAAe;IAE9C,OAAO,CAAC,WAAW,CAAQ;IAC3B,OAAO,CAAC,mBAAmB,CAAoB;IAC/C,OAAO,CAAC,aAAa,CAA6B;IAClD,OAAO,CAAC,QAAQ,CAAiC;IACjD,OAAO,CAAC,aAAa,CAA6B;IAClD,OAAO,CAAC,KAAK,CAAgC;IAC7C,OAAO,CAAC,OAAO,CAAkB;IACjC,OAAO,CAAC,aAAa,CAA6B;gBAEtC,OAAO,EAAE;QACnB,GAAG,EAAE,0BAA0B,CAAA;QAC/B,MAAM,CAAC,EAAE,SAAS,CAAA;QAClB,UAAU,CAAC,EAAE,aAAa,EAAE,CAAA;QAC5B,QAAQ,CAAC,EAAE,MAAM,CAAA;QACjB,cAAc,CAAC,EAAE,OAAO,CAAA;QACxB,aAAa,CAAC,EAAE,OAAO,CAAA;QACvB,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;QAC5B,oBAAoB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;QACpC,sBAAsB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;QACtC,eAAe,CAAC,EAAE,MAAM,EAAE,CAAA;QAC1B,eAAe,CAAC,EAAE,uBAAuB,EAAE,CAAA;QAC3C,gBAAgB,CAAC,EAAE,OAAO,CAAA;QAC1B,OAAO,CAAC,EAAE,OAAO,CAAA;QACjB,OAAO,CAAC,EAAE,gBAAgB,CAAA;QAC1B,oBAAoB,CAAC,EAAE,CAAC,MAAM,EAAE,SAAS,KAAK,aAAa,CAAA;KAC5D;IAiCY,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;YAiE1B,4BAA4B;IAuB1C,OAAO,CAAC,WAAW;IAyBZ,sBAAsB,IAAI,WAAW,EAAE;IAIvC,wBAAwB,IAAI,IAAI;IAIvC,OAAO,CAAC,YAAY;IAKb,gBAAgB,IAAI,aAAa,GAAG,IAAI;IAIxC,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAavC,kBAAkB,CAAC,eAAe,EAAE,MAAM,EAAE,GAAG,IAAI;IAQnD,kBAAkB,IAAI,MAAM,EAAE;YAIvB,iBAAiB;IAc/B;;OAEG;IACU,GAAG,CACd,KAAK,EAAE,MAAM,EACb,QAAQ,CAAC,EAAE,MAAM,EACjB,eAAe,CAAC,EAAE,OAAO,EACzB,eAAe,CAAC,EAAE,WAAW,EAAE,GAC9B,OAAO,CAAC,MAAM,CAAC;IAUlB;;OAEG;IACW,MAAM,CAClB,KAAK,EAAE,MAAM,EACb,QAAQ,CAAC,EAAE,MAAM,EACjB,eAAe,UAAO,EACtB,eAAe,CAAC,EAAE,WAAW,EAAE,GAC9B,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,CAAC;IA2J7B,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAyBpC"}

package/dist/src/agents/mcp_agent.js

@@ -4,6 +4,7 @@ import { ChatPromptTemplate, MessagesPlaceholder, } from '@langchain/core/prompt
 import { AgentExecutor, createToolCallingAgent, } from 'langchain/agents';
 import { LangChainAdapter } from '../adapters/langchain_adapter.js';
 import { logger } from '../logging.js';
+import { ServerManager } from '../managers/server_manager.js';
 import { createSystemMessage } from './prompts/system_prompt_builder.js';
 import { DEFAULT_SYSTEM_PROMPT_TEMPLATE, SERVER_MANAGER_SYSTEM_PROMPT_TEMPLATE } from './prompts/templates.js';
 export class MCPAgent {
@@ -14,6 +15,7 @@ export class MCPAgent {
     autoInitialize;
     memoryEnabled;
     disallowedTools;
+    additionalTools;
     useServerManager;
     verbose;
     systemPrompt;
@@ -38,6 +40,7 @@ export class MCPAgent {
         this.systemPromptTemplateOverride = options.systemPromptTemplate ?? null;
         this.additionalInstructions = options.additionalInstructions ?? null;
         this.disallowedTools = options.disallowedTools ?? [];
+        this.additionalTools = options.additionalTools ?? [];
         this.useServerManager = options.useServerManager ?? false;
         this.verbose = options.verbose ?? false;
         if (!this.client && this.connectors.length === 0) {
@@ -47,15 +50,13 @@ export class MCPAgent {
             if (!this.client) {
                 throw new Error('\'client\' must be provided when \'useServerManager\' is true.');
             }
-            if (options.serverManagerFactory) {
-                this.serverManager = options.serverManagerFactory(this.client);
-            }
-            else {
-                throw new Error('No serverManagerFactory passed to MCPAgent constructor.');
-            }
+            this.adapter = options.adapter ?? new LangChainAdapter(this.disallowedTools);
+            this.serverManager = options.serverManagerFactory?.(this.client) ?? new ServerManager(this.client, this.adapter);
         }
         // Let consumers swap allowed tools dynamically
-        this.adapter = options.adapter ?? new LangChainAdapter(this.disallowedTools);
+        else {
+            this.adapter = options.adapter ?? new LangChainAdapter(this.disallowedTools);
+        }
     }
     async initialize() {
         logger.info('🚀 Initializing MCP agent and connecting to services...');
@@ -65,6 +66,7 @@ export class MCPAgent {
             // Get server management tools
             const managementTools = this.serverManager.tools;
             this.tools = managementTools;
+            this.tools.push(...this.additionalTools);
             logger.info(`🔧 Server manager mode active with ${managementTools.length} management tools`);
             // Create the system message based on available tools
             await this.createSystemMessageFromTools(this.tools);
@@ -83,6 +85,7 @@ export class MCPAgent {
                 }
                 // Create LangChain tools directly from the client using the adapter
                 this.tools = await LangChainAdapter.createTools(this.client);
+                this.tools.push(...this.additionalTools);
                 logger.info(`🛠️ Created ${this.tools.length} LangChain tools from client`);
             }
             else {
@@ -95,6 +98,7 @@ export class MCPAgent {
                 }
                 // Create LangChain tools using the adapter with connectors
                 this.tools = await this.adapter.createToolsFromConnectors(this.connectors);
+                this.tools.push(...this.additionalTools);
                 logger.info(`🛠️ Created ${this.tools.length} LangChain tools from connectors`);
             }
             // Get all tools for system message generation
@@ -173,7 +177,28 @@ export class MCPAgent {
     getDisallowedTools() {
         return this.disallowedTools;
     }
-    async run(query, maxSteps, manageConnector = true, externalHistory) {
+    async _consumeAndReturn(generator) {
+        // Manually iterate through the generator to consume the steps.
+        // The for-await-of loop is not used because it discards the generator's
+        // final return value. We need to capture that value when `done` is true.
+        while (true) {
+            const { done, value } = await generator.next();
+            if (done) {
+                return value;
+            }
+        }
+    }
+    /**
+     * Runs the agent and returns a promise for the final result.
+     */
+    async run(query, maxSteps, manageConnector, externalHistory) {
+        const generator = this.stream(query, maxSteps, manageConnector, externalHistory);
+        return this._consumeAndReturn(generator);
+    }
+    /**
+     * Runs the agent and yields intermediate steps as an async generator.
+     */
+    async *stream(query, maxSteps, manageConnector = true, externalHistory) {
         let result = '';
         let initializedHere = false;
         try {
@@ -217,6 +242,7 @@ export class MCPAgent {
                     if (changed) {
                         logger.info(`🔄 Tools changed before step ${stepNum + 1}, updating agent. New tools: ${[...currentToolNames].join(', ')}`);
                         this.tools = currentTools;
+                        this.tools.push(...this.additionalTools);
                         await this.createSystemMessageFromTools(this.tools);
                         this.agentExecutor = this.createAgent();
                         this.agentExecutor.maxIterations = steps;
@@ -235,6 +261,7 @@ export class MCPAgent {
                     const stepArray = nextStepOutput;
                     intermediateSteps.push(...stepArray);
                     for (const step of stepArray) {
+                        yield step;
                         const { action, observation } = step;
                         const toolName = action.tool;
                         let toolInputStr = String(action.toolInput);

package/dist/src/client.d.ts

@@ -9,6 +9,8 @@ export declare class MCPClient {
     addServer(name: string, serverConfig: Record<string, any>): void;
     removeServer(name: string): void;
     getServerNames(): string[];
+    getServerConfig(name: string): Record<string, any>;
+    getConfig(): Record<string, any>;
     saveConfig(filepath: string): void;
     createSession(serverName: string, autoInitialize?: boolean): Promise<MCPSession>;
     createAllSessions(autoInitialize?: boolean): Promise<Record<string, MCPSession>>;

package/dist/src/client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/client.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAA;AAEzC,qBAAa,SAAS;IACpB,OAAO,CAAC,MAAM,CAA0B;IACxC,OAAO,CAAC,QAAQ,CAAiC;IAC1C,cAAc,EAAE,MAAM,EAAE,CAAK;gBAExB,MAAM,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;WAWnC,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,SAAS;WAI7C,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,SAAS;IAI9C,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI;IAKhE,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAOhC,cAAc,IAAI,MAAM,EAAE;IAI1B,UAAU,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAQ5B,aAAa,CACxB,UAAU,EAAE,MAAM,EAClB,cAAc,UAAO,GACpB,OAAO,CAAC,UAAU,CAAC;IAwBT,iBAAiB,CAC5B,cAAc,UAAO,GACpB,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAc/B,UAAU,CAAC,UAAU,EAAE,MAAM,GAAG,UAAU,GAAG,IAAI;IAWjD,oBAAoB,IAAI,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC;IAM5C,YAAY,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAmB/C,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;CAqB/C"}
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/client.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAA;AAEzC,qBAAa,SAAS;IACpB,OAAO,CAAC,MAAM,CAA0B;IACxC,OAAO,CAAC,QAAQ,CAAiC;IAC1C,cAAc,EAAE,MAAM,EAAE,CAAK;gBAExB,MAAM,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;WAWnC,QAAQ,CAAC,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,SAAS;WAI7C,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,SAAS;IAI9C,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI;IAKhE,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAOhC,cAAc,IAAI,MAAM,EAAE;IAI1B,eAAe,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;IAIlD,SAAS,IAAI,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;IAIhC,UAAU,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAQ5B,aAAa,CACxB,UAAU,EAAE,MAAM,EAClB,cAAc,UAAO,GACpB,OAAO,CAAC,UAAU,CAAC;IAwBT,iBAAiB,CAC5B,cAAc,UAAO,GACpB,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAc/B,UAAU,CAAC,UAAU,EAAE,MAAM,GAAG,UAAU,GAAG,IAAI;IAWjD,oBAAoB,IAAI,MAAM,CAAC,MAAM,EAAE,UAAU,CAAC;IAM5C,YAAY,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAmB/C,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;CAqB/C"}

package/dist/src/client.js

@@ -36,6 +36,12 @@ export class MCPClient {
     getServerNames() {
         return Object.keys(this.config.mcpServers ?? {});
     }
+    getServerConfig(name) {
+        return this.config.mcpServers?.[name];
+    }
+    getConfig() {
+        return this.config ?? {};
+    }
     saveConfig(filepath) {
         const dir = path.dirname(filepath);
         if (!fs.existsSync(dir)) {

package/dist/src/managers/server_manager.d.ts

@@ -7,7 +7,9 @@ export declare class ServerManager {
     readonly client: MCPClient;
     readonly adapter: LangChainAdapter;
     activeServer: string | null;
-    constructor(client: MCPClient, adapter: LangChainAdapter);
+    private overrideManagementTools?;
+    constructor(client: MCPClient, adapter: LangChainAdapter, managementTools?: StructuredToolInterface[]);
+    setManagementTools(tools: StructuredToolInterface[]): void;
     logState(context: string): void;
     initialize(): void;
     prefetchServerTools(): Promise<void>;

package/dist/src/managers/server_manager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"server_manager.d.ts","sourceRoot":"","sources":["../../../src/managers/server_manager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAA;AACpE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAA;AACxE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAA;AAW7C,qBAAa,aAAa;IACxB,SAAgB,kBAAkB,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAK;IAChE,SAAgB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,uBAAuB,EAAE,CAAC,CAAK;IAE3E,SAAgB,MAAM,EAAE,SAAS,CAAA;IACjC,SAAgB,OAAO,EAAE,gBAAgB,CAAA;IAClC,YAAY,EAAE,MAAM,GAAG,IAAI,CAAO;gBAE7B,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,gBAAgB;IAKjD,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAqBtC,UAAU,IAAI,IAAI;IAOZ,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAoD1C,IAAI,KAAK,IAAI,uBAAuB,EAAE,CAsBrC;CACF"}
+{"version":3,"file":"server_manager.d.ts","sourceRoot":"","sources":["../../../src/managers/server_manager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAA;AACpE,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAA;AACxE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,cAAc,CAAA;AAW7C,qBAAa,aAAa;IACxB,SAAgB,kBAAkB,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAK;IAChE,SAAgB,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,uBAAuB,EAAE,CAAC,CAAK;IAE3E,SAAgB,MAAM,EAAE,SAAS,CAAA;IACjC,SAAgB,OAAO,EAAE,gBAAgB,CAAA;IAClC,YAAY,EAAE,MAAM,GAAG,IAAI,CAAO;IACzC,OAAO,CAAC,uBAAuB,CAAC,CAA2B;gBAGzD,MAAM,EAAE,SAAS,EACjB,OAAO,EAAE,gBAAgB,EACzB,eAAe,CAAC,EAAE,uBAAuB,EAAE;IAOtC,kBAAkB,CAAC,KAAK,EAAE,uBAAuB,EAAE,GAAG,IAAI;IAO1D,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAqBtC,UAAU,IAAI,IAAI;IAOZ,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAoD1C,IAAI,KAAK,IAAI,uBAAuB,EAAE,CAsBrC;CACF"}

package/dist/src/managers/server_manager.js

@@ -1,7 +1,7 @@
 import { isEqual } from 'lodash-es';
 import { logger } from '../logging.js';
 import { AcquireActiveMCPServerTool } from './tools/acquire_active_mcp_server.js';
-import { AddMCPServerTool } from './tools/add_server.js';
+import { AddMCPServerFromConfigTool } from './tools/add_server_from_config.js';
 import { ConnectMCPServerTool } from './tools/connect_mcp_server.js';
 import { ListMCPServersTool } from './tools/list_mcp_servers.js';
 import { ReleaseMCPServerConnectionTool } from './tools/release_mcp_server_connection.js';
@@ -11,9 +11,15 @@ export class ServerManager {
     client;
     adapter;
     activeServer = null;
-    constructor(client, adapter) {
+    overrideManagementTools;
+    constructor(client, adapter, managementTools) {
         this.client = client;
         this.adapter = adapter;
+        this.overrideManagementTools = managementTools;
+    }
+    setManagementTools(tools) {
+        this.overrideManagementTools = tools;
+        logger.info(`Overriding default management tools with a new set of ${tools.length} tools.`);
     }
     logState(context) {
         const allServerNames = this.client.getServerNames();
@@ -83,8 +89,8 @@ export class ServerManager {
         if (logger.level === 'debug') {
             this.logState('Providing tools to agent');
         }
-        const managementTools = [
-            new AddMCPServerTool(this),
+        const managementTools = this.overrideManagementTools ?? [
+            new AddMCPServerFromConfigTool(this),
             new ListMCPServersTool(this),
             new ConnectMCPServerTool(this),
             new AcquireActiveMCPServerTool(this),

package/dist/src/managers/tools/{add_server.d.ts → add_server_from_config.d.ts}

@@ -1,7 +1,7 @@
 import type { ServerManager } from '../server_manager.js';
 import { StructuredTool } from 'langchain/tools';
 import { z } from 'zod';
-export declare class AddMCPServerTool extends StructuredTool {
+export declare class AddMCPServerFromConfigTool extends StructuredTool {
     name: string;
     description: string;
     schema: z.ZodObject<{
@@ -18,4 +18,4 @@ export declare class AddMCPServerTool extends StructuredTool {
     constructor(manager: ServerManager);
     protected _call({ serverName, serverConfig, }: z.infer<typeof this.schema>): Promise<string>;
 }
-//# sourceMappingURL=add_server.d.ts.map
+//# sourceMappingURL=add_server_from_config.d.ts.map

package/dist/src/managers/tools/add_server_from_config.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"add_server_from_config.d.ts","sourceRoot":"","sources":["../../../../src/managers/tools/add_server_from_config.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAA;AACzD,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAA;AAChD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAGvB,qBAAa,0BAA2B,SAAQ,cAAc;IAC5D,IAAI,SAA+B;IACnC,WAAW,SAC0G;IAErH,MAAM;;;;;;;;;OAOJ;IAEF,OAAO,CAAC,OAAO,CAAe;gBAElB,OAAO,EAAE,aAAa;cAKlB,KAAK,CAAC,EACpB,UAAU,EACV,YAAY,GACb,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;CA6BjD"}

package/dist/src/managers/tools/{add_server.js → add_server_from_config.js}

@@ -1,9 +1,9 @@
 import { StructuredTool } from 'langchain/tools';
 import { z } from 'zod';
 import { logger } from '../../logging.js';
-export class AddMCPServerTool extends StructuredTool {
-    name = 'add_mcp_server';
-    description = 'Adds a new MCP server to the client and connects to it, making its tools available.';
+export class AddMCPServerFromConfigTool extends StructuredTool {
+    name = 'add_mcp_server_from_config';
+    description = 'Adds a new MCP server to the client from a configuration object and connects to it, making its tools available.';
     schema = z.object({
         serverName: z.string().describe('The name for the new MCP server.'),
         serverConfig: z

package/dist/src/managers/tools/index.d.ts

@@ -1,4 +1,5 @@
 export * from './acquire_active_mcp_server.js';
+export * from './add_server_from_config.js';
 export * from './connect_mcp_server.js';
 export * from './list_mcp_servers.js';
 export * from './release_mcp_server_connection.js';

package/dist/src/managers/tools/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/managers/tools/index.ts"],"names":[],"mappings":"AAAA,cAAc,gCAAgC,CAAA;AAC9C,cAAc,yBAAyB,CAAA;AACvC,cAAc,uBAAuB,CAAA;AACrC,cAAc,oCAAoC,CAAA"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/managers/tools/index.ts"],"names":[],"mappings":"AAAA,cAAc,gCAAgC,CAAA;AAC9C,cAAc,6BAA6B,CAAA;AAC3C,cAAc,yBAAyB,CAAA;AACvC,cAAc,uBAAuB,CAAA;AACrC,cAAc,oCAAoC,CAAA"}

package/dist/src/managers/tools/index.js

@@ -1,4 +1,5 @@
 export * from './acquire_active_mcp_server.js';
+export * from './add_server_from_config.js';
 export * from './connect_mcp_server.js';
 export * from './list_mcp_servers.js';
 export * from './release_mcp_server_connection.js';

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "mcp-use",
   "type": "module",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "packageManager": "pnpm@10.6.1",
   "description": "A utility library for integrating Model Context Protocol (MCP) with LangChain, Zod, and related tools. Provides helpers for schema conversion, event streaming, and SDK usage.",
   "author": "Zane",

package/dist/src/managers/tools/add_server.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"add_server.d.ts","sourceRoot":"","sources":["../../../../src/managers/tools/add_server.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAA;AACzD,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAA;AAChD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAGvB,qBAAa,gBAAiB,SAAQ,cAAc;IAClD,IAAI,SAAmB;IACvB,WAAW,SAC8E;IAEzF,MAAM;;;;;;;;;OAOJ;IAEF,OAAO,CAAC,OAAO,CAAe;gBAElB,OAAO,EAAE,aAAa;cAKlB,KAAK,CAAC,EACpB,UAAU,EACV,YAAY,GACb,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;CA6BjD"}