From 8dd3cccfbe1282a0b520bbdc1f7ab3ad8527c305 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 27 Feb 2024 22:37:38 -0300 Subject: [PATCH] Refactor run_graph function to generate session_id if not provided --- src/backend/langflow/processing/process.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/processing/process.py b/src/backend/langflow/processing/process.py index aa419b962..0ab6fcadd 100644 --- a/src/backend/langflow/processing/process.py +++ b/src/backend/langflow/processing/process.py @@ -277,12 +277,19 @@ async def run_graph( ): """Run the graph and generate the result""" if isinstance(graph, dict): + graph_data = graph graph = Graph.from_payload(graph, flow_id=flow_id) + else: + graph_data = graph._graph_data + if not session_id: + session_id = session_service.generate_key( + session_id=flow_id, data_graph=graph_data + ) outputs = await graph.run(inputs) if session_id and session_service: session_service.update_session(session_id, (graph, artifacts)) - return outputs + return outputs, session_id def validate_input(