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

fix(chat_manager.py): handle case where client is not in active_connections dictionary and remove duplicate code in finally block

Browse source
This commit is contained in:
Gabriel Almeida 2023-05-10 12:17:52 -03:00
commit 7fde19b36a
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -4,6 +4,7 @@ from collections import defaultdict
from typing import Dict, List
from fastapi import WebSocket, status
from fastapi.websockets import WebSocketState
from langflow.api.schemas import ChatMessage, ChatResponse, FileResponse
from langflow.cache import cache_manager
@ -179,11 +180,13 @@ class ChatManager:
await self.active_connections[client_id].close(
code=status.WS_1011_INTERNAL_ERROR, reason=str(e)
)
self.disconnect(client_id)
finally:
try:
await self.active_connections[client_id].close(
code=1000, reason="Client disconnected"
)
connection = self.active_connections.get(client_id)
if connection:
await connection.close(code=1000, reason="Client disconnected")
self.disconnect(client_id)
except Exception as e:
logger.exception(e)
self.disconnect(client_id)