refactor: Remove unnecessary console logs from ToolAgent and MCP core functionalities

Author: Amrrx

packages/components/nodes/agents/ToolAgent/ToolAgent.ts

@@ -172,8 +172,6 @@ class ToolAgent_Agents implements INode {
             const handler = new CustomChainHandler(sseStreamer, chatId)
             const allCallbacks = [loggerHandler, handler, ...callbacks]
 
-            console.log("[Tool Agent] This is from the 'shouldStreamResponse", chatId, sseStreamer)
-
             // Add detailed streaming handler if enabled
             if (enableDetailedStreaming && customStreamingHandler) {
                 allCallbacks.push(customStreamingHandler)
@@ -212,8 +210,6 @@ class ToolAgent_Agents implements INode {
         } else {
             const allCallbacks = [loggerHandler, ...callbacks]
 
-            console.log("[Tool Agent] This is from the Else 'shouldStreamResponse", chatId, sseStreamer)
-
             // Add detailed streaming handler if enabled
             if (enableDetailedStreaming && customStreamingHandler) {
                 allCallbacks.push(customStreamingHandler)

packages/components/nodes/tools/MCP/core.ts

@@ -95,9 +95,8 @@ export class MCPToolkit extends BaseToolkit {
             hasStreaming =
                 (capabilities as any)?.notifications?.streaming === true ||
                 (capabilities as any)?.experimental?.notifications?.streaming === true
-            console.log(`🔍 [MCP Core] Server streaming capability:`, hasStreaming)
         } catch (error) {
-            console.log(`⚠️ [MCP Core] Could not detect streaming capabilities, falling back to non-streaming:`, error.message)
+            console.error(`⚠️ [MCP Core] Could not detect streaming capabilities, falling back to non-streaming:`, error.message)
         }
 
         return { client, hasStreaming }
@@ -126,21 +125,6 @@ export class MCPToolkit extends BaseToolkit {
                 throw new Error('Client is not initialized')
             }
 
-            // Log the incoming tool definition for debugging
-            console.log(
-                `📋 [MCP Tool Definition] ${tool.name}:`,
-                JSON.stringify(
-                    {
-                        name: tool.name,
-                        description: tool.description,
-                        annotations: tool.annotations,
-                        inputSchema: tool.inputSchema ? 'present' : 'missing'
-                    },
-                    null,
-                    2
-                )
-            )
-
             return await MCPTool({
                 toolkit: this,
                 name: tool.name,
@@ -175,14 +159,9 @@ export async function MCPTool({
     const { client, hasStreaming } = await toolkit.createClient()
     await client.close()
 
-    // Check if tool has streaming annotations
     const toolHasStreaming = annotations.streaming_enabled === true
     const shouldUseStreaming = hasStreaming && toolHasStreaming
 
-    console.log(
-        `🔍 [MCP Tool] ${name} - Server streaming: ${hasStreaming}, Tool streaming: ${toolHasStreaming}, Using streaming: ${shouldUseStreaming}`
-    )
-
     return tool(
         async (input, config): Promise<string> => {
             return await executeMCPTool(toolkit, name, input, config, annotations)
@@ -267,7 +246,6 @@ async function handleToolResponse(
     }
 
     // Streaming tools wait for completion
-    console.log(`🔄 [MCP Tool] Waiting for streaming completion for ${name}`)
     return waitForStreamingCompletion(contentString, sseStreamer, chatId, name, notifications)
 }
 
@@ -286,7 +264,6 @@ function waitForStreamingCompletion(
             completed = true
 
             const fullResponse = buildFullResponse(contentString, notifications)
-            console.log(`${reason} [MCP Tool] ${name} completed, returning full result`)
             resolve(fullResponse)
         }
 
@@ -342,15 +319,12 @@ function setupNotificationHandlers(
 
         // Detect completion based on tool's annotation signals
         if (completionSignals.includes(logger)) {
-            console.log(`🎯 [MCP Tool] Completion signal detected: ${logger} for ${toolName}`)
-
             // Add visual separation before LLM response
             sseStreamer.streamTokenEvent(chatId, '\n\n')
 
             // Trigger cleanup after brief delay
             setTimeout(() => {
                 sseStreamer.removeMcpConnection(chatId, toolName)
-                console.log(`🔧 [MCP Tool] Auto-cleanup completed for ${toolName}`)
             }, MCP_STREAMING_CONFIG.NOTIFICATION_DELAY)
         }
     })

packages/server/src/utils/SSEStreamer.ts

@@ -38,24 +38,20 @@ export class SSEStreamer implements IServerSideEventStreamer {
             startTime: Date.now(),
             status: 'active'
         })
-        console.log(`[SSEStreamer] Added MCP connection: ${connectionId}`)
     }
 
     markMcpConnectionCompleting(chatId: string, toolName?: string) {
         const connectionId = `${chatId}:${toolName || 'unknown'}`
         const connection = this.activeMcpConnections.get(connectionId)
         if (connection) {
             connection.status = 'completing'
-            console.log(`[SSEStreamer] Marking MCP connection as completing: ${connectionId}`)
         }
     }
 
     removeMcpConnection(chatId: string, toolName?: string) {
         const connectionId = `${chatId}:${toolName || 'unknown'}`
         this.activeMcpConnections.delete(connectionId)
-        console.log(`[SSEStreamer] Removed MCP connection: ${connectionId}`)
 
-        // Check if any active connections remain for this chatId
         const hasActiveConnections = Array.from(this.activeMcpConnections.keys()).some((key) => key.startsWith(`${chatId}:`))
 
         if (!hasActiveConnections && this.clients[chatId]?.pendingRemoval) {
@@ -72,8 +68,6 @@ export class SSEStreamer implements IServerSideEventStreamer {
         if (client) {
             // Check if there are active MCP connections for this chatId
             if (this.hasMcpConnections(chatId)) {
-                console.log(`[SSEStreamer] Delaying client removal for chatId: ${chatId} due to active MCP connections`)
-                // Mark client for pending removal but don't close it yet
                 client.pendingRemoval = true
                 return
             }
@@ -85,7 +79,6 @@ export class SSEStreamer implements IServerSideEventStreamer {
     forceRemoveClient(chatId: string) {
         const client = this.clients[chatId]
         if (client) {
-            console.log(`[SSEStreamer] Force removing client for chatId: ${chatId}`)
             const clientResponse = {
                 event: 'end',
                 data: '[DONE]'