Clean up all server connections.
(self)
| 485 | raise RuntimeError(f"Tool execution failed: {e}") |
| 486 | |
| 487 | async def cleanup(self): |
| 488 | """Clean up all server connections.""" |
| 489 | logger.info("Cleaning up MCP connections...") |
| 490 | await self.exit_stack.aclose() |
| 491 | self.sessions.clear() |
| 492 | self.tool_to_server.clear() |
| 493 | self._initialized = False |
| 494 | logger.info("MCP cleanup complete") |