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

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

@@ -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 → castrel_proxy-0.1.2}/README.md

@@ -1,6 +1,6 @@
-# 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/)
 
@@ -29,8 +29,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 .
 ```
 
@@ -181,15 +181,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
@@ -232,8 +228,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 → castrel_proxy-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "castrel-proxy"
-version = "0.1.0"
+version = "0.1.2"
 description = "A lightweight remote command execution bridge client with MCP integration"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -42,11 +42,11 @@ dev = [
 ]
 
 [project.urls]
-Homepage = "https://github.com/castrel-ai/castrel-bridge-proxy"
-Documentation = "https://github.com/castrel-ai/castrel-bridge-proxy#readme"
-Repository = "https://github.com/castrel-ai/castrel-bridge-proxy"
-Issues = "https://github.com/castrel-ai/castrel-bridge-proxy/issues"
-Changelog = "https://github.com/castrel-ai/castrel-bridge-proxy/blob/main/CHANGELOG.md"
+Homepage = "https://github.com/castrel-ai/castrel-proxy"
+Documentation = "https://github.com/castrel-ai/castrel-proxy#readme"
+Repository = "https://github.com/castrel-ai/castrel-proxy"
+Issues = "https://github.com/castrel-ai/castrel-proxy/issues"
+Changelog = "https://github.com/castrel-ai/castrel-proxy/blob/main/CHANGELOG.md"
 
 [project.scripts]
 castrel-proxy = "castrel_proxy.cli.commands:run"

{castrel_proxy-0.1.0 → castrel_proxy-0.1.2}/src/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"""