From 490c114159ebb83c948889b09138d5e41087cba6 Mon Sep 17 00:00:00 2001 From: Clare Liguori Date: Sun, 23 Mar 2025 12:54:22 -0700 Subject: [PATCH] Chatbot example: clean up servers in correct order --- .../clients/simple-chatbot/mcp_simple_chatbot/main.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/examples/clients/simple-chatbot/mcp_simple_chatbot/main.py b/examples/clients/simple-chatbot/mcp_simple_chatbot/main.py index 30bca722..82a2792d 100644 --- a/examples/clients/simple-chatbot/mcp_simple_chatbot/main.py +++ b/examples/clients/simple-chatbot/mcp_simple_chatbot/main.py @@ -282,13 +282,9 @@ def __init__(self, servers: list[Server], llm_client: LLMClient) -> None: async def cleanup_servers(self) -> None: """Clean up all servers properly.""" - cleanup_tasks = [] - for server in self.servers: - cleanup_tasks.append(asyncio.create_task(server.cleanup())) - - if cleanup_tasks: + for server in reversed(self.servers): try: - await asyncio.gather(*cleanup_tasks, return_exceptions=True) + await server.cleanup() except Exception as e: logging.warning(f"Warning during final cleanup: {e}")