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

🔄 refactor(custom_component.py): rename list_flow_names method to list_flows for better clarity and consistency

Browse source 
🐛 fix(custom_component.py): fix return value of list_flows method to return the actual flow objects instead of just their names
🔥 chore(custom_component.py): remove unused build method as it is not implemented
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-07-25 17:12:48 -03:00
commit b76141b3df
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -147,10 +147,10 @@ class CustomComponent(Component, extra=Extra.allow):
raise ValueError(f"Flow {flow_id} not found") raise ValueError(f"Flow {flow_id} not found")
return build_sorted_vertices_with_caching(data_graph) return build_sorted_vertices_with_caching(data_graph)
def list_flow_names(self): def list_flows(self):
with session_getter() as session: with session_getter() as session:
flows = session.query(Flow).all() flows = session.query(Flow).all()
return [flow.name for flow in flows] return flows
def build(self): def build(self):
raise NotImplementedError raise NotImplementedError