chuk-tool-processor 0.6.4__py3-none-any.whl → 0.9.7__py3-none-any.whl

chuk_tool_processor/registry/__init__.py

@@ -3,13 +3,11 @@
 Async-native tool registry package for managing and accessing tool implementations.
 """
 
-import asyncio
-from typing import Optional
-
+from chuk_tool_processor.registry.decorators import discover_decorated_tools, ensure_registrations, register_tool
 from chuk_tool_processor.registry.interface import ToolRegistryInterface
-from chuk_tool_processor.registry.metadata import ToolMetadata, StreamingToolMetadata
+from chuk_tool_processor.registry.metadata import StreamingToolMetadata, ToolMetadata
 from chuk_tool_processor.registry.provider import ToolRegistryProvider, get_registry
-from chuk_tool_processor.registry.decorators import register_tool, ensure_registrations, discover_decorated_tools
+
 
 # --------------------------------------------------------------------------- #
 # The default_registry is now an async function instead of direct property access
@@ -17,14 +15,15 @@ from chuk_tool_processor.registry.decorators import register_tool, ensure_regist
 async def get_default_registry() -> ToolRegistryInterface:
     """
     Get the default registry instance.
-    
+
     This is a convenience function that calls ToolRegistryProvider.get_registry()
-    
+
     Returns:
         The default tool registry
     """
     return await ToolRegistryProvider.get_registry()
 
+
 __all__ = [
     "ToolRegistryInterface",
     "ToolMetadata",
@@ -37,24 +36,25 @@ __all__ = [
     "get_registry",
 ]
 
+
 # --------------------------------------------------------------------------- #
 # Initialization helper that should be called at application startup
 # --------------------------------------------------------------------------- #
 async def initialize():
     """
     Initialize the registry system.
-    
+
     This function should be called during application startup to:
     1. Ensure the registry is created
     2. Register all tools decorated with @register_tool
-    
+
     Returns:
         The initialized registry
     """
     # Initialize registry
     registry = await get_default_registry()
-    
+
     # Process all pending tool registrations
     await ensure_registrations()
-    
-    return registry
+
+    return registry

chuk_tool_processor/registry/auto_register.py

@@ -11,10 +11,9 @@ These tools will immediately show up in the global registry.
 
 from __future__ import annotations
 
-import asyncio
 import inspect
-import types
-from typing import Callable, ForwardRef, Type, get_type_hints, Any, Optional, Dict, Union
+from collections.abc import Callable
+from typing import Any, ForwardRef, get_type_hints
 
 import anyio
 from pydantic import BaseModel, create_model
@@ -25,16 +24,14 @@ except ModuleNotFoundError:  # pragma: no cover
     BaseTool = None  # noqa: N816  - keep the name for isinstance() checks
 
 # registry
-from .decorators import register_tool
 from .provider import ToolRegistryProvider
 
-
 # ────────────────────────────────────────────────────────────────────────────
 # internals - build a Pydantic schema from an arbitrary callable
 # ────────────────────────────────────────────────────────────────────────────
 
 
-def _auto_schema(func: Callable) -> Type[BaseModel]:
+def _auto_schema(func: Callable) -> type[BaseModel]:
     """
     Turn a function signature into a `pydantic.BaseModel` subclass.
 
@@ -55,8 +52,7 @@ def _auto_schema(func: Callable) -> Type[BaseModel]:
         # couldn't resolve the type.
         hint: type = (
             raw_hint
-            if raw_hint not in (inspect._empty, None, str)
-            and not isinstance(raw_hint, (str, ForwardRef))
+            if raw_hint not in (inspect._empty, None, str) and not isinstance(raw_hint, str | ForwardRef)
             else str
         )
         fields[param.name] = (hint, ...)  # "..."  → required
@@ -78,7 +74,7 @@ async def register_fn_tool(
 ) -> None:
     """
     Register a plain function as a tool asynchronously.
-    
+
     Args:
         func: The function to register (can be sync or async)
         name: Optional name for the tool (defaults to function name)
@@ -88,21 +84,23 @@ async def register_fn_tool(
     schema = _auto_schema(func)
     tool_name = name or func.__name__
     tool_description = (description or func.__doc__ or "").strip()
-    
+
     # Create the tool wrapper class
     class _Tool:  # noqa: D401, N801 - internal auto-wrapper
         """Auto-generated tool wrapper for function."""
-        
+
         async def execute(self, **kwargs: Any) -> Any:
             """Execute the wrapped function."""
             if inspect.iscoroutinefunction(func):
                 return await func(**kwargs)
             # off-load blocking sync work
-            return await anyio.to_thread.run_sync(func, **kwargs)
-    
+            import functools
+
+            return await anyio.to_thread.run_sync(functools.partial(func, **kwargs))
+
     # Set the docstring
     _Tool.__doc__ = tool_description
-    
+
     # Get the registry and register directly
     registry = await ToolRegistryProvider.get_registry()
     await registry.register_tool(
@@ -115,7 +113,7 @@ async def register_fn_tool(
             "argument_schema": schema.model_json_schema(),
             "source": "function",
             "source_name": func.__qualname__,
-        }
+        },
     )
 
 
@@ -133,7 +131,7 @@ async def register_langchain_tool(
 ) -> None:
     """
     Register a **LangChain** `BaseTool` instance asynchronously.
-    
+
     Works with any object exposing `.run` / `.arun` methods.
 
     Args:
@@ -141,45 +139,39 @@ async def register_langchain_tool(
         name: Optional name for the tool (defaults to tool.name)
         description: Optional description (defaults to tool.description)
         namespace: Registry namespace (defaults to "default")
-        
+
     Raises:
         RuntimeError: If LangChain isn't installed
         TypeError: If the object isn't a LangChain BaseTool
     """
     if BaseTool is None:
-        raise RuntimeError(
-            "register_langchain_tool() requires LangChain - "
-            "install with `pip install langchain`"
-        )
+        raise RuntimeError("register_langchain_tool() requires LangChain - install with `pip install langchain`")
 
     if not isinstance(tool, BaseTool):  # pragma: no cover
-        raise TypeError(
-            "Expected a langchain.tools.base.BaseTool instance - got "
-            f"{type(tool).__name__}"
-        )
+        raise TypeError(f"Expected a langchain.tools.base.BaseTool instance - got {type(tool).__name__}")
 
     # Prefer async implementation if available
     fn = tool.arun if hasattr(tool, "arun") else tool.run
-    
+
     tool_name = name or tool.name or tool.__class__.__name__
     tool_description = description or tool.description or (tool.__doc__ or "")
-    
+
     await register_fn_tool(
         fn,
         name=tool_name,
         description=tool_description,
         namespace=namespace,
     )
-    
+
     # Update the metadata to include LangChain info
     registry = await ToolRegistryProvider.get_registry()
     metadata = await registry.get_metadata(tool_name, namespace)
-    
+
     if metadata:
         updated_metadata = metadata.model_copy()
         # Update source info
         updated_metadata.tags.add("langchain")
-        
+
         # Re-register with updated metadata
         await registry.register_tool(
             await registry.get_tool(tool_name, namespace),