From 738738808962dfe0cc09d10ce7d6f2133579d686 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 8 Feb 2024 15:30:50 -0300 Subject: [PATCH] Refactor create_credential function to use model_dump and model_validate methods --- src/backend/langflow/api/v1/credential.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/api/v1/credential.py b/src/backend/langflow/api/v1/credential.py index 1b7e04fe3..fbbb488bc 100644 --- a/src/backend/langflow/api/v1/credential.py +++ b/src/backend/langflow/api/v1/credential.py @@ -9,9 +9,6 @@ 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.user.model import User from langflow.services.deps import get_session, get_settings_service -from langflow.services.database.models.user.model import User -from langflow.services.deps import get_session, get_settings_service -from sqlmodel import Session, select router = APIRouter(prefix="/credentials", tags=["Credentials"]) @@ -33,7 +30,10 @@ def create_credential( if credential_exists: raise HTTPException(status_code=400, detail="Credential name already exists") - db_credential = Credential.model_validate(credential, from_attributes=True, update={"user_id": current_user.id}) + credential_dict = credential.model_dump() + credential_dict["user_id"] = current_user.id + + db_credential = Credential.model_validate(credential_dict) if not db_credential.value: raise HTTPException(status_code=400, detail="Credential value cannot be empty") encrypted = auth_utils.encrypt_api_key(db_credential.value, settings_service=settings_service)