jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor run_graph function to generate session_id if not provided

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-27 22:37:38 -03:00
commit 8dd3cccfbe
1 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/backend/langflow/processing/process.py
View file

@ -277,12 +277,19 @@ async def run_graph(
): ):
"""Run the graph and generate the result""" """Run the graph and generate the result"""
if isinstance(graph, dict): if isinstance(graph, dict):
graph_data = graph
graph = Graph.from_payload(graph, flow_id=flow_id) 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) outputs = await graph.run(inputs)
if session_id and session_service: if session_id and session_service:
session_service.update_session(session_id, (graph, artifacts)) session_service.update_session(session_id, (graph, artifacts))
return outputs return outputs, session_id
def validate_input( def validate_input(