chuk-tool-processor 0.6.5__tar.gz → 0.6.7__tar.gz

{chuk_tool_processor-0.6.5 → chuk_tool_processor-0.6.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chuk-tool-processor
-Version: 0.6.5
+Version: 0.6.7
 Summary: Async-native framework for registering, discovering, and executing tools referenced in LLM responses
 Author-email: CHUK Team <chrishayuk@somejunkmailbox.com>
 Maintainer-email: CHUK Team <chrishayuk@somejunkmailbox.com>

{chuk_tool_processor-0.6.5 → chuk_tool_processor-0.6.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "chuk-tool-processor"
-version = "0.6.5"
+version = "0.6.7"
 description = "Async-native framework for registering, discovering, and executing tools referenced in LLM responses"
 readme = "README.md"
 requires-python = ">=3.11"

{chuk_tool_processor-0.6.5 → chuk_tool_processor-0.6.7}/src/chuk_tool_processor/mcp/stream_manager.py

@@ -1,6 +1,6 @@
 # chuk_tool_processor/mcp/stream_manager.py
 """
-StreamManager for CHUK Tool Processor - Enhanced with robust shutdown handling
+StreamManager for CHUK Tool Processor - Enhanced with robust shutdown handling and headers support
 """
 from __future__ import annotations
 
@@ -27,12 +27,12 @@ class StreamManager:
     """
     Manager for MCP server streams with support for multiple transport types.
     
-    Enhanced with robust shutdown handling to prevent event loop closure issues.
+    Enhanced with robust shutdown handling and proper headers support.
     
     Updated to support the latest transports:
     - STDIO (process-based)
-    - SSE (Server-Sent Events) 
-    - HTTP Streamable (modern replacement for SSE, spec 2025-03-26)
+    - SSE (Server-Sent Events) with headers support
+    - HTTP Streamable (modern replacement for SSE, spec 2025-03-26) with headers support
     """
 
     def __init__(self) -> None:
@@ -193,14 +193,17 @@ class StreamManager:
                         if isinstance(params, dict) and 'url' in params:
                             sse_url = params['url']
                             api_key = params.get('api_key')
+                            headers = params.get('headers', {})
                         else:
                             sse_url = "http://localhost:8000"
                             api_key = None
+                            headers = {}
                             logger.warning("No URL configured for SSE transport, using default: %s", sse_url)
                         
                         transport = SSETransport(
                             sse_url,
                             api_key,
+                            headers=headers,
                             default_timeout=default_timeout
                         )
                     elif transport_type == "http_streamable":
@@ -210,16 +213,19 @@ class StreamManager:
                         if isinstance(params, dict) and 'url' in params:
                             http_url = params['url']
                             api_key = params.get('api_key')
+                            headers = params.get('headers', {})
                             session_id = params.get('session_id')
                         else:
                             http_url = "http://localhost:8000"
                             api_key = None
+                            headers = {}
                             session_id = None
                             logger.warning("No URL configured for HTTP Streamable transport, using default: %s", http_url)
                         
                         transport = HTTPStreamableTransport(
                             http_url,
                             api_key,
+                            headers=headers,
                             default_timeout=default_timeout,
                             session_id=session_id
                         )
@@ -271,6 +277,7 @@ class StreamManager:
         connection_timeout: float = 10.0,
         default_timeout: float = 30.0,
     ) -> None:
+        """Initialize with SSE transport with headers support."""
         if self._closed:
             raise RuntimeError("Cannot initialize a closed StreamManager")
             
@@ -283,9 +290,15 @@ class StreamManager:
                     logger.error("Bad server config: %s", cfg)
                     continue
                 try:
+                    # FIXED: Extract and pass headers from server config
+                    headers = cfg.get("headers", {})
+                    if headers:
+                        logger.debug("SSE %s: Using configured headers: %s", name, list(headers.keys()))
+                    
                     transport = SSETransport(
                         url, 
                         cfg.get("api_key"),
+                        headers=headers,  # ← FIXED: Pass headers
                         connection_timeout=connection_timeout,
                         default_timeout=default_timeout
                     )
@@ -326,7 +339,7 @@ class StreamManager:
         connection_timeout: float = 30.0,
         default_timeout: float = 30.0,
     ) -> None:
-        """Initialize with HTTP Streamable transport (modern MCP spec 2025-03-26)."""
+        """Initialize with HTTP Streamable transport with headers support."""
         if self._closed:
             raise RuntimeError("Cannot initialize a closed StreamManager")
             
@@ -339,9 +352,15 @@ class StreamManager:
                     logger.error("Bad server config: %s", cfg)
                     continue
                 try:
+                    # FIXED: Extract and pass headers from server config
+                    headers = cfg.get("headers", {})
+                    if headers:
+                        logger.debug("HTTP Streamable %s: Using configured headers: %s", name, list(headers.keys()))
+                    
                     transport = HTTPStreamableTransport(
                         url, 
                         cfg.get("api_key"),
+                        headers=headers,  # ← FIXED: Pass headers (requires HTTPStreamableTransport to be updated)
                         connection_timeout=connection_timeout,
                         default_timeout=default_timeout,
                         session_id=cfg.get("session_id")

{chuk_tool_processor-0.6.5 → chuk_tool_processor-0.6.7}/src/chuk_tool_processor/mcp/transport/sse_transport.py

@@ -1,7 +1,7 @@
 # chuk_tool_processor/mcp/transport/sse_transport.py
 """
 Fixed SSE transport that matches your server's actual behavior.
-Based on your working debug script.
+Based on your working debug script with smart URL detection.
 """
 from __future__ import annotations
 
@@ -26,13 +26,21 @@ class SSETransport(MCPBaseTransport):
     """
 
     def __init__(self, url: str, api_key: Optional[str] = None, 
+                 headers: Optional[Dict[str, str]] = None,
                  connection_timeout: float = 30.0, default_timeout: float = 30.0):
         """Initialize SSE transport."""
         self.url = url.rstrip('/')
         self.api_key = api_key
+        self.configured_headers = headers or {}
         self.connection_timeout = connection_timeout
         self.default_timeout = default_timeout
         
+        # DEBUG: Log what we received
+        logger.debug("SSE Transport initialized with:")
+        logger.debug("  URL: %s", self.url)
+        logger.debug("  API Key: %s", "***" if api_key else None)
+        logger.debug("  Headers: %s", {k: v[:10] + "..." if len(v) > 10 else v for k, v in self.configured_headers.items()})
+        
         # State
         self.session_id = None
         self.message_url = None
@@ -48,11 +56,41 @@ class SSETransport(MCPBaseTransport):
         self.sse_response = None
         self.sse_stream_context = None
 
+    def _construct_sse_url(self, base_url: str) -> str:
+        """
+        Construct the SSE endpoint URL from the base URL.
+        
+        Smart detection to avoid double-appending /sse if already present.
+        """
+        # Remove trailing slashes
+        base_url = base_url.rstrip('/')
+        
+        # Check if URL already ends with /sse
+        if base_url.endswith('/sse'):
+            # Already has /sse, use as-is
+            logger.debug("URL already contains /sse endpoint: %s", base_url)
+            return base_url
+        
+        # Append /sse to the base URL
+        sse_url = f"{base_url}/sse"
+        logger.debug("Appending /sse to base URL: %s -> %s", base_url, sse_url)
+        return sse_url
+
     def _get_headers(self) -> Dict[str, str]:
         """Get headers with auth if available."""
         headers = {}
+        
+        # Add configured headers first
+        if self.configured_headers:
+            headers.update(self.configured_headers)
+        
+        # Add API key as Bearer token if provided (this will override any Authorization header)
         if self.api_key:
             headers['Authorization'] = f'Bearer {self.api_key}'
+        
+        # DEBUG: Log what headers we're sending
+        logger.debug("Sending headers: %s", {k: v[:10] + "..." if len(v) > 10 else v for k, v in headers.items()})
+        
         return headers
 
     async def initialize(self) -> bool:
@@ -68,8 +106,8 @@ class SSETransport(MCPBaseTransport):
             self.stream_client = httpx.AsyncClient(timeout=self.connection_timeout)
             self.send_client = httpx.AsyncClient(timeout=self.default_timeout)
             
-            # Connect to SSE stream
-            sse_url = f"{self.url}/sse"
+            # Connect to SSE stream with smart URL construction
+            sse_url = self._construct_sse_url(self.url)
             logger.debug("Connecting to SSE: %s", sse_url)
             
             self.sse_stream_context = self.stream_client.stream(

{chuk_tool_processor-0.6.5 → chuk_tool_processor-0.6.7}/src/chuk_tool_processor.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chuk-tool-processor
-Version: 0.6.5
+Version: 0.6.7
 Summary: Async-native framework for registering, discovering, and executing tools referenced in LLM responses
 Author-email: CHUK Team <chrishayuk@somejunkmailbox.com>
 Maintainer-email: CHUK Team <chrishayuk@somejunkmailbox.com>