From 5894f32ab047dd374c896b8b8bb1dd0a840a98fc Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Tue, 25 Apr 2023 23:17:45 -0300 Subject: [PATCH] refactor(chat_manager.py): close connection in finally block to prevent memory leaks and improve code readability --- src/backend/langflow/api/chat_manager.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/api/chat_manager.py b/src/backend/langflow/api/chat_manager.py index ca7c135f5..4ee5ea760 100644 --- a/src/backend/langflow/api/chat_manager.py +++ b/src/backend/langflow/api/chat_manager.py @@ -166,11 +166,14 @@ class ChatManager: await self.process_message(client_id, payload) except Exception as e: # Handle any exceptions that might occur - print(f"Error: {e}") + logger.exception(e) # send a message to the client await self.send_message(client_id, f"Error: {e}") raise e finally: + await self.active_connections[client_id].close( + code=1000, reason="Client disconnected" + ) self.disconnect(client_id)