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