From df9312dfca17d4cdbbcd193f8fde571602211ea7 Mon Sep 17 00:00:00 2001 From: Yizhou Date: Tue, 15 Apr 2025 18:12:54 +0800 Subject: [PATCH] remove session close --- app/tool/mcp.py | 12 +----------- 1 file changed, 1 insertion(+), 11 deletions(-) diff --git a/app/tool/mcp.py b/app/tool/mcp.py index 86cec9d..94d3a68 100644 --- a/app/tool/mcp.py +++ b/app/tool/mcp.py @@ -131,18 +131,8 @@ class MCPClients(ToolCollection): if server_id in self.sessions: try: exit_stack = self.exit_stacks.get(server_id) - session = self.sessions.get(server_id) - # First close the session if it exists - if session: - try: - await session.close() - except Exception as e: - logger.warning( - f"Error closing session for {server_id}: {e}" - ) - - # Then close the exit stack + # Close the exit stack which will handle session cleanup if exit_stack: try: await exit_stack.aclose()