mirror of
https://github.com/affaan-m/everything-claude-code.git
synced 2026-05-11 23:27:25 +08:00
b39d2244cf
- add Vite and Redis pattern skills from closed stale PRs - add frontend-slides support assets - port skill-comply runner fixes and LLM prompt/provider regressions - harden agent frontmatter validation and sync catalog counts
112 lines
3.4 KiB
Python
112 lines
3.4 KiB
Python
from types import SimpleNamespace
|
|
from typing import Any
|
|
|
|
import pytest
|
|
|
|
from llm.core.types import LLMInput, Message, Role
|
|
from llm.providers.claude import ClaudeProvider
|
|
|
|
|
|
class FakeMessages:
|
|
def __init__(self, response: SimpleNamespace) -> None:
|
|
self.response = response
|
|
|
|
def create(self, **_params: object) -> SimpleNamespace:
|
|
return self.response
|
|
|
|
|
|
class FakeClient:
|
|
def __init__(self, response: SimpleNamespace) -> None:
|
|
self.messages = FakeMessages(response)
|
|
self.api_key = "test-key"
|
|
|
|
|
|
def make_provider(response: SimpleNamespace) -> ClaudeProvider:
|
|
provider = ClaudeProvider(api_key="test-key")
|
|
provider.client = FakeClient(response)
|
|
return provider
|
|
|
|
|
|
def make_response(content: list[SimpleNamespace], stop_reason: str = "tool_use") -> SimpleNamespace:
|
|
return SimpleNamespace(
|
|
content=content,
|
|
model="claude-test",
|
|
usage=SimpleNamespace(input_tokens=3, output_tokens=5),
|
|
stop_reason=stop_reason,
|
|
)
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_generate_collects_text_and_tool_use_blocks() -> None:
|
|
provider = make_provider(
|
|
make_response(
|
|
[
|
|
SimpleNamespace(type="text", text="I will search. "),
|
|
SimpleNamespace(type="tool_use", id="toolu_1", name="search", input={"query": "claude"}),
|
|
SimpleNamespace(type="text", text="Done."),
|
|
]
|
|
)
|
|
)
|
|
|
|
output = provider.generate(LLMInput(messages=[Message(role=Role.USER, content="Search")]))
|
|
|
|
assert output.content == "I will search. Done."
|
|
assert output.tool_calls is not None
|
|
assert len(output.tool_calls) == 1
|
|
assert output.tool_calls[0].id == "toolu_1"
|
|
assert output.tool_calls[0].name == "search"
|
|
assert output.tool_calls[0].arguments == {"query": "claude"}
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_generate_collects_multiple_tool_use_blocks() -> None:
|
|
provider = make_provider(
|
|
make_response(
|
|
[
|
|
SimpleNamespace(type="tool_use", id="toolu_1", name="search", input={"query": "claude"}),
|
|
SimpleNamespace(
|
|
type="tool_use",
|
|
id="toolu_2",
|
|
name="read",
|
|
input=SimpleNamespace(path="README.md"),
|
|
),
|
|
]
|
|
)
|
|
)
|
|
|
|
output = provider.generate(LLMInput(messages=[Message(role=Role.USER, content="Use tools")]))
|
|
|
|
assert output.content == ""
|
|
assert [call.id for call in output.tool_calls or []] == ["toolu_1", "toolu_2"]
|
|
assert (output.tool_calls or [])[1].arguments == {"path": "README.md"}
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_generate_copies_tool_use_dict_arguments() -> None:
|
|
raw_arguments: dict[str, Any] = {"query": "claude"}
|
|
provider = make_provider(
|
|
make_response(
|
|
[SimpleNamespace(type="tool_use", id="toolu_1", name="search", input=raw_arguments)]
|
|
)
|
|
)
|
|
|
|
output = provider.generate(LLMInput(messages=[Message(role=Role.USER, content="Use tools")]))
|
|
raw_arguments["query"] = "mutated"
|
|
|
|
assert (output.tool_calls or [])[0].arguments == {"query": "claude"}
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_generate_text_only_has_no_tool_calls() -> None:
|
|
provider = make_provider(
|
|
make_response(
|
|
[SimpleNamespace(type="text", text="Hello.")],
|
|
stop_reason="end_turn",
|
|
)
|
|
)
|
|
|
|
output = provider.generate(LLMInput(messages=[Message(role=Role.USER, content="Hi")]))
|
|
|
|
assert output.content == "Hello."
|
|
assert output.tool_calls is None
|