chuk-tool-processor 0.20__tar.gz → 0.22__tar.gz

{chuk_tool_processor-0.20/src/chuk_tool_processor.egg-info → chuk_tool_processor-0.22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chuk-tool-processor
-Version: 0.20
+Version: 0.22
 Summary: Async-native framework for registering, discovering, and executing tools referenced in LLM responses
 License: Apache-2.0
 Keywords: llm,tools,async,ai,openai,mcp,model-context-protocol,tool-calling,function-calling

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "chuk-tool-processor"
-version = "0.20"
+version = "0.22"
 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.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/__init__.py

@@ -20,10 +20,11 @@ Quick Start:
     >>> asyncio.run(main())
 """
 
+from importlib.metadata import version as _pkg_version
 from typing import TYPE_CHECKING
 
-# Version
-__version__ = "0.9.7"
+# Version — single source of truth is pyproject.toml
+__version__ = _pkg_version("chuk-tool-processor")
 
 # Core processor and context
 # Configuration
@@ -159,6 +160,7 @@ from chuk_tool_processor.registry import (
     ToolInfo,
     ToolRegistryProvider,
     create_registry,
+    ensure_registrations,
     get_default_registry,
     initialize,
 )
@@ -227,6 +229,7 @@ __all__ = [
     # Registry
     "ToolInfo",
     "initialize",
+    "ensure_registrations",
     "get_default_registry",
     "create_registry",
     "ToolRegistryProvider",

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/config.py

@@ -66,12 +66,13 @@ Example:
 from __future__ import annotations
 
 import os
-from dataclasses import dataclass, field
-from enum import Enum
+from enum import StrEnum
 from typing import Any
 
+from pydantic import BaseModel, Field
 
-class BackendType(str, Enum):
+
+class BackendType(StrEnum):
     """Backend type for registry and resilience features."""
 
     MEMORY = "memory"
@@ -115,8 +116,7 @@ def _get_float(key: str, default: float | None = None) -> float | None:
         return default
 
 
-@dataclass
-class CircuitBreakerConfig:
+class CircuitBreakerConfig(BaseModel):
     """Circuit breaker configuration."""
 
     enabled: bool = False
@@ -139,14 +139,13 @@ class CircuitBreakerConfig:
         )
 
 
-@dataclass
-class RateLimitConfig:
+class RateLimitConfig(BaseModel):
     """Rate limiting configuration."""
 
     enabled: bool = False
     global_limit: int | None = None
     global_period: float = 60.0
-    tool_limits: dict[str, tuple[int, float]] = field(default_factory=dict)
+    tool_limits: dict[str, tuple[int, float]] = Field(default_factory=dict)
 
     @classmethod
     def from_env(cls) -> RateLimitConfig:
@@ -172,8 +171,7 @@ class RateLimitConfig:
         )
 
 
-@dataclass
-class CacheConfig:
+class CacheConfig(BaseModel):
     """Caching configuration."""
 
     enabled: bool = True
@@ -188,8 +186,7 @@ class CacheConfig:
         )
 
 
-@dataclass
-class RetryConfig:
+class RetryConfig(BaseModel):
     """Retry configuration."""
 
     enabled: bool = True
@@ -210,8 +207,7 @@ class RetryConfig:
         )
 
 
-@dataclass
-class RegistryConfig:
+class RegistryConfig(BaseModel):
     """Registry configuration."""
 
     backend: BackendType = BackendType.MEMORY
@@ -236,8 +232,7 @@ class RegistryConfig:
         )
 
 
-@dataclass
-class ProcessorConfig:
+class ProcessorConfig(BaseModel):
     """
     Complete configuration for ToolProcessor with environment variable support.
 
@@ -266,7 +261,7 @@ class ProcessorConfig:
     """
 
     # Registry configuration
-    registry: RegistryConfig = field(default_factory=RegistryConfig)
+    registry: RegistryConfig = Field(default_factory=RegistryConfig)
 
     # Resilience backend selection (rate limiting, circuit breaker)
     resilience_backend: BackendType = BackendType.MEMORY
@@ -280,10 +275,10 @@ class ProcessorConfig:
     max_concurrency: int | None = None
 
     # Feature configs
-    circuit_breaker: CircuitBreakerConfig = field(default_factory=CircuitBreakerConfig)
-    rate_limit: RateLimitConfig = field(default_factory=RateLimitConfig)
-    cache: CacheConfig = field(default_factory=CacheConfig)
-    retry: RetryConfig = field(default_factory=RetryConfig)
+    circuit_breaker: CircuitBreakerConfig = Field(default_factory=CircuitBreakerConfig)
+    rate_limit: RateLimitConfig = Field(default_factory=RateLimitConfig)
+    cache: CacheConfig = Field(default_factory=CacheConfig)
+    retry: RetryConfig = Field(default_factory=RetryConfig)
 
     # Backwards compatibility alias
     @property

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/core/context.py

@@ -23,13 +23,13 @@ import contextlib
 import contextvars
 import uuid
 from datetime import UTC, datetime, timedelta
-from enum import Enum
+from enum import StrEnum
 from typing import Any, Self
 
 from pydantic import BaseModel, ConfigDict, Field, field_validator
 
 
-class ContextHeader(str, Enum):
+class ContextHeader(StrEnum):
     """Standard HTTP headers for context propagation."""
 
     REQUEST_ID = "X-Request-ID"
@@ -42,7 +42,7 @@ class ContextHeader(str, Enum):
     BUDGET = "X-Budget"
 
 
-class ContextKey(str, Enum):
+class ContextKey(StrEnum):
     """Standard keys for context dictionary representation."""
 
     REQUEST_ID = "request_id"

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/core/exceptions.py

@@ -9,7 +9,7 @@ that enable planners to make intelligent retry and fallback decisions.
 from __future__ import annotations
 
 from difflib import get_close_matches
-from enum import Enum
+from enum import StrEnum
 from typing import TYPE_CHECKING, Any, cast
 
 from pydantic import BaseModel, ConfigDict, Field
@@ -18,7 +18,7 @@ if TYPE_CHECKING:
     pass  # For future type hints if needed
 
 
-class ErrorCode(str, Enum):
+class ErrorCode(StrEnum):
     """Machine-readable error codes for tool processor errors."""
 
     # Tool registry errors
@@ -57,7 +57,7 @@ class ErrorCode(str, Enum):
     CONFIGURATION_ERROR = "CONFIGURATION_ERROR"
 
 
-class ErrorCategory(str, Enum):
+class ErrorCategory(StrEnum):
     """
     High-level error categories for planner decision-making.
 

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/discovery/dynamic_provider.py

@@ -20,7 +20,7 @@ from __future__ import annotations
 import json
 import logging
 from abc import ABC, abstractmethod
-from enum import Enum
+from enum import StrEnum
 from typing import Any, Generic, TypeVar
 
 from chuk_tool_processor.discovery.search import (
@@ -77,7 +77,7 @@ def _error_response(
     return response
 
 
-class DynamicToolName(str, Enum):
+class DynamicToolName(StrEnum):
     """Names of available dynamic tools - no magic strings!"""
 
     LIST_TOOLS = "list_tools"

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/execution/bulkhead.py

@@ -33,7 +33,7 @@ import asyncio
 import fnmatch
 from collections import defaultdict
 from contextlib import asynccontextmanager
-from enum import Enum
+from enum import StrEnum
 from functools import lru_cache
 from typing import Any
 
@@ -44,7 +44,7 @@ from chuk_tool_processor.logging import get_logger
 logger = get_logger("chuk_tool_processor.execution.bulkhead")
 
 
-class BulkheadLimitType(str, Enum):
+class BulkheadLimitType(StrEnum):
     """Types of bulkhead limits that can be exceeded."""
 
     TOOL = "tool"

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/execution/wrappers/circuit_breaker.py

@@ -18,7 +18,7 @@ from __future__ import annotations
 import asyncio
 import time
 from datetime import UTC, datetime
-from enum import Enum
+from enum import StrEnum
 from typing import Any
 
 from chuk_tool_processor.core.exceptions import ToolCircuitOpenError
@@ -50,7 +50,7 @@ except ImportError:
 # --------------------------------------------------------------------------- #
 # Circuit breaker state
 # --------------------------------------------------------------------------- #
-class CircuitState(str, Enum):
+class CircuitState(StrEnum):
     """Circuit breaker states."""
 
     CLOSED = "closed"  # Normal operation

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/execution/wrappers/factory.py

@@ -38,7 +38,7 @@ Example:
 from __future__ import annotations
 
 from dataclasses import dataclass
-from enum import Enum
+from enum import StrEnum
 from typing import TYPE_CHECKING, Any, Protocol, runtime_checkable
 
 from chuk_tool_processor.execution.wrappers.circuit_breaker import (
@@ -62,7 +62,7 @@ if TYPE_CHECKING:
 logger = get_logger("chuk_tool_processor.execution.wrappers.factory")
 
 
-class WrapperBackend(str, Enum):
+class WrapperBackend(StrEnum):
     """Backend type for production wrappers."""
 
     MEMORY = "memory"  # In-memory, single-instance

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/execution/wrappers/observable.py

@@ -162,7 +162,7 @@ class ObservableExecutor:
         decision = GuardDecision(
             guard_name=guard_name,
             guard_class=type(guard_result).__module__,
-            verdict=guard_result.verdict.value.upper(),
+            verdict=guard_result.verdict,
             reason=guard_result.reason,
             details=guard_result.details,
             duration_ms=duration_ms,

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/execution/wrappers/redis_circuit_breaker.py

@@ -37,7 +37,6 @@ from __future__ import annotations
 
 import time
 from dataclasses import dataclass
-from enum import Enum
 from typing import TYPE_CHECKING, Any
 
 from chuk_tool_processor.logging import get_logger
@@ -48,14 +47,6 @@ if TYPE_CHECKING:
 logger = get_logger("chuk_tool_processor.execution.wrappers.redis_circuit_breaker")
 
 
-class CircuitState(str, Enum):
-    """Circuit breaker states."""
-
-    CLOSED = "closed"  # Normal operation
-    OPEN = "open"  # Blocking requests due to failures
-    HALF_OPEN = "half_open"  # Testing recovery with limited requests
-
-
 @dataclass
 class RedisCircuitBreakerConfig:
     """Configuration for Redis circuit breaker behavior."""

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/guards/base.py

@@ -8,13 +8,13 @@ Each guard focuses on one concern and returns a typed result.
 from __future__ import annotations
 
 from abc import ABC, abstractmethod
-from enum import Enum
+from enum import StrEnum
 from typing import Any, Protocol, runtime_checkable
 
 from pydantic import BaseModel, Field
 
 
-class GuardVerdict(str, Enum):
+class GuardVerdict(StrEnum):
     """Verdict from a guard check."""
 
     ALLOW = "allow"  # Proceed with execution

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/guards/models.py

@@ -6,10 +6,10 @@ Defines tool classifications and enforcement levels used by guards.
 
 from __future__ import annotations
 
-from enum import Enum
+from enum import StrEnum
 
 
-class EnforcementLevel(str, Enum):
+class EnforcementLevel(StrEnum):
     """Enforcement level for guards and constraints."""
 
     OFF = "off"  # No enforcement

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/guards/network_policy.py

@@ -8,7 +8,7 @@ from __future__ import annotations
 
 import ipaddress
 import re
-from enum import Enum
+from enum import StrEnum
 from typing import Any
 from urllib.parse import urlparse
 
@@ -18,7 +18,7 @@ from chuk_tool_processor.guards.base import BaseGuard, GuardResult
 from chuk_tool_processor.guards.models import EnforcementLevel
 
 
-class NetworkViolationType(str, Enum):
+class NetworkViolationType(StrEnum):
     """Types of network policy violations."""
 
     DOMAIN_NOT_ALLOWED = "domain_not_allowed"

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/guards/output_size.py

@@ -7,7 +7,7 @@ Caps bytes, tokens, array lengths, and nesting depth.
 from __future__ import annotations
 
 import json
-from enum import Enum
+from enum import StrEnum
 from typing import Any
 
 from pydantic import BaseModel, Field
@@ -15,7 +15,7 @@ from pydantic import BaseModel, Field
 from chuk_tool_processor.guards.base import BaseGuard, GuardResult
 
 
-class TruncationMode(str, Enum):
+class TruncationMode(StrEnum):
     """How to handle oversized outputs."""
 
     ERROR = "error"  # Block with error
@@ -23,7 +23,7 @@ class TruncationMode(str, Enum):
     PAGINATE = "paginate"  # Return with pagination info
 
 
-class SizeViolationType(str, Enum):
+class SizeViolationType(StrEnum):
     """Types of size violations."""
 
     BYTES_EXCEEDED = "bytes_exceeded"

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/guards/plan_shape.py

@@ -6,7 +6,7 @@ Guards against overly long chains, excessive unique tools, and fan-out explosion
 
 from __future__ import annotations
 
-from enum import Enum
+from enum import StrEnum
 from typing import Any
 
 from pydantic import BaseModel, Field
@@ -15,7 +15,7 @@ from chuk_tool_processor.guards.base import BaseGuard, GuardResult
 from chuk_tool_processor.guards.models import EnforcementLevel
 
 
-class PlanShapeViolationType(str, Enum):
+class PlanShapeViolationType(StrEnum):
     """Types of plan shape violations."""
 
     CHAIN_TOO_LONG = "chain_too_long"

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/guards/retry_safety.py

@@ -8,7 +8,7 @@ from __future__ import annotations
 
 import hashlib
 import time
-from enum import Enum
+from enum import StrEnum
 from typing import Any
 
 from pydantic import BaseModel, Field
@@ -17,7 +17,7 @@ from chuk_tool_processor.guards.base import BaseGuard, GuardResult
 from chuk_tool_processor.guards.models import EnforcementLevel
 
 
-class ErrorClass(str, Enum):
+class ErrorClass(StrEnum):
     """Classification of errors for retry decisions."""
 
     VALIDATION = "validation"

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/guards/schema_strictness.py

@@ -7,7 +7,7 @@ Blocks or auto-fixes calls that don't conform to tool JSON schemas.
 from __future__ import annotations
 
 from collections.abc import Awaitable, Callable
-from enum import Enum
+from enum import StrEnum
 from typing import Any
 
 from pydantic import BaseModel, Field
@@ -16,7 +16,7 @@ from chuk_tool_processor.guards.base import BaseGuard, GuardResult
 from chuk_tool_processor.guards.models import EnforcementLevel
 
 
-class SchemaViolationType(str, Enum):
+class SchemaViolationType(StrEnum):
     """Types of schema violations."""
 
     MISSING_REQUIRED = "missing_required"

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/guards/sensitive_data.py

@@ -8,7 +8,7 @@ from __future__ import annotations
 
 import hashlib
 import re
-from enum import Enum
+from enum import StrEnum
 from typing import Any
 
 from pydantic import BaseModel, Field
@@ -17,7 +17,7 @@ from chuk_tool_processor.guards.base import BaseGuard, GuardResult
 from chuk_tool_processor.guards.models import EnforcementLevel
 
 
-class SensitiveDataType(str, Enum):
+class SensitiveDataType(StrEnum):
     """Types of sensitive data patterns."""
 
     API_KEY = "api_key"
@@ -30,7 +30,7 @@ class SensitiveDataType(str, Enum):
     BASIC_AUTH = "basic_auth"
 
 
-class RedactMode(str, Enum):
+class RedactMode(StrEnum):
     """How to handle detected secrets."""
 
     BLOCK = "block"  # Block execution entirely

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/guards/side_effect.py

@@ -7,7 +7,7 @@ Labels tools as read_only/write/destructive and enforces policies.
 from __future__ import annotations
 
 from collections.abc import Awaitable, Callable
-from enum import Enum
+from enum import StrEnum
 from typing import Any
 
 from pydantic import BaseModel, Field
@@ -16,7 +16,7 @@ from chuk_tool_processor.guards.base import BaseGuard, GuardResult
 from chuk_tool_processor.guards.models import EnforcementLevel, ToolClassification
 
 
-class SideEffectClass(str, Enum):
+class SideEffectClass(StrEnum):
     """Classification of tool side effects."""
 
     READ_ONLY = "read_only"
@@ -24,7 +24,7 @@ class SideEffectClass(str, Enum):
     DESTRUCTIVE = "destructive"
 
 
-class ExecutionMode(str, Enum):
+class ExecutionMode(StrEnum):
     """Allowed execution modes."""
 
     READ_ONLY = "read_only"  # Only read operations allowed
@@ -32,7 +32,7 @@ class ExecutionMode(str, Enum):
     DESTRUCTIVE_ALLOWED = "destructive_allowed"  # All operations allowed
 
 
-class Environment(str, Enum):
+class Environment(StrEnum):
     """Deployment environment."""
 
     DEV = "dev"

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/guards/timeout_budget.py

@@ -7,7 +7,7 @@ Enforces per-turn and per-plan time budgets with soft/hard limits.
 from __future__ import annotations
 
 import time
-from enum import Enum
+from enum import StrEnum
 from typing import Any
 
 from pydantic import BaseModel, Field
@@ -16,7 +16,7 @@ from chuk_tool_processor.guards.base import BaseGuard, GuardResult
 from chuk_tool_processor.guards.models import EnforcementLevel
 
 
-class DegradeAction(str, Enum):
+class DegradeAction(StrEnum):
     """Actions to take when soft budget is exceeded."""
 
     DISABLE_RETRIES = "disable_retries"
@@ -25,7 +25,7 @@ class DegradeAction(str, Enum):
     REDUCE_TIMEOUT = "reduce_timeout"
 
 
-class BudgetStatus(str, Enum):
+class BudgetStatus(StrEnum):
     """Current budget status."""
 
     OK = "ok"

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/mcp/mcp_tool.py

@@ -393,7 +393,7 @@ class MCPTool:
         # Check if we should open the circuit breaker
         if self._consecutive_failures >= self.recovery_config.circuit_breaker_threshold and not self._circuit_open:
             self._circuit_open = True
-            self._circuit_open_time = time.time()
+            self._circuit_open_time = time.monotonic()
             self.connection_state = ConnectionState.FAILED
             logger.error(
                 f"Circuit breaker opened for tool '{self.tool_name}' after {self._consecutive_failures} consecutive failures"
@@ -407,7 +407,7 @@ class MCPTool:
         # Check if enough time has passed to close the circuit
         if (
             self._circuit_open_time
-            and time.time() - self._circuit_open_time >= self.recovery_config.circuit_breaker_timeout
+            and time.monotonic() - self._circuit_open_time >= self.recovery_config.circuit_breaker_timeout
         ):
             self._circuit_open = False
             self._circuit_open_time = None

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/mcp/middleware.py

@@ -11,7 +11,7 @@ Async-native, Pydantic-based middleware configuration for:
 from __future__ import annotations
 
 from datetime import UTC, datetime
-from enum import Enum
+from enum import StrEnum
 from typing import TYPE_CHECKING, Any
 
 from pydantic import BaseModel, Field
@@ -43,7 +43,7 @@ logger = get_logger("chuk_tool_processor.mcp.middleware")
 # ============================================================================
 
 
-class MiddlewareLayer(str, Enum):
+class MiddlewareLayer(StrEnum):
     """Middleware layer identifiers."""
 
     RETRY = "retry"
@@ -51,7 +51,7 @@ class MiddlewareLayer(str, Enum):
     RATE_LIMITING = "rate_limiting"
 
 
-class RetryableError(str, Enum):
+class RetryableError(StrEnum):
     """Error patterns that should trigger retry."""
 
     TRANSPORT_NOT_INITIALIZED = "Transport not initialized"
@@ -62,7 +62,7 @@ class RetryableError(str, Enum):
     CLOSED = "closed"
 
 
-class NonRetryableError(str, Enum):
+class NonRetryableError(StrEnum):
     """Error patterns that should NOT be retried."""
 
     OAUTH = "oauth"

{chuk_tool_processor-0.20 → chuk_tool_processor-0.22}/src/chuk_tool_processor/mcp/models.py

@@ -6,18 +6,19 @@ Pydantic models for MCP server configurations.
 
 from __future__ import annotations
 
-from enum import Enum
+from enum import StrEnum
 from typing import Any
 
 from pydantic import BaseModel, Field, model_validator
 
 
-class MCPTransport(str, Enum):
+class MCPTransport(StrEnum):
     """Supported MCP transport types."""
 
     STDIO = "stdio"
     SSE = "sse"
     HTTP = "http"
+    HTTP_STREAMABLE = "http_streamable"
 
 
 class MCPServerConfig(BaseModel):
@@ -46,7 +47,7 @@ class MCPServerConfig(BaseModel):
             if not self.command:
                 raise ValueError("command is required for stdio transport")
         else:
-            # SSE/HTTP
+            # SSE/HTTP/HTTP_STREAMABLE
             if not self.url:
                 raise ValueError(f"url is required for {self.transport} transport")
             # Extract API key from Authorization header if present