fix: wrong title location

This commit is contained in:
GareArc 2025-03-17 15:00:04 -04:00
parent ea769c6483
commit d72b4752c9
4 changed files with 14 additions and 6 deletions

View File

@ -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

View File

@ -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:

View File

@ -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,

View File

@ -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: