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

refactor(chat_manager.py): use dict.pop() method to remove client_id from active_connections dictionary

Browse source
This commit is contained in:
Gabriel Almeida 2023-05-11 11:15:51 -03:00
commit 2e16618e47
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -90,7 +90,7 @@ class ChatManager:
self.active_connections[client_id] = websocket self.active_connections[client_id] = websocket
def disconnect(self, client_id: str): def disconnect(self, client_id: str):
del self.active_connections[client_id] self.active_connections.pop(client_id, None)
async def send_message(self, client_id: str, message: str): async def send_message(self, client_id: str, message: str):
websocket = self.active_connections[client_id] websocket = self.active_connections[client_id]