From 89880b14a09af8054806e97ddb4de9847c29bf7c Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 29 Sep 2023 10:24:40 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20fix(schemas.py):=20add=20optiona?= =?UTF-8?q?l=20chatKey=20field=20to=20ChatMessage=20schema=20to=20support?= =?UTF-8?q?=20additional=20metadata=20for=20chat=20messages=20=F0=9F=94=A7?= =?UTF-8?q?=20fix(manager.py):=20add=20support=20for=20chatKey=20field=20i?= =?UTF-8?q?n=20ChatService=20to=20store=20additional=20metadata=20for=20ch?= =?UTF-8?q?at=20messages?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/api/v1/schemas.py | 1 + src/backend/langflow/services/chat/manager.py | 3 ++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/api/v1/schemas.py b/src/backend/langflow/api/v1/schemas.py index 1ccb79063..9a4df7c4f 100644 --- a/src/backend/langflow/api/v1/schemas.py +++ b/src/backend/langflow/api/v1/schemas.py @@ -78,6 +78,7 @@ class ChatMessage(BaseModel): is_bot: bool = False message: Union[str, None, dict] = None + chatKey: Optional[str] = None type: str = "human" diff --git a/src/backend/langflow/services/chat/manager.py b/src/backend/langflow/services/chat/manager.py index 2a1bb5694..ee4e6a9fc 100644 --- a/src/backend/langflow/services/chat/manager.py +++ b/src/backend/langflow/services/chat/manager.py @@ -126,7 +126,8 @@ class ChatService(Service): ): # Process the graph data and chat message chat_inputs = payload.pop("inputs", {}) - chat_inputs = ChatMessage(message=chat_inputs) + chatkey = payload.pop("chatKey", None) + chat_inputs = ChatMessage(message=chat_inputs, chatKey=chatkey) self.chat_history.add_message(client_id, chat_inputs) # graph_data = payload