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:
|
async def cleanup_servers(self) -> None:
|
||||||
"""Clean up all servers properly."""
|
"""Clean up all servers properly."""
|
||||||
cleanup_tasks = [
|
for server in reversed(self.servers):
|
||||||
asyncio.create_task(server.cleanup()) for server in self.servers
|
|
||||||
]
|
|
||||||
if cleanup_tasks:
|
|
||||||
try:
|
try:
|
||||||
await asyncio.gather(*cleanup_tasks, return_exceptions=True)
|
await server.cleanup()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logging.warning(f"Warning during final cleanup: {e}")
|
logging.warning(f"Warning during final cleanup: {e}")
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue