mcp-use 0.1.6 → 0.1.8

package/dist/index.js

@@ -8,6 +8,8 @@ import { WebSocketConnector } from './src/connectors/websocket.js';
 import { Logger, logger } from './src/logging.js';
 import { MCPSession } from './src/session.js';
 export { BaseAdapter, LangChainAdapter } from './src/adapters/index.js';
+// Export AI SDK utilities
+export * from './src/agents/utils/index.js';
 export { ServerManager } from './src/managers/server_manager.js';
 export * from './src/managers/tools/index.js';
 // Export telemetry utilities

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

@@ -1,6 +1,7 @@
 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 { StreamEvent } from '@langchain/core/tracers/log_stream';
 import type { AgentStep } from 'langchain/agents';
 import type { MCPClient } from '../client.js';
 import type { BaseConnector } from '../connectors/base.js';
@@ -21,12 +22,12 @@ export declare class MCPAgent {
     private systemPrompt?;
     private systemPromptTemplateOverride?;
     private additionalInstructions?;
-    private initialized;
+    private _initialized;
     private conversationHistory;
-    private agentExecutor;
+    private _agentExecutor;
     private sessions;
     private systemMessage;
-    private tools;
+    private _tools;
     private adapter;
     private serverManager;
     private telemetry;
@@ -69,5 +70,10 @@ export declare class MCPAgent {
      */
     stream(query: string, maxSteps?: number, manageConnector?: boolean, externalHistory?: BaseMessage[]): AsyncGenerator<AgentStep, string, void>;
     close(): Promise<void>;
+    /**
+     * Yields LangChain StreamEvent objects from the underlying streamEvents() method.
+     * This provides token-level streaming and fine-grained event updates.
+     */
+    streamEvents(query: string, maxSteps?: number, manageConnector?: boolean, externalHistory?: BaseMessage[]): AsyncGenerator<StreamEvent, void, 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;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"}
+{"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,WAAW,EAAE,MAAM,oCAAoC,CAAA;AACrE,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,YAAY,CAAQ;IAC5B,OAAO,CAAC,mBAAmB,CAAoB;IAC/C,OAAO,CAAC,cAAc,CAA6B;IACnD,OAAO,CAAC,QAAQ,CAAiC;IACjD,OAAO,CAAC,aAAa,CAA6B;IAClD,OAAO,CAAC,MAAM,CAAgC;IAC9C,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;IAsDY,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;IA0M7B,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IA0BnC;;;OAGG;IACW,YAAY,CACxB,KAAK,EAAE,MAAM,EACb,QAAQ,CAAC,EAAE,MAAM,EACjB,eAAe,UAAO,EACtB,eAAe,CAAC,EAAE,WAAW,EAAE,GAC9B,cAAc,CAAC,WAAW,EAAE,IAAI,EAAE,IAAI,CAAC;CAwI3C"}

package/dist/src/agents/mcp_agent.js

@@ -22,12 +22,12 @@ export class MCPAgent {
     systemPrompt;
     systemPromptTemplateOverride;
     additionalInstructions;
-    initialized = false;
+    _initialized = false;
     conversationHistory = [];
-    agentExecutor = null;
+    _agentExecutor = null;
     sessions = {};
     systemMessage = null;
-    tools = [];
+    _tools = [];
     adapter;
     serverManager = null;
     telemetry;
@@ -67,6 +67,19 @@ export class MCPAgent {
         const [provider, name] = extractModelInfo(this.llm);
         this.modelProvider = provider;
         this.modelName = name;
+        // Make getters configurable for test mocking
+        Object.defineProperty(this, 'agentExecutor', {
+            get: () => this._agentExecutor,
+            configurable: true,
+        });
+        Object.defineProperty(this, 'tools', {
+            get: () => this._tools,
+            configurable: true,
+        });
+        Object.defineProperty(this, 'initialized', {
+            get: () => this._initialized,
+            configurable: true,
+        });
     }
     async initialize() {
         logger.info('🚀 Initializing MCP agent and connecting to services...');
@@ -75,11 +88,11 @@ export class MCPAgent {
             await this.serverManager.initialize();
             // Get server management tools
             const managementTools = this.serverManager.tools;
-            this.tools = managementTools;
-            this.tools.push(...this.additionalTools);
+            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);
+            await this.createSystemMessageFromTools(this._tools);
         }
         else {
             // Standard initialization - if using client, get or create sessions
@@ -94,9 +107,9 @@ export class MCPAgent {
                     logger.info(`✅ Created ${Object.keys(this.sessions).length} new sessions`);
                 }
                 // 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`);
+                this._tools = await LangChainAdapter.createTools(this.client);
+                this._tools.push(...this.additionalTools);
+                logger.info(`🛠️ Created ${this._tools.length} LangChain tools from client`);
             }
             else {
                 // Using direct connector - only establish connection
@@ -107,18 +120,18 @@ 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`);
+                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
-            logger.info(`🧰 Found ${this.tools.length} tools across all connectors`);
+            logger.info(`🧰 Found ${this._tools.length} tools across all connectors`);
             // Create the system message based on available tools
-            await this.createSystemMessageFromTools(this.tools);
+            await this.createSystemMessageFromTools(this._tools);
         }
         // Create the agent executor and mark initialized
-        this.agentExecutor = this.createAgent();
-        this.initialized = true;
+        this._agentExecutor = this.createAgent();
+        this._initialized = true;
         logger.info('✨ Agent initialization complete');
     }
     async createSystemMessageFromTools(tools) {
@@ -142,12 +155,12 @@ export class MCPAgent {
         ]);
         const agent = createToolCallingAgent({
             llm: this.llm,
-            tools: this.tools,
+            tools: this._tools,
             prompt,
         });
         return new AgentExecutor({
             agent,
-            tools: this.tools,
+            tools: this._tools,
             maxIterations: this.maxSteps,
             verbose: this.verbose,
             returnIntermediateSteps: true,
@@ -172,15 +185,15 @@ export class MCPAgent {
             this.conversationHistory = this.conversationHistory.filter(m => !(m instanceof SystemMessage));
             this.conversationHistory.unshift(this.systemMessage);
         }
-        if (this.initialized && this.tools.length) {
-            this.agentExecutor = this.createAgent();
+        if (this._initialized && this._tools.length) {
+            this._agentExecutor = this.createAgent();
             logger.debug('Agent recreated with new system message');
         }
     }
     setDisallowedTools(disallowedTools) {
         this.disallowedTools = disallowedTools;
         this.adapter = new LangChainAdapter(this.disallowedTools);
-        if (this.initialized) {
+        if (this._initialized) {
             logger.debug('Agent already initialized. Changes will take effect on next initialization.');
         }
     }
@@ -216,19 +229,19 @@ export class MCPAgent {
         let stepsTaken = 0;
         let success = false;
         try {
-            if (manageConnector && !this.initialized) {
+            if (manageConnector && !this._initialized) {
                 await this.initialize();
                 initializedHere = true;
             }
-            else if (!this.initialized && this.autoInitialize) {
+            else if (!this._initialized && this.autoInitialize) {
                 await this.initialize();
                 initializedHere = true;
             }
-            if (!this.agentExecutor) {
+            if (!this._agentExecutor) {
                 throw new Error('MCP agent failed to initialize');
             }
             const steps = maxSteps ?? this.maxSteps;
-            this.agentExecutor.maxIterations = steps;
+            this._agentExecutor.maxIterations = steps;
             const display_query = query.length > 50 ? `${query.slice(0, 50).replace(/\n/g, ' ')}...` : query.replace(/\n/g, ' ');
             logger.info(`💬 Received query: '${display_query}'`);
             // —–– Record user message
@@ -244,30 +257,30 @@ export class MCPAgent {
             }
             const intermediateSteps = [];
             const inputs = { input: query, chat_history: langchainHistory };
-            let nameToToolMap = Object.fromEntries(this.tools.map(t => [t.name, t]));
+            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));
-                    const existingToolNames = new Set(this.tools.map(t => t.name));
-                    const changed = currentTools.length !== this.tools.length
+                    const existingToolNames = new Set(this._tools.map(t => t.name));
+                    const changed = currentTools.length !== this._tools.length
                         || [...currentToolNames].some(n => !existingToolNames.has(n));
                     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;
-                        nameToToolMap = Object.fromEntries(this.tools.map(t => [t.name, t]));
+                        this._tools = currentTools;
+                        this._tools.push(...this.additionalTools);
+                        await this.createSystemMessageFromTools(this._tools);
+                        this._agentExecutor = this.createAgent();
+                        this._agentExecutor.maxIterations = steps;
+                        nameToToolMap = Object.fromEntries(this._tools.map(t => [t.name, t]));
                     }
                 }
                 logger.info(`👣 Step ${stepNum + 1}/${steps}`);
                 try {
                     logger.debug('Starting agent step execution');
-                    const nextStepOutput = await this.agentExecutor._takeNextStep(nameToToolMap, inputs, intermediateSteps);
+                    const nextStepOutput = await this._agentExecutor._takeNextStep(nameToToolMap, inputs, intermediateSteps);
                     if (nextStepOutput.returnValues) {
                         logger.info(`✅ Agent finished at step ${stepNum + 1}`);
                         result = nextStepOutput.returnValues?.output ?? 'No output generated';
@@ -280,7 +293,9 @@ export class MCPAgent {
                         const { action, observation } = step;
                         const toolName = action.tool;
                         toolsUsedNames.push(toolName);
-                        let toolInputStr = String(action.toolInput);
+                        let toolInputStr = typeof action.toolInput === 'string'
+                            ? action.toolInput
+                            : JSON.stringify(action.toolInput, null, 2);
                         if (toolInputStr.length > 100)
                             toolInputStr = `${toolInputStr.slice(0, 97)}...`;
                         logger.info(`🔧 Tool call: ${toolName} with input: ${toolInputStr}`);
@@ -293,7 +308,7 @@ export class MCPAgent {
                     // Detect direct return
                     if (stepArray.length) {
                         const lastStep = stepArray[stepArray.length - 1];
-                        const toolReturn = await this.agentExecutor._getToolReturn(lastStep);
+                        const toolReturn = await this._agentExecutor._getToolReturn(lastStep);
                         if (toolReturn) {
                             logger.info(`🏆 Tool returned directly at step ${stepNum + 1}`);
                             result = toolReturn.returnValues?.output ?? 'No output generated';
@@ -352,8 +367,8 @@ export class MCPAgent {
                 modelName: this.modelName,
                 serverCount,
                 serverIdentifiers: this.connectors.map(connector => connector.publicIdentifier),
-                totalToolsAvailable: this.tools.length,
-                toolsAvailableNames: this.tools.map(t => t.name),
+                totalToolsAvailable: this._tools.length,
+                toolsAvailableNames: this._tools.map(t => t.name),
                 maxStepsConfigured: this.maxSteps,
                 memoryEnabled: this.memoryEnabled,
                 useServerManager: this.useServerManager,
@@ -377,8 +392,8 @@ export class MCPAgent {
     async close() {
         logger.info('🔌 Closing MCPAgent resources…');
         try {
-            this.agentExecutor = null;
-            this.tools = [];
+            this._agentExecutor = null;
+            this._tools = [];
             if (this.client) {
                 logger.info('🔄 Closing sessions through client');
                 await this.client.closeAllSessions();
@@ -395,8 +410,126 @@ export class MCPAgent {
             }
         }
         finally {
-            this.initialized = false;
+            this._initialized = false;
             logger.info('👋 Agent closed successfully');
         }
     }
+    /**
+     * Yields LangChain StreamEvent objects from the underlying streamEvents() method.
+     * This provides token-level streaming and fine-grained event updates.
+     */
+    async *streamEvents(query, maxSteps, manageConnector = true, externalHistory) {
+        let initializedHere = false;
+        const startTime = Date.now();
+        let success = false;
+        let eventCount = 0;
+        let totalResponseLength = 0;
+        try {
+            // Initialize if needed
+            if (manageConnector && !this._initialized) {
+                await this.initialize();
+                initializedHere = true;
+            }
+            else if (!this._initialized && this.autoInitialize) {
+                await this.initialize();
+                initializedHere = true;
+            }
+            const agentExecutor = this.agentExecutor;
+            if (!agentExecutor) {
+                throw new Error('MCP agent failed to initialize');
+            }
+            // Set max iterations
+            const steps = maxSteps ?? this.maxSteps;
+            agentExecutor.maxIterations = steps;
+            const display_query = query.length > 50 ? `${query.slice(0, 50).replace(/\n/g, ' ')}...` : query.replace(/\n/g, ' ');
+            logger.info(`💬 Received query for streamEvents: '${display_query}'`);
+            // Add user message to history if memory enabled
+            if (this.memoryEnabled) {
+                this.addToHistory(new HumanMessage(query));
+            }
+            // Prepare history
+            const historyToUse = externalHistory ?? this.conversationHistory;
+            const langchainHistory = [];
+            for (const msg of historyToUse) {
+                if (msg instanceof HumanMessage || msg instanceof AIMessage) {
+                    langchainHistory.push(msg);
+                }
+            }
+            // Prepare inputs
+            const inputs = { input: query, chat_history: langchainHistory };
+            // Stream events from the agent executor
+            const eventStream = agentExecutor.streamEvents(inputs, { version: 'v2' });
+            // Yield each event
+            for await (const event of eventStream) {
+                eventCount++;
+                // Skip null or invalid events
+                if (!event || typeof event !== 'object') {
+                    continue;
+                }
+                // Track response length for telemetry
+                if (event.event === 'on_chat_model_stream' && event.data?.chunk?.content) {
+                    totalResponseLength += event.data.chunk.content.length;
+                }
+                yield event;
+                // Handle final message for history
+                if (event.event === 'on_chain_end' && event.data?.output) {
+                    const output = event.data.output;
+                    if (typeof output === 'string' && this.memoryEnabled) {
+                        this.addToHistory(new AIMessage(output));
+                    }
+                    else if (output?.output && typeof output.output === 'string' && this.memoryEnabled) {
+                        this.addToHistory(new AIMessage(output.output));
+                    }
+                }
+            }
+            logger.info(`🎉 StreamEvents complete - ${eventCount} events emitted`);
+            success = true;
+        }
+        catch (e) {
+            logger.error(`❌ Error during streamEvents: ${e}`);
+            if (initializedHere && manageConnector) {
+                logger.info('🧹 Cleaning up resources after initialization error in streamEvents');
+                await this.close();
+            }
+            throw e;
+        }
+        finally {
+            // Track telemetry
+            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: 'streamEvents',
+                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,
+                response: `[STREAMED RESPONSE - ${totalResponseLength} chars]`,
+                executionTimeMs,
+                errorType: success ? null : 'streaming_error',
+                conversationHistoryLength,
+            });
+            // Clean up if needed
+            if (manageConnector && !this.client && initializedHere) {
+                logger.info('🧹 Closing agent after streamEvents completion');
+                await this.close();
+            }
+        }
+    }
 }

package/dist/src/agents/utils/ai_sdk.d.ts

@@ -0,0 +1,22 @@
+/**
+ * AI SDK Integration Utilities
+ *
+ * Utility functions for integrating MCPAgent's streamEvents with Vercel AI SDK.
+ * These utilities help convert stream events to AI SDK compatible formats.
+ */
+import type { StreamEvent } from '@langchain/core/tracers/log_stream';
+/**
+ * Converts streamEvents to AI SDK compatible stream (basic version)
+ * Only yields the actual content tokens from chat model streams
+ */
+export declare function streamEventsToAISDK(streamEvents: AsyncGenerator<StreamEvent, void, void>): AsyncGenerator<string, void, void>;
+/**
+ * Converts async generator to ReadableStream for AI SDK compatibility
+ */
+export declare function createReadableStreamFromGenerator(generator: AsyncGenerator<string, void, void>): ReadableStream<string>;
+/**
+ * Enhanced adapter that includes tool information along with chat content
+ * Yields both content tokens and tool usage notifications
+ */
+export declare function streamEventsToAISDKWithTools(streamEvents: AsyncGenerator<StreamEvent, void, void>): AsyncGenerator<string, void, void>;
+//# sourceMappingURL=ai_sdk.d.ts.map

package/dist/src/agents/utils/ai_sdk.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai_sdk.d.ts","sourceRoot":"","sources":["../../../../src/agents/utils/ai_sdk.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,oCAAoC,CAAA;AAErE;;;GAGG;AACH,wBAAuB,mBAAmB,CACxC,YAAY,EAAE,cAAc,CAAC,WAAW,EAAE,IAAI,EAAE,IAAI,CAAC,GACpD,cAAc,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,CASpC;AAED;;GAEG;AACH,wBAAgB,iCAAiC,CAC/C,SAAS,EAAE,cAAc,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,GAC5C,cAAc,CAAC,MAAM,CAAC,CAcxB;AAED;;;GAGG;AACH,wBAAuB,4BAA4B,CACjD,YAAY,EAAE,cAAc,CAAC,WAAW,EAAE,IAAI,EAAE,IAAI,CAAC,GACpD,cAAc,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,CAqBpC"}

package/dist/src/agents/utils/ai_sdk.js

@@ -0,0 +1,62 @@
+/**
+ * AI SDK Integration Utilities
+ *
+ * Utility functions for integrating MCPAgent's streamEvents with Vercel AI SDK.
+ * These utilities help convert stream events to AI SDK compatible formats.
+ */
+/**
+ * Converts streamEvents to AI SDK compatible stream (basic version)
+ * Only yields the actual content tokens from chat model streams
+ */
+export async function* streamEventsToAISDK(streamEvents) {
+    for await (const event of streamEvents) {
+        if (event.event === 'on_chat_model_stream' && event.data?.chunk?.text) {
+            const textContent = event.data.chunk.text;
+            if (typeof textContent === 'string' && textContent.length > 0) {
+                yield textContent;
+            }
+        }
+    }
+}
+/**
+ * Converts async generator to ReadableStream for AI SDK compatibility
+ */
+export function createReadableStreamFromGenerator(generator) {
+    return new ReadableStream({
+        async start(controller) {
+            try {
+                for await (const chunk of generator) {
+                    controller.enqueue(chunk);
+                }
+                controller.close();
+            }
+            catch (error) {
+                controller.error(error);
+            }
+        },
+    });
+}
+/**
+ * Enhanced adapter that includes tool information along with chat content
+ * Yields both content tokens and tool usage notifications
+ */
+export async function* streamEventsToAISDKWithTools(streamEvents) {
+    for await (const event of streamEvents) {
+        switch (event.event) {
+            case 'on_chat_model_stream':
+                if (event.data?.chunk?.text) {
+                    const textContent = event.data.chunk.text;
+                    if (typeof textContent === 'string' && textContent.length > 0) {
+                        yield textContent;
+                    }
+                }
+                break;
+            case 'on_tool_start':
+                yield `\n🔧 Using tool: ${event.name}\n`;
+                break;
+            case 'on_tool_end':
+                yield `\n✅ Tool completed: ${event.name}\n`;
+                break;
+        }
+    }
+}

package/dist/src/agents/utils/index.d.ts

@@ -0,0 +1,2 @@
+export { createReadableStreamFromGenerator, streamEventsToAISDK, streamEventsToAISDKWithTools, } from './ai_sdk.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/src/agents/utils/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/agents/utils/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,iCAAiC,EACjC,mBAAmB,EACnB,4BAA4B,GAC7B,MAAM,aAAa,CAAA"}

package/dist/src/agents/utils/index.js

@@ -0,0 +1 @@
+export { createReadableStreamFromGenerator, streamEventsToAISDK, streamEventsToAISDKWithTools, } from './ai_sdk.js';

package/dist/tests/ai_sdk_compatibility.test.d.ts

@@ -0,0 +1,13 @@
+/**
+ * Tests for AI SDK compatibility with MCPAgent streamEvents()
+ *
+ * These tests verify that streamEvents() can be used with the AI SDK's
+ * LangChainAdapter for creating data stream responses compatible with
+ * Vercel AI SDK hooks like useCompletion and useChat.
+ */
+import type { StreamEvent } from '../index.js';
+declare function streamEventsToAISDK(streamEvents: AsyncGenerator<StreamEvent, void, void>): AsyncGenerator<string, void, void>;
+declare function streamEventsToCompleteContent(streamEvents: AsyncGenerator<StreamEvent, void, void>): AsyncGenerator<string, void, void>;
+declare function createReadableStreamFromGenerator(generator: AsyncGenerator<string, void, void>): ReadableStream<string>;
+export { createReadableStreamFromGenerator, streamEventsToAISDK, streamEventsToCompleteContent };
+//# sourceMappingURL=ai_sdk_compatibility.test.d.ts.map

package/dist/tests/ai_sdk_compatibility.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai_sdk_compatibility.test.d.ts","sourceRoot":"","sources":["../../tests/ai_sdk_compatibility.test.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,aAAa,CAAA;AAmD9C,iBAAgB,mBAAmB,CACjC,YAAY,EAAE,cAAc,CAAC,WAAW,EAAE,IAAI,EAAE,IAAI,CAAC,GACpD,cAAc,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,CAOpC;AAGD,iBAAgB,6BAA6B,CAC3C,YAAY,EAAE,cAAc,CAAC,WAAW,EAAE,IAAI,EAAE,IAAI,CAAC,GACpD,cAAc,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,CAoBpC;AAuJD,iBAAS,iCAAiC,CACxC,SAAS,EAAE,cAAc,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,GAC5C,cAAc,CAAC,MAAM,CAAC,CAcxB;AAGD,OAAO,EAAE,iCAAiC,EAAE,mBAAmB,EAAE,6BAA6B,EAAE,CAAA"}