jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor(chat_manager.py): close connection in finally block to prevent memory leaks and improve code readability

Browse source
This commit is contained in:
Gabriel Almeida 2023-04-25 23:17:45 -03:00
commit 5894f32ab0
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/langflow/api/chat_manager.py
View file

@ -166,11 +166,14 @@ class ChatManager:
await self.process_message(client_id, payload) await self.process_message(client_id, payload)
except Exception as e: except Exception as e:
# Handle any exceptions that might occur # Handle any exceptions that might occur
print(f"Error: {e}") logger.exception(e)
# send a message to the client # send a message to the client
await self.send_message(client_id, f"Error: {e}") await self.send_message(client_id, f"Error: {e}")
raise e raise e
finally: finally:
await self.active_connections[client_id].close(
code=1000, reason="Client disconnected"
)
self.disconnect(client_id) self.disconnect(client_id)