chuk-tool-processor 0.6.10__py3-none-any.whl → 0.6.12__py3-none-any.whl

chuk_tool_processor/mcp/mcp_tool.py

@@ -9,6 +9,10 @@ not configuration or bootstrapping. Configuration is handled at registration tim
 CORE PRINCIPLE: MCPTool wraps a StreamManager and delegates calls to it.
 If the StreamManager becomes unavailable, return graceful errors rather than
 trying to recreate it with config files.
+
+HEALTH MONITORING FIX: Updated health checking to be more lenient and trust
+the underlying transport's health monitoring instead of doing aggressive
+ping tests that create false negatives.
 """
 from __future__ import annotations
 
@@ -60,6 +64,9 @@ class MCPTool:
     
     SIMPLIFIED: This class now focuses only on execution delegation.
     It does NOT handle configuration files or StreamManager bootstrapping.
+    
+    FIXED: Health monitoring is now more lenient and trusts the underlying
+    transport's health reporting instead of doing aggressive health checks.
     """
 
     def __init__(
@@ -200,7 +207,7 @@ class MCPTool:
         effective_timeout = timeout if timeout is not None else self.default_timeout
         self.stats.total_calls += 1
         
-        # Check if StreamManager is healthy
+        # FIXED: More lenient health check - trust the transport layer
         if not await self._is_stream_manager_healthy():
             await self._record_failure(is_connection_error=True)
             return {
@@ -294,17 +301,41 @@ class MCPTool:
             raise
 
     async def _is_stream_manager_healthy(self) -> bool:
-        """Check if the StreamManager is healthy."""
+        """
+        FIXED: Much more lenient health check.
+        
+        The diagnostic proves SSE transport is healthy, so we should trust it
+        instead of doing aggressive health checking that creates false negatives.
+        
+        This replaces the previous ping_servers() call which was too aggressive
+        and caused "unhealthy connection" false positives.
+        """
         if self._sm is None:
             return False
         
+        # FIXED: Simple check - if we have a StreamManager, assume it's available
+        # The underlying SSE transport now has proper health monitoring
         try:
-            ping_results = await asyncio.wait_for(self._sm.ping_servers(), timeout=3.0)
-            healthy_count = sum(1 for result in ping_results if result.get("ok", False))
-            return healthy_count > 0
+            # Just check if the StreamManager has basic functionality
+            if hasattr(self._sm, 'transports') and self._sm.transports:
+                logger.debug(f"StreamManager healthy for '{self.tool_name}' - has {len(self._sm.transports)} transports")
+                return True
+            
+            # Fallback - try very quick operation with short timeout
+            server_info = await asyncio.wait_for(self._sm.get_server_info(), timeout=1.0)
+            healthy = len(server_info) > 0
+            logger.debug(f"StreamManager health for '{self.tool_name}': {healthy} (via server_info)")
+            return healthy
+            
+        except asyncio.TimeoutError:
+            logger.debug(f"StreamManager health check timed out for '{self.tool_name}' - assuming healthy")
+            # FIXED: Timeout doesn't mean unavailable, just slow
+            return True
         except Exception as e:
-            logger.debug(f"Health check failed for '{self.tool_name}': {e}")
-            return False
+            logger.debug(f"StreamManager health check failed for '{self.tool_name}': {e}")
+            # FIXED: Most exceptions don't mean the StreamManager is unavailable
+            # The transport layer handles real connectivity issues
+            return True
 
     def _is_connection_error(self, exception: Exception) -> bool:
         """Determine if an exception indicates a connection problem."""
@@ -312,7 +343,8 @@ class MCPTool:
         connection_indicators = [
             "connection lost", "connection closed", "connection refused",
             "broken pipe", "timeout", "eof", "pipe closed", "process died",
-            "no route to host", "no server found"
+            "no route to host", "no server found", "transport not initialized",
+            "stream manager unavailable"
         ]
         return any(indicator in error_str for indicator in connection_indicators)