From 031fb8234a62fc88e1d309b3bc060b4236974154 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 --- examples/clients/simple-chatbot/mcp_simple_chatbot/main.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/examples/clients/simple-chatbot/mcp_simple_chatbot/main.py b/examples/clients/simple-chatbot/mcp_simple_chatbot/main.py index ef72d78f9..cc76a1635 100644 --- a/examples/clients/simple-chatbot/mcp_simple_chatbot/main.py +++ b/examples/clients/simple-chatbot/mcp_simple_chatbot/main.py @@ -284,12 +284,9 @@ def __init__(self, servers: list[Server], llm_client: LLMClient) -> None: async def cleanup_servers(self) -> None: """Clean up all servers properly.""" - cleanup_tasks = [ - asyncio.create_task(server.cleanup()) for server in self.servers - ] - 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}")