chuk-tool-processor 0.9__tar.gz → 0.9.2__tar.gz

{chuk_tool_processor-0.9 → chuk_tool_processor-0.9.2}/PKG-INFO

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

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

@@ -176,17 +176,24 @@ class StreamManager:
             for idx, server_name in enumerate(servers):
                 try:
                     if transport_type == "stdio":
-                        params = await load_config(config_file, server_name)
+                        params, server_timeout = await load_config(config_file, server_name)
+                        # Use per-server timeout if specified, otherwise use global default
+                        effective_timeout = server_timeout if server_timeout is not None else default_timeout
+                        logger.info(
+                            f"Server '{server_name}' using timeout: {effective_timeout}s (per-server: {server_timeout}, default: {default_timeout})"
+                        )
                         # Use initialization_timeout for connection_timeout since subprocess
                         # launch can take time (e.g., uvx downloading packages)
                         transport: MCPBaseTransport = StdioTransport(
-                            params, connection_timeout=initialization_timeout, default_timeout=default_timeout
+                            params, connection_timeout=initialization_timeout, default_timeout=effective_timeout
                         )
                     elif transport_type == "sse":
                         logger.debug(
                             "Using SSE transport in initialize() - consider using initialize_with_sse() instead"
                         )
-                        params = await load_config(config_file, server_name)
+                        params, server_timeout = await load_config(config_file, server_name)
+                        # Use per-server timeout if specified, otherwise use global default
+                        effective_timeout = server_timeout if server_timeout is not None else default_timeout
 
                         if isinstance(params, dict) and "url" in params:
                             sse_url = params["url"]
@@ -199,7 +206,7 @@ class StreamManager:
                             logger.debug("No URL configured for SSE transport, using default: %s", sse_url)
 
                         # Build SSE transport with optional headers
-                        transport_params = {"url": sse_url, "api_key": api_key, "default_timeout": default_timeout}
+                        transport_params = {"url": sse_url, "api_key": api_key, "default_timeout": effective_timeout}
                         if headers:
                             transport_params["headers"] = headers
 
@@ -209,7 +216,9 @@ class StreamManager:
                         logger.debug(
                             "Using HTTP Streamable transport in initialize() - consider using initialize_with_http_streamable() instead"
                         )
-                        params = await load_config(config_file, server_name)
+                        params, server_timeout = await load_config(config_file, server_name)
+                        # Use per-server timeout if specified, otherwise use global default
+                        effective_timeout = server_timeout if server_timeout is not None else default_timeout
 
                         if isinstance(params, dict) and "url" in params:
                             http_url = params["url"]
@@ -227,7 +236,7 @@ class StreamManager:
                         transport_params = {
                             "url": http_url,
                             "api_key": api_key,
-                            "default_timeout": default_timeout,
+                            "default_timeout": effective_timeout,
                             "session_id": session_id,
                         }
                         # Note: headers not added until HTTPStreamableTransport supports them

{chuk_tool_processor-0.9 → chuk_tool_processor-0.9.2}/src/chuk_tool_processor/observability/metrics.py

@@ -12,7 +12,7 @@ from typing import TYPE_CHECKING
 from chuk_tool_processor.logging import get_logger
 
 if TYPE_CHECKING:
-    from prometheus_client import Counter, Gauge, Histogram
+    from prometheus_client import Counter, Gauge, Histogram  # type: ignore[import-not-found]
 
 logger = get_logger("chuk_tool_processor.observability.metrics")
 

{chuk_tool_processor-0.9 → chuk_tool_processor-0.9.2}/src/chuk_tool_processor/observability/tracing.py

@@ -12,7 +12,7 @@ from typing import TYPE_CHECKING, Any
 from chuk_tool_processor.logging import get_logger
 
 if TYPE_CHECKING:
-    from opentelemetry.trace import Span, Tracer
+    from opentelemetry.trace import Span, Tracer  # type: ignore[import-not-found]
 
 logger = get_logger("chuk_tool_processor.observability.tracing")
 
@@ -34,11 +34,13 @@ def init_tracer(service_name: str = "chuk-tool-processor") -> Tracer | NoOpTrace
     global _tracer, _tracing_enabled
 
     try:
-        from opentelemetry import trace
-        from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
-        from opentelemetry.sdk.resources import Resource
-        from opentelemetry.sdk.trace import TracerProvider
-        from opentelemetry.sdk.trace.export import BatchSpanProcessor
+        from opentelemetry import trace  # type: ignore[import-not-found]
+        from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import (  # type: ignore[import-not-found]
+            OTLPSpanExporter,
+        )
+        from opentelemetry.sdk.resources import Resource  # type: ignore[import-not-found]
+        from opentelemetry.sdk.trace import TracerProvider  # type: ignore[import-not-found]
+        from opentelemetry.sdk.trace.export import BatchSpanProcessor  # type: ignore[import-not-found]
 
         # Create resource with service name
         resource = Resource.create({"service.name": service_name})

{chuk_tool_processor-0.9 → chuk_tool_processor-0.9.2}/src/chuk_tool_processor.egg-info/PKG-INFO

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