From 782fc332e1f772dfc78daa275454623dd7264c3c Mon Sep 17 00:00:00 2001
From: Yansong Zhang <916125788@qq.com>
Date: Wed, 21 Jan 2026 14:07:11 +0800
Subject: [PATCH] fix const

---
 api/enums/hosted_provider.py    | 19 +++++++++++++------
 api/services/feature_service.py |  6 +++---
 2 files changed, 16 insertions(+), 9 deletions(-)

diff --git a/api/enums/hosted_provider.py b/api/enums/hosted_provider.py
index 17d115f4b5..f7a3683738 100644
--- a/api/enums/hosted_provider.py
+++ b/api/enums/hosted_provider.py
@@ -6,9 +6,16 @@ class HostedTrialProvider(StrEnum):
     Enum representing hosted model provider names for trial access.
     """
 
-    OPENAI = "openai"
-    DEEPSEEK = "deepseek"
-    ANTHROPIC = "anthropic"
-    GEMINI = "gemini"
-    XAI = "xai"
-    TONGYI = "tongyi"
+    OPENAI = "langgenius/openai"
+    ANTHROPIC = "langgenius/anthropic"
+    GEMINI = "langgenius/gemini"
+    X = "langgenius/x"
+    DEEPSEEK = "langgenius/deepseek"
+    TONGYI = "langgenius/tongyi"
+
+    @property
+    def config_key(self) -> str:
+        """Return the config key used in dify_config (e.g., HOSTED_{config_key}_PAID_ENABLED)."""
+        if self == HostedTrialProvider.X:
+            return "XAI"
+        return self.name
diff --git a/api/services/feature_service.py b/api/services/feature_service.py
index 1e9b5aaf79..d61628c496 100644
--- a/api/services/feature_service.py
+++ b/api/services/feature_service.py
@@ -232,11 +232,11 @@ class FeatureService:
     @classmethod
     def _fulfill_trial_models_from_env(cls) -> list[str]:
         return [
-            provider
+            provider.value
             for provider in HostedTrialProvider
             if (
-                getattr(dify_config, f"HOSTED_{provider.upper()}_PAID_ENABLED", False)
-                and getattr(dify_config, f"HOSTED_{provider.upper()}_TRIAL_ENABLED", False)
+                getattr(dify_config, f"HOSTED_{provider.config_key}_PAID_ENABLED", False)
+                and getattr(dify_config, f"HOSTED_{provider.config_key}_TRIAL_ENABLED", False)
             )
         ]