From 56aaee5558f52eeeac3846f6d17f465daf36336d Mon Sep 17 00:00:00 2001 From: GareArc Date: Mon, 17 Mar 2025 15:01:31 -0400 Subject: [PATCH] fix: wrong branding title --- api/services/feature_service.py | 8 ++++++++ api/tasks/mail_email_code_login.py | 4 ++-- api/tasks/mail_invite_member_task.py | 4 ++-- api/tasks/mail_reset_password_task.py | 4 ++-- 4 files changed, 14 insertions(+), 6 deletions(-) diff --git a/api/services/feature_service.py b/api/services/feature_service.py index 404dbc46ac..8335159999 100644 --- a/api/services/feature_service.py +++ b/api/services/feature_service.py @@ -195,3 +195,11 @@ class FeatureService: if "expired_at" in license_info: features.license.expired_at = license_info["expiredAt"] + + @classmethod + def get_enterprise_application_title(cls): + branding = cls.get_system_features().get("branding", None) + application_title = "Dify" + if branding and branding.get("enabled", False): + application_title = branding.get("application_title", "Dify") + return application_title diff --git a/api/tasks/mail_email_code_login.py b/api/tasks/mail_email_code_login.py index 4bfaaca453..6c52b725dc 100644 --- a/api/tasks/mail_email_code_login.py +++ b/api/tasks/mail_email_code_login.py @@ -29,7 +29,7 @@ def send_email_code_login_mail_task(language: str, to: str, code: str): if language == "zh-Hans": template = "email_code_login_mail_template_zh-CN.html" if dify_config.ENTERPRISE_ENABLED: - application_title = FeatureService.get_system_features().get("application_title", "Dify") + application_title = FeatureService.get_enterprise_application_title() template = "without-brand/email_code_login_mail_template_zh-CN_enterprise.html" html_content = render_template(template, to=to, code=code, application_title=application_title) else: @@ -38,7 +38,7 @@ def send_email_code_login_mail_task(language: str, to: str, code: str): else: template = "email_code_login_mail_template_en-US.html" if dify_config.ENTERPRISE_ENABLED: - application_title = FeatureService.get_system_features().get("application_title", "Dify") + application_title = FeatureService.get_enterprise_application_title() template = "without-brand/email_code_login_mail_template_en-US_enterprise.html" html_content = render_template(template, to=to, code=code, application_title=application_title) else: diff --git a/api/tasks/mail_invite_member_task.py b/api/tasks/mail_invite_member_task.py index 92c8250586..836fc3f991 100644 --- a/api/tasks/mail_invite_member_task.py +++ b/api/tasks/mail_invite_member_task.py @@ -36,7 +36,7 @@ def send_invite_member_mail_task(language: str, to: str, token: str, inviter_nam if language == "zh-Hans": template = "invite_member_mail_template_zh-CN.html" if dify_config.ENTERPRISE_ENABLED: - application_title = FeatureService.get_system_features().get("application_title", "Dify") + application_title = FeatureService.get_enterprise_application_title() template = "without-brand/invite_member_mail_template_zh-CN.html" html_content = render_template( template, @@ -54,7 +54,7 @@ def send_invite_member_mail_task(language: str, to: str, token: str, inviter_nam else: template = "invite_member_mail_template_en-US.html" if dify_config.ENTERPRISE_ENABLED: - application_title = FeatureService.get_system_features().get("application_title", "Dify") + application_title = FeatureService.get_enterprise_application_title() template = "without-brand/invite_member_mail_template_en-US.html" html_content = render_template( template, diff --git a/api/tasks/mail_reset_password_task.py b/api/tasks/mail_reset_password_task.py index d892dbe345..37c6fa99e5 100644 --- a/api/tasks/mail_reset_password_task.py +++ b/api/tasks/mail_reset_password_task.py @@ -29,7 +29,7 @@ def send_reset_password_mail_task(language: str, to: str, code: str): if language == "zh-Hans": template = "reset_password_mail_template_zh-CN.html" if dify_config.ENTERPRISE_ENABLED: - application_title = FeatureService.get_system_features().get("application_title", "Dify") + application_title = FeatureService.get_enterprise_application_title() template = "without-brand/reset_password_mail_template_zh-CN.html" html_content = render_template(template, to=to, code=code, application_title=application_title) else: @@ -38,7 +38,7 @@ def send_reset_password_mail_task(language: str, to: str, code: str): else: template = "reset_password_mail_template_en-US.html" if dify_config.ENTERPRISE_ENABLED: - application_title = FeatureService.get_system_features().get("application_title", "Dify") + application_title = FeatureService.get_enterprise_application_title() template = "without-brand/reset_password_mail_template_en-US.html" html_content = render_template(template, to=to, code=code, application_title=application_title) else: