diff --git a/src/backend/langflow/interface/custom/custom_component.py b/src/backend/langflow/interface/custom/custom_component.py index 79ca64bdc..1cc0ca620 100644 --- a/src/backend/langflow/interface/custom/custom_component.py +++ b/src/backend/langflow/interface/custom/custom_component.py @@ -147,10 +147,10 @@ class CustomComponent(Component, extra=Extra.allow): raise ValueError(f"Flow {flow_id} not found") return build_sorted_vertices_with_caching(data_graph) - def list_flow_names(self): + def list_flows(self): with session_getter() as session: flows = session.query(Flow).all() - return [flow.name for flow in flows] + return flows def build(self): raise NotImplementedError