chuk-tool-processor 0.6.9__tar.gz → 0.6.10__tar.gz

{chuk_tool_processor-0.6.9 → chuk_tool_processor-0.6.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chuk-tool-processor
-Version: 0.6.9
+Version: 0.6.10
 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.9 → chuk_tool_processor-0.6.10}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "chuk-tool-processor"
-version = "0.6.9"
+version = "0.6.10"
 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.9 → chuk_tool_processor-0.6.10}/src/chuk_tool_processor/mcp/transport/sse_transport.py

@@ -8,6 +8,9 @@ Implements Server-Sent Events transport with two-step async pattern:
 
 Note: This transport is deprecated in favor of HTTP Streamable (spec 2025-03-26)
 but remains supported for backward compatibility.
+
+FIXED: Updated to support both old format (/messages/) and new event-based format
+(event: endpoint + data: https://...) for session discovery.
 """
 from __future__ import annotations
 
@@ -32,6 +35,8 @@ class SSETransport(MCPBaseTransport):
     This transport uses a dual-connection approach:
     - SSE stream for receiving responses
     - HTTP POST for sending requests
+    
+    FIXED: Supports both old and new session discovery formats.
     """
 
     def __init__(self, url: str, api_key: Optional[str] = None, 
@@ -215,33 +220,61 @@ class SSETransport(MCPBaseTransport):
             return False
 
     async def _process_sse_stream(self):
-        """Process the persistent SSE stream for responses and session discovery."""
+        """
+        Process the persistent SSE stream for responses and session discovery.
+        
+        FIXED: Supports both old format (/messages/) and new event-based format
+        (event: endpoint + data: https://...) for session discovery.
+        """
         try:
             logger.debug("Starting SSE stream processing...")
             
+            current_event = None  # Track current event type
+            
             async for line in self.sse_response.aiter_lines():
                 line = line.strip()
                 if not line:
                     continue
                 
-                # Handle session endpoint discovery
-                if not self.message_url and line.startswith('data:') and '/messages/' in line:
-                    endpoint_path = line.split(':', 1)[1].strip()
-                    self.message_url = f"{self.url}{endpoint_path}"
+                # Handle event type declarations
+                if line.startswith('event:'):
+                    current_event = line.split(':', 1)[1].strip()
+                    logger.debug("SSE event type: %s", current_event)
+                    continue
+                
+                # Handle session endpoint discovery (BOTH FORMATS)
+                if not self.message_url and line.startswith('data:'):
+                    data_part = line.split(':', 1)[1].strip()
                     
-                    # Extract session ID if present
-                    if 'session_id=' in endpoint_path:
-                        self.session_id = endpoint_path.split('session_id=')[1].split('&')[0]
+                    # NEW FORMAT: event: endpoint + data: https://...
+                    if current_event == "endpoint" and data_part.startswith('http'):
+                        self.message_url = data_part
+                        
+                        # Extract session ID from URL if present
+                        if 'session_id=' in data_part:
+                            self.session_id = data_part.split('session_id=')[1].split('&')[0]
+                        
+                        logger.debug("Session endpoint discovered via event format: %s", self.session_id)
+                        continue
                     
-                    logger.debug("Session endpoint discovered: %s", self.session_id)
-                    continue
+                    # OLD FORMAT: data: /messages/... (backwards compatibility)
+                    elif '/messages/' in data_part:
+                        endpoint_path = data_part
+                        self.message_url = f"{self.url}{endpoint_path}"
+                        
+                        # Extract session ID if present
+                        if 'session_id=' in endpoint_path:
+                            self.session_id = endpoint_path.split('session_id=')[1].split('&')[0]
+                        
+                        logger.debug("Session endpoint discovered via old format: %s", self.session_id)
+                        continue
                 
                 # Handle JSON-RPC responses
                 if line.startswith('data:'):
                     data_part = line.split(':', 1)[1].strip()
                     
                     # Skip keepalive pings and empty data
-                    if not data_part or data_part.startswith('ping'):
+                    if not data_part or data_part.startswith('ping') or data_part in ('{}', '[]'):
                         continue
                     
                     try:
@@ -260,6 +293,10 @@ class SSETransport(MCPBaseTransport):
                     
                     except json.JSONDecodeError as e:
                         logger.debug("Non-JSON data in SSE stream (ignoring): %s", e)
+                
+                # Reset event type after processing data (only if we processed JSON-RPC)
+                if line.startswith('data:') and current_event not in ("endpoint",):
+                    current_event = None
                         
         except Exception as e:
             if self.enable_metrics:
@@ -403,7 +440,7 @@ class SSETransport(MCPBaseTransport):
 
         start_time = time.time()
         if self.enable_metrics:
-            self._metrics["total_calls"] += 1  # FIXED: INCREMENT FIRST
+            self._metrics["total_calls"] += 1
 
         try:
             logger.debug("Calling tool '%s' with arguments: %s", tool_name, arguments)
@@ -461,7 +498,6 @@ class SSETransport(MCPBaseTransport):
             self._metrics["failed_calls"] += 1
             
         self._metrics["total_time"] += response_time
-        # FIXED: Only calculate average if we have total calls
         if self._metrics["total_calls"] > 0:
             self._metrics["avg_response_time"] = (
                 self._metrics["total_time"] / self._metrics["total_calls"]
@@ -582,13 +618,13 @@ class SSETransport(MCPBaseTransport):
         return []
 
     # ------------------------------------------------------------------ #
-    #  Context manager support (now uses base class with fixed error)   #
+    #  Context manager support                                           #
     # ------------------------------------------------------------------ #
     async def __aenter__(self):
         """Context manager entry."""
         success = await self.initialize()
         if not success:
-            raise RuntimeError("Failed to initialize SSETransport")  # FIXED: message
+            raise RuntimeError("Failed to initialize SSETransport")
         return self
 
     async def __aexit__(self, exc_type, exc_val, exc_tb):

{chuk_tool_processor-0.6.9 → chuk_tool_processor-0.6.10}/src/chuk_tool_processor.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chuk-tool-processor
-Version: 0.6.9
+Version: 0.6.10
 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>