freemo
2980b14e7b
Add ~60 new behave scenarios across output rendering, tool router, file ops, and skill search features targeting uncovered lines and branches. Key additions: - ElementHandle validation guards (empty id, type, negative index, None args) - Handle close/context-manager edge cases - Table list-row and batch-row coverage - Color/box-draw/JSON/YAML materializer edge cases - Format selection paths (detect capabilities, explicit flag, empty format) - Session ColumnDef, double-close guard, force-close open handles - Tool router schema export and provider format scenarios - File ops edge cases and search stat-failure/glob-include tests All nox checks pass: lint, typecheck (0 errors), unit_tests (4730 scenarios), coverage_report (97.0% >= 97% threshold).
1019 lines
34 KiB
Python
1019 lines
34 KiB
Python
"""Step implementations for tool_router.feature."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
from typing import Any
|
|
|
|
from behave import given, then, when
|
|
from behave.runner import Context
|
|
|
|
from cleveragents.domain.models.core.tool import ToolCapability
|
|
from cleveragents.tool.registry import ToolRegistry
|
|
from cleveragents.tool.router import (
|
|
NormalizedToolCallResult,
|
|
ProviderFormat,
|
|
StreamingStatus,
|
|
StreamingToolUpdate,
|
|
ToolCallRouter,
|
|
classify_tool_error,
|
|
detect_provider_format,
|
|
generate_tool_call_id,
|
|
normalize_tool_call,
|
|
normalize_tool_schema_for_provider,
|
|
)
|
|
from cleveragents.tool.runner import ToolRunner
|
|
from cleveragents.tool.runtime import ToolResult, ToolSpec
|
|
|
|
# -- Helpers ----------------------------------------------------------------
|
|
|
|
|
|
def _echo_handler(inputs: dict[str, Any]) -> dict[str, Any]:
|
|
return dict(inputs)
|
|
|
|
|
|
def _fail_handler(inputs: dict[str, Any]) -> dict[str, Any]:
|
|
raise RuntimeError("handler exploded")
|
|
|
|
|
|
def _validation_pass_handler(inputs: dict[str, Any]) -> dict[str, Any]:
|
|
return {"passed": True, "message": "All checks passed"}
|
|
|
|
|
|
def _validation_fail_handler(inputs: dict[str, Any]) -> dict[str, Any]:
|
|
return {"passed": False, "message": "Check failed"}
|
|
|
|
|
|
def _make_echo_spec(name: str) -> ToolSpec:
|
|
return ToolSpec(
|
|
name=name,
|
|
description="Echo tool for testing",
|
|
handler=_echo_handler,
|
|
)
|
|
|
|
|
|
def _make_fail_spec(name: str) -> ToolSpec:
|
|
return ToolSpec(
|
|
name=name,
|
|
description="Failing tool for testing",
|
|
handler=_fail_handler,
|
|
)
|
|
|
|
|
|
def _make_validation_pass_spec(name: str) -> ToolSpec:
|
|
return ToolSpec(
|
|
name=name,
|
|
description="Passing validation tool",
|
|
handler=_validation_pass_handler,
|
|
capabilities=ToolCapability(read_only=True, writes=False),
|
|
output_schema={"validation_mode": "required"},
|
|
)
|
|
|
|
|
|
def _make_validation_fail_spec(name: str) -> ToolSpec:
|
|
return ToolSpec(
|
|
name=name,
|
|
description="Failing validation tool",
|
|
handler=_validation_fail_handler,
|
|
capabilities=ToolCapability(read_only=True, writes=False),
|
|
output_schema={"validation_mode": "required"},
|
|
)
|
|
|
|
|
|
# -- Provider Format Detection -----------------------------------------------
|
|
|
|
|
|
@given("a tool call payload with name \"{name}\" and arguments '{args}'")
|
|
def step_payload_openai(context: Context, name: str, args: str) -> None:
|
|
context.payload = {"name": name, "arguments": args}
|
|
|
|
|
|
@given('a tool call payload with name "{name}" and input {input_dict}')
|
|
def step_payload_anthropic(context: Context, name: str, input_dict: str) -> None:
|
|
context.payload = {"name": name, "input": json.loads(input_dict)}
|
|
|
|
|
|
@given(
|
|
'a tool call payload with name "{name}" and type "{type_val}" and args {args_dict}'
|
|
)
|
|
def step_payload_langchain(
|
|
context: Context, name: str, type_val: str, args_dict: str
|
|
) -> None:
|
|
context.payload = {"name": name, "type": type_val, "args": json.loads(args_dict)}
|
|
|
|
|
|
@given('a tool call payload with name "{name}" and args key only')
|
|
def step_payload_langchain_args_only(context: Context, name: str) -> None:
|
|
context.payload = {"name": name, "args": {"key": "value"}}
|
|
|
|
|
|
@given("an empty tool call payload")
|
|
def step_empty_payload(context: Context) -> None:
|
|
context.payload = {}
|
|
|
|
|
|
@given('a tool call payload with name "{name}" and dict arguments {args_dict}')
|
|
def step_payload_openai_dict_args(context: Context, name: str, args_dict: str) -> None:
|
|
context.payload = {"name": name, "arguments": json.loads(args_dict)}
|
|
|
|
|
|
@given('a tool call payload with name "{name}" and parameters key {params_dict}')
|
|
def step_payload_unknown_with_params(
|
|
context: Context, name: str, params_dict: str
|
|
) -> None:
|
|
context.payload = {"name": name, "parameters": json.loads(params_dict)}
|
|
|
|
|
|
@given("a tool call payload without a name")
|
|
def step_payload_no_name(context: Context) -> None:
|
|
context.payload = {"arguments": "{}"}
|
|
|
|
|
|
@when("I detect the provider format")
|
|
def step_detect_format(context: Context) -> None:
|
|
context.detected_format = detect_provider_format(context.payload)
|
|
|
|
|
|
@then('the detected format should be "{expected}"')
|
|
def step_check_format(context: Context, expected: str) -> None:
|
|
assert context.detected_format.value == expected, (
|
|
f"Expected {expected}, got {context.detected_format.value}"
|
|
)
|
|
|
|
|
|
# -- Payload Normalization ---------------------------------------------------
|
|
|
|
|
|
@when("I normalize the tool call")
|
|
def step_normalize(context: Context) -> None:
|
|
context.normalized = normalize_tool_call(context.payload)
|
|
|
|
|
|
@when("I try to normalize the tool call")
|
|
def step_try_normalize(context: Context) -> None:
|
|
try:
|
|
context.normalized = normalize_tool_call(context.payload)
|
|
context.raised_error = None
|
|
except ValueError as exc:
|
|
context.raised_error = exc
|
|
|
|
|
|
@when("I try to normalize a non-dict payload")
|
|
def step_try_normalize_non_dict(context: Context) -> None:
|
|
try:
|
|
normalize_tool_call("not a dict") # type: ignore[arg-type]
|
|
context.raised_error = None
|
|
except ValueError as exc:
|
|
context.raised_error = exc
|
|
|
|
|
|
@then('the normalized request tool_name should be "{expected}"')
|
|
def step_check_normalized_name(context: Context, expected: str) -> None:
|
|
assert context.normalized.tool_name == expected
|
|
|
|
|
|
@then('the normalized request arguments should have key "{key}"')
|
|
def step_check_normalized_args_key(context: Context, key: str) -> None:
|
|
assert key in context.normalized.arguments
|
|
|
|
|
|
@then('the normalized request provider_format should be "{expected}"')
|
|
def step_check_normalized_format(context: Context, expected: str) -> None:
|
|
assert context.normalized.provider_format.value == expected
|
|
|
|
|
|
@then('a router ValueError should be raised containing "{text}"')
|
|
def step_check_router_value_error(context: Context, text: str) -> None:
|
|
assert context.raised_error is not None, "Expected ValueError but none was raised"
|
|
assert text.lower() in str(context.raised_error).lower(), (
|
|
f"Expected '{text}' in error message, got: {context.raised_error}"
|
|
)
|
|
|
|
|
|
# -- Stable ID Generation ---------------------------------------------------
|
|
|
|
|
|
@when('I generate a tool call ID for plan "{plan}" sequence {seq:d}')
|
|
def step_generate_id(context: Context, plan: str, seq: int) -> None:
|
|
context.tool_call_id = generate_tool_call_id(plan, seq)
|
|
|
|
|
|
@when('I generate another tool call ID for plan "{plan}" sequence {seq:d}')
|
|
def step_generate_another_id(context: Context, plan: str, seq: int) -> None:
|
|
context.tool_call_id_2 = generate_tool_call_id(plan, seq)
|
|
|
|
|
|
@then('the tool call ID should start with "{prefix}"')
|
|
def step_check_id_prefix(context: Context, prefix: str) -> None:
|
|
assert context.tool_call_id.startswith(prefix), (
|
|
f"Expected prefix '{prefix}', got: {context.tool_call_id}"
|
|
)
|
|
|
|
|
|
@then("the tool call ID should have length {length:d}")
|
|
def step_check_id_length(context: Context, length: int) -> None:
|
|
assert len(context.tool_call_id) == length, (
|
|
f"Expected length {length}, got {len(context.tool_call_id)}"
|
|
)
|
|
|
|
|
|
@then("both tool call IDs should be identical")
|
|
def step_check_ids_same(context: Context) -> None:
|
|
assert context.tool_call_id == context.tool_call_id_2
|
|
|
|
|
|
@then("the tool call IDs should differ")
|
|
def step_check_ids_differ(context: Context) -> None:
|
|
assert context.tool_call_id != context.tool_call_id_2
|
|
|
|
|
|
@when("I try to generate a tool call ID with empty plan_id")
|
|
def step_try_gen_empty_plan(context: Context) -> None:
|
|
try:
|
|
generate_tool_call_id("", 0)
|
|
context.raised_error = None
|
|
except ValueError as exc:
|
|
context.raised_error = exc
|
|
|
|
|
|
@when("I try to generate a tool call ID with negative sequence")
|
|
def step_try_gen_neg_seq(context: Context) -> None:
|
|
try:
|
|
generate_tool_call_id("plan-1", -1)
|
|
context.raised_error = None
|
|
except ValueError as exc:
|
|
context.raised_error = exc
|
|
|
|
|
|
# -- Router Execution --------------------------------------------------------
|
|
|
|
|
|
@given('a tool registry with an echo tool "{name}"')
|
|
def step_registry_with_echo(context: Context, name: str) -> None:
|
|
context.registry = ToolRegistry()
|
|
spec = _make_echo_spec(name)
|
|
context.registry.register(spec)
|
|
context.runner = ToolRunner(context.registry)
|
|
|
|
|
|
@given('a tool registry with a failing tool "{name}"')
|
|
def step_registry_with_fail(context: Context, name: str) -> None:
|
|
context.registry = ToolRegistry()
|
|
spec = _make_fail_spec(name)
|
|
context.registry.register(spec)
|
|
context.runner = ToolRunner(context.registry)
|
|
|
|
|
|
@given('a tool registry with a passing validation tool "{name}"')
|
|
def step_registry_with_passing_val(context: Context, name: str) -> None:
|
|
context.registry = ToolRegistry()
|
|
spec = _make_validation_pass_spec(name)
|
|
context.registry.register(spec)
|
|
context.runner = ToolRunner(context.registry)
|
|
|
|
|
|
@given('a tool registry with a failing validation tool "{name}"')
|
|
def step_registry_with_failing_val(context: Context, name: str) -> None:
|
|
context.registry = ToolRegistry()
|
|
spec = _make_validation_fail_spec(name)
|
|
context.registry.register(spec)
|
|
context.runner = ToolRunner(context.registry)
|
|
|
|
|
|
@given('a tool call router for plan "{plan_id}"')
|
|
def step_create_router(context: Context, plan_id: str) -> None:
|
|
registry = getattr(context, "tool_registry", None) or context.registry
|
|
context.router = ToolCallRouter(
|
|
registry=registry,
|
|
runner=context.runner,
|
|
plan_id=plan_id,
|
|
)
|
|
|
|
|
|
@given("an OpenAI-format tool call for \"{name}\" with arguments '{args}'")
|
|
def step_openai_call(context: Context, name: str, args: str) -> None:
|
|
context.payload = {"name": name, "arguments": args}
|
|
|
|
|
|
@given('an Anthropic-format tool call for "{name}" with input {input_dict}')
|
|
def step_anthropic_call(context: Context, name: str, input_dict: str) -> None:
|
|
context.payload = {"name": name, "input": json.loads(input_dict)}
|
|
|
|
|
|
@given('a LangChain-format tool call for "{name}" with args {args_dict}')
|
|
def step_langchain_call(context: Context, name: str, args_dict: str) -> None:
|
|
context.payload = {"name": name, "type": "tool_call", "args": json.loads(args_dict)}
|
|
|
|
|
|
@given('provider metadata with model "{model}" and provider_id "{pid}"')
|
|
def step_provider_metadata(context: Context, model: str, pid: str) -> None:
|
|
context.provider_metadata = {"model": model, "provider_id": pid}
|
|
|
|
|
|
@given('a batch of {count:d} OpenAI-format tool calls for "{name}"')
|
|
def step_batch_calls(context: Context, count: int, name: str) -> None:
|
|
context.batch_payloads = [
|
|
{"name": name, "arguments": json.dumps({"index": i})} for i in range(count)
|
|
]
|
|
|
|
|
|
@when("I route the tool call")
|
|
def step_route(context: Context) -> None:
|
|
context.route_result = context.router.route(context.payload)
|
|
|
|
|
|
@when("I route the tool call with provider metadata")
|
|
def step_route_with_metadata(context: Context) -> None:
|
|
context.route_result = context.router.route(
|
|
context.payload, provider_metadata=context.provider_metadata
|
|
)
|
|
|
|
|
|
@when("I route the batch")
|
|
def step_route_batch(context: Context) -> None:
|
|
context.batch_results = context.router.route_batch(context.batch_payloads)
|
|
|
|
|
|
@when("I route an empty batch")
|
|
def step_route_empty_batch(context: Context) -> None:
|
|
context.batch_results = context.router.route_batch([])
|
|
|
|
|
|
@when("I route the tool call with streaming")
|
|
def step_route_streaming(context: Context) -> None:
|
|
context.stream_items = list(context.router.route_streaming(context.payload))
|
|
|
|
|
|
@then("the normalized result should be successful")
|
|
def step_check_result_success(context: Context) -> None:
|
|
assert context.route_result.result.success, (
|
|
f"Expected success, got error: {context.route_result.result.error}"
|
|
)
|
|
|
|
|
|
@then("the normalized result should not be successful")
|
|
def step_check_result_failure(context: Context) -> None:
|
|
assert not context.route_result.result.success
|
|
|
|
|
|
@then('the normalized result tool_name should be "{expected}"')
|
|
def step_check_result_name(context: Context, expected: str) -> None:
|
|
assert context.route_result.tool_name == expected
|
|
|
|
|
|
@then('the normalized result provider_format should be "{expected}"')
|
|
def step_check_result_provider(context: Context, expected: str) -> None:
|
|
assert context.route_result.provider_format.value == expected
|
|
|
|
|
|
@then('the normalized result tool_call_id should start with "{prefix}"')
|
|
def step_check_result_id_prefix(context: Context, prefix: str) -> None:
|
|
assert context.route_result.tool_call_id.startswith(prefix)
|
|
|
|
|
|
@then('the normalized result error_category should be "{expected}"')
|
|
def step_check_result_error_cat(context: Context, expected: str) -> None:
|
|
assert context.route_result.error_category is not None
|
|
assert context.route_result.error_category.value == expected
|
|
|
|
|
|
@then('the normalized result provider_metadata should contain key "{key}"')
|
|
def step_check_result_metadata_key(context: Context, key: str) -> None:
|
|
assert key in context.route_result.provider_metadata
|
|
|
|
|
|
@then("the normalized result is_validation should be True")
|
|
def step_check_is_validation_true(context: Context) -> None:
|
|
assert context.route_result.is_validation is True
|
|
|
|
|
|
@then("the normalized result validation_passed should be True")
|
|
def step_check_validation_passed_true(context: Context) -> None:
|
|
assert context.route_result.validation_passed is True
|
|
|
|
|
|
@then("the normalized result validation_passed should be False")
|
|
def step_check_validation_passed_false(context: Context) -> None:
|
|
assert context.route_result.validation_passed is False
|
|
|
|
|
|
# -- Batch results -----------------------------------------------------------
|
|
|
|
|
|
@then("the batch should return {count:d} results")
|
|
def step_check_batch_count(context: Context, count: int) -> None:
|
|
assert len(context.batch_results) == count
|
|
|
|
|
|
@then("all batch results should be successful")
|
|
def step_check_batch_all_success(context: Context) -> None:
|
|
for r in context.batch_results:
|
|
assert r.result.success, f"Batch item failed: {r.result.error}"
|
|
|
|
|
|
# -- Streaming results -------------------------------------------------------
|
|
|
|
|
|
@then("the stream should emit a pending update")
|
|
def step_check_stream_pending(context: Context) -> None:
|
|
pending = [
|
|
i
|
|
for i in context.stream_items
|
|
if isinstance(i, StreamingToolUpdate) and i.status == StreamingStatus.PENDING
|
|
]
|
|
assert len(pending) >= 1, "No pending update found"
|
|
|
|
|
|
@then("the stream should emit a running update")
|
|
def step_check_stream_running(context: Context) -> None:
|
|
running = [
|
|
i
|
|
for i in context.stream_items
|
|
if isinstance(i, StreamingToolUpdate) and i.status == StreamingStatus.RUNNING
|
|
]
|
|
assert len(running) >= 1, "No running update found"
|
|
|
|
|
|
@then("the stream should emit a complete update")
|
|
def step_check_stream_complete(context: Context) -> None:
|
|
complete = [
|
|
i
|
|
for i in context.stream_items
|
|
if isinstance(i, StreamingToolUpdate) and i.status == StreamingStatus.COMPLETE
|
|
]
|
|
assert len(complete) >= 1, "No complete update found"
|
|
|
|
|
|
@then("the stream should emit a final result")
|
|
def step_check_stream_final(context: Context) -> None:
|
|
results = [
|
|
i for i in context.stream_items if isinstance(i, NormalizedToolCallResult)
|
|
]
|
|
assert len(results) >= 1, "No final result found"
|
|
|
|
|
|
@then("the stream should emit an error update")
|
|
def step_check_stream_error(context: Context) -> None:
|
|
errors = [
|
|
i
|
|
for i in context.stream_items
|
|
if isinstance(i, StreamingToolUpdate) and i.status == StreamingStatus.ERROR
|
|
]
|
|
assert len(errors) >= 1, "No error update found"
|
|
|
|
|
|
@then("the stream should emit a final result that is not successful")
|
|
def step_check_stream_final_failure(context: Context) -> None:
|
|
results = [
|
|
i for i in context.stream_items if isinstance(i, NormalizedToolCallResult)
|
|
]
|
|
assert len(results) >= 1, "No final result found"
|
|
assert not results[-1].result.success
|
|
|
|
|
|
# -- Error Classification ---------------------------------------------------
|
|
|
|
|
|
@when('I classify the error "{error_msg}"')
|
|
def step_classify_error(context: Context, error_msg: str) -> None:
|
|
context.error_category = classify_tool_error(error_msg)
|
|
|
|
|
|
@then('the error category should be "{expected}"')
|
|
def step_check_error_cat(context: Context, expected: str) -> None:
|
|
assert context.error_category.value == expected
|
|
|
|
|
|
# -- Schema Normalization ---------------------------------------------------
|
|
|
|
|
|
@given('a tool spec named "{name}" with description "{desc}"')
|
|
def step_tool_spec_with_desc(context: Context, name: str, desc: str) -> None:
|
|
context.tool_spec = ToolSpec(
|
|
name=name,
|
|
description=desc,
|
|
handler=_echo_handler,
|
|
)
|
|
|
|
|
|
@given('a tool spec named "{name}" with a very long description')
|
|
def step_tool_spec_long_desc(context: Context, name: str) -> None:
|
|
context.tool_spec = ToolSpec(
|
|
name=name,
|
|
description="A" * 2000,
|
|
handler=_echo_handler,
|
|
)
|
|
|
|
|
|
@when('I normalize the schema for "{provider}" provider')
|
|
def step_normalize_schema(context: Context, provider: str) -> None:
|
|
context.normalized_schema = normalize_tool_schema_for_provider(
|
|
context.tool_spec, ProviderFormat(provider)
|
|
)
|
|
|
|
|
|
@when('I normalize the schema for "{provider}" provider with max length {length:d}')
|
|
def step_normalize_schema_max(context: Context, provider: str, length: int) -> None:
|
|
context.normalized_schema = normalize_tool_schema_for_provider(
|
|
context.tool_spec, ProviderFormat(provider), max_description_length=length
|
|
)
|
|
|
|
|
|
@when("I try to normalize the schema with max_description_length {length:d}")
|
|
def step_try_normalize_schema_bad_len(context: Context, length: int) -> None:
|
|
try:
|
|
normalize_tool_schema_for_provider(
|
|
context.tool_spec, ProviderFormat.OPENAI, max_description_length=length
|
|
)
|
|
context.raised_error = None
|
|
except ValueError as exc:
|
|
context.raised_error = exc
|
|
|
|
|
|
@then('the normalized schema should have key "{key}"')
|
|
def step_check_schema_key(context: Context, key: str) -> None:
|
|
assert key in context.normalized_schema
|
|
|
|
|
|
@then('the normalized schema should have name "{name}"')
|
|
def step_check_schema_name(context: Context, name: str) -> None:
|
|
assert context.normalized_schema["name"] == name
|
|
|
|
|
|
@then("the normalized schema description should be at most {length:d} characters")
|
|
def step_check_schema_desc_len(context: Context, length: int) -> None:
|
|
assert len(context.normalized_schema["description"]) <= length
|
|
|
|
|
|
# -- Schema Export -----------------------------------------------------------
|
|
|
|
|
|
@when('I export schemas for "{provider}" provider')
|
|
def step_export_schemas(context: Context, provider: str) -> None:
|
|
context.exported_schemas = context.router.export_schemas(ProviderFormat(provider))
|
|
|
|
|
|
@then("the exported schemas should contain {count:d} schema")
|
|
def step_check_exported_count(context: Context, count: int) -> None:
|
|
assert len(context.exported_schemas) == count
|
|
|
|
|
|
@then('the first exported schema should have tool_type "{expected}"')
|
|
def step_check_exported_type(context: Context, expected: str) -> None:
|
|
assert context.exported_schemas[0]["tool_type"] == expected
|
|
|
|
|
|
# -- Sequence counter -------------------------------------------------------
|
|
|
|
|
|
@then("the router sequence should be {expected:d}")
|
|
def step_check_sequence(context: Context, expected: int) -> None:
|
|
assert context.router.sequence == expected
|
|
|
|
|
|
# -- Router construction validation -----------------------------------------
|
|
|
|
|
|
@when("I try to create a router with empty plan_id")
|
|
def step_try_create_router_empty(context: Context) -> None:
|
|
try:
|
|
ToolCallRouter(
|
|
registry=context.registry,
|
|
runner=context.runner,
|
|
plan_id="",
|
|
)
|
|
context.raised_error = None
|
|
except ValueError as exc:
|
|
context.raised_error = exc
|
|
|
|
|
|
@when("I try to route a non-dict payload")
|
|
def step_try_route_non_dict(context: Context) -> None:
|
|
try:
|
|
context.router.route("not a dict") # type: ignore[arg-type]
|
|
context.raised_error = None
|
|
except ValueError as exc:
|
|
context.raised_error = exc
|
|
|
|
|
|
@when("I try to route_batch a non-list payload")
|
|
def step_try_route_batch_non_list(context: Context) -> None:
|
|
try:
|
|
context.router.route_batch("not a list") # type: ignore[arg-type]
|
|
context.raised_error = None
|
|
except ValueError as exc:
|
|
context.raised_error = exc
|
|
|
|
|
|
# -- Detect format non-dict -------------------------------------------------
|
|
|
|
|
|
@when("I detect format for a non-dict value")
|
|
def step_detect_non_dict(context: Context) -> None:
|
|
context.detected_format = detect_provider_format("not-a-dict") # type: ignore[arg-type]
|
|
|
|
|
|
@when("I classify an empty error message")
|
|
def step_classify_empty_error(context: Context) -> None:
|
|
context.error_category = classify_tool_error("")
|
|
|
|
|
|
@given("a tool call with invalid JSON arguments")
|
|
def step_payload_invalid_json(context: Context) -> None:
|
|
context.payload = {"name": "test/echo", "arguments": "not valid json"}
|
|
|
|
|
|
# -- Coverage: provider branches and validation ---------------------------------
|
|
|
|
|
|
@given('a tool call payload with name "{name}" and dict arguments')
|
|
def step_payload_dict_args(context: Context, name: str) -> None:
|
|
context.payload = {"name": name, "arguments": {"msg": "hello"}}
|
|
|
|
|
|
@given("an OpenAI payload with integer arguments")
|
|
def step_payload_int_args(context: Context) -> None:
|
|
context.payload = {"name": "test/echo", "arguments": 42}
|
|
|
|
|
|
@when("I try to normalize the invalid openai arguments")
|
|
def step_normalize_invalid_openai(context: Context) -> None:
|
|
try:
|
|
normalize_tool_call(context.payload)
|
|
context.raised_error = None
|
|
except ValueError as exc:
|
|
context.raised_error = exc
|
|
|
|
|
|
@given("an Anthropic payload with string input")
|
|
def step_payload_anthropic_str(context: Context) -> None:
|
|
context.payload = {
|
|
"name": "test/echo",
|
|
"id": "call-1",
|
|
"type": "tool_use",
|
|
"input": "bad",
|
|
}
|
|
|
|
|
|
@when("I try to normalize the invalid anthropic input")
|
|
def step_normalize_invalid_anthropic(context: Context) -> None:
|
|
try:
|
|
normalize_tool_call(context.payload)
|
|
context.raised_error = None
|
|
except ValueError as exc:
|
|
context.raised_error = exc
|
|
|
|
|
|
@given("a LangChain payload with dict args")
|
|
def step_payload_langchain_dict(context: Context) -> None:
|
|
context.payload = {"name": "test/echo", "args": {"key": "val"}, "id": "lc-1"}
|
|
|
|
|
|
@given("a LangChain payload with list args")
|
|
def step_payload_langchain_list(context: Context) -> None:
|
|
context.payload = {"name": "test/echo", "args": [1, 2, 3], "id": "lc-2"}
|
|
|
|
|
|
@when("I try to normalize the invalid langchain args")
|
|
def step_normalize_invalid_langchain(context: Context) -> None:
|
|
try:
|
|
normalize_tool_call(context.payload)
|
|
context.raised_error = None
|
|
except ValueError as exc:
|
|
context.raised_error = exc
|
|
|
|
|
|
@given("an unknown format payload with parameters key")
|
|
def step_payload_unknown_params(context: Context) -> None:
|
|
context.payload = {"name": "test/echo", "parameters": {"query": "hello"}}
|
|
|
|
|
|
@given("an unknown format payload with JSON string in args key")
|
|
def step_payload_unknown_json_str(context: Context) -> None:
|
|
# Use "arguments" with a non-parseable string first then "input" as JSON string
|
|
# to trigger the unknown branch with a JSON string in a fallback key.
|
|
# "arguments" with int makes it not str/dict -> unknown; "input" as JSON string.
|
|
context.payload = {"name": "test/echo", "input": '{"data": "value"}'}
|
|
|
|
|
|
@given("an unknown format payload with bad JSON and valid fallback")
|
|
def step_payload_unknown_bad_json(context: Context) -> None:
|
|
# arguments=42 makes detect_provider_format return UNKNOWN.
|
|
# In normalize unknown branch: "arguments" (int) skipped,
|
|
# "input" (bad JSON str) triggers JSONDecodeError→continue,
|
|
# "parameters" (dict) is used successfully.
|
|
context.payload = {
|
|
"name": "test/echo",
|
|
"arguments": 42,
|
|
"input": "not json",
|
|
"parameters": {"ok": True},
|
|
}
|
|
|
|
|
|
@when('I normalize the tool call for "{fmt}" format')
|
|
def step_normalize_for_format(context: Context, fmt: str) -> None:
|
|
context.normalized_request = normalize_tool_call(context.payload)
|
|
|
|
|
|
@then('the normalized request should have tool name "{name}"')
|
|
def step_then_request_tool_name(context: Context, name: str) -> None:
|
|
assert context.normalized_request.tool_name == name, (
|
|
f"Expected '{name}', got '{context.normalized_request.tool_name}'"
|
|
)
|
|
|
|
|
|
@then('the normalized arguments should be a dict with key "{key}"')
|
|
def step_then_args_dict_key(context: Context, key: str) -> None:
|
|
assert key in context.normalized_request.arguments
|
|
|
|
|
|
@then('the normalized arguments should have key "{key}"')
|
|
def step_then_args_key(context: Context, key: str) -> None:
|
|
assert key in context.normalized_request.arguments
|
|
|
|
|
|
@given('a tool spec named "{name}"')
|
|
def step_given_tool_spec(context: Context, name: str) -> None:
|
|
context.tool_spec = ToolSpec(
|
|
name=name,
|
|
description="test tool",
|
|
input_schema={"type": "object", "properties": {"msg": {"type": "string"}}},
|
|
output_schema={},
|
|
capabilities=ToolCapability(read_only=False, writes=True),
|
|
handler=lambda inputs: inputs,
|
|
)
|
|
|
|
|
|
@then('the schema should have key "{key}"')
|
|
def step_then_schema_key(context: Context, key: str) -> None:
|
|
assert key in context.normalized_schema, (
|
|
f"Missing key '{key}' in {context.normalized_schema}"
|
|
)
|
|
|
|
|
|
@then('the router plan_id should be "{expected}"')
|
|
def step_then_router_plan_id(context: Context, expected: str) -> None:
|
|
assert context.router.plan_id == expected
|
|
|
|
|
|
# -- Tools that raise or fail for coverage tests ---------
|
|
|
|
|
|
@given("a tool registry with a tool that raises RuntimeError")
|
|
def step_registry_runtime_error(context: Context) -> None:
|
|
registry = ToolRegistry()
|
|
|
|
def _noop(inputs: dict[str, Any]) -> dict[str, Any]:
|
|
return {}
|
|
|
|
registry.register(
|
|
ToolSpec(
|
|
name="test/boom",
|
|
description="exploding tool",
|
|
input_schema={},
|
|
output_schema={},
|
|
capabilities=ToolCapability(read_only=True),
|
|
handler=_noop,
|
|
)
|
|
)
|
|
context.tool_registry = registry
|
|
runner = ToolRunner(registry)
|
|
|
|
# Override execute to raise RuntimeError, bypassing ToolRunner's catch
|
|
def _raising_execute(name: str, arguments: dict[str, Any]) -> ToolResult:
|
|
raise RuntimeError("boom")
|
|
|
|
runner.execute = _raising_execute # type: ignore[assignment]
|
|
context.runner = runner
|
|
|
|
|
|
@when("I route an OpenAI payload for the error tool")
|
|
def step_route_error_tool(context: Context) -> None:
|
|
context.route_result = context.router.route(
|
|
{"name": "test/boom", "arguments": "{}"}
|
|
)
|
|
|
|
|
|
@then("the route result should have success false")
|
|
def step_then_result_fail(context: Context) -> None:
|
|
assert context.route_result.result.success is False
|
|
|
|
|
|
@then('the route result error should contain "{text}"')
|
|
def step_then_result_error_contains(context: Context, text: str) -> None:
|
|
assert text in (context.route_result.result.error or "")
|
|
|
|
|
|
@given("a tool registry with a tool that returns failed result")
|
|
def step_registry_fail_result(context: Context) -> None:
|
|
registry = ToolRegistry()
|
|
|
|
def _fail(inputs: dict[str, Any]) -> dict[str, Any]:
|
|
return {"error": "tool not found: missing"}
|
|
|
|
spec = ToolSpec(
|
|
name="test/fail",
|
|
description="failing tool",
|
|
input_schema={},
|
|
output_schema={},
|
|
capabilities=ToolCapability(read_only=True),
|
|
handler=_fail,
|
|
)
|
|
registry.register(spec)
|
|
context.tool_registry = registry
|
|
|
|
runner = ToolRunner(registry)
|
|
|
|
# Override execute to return a ToolResult with success=False
|
|
def _execute_fail(name: str, arguments: dict[str, Any]) -> ToolResult:
|
|
return ToolResult(
|
|
success=False,
|
|
output={},
|
|
error="tool not found: missing",
|
|
duration_ms=1.0,
|
|
)
|
|
|
|
runner.execute = _execute_fail # type: ignore[assignment]
|
|
context.runner = runner
|
|
|
|
|
|
@when("I route an OpenAI payload for the fail tool")
|
|
def step_route_fail_tool(context: Context) -> None:
|
|
context.route_result = context.router.route(
|
|
{"name": "test/fail", "arguments": "{}"}
|
|
)
|
|
|
|
|
|
@then("the route result should have error_category set")
|
|
def step_then_error_category_set(context: Context) -> None:
|
|
assert context.route_result.error_category is not None
|
|
|
|
|
|
@given("a tool registry with a validation tool")
|
|
def step_registry_validation_tool(context: Context) -> None:
|
|
registry = ToolRegistry()
|
|
|
|
def _validate(inputs: dict[str, Any]) -> dict[str, Any]:
|
|
return {"passed": True}
|
|
|
|
registry.register(
|
|
ToolSpec(
|
|
name="validate/check",
|
|
description="validate something",
|
|
input_schema={},
|
|
output_schema={"validation_mode": "strict"},
|
|
capabilities=ToolCapability(read_only=True, writes=False),
|
|
handler=_validate,
|
|
)
|
|
)
|
|
context.tool_registry = registry
|
|
context.runner = ToolRunner(registry)
|
|
|
|
|
|
@when("I route an OpenAI payload for the validation tool")
|
|
def step_route_validation_tool(context: Context) -> None:
|
|
context.route_result = context.router.route(
|
|
{"name": "validate/check", "arguments": "{}"}
|
|
)
|
|
|
|
|
|
@then("the route result should have is_validation true")
|
|
def step_then_is_validation(context: Context) -> None:
|
|
assert context.route_result.is_validation is True
|
|
|
|
|
|
@then("the route result should have validation_passed true")
|
|
def step_then_validation_passed(context: Context) -> None:
|
|
assert context.route_result.validation_passed is True
|
|
|
|
|
|
# -- Streaming coverage ----
|
|
|
|
|
|
@when("I stream-route an OpenAI payload for the error tool")
|
|
def step_stream_route_error(context: Context) -> None:
|
|
results = list(
|
|
context.router.route_streaming({"name": "test/boom", "arguments": "{}"})
|
|
)
|
|
context.streaming_results = results
|
|
|
|
|
|
@then("the streaming updates should include ERROR status")
|
|
def step_then_streaming_error_status(context: Context) -> None:
|
|
statuses = [
|
|
r.status
|
|
for r in context.streaming_results
|
|
if isinstance(r, StreamingToolUpdate)
|
|
]
|
|
assert StreamingStatus.ERROR in statuses
|
|
|
|
|
|
@then("the streaming final result should have success false")
|
|
def step_then_streaming_fail(context: Context) -> None:
|
|
final = [
|
|
r for r in context.streaming_results if isinstance(r, NormalizedToolCallResult)
|
|
]
|
|
assert len(final) == 1
|
|
assert final[0].result.success is False
|
|
|
|
|
|
@when("I try to stream-route a non-dict payload")
|
|
def step_try_stream_non_dict(context: Context) -> None:
|
|
try:
|
|
list(context.router.route_streaming("not a dict")) # type: ignore[arg-type]
|
|
context.raised_error = None
|
|
except ValueError as exc:
|
|
context.raised_error = exc
|
|
|
|
|
|
@when("I stream-route an OpenAI payload for the validation tool")
|
|
def step_stream_route_validation(context: Context) -> None:
|
|
results = list(
|
|
context.router.route_streaming({"name": "validate/check", "arguments": "{}"})
|
|
)
|
|
context.streaming_results = results
|
|
|
|
|
|
@then("the streaming final result should have is_validation true")
|
|
def step_then_streaming_validation(context: Context) -> None:
|
|
final = [
|
|
r for r in context.streaming_results if isinstance(r, NormalizedToolCallResult)
|
|
]
|
|
assert len(final) == 1
|
|
assert final[0].is_validation is True
|
|
|
|
|
|
# -- Schema export coverage ----
|
|
|
|
|
|
@when('I export tool schemas for "{provider}"')
|
|
def step_export_schemas(context: Context, provider: str) -> None:
|
|
fmt = ProviderFormat(provider)
|
|
context.exported_schemas = context.router.export_schemas(fmt)
|
|
|
|
|
|
@then('the exported schemas should include a tool with type "{tool_type}"')
|
|
def step_then_schema_tool_type(context: Context, tool_type: str) -> None:
|
|
types = [s.get("tool_type") for s in context.exported_schemas]
|
|
assert tool_type in types, f"Expected type '{tool_type}' in {types}"
|
|
|
|
|
|
@given('a tool registry with a validation tool with mode "{mode}"')
|
|
def step_registry_validation_with_mode(context: Context, mode: str) -> None:
|
|
registry = ToolRegistry()
|
|
|
|
def _validate(inputs: dict[str, Any]) -> dict[str, Any]:
|
|
return {"passed": True}
|
|
|
|
registry.register(
|
|
ToolSpec(
|
|
name="validate/strict",
|
|
description="strict validator",
|
|
input_schema={},
|
|
output_schema={"validation_mode": mode},
|
|
capabilities=ToolCapability(read_only=True, writes=False),
|
|
handler=_validate,
|
|
)
|
|
)
|
|
context.tool_registry = registry
|
|
context.runner = ToolRunner(registry)
|
|
|
|
|
|
@then('the exported validation schema should have mode "{mode}"')
|
|
def step_then_schema_mode(context: Context, mode: str) -> None:
|
|
validation_schemas = [
|
|
s for s in context.exported_schemas if s.get("tool_type") == "validation"
|
|
]
|
|
assert len(validation_schemas) >= 1
|
|
assert validation_schemas[0].get("validation_mode") == mode
|
|
|
|
|
|
@given("a tool registry with a validation tool with numeric mode")
|
|
def step_registry_validation_numeric_mode(context: Context) -> None:
|
|
registry = ToolRegistry()
|
|
|
|
def _validate(inputs: dict[str, Any]) -> dict[str, Any]:
|
|
return {"passed": True}
|
|
|
|
registry.register(
|
|
ToolSpec(
|
|
name="validate/numeric",
|
|
description="numeric mode validator",
|
|
input_schema={},
|
|
output_schema={"validation_mode": 123},
|
|
capabilities=ToolCapability(read_only=True, writes=False),
|
|
handler=_validate,
|
|
)
|
|
)
|
|
context.tool_registry = registry
|
|
context.runner = ToolRunner(registry)
|
|
|
|
|
|
@then("the exported validation schema should not have a mode")
|
|
def step_then_schema_no_mode(context: Context) -> None:
|
|
validation_schemas = [
|
|
s for s in context.exported_schemas if s.get("tool_type") == "validation"
|
|
]
|
|
assert len(validation_schemas) >= 1
|
|
assert "validation_mode" not in validation_schemas[0]
|