castrel-proxy 0.1.1__tar.gz → 0.1.2__tar.gz

{castrel_proxy-0.1.1 → castrel_proxy-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: castrel-proxy
-Version: 0.1.1
+Version: 0.1.2
 Summary: A lightweight remote command execution bridge client with MCP integration
 Project-URL: Homepage, https://github.com/castrel-ai/castrel-proxy
 Project-URL: Documentation, https://github.com/castrel-ai/castrel-proxy#readme

{castrel_proxy-0.1.1 → castrel_proxy-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "castrel-proxy"
-version = "0.1.1"
+version = "0.1.2"
 description = "A lightweight remote command execution bridge client with MCP integration"
 readme = "README.md"
 requires-python = ">=3.10"

{castrel_proxy-0.1.1 → castrel_proxy-0.1.2}/src/castrel_proxy/mcp/manager.py

@@ -213,21 +213,18 @@ class MCPManager:
         Get tools from all MCP services
 
         Returns:
-            Dict[str, List[Dict]]: All tools list
-
-        Raises:
-            SystemExit: When MCP client is not initialized or tools retrieval fails
+            Dict[str, List[Dict]]: All tools list (may be partial if some servers fail)
         """
         if not self.client:
             logger.error("MCP client not initialized")
-            logger.error("Exiting due to uninitialized MCP client")
-            sys.exit(1)
+            return {}
 
-        try:
-            # Use MultiServerMCPClient to get all tools
-            result = {}
-            count = 0
-            for server_name in self.server_configs:
+        result = {}
+        total_count = 0
+        failed_servers = []
+
+        for server_name in self.server_configs:
+            try:
                 tools = await self.client.get_tools(server_name=server_name)
                 # Convert to required format
                 formatted_tools = []
@@ -240,27 +237,29 @@ class MCPManager:
                         "mcp_server": getattr(tool, "server_name", "unknown"),
                     }
                     formatted_tools.append(tool_info)
-                count = count + len(formatted_tools)
+                total_count += len(formatted_tools)
                 result[server_name] = formatted_tools
+                logger.info(f"Retrieved {len(formatted_tools)} tool(s) from '{server_name}'")
 
-            if count == 0:
-                logger.warning("No tools retrieved from any MCP server")
-                logger.warning("Exiting due to zero tools retrieved")
-                sys.exit(1)
+            except Exception as e:
+                failed_servers.append(server_name)
+                # Log error but continue with other servers
+                if "Configuration error" in str(e) or "Missing 'transport' key" in str(e):
+                    logger.error(f"Configuration error for server '{server_name}': {e}")
+                else:
+                    logger.error(f"Failed to get tools from server '{server_name}': {e}")
+
+        if failed_servers:
+            logger.warning(
+                f"Failed to retrieve tools from {len(failed_servers)} server(s): {', '.join(failed_servers)}"
+            )
 
-            logger.info(f"Retrieved {count} tool(s)")
-            return result
+        if total_count == 0:
+            logger.warning("No tools retrieved from any MCP server")
+        else:
+            logger.info(f"Retrieved {total_count} tool(s) from {len(result)} server(s)")
 
-        except Exception as e:
-            # Check if it's a configuration error
-            if "Configuration error" in str(e) or "Missing 'transport' key" in str(e):
-                logger.error(f"Configuration error while getting MCP tools: {e}")
-                logger.error("Exiting due to invalid MCP configuration")
-                sys.exit(1)
-            else:
-                logger.error(f"Failed to get MCP tools: {e}")
-                logger.error("Exiting due to MCP tools retrieval failure")
-                sys.exit(1)
+        return result
 
     async def disconnect_all(self):
         """Disconnect all MCP connections"""