Update ai-routes.js

ai-routes.js

@@ -214,10 +214,13 @@ async function streamDoubao(baseParams, res, username, mode = 'chat', config, en
     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;
     }
@@ -225,15 +228,19 @@ async function streamDoubao(baseParams, res, username, mode = 'chat', config, en
     // 3. Build Input
     let inputPayload = [];
     if (idToSend) {
-        // INCREMENTAL MODE
+        // INCREMENTAL MODE: We assume the server has context.
+        // We need to send System Prompt + NEWEST Message only.
+        
         const fullInputs = convertGeminiToDoubaoInput(baseParams);
+        
         const systemMsg = fullInputs.find(m => m.role === 'system');
+        // Get the last message (User's new question)
         const lastMsg = fullInputs[fullInputs.length - 1]; 
         
         if (systemMsg) inputPayload.push(systemMsg);
         if (lastMsg) inputPayload.push(lastMsg);
     } else {
-        // FULL MODE
+        // FULL MODE: ID Invalid/Missing/SearchOn -> Send Full History
         inputPayload = convertGeminiToDoubaoInput(baseParams);
     }
     // --- CONTEXT LOGIC END ---
@@ -242,10 +249,15 @@ async function streamDoubao(baseParams, res, username, mode = 'chat', config, en
     const requestBody = {
         model: endpointId,
         stream: true,
+        // Doubao Bot API uses 'input' parameter with specific content types
         input: inputPayload, 
+        
+        // CRITICAL: Explicitly send disabled/enabled as requested by user
         thinking: { type: enableThinking ? "enabled" : "disabled" },
         caching: { type: requestCachingType },
+        // Add ID if valid
         ...(idToSend && { previous_response_id: idToSend }),
+        // Add Search Tools if enabled
         // CORRECTED TOOL STRUCTURE: just type
         ...(enableSearch && { tools: [{ type: "web_search" }] })
     };
@@ -302,7 +314,7 @@ async function streamDoubao(baseParams, res, username, mode = 'chat', config, en
                             const jsonStr = trimmed.substring(6);
                             const json = JSON.parse(jsonStr);
                             
-                            // --- V3 EVENT PARSING ---
+                            // --- V3 EVENT PARSING (Flat Structure) ---
                             const eventType = json.type;
 
                             // 1. Response ID
@@ -319,13 +331,14 @@ async function streamDoubao(baseParams, res, username, mode = 'chat', config, en
                             }
 
                             // 3. Reasoning / Thinking (Reasoning Delta)
-                            // Event: response.reasoning_text.delta OR response.reasoning_summary_text.delta
+                            // Support both regular reasoning and summary reasoning
                             if ((eventType === 'response.reasoning_text.delta' || eventType === 'response.reasoning_summary_text.delta') && json.delta) {
                                 res.write(`data: ${JSON.stringify({ type: 'thinking', content: json.delta })}\n\n`);
                                 if (res.flush) res.flush();
                             }
 
                             // 4. Search Status
+                            // Explicitly catch the searching event
                             if (eventType === 'response.web_search_call.searching') {
                                 res.write(`data: ${JSON.stringify({ type: 'search', status: 'searching' })}\n\n`);
                                 if (res.flush) res.flush();
@@ -339,7 +352,7 @@ async function streamDoubao(baseParams, res, username, mode = 'chat', config, en
                                 }
                             }
                             
-                            // Fallback for Legacy/Chat Format (Safety net)
+                            // Fallback for Legacy/Chat Format (Safety net, in case they revert or mix formats)
                             if (!eventType && json.response?.output_text?.delta) {
                                 const content = json.response.output_text.delta;
                                 fullText += content;
@@ -366,6 +379,7 @@ async function streamDoubao(baseParams, res, username, mode = 'chat', config, en
             }
 
             // 4. Update User State (Context Persistence)
+            // Rules: Only save ID if Search was OFF.
             if (!enableSearch && newResponseID) {
                 await User.findOneAndUpdate({ username }, {
                     doubaoState: {
@@ -375,6 +389,7 @@ async function streamDoubao(baseParams, res, username, mode = 'chat', config, en
                 });
                 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 }
                 });
@@ -386,15 +401,22 @@ async function streamDoubao(baseParams, res, username, mode = 'chat', config, en
         } catch (e) {
             console.error(`[AI] ❌ Doubao Request Failed!`);
             
+            // ERROR RESPONSE BODY CAPTURE (Stream Handling)
             if (e.response) {
                 console.error(`[AI] ❌ Status: ${e.response.status}`);
+                console.error(`[AI] ❌ Headers:`, JSON.stringify(e.response.headers, null, 2));
+                
                 if (e.response.data) {
                     try {
+                        // Check if it's a stream we need to consume
                         if (typeof e.response.data.on === 'function' || (e.response.data && typeof e.response.data.read === 'function')) {
                              let errData = '';
-                             for await (const chunk of e.response.data) { errData += chunk.toString(); }
+                             for await (const chunk of e.response.data) {
+                                 errData += chunk.toString();
+                             }
                              console.error(`[AI] ❌ Response Body (Stream):`, errData);
                         } else {
+                             // It's likely a JSON object already if axios parsed it (but responseType='stream' usually prevents this)
                              console.error(`[AI] ❌ Response Body (Object):`, JSON.stringify(e.response.data, null, 2));
                         }
                     } catch (readErr) {
@@ -407,10 +429,14 @@ async function streamDoubao(baseParams, res, username, mode = 'chat', config, en
                 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;