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

🐛 fix(custom_component.py): fix session context and add error handling for flow not found

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-07-20 15:38:22 -03:00
commit 7193bb4488
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/langflow/interface/custom/custom_component.py
View file

@ -133,8 +133,10 @@ class CustomComponent(Component):
def load_flow(self, flow_id: UUID = None):
from langflow.processing.process import build_sorted_vertices_with_caching
session = next(get_session())
data_graph = flow.data if (flow := session.get(Flow, flow_id)) else None
with get_session() as session:
data_graph = flow.data if (flow := session.get(Flow, flow_id)) else None
if not data_graph:
raise ValueError(f"Flow {flow_id} not found")
return build_sorted_vertices_with_caching(data_graph)
def build(self):