feat(mcp): Routing mcp log to file

Author: daltonnyx

AgentCrew/modules/mcpclient/service.py

@@ -1,5 +1,5 @@
 from AgentCrew.modules import logger
-from typing import Dict, Any, List, Optional, Callable
+from typing import Dict, Any, List, Optional, Callable, TextIO, AnyStr
 from mcp import ClientSession, StdioServerParameters
 from mcp.types import Prompt, ContentBlock, TextContent, ImageContent
 from mcp.client.stdio import stdio_client
@@ -8,6 +8,46 @@
 from .config import MCPServerConfig
 import asyncio
 import threading
+import tempfile
+from datetime import datetime
+
+
+class MCPLogIO(TextIO):
+    """File-like object compatible with sys.stderr for MCP logging."""
+
+    def __init__(self):
+        self.log_path = (
+            tempfile.gettempdir() + f"/mcp_agentcrew_{datetime.now().timestamp()}.log"
+        )
+        print(f"Routing MCP logs to {self.log_path}")
+        self.file = open(self.log_path, "w+")
+
+    def write(self, data: AnyStr) -> int:
+        """Write data to the log file."""
+        if isinstance(data, bytes):
+            # Convert bytes to string for writing
+            str_data = data.decode("utf-8", errors="replace")
+        else:
+            str_data = str(data)
+        self.file.write(str_data)
+        self.file.flush()  # Ensure data is written immediately
+        return 0
+
+    def flush(self):
+        """Flush the file buffer."""
+        self.file.flush()
+
+    def close(self):
+        """Close the file."""
+        self.file.close()
+
+    def fileno(self):
+        """Return the file descriptor."""
+        return self.file.fileno()
+
+
+# Initialize the logger
+mcp_log_io = MCPLogIO()
 
 
 class MCPService:
@@ -69,7 +109,7 @@ async def _manage_single_connection(self, server_config: MCPServerConfig):
                             self.server_prompts[server_id] = prompts.prompts
 
                         except Exception as e:
-                            print(f"{str(e)}")
+                            logger.warning(f"{str(e)}")
 
                         logger.info(
                             f"MCPService: {server_id} setup complete. Waiting for shutdown signal."
@@ -83,9 +123,12 @@ async def _manage_single_connection(self, server_config: MCPServerConfig):
                     env=server_config.env,
                 )
 
-                async with stdio_client(server_params) as (stdio, write_stream):
+                async with stdio_client(server_params, errlog=mcp_log_io) as (
+                    read_stream,
+                    write_stream,
+                ):
                     logger.info(f"MCPService: stdio_client established for {server_id}")
-                    async with ClientSession(stdio, write_stream) as session:
+                    async with ClientSession(read_stream, write_stream) as session:
                         logger.info(
                             f"MCPService: ClientSession established for {server_id}"
                         )
@@ -106,7 +149,7 @@ async def _manage_single_connection(self, server_config: MCPServerConfig):
                             self.server_prompts[server_id] = prompts.prompts
 
                         except Exception as e:
-                            print(f"{str(e)}")
+                            logger.warning(f"{str(e)}")
 
                         logger.info(
                             f"MCPService: {server_id} setup complete. Waiting for shutdown signal."
@@ -421,7 +464,6 @@ async def get_prompt(
             Prompt object if found, None otherwise
         """
 
-        print(server_id, prompt_name)
         if server_id not in self.sessions or not self.connected_servers.get(server_id):
             return {
                 "content": f"Cannot call tool: Server '{server_id}' is not connected",