chuk-tool-processor 0.6.5__py3-none-any.whl → 0.6.6__py3-none-any.whl

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.dist-info → chuk_tool_processor-0.6.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chuk-tool-processor
-Version: 0.6.5
+Version: 0.6.6
 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.dist-info → chuk_tool_processor-0.6.6.dist-info}/RECORD

@@ -26,7 +26,7 @@ chuk_tool_processor/mcp/stream_manager.py,sha256=DN58d76J3Xkg9DI3f6EFe9qAG-ZsFa4
 chuk_tool_processor/mcp/transport/__init__.py,sha256=0DX7m_VvlXPxijc-88_QTLhq4ZqAgUgzBjSMGL9C_lM,963
 chuk_tool_processor/mcp/transport/base_transport.py,sha256=bqId34OMQMxzMXtrKq_86sot0_x0NS_ecaIllsCyy6I,3423
 chuk_tool_processor/mcp/transport/http_streamable_transport.py,sha256=3I3tNYU8r4YqCbNhMCkoucvZc6VS2ulzeUjDe2FbcRk,19108
-chuk_tool_processor/mcp/transport/sse_transport.py,sha256=smsBrKm-U-nOwsWLz_jbAF7MkmQYx9jkZLWyEw3JFRE,15594
+chuk_tool_processor/mcp/transport/sse_transport.py,sha256=s8an1sHLFsIJwqxUcdSvY6bYBkdyNQ-LkQMqzD5okdI,17253
 chuk_tool_processor/mcp/transport/stdio_transport.py,sha256=DPXLR_OxuCJ2bgwYDuT_iYC_CDcUIySvLNO7JzyiPyc,9099
 chuk_tool_processor/models/__init__.py,sha256=TC__rdVa0lQsmJHM_hbLDPRgToa_pQT_UxRcPZk6iVw,40
 chuk_tool_processor/models/execution_strategy.py,sha256=UVW35YIeMY2B3mpIKZD2rAkyOPayI6ckOOUALyf0YiQ,2115
@@ -54,7 +54,7 @@ chuk_tool_processor/registry/providers/__init__.py,sha256=eigwG_So11j7WbDGSWaKd3
 chuk_tool_processor/registry/providers/memory.py,sha256=6cMtUwLO6zrk3pguQRgxJ2CReHAzewgZsizWZhsoStk,5184
 chuk_tool_processor/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chuk_tool_processor/utils/validation.py,sha256=V5N1dH9sJlHepFIbiI2k2MU82o7nvnh0hKyIt2jdgww,4136
-chuk_tool_processor-0.6.5.dist-info/METADATA,sha256=zwmOSBOSAbk0A8oTfY34RokEr7VjtvUaEfWpGmuXSyg,23463
-chuk_tool_processor-0.6.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-chuk_tool_processor-0.6.5.dist-info/top_level.txt,sha256=7lTsnuRx4cOW4U2sNJWNxl4ZTt_J1ndkjTbj3pHPY5M,20
-chuk_tool_processor-0.6.5.dist-info/RECORD,,
+chuk_tool_processor-0.6.6.dist-info/METADATA,sha256=xjCU03q8dyRa8VbmtEO5ZjLiJbWzid6ccy5TItWokSo,23463
+chuk_tool_processor-0.6.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+chuk_tool_processor-0.6.6.dist-info/top_level.txt,sha256=7lTsnuRx4cOW4U2sNJWNxl4ZTt_J1ndkjTbj3pHPY5M,20
+chuk_tool_processor-0.6.6.dist-info/RECORD,,