57826f1248
fix(api/chat_manager.py): remove async from ChatHistory.add_message and on_chat_history_update fix(interface/run.py): remove async from async_get_result_and_steps refactor(utils/util.py): remove unused code and simplify sync_to_async decorator
18 lines
504 B
Python
18 lines
504 B
Python
from fastapi import APIRouter, WebSocket
|
|
|
|
from langflow.api.chat_manager import ChatManager
|
|
from langflow.utils.logger import logger
|
|
|
|
router = APIRouter()
|
|
chat_manager = ChatManager()
|
|
|
|
|
|
@router.websocket("/chat/{client_id}")
|
|
async def websocket_endpoint(client_id: str, websocket: WebSocket):
|
|
"""Websocket endpoint for chat."""
|
|
try:
|
|
await chat_manager.handle_websocket(client_id, websocket)
|
|
except Exception as e:
|
|
# Log stack trace
|
|
logger.exception(e)
|
|
raise e
|