From eeb3077b6f9252760d2c0b7aff92c03d7eaccace Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 21:20:20 -0300 Subject: [PATCH] refactor: Improve endpoint uniqueness check in create_flow function --- src/backend/base/langflow/api/v1/flows.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/api/v1/flows.py b/src/backend/base/langflow/api/v1/flows.py index be074be07..2cd16a003 100644 --- a/src/backend/base/langflow/api/v1/flows.py +++ b/src/backend/base/langflow/api/v1/flows.py @@ -51,9 +51,12 @@ def create_flow( else: flow.name = f"{flow.name} (1)" # Now check if the endpoint is unique - if session.exec( - select(Flow).where(Flow.endpoint_name == flow.endpoint_name).where(Flow.user_id == current_user.id) - ).first(): + if ( + flow.endpoint_name + and session.exec( + select(Flow).where(Flow.endpoint_name == flow.endpoint_name).where(Flow.user_id == current_user.id) + ).first() + ): flows = session.exec( select(Flow) .where(Flow.endpoint_name.like(f"{flow.endpoint_name}-%")) # type: ignore