Upload 63 files

ai-routes.js

@@ -74,7 +74,7 @@ function convertGeminiToOpenAI(baseParams) {
 
 const PROVIDERS = { GEMINI: 'GEMINI', OPENROUTER: 'OPENROUTER', GEMMA: 'GEMMA', DOUBAO: 'DOUBAO' };
 const DEFAULT_OPENROUTER_MODELS = ['qwen/qwen3-coder:free', 'openai/gpt-oss-120b:free', 'qwen/qwen3-235b-a22b:free', 'tngtech/deepseek-r1t-chimera:free'];
-const DEFAULT_DOUBAO_MODEL = 'doubao-seed-1-6-251015'; 
+const DEFAULT_DOUBAO_MODEL = 'doubao-seed-1-6-251015'; // Default fallback
 
 // Runtime override logic
 let runtimeProviderOrder = [];
@@ -173,7 +173,7 @@ async function getDoubaoContextId(apiKey, model, systemMessage, userId) {
     return null;
 }
 
-async function streamDoubao(baseParams, res, userId, mode = 'chat') {
+async function streamDoubao(baseParams, res, userId, mode = 'chat', config) {
     const keys = await getKeyPool('doubao');
     if (keys.length === 0) throw new Error("No Doubao API keys configured");
     
@@ -181,7 +181,12 @@ async function streamDoubao(baseParams, res, userId, mode = 'chat') {
     const messages = convertGeminiToOpenAI(baseParams);
     if (messages.length === 0) throw new Error("Doubao: Empty messages");
 
-    const model = DEFAULT_DOUBAO_MODEL;
+    // DETERMINE MODEL: Use configured custom model (endpoint) if available, otherwise default
+    let model = DEFAULT_DOUBAO_MODEL;
+    if (config && config.doubaoModels && config.doubaoModels.length > 0) {
+        model = config.doubaoModels[0].id;
+    }
+
     const apiUrl = 'https://ark.cn-beijing.volces.com/api/v3/chat/completions';
 
     // Extract System Prompt for Context Creation if needed
@@ -220,7 +225,7 @@ async function streamDoubao(baseParams, res, userId, mode = 'chat') {
                 }
             }
 
-            console.log(`[AI] 🚀 Requesting Doubao...`);
+            console.log(`[AI] 🚀 Requesting Doubao (Model: ${model})...`);
             
             const response = await axios.post(apiUrl, requestBody, {
                 headers: {
@@ -424,7 +429,7 @@ async function streamContentWithSmartFallback(baseParams, res, userId, mode = 'c
             console.log(`[AI] 👉 Trying Provider: ${provider}... Mode: ${mode}`);
             if (provider === PROVIDERS.GEMINI) return await streamGemini(baseParams, res);
             else if (provider === PROVIDERS.OPENROUTER) return await streamOpenRouter(baseParams, res);
-            else if (provider === PROVIDERS.DOUBAO) return await streamDoubao(baseParams, res, userId, mode);
+            else if (provider === PROVIDERS.DOUBAO) return await streamDoubao(baseParams, res, userId, mode, config);
             else if (provider === PROVIDERS.GEMMA) return await streamGemma(baseParams, res);
         } catch (e) {
             console.error(`[AI] ❌ Provider ${provider} Failed: ${e.message}`);

components/ai/AdminPanel.tsx

@@ -1,8 +1,8 @@
 
 import React, { useState, useEffect } from 'react';
 import { api } from '../../services/api';
-import { SystemConfig, OpenRouterModelConfig } from '../../types';
-import { Bot, CheckCircle, X, BarChart2, Brain, Activity, ShieldCheck, Power, Key, Trash2, Plus, Save, Layers, ArrowUp, ArrowDown } from 'lucide-react';
+import { SystemConfig, OpenRouterModelConfig, DoubaoModelConfig } from '../../types';
+import { Bot, CheckCircle, X, BarChart2, Brain, Activity, ShieldCheck, Power, Key, Trash2, Plus, Save, Layers, ArrowUp, ArrowDown, Server } from 'lucide-react';
 import { BarChart, Bar, XAxis, YAxis, CartesianGrid, Tooltip, ResponsiveContainer, PieChart, Pie, Cell, Legend } from 'recharts';
 import { Toast, ToastState } from '../Toast';
 
@@ -36,12 +36,17 @@ export const AdminPanel: React.FC = () => {
     const [newOpenRouterKey, setNewOpenRouterKey] = useState('');
     const [newDoubaoKey, setNewDoubaoKey] = useState(''); // New Doubao Input
     
-    // Model Management
+    // Model Management - OpenRouter
     const [orModels, setOrModels] = useState<OpenRouterModelConfig[]>([]);
     const [newModelId, setNewModelId] = useState('');
     const [newModelName, setNewModelName] = useState('');
     const [newModelApiUrl, setNewModelApiUrl] = useState('');
 
+    // Model Management - Doubao
+    const [doubaoModels, setDoubaoModels] = useState<DoubaoModelConfig[]>([]);
+    const [newDoubaoModelId, setNewDoubaoModelId] = useState(''); // Endpoint ID
+    const [newDoubaoModelName, setNewDoubaoModelName] = useState('');
+
     // Provider Priority
     const [providerOrder, setProviderOrder] = useState<string[]>(['GEMINI', 'OPENROUTER', 'DOUBAO', 'GEMMA']);
 
@@ -60,6 +65,9 @@ export const AdminPanel: React.FC = () => {
             }
             setOrModels(cfg.openRouterModels && cfg.openRouterModels.length > 0 ? cfg.openRouterModels : DEFAULT_OR_MODELS);
             
+            // Load Doubao Models
+            setDoubaoModels(cfg.doubaoModels || []);
+
             if (cfg.aiProviderOrder && cfg.aiProviderOrder.length > 0) {
                 setProviderOrder(cfg.aiProviderOrder);
             }
@@ -99,6 +107,9 @@ export const AdminPanel: React.FC = () => {
         else if (type === 'doubao') setDoubaoKeys(doubaoKeys.filter((_, i) => i !== index));
     };
 
+    // --- Model Management Functions ---
+    
+    // OpenRouter
     const handleAddModel = () => {
         if (!newModelId.trim()) return;
         setOrModels([...orModels, { 
@@ -109,7 +120,6 @@ export const AdminPanel: React.FC = () => {
         }]);
         setNewModelId(''); setNewModelName(''); setNewModelApiUrl('');
     };
-
     const handleRemoveModel = (idx: number) => setOrModels(orModels.filter((_, i) => i !== idx));
     const handleMoveModel = (idx: number, direction: -1 | 1) => {
         const newArr = [...orModels]; const targetIdx = idx + direction;
@@ -118,6 +128,23 @@ export const AdminPanel: React.FC = () => {
         setOrModels(newArr);
     };
 
+    // Doubao
+    const handleAddDoubaoModel = () => {
+        if (!newDoubaoModelId.trim()) return;
+        setDoubaoModels([...doubaoModels, {
+            id: newDoubaoModelId.trim(),
+            name: newDoubaoModelName.trim() || newDoubaoModelId.trim()
+        }]);
+        setNewDoubaoModelId(''); setNewDoubaoModelName('');
+    };
+    const handleRemoveDoubaoModel = (idx: number) => setDoubaoModels(doubaoModels.filter((_, i) => i !== idx));
+    const handleMoveDoubaoModel = (idx: number, direction: -1 | 1) => {
+        const newArr = [...doubaoModels]; const targetIdx = idx + direction;
+        if (targetIdx < 0 || targetIdx >= newArr.length) return;
+        [newArr[idx], newArr[targetIdx]] = [newArr[targetIdx], newArr[idx]];
+        setDoubaoModels(newArr);
+    };
+
     const handleMoveProviderOrder = (idx: number, direction: -1 | 1) => {
         const newArr = [...providerOrder]; const targetIdx = idx + direction;
         if (targetIdx < 0 || targetIdx >= newArr.length) return;
@@ -132,6 +159,7 @@ export const AdminPanel: React.FC = () => {
                 ...systemConfig, 
                 apiKeys: { gemini: geminiKeys, openrouter: openRouterKeys, doubao: doubaoKeys }, 
                 openRouterModels: orModels,
+                doubaoModels: doubaoModels, // NEW
                 aiProviderOrder: providerOrder
             });
             await api.ai.resetPool();
@@ -239,6 +267,39 @@ export const AdminPanel: React.FC = () => {
                     </div>
                 </div>
 
+                {/* DOUBAO 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 flex items-center"><Server className="mr-2" size={16}/> 豆包/火山引擎模型管理 (接入点 ID)</h4></div>
+                    <div className="bg-blue-50 border border-blue-100 p-3 rounded-lg mb-4 text-xs text-blue-700">
+                        火山引擎使用 <b>接入点 ID (Endpoint ID)</b> 来标识模型部署。格式如 <code>ep-2024060401xxxx-xxxxx</code>。请在火山引擎控制台创建接入点后在此添加。
+                        <br/>系统将默认使用列表中的<b>第一个模型</b>。
+                    </div>
+                    <div className="space-y-2 mb-4 bg-gray-50 p-3 rounded-lg border border-gray-200">
+                        {doubaoModels.length === 0 ? <div className="text-center text-gray-400 text-xs py-2">暂无配置模型，将使用系统默认种子模型。</div> : doubaoModels.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={()=>handleMoveDoubaoModel(idx, -1)} className="text-gray-400 hover:text-blue-500 disabled:opacity-30" disabled={idx===0}><ArrowUp size={12}/></button>
+                                    <button onClick={()=>handleMoveDoubaoModel(idx, 1)} className="text-gray-400 hover:text-blue-500 disabled:opacity-30" disabled={idx===doubaoModels.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>
+                                </div>
+                                <div className="flex items-center gap-2">
+                                    {idx === 0 && <span className="text-[10px] bg-green-50 text-green-600 px-2 py-0.5 rounded font-bold">默认</span>}
+                                    <button onClick={() => handleRemoveDoubaoModel(idx)} className="p-1.5 rounded text-gray-400 hover:text-red-500 hover:bg-red-50"><Trash2 size={16}/></button>
+                                </div>
+                            </div>
+                        ))}
+                    </div>
+                    <div className="flex flex-col md:flex-row gap-2 items-end bg-gray-50 p-3 rounded-lg border border-gray-200">
+                        <div className="flex-1 w-full"><label className="text-xs text-gray-500 mb-1 block">接入点 ID (Endpoint ID) *</label><input className="w-full border border-gray-300 rounded px-2 py-1.5 text-sm" value={newDoubaoModelId} onChange={e=>setNewDoubaoModelId(e.target.value)} placeholder="如: ep-2024xxxx-xxxxx"/></div>
+                        <div className="flex-1 w-full"><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={newDoubaoModelName} onChange={e=>setNewDoubaoModelName(e.target.value)} placeholder="如: Doubao Pro 32k"/></div>
+                        <button onClick={handleAddDoubaoModel} className="bg-indigo-600 text-white px-4 py-1.5 rounded text-sm hover:bg-indigo-700 h-9 w-full md:w-auto">添加</button>
+                    </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">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>
@@ -252,4 +313,4 @@ export const AdminPanel: React.FC = () => {
             </div>
         </div>
     );
-};
+};

models.js

@@ -122,13 +122,19 @@ 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
+    // Updated default to include DOUBAO
+    aiProviderOrder: { type: [String], default: ['GEMINI', 'OPENROUTER', 'DOUBAO', 'GEMMA'] }, 
     periodConfig: [{ period: Number, name: String, startTime: String, endTime: String }],
     apiKeys: {
         gemini: [String],
         openrouter: [String],
         doubao: [String] // NEW: Doubao Key Pool
     },
+    // NEW: Doubao Custom Models (Endpoints)
+    doubaoModels: [{
+        id: String, // Endpoint ID (e.g. ep-2024xxxx)
+        name: String // Friendly Name
+    }],
     openRouterModels: [{ 
         id: String, 
         name: String, 

types.ts

@@ -177,6 +177,11 @@ export interface OpenRouterModelConfig {
   isCustom: boolean; // Built-in vs Custom
 }
 
+export interface DoubaoModelConfig {
+  id: string; // Endpoint ID (e.g., ep-2024...)
+  name?: string;
+}
+
 export interface SystemConfig {
   systemName: string;
   semester: string;
@@ -196,6 +201,7 @@ export interface SystemConfig {
       openrouter?: string[];
       doubao?: string[]; // NEW: Doubao keys
   };
+  doubaoModels?: DoubaoModelConfig[]; // NEW
   openRouterModels?: OpenRouterModelConfig[];
 }