wehub-resource-sync
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
import pytest
|
|
|
|
from free_claude_code.core.anthropic.stream_contracts import (
|
|
assert_anthropic_stream_contract,
|
|
has_tool_use,
|
|
)
|
|
from smoke.lib.config import SmokeConfig
|
|
from smoke.lib.http import collect_message_stream, message_payload
|
|
from smoke.lib.server import start_server
|
|
from smoke.lib.skips import (
|
|
skip_if_upstream_unavailable_events,
|
|
skip_if_upstream_unavailable_exception,
|
|
)
|
|
|
|
pytestmark = [pytest.mark.live, pytest.mark.smoke_target("tools")]
|
|
|
|
|
|
def test_live_tool_use_when_configured_model_supports_tools(
|
|
smoke_config: SmokeConfig,
|
|
) -> None:
|
|
models = smoke_config.provider_models()
|
|
if not models:
|
|
pytest.skip("no configured provider model available for tool-use smoke")
|
|
provider_model = models[0]
|
|
|
|
payload = message_payload(
|
|
"Use the echo_smoke tool once with value FCC_SMOKE_TOOL.",
|
|
model="fcc-smoke-default",
|
|
max_tokens=256,
|
|
extra={
|
|
"tools": [
|
|
{
|
|
"name": "echo_smoke",
|
|
"description": "Echo a test value.",
|
|
"input_schema": {
|
|
"type": "object",
|
|
"properties": {"value": {"type": "string"}},
|
|
"required": ["value"],
|
|
},
|
|
}
|
|
],
|
|
"tool_choice": {"type": "tool", "name": "echo_smoke"},
|
|
},
|
|
)
|
|
|
|
with start_server(
|
|
smoke_config,
|
|
env_overrides={
|
|
"MODEL": provider_model.full_model,
|
|
"MESSAGING_PLATFORM": "none",
|
|
},
|
|
name="tools",
|
|
) as server:
|
|
try:
|
|
events = collect_message_stream(server, payload, smoke_config)
|
|
except Exception as exc:
|
|
skip_if_upstream_unavailable_exception(exc)
|
|
raise
|
|
skip_if_upstream_unavailable_events(events)
|
|
assert_anthropic_stream_contract(events)
|
|
assert has_tool_use(events), "model did not emit a tool_use block"
|