diff --git a/api/configs/license/__init__.py b/api/configs/license/__init__.py index 408926fcc3..7cb4873b76 100644 --- a/api/configs/license/__init__.py +++ b/api/configs/license/__init__.py @@ -3,19 +3,10 @@ from enum import StrEnum from pydantic import Field from pydantic_settings import BaseSettings - -class LicenseStatus(StrEnum): - NONE = "none" - INACTIVE = "inactive" - ACTIVE = "active" - EXPIRING = "expiring" - EXPIRED = "expired" - LOST = "lost" - class LicenseConfig(BaseSettings): - LICENSE_STATUS: LicenseStatus = Field( + LICENSE_STATUS: str = Field( description="license status", - default=LicenseStatus.NONE, + default="none", ) LICENSE_EXPIRED_AT: str = Field( diff --git a/api/services/feature_service.py b/api/services/feature_service.py index 26d6c85efd..9557622310 100644 --- a/api/services/feature_service.py +++ b/api/services/feature_service.py @@ -39,7 +39,7 @@ class LicenseStatus(StrEnum): class LicenseModel(BaseModel): # status: LicenseStatus = LicenseStatus.NONE # expired_at: str = "" - status: LicenseStatus = dify_config.LICENSE_STATUS + status: LicenseStatus = LicenseStatus(dify_config.LICENSE_STATUS) expired_at: str = dify_config.LICENSE_EXPIRED_AT class FeatureModel(BaseModel):