fix: store original MCP tool name, make close_mcp public

This commit is contained in:
Re-bin
2026-02-15 07:00:27 +00:00
parent 54d5f637e7
commit 52cf1da30a
4 changed files with 41 additions and 9 deletions

View File

@@ -405,7 +405,7 @@ def gateway(
except KeyboardInterrupt:
console.print("\nShutting down...")
finally:
await agent._close_mcp()
await agent.close_mcp()
heartbeat.stop()
cron.stop()
agent.stop()
@@ -473,7 +473,7 @@ def agent(
with _thinking_ctx():
response = await agent_loop.process_direct(message, session_id)
_print_agent_response(response, render_markdown=markdown)
await agent_loop._close_mcp()
await agent_loop.close_mcp()
asyncio.run(run_once())
else:
@@ -515,7 +515,7 @@ def agent(
console.print("\nGoodbye!")
break
finally:
await agent_loop._close_mcp()
await agent_loop.close_mcp()
asyncio.run(run_interactive())