mirror of
https://github.com/affaan-m/everything-claude-code.git
synced 2026-04-14 05:43:29 +08:00
87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
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"
|