updated version and tests

Author: chrishayuk

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "chuk-tool-processor"
-version = "0.6.13"
+version = "0.6.14"
 description = "Async-native framework for registering, discovering, and executing tools referenced in LLM responses"
 readme = "README.md"
 requires-python = ">=3.11"

tests/registry/test_decorators.py

@@ -504,7 +504,7 @@ class TestEnsureRegistrations:
     @pytest.mark.asyncio
     async def test_ensure_registrations_empty(self):
         """Test ensure_registrations with no pending tools."""
-        with patch("chuk_tool_processor.registry.decorators._pending_registrations", []):
+        with patch("chuk_tool_processor.registry.decorators._PENDING_REGISTRATIONS", []):
             await ensure_registrations()
             # Should complete without error
 
@@ -514,14 +514,17 @@ async def test_ensure_registrations_with_pending(self):
         mock_registry = AsyncMock()
         mock_registry.register_tool = AsyncMock()
 
-        # Create pending registrations
-        pending = [
-            ("tool1", ValidatedTool, {"namespace": "default"}),
-            ("tool2", ValidatedTool, {"namespace": "custom"}),
-        ]
+        # Create async registration functions
+        async def reg1():
+            await mock_registry.register_tool(ValidatedTool, name="tool1", namespace="default", metadata={})
+
+        async def reg2():
+            await mock_registry.register_tool(ValidatedTool, name="tool2", namespace="custom", metadata={})
+
+        pending = [reg1, reg2]
 
         with (
-            patch("chuk_tool_processor.registry.decorators._pending_registrations", pending),
+            patch("chuk_tool_processor.registry.decorators._PENDING_REGISTRATIONS", pending),
             patch(
                 "chuk_tool_processor.registry.provider.ToolRegistryProvider.get_registry", return_value=mock_registry
             ),
@@ -537,10 +540,14 @@ async def test_ensure_registrations_clears_pending(self):
         mock_registry = AsyncMock()
         mock_registry.register_tool = AsyncMock()
 
-        pending = [("tool1", ValidatedTool, {})]
+        # Create async registration function
+        async def reg1():
+            await mock_registry.register_tool(ValidatedTool, name="tool1", namespace="default", metadata={})
+
+        pending = [reg1]
 
         with (
-            patch("chuk_tool_processor.registry.decorators._pending_registrations", pending) as pending_list,
+            patch("chuk_tool_processor.registry.decorators._PENDING_REGISTRATIONS", pending) as pending_list,
             patch(
                 "chuk_tool_processor.registry.provider.ToolRegistryProvider.get_registry", return_value=mock_registry
             ),
@@ -692,21 +699,17 @@ class Arguments(BaseModel):
                 async def _execute(self, value: str, count: int):
                     return {"result": value * count}
 
+            # Process pending registrations
+            await ensure_registrations()
+
             # Tool should be registered
             mock_registry.register_tool.assert_called()
 
-            # Should be in decorated tools
-            with patch("chuk_tool_processor.registry.decorators._decorated_tools", []) as decorated:
-
-                @register_tool(name="another")
-                class AnotherTool(ValidatedTool):
-                    class Arguments(BaseModel):
-                        x: int
-
-                    async def _execute(self, x: int):
-                        return {"result": x}
-
-                assert len(decorated) == 1
+            # Verify the tool was registered with correct parameters
+            call_args = mock_registry.register_tool.call_args
+            assert call_args is not None
+            assert call_args.kwargs["name"] == "integration_tool"
+            assert call_args.kwargs["namespace"] == "test"
 
     def test_function_to_tool_conversion(self):
         """Test converting function to tool via decorator."""

tests/registry/test_registry_init.py

@@ -65,4 +65,6 @@ async def execute(self, x: int) -> int:
         # Check if the tool was registered
         tool = await registry.get_tool("test_init_tool_unique", namespace="test_init_ns")
         assert tool is not None
+        # Check if the tool is a class or has __name__ attribute
+        assert hasattr(tool, "__name__"), f"Tool should have __name__, got {type(tool)}: {tool}"
         assert tool.__name__ == "TestInitTool"

tests/registry/test_tool_export_simple.py

@@ -92,9 +92,11 @@ async def test_tool_by_openai_name(self):
             tool = await tool_by_openai_name("MockTool")
             assert tool == MockTool
 
-            # Unknown tool returns None
-            tool = await tool_by_openai_name("UnknownTool")
-            assert tool is None
+            # Unknown tool raises KeyError
+            import pytest
+
+            with pytest.raises(KeyError, match="No tool registered for OpenAI name 'UnknownTool'"):
+                await tool_by_openai_name("UnknownTool")
 
     @pytest.mark.asyncio
     async def test_openai_functions(self):