Upload 53 files

ai-routes.js

@@ -95,6 +95,13 @@ const PROVIDERS = {
     GEMMA: 'GEMMA'
 };
 
+const DEFAULT_OPENROUTER_MODELS = [
+    'qwen/qwen3-coder:free',
+    'openai/gpt-oss-120b:free',
+    'qwen/qwen3-235b-a22b:free',
+    'tngtech/deepseek-r1t-chimera:free'
+];
+
 let activeProviderOrder = [PROVIDERS.GEMINI, PROVIDERS.OPENROUTER, PROVIDERS.GEMMA];
 
 function deprioritizeProvider(providerName) {
@@ -159,14 +166,16 @@ async function callGeminiProvider(baseParams) {
 }
 
 async function callOpenRouterProvider(baseParams) {
-    const openRouterModels = [
-        'qwen/qwen3-coder:free',
-        'openai/gpt-oss-120b:free',
-        'qwen/qwen3-235b-a22b:free',
-        'tngtech/deepseek-r1t-chimera:free'
-    ];
+    // Retrieve model list from DB
+    const config = await ConfigModel.findOne({ key: 'main' });
+    let openRouterModels = [];
+    if (config && config.openRouterModels && config.openRouterModels.length > 0) {
+        openRouterModels = config.openRouterModels.map(m => m.id);
+    } else {
+        openRouterModels = DEFAULT_OPENROUTER_MODELS;
+    }
+
     const openAIMessages = convertGeminiToOpenAI(baseParams);
-    
     const keys = await getKeyPool('openrouter');
     if (keys.length === 0) throw new Error("No OpenRouter API keys configured");
 
@@ -310,9 +319,16 @@ async function streamGemini(baseParams, res) {
 }
 
 async function streamOpenRouter(baseParams, res) {
-    const openRouterModels = ['qwen/qwen3-coder:free', 'openai/gpt-oss-120b:free'];
+    // Retrieve model list from DB
+    const config = await ConfigModel.findOne({ key: 'main' });
+    let openRouterModels = [];
+    if (config && config.openRouterModels && config.openRouterModels.length > 0) {
+        openRouterModels = config.openRouterModels.map(m => m.id);
+    } else {
+        openRouterModels = DEFAULT_OPENROUTER_MODELS;
+    }
+
     const messages = convertGeminiToOpenAI(baseParams);
-    
     const keys = await getKeyPool('openrouter');
     if (keys.length === 0) throw new Error("No OpenRouter API keys");
 

models.js

@@ -120,7 +120,12 @@ const ConfigSchema = new mongoose.Schema({
     apiKeys: {
         gemini: [String],
         openrouter: [String]
-    }
+    },
+    openRouterModels: [{ 
+        id: String, 
+        name: String, 
+        isCustom: { type: Boolean, default: false } 
+    }]
 });
 const ConfigModel = mongoose.model('Config', ConfigSchema);
 

pages/AIAssistant.tsx

@@ -1,8 +1,8 @@
 
 import React, { useState, useRef, useEffect } from 'react';
 import { api } from '../services/api';
-import { AIChatMessage, SystemConfig, UserRole } from '../types';
-import { Bot, Mic, Square, Play, Volume2, Send, CheckCircle, Brain, Sparkles, Loader2, StopCircle, Trash2, Image as ImageIcon, X, AlertTriangle, ShieldCheck, Activity, Power, ExternalLink, Key, Plus, Save } from 'lucide-react';
+import { AIChatMessage, SystemConfig, UserRole, OpenRouterModelConfig } from '../types';
+import { Bot, Mic, Square, Play, Volume2, Send, CheckCircle, Brain, Sparkles, Loader2, StopCircle, Trash2, Image as ImageIcon, X, AlertTriangle, ShieldCheck, Activity, Power, ExternalLink, Key, Plus, Save, ArrowUp, ArrowDown } from 'lucide-react';
 import ReactMarkdown from 'react-markdown';
 import remarkGfm from 'remark-gfm';
 import { Toast, ToastState } from '../components/Toast';
@@ -85,6 +85,13 @@ const cleanTextForTTS = (text: string) => {
         .trim();
 };
 
+const DEFAULT_OR_MODELS = [
+    { id: 'qwen/qwen3-coder:free', name: 'Qwen 3 Coder', isCustom: false },
+    { id: 'openai/gpt-oss-120b:free', name: 'GPT OSS 120B', isCustom: false },
+    { id: 'qwen/qwen3-235b-a22b:free', name: 'Qwen 3 235B', isCustom: false },
+    { id: 'tngtech/deepseek-r1t-chimera:free', name: 'DeepSeek R1T', isCustom: false }
+];
+
 export const AIAssistant: React.FC = () => {
     const currentUser = api.auth.getCurrentUser();
     const isAdmin = currentUser?.role === UserRole.ADMIN;
@@ -97,13 +104,22 @@ export const AIAssistant: React.FC = () => {
     // Chat State with Persistence
     const [activeTab, setActiveTab] = useState<'chat' | 'assessment'>('chat');
     const [messages, setMessages] = useState<AIChatMessage[]>(() => {
-        const saved = localStorage.getItem('ai_chat_history');
-        return saved ? JSON.parse(saved) : [{ 
-            id: 'welcome', 
-            role: 'model', 
-            text: '你好！我是你的 AI 智能助教。有什么可以帮你的吗？', 
-            timestamp: Date.now() 
-        }];
+        try {
+            const saved = localStorage.getItem('ai_chat_history');
+            return saved ? JSON.parse(saved) : [{ 
+                id: 'welcome', 
+                role: 'model', 
+                text: '你好！我是你的 AI 智能助教。有什么可以帮你的吗？', 
+                timestamp: Date.now() 
+            }];
+        } catch (e) {
+            return [{ 
+                id: 'welcome', 
+                role: 'model', 
+                text: '你好！我是你的 AI 智能助教。有什么可以帮你的吗？', 
+                timestamp: Date.now() 
+            }];
+        }
     });
     const [textInput, setTextInput] = useState('');
     const [inputMode, setInputMode] = useState<'text' | 'audio'>('text');
@@ -121,6 +137,11 @@ export const AIAssistant: React.FC = () => {
     const [openRouterKeys, setOpenRouterKeys] = useState<string[]>([]);
     const [newGeminiKey, setNewGeminiKey] = useState('');
     const [newOpenRouterKey, setNewOpenRouterKey] = useState('');
+    
+    // Admin Model Management State
+    const [orModels, setOrModels] = useState<OpenRouterModelConfig[]>([]);
+    const [newModelId, setNewModelId] = useState('');
+    const [newModelName, setNewModelName] = useState('');
 
     // Audio Refs
     const mediaRecorderRef = useRef<MediaRecorder | null>(null);
@@ -143,6 +164,9 @@ export const AIAssistant: React.FC = () => {
                     setGeminiKeys(cfg.apiKeys.gemini || []);
                     setOpenRouterKeys(cfg.apiKeys.openrouter || []);
                 }
+                if (isAdmin) {
+                    setOrModels(cfg.openRouterModels && cfg.openRouterModels.length > 0 ? cfg.openRouterModels : DEFAULT_OR_MODELS);
+                }
             } catch (e) {
                 console.error("Init failed", e);
             } finally {
@@ -157,9 +181,25 @@ export const AIAssistant: React.FC = () => {
         };
     }, []);
 
-    // Persist messages
+    // Persist messages with strict count limit
     useEffect(() => {
-        localStorage.setItem('ai_chat_history', JSON.stringify(messages));
+        try {
+            // Force strict limit: Welcome message + Last 50 messages
+            const MAX_COUNT = 50;
+            const welcome = messages.find(m => m.id === 'welcome');
+            const others = messages.filter(m => m.id !== 'welcome');
+            
+            // Slice the last 50
+            const recent = others.slice(-MAX_COUNT);
+            
+            // Reconstruct array: Welcome (if exists) + Recent
+            const messagesToSave = (welcome ? [welcome] : []).concat(recent);
+
+            localStorage.setItem('ai_chat_history', JSON.stringify(messagesToSave));
+        } catch (e) {
+            console.warn("LocalStorage save failed despite count limit (Quota Exceeded). History not persisted.", e);
+            // Catch error to prevent crash. This usually happens if the 50 messages contain VERY large audio files exceeding 5MB total.
+        }
     }, [messages]);
 
     useEffect(() => {
@@ -455,6 +495,30 @@ export const AIAssistant: React.FC = () => {
         }
     };
 
+    // --- Model Management Handlers ---
+    const handleAddModel = () => {
+        if (!newModelId.trim()) return;
+        setOrModels([...orModels, {
+            id: newModelId.trim(),
+            name: newModelName.trim() || newModelId.trim(),
+            isCustom: true
+        }]);
+        setNewModelId('');
+        setNewModelName('');
+    };
+
+    const handleRemoveModel = (idx: number) => {
+        setOrModels(orModels.filter((_, i) => i !== idx));
+    };
+
+    const handleMoveModel = (idx: number, direction: -1 | 1) => {
+        const newArr = [...orModels];
+        const targetIdx = idx + direction;
+        if (targetIdx < 0 || targetIdx >= newArr.length) return;
+        [newArr[idx], newArr[targetIdx]] = [newArr[targetIdx], newArr[idx]];
+        setOrModels(newArr);
+    };
+
     const saveApiKeys = async () => {
         if (!systemConfig) return;
         try {
@@ -463,12 +527,13 @@ export const AIAssistant: React.FC = () => {
                 apiKeys: {
                     gemini: geminiKeys,
                     openrouter: openRouterKeys
-                }
+                },
+                openRouterModels: orModels
             });
             // NEW: Reset provider priority pool immediately after saving
             await api.ai.resetPool();
             
-            setToast({ show: true, message: 'API 密钥配置已保存并生效 (线路优先级已重置)', type: 'success' });
+            setToast({ show: true, message: 'API 配置及模型列表已保存', type: 'success' });
         } catch (e) {
             setToast({ show: true, message: '保存失败', type: 'error' });
         }
@@ -528,7 +593,7 @@ export const AIAssistant: React.FC = () => {
                     <div className="flex justify-between items-center mb-6">
                         <h3 className="font-bold text-gray-800 flex items-center"><Key className="mr-2 text-amber-500"/> 多线路密钥池配置</h3>
                         <button onClick={saveApiKeys} className="bg-blue-600 text-white px-4 py-2 rounded-lg text-sm font-bold flex items-center gap-2 hover:bg-blue-700 shadow-sm">
-                            <Save size={16}/> 保存密钥配置
+                            <Save size={16}/> 保存所有配置
                         </button>
                     </div>
                     
@@ -589,6 +654,54 @@ export const AIAssistant: React.FC = () => {
                             </div>
                         </div>
                     </div>
+
+                    {/* OpenRouter Model Management */}
+                    <div className="mt-8 border-t border-gray-100 pt-6">
+                        <div className="flex justify-between items-center mb-4">
+                            <h4 className="font-bold text-gray-700 text-sm">OpenRouter 大模型列表管理</h4>
+                            <span className="text-xs text-gray-400">优先尝试列表顶部的模型</span>
+                        </div>
+                        <div className="space-y-2 mb-4 bg-gray-50 p-3 rounded-lg border border-gray-200">
+                            {orModels.map((m, idx) => (
+                                <div key={idx} className="flex items-center gap-2 bg-white p-2 rounded border border-gray-100 shadow-sm">
+                                    <div className="flex flex-col gap-0.5 px-1">
+                                        <button onClick={()=>handleMoveModel(idx, -1)} className="text-gray-400 hover:text-blue-500 disabled:opacity-30" disabled={idx===0}><ArrowUp size={12}/></button>
+                                        <button onClick={()=>handleMoveModel(idx, 1)} className="text-gray-400 hover:text-blue-500 disabled:opacity-30" disabled={idx===orModels.length-1}><ArrowDown size={12}/></button>
+                                    </div>
+                                    <div className="flex-1">
+                                        <div className="text-sm font-bold text-gray-800">{m.name || m.id}</div>
+                                        <div className="text-xs text-gray-400 font-mono">{m.id}</div>
+                                    </div>
+                                    <div className="flex items-center gap-2">
+                                        {m.isCustom ? (
+                                            <span className="text-[10px] bg-blue-50 text-blue-600 px-2 py-0.5 rounded">自定义</span>
+                                        ) : (
+                                            <span className="text-[10px] bg-gray-100 text-gray-500 px-2 py-0.5 rounded">内置</span>
+                                        )}
+                                        <button 
+                                            onClick={() => handleRemoveModel(idx)} 
+                                            className={`p-1.5 rounded transition-colors ${m.isCustom ? 'text-gray-400 hover:text-red-500 hover:bg-red-50' : 'text-gray-200 cursor-not-allowed'}`}
+                                            disabled={!m.isCustom}
+                                            title={!m.isCustom ? "内置模型不可删除" : "删除模型"}
+                                        >
+                                            <Trash2 size={16}/>
+                                        </button>
+                                    </div>
+                                </div>
+                            ))}
+                        </div>
+                        <div className="flex gap-2 items-end bg-gray-50 p-3 rounded-lg border border-gray-200">
+                            <div className="flex-1">
+                                <label className="text-xs text-gray-500 mb-1 block">模型 ID (如: openai/gpt-4o)</label>
+                                <input className="w-full border border-gray-300 rounded px-2 py-1.5 text-sm" value={newModelId} onChange={e=>setNewModelId(e.target.value)} placeholder="输入 OpenRouter 模型 ID"/>
+                            </div>
+                            <div className="flex-1">
+                                <label className="text-xs text-gray-500 mb-1 block">显示名称 (可选)</label>
+                                <input className="w-full border border-gray-300 rounded px-2 py-1.5 text-sm" value={newModelName} onChange={e=>setNewModelName(e.target.value)} placeholder="友好显示的名称"/>
+                            </div>
+                            <button onClick={handleAddModel} className="bg-indigo-600 text-white px-4 py-1.5 rounded text-sm hover:bg-indigo-700 h-9">添加</button>
+                        </div>
+                    </div>
                 </div>
             </div>
         );

types.ts

@@ -165,6 +165,12 @@ export interface Notification {
   createTime: string;
 }
 
+export interface OpenRouterModelConfig {
+  id: string; // The model name (e.g., qwen/qwen3-coder:free)
+  name?: string; // Optional friendly name
+  isCustom: boolean; // Built-in vs Custom
+}
+
 export interface SystemConfig {
   systemName: string;
   semester: string;
@@ -182,6 +188,7 @@ export interface SystemConfig {
       gemini?: string[];
       openrouter?: string[];
   };
+  openRouterModels?: OpenRouterModelConfig[];
 }
 
 export interface SchoolCalendarEntry {
@@ -372,4 +379,4 @@ export interface AIChatMessage {
   audio?: string;
   isAudioMessage?: boolean;
   timestamp: number;
-}
+}