improve stdio server logging

Author: SecretiveShell

mcp_bridge/mcp_clients/AbstractClient.py

@@ -29,6 +29,8 @@ def __init__(self, name: str) -> None:
         self.session = None
         self.name = name
 
+        logger.debug(f"initializing client class for {name}")
+
     @abstractmethod
     async def _maintain_session(self):
         pass
@@ -39,7 +41,9 @@ async def _session_maintainer(self):
                 await self._maintain_session()
             except Exception as e:
                 logger.trace(f"failed to maintain session for {self.name}: {e}")
-                await asyncio.sleep(0.5)
+
+            logger.debug(f"restarting session for {self.name}")
+            await asyncio.sleep(0.5)
 
     async def start(self):
         asyncio.create_task(self._session_maintainer())
@@ -122,7 +126,7 @@ async def list_prompts(self) -> ListPromptsResult:
             logger.error(f"error listing prompts: {e}")
             return ListPromptsResult(prompts=[])
 
-    async def _wait_for_session(self, timeout: int = 10, http_error: bool = True):
+    async def _wait_for_session(self, timeout: int = 5, http_error: bool = True):
         try:
             async with asyncio.timeout(timeout):
                 while self.session is None:
@@ -132,10 +136,10 @@ async def _wait_for_session(self, timeout: int = 10, http_error: bool = True):
         except asyncio.TimeoutError:
             if http_error:
                 raise HTTPException(
-                    status_code=500, detail="Could not connect to MCP server."
+                    status_code=500, detail=f"Could not connect to MCP server \"{self.name}\"." 
                 )
 
-            raise TimeoutError("Session initialization timed out.")
+            raise TimeoutError(f"Could not connect to MCP server \"{self.name}\"." )
 
         assert self.session is not None, "Session is None"
 

mcp_bridge/mcp_clients/StdioClient.py

@@ -24,7 +24,7 @@ def __init__(self, name: str, config: StdioServerParameters) -> None:
             if not any(keyword in key for keyword in venv_keywords)
         }
 
-        logger.debug(f"env: {env}")
+        # logger.debug(f"env: {env}")
 
         if config.env is not None:
             env.update(config.env)
@@ -43,8 +43,13 @@ def __init__(self, name: str, config: StdioServerParameters) -> None:
         self.config = own_config
 
     async def _maintain_session(self):
+        logger.debug(f"starting maintain session for {self.name}")
         async with stdio_client(self.config) as client:
+            logger.debug(f"entered stdio_client context manager for {self.name}")
+            assert client[0] is not None, f"missing read stream for {self.name}"
+            assert client[1] is not None, f"missing write stream for {self.name}"
             async with ClientSession(*client) as session:
+                logger.debug(f"entered client session context manager for {self.name}")
                 await session.initialize()
                 logger.debug(f"finished initialise session for {self.name}")
                 self.session = session