From 695bc4c9bf38ebe7f1f8f8299cdc8de392182f32 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 28 Jun 2023 12:01:36 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(manager.py):=20rename=20vari?= =?UTF-8?q?ables=20in=20ChatManager=20class=20to=20improve=20clarity=20and?= =?UTF-8?q?=20consistency=20=F0=9F=90=9B=20fix(utils.py):=20rename=20varia?= =?UTF-8?q?ble=20in=20process=5Fgraph=20function=20to=20improve=20clarity?= =?UTF-8?q?=20and=20consistency?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit In the `ChatManager` class, the variable `chat_message` has been renamed to `chat_inputs` to better reflect its purpose of storing the input messages from the chat payload. This improves clarity and consistency within the class. In the `process_graph` function in `utils.py`, the variable `chat_message` has been renamed to `chat_inputs` to better reflect its purpose of storing the input messages from the chat payload. This improves clarity and consistency within the function. --- src/backend/langflow/chat/manager.py | 8 ++++---- src/backend/langflow/chat/utils.py | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/src/backend/langflow/chat/manager.py b/src/backend/langflow/chat/manager.py index 4a1b8e77c..856ee226e 100644 --- a/src/backend/langflow/chat/manager.py +++ b/src/backend/langflow/chat/manager.py @@ -111,9 +111,9 @@ class ChatManager: self, client_id: str, payload: Dict, langchain_object: Any ): # Process the graph data and chat message - chat_message = payload.pop("message", "") - chat_message = ChatMessage(message=chat_message) - self.chat_history.add_message(client_id, chat_message) + chat_inputs = payload.pop("inputs", "") + chat_inputs = ChatMessage(message=chat_inputs) + self.chat_history.add_message(client_id, chat_inputs) # graph_data = payload start_resp = ChatResponse(message=None, type="start", intermediate_steps="") @@ -126,7 +126,7 @@ class ChatManager: result, intermediate_steps = await process_graph( langchain_object=langchain_object, - chat_message=chat_message, + chat_inputs=chat_inputs, websocket=self.active_connections[client_id], ) except Exception as e: diff --git a/src/backend/langflow/chat/utils.py b/src/backend/langflow/chat/utils.py index 2e2ee367f..5047438f2 100644 --- a/src/backend/langflow/chat/utils.py +++ b/src/backend/langflow/chat/utils.py @@ -7,7 +7,7 @@ from langflow.utils.logger import logger async def process_graph( langchain_object, - chat_message: ChatMessage, + chat_inputs: ChatMessage, websocket: WebSocket, ): langchain_object = try_setting_streaming_options(langchain_object, websocket) @@ -23,7 +23,7 @@ async def process_graph( try: logger.debug("Generating result and thought") result, intermediate_steps = await get_result_and_steps( - langchain_object, chat_message.message or "", websocket=websocket + langchain_object, chat_inputs.message or "", websocket=websocket ) logger.debug("Generated result and intermediate_steps") return result, intermediate_steps