fix(chat_manager.py): add missing import statement
feat(chat_manager.py): add chat message to chat history when received and sent
This commit is contained in:
parent
956b28b6e9
commit
f6bc152dab
1 changed files with 3 additions and 1 deletions
|
|
@ -5,7 +5,7 @@ from typing import Dict, List
|
|||
|
||||
from fastapi import WebSocket
|
||||
|
||||
from langflow.api.callback import StreamingLLMCallbackHandler
|
||||
|
||||
from langflow.api.schemas import ChatMessage, ChatResponse, FileResponse
|
||||
from langflow.cache import cache_manager
|
||||
from langflow.cache.manager import Subject
|
||||
|
|
@ -107,6 +107,7 @@ class ChatManager:
|
|||
chat_message = payload.pop("message", "")
|
||||
chat_message = ChatMessage(message=chat_message)
|
||||
await self.send_json(client_id, chat_message)
|
||||
self.chat_history.add_message(client_id, chat_message)
|
||||
|
||||
graph_data = payload
|
||||
start_resp = ChatResponse(message=None, type="start", intermediate_steps="")
|
||||
|
|
@ -150,6 +151,7 @@ class ChatManager:
|
|||
files=file_responses,
|
||||
)
|
||||
await self.send_json(client_id, response)
|
||||
self.chat_history.add_message(client_id, chat_message)
|
||||
|
||||
async def handle_websocket(self, client_id: str, websocket: WebSocket):
|
||||
await self.connect(client_id, websocket)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue