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

🔀 refactor(custom_component.py): modify get_flow method to accept keyword arguments for flow_name and flow_id to improve flexibility and readability

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-07-31 10:35:48 -03:00
commit 566c48166e
1 changed files with 5 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -158,15 +158,13 @@ class CustomComponent(Component, extra=Extra.allow):
flows = session.query(Flow).all() flows = session.query(Flow).all()
return flows return flows
def get_flow(self, flow_name: str) -> Flow: def get_flow(self, *, flow_name: str = None, flow_id: str = None) -> Flow:
with session_getter() as session: with session_getter() as session:
if flow_id:
flow = session.query(Flow).get(flow_id)
elif flow_name:
flow = session.query(Flow).filter(Flow.name == flow_name).first() flow = session.query(Flow).filter(Flow.name == flow_name).first()
return flow return flow
def get_flow_by_id(self, flow_id: str) -> Flow:
with session_getter() as session:
flow = session.query(Flow).filter(Flow.id == flow_id).first()
return flow
def build(self): def build(self):
raise NotImplementedError raise NotImplementedError