Upload app.py

app.py

@@ -1358,7 +1358,9 @@ def _clear_chat_history_db(username):
 # =========================
 # 7. 侧边栏 UI & 逻辑
 # =========================
-model_mapping = {
+
+# 游客态模型（免费模型，有优先级策略）
+_guest_models = {
     "⭐ Step-3.5 (首选)": "stepfun/step-3.5-flash:free",
     "🌐 OR-Auto (避堵)": "openrouter/free",
     "🧠 GLM-4.5 (推理)": "z-ai/glm-4.5-air:free",
@@ -1375,16 +1377,34 @@ model_mapping = {
     "🪶 Gemma-3-4B": "google/gemma-3-4b-it:free",
     "🫧 Gemma-3n-e2b": "google/gemma-3n-e2b-it:free",
     "📷 Nemotron-VL": "nvidia/nemotron-nano-12b-v2-vl:free",
-    "🛡️ DeepSeek (官方)": "deepseek-chat",
-    "🏢 百度文心 (官方)": "ernie-3.5-8k",
 }
 
-# 游客模式：屏蔽收费模型
-_GUEST_BLOCKED_MODELS = {"🛡️ DeepSeek (官方)", "🏢 百度文心 (官方)"}
+# 登录态模型（付费/官方模型，无优先级策略）
+_login_models = {
+    # 顶级推理
+    "🔥 文心5.0 (旗舰)": "ernie-5.0",
+    "🧠 文心5.0思考": "ernie-5.0-thinking-latest",
+    "🛡️ DeepSeek R1 (推理)": "deepseek-r1",
+    "💭 DeepSeek V3.2思考": "deepseek-v3.2-think",
+    "🚀 千问3.5-397B (超大)": "qwen3.5-397b-a17b",
+    "✨ 智谱GLM-5": "glm-5",
+    "🌙 Kimi K2.5": "kimi-k2.5",
+    # 性价比
+    "⚡ 文心4.5T (快速)": "ernie-4.5-turbo-latest",
+    "⚖️ 千问3.5-35B (平衡)": "qwen3.5-35b-a3b",
+    "💎 MiniMax M2.5": "minimax-m2.5",
+    # 视觉多模态
+    "📷 千问3视觉-235B": "qwen3-vl-235b-a22b-instruct",
+    "🖼️ 文心4.5视觉": "ernie-4.5-turbo-vl-latest",
+    # DeepSeek官方
+    "🔰 DeepSeek (官方)": "deepseek-chat",
+}
+
+# 根据登录状态选择模型列表
 if IS_GUEST:
-    _active_models = {k: v for k, v in model_mapping.items() if k not in _GUEST_BLOCKED_MODELS}
+    _active_models = _guest_models
 else:
-    _active_models = model_mapping
+    _active_models = _login_models
 
 
 with st.sidebar:
@@ -1698,7 +1718,28 @@ def llm_answer(query, context_docs, selected_display_name, web_enabled, is_guest
     ds_client = get_ds_client()
     baidu_client = get_baidu_client()
 
-    special_clients = {"deepseek-chat": ds_client, "ernie-3.5-8k": baidu_client}
+    # 根据模型ID选择对应的客户端（登录态模型都走百度千帆）
+    special_clients = {
+        # 游客兜底
+        "deepseek-chat": ds_client,
+        "ernie-3.5-8k": baidu_client,
+        # 登录态模型（全部走百度千帆）
+        "ernie-5.0": baidu_client,
+        "ernie-5.0-thinking-latest": baidu_client,
+        "deepseek-r1": baidu_client,
+        "deepseek-v3.2-think": baidu_client,
+        "qwen3.5-397b-a17b": baidu_client,
+        "glm-5": baidu_client,
+        "kimi-k2.5": baidu_client,
+        "ernie-4.5-turbo-latest": baidu_client,
+        "qwen3.5-35b-a3b": baidu_client,
+        "minimax-m2.5": baidu_client,
+        "qwen3-vl-235b-a22b-instruct": baidu_client,
+        "ernie-4.5-turbo-vl-latest": baidu_client,
+    }
+    
+    # 根据登录状态选择模型映射
+    model_mapping = _guest_models if is_guest else _login_models
     selected_id = model_mapping[selected_display_name]
 
     retry_queue = []
@@ -1706,19 +1747,21 @@ def llm_answer(query, context_docs, selected_display_name, web_enabled, is_guest
         (special_clients.get(selected_id, or_client), selected_id, f"首选-{selected_display_name}")
     )
 
-    if selected_id != "stepfun/step-3.5-flash:free":
-        retry_queue.append((or_client, "stepfun/step-3.5-flash:free", "⚡ 快速备选-Step3.5"))
+    # 仅游客态使用优先级和兜底策略
+    if is_guest:
+        if selected_id != "stepfun/step-3.5-flash:free":
+            retry_queue.append((or_client, "stepfun/step-3.5-flash:free", "⚡ 快速备选-Step3.5"))
 
-    if selected_id != "openrouter/free":
-        retry_queue.append((or_client, "openrouter/free", "OR-Auto 免费避堵"))
+        if selected_id != "openrouter/free":
+            retry_queue.append((or_client, "openrouter/free", "OR-Auto 免费避堵"))
 
-    paid_backups = [
-        ("deepseek-chat", "🛡️ DeepSeek 官方", ds_client),
-        ("ernie-3.5-8k", "🏢 百度文心", baidu_client),
-    ]
-    for p_id, p_label, p_client in paid_backups:
-        if selected_id != p_id:
-            retry_queue.append((p_client, p_id, f"💰 收费兜底-{p_label}"))
+        paid_backups = [
+            ("deepseek-chat", "🛡️ DeepSeek 官方", ds_client),
+            ("ernie-3.5-8k", "🏢 百度文心", baidu_client),
+        ]
+        for p_id, p_label, p_client in paid_backups:
+            if selected_id != p_id:
+                retry_queue.append((p_client, p_id, f"💰 收费兜底-{p_label}"))
 
     messages = [
         {"role": "system", "content": system_prompt},