Upload 58 files

ai-routes.js

@@ -60,13 +60,18 @@ function convertGeminiToOpenAI(baseParams) {
 
 const PROVIDERS = { GEMINI: 'GEMINI', OPENROUTER: 'OPENROUTER', 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];
+
+// Runtime override logic
+let runtimeProviderOrder = [];
 
 function deprioritizeProvider(providerName) {
-    if (activeProviderOrder[activeProviderOrder.length - 1] === providerName) return;
-    activeProviderOrder = activeProviderOrder.filter(p => p !== providerName).concat(providerName);
+    // If the provider is already last, do nothing
+    if (runtimeProviderOrder.length > 0 && runtimeProviderOrder[runtimeProviderOrder.length - 1] === providerName) return;
+    
+    // Move to end
+    runtimeProviderOrder = runtimeProviderOrder.filter(p => p !== providerName).concat(providerName);
 }
-function resetProviderOrder() { activeProviderOrder = [PROVIDERS.GEMINI, PROVIDERS.OPENROUTER, PROVIDERS.GEMMA]; }
+
 function isQuotaError(e) {
     const msg = (e.message || '').toLowerCase();
     return e.status === 429 || e.status === 503 || msg.includes('quota') || msg.includes('overloaded') || msg.includes('resource_exhausted') || msg.includes('rate limit') || msg.includes('credits');
@@ -171,8 +176,24 @@ async function streamContentWithSmartFallback(baseParams, res) {
     if (hasAudio) {
         try { return await streamGemini(baseParams, res); } catch(e) { deprioritizeProvider(PROVIDERS.GEMINI); throw new Error('QUOTA_EXCEEDED_AUDIO'); }
     }
+    
+    // FETCH CONFIG AND SET PROVIDER ORDER
+    const config = await ConfigModel.findOne({ key: 'main' });
+    const configuredOrder = config?.aiProviderOrder && config.aiProviderOrder.length > 0 
+        ? config.aiProviderOrder 
+        : [PROVIDERS.GEMINI, PROVIDERS.OPENROUTER, PROVIDERS.GEMMA];
+
+    // If runtime order is empty or contains different elements (e.g. config changed), reset it
+    const runtimeSet = new Set(runtimeProviderOrder);
+    const configSet = new Set(configuredOrder);
+    if (runtimeProviderOrder.length === 0 || runtimeProviderOrder.length !== configuredOrder.length || !configuredOrder.every(p => runtimeSet.has(p))) {
+        runtimeProviderOrder = [...configuredOrder];
+    }
+
     let finalError = null;
-    for (const provider of activeProviderOrder) {
+    
+    // Use runtimeProviderOrder which might have been adjusted due to quota errors in previous calls
+    for (const provider of runtimeProviderOrder) {
         try {
             if (provider === PROVIDERS.GEMINI) return await streamGemini(baseParams, res);
             else if (provider === PROVIDERS.OPENROUTER) return await streamOpenRouter(baseParams, res);
@@ -226,7 +247,10 @@ router.get('/stats', checkAIAccess, async (req, res) => {
     } catch (e) { res.status(500).json({ error: e.message }); }
 });
 
-router.post('/reset-pool', checkAIAccess, (req, res) => { resetProviderOrder(); res.json({ success: true }); });
+router.post('/reset-pool', checkAIAccess, (req, res) => { 
+    runtimeProviderOrder = []; // Will be re-initialized from DB on next call
+    res.json({ success: true }); 
+});
 
 router.post('/chat', checkAIAccess, async (req, res) => {
     const { text, audio, history } = req.body;

models.js

@@ -115,6 +115,7 @@ const ConfigSchema = new mongoose.Schema({
     emailNotify: Boolean,
     enableAI: { type: Boolean, default: true },
     aiTotalCalls: { type: Number, default: 0 },
+    aiProviderOrder: { type: [String], default: ['GEMINI', 'OPENROUTER', 'GEMMA'] }, // NEW
     periodConfig: [{ period: Number, name: String, startTime: String, endTime: String }],
     apiKeys: {
         gemini: [String],

pages/AIAssistant.tsx

@@ -2,7 +2,7 @@
 import React, { useState, useRef, useEffect } from 'react';
 import { api } from '../services/api';
 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, BarChart2, Zap } from 'lucide-react';
+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, BarChart2, Zap, Layers } from 'lucide-react';
 import ReactMarkdown from 'react-markdown';
 import remarkGfm from 'remark-gfm';
 import { Toast, ToastState } from '../components/Toast';
@@ -206,6 +206,9 @@ export const AIAssistant: React.FC = () => {
     const [newModelName, setNewModelName] = useState('');
     const [newModelApiUrl, setNewModelApiUrl] = useState('');
 
+    // Admin Provider Priority State
+    const [providerOrder, setProviderOrder] = useState<string[]>(['GEMINI', 'OPENROUTER', 'GEMMA']);
+
     // Audio Refs
     const mediaRecorderRef = useRef<MediaRecorder | null>(null);
     const audioChunksRef = useRef<Blob[]>([]);
@@ -230,6 +233,10 @@ export const AIAssistant: React.FC = () => {
                     }
                     setOrModels(cfg.openRouterModels && cfg.openRouterModels.length > 0 ? cfg.openRouterModels : DEFAULT_OR_MODELS);
                     
+                    if (cfg.aiProviderOrder && cfg.aiProviderOrder.length > 0) {
+                        setProviderOrder(cfg.aiProviderOrder);
+                    }
+
                     // Fetch Detailed Stats
                     const stats = await api.ai.getStats();
                     setDetailedStats(stats);
@@ -616,10 +623,21 @@ export const AIAssistant: React.FC = () => {
         [newArr[idx], newArr[targetIdx]] = [newArr[targetIdx], newArr[idx]];
         setOrModels(newArr);
     };
+    const handleMoveProviderOrder = (idx: number, direction: -1 | 1) => {
+        const newArr = [...providerOrder]; const targetIdx = idx + direction;
+        if (targetIdx < 0 || targetIdx >= newArr.length) return;
+        [newArr[idx], newArr[targetIdx]] = [newArr[targetIdx], newArr[idx]];
+        setProviderOrder(newArr);
+    };
     const saveApiKeys = async () => {
         if (!systemConfig) return;
         try {
-            await api.config.save({ ...systemConfig, apiKeys: { gemini: geminiKeys, openrouter: openRouterKeys }, openRouterModels: orModels });
+            await api.config.save({ 
+                ...systemConfig, 
+                apiKeys: { gemini: geminiKeys, openrouter: openRouterKeys }, 
+                openRouterModels: orModels,
+                aiProviderOrder: providerOrder
+            });
             await api.ai.resetPool();
             setToast({ show: true, message: 'API 配置及模型列表已保存', type: 'success' });
         } catch (e) { setToast({ show: true, message: '保存失败', type: 'error' }); }
@@ -699,6 +717,27 @@ export const AIAssistant: React.FC = () => {
                             <div className="flex gap-2"><input className="flex-1 border border-gray-300 rounded px-3 py-1.5 text-sm outline-none focus:ring-2 focus:ring-purple-500" placeholder="输入 API Key" value={newOpenRouterKey} onChange={e => setNewOpenRouterKey(e.target.value)}/><button onClick={() => handleAddKey('openrouter')} className="bg-gray-100 hover:bg-gray-200 text-gray-600 px-3 py-1.5 rounded border border-gray-300"><Plus size={16}/></button></div>
                         </div>
                     </div>
+                    
+                    {/* Provider Order 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 flex items-center"><Layers className="mr-2" size={16}/> 大模型调用优先级</h4></div>
+                        <div className="bg-amber-50 p-4 rounded-lg border border-amber-100 mb-4">
+                            <p className="text-xs text-amber-800">系统将按照以下顺序尝试调用大模型。如果前一个服务商额度耗尽或报错，会自动切换到下一个。</p>
+                        </div>
+                        <div className="space-y-2 max-w-md">
+                            {providerOrder.map((provider, idx) => (
+                                <div key={provider} className="flex items-center gap-3 bg-white p-3 rounded-lg border border-gray-200 shadow-sm">
+                                    <div className="bg-gray-100 text-gray-500 w-6 h-6 flex items-center justify-center rounded-full text-xs font-bold">{idx + 1}</div>
+                                    <div className="flex-1 font-bold text-gray-700">{provider}</div>
+                                    <div className="flex gap-1">
+                                        <button onClick={() => handleMoveProviderOrder(idx, -1)} disabled={idx === 0} className="p-1 hover:bg-gray-100 rounded text-gray-400 hover:text-blue-500 disabled:opacity-30"><ArrowUp size={16}/></button>
+                                        <button onClick={() => handleMoveProviderOrder(idx, 1)} disabled={idx === providerOrder.length - 1} className="p-1 hover:bg-gray-100 rounded text-gray-400 hover:text-blue-500 disabled:opacity-30"><ArrowDown size={16}/></button>
+                                    </div>
+                                </div>
+                            ))}
+                        </div>
+                    </div>
+
                     <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">OpenAI 格式大模型列表管理</h4></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 min-w-0"><div className="text-sm font-bold text-gray-800">{m.name || m.id}</div><div className="text-xs text-gray-400 font-mono truncate" title={m.id}>ID: {m.id}</div>{m.apiUrl && <div className="text-[10px] text-blue-500 truncate" title={m.apiUrl}>API: {m.apiUrl}</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}><Trash2 size={16}/></button></div></div>))}</div>

types.ts

@@ -184,6 +184,7 @@ export interface SystemConfig {
   emailNotify: boolean;
   enableAI?: boolean;
   aiTotalCalls?: number;
+  aiProviderOrder?: string[]; // 'GEMINI', 'OPENROUTER', 'GEMMA'
   periodConfig?: PeriodConfig[]; 
   apiKeys?: {
       gemini?: string[];