castrel-proxy 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

castrel_proxy/mcp/manager.py

@@ -69,18 +69,22 @@ def convert_config_to_langchain_format(config_data: dict) -> dict:
                 "url": server_config.get("url"),
             }
 
-        elif transport in ["sse", "websocket"]:
-            error_msg = (
-                f"Configuration error for server '{name}': Transport type '{transport}' is not yet supported. "
-                f"Currently supported transports: 'stdio', 'http'"
-            )
-            logger.error(error_msg)
-            raise ValueError(error_msg)
+        elif transport == "sse":
+            # sse type: use url
+            if not server_config.get("url"):
+                error_msg = f"Configuration error for server '{name}': Missing 'url' for sse transport"
+                logger.error(error_msg)
+                raise ValueError(error_msg)
+
+            langchain_config[name] = {
+                "transport": "sse",
+                "url": server_config.get("url"),
+            }
 
         else:
             error_msg = (
                 f"Configuration error for server '{name}': Unknown transport type '{transport}'. "
-                f"Supported transports: 'stdio', 'http'"
+                f"Supported transports: 'stdio', 'http', 'sse'"
             )
             logger.error(error_msg)
             raise ValueError(error_msg)
@@ -209,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 = []
@@ -236,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"""

{castrel_proxy-0.1.0.dist-info → castrel_proxy-0.1.2.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: castrel-proxy
-Version: 0.1.0
+Version: 0.1.2
 Summary: A lightweight remote command execution bridge client with MCP integration
-Project-URL: Homepage, https://github.com/castrel-ai/castrel-bridge-proxy
-Project-URL: Documentation, https://github.com/castrel-ai/castrel-bridge-proxy#readme
-Project-URL: Repository, https://github.com/castrel-ai/castrel-bridge-proxy
-Project-URL: Issues, https://github.com/castrel-ai/castrel-bridge-proxy/issues
-Project-URL: Changelog, https://github.com/castrel-ai/castrel-bridge-proxy/blob/main/CHANGELOG.md
+Project-URL: Homepage, https://github.com/castrel-ai/castrel-proxy
+Project-URL: Documentation, https://github.com/castrel-ai/castrel-proxy#readme
+Project-URL: Repository, https://github.com/castrel-ai/castrel-proxy
+Project-URL: Issues, https://github.com/castrel-ai/castrel-proxy/issues
+Project-URL: Changelog, https://github.com/castrel-ai/castrel-proxy/blob/main/CHANGELOG.md
 Author: Castrel Team
 License: MIT License
         
@@ -58,9 +58,9 @@ Requires-Dist: pytest-cov>=4.0.0; extra == 'dev'
 Requires-Dist: pytest>=7.0.0; extra == 'dev'
 Description-Content-Type: text/markdown
 
-# Castrel Bridge Proxy
+# Castrel Proxy
 
-[![CI](https://github.com/castrel-ai/castrel-bridge-proxy/workflows/CI/badge.svg)](https://github.com/castrel-ai/castrel-bridge-proxy/actions)
+[![CI](https://github.com/castrel-ai/castrel-proxy/workflows/CI/badge.svg)](https://github.com/castrel-ai/castrel-proxy/actions)
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
 [![Python Version](https://img.shields.io/pypi/pyversions/castrel-proxy)](https://pypi.org/project/castrel-proxy/)
 
@@ -89,8 +89,8 @@ pip install castrel-proxy
 ### From source
 
 ```bash
-git clone https://github.com/castrel-ai/castrel-bridge-proxy.git
-cd castrel-bridge-proxy
+git clone https://github.com/castrel-ai/castrel-proxy.git
+cd castrel-proxy
 pip install -e .
 ```
 
@@ -241,15 +241,11 @@ python
 
 ```bash
 # Clone repository
-git clone https://github.com/castrel-ai/castrel-bridge-proxy.git
-cd castrel-bridge-proxy
+git clone https://github.com/castrel-ai/castrel-proxy.git
+cd castrel-proxy
 
-# Create virtual environment
-python -m venv venv
-source venv/bin/activate  # On Windows: venv\Scripts\activate
-
-# Install dependencies
-pip install -e ".[dev]"
+# Install dependencies and sync environment
+uv sync
 ```
 
 ### Run Tests
@@ -292,8 +288,8 @@ For security concerns, please see [SECURITY.md](SECURITY.md) or contact security
 
 ## 📮 Contact
 
-- Issues: [GitHub Issues](https://github.com/castrel-ai/castrel-bridge-proxy/issues)
-- Discussions: [GitHub Discussions](https://github.com/castrel-ai/castrel-bridge-proxy/discussions)
+- Issues: [GitHub Issues](https://github.com/castrel-ai/castrel-proxy/issues)
+- Discussions: [GitHub Discussions](https://github.com/castrel-ai/castrel-proxy/discussions)
 
 ## 🙏 Acknowledgments
 

{castrel_proxy-0.1.0.dist-info → castrel_proxy-0.1.2.dist-info}/RECORD

@@ -9,7 +9,7 @@ castrel_proxy/core/executor.py,sha256=Z1TBW6rRJMDuls0dF0_W32OuOPaZNmlB4DpCC55IJJ
 castrel_proxy/data/__init__.py,sha256=NTWvSsQ2vYDYvJV4Xng2q2q6Qyj1Z4VikLeQPlOre4o,42
 castrel_proxy/data/default_whitelist.txt,sha256=tkjvX6Q-kXzy8bgkPssWzBbakDJvUZL7BQbIDDj7XbM,2135
 castrel_proxy/mcp/__init__.py,sha256=aPUkCh4im-MlU3byXn_nXp9k85MUY1wfFLvDwH7eDjM,161
-castrel_proxy/mcp/manager.py,sha256=BTO_DA-xLEAhuL795t1bBb-eSX1Jv9v32Pp_px0jDK8,9445
+castrel_proxy/mcp/manager.py,sha256=epxT4zDMGmgzPzV7F3swM6ExMFlbx_fz7OIqsmgXZsY,9506
 castrel_proxy/network/__init__.py,sha256=C7PPMqB44X8yMSNMv8vuILPI26xCfLf5levuzIrjdXk,329
 castrel_proxy/network/api_client.py,sha256=eMpRTlnvphakTgufqxmLFGr5PLEPTR9FtZ1oBrZrQ68,9898
 castrel_proxy/network/websocket_client.py,sha256=-PJ25FVDcEaWod7pJjSQzjoTQOygVcw5buO8g8xUyKo,43470
@@ -17,8 +17,8 @@ castrel_proxy/operations/__init__.py,sha256=NGv4Z-8Fek9KiDv630CmekWvbC2n-BKlc2yq
 castrel_proxy/operations/document.py,sha256=UnybzI3FCOVteMHkd88JaR6WvU2YS1e9Rf45wTttGss,10673
 castrel_proxy/security/__init__.py,sha256=aC0CcnM4mfuuXfYJniMbn47JN1eQUXU-k3ByqfWuK4s,352
 castrel_proxy/security/whitelist.py,sha256=nU5_ovpGZOgZfhMofoxt9ImxJR8boYdk937Tz-K34C4,11493
-castrel_proxy-0.1.0.dist-info/METADATA,sha256=NW8c2nvNawhElDsYCG9GP1aP9LcvOdjol55BwFMtOaA,10246
-castrel_proxy-0.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-castrel_proxy-0.1.0.dist-info/entry_points.txt,sha256=5X5vHpAFgIFpvt2CmI5U0d_naKA64eE-IZKEqvnS7VE,65
-castrel_proxy-0.1.0.dist-info/licenses/LICENSE,sha256=cqP1Kg_ECZMt4RPE8h5uL4-qpEaYTgca7r4aHtMIDGk,1066
-castrel_proxy-0.1.0.dist-info/RECORD,,
+castrel_proxy-0.1.2.dist-info/METADATA,sha256=RtmzYM-E2d3CeDerhRqjgnSRbPyzc1jRr4Oj7EGaxWI,10041
+castrel_proxy-0.1.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+castrel_proxy-0.1.2.dist-info/entry_points.txt,sha256=5X5vHpAFgIFpvt2CmI5U0d_naKA64eE-IZKEqvnS7VE,65
+castrel_proxy-0.1.2.dist-info/licenses/LICENSE,sha256=cqP1Kg_ECZMt4RPE8h5uL4-qpEaYTgca7r4aHtMIDGk,1066
+castrel_proxy-0.1.2.dist-info/RECORD,,