From 114809515655a81e4f97d3cc7ea32a0b4020da72 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 8 Feb 2024 15:07:01 -0300 Subject: [PATCH] Refactor credential.py: update create_credential() method --- src/backend/langflow/api/v1/credential.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/api/v1/credential.py b/src/backend/langflow/api/v1/credential.py index e8b6c0e5d..1b7e04fe3 100644 --- a/src/backend/langflow/api/v1/credential.py +++ b/src/backend/langflow/api/v1/credential.py @@ -2,11 +2,15 @@ from datetime import datetime from uuid import UUID from fastapi import APIRouter, Depends, HTTPException +from sqlmodel import Session, select + 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.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"]) @@ -29,7 +33,7 @@ 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) + db_credential = Credential.model_validate(credential, from_attributes=True, update={"user_id": current_user.id}) 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)