From cf5bf746ef74e80603de21cde1ed6cf0189fee74 Mon Sep 17 00:00:00 2001
From: Gsh <15170702455@163.com>
Date: Tue, 25 Nov 2025 22:14:48 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E6=A8=A1=E5=9E=8B=E5=B0=8A=E4=BA=AB?=
 =?UTF-8?q?=E6=A0=87=E8=AF=86=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Yi.Ai.Vue3/src/components/ModelSelect/index.vue | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/Yi.Ai.Vue3/src/components/ModelSelect/index.vue b/Yi.Ai.Vue3/src/components/ModelSelect/index.vue
index ce697df9..4f632d51 100644
--- a/Yi.Ai.Vue3/src/components/ModelSelect/index.vue
+++ b/Yi.Ai.Vue3/src/components/ModelSelect/index.vue
@@ -99,14 +99,16 @@ function handleModelClick(item: GetSessionListVO) {
    规则2：金色光泽（VIP/付费）
    规则3：彩色流光（尊享/高级）
 -------------------------------- */
-function getModelStyleClass(modelName: any) {
-  if (!modelName) {
+function getModelStyleClass(mode: any) {
+  if (!mode) {
     return;
   }
-  const name = modelName.toLowerCase();
+  // isPremiumPackage
+  const name = mode.modelName.toLowerCase();
+  const isPremiumPackage = mode.isPremiumPackage;
 
   // 规则3：彩色流光
-  if (name.includes('claude-sonnet-4-5-20250929')) {
+  if (isPremiumPackage) {
     return `
       text-transparent bg-clip-text
       bg-[linear-gradient(45deg,#ff0000,#ff8000,#ffff00,#00ff00,#00ffff,#0000ff,#8000ff,#ff0080)]
@@ -167,7 +169,7 @@ function getWrapperClass(item: GetSessionListVO) {
           <div class="model-select-box-icon">
             <SvgIcon name="models" size="12" />
           </div>
-          <div :class="getModelStyleClass(currentModelName)" class="model-select-box-text font-size-12px">
+          <div :class="getModelStyleClass(modelStore.currentModelInfo)" class="model-select-box-text font-size-12px">
             {{ currentModelName }}
           </div>
         </div>
@@ -188,7 +190,7 @@ function getWrapperClass(item: GetSessionListVO) {
             :offset="[12, 0]"
           >
             <template #trigger>
-              <span :class="getModelStyleClass(item.modelName)">
+              <span :class="getModelStyleClass(item)">
                 {{ item.modelName }}
               </span>
             </template>