mirror of
https://github.com/affaan-m/ECC.git
synced 2026-08-17 21:15:40 +02:00
a15c8e8533
* ci: add ruff + mypy to the Python CI job and fix pyproject tool config The python-tests job runs pytest but not lint/type checks, and the ruff and mypy configuration in pyproject.toml was silently broken, so neither tool could run at all. - add ruff and mypy steps to the existing python-tests job - fix invalid pyproject keys: [tool.ruff] src-path -> src, [tool.mypy] src_paths -> mypy_path - ignore ruff UP042 (the (str, Enum) mixin is intentional) - resolve ruff findings (unused/unsorted imports) across src and tests - fix mypy errors in tools/executor.py and prompt/builder.py * fix(ci): satisfy Python lint after main refresh --------- Co-authored-by: haelyra <49814733+haelyra@users.noreply.github.com>
86 lines
2.6 KiB
Python
86 lines
2.6 KiB
Python
from llm.core.types import ToolCall, ToolDefinition
|
|
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"
|