From aee00ab07f8cc572727e211eb109fd4c510030bd Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 5 Dec 2023 18:15:40 -0300 Subject: [PATCH] Fix credential query in create_credential and update_credential functions --- src/backend/langflow/api/v1/credential.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/api/v1/credential.py b/src/backend/langflow/api/v1/credential.py index d828c4511..3c71e7e3a 100644 --- a/src/backend/langflow/api/v1/credential.py +++ b/src/backend/langflow/api/v1/credential.py @@ -27,8 +27,7 @@ def create_credential( try: # check if credential name already exists credential_exists = ( - session.query(Credential) - .filter(Credential.name == credential.name, Credential.user_id == current_user.id) + session.exec(select(Credential).where(Credential.name == credential.name, Credential.user_id == current_user.id)) .first() ) if credential_exists: @@ -75,7 +74,7 @@ def update_credential( """Update a credential.""" try: db_credential = ( - session.exec(select(Credential).where(Credential.id == credential_id and Credential.user_id == current_user.id)) + session.exec(select(Credential).where(Credential.id == credential_id, Credential.user_id == current_user.id)) .first() ) if not db_credential: