Chatbot example: clean up servers in correct order (#353)
This commit is contained in:
parent
0e2a34189c
commit
4cd4fe006c
1 changed files with 2 additions and 5 deletions
|
|
@ -284,12 +284,9 @@ class ChatSession:
|
|||
|
||||
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}")
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue