From 71bb7c18e4f8b0a0dbb0d6a975a1d618d8a6dcde Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 14 Jan 2024 17:50:03 -0300 Subject: [PATCH] Refactor process_graph_cached function --- src/backend/langflow/processing/process.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/processing/process.py b/src/backend/langflow/processing/process.py index cf3e08d1f..2fa14e5e5 100644 --- a/src/backend/langflow/processing/process.py +++ b/src/backend/langflow/processing/process.py @@ -7,14 +7,13 @@ from langchain.schema import AgentAction, Document from langchain.vectorstores.base import VectorStore from langchain_core.messages import AIMessage from langchain_core.runnables.base import Runnable -from loguru import logger -from pydantic import BaseModel - from langflow.graph.graph.base import Graph from langflow.interface.custom.custom_component import CustomComponent from langflow.interface.run import build_sorted_vertices, get_memory_key, update_memory_keys from langflow.services.deps import get_session_service from langflow.services.session.service import SessionService +from loguru import logger +from pydantic import BaseModel def fix_memory_inputs(langchain_object): @@ -224,7 +223,9 @@ async def process_graph_cached( if not graph: raise ValueError("Graph not found in the session") - result = await build_graph_and_generate_result(graph, session_id, inputs, artifacts, session_service) + result = await build_graph_and_generate_result( + graph=graph, session_id=session_id, inputs=inputs, artifacts=artifacts, session_service=session_service + ) return result