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>