Spaces:
Running
Running
File size: 1,637 Bytes
79df050 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | import pytest
from features.mcp.agent_bridge import MCPAgentBridge
from features.mcp.tool_models import ToolResult
class RecordingCoordinator:
def __init__(self):
self.calls = []
async def invoke(self, tool_name, arguments, *, user_id, original_message):
self.calls.append((tool_name, arguments, user_id, original_message))
return ToolResult(
name=tool_name,
message=f"{tool_name} ok",
data={"arguments": arguments},
)
@pytest.mark.asyncio
async def test_process_intent_executes_multiple_tool_calls_in_order():
bridge = MCPAgentBridge.__new__(MCPAgentBridge)
coordinator = RecordingCoordinator()
bridge._tool_coordinator = coordinator
result = await bridge.process_intent(
{
"type": "mcp_tool",
"tool_calls": [
{"tool_name": "weather_query", "arguments": {"city": "Taipei"}},
{"tool_name": "exchange_query", "arguments": {"from_currency": "USD", "to_currency": "TWD"}},
],
},
user_id="u1",
original_message="台北天氣和美元匯率",
chat_id="c1",
)
assert coordinator.calls == [
("weather_query", {"city": "Taipei"}, "u1", "台北天氣和美元匯率"),
("exchange_query", {"from_currency": "USD", "to_currency": "TWD"}, "u1", "台北天氣和美元匯率"),
]
assert result["tool_name"] == "multi_tool"
assert result["tool_data"]["tool_names"] == ["weather_query", "exchange_query"]
assert "weather_query ok" in result["message"]
assert "exchange_query ok" in result["message"]
|