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

🐛 fix(flow-runner.mdx): remove unnecessary load_flow calls and update get_flow method signature to include tweaks parameter

Browse source 
🚚 chore(flow-runner.mdx): update load_flow references to get_flow
🚚 chore(flow-runner.mdx): remove commented out code
🚚 chore(flow-runner.mdx): update load_flow references to get_flow
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-07-31 16:03:51 -03:00
commit b2979bce17
2 changed files with 19 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -159,14 +159,23 @@ class CustomComponent(Component, extra=Extra.allow):
return flows
def get_flow(
self, *, flow_name: Optional[str] = None, flow_id: Optional[str] = None
self,
*,
flow_name: Optional[str] = None,
flow_id: Optional[str] = None,
tweaks: Optional[dict] = None,
) -> Flow:
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()
return flow
else:
raise ValueError("Either flow_name or flow_id must be provided")
if not flow:
raise ValueError(f"Flow {flow_name or flow_id} not found")
return self.load_flow(flow.id, tweaks)
def build(self):
raise NotImplementedError