mcp-use 0.1.0 → 0.1.2

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,7 @@ 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 { setTelemetrySource, Telemetry } from './src/telemetry/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;AAG7C,OAAO,EAAE,kBAAkB,EAAE,SAAS,EAAE,MAAM,0BAA0B,CAAA;AAExE,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,7 @@ 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 telemetry utilities
+export { setTelemetrySource, Telemetry } from './src/telemetry/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?;
@@ -26,6 +29,9 @@ export declare class MCPAgent {
     private tools;
     private adapter;
     private serverManager;
+    private telemetry;
+    private modelProvider;
+    private modelName;
     constructor(options: {
         llm: BaseLanguageModelInterface;
         client?: MCPClient;
@@ -37,6 +43,7 @@ export declare class MCPAgent {
         systemPromptTemplate?: string | null;
         additionalInstructions?: string | null;
         disallowedTools?: string[];
+        additionalTools?: StructuredToolInterface[];
         useServerManager?: boolean;
         verbose?: boolean;
         adapter?: LangChainAdapter;
@@ -52,7 +59,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;AAK7D,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;IAClD,OAAO,CAAC,SAAS,CAAW;IAC5B,OAAO,CAAC,aAAa,CAAQ;IAC7B,OAAO,CAAC,SAAS,CAAQ;gBAEb,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;IAwCY,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;IAwM7B,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAyBpC"}

package/dist/src/agents/mcp_agent.js

@@ -4,6 +4,8 @@ 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 { extractModelInfo, Telemetry } from '../telemetry/index.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 +16,7 @@ export class MCPAgent {
     autoInitialize;
     memoryEnabled;
     disallowedTools;
+    additionalTools;
     useServerManager;
     verbose;
     systemPrompt;
@@ -27,6 +30,9 @@ export class MCPAgent {
     tools = [];
     adapter;
     serverManager = null;
+    telemetry;
+    modelProvider;
+    modelName;
     constructor(options) {
         this.llm = options.llm;
         this.client = options.client;
@@ -38,6 +44,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 +54,19 @@ 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);
+        }
+        // Initialize telemetry
+        this.telemetry = Telemetry.getInstance();
+        // Track model info for telemetry
+        const [provider, name] = extractModelInfo(this.llm);
+        this.modelProvider = provider;
+        this.modelName = name;
     }
     async initialize() {
         logger.info('🚀 Initializing MCP agent and connecting to services...');
@@ -65,6 +76,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 +95,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 +108,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,9 +187,34 @@ 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;
+        const startTime = Date.now();
+        const toolsUsedNames = [];
+        let stepsTaken = 0;
+        let success = false;
         try {
             if (manageConnector && !this.initialized) {
                 await this.initialize();
@@ -208,6 +247,7 @@ export class MCPAgent {
             let nameToToolMap = Object.fromEntries(this.tools.map(t => [t.name, t]));
             logger.info(`🏁 Starting agent execution with max_steps=${steps}`);
             for (let stepNum = 0; stepNum < steps; stepNum++) {
+                stepsTaken = stepNum + 1;
                 if (this.useServerManager && this.serverManager) {
                     const currentTools = this.serverManager.tools;
                     const currentToolNames = new Set(currentTools.map(t => t.name));
@@ -217,6 +257,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,8 +276,10 @@ export class MCPAgent {
                     const stepArray = nextStepOutput;
                     intermediateSteps.push(...stepArray);
                     for (const step of stepArray) {
+                        yield step;
                         const { action, observation } = step;
                         const toolName = action.tool;
+                        toolsUsedNames.push(toolName);
                         let toolInputStr = String(action.toolInput);
                         if (toolInputStr.length > 100)
                             toolInputStr = `${toolInputStr.slice(0, 97)}...`;
@@ -279,6 +322,7 @@ export class MCPAgent {
                 this.addToHistory(new AIMessage(result));
             }
             logger.info('🎉 Agent execution complete');
+            success = true;
             return result;
         }
         catch (e) {
@@ -290,6 +334,40 @@ export class MCPAgent {
             throw e;
         }
         finally {
+            // Track comprehensive execution data
+            const executionTimeMs = Date.now() - startTime;
+            let serverCount = 0;
+            if (this.client) {
+                serverCount = Object.keys(await this.client.getAllActiveSessions()).length;
+            }
+            else if (this.connectors) {
+                serverCount = this.connectors.length;
+            }
+            const conversationHistoryLength = this.memoryEnabled ? this.conversationHistory.length : 0;
+            await this.telemetry.trackAgentExecution({
+                executionMethod: 'stream',
+                query,
+                success,
+                modelProvider: this.modelProvider,
+                modelName: this.modelName,
+                serverCount,
+                serverIdentifiers: this.connectors.map(connector => connector.publicIdentifier),
+                totalToolsAvailable: this.tools.length,
+                toolsAvailableNames: this.tools.map(t => t.name),
+                maxStepsConfigured: this.maxSteps,
+                memoryEnabled: this.memoryEnabled,
+                useServerManager: this.useServerManager,
+                maxStepsUsed: maxSteps ?? null,
+                manageConnector,
+                externalHistoryUsed: externalHistory !== undefined,
+                stepsTaken,
+                toolsUsedCount: toolsUsedNames.length,
+                toolsUsedNames,
+                response: result,
+                executionTimeMs,
+                errorType: success ? null : 'execution_error',
+                conversationHistoryLength,
+            });
             if (manageConnector && !this.client && initializedHere) {
                 logger.info('🧹 Closing agent after query completion');
                 await this.close();

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/connectors/base.d.ts

@@ -25,6 +25,8 @@ export declare abstract class BaseConnector {
     constructor(opts?: ConnectorInitOptions);
     /** Establish the connection and create the SDK client. */
     abstract connect(): Promise<void>;
+    /** Get the identifier for the connector. */
+    abstract get publicIdentifier(): Record<string, string>;
     /** Disconnect and release resources. */
     disconnect(): Promise<void>;
     /** Check if the client is connected */

package/dist/src/connectors/base.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../src/connectors/base.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,2CAA2C,CAAA;AACtF,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,8CAA8C,CAAA;AAClF,OAAO,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,oCAAoC,CAAA;AAC9E,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAA;AAGjE,MAAM,WAAW,oBAAoB;IACnC;;OAEG;IACH,aAAa,CAAC,EAAE,aAAa,CAAA;IAC7B;;;OAGG;IACH,qBAAqB,CAAC,EAAE,cAAc,CAAA;CACvC;AAED;;GAEG;AACH,8BAAsB,aAAa;IACjC,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAO;IACtC,SAAS,CAAC,iBAAiB,EAAE,iBAAiB,CAAC,GAAG,CAAC,GAAG,IAAI,CAAO;IACjE,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,GAAG,IAAI,CAAO;IAC1C,SAAS,CAAC,SAAS,UAAQ;IAC3B,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,oBAAoB,CAAA;gBAEjC,IAAI,GAAE,oBAAyB;IAI3C,0DAA0D;IAC1D,QAAQ,CAAC,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAEjC,wCAAwC;IAClC,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAYjC,uCAAuC;IACvC,IAAI,iBAAiB,IAAI,OAAO,CAE/B;IAED;;;;;;OAMG;IACG,UAAU,CAAC,qBAAqB,GAAE,cAAsD,GAAG,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,CAAC;IAkBrJ,2CAA2C;IAC3C,IAAI,KAAK,IAAI,IAAI,EAAE,CAKlB;IAED,iCAAiC;IAC3B,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,OAAO,CAAC,EAAE,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;IAW1G,sCAAsC;IAChC,aAAa,CAAC,OAAO,CAAC,EAAE,cAAc;;;;;;;;;;;;;;;;;;;;;;IAS5C,8BAA8B;IACxB,YAAY,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,cAAc;;;;IAUlD,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IASX,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IASvD,6CAA6C;IACvC,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAW,EAAE,OAAO,CAAC,EAAE,cAAc;IASjG;;OAEG;cACa,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;CAsClD"}
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../src/connectors/base.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,2CAA2C,CAAA;AACtF,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,8CAA8C,CAAA;AAClF,OAAO,KAAK,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,oCAAoC,CAAA;AAC9E,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAA;AAGjE,MAAM,WAAW,oBAAoB;IACnC;;OAEG;IACH,aAAa,CAAC,EAAE,aAAa,CAAA;IAC7B;;;OAGG;IACH,qBAAqB,CAAC,EAAE,cAAc,CAAA;CACvC;AAED;;GAEG;AACH,8BAAsB,aAAa;IACjC,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAO;IACtC,SAAS,CAAC,iBAAiB,EAAE,iBAAiB,CAAC,GAAG,CAAC,GAAG,IAAI,CAAO;IACjE,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,GAAG,IAAI,CAAO;IAC1C,SAAS,CAAC,SAAS,UAAQ;IAC3B,SAAS,CAAC,QAAQ,CAAC,IAAI,EAAE,oBAAoB,CAAA;gBAEjC,IAAI,GAAE,oBAAyB;IAI3C,0DAA0D;IAC1D,QAAQ,CAAC,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAEjC,4CAA4C;IAC5C,QAAQ,KAAK,gBAAgB,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAEvD,wCAAwC;IAClC,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAYjC,uCAAuC;IACvC,IAAI,iBAAiB,IAAI,OAAO,CAE/B;IAED;;;;;;OAMG;IACG,UAAU,CAAC,qBAAqB,GAAE,cAAsD,GAAG,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,CAAC;IAkBrJ,2CAA2C;IAC3C,IAAI,KAAK,IAAI,IAAI,EAAE,CAKlB;IAED,iCAAiC;IAC3B,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,OAAO,CAAC,EAAE,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;IAW1G,sCAAsC;IAChC,aAAa,CAAC,OAAO,CAAC,EAAE,cAAc;;;;;;;;;;;;;;;;;;;;;;IAS5C,8BAA8B;IACxB,YAAY,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,cAAc;;;;IAUlD,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IASX,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;IASvD,6CAA6C;IACvC,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAW,EAAE,OAAO,CAAC,EAAE,cAAc;IASjG;;OAEG;cACa,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;CAsClD"}

package/dist/src/connectors/http.d.ts

@@ -19,5 +19,6 @@ export declare class HttpConnector extends BaseConnector {
     constructor(baseUrl: string, opts?: HttpConnectorOptions);
     /** Establish connection to the MCP implementation via SSE. */
     connect(): Promise<void>;
+    get publicIdentifier(): Record<string, string>;
 }
 //# sourceMappingURL=http.d.ts.map

package/dist/src/connectors/http.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"http.d.ts","sourceRoot":"","sources":["../../../src/connectors/http.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,WAAW,CAAA;AAIrD,OAAO,EAAE,aAAa,EAAE,MAAM,WAAW,CAAA;AAEzC,MAAM,WAAW,oBAAqB,SAAQ,oBAAoB;IAChE,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAChC,OAAO,CAAC,EAAE,MAAM,CAAA;IAChB,cAAc,CAAC,EAAE,MAAM,CAAA;IACvB,UAAU,CAAC,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAA;CAC/C;AAED,qBAAa,aAAc,SAAQ,aAAa;IAC9C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAQ;IAChC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAwB;IAChD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAQ;IAChC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAQ;IACvC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAmC;gBAElD,OAAO,EAAE,MAAM,EAAE,IAAI,GAAE,oBAAyB;IAc5D,8DAA8D;IACxD,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CAoC/B"}
+{"version":3,"file":"http.d.ts","sourceRoot":"","sources":["../../../src/connectors/http.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,WAAW,CAAA;AAIrD,OAAO,EAAE,aAAa,EAAE,MAAM,WAAW,CAAA;AAEzC,MAAM,WAAW,oBAAqB,SAAQ,oBAAoB;IAChE,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAChC,OAAO,CAAC,EAAE,MAAM,CAAA;IAChB,cAAc,CAAC,EAAE,MAAM,CAAA;IACvB,UAAU,CAAC,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAA;CAC/C;AAED,qBAAa,aAAc,SAAQ,aAAa;IAC9C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAQ;IAChC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAwB;IAChD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAQ;IAChC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAQ;IACvC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAmC;gBAElD,OAAO,EAAE,MAAM,EAAE,IAAI,GAAE,oBAAyB;IAc5D,8DAA8D;IACxD,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAqC9B,IAAI,gBAAgB,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAK7C;CACF"}

package/dist/src/connectors/http.js

@@ -48,4 +48,10 @@ export class HttpConnector extends BaseConnector {
             throw err;
         }
     }
+    get publicIdentifier() {
+        return {
+            type: 'http',
+            url: this.baseUrl,
+        };
+    }
 }

package/dist/src/connectors/stdio.d.ts

@@ -21,5 +21,6 @@ export declare class StdioConnector extends BaseConnector {
     } & StdioConnectorOptions);
     /** Establish connection to the MCP implementation. */
     connect(): Promise<void>;
+    get publicIdentifier(): Record<string, string>;
 }
 //# sourceMappingURL=stdio.d.ts.map

package/dist/src/connectors/stdio.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"stdio.d.ts","sourceRoot":"","sources":["../../../src/connectors/stdio.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAA;AAE3C,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,WAAW,CAAA;AAMrD,OAAO,EAAE,aAAa,EAAE,MAAM,WAAW,CAAA;AAEzC,MAAM,WAAW,qBAAsB,SAAQ,oBAAoB;IACjE,UAAU,CAAC,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAA;CAC/C;AAED,qBAAa,cAAe,SAAQ,aAAa;IAC/C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAQ;IAChC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAU;IAC/B,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAwB;IAC7C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAU;IACjC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAmC;gBAG5D,EACE,OAAe,EACf,IAAS,EACT,GAAG,EACH,MAAuB,EACvB,GAAG,IAAI,EACR,GAAE;QACD,OAAO,CAAC,EAAE,MAAM,CAAA;QAChB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAA;QACf,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;QAC5B,MAAM,CAAC,EAAE,QAAQ,CAAA;KAClB,GAAG,qBAA0B;IAWhC,sDAAsD;IAChD,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;CA+C/B"}
+{"version":3,"file":"stdio.d.ts","sourceRoot":"","sources":["../../../src/connectors/stdio.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAA;AAE3C,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,WAAW,CAAA;AAMrD,OAAO,EAAE,aAAa,EAAE,MAAM,WAAW,CAAA;AAEzC,MAAM,WAAW,qBAAsB,SAAQ,oBAAoB;IACjE,UAAU,CAAC,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAA;CAC/C;AAED,qBAAa,cAAe,SAAQ,aAAa;IAC/C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAQ;IAChC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAU;IAC/B,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAwB;IAC7C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAU;IACjC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAmC;gBAG5D,EACE,OAAe,EACf,IAAS,EACT,GAAG,EACH,MAAuB,EACvB,GAAG,IAAI,EACR,GAAE;QACD,OAAO,CAAC,EAAE,MAAM,CAAA;QAChB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAA;QACf,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;QAC5B,MAAM,CAAC,EAAE,QAAQ,CAAA;KAClB,GAAG,qBAA0B;IAWhC,sDAAsD;IAChD,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAgD9B,IAAI,gBAAgB,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAK7C;CACF"}

package/dist/src/connectors/stdio.js

@@ -59,4 +59,10 @@ export class StdioConnector extends BaseConnector {
             throw err;
         }
     }
+    get publicIdentifier() {
+        return {
+            'type': 'stdio',
+            'command&args': `${this.command} ${this.args.join(' ')}`,
+        };
+    }
 }

package/dist/src/connectors/websocket.d.ts

@@ -30,5 +30,6 @@ export declare class WebSocketConnector extends BaseConnector {
     request(method: string, params?: Record<string, any> | null): Promise<any>;
     get tools(): Tool[];
     protected cleanupResources(): Promise<void>;
+    get publicIdentifier(): Record<string, string>;
 }
 //# sourceMappingURL=websocket.d.ts.map

package/dist/src/connectors/websocket.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"websocket.d.ts","sourceRoot":"","sources":["../../../src/connectors/websocket.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,cAAc,EACd,IAAI,EACL,MAAM,oCAAoC,CAAA;AAG3C,OAAO,EAAE,0BAA0B,EAAE,MAAM,+BAA+B,CAAA;AAE1E,OAAO,EAAE,aAAa,EAAE,MAAM,WAAW,CAAA;AAEzC,MAAM,WAAW,yBAAyB;IACxC,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;CACjC;AAED,qBAAa,kBAAmB,SAAQ,aAAa;IACnD,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAQ;IAC5B,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAwB;IAEhD,SAAS,CAAC,iBAAiB,EAAE,0BAA0B,GAAG,IAAI,CAAO;IACrE,OAAO,CAAC,EAAE,CAAkD;IAC5D,OAAO,CAAC,YAAY,CAA6B;IACjD,OAAO,CAAC,OAAO,CACF;IAEb,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,GAAG,IAAI,CAAO;gBAE9B,GAAG,EAAE,MAAM,EAAE,IAAI,GAAE,yBAA8B;IAQvD,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAqBxB,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAUjC,OAAO,CAAC,WAAW;YAiBL,WAAW;IAyCzB,OAAO,CAAC,SAAS;IAKX,UAAU,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAS1C,SAAS,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;IAK5B,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,OAAO,CAAC,cAAc,CAAC;IAI1E,aAAa,IAAI,OAAO,CAAC,GAAG,CAAC;IAK7B,YAAY,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,OAAO,EAAE,WAAW,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC;IAK9E,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAW,GAAG,OAAO,CAAC,GAAG,CAAC;IAItF,IAAI,KAAK,IAAI,IAAI,EAAE,CAIlB;cAEe,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;CAkBlD"}
+{"version":3,"file":"websocket.d.ts","sourceRoot":"","sources":["../../../src/connectors/websocket.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,cAAc,EACd,IAAI,EACL,MAAM,oCAAoC,CAAA;AAG3C,OAAO,EAAE,0BAA0B,EAAE,MAAM,+BAA+B,CAAA;AAE1E,OAAO,EAAE,aAAa,EAAE,MAAM,WAAW,CAAA;AAEzC,MAAM,WAAW,yBAAyB;IACxC,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;CACjC;AAED,qBAAa,kBAAmB,SAAQ,aAAa;IACnD,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAQ;IAC5B,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAwB;IAEhD,SAAS,CAAC,iBAAiB,EAAE,0BAA0B,GAAG,IAAI,CAAO;IACrE,OAAO,CAAC,EAAE,CAAkD;IAC5D,OAAO,CAAC,YAAY,CAA6B;IACjD,OAAO,CAAC,OAAO,CACF;IAEb,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,GAAG,IAAI,CAAO;gBAE9B,GAAG,EAAE,MAAM,EAAE,IAAI,GAAE,yBAA8B;IAQvD,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAqBxB,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAUjC,OAAO,CAAC,WAAW;YAiBL,WAAW;IAyCzB,OAAO,CAAC,SAAS;IAKX,UAAU,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAS1C,SAAS,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;IAK5B,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,OAAO,CAAC,cAAc,CAAC;IAI1E,aAAa,IAAI,OAAO,CAAC,GAAG,CAAC;IAK7B,YAAY,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,OAAO,EAAE,WAAW,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,CAAC;IAK9E,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,GAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,IAAW,GAAG,OAAO,CAAC,GAAG,CAAC;IAItF,IAAI,KAAK,IAAI,IAAI,EAAE,CAIlB;cAEe,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;IAmBjD,IAAI,gBAAgB,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAK7C;CACF"}

package/dist/src/connectors/websocket.js

@@ -148,4 +148,10 @@ export class WebSocketConnector extends BaseConnector {
         }
         this.toolsCache = null;
     }
+    get publicIdentifier() {
+        return {
+            type: 'websocket',
+            url: this.url,
+        };
+    }
 }

package/dist/src/logging.js

@@ -10,7 +10,7 @@ function resolveLevel(env) {
         case '1':
             return 'info';
         default:
-            return 'warn';
+            return 'info';
     }
 }
 const minimalFormatter = printf(({ level, message, label, timestamp }) => {
@@ -77,7 +77,7 @@ export class Logger {
         else if (enabled === 1)
             level = 'info';
         else
-            level = 'warn';
+            level = 'info';
         Object.values(this.instances).forEach((logger) => {
             logger.level = level;
         });

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/dist/src/telemetry/events.d.ts

@@ -0,0 +1,35 @@
+export declare abstract class BaseTelemetryEvent {
+    abstract get name(): string;
+    abstract get properties(): Record<string, any>;
+}
+export interface MCPAgentExecutionEventData {
+    executionMethod: string;
+    query: string;
+    success: boolean;
+    modelProvider: string;
+    modelName: string;
+    serverCount: number;
+    serverIdentifiers: Array<Record<string, string>>;
+    totalToolsAvailable: number;
+    toolsAvailableNames: string[];
+    maxStepsConfigured: number;
+    memoryEnabled: boolean;
+    useServerManager: boolean;
+    maxStepsUsed: number | null;
+    manageConnector: boolean;
+    externalHistoryUsed: boolean;
+    stepsTaken?: number | null;
+    toolsUsedCount?: number | null;
+    toolsUsedNames?: string[] | null;
+    response?: string | null;
+    executionTimeMs?: number | null;
+    errorType?: string | null;
+    conversationHistoryLength?: number | null;
+}
+export declare class MCPAgentExecutionEvent extends BaseTelemetryEvent {
+    private data;
+    constructor(data: MCPAgentExecutionEventData);
+    get name(): string;
+    get properties(): Record<string, any>;
+}
+//# sourceMappingURL=events.d.ts.map

package/dist/src/telemetry/events.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"events.d.ts","sourceRoot":"","sources":["../../../src/telemetry/events.ts"],"names":[],"mappings":"AAAA,8BAAsB,kBAAkB;IACtC,QAAQ,KAAK,IAAI,IAAI,MAAM,CAAA;IAC3B,QAAQ,KAAK,UAAU,IAAI,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;CAC/C;AAED,MAAM,WAAW,0BAA0B;IAEzC,eAAe,EAAE,MAAM,CAAA;IACvB,KAAK,EAAE,MAAM,CAAA;IACb,OAAO,EAAE,OAAO,CAAA;IAGhB,aAAa,EAAE,MAAM,CAAA;IACrB,SAAS,EAAE,MAAM,CAAA;IACjB,WAAW,EAAE,MAAM,CAAA;IACnB,iBAAiB,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAA;IAChD,mBAAmB,EAAE,MAAM,CAAA;IAC3B,mBAAmB,EAAE,MAAM,EAAE,CAAA;IAC7B,kBAAkB,EAAE,MAAM,CAAA;IAC1B,aAAa,EAAE,OAAO,CAAA;IACtB,gBAAgB,EAAE,OAAO,CAAA;IAGzB,YAAY,EAAE,MAAM,GAAG,IAAI,CAAA;IAC3B,eAAe,EAAE,OAAO,CAAA;IACxB,mBAAmB,EAAE,OAAO,CAAA;IAG5B,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;IAC1B,cAAc,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;IAC9B,cAAc,CAAC,EAAE,MAAM,EAAE,GAAG,IAAI,CAAA;IAChC,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;IACxB,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;IAC/B,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;IAGzB,yBAAyB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAA;CAC1C;AAED,qBAAa,sBAAuB,SAAQ,kBAAkB;IAChD,OAAO,CAAC,IAAI;gBAAJ,IAAI,EAAE,0BAA0B;IAIpD,IAAI,IAAI,IAAI,MAAM,CAEjB;IAED,IAAI,UAAU,IAAI,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CA+BpC;CACF"}

package/dist/src/telemetry/events.js

@@ -0,0 +1,44 @@
+export class BaseTelemetryEvent {
+}
+export class MCPAgentExecutionEvent extends BaseTelemetryEvent {
+    data;
+    constructor(data) {
+        super();
+        this.data = data;
+    }
+    get name() {
+        return 'mcp_agent_execution';
+    }
+    get properties() {
+        return {
+            // Core execution info
+            execution_method: this.data.executionMethod,
+            query: this.data.query,
+            query_length: this.data.query.length,
+            success: this.data.success,
+            // Agent configuration
+            model_provider: this.data.modelProvider,
+            model_name: this.data.modelName,
+            server_count: this.data.serverCount,
+            server_identifiers: this.data.serverIdentifiers,
+            total_tools_available: this.data.totalToolsAvailable,
+            tools_available_names: this.data.toolsAvailableNames,
+            max_steps_configured: this.data.maxStepsConfigured,
+            memory_enabled: this.data.memoryEnabled,
+            use_server_manager: this.data.useServerManager,
+            // Execution parameters (always include, even if null)
+            max_steps_used: this.data.maxStepsUsed,
+            manage_connector: this.data.manageConnector,
+            external_history_used: this.data.externalHistoryUsed,
+            // Execution results (always include, even if null)
+            steps_taken: this.data.stepsTaken ?? null,
+            tools_used_count: this.data.toolsUsedCount ?? null,
+            tools_used_names: this.data.toolsUsedNames ?? null,
+            response: this.data.response ?? null,
+            response_length: this.data.response ? this.data.response.length : null,
+            execution_time_ms: this.data.executionTimeMs ?? null,
+            error_type: this.data.errorType ?? null,
+            conversation_history_length: this.data.conversationHistoryLength ?? null,
+        };
+    }
+}

package/dist/src/telemetry/index.d.ts

@@ -0,0 +1,5 @@
+export { BaseTelemetryEvent, MCPAgentExecutionEvent, MCPAgentExecutionEventData } from './events.js';
+export { Telemetry } from './telemetry.js';
+export { extractModelInfo, getModelName, getModelProvider, getPackageVersion } from './utils.js';
+export declare function setTelemetrySource(source: string): void;
+//# sourceMappingURL=index.d.ts.map

package/dist/src/telemetry/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/telemetry/index.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,kBAAkB,EAAE,sBAAsB,EAAE,0BAA0B,EAAE,MAAM,aAAa,CAAA;AACpG,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAA;AAC1C,OAAO,EAAE,gBAAgB,EAAE,YAAY,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAA;AAGhG,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAEvD"}

package/dist/src/telemetry/index.js

@@ -0,0 +1,8 @@
+import { Telemetry } from './telemetry.js';
+export { BaseTelemetryEvent, MCPAgentExecutionEvent } from './events.js';
+export { Telemetry } from './telemetry.js';
+export { extractModelInfo, getModelName, getModelProvider, getPackageVersion } from './utils.js';
+// Convenience function to set telemetry source globally
+export function setTelemetrySource(source) {
+    Telemetry.getInstance().setSource(source);
+}

package/dist/src/telemetry/telemetry.d.ts

@@ -0,0 +1,33 @@
+import type { BaseTelemetryEvent, MCPAgentExecutionEventData } from './events.js';
+export declare class Telemetry {
+    private static instance;
+    private readonly USER_ID_PATH;
+    private readonly VERSION_DOWNLOAD_PATH;
+    private readonly PROJECT_API_KEY;
+    private readonly HOST;
+    private readonly SCARF_GATEWAY_URL;
+    private readonly UNKNOWN_USER_ID;
+    private _currUserId;
+    private _posthogClient;
+    private _scarfClient;
+    private _source;
+    private constructor();
+    static getInstance(): Telemetry;
+    /**
+     * Set the source identifier for telemetry events.
+     * This allows tracking usage from different applications.
+     * @param source - The source identifier (e.g., "my-app", "cli", "vs-code-extension")
+     */
+    setSource(source: string): void;
+    /**
+     * Get the current source identifier.
+     */
+    getSource(): string;
+    get userId(): string;
+    capture(event: BaseTelemetryEvent): Promise<void>;
+    trackPackageDownload(properties?: Record<string, any>): Promise<void>;
+    trackAgentExecution(data: MCPAgentExecutionEventData): Promise<void>;
+    flush(): void;
+    shutdown(): void;
+}
+//# sourceMappingURL=telemetry.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"telemetry.d.ts","sourceRoot":"","sources":["../../../src/telemetry/telemetry.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,0BAA0B,EAAE,MAAM,aAAa,CAAA;AA0EjF,qBAAa,SAAS;IACpB,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAyB;IAEhD,OAAO,CAAC,QAAQ,CAAC,YAAY,CAA8D;IAC3F,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CAA2D;IACjG,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAoD;IACpF,OAAO,CAAC,QAAQ,CAAC,IAAI,CAA6B;IAClD,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAA8C;IAChF,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAoB;IAEpD,OAAO,CAAC,WAAW,CAAsB;IACzC,OAAO,CAAC,cAAc,CAAuB;IAC7C,OAAO,CAAC,YAAY,CAAgC;IACpD,OAAO,CAAC,OAAO,CAAuB;IAEtC,OAAO;IAwCP,MAAM,CAAC,WAAW,IAAI,SAAS;IAO/B;;;;OAIG;IACH,SAAS,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAK/B;;OAEG;IACH,SAAS,IAAI,MAAM;IAInB,IAAI,MAAM,IAAI,MAAM,CAgCnB;IAEK,OAAO,CAAC,KAAK,EAAE,kBAAkB,GAAG,OAAO,CAAC,IAAI,CAAC;IA4CjD,oBAAoB,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAqDrE,mBAAmB,CAAC,IAAI,EAAE,0BAA0B,GAAG,OAAO,CAAC,IAAI,CAAC;IAK1E,KAAK,IAAI,IAAI;IAkBb,QAAQ,IAAI,IAAI;CAiBjB"}

package/dist/src/telemetry/telemetry.js

@@ -0,0 +1,278 @@
+import * as fs from 'node:fs';
+import * as os from 'node:os';
+import * as path from 'node:path';
+import { PostHog } from 'posthog-node';
+import { v4 as uuidv4 } from 'uuid';
+import { logger } from '../logging.js';
+import { MCPAgentExecutionEvent } from './events.js';
+import { getPackageVersion } from './utils.js';
+// Simple Scarf event logger implementation
+class ScarfEventLogger {
+    endpoint;
+    timeout;
+    constructor(endpoint, timeout = 3000) {
+        this.endpoint = endpoint;
+        this.timeout = timeout;
+    }
+    async logEvent(properties) {
+        try {
+            const controller = new AbortController();
+            const timeoutId = setTimeout(() => controller.abort(), this.timeout);
+            const response = await fetch(this.endpoint, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+                body: JSON.stringify(properties),
+                signal: controller.signal,
+            });
+            clearTimeout(timeoutId);
+            if (!response.ok) {
+                throw new Error(`HTTP error! status: ${response.status}`);
+            }
+        }
+        catch (error) {
+            // Silently fail - telemetry should not break the application
+            logger.debug(`Failed to send Scarf event: ${error}`);
+        }
+    }
+}
+function getCacheHome() {
+    // XDG_CACHE_HOME for Linux and manually set envs
+    const envVar = process.env.XDG_CACHE_HOME;
+    if (envVar && path.isAbsolute(envVar)) {
+        return envVar;
+    }
+    const platform = process.platform;
+    const homeDir = os.homedir();
+    if (platform === 'win32') {
+        const appdata = process.env.LOCALAPPDATA || process.env.APPDATA;
+        if (appdata) {
+            return appdata;
+        }
+        return path.join(homeDir, 'AppData', 'Local');
+    }
+    else if (platform === 'darwin') {
+        // macOS
+        return path.join(homeDir, 'Library', 'Caches');
+    }
+    else {
+        // Linux or other Unix
+        return path.join(homeDir, '.cache');
+    }
+}
+export class Telemetry {
+    static instance = null;
+    USER_ID_PATH = path.join(getCacheHome(), 'mcp_use_3', 'telemetry_user_id');
+    VERSION_DOWNLOAD_PATH = path.join(getCacheHome(), 'mcp_use', 'download_version');
+    PROJECT_API_KEY = 'phc_lyTtbYwvkdSbrcMQNPiKiiRWrrM1seyKIMjycSvItEI';
+    HOST = 'https://eu.i.posthog.com';
+    SCARF_GATEWAY_URL = 'https://mcpuse.gateway.scarf.sh/events-ts';
+    UNKNOWN_USER_ID = 'UNKNOWN_USER_ID';
+    _currUserId = null;
+    _posthogClient = null;
+    _scarfClient = null;
+    _source = 'typescript';
+    constructor() {
+        const telemetryDisabled = process.env.MCP_USE_ANONYMIZED_TELEMETRY?.toLowerCase() === 'false';
+        // Check for source from environment variable, default to 'typescript'
+        this._source = process.env.MCP_USE_TELEMETRY_SOURCE || 'typescript';
+        if (telemetryDisabled) {
+            this._posthogClient = null;
+            this._scarfClient = null;
+            logger.debug('Telemetry disabled');
+        }
+        else {
+            logger.info('Anonymized telemetry enabled. Set MCP_USE_ANONYMIZED_TELEMETRY=false to disable.');
+            // Initialize PostHog
+            try {
+                this._posthogClient = new PostHog(this.PROJECT_API_KEY, {
+                    host: this.HOST,
+                    disableGeoip: false,
+                });
+            }
+            catch (e) {
+                logger.warn(`Failed to initialize PostHog telemetry: ${e}`);
+                this._posthogClient = null;
+            }
+            // Initialize Scarf
+            try {
+                this._scarfClient = new ScarfEventLogger(this.SCARF_GATEWAY_URL, 3000);
+            }
+            catch (e) {
+                logger.warn(`Failed to initialize Scarf telemetry: ${e}`);
+                this._scarfClient = null;
+            }
+        }
+    }
+    static getInstance() {
+        if (!Telemetry.instance) {
+            Telemetry.instance = new Telemetry();
+        }
+        return Telemetry.instance;
+    }
+    /**
+     * Set the source identifier for telemetry events.
+     * This allows tracking usage from different applications.
+     * @param source - The source identifier (e.g., "my-app", "cli", "vs-code-extension")
+     */
+    setSource(source) {
+        this._source = source;
+        logger.debug(`Telemetry source set to: ${source}`);
+    }
+    /**
+     * Get the current source identifier.
+     */
+    getSource() {
+        return this._source;
+    }
+    get userId() {
+        if (this._currUserId) {
+            return this._currUserId;
+        }
+        try {
+            const isFirstTime = !fs.existsSync(this.USER_ID_PATH);
+            if (isFirstTime) {
+                logger.debug(`Creating user ID path: ${this.USER_ID_PATH}`);
+                fs.mkdirSync(path.dirname(this.USER_ID_PATH), { recursive: true });
+                const newUserId = uuidv4();
+                fs.writeFileSync(this.USER_ID_PATH, newUserId);
+                this._currUserId = newUserId;
+                logger.debug(`User ID path created: ${this.USER_ID_PATH}`);
+            }
+            else {
+                this._currUserId = fs.readFileSync(this.USER_ID_PATH, 'utf-8').trim();
+            }
+            // Always check for version-based download tracking
+            // Note: We can't await here since this is a getter, so we fire and forget
+            this.trackPackageDownload({
+                triggered_by: 'user_id_property',
+            }).catch(e => logger.debug(`Failed to track package download: ${e}`));
+        }
+        catch (e) {
+            logger.debug(`Failed to get/create user ID: ${e}`);
+            this._currUserId = this.UNKNOWN_USER_ID;
+        }
+        return this._currUserId;
+    }
+    async capture(event) {
+        if (!this._posthogClient && !this._scarfClient) {
+            return;
+        }
+        // Send to PostHog
+        if (this._posthogClient) {
+            try {
+                // Add package version, language flag, and source to all events
+                const properties = { ...event.properties };
+                properties.mcp_use_version = getPackageVersion();
+                properties.language = 'typescript';
+                properties.source = this._source;
+                this._posthogClient.capture({
+                    distinctId: this.userId,
+                    event: event.name,
+                    properties,
+                });
+            }
+            catch (e) {
+                logger.debug(`Failed to track PostHog event ${event.name}: ${e}`);
+            }
+        }
+        // Send to Scarf (when implemented)
+        if (this._scarfClient) {
+            try {
+                // Add package version, user_id, language flag, and source to all events
+                const properties = {};
+                properties.mcp_use_version = getPackageVersion();
+                properties.user_id = this.userId;
+                properties.event = event.name;
+                properties.language = 'typescript';
+                properties.source = this._source;
+                await this._scarfClient.logEvent(properties);
+            }
+            catch (e) {
+                logger.debug(`Failed to track Scarf event ${event.name}: ${e}`);
+            }
+        }
+    }
+    async trackPackageDownload(properties) {
+        if (!this._scarfClient) {
+            return;
+        }
+        try {
+            const currentVersion = getPackageVersion();
+            let shouldTrack = false;
+            let firstDownload = false;
+            // Check if version file exists
+            if (!fs.existsSync(this.VERSION_DOWNLOAD_PATH)) {
+                // First download
+                shouldTrack = true;
+                firstDownload = true;
+                // Create directory and save version
+                fs.mkdirSync(path.dirname(this.VERSION_DOWNLOAD_PATH), { recursive: true });
+                fs.writeFileSync(this.VERSION_DOWNLOAD_PATH, currentVersion);
+            }
+            else {
+                // Read saved version
+                const savedVersion = fs.readFileSync(this.VERSION_DOWNLOAD_PATH, 'utf-8').trim();
+                // Compare versions (simple string comparison for now)
+                if (currentVersion > savedVersion) {
+                    shouldTrack = true;
+                    firstDownload = false;
+                    // Update saved version
+                    fs.writeFileSync(this.VERSION_DOWNLOAD_PATH, currentVersion);
+                }
+            }
+            if (shouldTrack) {
+                logger.debug(`Tracking package download event with properties: ${JSON.stringify(properties)}`);
+                // Add package version, user_id, language flag, and source to event
+                const eventProperties = { ...(properties || {}) };
+                eventProperties.mcp_use_version = currentVersion;
+                eventProperties.user_id = this.userId;
+                eventProperties.event = 'package_download';
+                eventProperties.first_download = firstDownload;
+                eventProperties.language = 'typescript';
+                eventProperties.source = this._source;
+                await this._scarfClient.logEvent(eventProperties);
+            }
+        }
+        catch (e) {
+            logger.debug(`Failed to track Scarf package_download event: ${e}`);
+        }
+    }
+    async trackAgentExecution(data) {
+        const event = new MCPAgentExecutionEvent(data);
+        await this.capture(event);
+    }
+    flush() {
+        // Flush PostHog
+        if (this._posthogClient) {
+            try {
+                this._posthogClient.flush();
+                logger.debug('PostHog client telemetry queue flushed');
+            }
+            catch (e) {
+                logger.debug(`Failed to flush PostHog client: ${e}`);
+            }
+        }
+        // Scarf events are sent immediately, no flush needed
+        if (this._scarfClient) {
+            logger.debug('Scarf telemetry events sent immediately (no flush needed)');
+        }
+    }
+    shutdown() {
+        // Shutdown PostHog
+        if (this._posthogClient) {
+            try {
+                this._posthogClient.shutdown();
+                logger.debug('PostHog client shutdown successfully');
+            }
+            catch (e) {
+                logger.debug(`Error shutting down PostHog client: ${e}`);
+            }
+        }
+        // Scarf doesn't require explicit shutdown
+        if (this._scarfClient) {
+            logger.debug('Scarf telemetry client shutdown (no action needed)');
+        }
+    }
+}

package/dist/src/telemetry/utils.d.ts

@@ -0,0 +1,6 @@
+import type { BaseLanguageModel } from '@langchain/core/language_models/base';
+export declare function getPackageVersion(): string;
+export declare function getModelProvider(llm: BaseLanguageModel): string;
+export declare function getModelName(llm: BaseLanguageModel): string;
+export declare function extractModelInfo(llm: BaseLanguageModel): [string, string];
+//# sourceMappingURL=utils.d.ts.map

package/dist/src/telemetry/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/telemetry/utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,sCAAsC,CAAA;AAI7E,wBAAgB,iBAAiB,IAAI,MAAM,CAS1C;AAED,wBAAgB,gBAAgB,CAAC,GAAG,EAAE,iBAAiB,GAAG,MAAM,CAG/D;AAED,wBAAgB,YAAY,CAAC,GAAG,EAAE,iBAAiB,GAAG,MAAM,CAgB3D;AAED,wBAAgB,gBAAgB,CAAC,GAAG,EAAE,iBAAiB,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAEzE"}

package/dist/src/telemetry/utils.js

@@ -0,0 +1,35 @@
+import * as fs from 'node:fs';
+import * as path from 'node:path';
+export function getPackageVersion() {
+    try {
+        const packagePath = path.join(__dirname, '../../package.json');
+        const packageJson = JSON.parse(fs.readFileSync(packagePath, 'utf-8'));
+        return packageJson.version || 'unknown';
+    }
+    catch {
+        return 'unknown';
+    }
+}
+export function getModelProvider(llm) {
+    // Use LangChain's standard _llm_type property for identification
+    return llm._llm_type || llm.constructor.name.toLowerCase();
+}
+export function getModelName(llm) {
+    // First try _identifying_params which may contain model info
+    if ('_identifyingParams' in llm) {
+        const identifyingParams = llm._identifyingParams;
+        if (typeof identifyingParams === 'object' && identifyingParams !== null) {
+            // Common keys that contain model names
+            for (const key of ['model', 'modelName', 'model_name', 'modelId', 'model_id', 'deploymentName', 'deployment_name']) {
+                if (key in identifyingParams) {
+                    return String(identifyingParams[key]);
+                }
+            }
+        }
+    }
+    // Fallback to direct model attributes
+    return llm.model || llm.modelName || llm.constructor.name;
+}
+export function extractModelInfo(llm) {
+    return [getModelProvider(llm), getModelName(llm)];
+}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "mcp-use",
   "type": "module",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "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",
@@ -75,11 +75,13 @@
     "@langchain/core": "0.3.58",
     "@langchain/openai": "^0.5.15",
     "@modelcontextprotocol/sdk": "1.12.1",
+    "@scarf/scarf": "^1.4.0",
     "dotenv": "^16.5.0",
     "eventsource": "^3.0.6",
     "fastembed": "^1.14.4",
     "langchain": "^0.3.27",
     "lodash-es": "^4.17.21",
+    "posthog-node": "^5.1.1",
     "uuid": "^11.1.0",
     "winston": "^3.17.0",
     "ws": "^8.18.2",

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"}