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

fix(chat_manager.py): close websocket connection with status code 1011 and error message on exception

Browse source
This commit is contained in:
Gabriel Almeida 2023-05-10 12:09:14 -03:00
commit 6b6ac17167
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -3,7 +3,7 @@ import json
from collections import defaultdict from collections import defaultdict
from typing import Dict, List from typing import Dict, List
from fastapi import WebSocket from fastapi import WebSocket, status
from langflow.api.schemas import ChatMessage, ChatResponse, FileResponse from langflow.api.schemas import ChatMessage, ChatResponse, FileResponse
from langflow.cache import cache_manager from langflow.cache import cache_manager
@ -176,7 +176,9 @@ class ChatManager:
# Handle any exceptions that might occur # Handle any exceptions that might occur
logger.exception(e) logger.exception(e)
# send a message to the client # send a message to the client
raise e await self.active_connections[client_id].close(
code=status.WS_1011_INTERNAL_ERROR, reason=str(e)
)
finally: finally:
try: try:
await self.active_connections[client_id].close( await self.active_connections[client_id].close(