Teknium
3340bbbdad
* feat(a2a): outbound client tools are config-gated — served only when a2a_agents configured, inbound platform enabled, or A2A_PORT set (-561 tok/call on unconfigured installs) * ci: retrigger after runner startup_failure on rerun attempt
55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
"""Regression tests for A2A client-tool schema registration."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
|
|
from plugins.platforms.a2a import tools as a2a_tools
|
|
from tools import tool_search
|
|
from tools.registry import ToolRegistry
|
|
|
|
|
|
def test_a2a_call_schema_round_trips_through_tool_describe(monkeypatch):
|
|
registry = ToolRegistry()
|
|
|
|
# The client tools are config-gated now (test_a2a_tools_gate.py):
|
|
# open the gate the way a real install would — configure a peer.
|
|
monkeypatch.setattr(
|
|
a2a_tools,
|
|
"_load_config",
|
|
lambda: {"a2a_agents": {"peer": {"url": "http://localhost:9999"}}},
|
|
)
|
|
|
|
class _Context:
|
|
def register_tool(self, name, toolset, schema, handler, **kwargs):
|
|
registry.register(
|
|
name=name,
|
|
toolset=toolset,
|
|
schema=schema,
|
|
handler=handler,
|
|
**kwargs,
|
|
)
|
|
|
|
a2a_tools.register_tools(_Context())
|
|
definitions = registry.get_definitions({"a2a_call"})
|
|
monkeypatch.setattr(
|
|
tool_search,
|
|
"is_deferrable_tool_name",
|
|
lambda name: name == "a2a_call",
|
|
)
|
|
|
|
described = json.loads(
|
|
tool_search.dispatch_tool_describe(
|
|
{"names": ["a2a_call"]},
|
|
current_tool_defs=definitions,
|
|
)
|
|
)["tools"]["a2a_call"]
|
|
|
|
assert described["description"]
|
|
assert described["parameters"]["required"] == ["agent", "message"]
|
|
assert set(described["parameters"]["properties"]) == {
|
|
"agent",
|
|
"message",
|
|
"context_id",
|
|
}
|