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

refactor: Improve endpoint uniqueness check in create_flow function

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 21:20:20 -03:00
commit eeb3077b6f
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/base/langflow/api/v1/flows.py
View file

@ -51,9 +51,12 @@ def create_flow(
else: else:
flow.name = f"{flow.name} (1)" flow.name = f"{flow.name} (1)"
# Now check if the endpoint is unique # Now check if the endpoint is unique
if session.exec( if (
flow.endpoint_name
and session.exec(
select(Flow).where(Flow.endpoint_name == flow.endpoint_name).where(Flow.user_id == current_user.id) select(Flow).where(Flow.endpoint_name == flow.endpoint_name).where(Flow.user_id == current_user.id)
).first(): ).first()
):
flows = session.exec( flows = session.exec(
select(Flow) select(Flow)
.where(Flow.endpoint_name.like(f"{flow.endpoint_name}-%")) # type: ignore .where(Flow.endpoint_name.like(f"{flow.endpoint_name}-%")) # type: ignore