支持配置读取许可证

pull/19211/head
JJ-Choi1999 1 year ago
parent 73a94bae14
commit 4635929653

@ -8,6 +8,7 @@ from .deploy import DeploymentConfig
from .enterprise import EnterpriseFeatureConfig from .enterprise import EnterpriseFeatureConfig
from .extra import ExtraServiceConfig from .extra import ExtraServiceConfig
from .feature import FeatureConfig from .feature import FeatureConfig
from .license import LicenseConfig
from .middleware import MiddlewareConfig from .middleware import MiddlewareConfig
from .observability import ObservabilityConfig from .observability import ObservabilityConfig
from .packaging import PackagingInfo from .packaging import PackagingInfo
@ -70,6 +71,7 @@ class DifyConfig(
# Enterprise feature configs # Enterprise feature configs
# **Before using, please contact business@dify.ai by email to inquire about licensing matters.** # **Before using, please contact business@dify.ai by email to inquire about licensing matters.**
EnterpriseFeatureConfig, EnterpriseFeatureConfig,
LicenseConfig
): ):
model_config = SettingsConfigDict( model_config = SettingsConfigDict(
# read from dotenv format config file # read from dotenv format config file

@ -0,0 +1,22 @@
from pydantic import Field
from pydantic_settings import BaseSettings
from enum import StrEnum
class LicenseStatus(StrEnum):
NONE = "none"
INACTIVE = "inactive"
ACTIVE = "active"
EXPIRING = "expiring"
EXPIRED = "expired"
LOST = "lost"
class LicenseConfig(BaseSettings):
LICENSE_STATUS: LicenseStatus = Field(
description="license status",
default=LicenseStatus.NONE,
)
LICENSE_EXPIRED_AT: str = Field(
description="license expired at",
default="",
)

@ -6,7 +6,6 @@ from configs import dify_config
from services.billing_service import BillingService from services.billing_service import BillingService
from services.enterprise.enterprise_service import EnterpriseService from services.enterprise.enterprise_service import EnterpriseService
class SubscriptionModel(BaseModel): class SubscriptionModel(BaseModel):
plan: str = "sandbox" plan: str = "sandbox"
interval: str = "" interval: str = ""
@ -37,9 +36,10 @@ class LicenseStatus(StrEnum):
class LicenseModel(BaseModel): class LicenseModel(BaseModel):
status: LicenseStatus = LicenseStatus.NONE # status: LicenseStatus = LicenseStatus.NONE
expired_at: str = "" # expired_at: str = ""
status: LicenseStatus = dify_config.LICENSE_STATUS
expired_at: str = dify_config.LICENSE_EXPIRED_AT
class FeatureModel(BaseModel): class FeatureModel(BaseModel):
billing: BillingModel = BillingModel() billing: BillingModel = BillingModel()

Loading…
Cancel
Save