From 84f4c320769e02719b800e4d2a833f9bc196b6c7 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 6 Feb 2024 19:14:52 -0300 Subject: [PATCH] Refactor Credential model to use Optional for provider field --- .../langflow/services/database/models/credential/model.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/services/database/models/credential/model.py b/src/backend/langflow/services/database/models/credential/model.py index 95bd4b829..5c9f7d9b0 100644 --- a/src/backend/langflow/services/database/models/credential/model.py +++ b/src/backend/langflow/services/database/models/credential/model.py @@ -2,9 +2,8 @@ from datetime import datetime from typing import TYPE_CHECKING, Optional from uuid import UUID, uuid4 -from sqlmodel import Field, Relationship, SQLModel - from langflow.services.database.models.credential.schema import CredentialType +from sqlmodel import Field, Relationship, SQLModel if TYPE_CHECKING: from langflow.services.database.models.user import User @@ -28,7 +27,7 @@ class Credential(CredentialBase, table=True): class CredentialCreate(CredentialBase): # AcceptedProviders is a custom Enum - provider: CredentialType = Field(description="Provider of the credential (e.g OpenAI)") + provider: Optional[CredentialType] = Field(None, description="Provider of the credential (e.g OpenAI)") class CredentialRead(SQLModel):