Upload 67 files

ai-routes.js

@@ -2,7 +2,7 @@
 const express = require('express');
 const router = express.Router();
 const OpenAI = require('openai');
-const axios = require('axios'); // Imported Axios for Doubao Context (still needed for context creation)
+const axios = require('axios'); // Imported Axios for Doubao Context
 const { ConfigModel, User, AIUsageModel, ChatHistoryModel } = require('./models');
 const { buildUserContext } = require('./ai-context');
 
@@ -76,9 +76,49 @@ function convertGeminiToOpenAI(baseParams) {
     return messages;
 }
 
+// Convert Gemini params to Doubao Native Input Format
+function convertGeminiToDoubaoInput(baseParams) {
+    const inputs = [];
+    
+    // 1. System Prompt
+    const sysInstruction = baseParams.config?.systemInstruction || baseParams.systemInstruction;
+    if (sysInstruction) {
+        inputs.push({ role: 'system', content: [{ type: 'text', text: sysInstruction }] });
+    }
+
+    // 2. Chat History & Current Message
+    let contents = baseParams.contents;
+    if (contents && !Array.isArray(contents)) {
+        contents = [contents];
+    }
+
+    if (contents) {
+        contents.forEach(content => {
+            let role = (content.role === 'model' || content.role === 'assistant') ? 'assistant' : 'user';
+            const contentParts = [];
+            
+            if (content.parts) {
+                content.parts.forEach(p => {
+                    if (p.text) {
+                        // SPEC: User text must be "input_text", Assistant must be "text"
+                        const type = role === 'user' ? 'input_text' : 'text';
+                        contentParts.push({ type: type, text: p.text });
+                    }
+                    else if (p.inlineData && p.inlineData.mimeType.startsWith('image/')) {
+                        contentParts.push({ type: 'image_url', image_url: { url: `data:${p.inlineData.mimeType};base64,${p.inlineData.data}` } });
+                    }
+                });
+            }
+            if (contentParts.length > 0) {
+                inputs.push({ role: role, content: contentParts });
+            }
+        });
+    }
+    return inputs;
+}
+
 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'];
-// No hardcoded default endpoint because it varies per user
 
 // Runtime override logic
 let runtimeProviderOrder = [];
@@ -109,11 +149,9 @@ async function streamGemini(baseParams, res, enableThinking = false) {
             try {
                 console.log(`[AI] 🚀 Attempting Gemini Model: ${modelName} (Key ends with ...${apiKey.slice(-4)})`);
                 
-                // Add thinking config if requested
                 const requestParams = { ...baseParams, model: modelName };
                 if (enableThinking) {
                     requestParams.config = requestParams.config || {};
-                    // Gemini 2.5 Flash supports thinking config (set budget)
                     requestParams.config.thinkingConfig = { thinkingBudget: 1024 }; 
                 }
 
@@ -128,11 +166,6 @@ async function streamGemini(baseParams, res, enableThinking = false) {
                         recordUsage(modelName, PROVIDERS.GEMINI);
                         hasStarted = true;
                     }
-                    
-                    // Check for thought content (provider specific, Gemini SDK usually keeps it in candidates)
-                    // Note: Current Google GenAI Node SDK might not separate thought perfectly in stream without checking complex response structure.
-                    // For now, we assume Gemini just streams text. If Gemini adds explicit thought parts in future SDKs, we parse here.
-                    
                     if (chunk.text) {
                         fullText += chunk.text;
                         res.write(`data: ${JSON.stringify({ type: 'text', content: chunk.text })}\n\n`);
@@ -153,46 +186,86 @@ async function streamGemini(baseParams, res, enableThinking = false) {
     throw new Error("Gemini streaming failed (All keys/models exhausted)");
 }
 
-// --- DOUBAO DIRECT STREAMING (Axios) ---
-
-async function streamDoubao(baseParams, res, userId, mode = 'chat', config, enableThinking = false) {
+// --- DOUBAO DIRECT STREAMING (Ark /responses) ---
+async function streamDoubao(baseParams, res, username, mode = 'chat', config, enableThinking = false, enableSearch = false) {
     const keys = await getKeyPool('doubao');
     if (keys.length === 0) throw new Error("No Doubao API keys configured");
-    
-    // Convert to OpenAI format
-    const messages = convertGeminiToOpenAI(baseParams);
-    if (messages.length === 0) throw new Error("Doubao: Empty messages");
 
-    // DETERMINE ENDPOINT
+    // DETERMINE ENDPOINT & MODEL
     if (!config || !config.doubaoModels || config.doubaoModels.length === 0) {
         throw new Error("Doubao requires an Endpoint ID configured in Admin Panel.");
     }
     const doubaoConfig = config.doubaoModels[0];
-    const endpointId = doubaoConfig.endpointId; // MUST be ep-xxxx
+    const endpointId = doubaoConfig.endpointId; 
     const modelId = doubaoConfig.modelId || 'Doubao'; 
 
+    // --- CONTEXT LOGIC START ---
+    const user = await User.findOne({ username });
+    const lastId = user?.doubaoState?.responseId;
+    const lastThinking = user?.doubaoState?.thinkingState;
+
+    // 1. Decide on Caching
+    // Rule: Search disables cache. Otherwise enable it.
+    const requestCachingType = enableSearch ? "disabled" : "enabled";
+
+    // 2. Decide on ID Reuse
+    // Rule: Must have ID + No Search + Thinking State Unchanged
+    let idToSend = null;
+    // Note: 'undefined' matches 'false' effectively for our boolean logic check if we cast both
+    if (lastId && !enableSearch && (!!lastThinking === !!enableThinking)) {
+        idToSend = lastId;
+    }
+
+    // 3. Build Input
+    let inputPayload = [];
+    if (idToSend) {
+        // INCREMENTAL MODE: We assume the server has context.
+        // We need to send System Prompt + NEWEST Message only.
+        // However, 'convertGeminiToDoubaoInput' converts the whole baseParams list.
+        // We need to extract just the System and the last User message.
+        
+        const fullInputs = convertGeminiToDoubaoInput(baseParams);
+        
+        const systemMsg = fullInputs.find(m => m.role === 'system');
+        // Get the last message (User's new question)
+        // Gemini format 'contents' usually ends with the new prompt.
+        const lastMsg = fullInputs[fullInputs.length - 1]; 
+        
+        if (systemMsg) inputPayload.push(systemMsg);
+        if (lastMsg) inputPayload.push(lastMsg);
+    } else {
+        // FULL MODE: ID Invalid/Missing/SearchOn -> Send Full History
+        inputPayload = convertGeminiToDoubaoInput(baseParams);
+    }
+    // --- CONTEXT LOGIC END ---
+
+    // Build Request Body
+    const requestBody = {
+        model: endpointId,
+        stream: true,
+        input: inputPayload,
+        thinking: { type: enableThinking ? "enabled" : "disabled" },
+        caching: { type: requestCachingType },
+        // Add ID if valid
+        ...(idToSend && { previous_response_id: idToSend }),
+        // Add Search Tools if enabled
+        ...(enableSearch && { tools: [{ type: "web_search", web_search: { search_result: true } }] })
+    };
+
+    console.log(`[AI] 🚀 Doubao Ark Request: EP=${endpointId}, Cache=${requestCachingType}, Think=${enableThinking}, Search=${enableSearch}, PrevID=${idToSend ? 'YES' : 'NO'}`);
+
     for (const apiKey of keys) {
         try {
-            console.log(`[AI] 🚀 Calling Doubao API (Axios): ${endpointId}, Thinking: ${enableThinking}`);
-            
-            // Payload matching user's CURL example
-            const payload = {
-                model: endpointId,
-                messages: messages,
-                stream: true,
-                thinking: { type: enableThinking ? "enabled" : "disabled" } 
-            };
-
             const response = await axios.post(
-                'https://ark.cn-beijing.volces.com/api/v3/chat/completions',
-                payload,
+                'https://ark.cn-beijing.volces.com/api/v3/responses',
+                requestBody,
                 {
                     headers: {
                         'Content-Type': 'application/json',
                         'Authorization': `Bearer ${apiKey}`
                     },
                     responseType: 'stream',
-                    timeout: 60000 // 60s timeout
+                    timeout: 60000
                 }
             );
 
@@ -200,19 +273,18 @@ async function streamDoubao(baseParams, res, userId, mode = 'chat', config, enab
             recordUsage(modelId, PROVIDERS.DOUBAO);
 
             let fullText = "";
-            let fullThought = "";
+            let newResponseID = null;
             let hasStarted = false;
             let buffer = "";
 
             const stream = response.data;
 
-            // Handle Node.js stream
             for await (const chunk of stream) {
                 if (!hasStarted) hasStarted = true;
                 
                 buffer += chunk.toString();
                 const lines = buffer.split('\n');
-                buffer = lines.pop(); // Keep incomplete line
+                buffer = lines.pop(); 
 
                 for (const line of lines) {
                     const trimmed = line.trim();
@@ -222,29 +294,71 @@ async function streamDoubao(baseParams, res, userId, mode = 'chat', config, enab
                         try {
                             const jsonStr = trimmed.substring(6);
                             const json = JSON.parse(jsonStr);
-                            const delta = json.choices?.[0]?.delta;
                             
-                            // Handle Thinking Content (DeepSeek/Doubao format usually puts it in reasoning_content)
-                            const reasoning = delta?.reasoning_content;
-                            const content = delta?.content;
+                            // 1. Get Response ID (Anchor)
+                            if (json.response && json.response.id) {
+                                newResponseID = json.response.id;
+                            }
 
-                            if (reasoning) {
-                                fullThought += reasoning;
-                                res.write(`data: ${JSON.stringify({ type: 'thinking', content: reasoning })}\n\n`);
+                            // 2. Handle Content (Reasoning vs Output)
+                            if (json.response && json.response.reasoning_text && json.response.reasoning_text.delta) {
+                                res.write(`data: ${JSON.stringify({ type: 'thinking', content: json.response.reasoning_text.delta })}\n\n`);
                                 if (res.flush) res.flush();
                             }
                             
-                            if (content) {
+                            if (json.response && json.response.output_text && json.response.output_text.delta) {
+                                const content = json.response.output_text.delta;
                                 fullText += content;
                                 res.write(`data: ${JSON.stringify({ type: 'text', content: content })}\n\n`);
                                 if (res.flush) res.flush();
                             }
+
+                            // 3. Handle Search Status
+                            if (json.response && json.response.web_search_call && json.response.web_search_call.searching) {
+                                // Send a status update to UI (formatted as text or custom type)
+                                // We send a temporary thinking-like message or just custom type
+                                res.write(`data: ${JSON.stringify({ type: 'thinking', content: '\n\n🌐 正在联网搜索...\n\n' })}\n\n`); 
+                                if (res.flush) res.flush();
+                            }
+
+                            // 4. Handle References (End of stream usually)
+                            if (json.response && json.response.content_part && json.response.content_part.done) {
+                                const annotations = json.response.content_part.annotations || [];
+                                if (annotations.length > 0) {
+                                    let refText = "\n\n**引用来源:**\n";
+                                    annotations.forEach((ref, idx) => {
+                                        refText += `[${idx + 1}] [${ref.title}](${ref.url})\n`;
+                                    });
+                                    fullText += refText;
+                                    res.write(`data: ${JSON.stringify({ type: 'text', content: refText })}\n\n`);
+                                    if (res.flush) res.flush();
+                                }
+                            }
+
                         } catch (e) {
-                            // Ignore parse errors for partial chunks or keepalives
+                            // Ignore parse errors for partial chunks
                         }
                     }
                 }
             }
+
+            // 4. Update User State (Context Persistence)
+            // Rules: Only save ID if Search was OFF.
+            if (!enableSearch && newResponseID) {
+                await User.findOneAndUpdate({ username }, {
+                    doubaoState: {
+                        responseId: newResponseID,
+                        thinkingState: enableThinking
+                    }
+                });
+                console.log(`[AI] 💾 Doubao Context Updated: ID=${newResponseID}`);
+            } else {
+                // If search was ON, ID is invalid for next turn. Clear it to force full history next time.
+                await User.findOneAndUpdate({ username }, {
+                    doubaoState: { responseId: null, thinkingState: null }
+                });
+                console.log(`[AI] 🧹 Doubao Context Cleared (Search Active or No ID)`);
+            }
             
             return fullText;
 
@@ -254,6 +368,16 @@ async function streamDoubao(baseParams, res, userId, mode = 'chat', config, enab
                 console.log(`[AI] 🔄 Quota exceeded, trying next key...`);
                 continue;
             }
+            // If error is 400 (Bad Request), it might be due to invalid ID.
+            // Retry ONCE with ID=null (Full History) if we tried with ID
+            if (e.response?.status === 400 && idToSend) {
+                console.log(`[AI] ⚠️ 400 Error with ID. Retrying with full history...`);
+                idToSend = null;
+                // Recursive retry with modified params (careful with infinite loops, handled by loop structure somewhat, but strictly we should just clear DB and fail this request gracefully or implement simpler retry)
+                await User.findOneAndUpdate({ username }, { doubaoState: { responseId: null } });
+                // We won't recurse here to keep it simple, just fail and let user retry (which will pick up null ID)
+                throw new Error("Context invalid. Please retry.");
+            }
             throw e;
         }
     }
@@ -279,22 +403,13 @@ async function streamOpenRouter(baseParams, res) {
             
             const client = new OpenAI({ baseURL, apiKey, defaultHeaders: { "HTTP-Referer": "https://smart.com", "X-Title": "Smart School" } });
             
-            const extraBody = {};
-            // If user uses Doubao via OpenRouter/Custom Proxy, also try to apply cache/thinking params
-            if (modelName.toLowerCase().includes('doubao')) {
-                extraBody.caching = { type: "enabled", prefix: true };
-                // We rely on the request passing thinking enabled, but OpenRouter implementation might vary
-                // For now, standard OpenRouter doesn't support 'thinking' param standardly, mostly handled via model choice (e.g. R1)
-            }
-
             try {
-                console.log(`[AI] 🚀 Attempting ${providerLabel} Model: ${modelName} (URL: ${baseURL})`);
+                console.log(`[AI] 🚀 Attempting ${providerLabel} Model: ${modelName}`);
                 
                 const stream = await client.chat.completions.create({ 
                     model: modelName, 
                     messages, 
-                    stream: true,
-                    ...extraBody 
+                    stream: true
                 });
                 
                 console.log(`[AI] ✅ Connected to ${providerLabel}: ${modelName}`);
@@ -305,7 +420,6 @@ async function streamOpenRouter(baseParams, res) {
                     const text = chunk.choices[0]?.delta?.content || '';
                     if (text) {
                         fullText += text;
-                        // FIX: Use { type: 'text', content: ... }
                         res.write(`data: ${JSON.stringify({ type: 'text', content: text })}\n\n`);
                         if (res.flush) res.flush();
                     }
@@ -313,14 +427,11 @@ async function streamOpenRouter(baseParams, res) {
                 return fullText;
             } catch (e) { 
                 console.warn(`[AI] ⚠️ ${providerLabel} ${modelName} Error: ${e.message}`);
-                if (isQuotaError(e)) {
-                    console.log(`[AI] 🔄 Rate limit/Quota for ${modelName}, switching...`);
-                    break;
-                }
+                if (isQuotaError(e)) break;
             }
         }
     }
-    throw new Error("OpenRouter/Custom stream failed (All models exhausted)");
+    throw new Error("OpenRouter stream failed");
 }
 
 async function streamGemma(baseParams, res) {
@@ -346,7 +457,6 @@ async function streamGemma(baseParams, res) {
                     }
                     if (chunk.text) {
                         fullText += chunk.text;
-                        // FIX: Use { type: 'text', content: ... }
                         res.write(`data: ${JSON.stringify({ type: 'text', content: chunk.text })}\n\n`);
                         if (res.flush) res.flush();
                     }
@@ -361,7 +471,7 @@ async function streamGemma(baseParams, res) {
     throw new Error("Gemma stream failed");
 }
 
-async function streamContentWithSmartFallback(baseParams, res, userId, mode = 'chat', enableThinking = false) {
+async function streamContentWithSmartFallback(baseParams, res, username, mode = 'chat', enableThinking = false, enableSearch = false) {
     let hasAudio = false;
     const contentsArray = Array.isArray(baseParams.contents) ? baseParams.contents : [baseParams.contents];
     
@@ -395,10 +505,10 @@ async function streamContentWithSmartFallback(baseParams, res, userId, mode = 'c
     let finalError = null;
     for (const provider of runtimeProviderOrder) {
         try {
-            console.log(`[AI] 👉 Trying Provider: ${provider}... Mode: ${mode}, Thinking: ${enableThinking}`);
+            console.log(`[AI] 👉 Trying Provider: ${provider}... Mode: ${mode}, Thinking: ${enableThinking}, Search: ${enableSearch}`);
             if (provider === PROVIDERS.GEMINI) return await streamGemini(baseParams, res, enableThinking);
             else if (provider === PROVIDERS.OPENROUTER) return await streamOpenRouter(baseParams, res);
-            else if (provider === PROVIDERS.DOUBAO) return await streamDoubao(baseParams, res, userId, mode, config, enableThinking);
+            else if (provider === PROVIDERS.DOUBAO) return await streamDoubao(baseParams, res, username, mode, config, enableThinking, enableSearch);
             else if (provider === PROVIDERS.GEMMA) return await streamGemma(baseParams, res);
         } catch (e) {
             console.error(`[AI] ❌ Provider ${provider} Failed: ${e.message}`);
@@ -457,20 +567,16 @@ router.post('/reset-pool', checkAIAccess, (req, res) => {
     console.log('[AI] 🔄 Provider priority pool reset.');
     res.json({ success: true }); 
 });
-// --- PERSISTENT CHAT HISTORY HANDLER ---
-// Instead of relying on client-side 'history', we use MongoDB to ensure cross-device memory.
+
 router.post('/chat', async (req, res) => {
-    const { text, audio, images, history, enableThinking, overrideSystemPrompt } = req.body; // Added images, enableThinking, overrideSystemPrompt
+    const { text, audio, images, history, enableThinking, overrideSystemPrompt, enableSearch } = req.body; 
     const userRole = req.headers['x-user-role'];
     const username = req.headers['x-user-username'];
     const schoolId = req.headers['x-school-id'];
 
-    // ... (Keep Context building logic same) ...
-    // If overrideSystemPrompt is provided (Work Assistant), use it. Otherwise build standard context.
+    // ... (Context building logic)
     const systemInstruction = overrideSystemPrompt || await buildUserContext(username, userRole, schoolId);
 
-    // Build History
-    // Filter out messages with empty text to prevent API errors
     const geminiHistory = (history || [])
         .filter(msg => msg.text && msg.text.trim() !== '')
         .map(msg => ({
@@ -478,23 +584,15 @@ router.post('/chat', async (req, res) => {
             parts: [{ text: msg.text }]
         }));
 
-    // Build Current Message Parts
     const currentParts = [];
-    
-    // 1. Text Instruction (Priority)
-    // FIX: Put text FIRST to ensure model pays attention to instructions before processing media
     if (text && text.trim()) {
         currentParts.push({ text: text });
     } else if (audio) {
-        // Fix: Explicit instruction for audio, placed before audio data if possible
         currentParts.push({ text: "用户发送了语音消息，请听录音并回答。" });
     } else {
-        // Fallback
         currentParts.push({ text: "." });
     }
 
-    // 2. Media (Images/Audio)
-    // Add Images
     if (images && Array.isArray(images)) {
         images.forEach(base64 => {
             if (base64) {
@@ -508,7 +606,6 @@ router.post('/chat', async (req, res) => {
         });
     }
 
-    // Add Audio
     if (audio) {
         currentParts.push({
             inlineData: {
@@ -518,7 +615,6 @@ router.post('/chat', async (req, res) => {
         });
     }
 
-    // Prepare Response Stream
     res.setHeader('Content-Type', 'text/event-stream');
     res.setHeader('Cache-Control', 'no-cache');
     res.setHeader('Connection', 'keep-alive');
@@ -527,18 +623,13 @@ router.post('/chat', async (req, res) => {
     let fullText = "";
 
     try {
-        // 1. Generate Text (Stream)
-        // FIX: Place systemInstruction inside `config` object to comply with Gemini SDK and ensure visibility for Doubao conversion
         fullText = await streamContentWithSmartFallback({
             contents: [...geminiHistory, { role: 'user', parts: currentParts }],
             config: {
                 systemInstruction: systemInstruction
             }
-        }, res, req.headers['x-user-username'], 'chat', enableThinking);
+        }, res, req.headers['x-user-username'], 'chat', enableThinking, enableSearch);
         
-        // 2. Save User Message to DB
-        // For Work Assistant, we might want to segregate history, but for simplicity we save all to same collection
-        // Client-side can filter or use distinct states
         await ChatHistoryModel.create({
             userId: req.headers['x-user-username'], 
             role: 'user',
@@ -546,9 +637,6 @@ router.post('/chat', async (req, res) => {
             timestamp: Date.now()
         });
 
-        // 3. Generate TTS (Independent of Text Model)
-        // Only generate audio if text is sufficient length, not just punctuation, AND NO Thinking logic (Work Assistant disables audio usually)
-        // Check req.body.disableAudio which might be passed by Work Assistant
         if (fullText && fullText.length > 2 && !req.body.disableAudio) {
             res.write(`data: ${JSON.stringify({ type: 'status', status: 'tts' })}\n\n`);
             try {
@@ -556,13 +644,12 @@ router.post('/chat', async (req, res) => {
                 const keys = await getKeyPool('gemini');
                 let audioData = null;
                 
-                // Retry logic for TTS
                 for (const apiKey of keys) {
                     try {
                         const client = new GoogleGenAI({ apiKey });
                         const ttsResponse = await client.models.generateContent({
                             model: "gemini-2.5-flash-preview-tts",
-                            contents: [{ parts: [{ text: fullText.substring(0, 500) }] }], // Limit TTS length for speed
+                            contents: [{ parts: [{ text: fullText.substring(0, 500) }] }],
                             config: { responseModalities: ['AUDIO'], speechConfig: { voiceConfig: { prebuiltVoiceConfig: { voiceName: 'Kore' } } } }
                         });
                         audioData = ttsResponse.candidates?.[0]?.content?.parts?.[0]?.inlineData?.data;
@@ -576,16 +663,12 @@ router.post('/chat', async (req, res) => {
                     res.write(`data: ${JSON.stringify({ type: 'status', ttsSkipped: true })}\n\n`);
                 }
             } catch (ttsErr) {
-                console.error("TTS Error:", ttsErr);
                 res.write(`data: ${JSON.stringify({ type: 'status', ttsSkipped: true })}\n\n`);
             }
         } else {
              res.write(`data: ${JSON.stringify({ type: 'status', ttsSkipped: true })}\n\n`);
         }
 
-        // 4. Save Model Message to DB
-        // If we have audio, we could technically save it, but MongoDB limits document size. 
-        // For now, we rely on text history and client-side TTS fallback if audio is lost on refresh.
         await ChatHistoryModel.create({
             userId: req.headers['x-user-username'],
             role: 'model',
@@ -605,11 +688,8 @@ router.post('/chat', async (req, res) => {
     }
 });
 
-// STREAMING ASSESSMENT ENDPOINT
 router.post('/evaluate', checkAIAccess, async (req, res) => {
     const { question, audio, image, images } = req.body;
-    
-    // Extract User info for userId passing
     const username = req.headers['x-user-username'];
     const user = await User.findOne({ username });
     
@@ -627,19 +707,16 @@ router.post('/evaluate', checkAIAccess, async (req, res) => {
             evalParts.push({ inlineData: { mimeType: 'audio/webm', data: audio } });
         }
         
-        // Support multiple images
         if (images && Array.isArray(images) && images.length > 0) {
             evalParts.push({ text: "学生的回答写在以下图片中，请识别所有图片中的文字内容并进行批改：" });
             images.forEach(img => {
                 if(img) evalParts.push({ inlineData: { mimeType: 'image/jpeg', data: img } });
             });
         } else if (image) {
-            // Legacy single image support
             evalParts.push({ text: "学生的回答写在图片中，请识别图片中的文字内容并进行批改。" });
             evalParts.push({ inlineData: { mimeType: 'image/jpeg', data: image } });
         }
 
-        // Force structured markdown output for streaming parsing
         evalParts.push({ text: `请分析：1. 内容准确性 2. 表达/书写规范。
         必须严格按照以下格式输出(不要使用Markdown代码块包裹):
         
@@ -652,18 +729,13 @@ router.post('/evaluate', checkAIAccess, async (req, res) => {
         ## Score
         (在此处仅输出一个0-100的数字)` });
 
-        // Stream Text
         const fullText = await streamContentWithSmartFallback({
-            // CRITICAL FIX: Pass as array of objects for OpenRouter compatibility
             contents: [{ role: 'user', parts: evalParts }],
-            // NO JSON MODE to allow progressive text streaming
-        }, res, user?._id, 'evaluate'); // Pass mode='evaluate'
+        }, res, user?._id, 'evaluate'); 
 
-        // Extract Feedback for TTS
         const feedbackMatch = fullText.match(/## Feedback\s+([\s\S]*?)(?=## Score|$)/i);
         const feedbackText = feedbackMatch ? feedbackMatch[1].trim() : "";
 
-        // Generate TTS if feedback exists
         if (feedbackText) {
             res.write(`data: ${JSON.stringify({ status: 'tts' })}\n\n`);
             try {

components/ai/ChatPanel.tsx

@@ -1,7 +1,7 @@
 
 import React, { useState, useRef, useEffect } from 'react';
 import { AIChatMessage, User } from '../../types';
-import { Bot, Mic, Square, Volume2, Send, Sparkles, Loader2, Image as ImageIcon, Trash2, X, StopCircle } from 'lucide-react';
+import { Bot, Mic, Square, Volume2, Send, Sparkles, Loader2, Image as ImageIcon, Trash2, X, StopCircle, Globe, Brain } from 'lucide-react';
 import ReactMarkdown from 'react-markdown';
 import remarkGfm from 'remark-gfm';
 import { blobToBase64, base64ToUint8Array, decodePCM, cleanTextForTTS, compressImage } from '../../utils/mediaHelpers';
@@ -30,6 +30,10 @@ export const ChatPanel: React.FC<ChatPanelProps> = ({ currentUser }) => {
     const [textInput, setTextInput] = useState('');
     const [isRecording, setIsRecording] = useState(false);
     
+    // Config States
+    const [enableThinking, setEnableThinking] = useState(false);
+    const [enableSearch, setEnableSearch] = useState(false);
+    
     // Attachments
     const [selectedImages, setSelectedImages] = useState<File[]>([]);
     const [audioAttachment, setAudioAttachment] = useState<string | null>(null); // Base64
@@ -37,6 +41,7 @@ export const ChatPanel: React.FC<ChatPanelProps> = ({ currentUser }) => {
     const [isChatProcessing, setIsChatProcessing] = useState(false);
     const [playingMessageId, setPlayingMessageId] = useState<string | null>(null);
     const [toast, setToast] = useState<ToastState>({ show: false, message: '', type: 'success' });
+    const [isThinkingExpanded, setIsThinkingExpanded] = useState<Record<string, boolean>>({});
     
     const mediaRecorderRef = useRef<MediaRecorder | null>(null);
     const audioChunksRef = useRef<Blob[]>([]);
@@ -70,7 +75,7 @@ export const ChatPanel: React.FC<ChatPanelProps> = ({ currentUser }) => {
 
     useEffect(() => {
         messagesEndRef.current?.scrollIntoView({ behavior: 'smooth', block: 'end' });
-    }, [messages, isChatProcessing]);
+    }, [messages, isChatProcessing, isThinkingExpanded]);
 
     const stopPlayback = () => {
         if (currentSourceRef.current) {
@@ -179,14 +184,11 @@ export const ChatPanel: React.FC<ChatPanelProps> = ({ currentUser }) => {
         
         setIsChatProcessing(true);
 
-        // Define ID outside try block so it is available in catch block
         const newAiMsgId = (Date.now() + 1).toString();
 
         try {
-            // Process Images to Base64 first to update UI correctly
             const base64Images = await Promise.all(currentImages.map(f => compressImage(f)));
 
-            // Optimistic UI Update with Base64 Images
             const newUserMsg: AIChatMessage = { 
                 id: Date.now().toString(), 
                 role: 'user', 
@@ -196,9 +198,10 @@ export const ChatPanel: React.FC<ChatPanelProps> = ({ currentUser }) => {
                 timestamp: Date.now() 
             };
             
-            const newAiMsg: AIChatMessage = { id: newAiMsgId, role: 'model', text: '', timestamp: Date.now() };
+            const newAiMsg: AIChatMessage = { id: newAiMsgId, role: 'model', text: '', thought: '', timestamp: Date.now() };
 
             setMessages(prev => [...prev, newUserMsg, newAiMsg]);
+            if (enableThinking) setIsThinkingExpanded(prev => ({ ...prev, [newAiMsgId]: true }));
 
             const response = await fetch('/api/ai/chat', {
                 method: 'POST',
@@ -212,7 +215,9 @@ export const ChatPanel: React.FC<ChatPanelProps> = ({ currentUser }) => {
                     text: currentText, 
                     audio: currentAudio, 
                     images: base64Images, 
-                    history: messages.filter(m => m.id !== 'welcome').map(m => ({ role: m.role, text: m.text }))
+                    history: messages.filter(m => m.id !== 'welcome').map(m => ({ role: m.role, text: m.text })),
+                    enableThinking,
+                    enableSearch
                 })
             });
 
@@ -222,6 +227,7 @@ export const ChatPanel: React.FC<ChatPanelProps> = ({ currentUser }) => {
             const reader = response.body.getReader();
             const decoder = new TextDecoder();
             let aiTextAccumulated = '';
+            let aiThoughtAccumulated = '';
             let buffer = '';
 
             while (true) {
@@ -238,37 +244,32 @@ export const ChatPanel: React.FC<ChatPanelProps> = ({ currentUser }) => {
                         try {
                             const data = JSON.parse(jsonStr);
                             
-                            // 1. Text Update
                             if (data.type === 'text') {
+                                if (aiTextAccumulated === '' && aiThoughtAccumulated !== '') {
+                                    setIsThinkingExpanded(prev => ({ ...prev, [newAiMsgId]: false }));
+                                }
                                 aiTextAccumulated += data.content;
                                 setMessages(prev => prev.map(m => m.id === newAiMsgId ? { ...m, text: aiTextAccumulated } : m));
                             } 
-                            
-                            // 2. Status Update (e.g. Generating TTS)
+                            else if (data.type === 'thinking') {
+                                aiThoughtAccumulated += data.content;
+                                setMessages(prev => prev.map(m => m.id === newAiMsgId ? { ...m, thought: aiThoughtAccumulated } : m));
+                            }
                             else if (data.type === 'status' && data.status === 'tts') {
                                 setMessages(prev => prev.map(m => m.id === newAiMsgId ? { ...m, isGeneratingAudio: true } : m));
                             }
-
-                            // 3. Audio Arrived
                             else if (data.type === 'audio') {
-                                // Update state: save audio, clear generating flag
                                 setMessages(prev => prev.map(m => m.id === newAiMsgId ? { ...m, audio: data.audio, isGeneratingAudio: false } : m));
-                                // Auto-play
                                 const tempMsg = { ...newAiMsg, text: aiTextAccumulated, audio: data.audio };
                                 playAudio(tempMsg);
                             }
-                            
-                            // 4. TTS Skipped (play browser TTS)
                             else if (data.type === 'status' && data.ttsSkipped) {
                                 setMessages(prev => prev.map(m => m.id === newAiMsgId ? { ...m, isGeneratingAudio: false } : m));
-                                // Fallback play
                                 if (aiTextAccumulated) {
                                     const tempMsg = { ...newAiMsg, text: aiTextAccumulated };
                                     playAudio(tempMsg);
                                 }
                             }
-                            
-                            // 5. Error
                             else if (data.type === 'error') {
                                 setMessages(prev => prev.map(m => m.id === newAiMsgId ? { ...m, text: `⚠️ 错误: ${data.message}`, isGeneratingAudio: false } : m));
                             }
@@ -291,12 +292,13 @@ export const ChatPanel: React.FC<ChatPanelProps> = ({ currentUser }) => {
         <div className="flex-1 flex flex-col max-w-4xl mx-auto w-full min-h-0 relative overflow-hidden h-full">
             {toast.show && <Toast message={toast.message} type={toast.type} onClose={()=>setToast({...toast, show: false})}/>}
             
-            <div className="absolute top-2 right-4 z-10">
+            <div className="absolute top-2 right-4 z-10 flex gap-2">
                 <button onClick={clearHistory} className="text-xs text-gray-400 hover:text-red-500 flex items-center gap-1 bg-white/80 p-1.5 rounded-lg border border-transparent hover:border-red-100 transition-all shadow-sm backdrop-blur">
                     <Trash2 size={14}/> 清除
                 </button>
             </div>
 
+            {/* Chat History */}
             <div className="flex-1 overflow-y-auto p-4 space-y-4 pb-4 custom-scrollbar">
                 {messages.map(msg => (
                     <div key={msg.id} className={`flex gap-3 ${msg.role === 'user' ? 'flex-row-reverse' : ''}`}>
@@ -304,6 +306,24 @@ export const ChatPanel: React.FC<ChatPanelProps> = ({ currentUser }) => {
                             {msg.role === 'model' ? <Sparkles size={20}/> : <Bot size={20}/>}
                         </div>
                         <div className={`max-w-[85%] flex flex-col items-start ${msg.role === 'user' ? 'items-end' : ''}`}>
+                            {msg.role === 'model' && msg.thought && (
+                                <div className="w-full bg-purple-50 rounded-xl border border-purple-100 overflow-hidden mb-2 max-w-full">
+                                    <button 
+                                        onClick={() => setIsThinkingExpanded(prev => ({ ...prev, [msg.id]: !prev[msg.id] }))}
+                                        className="w-full px-4 py-2 flex items-center gap-2 text-xs font-bold text-purple-700 bg-purple-100/50 hover:bg-purple-100 transition-colors"
+                                    >
+                                        <Brain size={14}/> 
+                                        <span>深度思考过程</span>
+                                        <div className="ml-auto text-xs opacity-50">{isThinkingExpanded[msg.id] ? '收起' : '展开'}</div>
+                                    </button>
+                                    {isThinkingExpanded[msg.id] && (
+                                        <div className="p-4 text-xs text-purple-800 whitespace-pre-wrap leading-relaxed border-t border-purple-100 font-mono bg-white/50">
+                                            {msg.thought}
+                                        </div>
+                                    )}
+                                </div>
+                            )}
+
                             <div className={`p-3 rounded-2xl text-sm overflow-hidden shadow-sm ${msg.role === 'user' ? 'bg-blue-600 text-white rounded-tr-none' : 'bg-white border border-gray-200 text-gray-800 rounded-tl-none'}`}>
                                 {msg.images && msg.images.length > 0 && (
                                     <div className="flex gap-2 mb-2 flex-wrap">
@@ -327,7 +347,6 @@ export const ChatPanel: React.FC<ChatPanelProps> = ({ currentUser }) => {
                                     </div>
                                 )}
                                 
-                                {/* Play Button Logic */}
                                 {(msg.role === 'model' && (msg.text || msg.audio) && !isChatProcessing && !msg.isGeneratingAudio) && (
                                     <button 
                                         onClick={() => playingMessageId === msg.id ? stopPlayback() : playAudio(msg)} 
@@ -351,6 +370,27 @@ export const ChatPanel: React.FC<ChatPanelProps> = ({ currentUser }) => {
             {/* Improved Input Area */}
             <div className="p-4 bg-white border-t border-gray-200 shrink-0 z-20">
                 <div className="max-w-4xl mx-auto flex flex-col gap-2">
+                    
+                    {/* Toolbar */}
+                    <div className="flex justify-between items-center px-1">
+                        <div className="flex gap-3">
+                            <button 
+                                onClick={() => setEnableThinking(!enableThinking)}
+                                className={`flex items-center gap-1 text-xs px-2 py-1 rounded-md border transition-all ${enableThinking ? 'bg-purple-50 text-purple-600 border-purple-200' : 'bg-gray-50 text-gray-500 border-transparent hover:bg-gray-100'}`}
+                                title="开启后AI将进行深度思考 (仅部分模型支持)"
+                            >
+                                <Brain size={14} className={enableThinking ? "fill-current" : ""}/> 深度思考
+                            </button>
+                            <button 
+                                onClick={() => setEnableSearch(!enableSearch)}
+                                className={`flex items-center gap-1 text-xs px-2 py-1 rounded-md border transition-all ${enableSearch ? 'bg-blue-50 text-blue-600 border-blue-200' : 'bg-gray-50 text-gray-500 border-transparent hover:bg-gray-100'}`}
+                                title="开启后AI将联网搜索最新信息 (Doubao模型支持)"
+                            >
+                                <Globe size={14}/> 联网搜索
+                            </button>
+                        </div>
+                    </div>
+
                     {/* Attachments Preview */}
                     {(selectedImages.length > 0 || audioAttachment) && (
                         <div className="flex gap-2 overflow-x-auto pb-2">

components/ai/WorkAssistantPanel.tsx

@@ -1,7 +1,7 @@
 
 import React, { useState, useRef, useEffect } from 'react';
 import { AIChatMessage, User } from '../../types';
-import { Bot, Send, Sparkles, Loader2, Image as ImageIcon, X, Trash2, Brain, ChevronDown, ChevronRight, Copy, Check, FileText, Plus, Paperclip, File } from 'lucide-react';
+import { Bot, Send, Sparkles, Loader2, Image as ImageIcon, X, Trash2, Brain, ChevronDown, ChevronRight, Copy, Check, FileText, Plus, Paperclip, File, Globe } from 'lucide-react';
 import ReactMarkdown from 'react-markdown';
 import remarkGfm from 'remark-gfm';
 import { compressImage } from '../../utils/mediaHelpers';
@@ -142,6 +142,7 @@ const ROLES = [
 export const WorkAssistantPanel: React.FC<WorkAssistantPanelProps> = ({ currentUser }) => {
     const [selectedRole, setSelectedRole] = useState(ROLES[0]);
     const [enableThinking, setEnableThinking] = useState(false);
+    const [enableSearch, setEnableSearch] = useState(false);
     
     // Chat State
     const [messages, setMessages] = useState<AIChatMessage[]>([]);
@@ -217,7 +218,6 @@ export const WorkAssistantPanel: React.FC<WorkAssistantPanelProps> = ({ currentU
         setSelectedImages([]);
         clearDoc();
 
-        // ID generation using UUID to prevent collisions
         const userMsgId = crypto.randomUUID();
         const aiMsgId = crypto.randomUUID();
 
@@ -241,7 +241,6 @@ export const WorkAssistantPanel: React.FC<WorkAssistantPanelProps> = ({ currentU
                 text: '', 
                 thought: '', 
                 timestamp: Date.now(),
-                // Pass user images to AI message so it can render the placeholders using the source images
                 images: base64Images 
             };
 
@@ -282,6 +281,7 @@ export const WorkAssistantPanel: React.FC<WorkAssistantPanelProps> = ({ currentU
                     images: base64Images, 
                     history: [], // Work assistant keeps context short
                     enableThinking,
+                    enableSearch,
                     overrideSystemPrompt: dynamicSystemPrompt, 
                     disableAudio: true 
                 })
@@ -312,7 +312,6 @@ export const WorkAssistantPanel: React.FC<WorkAssistantPanelProps> = ({ currentU
                         try {
                             const data = JSON.parse(jsonStr);
                             
-                            // Using functional update with ID check to ensure we only update the AI message
                             if (data.type === 'thinking') {
                                 aiThoughtAccumulated += data.content;
                                 setMessages(prev => prev.map(m => m.id === aiMsgId ? { ...m, thought: aiThoughtAccumulated } : m));
@@ -436,16 +435,22 @@ export const WorkAssistantPanel: React.FC<WorkAssistantPanelProps> = ({ currentU
                     </div>
                 </div>
 
-                <div className="flex items-center gap-4">
-                    <div className="flex items-center gap-2" title="开启后AI会进行更深入的逻辑推演">
-                        <span className="text-sm text-gray-600 font-medium flex items-center gap-1">
-                            <Brain size={16} className={enableThinking ? "text-purple-600" : "text-gray-400"}/> 深度思考
-                        </span>
+                <div className="flex items-center gap-3">
+                    <div className="flex items-center gap-1" title="开启后AI会进行更深入的逻辑推演">
                         <label className="relative inline-flex items-center cursor-pointer">
                             <input type="checkbox" checked={enableThinking} onChange={e => setEnableThinking(e.target.checked)} className="sr-only peer"/>
                             <div className="w-9 h-5 bg-gray-200 peer-focus:outline-none rounded-full peer peer-checked:after:translate-x-full peer-checked:after:border-white after:content-[''] after:absolute after:top-[2px] after:left-[2px] after:bg-white after:border-gray-300 after:border after:rounded-full after:h-4 after:w-4 after:transition-all peer-checked:bg-purple-600"></div>
                         </label>
+                        <span className="text-xs text-gray-600 font-bold">深度思考</span>
                     </div>
+                    <div className="flex items-center gap-1" title="开启联网搜索">
+                        <label className="relative inline-flex items-center cursor-pointer">
+                            <input type="checkbox" checked={enableSearch} onChange={e => setEnableSearch(e.target.checked)} className="sr-only peer"/>
+                            <div className="w-9 h-5 bg-gray-200 peer-focus:outline-none rounded-full peer peer-checked:after:translate-x-full peer-checked:after:border-white after:content-[''] after:absolute after:top-[2px] after:left-[2px] after:bg-white after:border-gray-300 after:border after:rounded-full after:h-4 after:w-4 after:transition-all peer-checked:bg-blue-600"></div>
+                        </label>
+                        <span className="text-xs text-gray-600 font-bold">联网搜索</span>
+                    </div>
+                    <div className="w-px h-6 bg-gray-200 mx-1"></div>
                     <button onClick={() => setMessages([])} className="text-gray-400 hover:text-red-500 p-2 rounded-full hover:bg-red-50 transition-colors" title="清空对话">
                         <Trash2 size={18}/>
                     </button>
@@ -466,12 +471,9 @@ export const WorkAssistantPanel: React.FC<WorkAssistantPanelProps> = ({ currentU
                 )}
                 
                 {messages.map((msg, index) => {
-                    // Logic to retrieve images from the AI message itself (passed during creation) 
-                    // OR from the preceding user message if it's a model response
                     let sourceImages: string[] = msg.images || [];
                     
                     if (msg.role === 'model' && (!sourceImages || sourceImages.length === 0)) {
-                        // Fallback: look at previous user message
                         const prevMsg = messages[index - 1];
                         if (prevMsg && prevMsg.role === 'user' && prevMsg.images) {
                             sourceImages = prevMsg.images;
@@ -503,7 +505,7 @@ export const WorkAssistantPanel: React.FC<WorkAssistantPanelProps> = ({ currentU
                                 )}
 
                                 <div className={`p-5 rounded-2xl shadow-sm text-sm overflow-hidden relative group w-full ${msg.role === 'user' ? 'bg-blue-600 text-white rounded-tr-none' : 'bg-white text-gray-800 border border-gray-100 rounded-tl-none'}`}>
-                                    {/* User Image Preview Grid (Only for user messages) */}
+                                    {/* User Image Preview Grid */}
                                     {msg.role === 'user' && sourceImages.length > 0 && (
                                         <div className="grid grid-cols-4 gap-2 mb-3">
                                             {sourceImages.map((img, i) => (
@@ -577,8 +579,6 @@ export const WorkAssistantPanel: React.FC<WorkAssistantPanelProps> = ({ currentU
                                     <button onClick={clearDoc} className="absolute top-0.5 right-0.5 text-indigo-300 hover:text-red-500 p-0.5"><X size={12}/></button>
                                 </div>
                             )}
-
-                            {/* Add More Button (if needed logic) */}
                         </div>
                     )}
 

models.js

@@ -26,18 +26,21 @@ const UserSchema = new mongoose.Schema({
     seatNo: String, 
     idCard: String,
     aiAccess: { type: Boolean, default: false },
-    doubaoContextId: { type: String }, // NEW: Stores the Doubao Context ID for this user
-    menuOrder: [String], // NEW
+    // UPDATED: Stores Doubao Context State
+    doubaoState: {
+        responseId: String,       // previous_response_id
+        thinkingState: Boolean    // Last thinking state (enabled/disabled)
+    },
+    menuOrder: [String], 
     classApplication: { 
         type: { type: String }, 
         targetClass: String, 
         status: String 
     },
-    // NEW: For Principal creating a new school
     pendingSchoolData: {
         name: String,
         code: String,
-        type: { type: String } // FIXED: Wrapped in object to avoid Mongoose interpreting parent as String type
+        type: { type: String } 
     }
 });
 const User = mongoose.model('User', UserSchema);