refactor: move MCP lifecycle out of AgentLoop (#5343)

2026-08-15 08:39:16 +03:00 · 2026-08-12 17:51:04 +08:00
parent 686dd0603e
commit 19997d20bb
39 changed files with 1192 additions and 846 deletions
@@ -32,8 +32,7 @@ AUTH_HEADERS = {"Authorization": f"Bearer {API_KEY}"}
 def _make_mock_agent(response_text: str = "mock response") -> MagicMock:
    agent = MagicMock()
    agent.process_direct = AsyncMock(return_value=response_text)
-    agent._connect_mcp = AsyncMock()
-    agent.close_mcp = AsyncMock()
+    agent.aclose = AsyncMock()
    agent._last_usage = {}
    return agent