youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-03 17:08:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add branding support

Browse Source
This commit is contained in:
GareArc
2025-03-11 07:14:52 -04:00
parent 0856792a57
commit fdb1e649d4
5 changed files with 50 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
api/services/feature_service.py
View File

@ -36,6 +36,13 @@ class LicenseModel(BaseModel):
expired_at: str = ""
class BrandingModel(BaseModel):
enabled: bool = False
login_page_logo: str = ("",)
workspace_logo: str = ("",)
favicon: str = ("",)
class FeatureModel(BaseModel):
billing: BillingModel = BillingModel()
members: LimitationModel = LimitationModel(size=0, limit=1)
@ -65,6 +72,8 @@ class SystemFeatureModel(BaseModel):
is_allow_create_workspace: bool = False
is_email_setup: bool = False
license: LicenseModel = LicenseModel()
is_enterprise: bool = False
branding: BrandingModel = BrandingModel()
class FeatureService:
@ -87,6 +96,7 @@ class FeatureService:
if dify_config.ENTERPRISE_ENABLED:
system_features.enable_web_sso_switch_component = True
system_features.is_enterprise = True
cls._fulfill_params_from_enterprise(system_features)
@ -172,6 +182,9 @@ class FeatureService:
if "is_allow_create_workspace" in enterprise_info:
features.is_allow_create_workspace = enterprise_info["is_allow_create_workspace"]
if "branding" in enterprise_info:
features.branding = enterprise_info["branding"]
if "license" in enterprise_info:
license_info = enterprise_info["license"]