From 62c053030bba0dce4d98b85350833b89f5fd2724 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 5 Dec 2023 18:00:59 -0300 Subject: [PATCH] Refactor credential.py to use SQLModel's select method --- src/backend/langflow/api/v1/credential.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/src/backend/langflow/api/v1/credential.py b/src/backend/langflow/api/v1/credential.py index f4347afcd..d828c4511 100644 --- a/src/backend/langflow/api/v1/credential.py +++ b/src/backend/langflow/api/v1/credential.py @@ -4,10 +4,13 @@ from uuid import UUID from fastapi import APIRouter, Depends, HTTPException from langflow.services.auth import utils as auth_utils from langflow.services.auth.utils import get_current_active_user -from langflow.services.database.models.credential import Credential, CredentialCreate, CredentialRead, CredentialUpdate +from langflow.services.database.models.credential import (Credential, + CredentialCreate, + CredentialRead, + CredentialUpdate) from langflow.services.database.models.user.model import User from langflow.services.deps import get_session, get_settings_service -from sqlmodel import Session +from sqlmodel import Session, select router = APIRouter(prefix="/credentials", tags=["Credentials"]) @@ -55,7 +58,7 @@ def read_credentials( ): """Read all credentials.""" try: - credentials = session.query(Credential).filter(Credential.user_id == current_user.id).all() + credentials = session.exec(select(Credential).where(Credential.user_id == current_user.id)).all() return credentials except Exception as e: raise HTTPException(status_code=500, detail=str(e)) from e @@ -72,8 +75,7 @@ def update_credential( """Update a credential.""" try: db_credential = ( - session.query(Credential) - .filter(Credential.id == credential_id, Credential.user_id == current_user.id) + session.exec(select(Credential).where(Credential.id == credential_id and Credential.user_id == current_user.id)) .first() ) if not db_credential: