From da5fe778640521f102b7b54d791a9e4295771fa2 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 22 Nov 2023 21:14:42 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20chore(utils.py):=20reorder=20imp?= =?UTF-8?q?orts=20to=20group=20loguru=20and=20langflow=20imports=20togethe?= =?UTF-8?q?r=20for=20better=20readability=20=F0=9F=90=9B=20fix(utils.py):?= =?UTF-8?q?=20add=20raw=5Foutput=20parameter=20to=20get=5Fresult=5Fand=5Fs?= =?UTF-8?q?teps=20function=20call=20to=20fix=20a=20bug=20where=20raw=20out?= =?UTF-8?q?put=20was=20not=20being=20returned=20correctly=20=E2=9C=A8=20fe?= =?UTF-8?q?at(utils.py):=20add=20raw=5Foutput=20to=20the=20return=20statem?= =?UTF-8?q?ent=20of=20process=5Fgraph=20function=20to=20expose=20the=20raw?= =?UTF-8?q?=20output=20of=20get=5Fresult=5Fand=5Fsteps=20function?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/services/chat/utils.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/src/backend/langflow/services/chat/utils.py b/src/backend/langflow/services/chat/utils.py index 604f4f5a5..7970e5d89 100644 --- a/src/backend/langflow/services/chat/utils.py +++ b/src/backend/langflow/services/chat/utils.py @@ -1,8 +1,9 @@ -from langflow.api.v1.schemas import ChatMessage -from langflow.processing.base import get_result_and_steps -from langflow.interface.utils import try_setting_streaming_options from loguru import logger +from langflow.api.v1.schemas import ChatMessage +from langflow.interface.utils import try_setting_streaming_options +from langflow.processing.base import get_result_and_steps + async def process_graph( langchain_object, @@ -24,14 +25,14 @@ async def process_graph( chat_inputs.message = {} logger.debug("Generating result and thought") - result, intermediate_steps = await get_result_and_steps( + result, intermediate_steps, raw_output = await get_result_and_steps( langchain_object, chat_inputs.message, client_id=client_id, session_id=session_id, ) logger.debug("Generated result and intermediate_steps") - return result, intermediate_steps + return result, intermediate_steps, raw_output except Exception as e: # Log stack trace logger.exception(e)