From f6bc152dabfb4fe6fd0524e6705dbb6f9edc2fce Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Tue, 9 May 2023 12:37:15 -0300 Subject: [PATCH] fix(chat_manager.py): add missing import statement feat(chat_manager.py): add chat message to chat history when received and sent --- src/backend/langflow/api/chat_manager.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/api/chat_manager.py b/src/backend/langflow/api/chat_manager.py index 73d9ad6da..0ea430688 100644 --- a/src/backend/langflow/api/chat_manager.py +++ b/src/backend/langflow/api/chat_manager.py @@ -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)