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

fix(chat_manager.py): add missing import statement

Browse source 
feat(chat_manager.py): add chat message to chat history when received and sent
This commit is contained in:
Gabriel Almeida 2023-05-09 12:37:15 -03:00
commit f6bc152dab
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -5,7 +5,7 @@ from typing import Dict, List
from fastapi import WebSocket from fastapi import WebSocket
from langflow.api.callback import StreamingLLMCallbackHandler
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
from langflow.cache.manager import Subject from langflow.cache.manager import Subject
@ -107,6 +107,7 @@ class ChatManager:
chat_message = payload.pop("message", "") chat_message = payload.pop("message", "")
chat_message = ChatMessage(message=chat_message) chat_message = ChatMessage(message=chat_message)
await self.send_json(client_id, chat_message) await self.send_json(client_id, chat_message)
self.chat_history.add_message(client_id, chat_message)
graph_data = payload graph_data = payload
start_resp = ChatResponse(message=None, type="start", intermediate_steps="") start_resp = ChatResponse(message=None, type="start", intermediate_steps="")
@ -150,6 +151,7 @@ class ChatManager:
files=file_responses, files=file_responses,
) )
await self.send_json(client_id, response) 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): async def handle_websocket(self, client_id: str, websocket: WebSocket):
await self.connect(client_id, websocket) await self.connect(client_id, websocket)