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

Refactor run_flow function to improve readability and maintainability

Browse source
This commit is contained in:
ogabrielluiz 2024-06-07 10:04:39 -03:00
commit 3d82417068
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/base/langflow/helpers/flow.py
View file

@ -90,7 +90,9 @@ async def run_flow(
fallback_to_env_vars = get_settings_service().settings.fallback_to_env_var fallback_to_env_vars = get_settings_service().settings.fallback_to_env_var
return await graph.arun(inputs_list, inputs_components=inputs_components, types=types, fallback_to_env_vars=fallback_to_env_vars) return await graph.arun(
inputs_list, inputs_components=inputs_components, types=types, fallback_to_env_vars=fallback_to_env_vars
)
def generate_function_for_flow( def generate_function_for_flow(
@ -249,7 +251,7 @@ def get_flow_by_id_or_endpoint_name(
flow = db.get(Flow, flow_id) flow = db.get(Flow, flow_id)
except ValueError: except ValueError:
endpoint_name = flow_id_or_name endpoint_name = flow_id_or_name
stmt = select(Flow).where(Flow.name == endpoint_name) stmt = select(Flow).where(Flow.endpoint_name == endpoint_name)
if user_id: if user_id:
stmt = stmt.where(Flow.user_id == user_id) stmt = stmt.where(Flow.user_id == user_id)
flow = db.exec(stmt).first() flow = db.exec(stmt).first()