Small changes

tests/test_executor.py

@@ -0,0 +1,86 @@
+import pytest
+from llm.core.types import ToolCall, ToolDefinition, ToolResult
+from llm.tools import ToolExecutor, ToolRegistry
+
+
+class TestToolRegistry:
+    def test_register_and_get(self):
+        registry = ToolRegistry()
+
+        def dummy_func() -> str:
+            return "result"
+
+        tool_def = ToolDefinition(
+            name="dummy",
+            description="A dummy tool",
+            parameters={"type": "object"},
+        )
+        registry.register(tool_def, dummy_func)
+
+        assert registry.has("dummy") is True
+        assert registry.get("dummy") is dummy_func
+        assert registry.get_definition("dummy") == tool_def
+
+    def test_list_tools(self):
+        registry = ToolRegistry()
+        tool_def = ToolDefinition(name="test", description="Test", parameters={})
+        registry.register(tool_def, lambda: None)
+
+        tools = registry.list_tools()
+        assert len(tools) == 1
+        assert tools[0].name == "test"
+
+
+class TestToolExecutor:
+    def test_execute_success(self):
+        registry = ToolRegistry()
+
+        def search(query: str) -> str:
+            return f"Results for: {query}"
+
+        registry.register(
+            ToolDefinition(
+                name="search",
+                description="Search",
+                parameters={"type": "object", "properties": {"query": {"type": "string"}}},
+            ),
+            search,
+        )
+
+        executor = ToolExecutor(registry)
+        result = executor.execute(ToolCall(id="1", name="search", arguments={"query": "test"}))
+
+        assert result.tool_call_id == "1"
+        assert result.content == "Results for: test"
+        assert result.is_error is False
+
+    def test_execute_unknown_tool(self):
+        registry = ToolRegistry()
+        executor = ToolExecutor(registry)
+
+        result = executor.execute(ToolCall(id="1", name="unknown", arguments={}))
+
+        assert result.is_error is True
+        assert "not found" in result.content
+
+    def test_execute_all(self):
+        registry = ToolRegistry()
+
+        def tool1() -> str:
+            return "result1"
+
+        def tool2() -> str:
+            return "result2"
+
+        registry.register(ToolDefinition(name="t1", description="", parameters={}), tool1)
+        registry.register(ToolDefinition(name="t2", description="", parameters={}), tool2)
+
+        executor = ToolExecutor(registry)
+        results = executor.execute_all([
+            ToolCall(id="1", name="t1", arguments={}),
+            ToolCall(id="2", name="t2", arguments={}),
+        ])
+
+        assert len(results) == 2
+        assert results[0].content == "result1"
+        assert results[1].content == "result2"