From 088c57e2299912855a945b697c6ddfb9082c3b69 Mon Sep 17 00:00:00 2001
From: Silence_Lurker <slvccans@gmail.com>
Date: Tue, 20 Jan 2026 11:56:58 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=81=87=E6=B5=81=E5=8A=9F?=
 =?UTF-8?q?=E8=83=BD=E7=9A=84=E8=AE=BE=E7=BD=AE=E9=80=89=E9=A1=B9=E5=92=8C?=
 =?UTF-8?q?=E4=BA=8B=E4=BB=B6=E7=BB=91=E5=AE=9A?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../Amily2-AdditionalFeatures.html            |  8 ++
 core/api/Ngms_api.js                          | 74 ++++++++++++++++++-
 ui/historiography-bindings.js                 |  8 ++
 3 files changed, 86 insertions(+), 4 deletions(-)

diff --git a/assets/amily-additional-features/Amily2-AdditionalFeatures.html b/assets/amily-additional-features/Amily2-AdditionalFeatures.html
index dcd5b1a..b0a3718 100644
--- a/assets/amily-additional-features/Amily2-AdditionalFeatures.html
+++ b/assets/amily-additional-features/Amily2-AdditionalFeatures.html
@@ -139,6 +139,14 @@
         </label>
     </div>
 
+    <div class="amily2_settings_block">
+        <label for="amily2_ngms_fakestream_enabled">启用 Fake Stream (防超时)</label>
+        <label class="toggle-switch">
+            <input id="amily2_ngms_fakestream_enabled" type="checkbox" />
+            <span class="slider"></span>
+        </label>
+    </div>
+
     <div class="amily2_settings_block" id="amily2_ngms_content" style="display: none;">
         <div class="control-group">
             <label for="amily2_ngms_api_mode">API调用模式：</label>
diff --git a/core/api/Ngms_api.js b/core/api/Ngms_api.js
index bd97c58..b6d3d99 100644
--- a/core/api/Ngms_api.js
+++ b/core/api/Ngms_api.js
@@ -50,7 +50,8 @@ export function getNgmsApiSettings() {
         model: extension_settings[extensionName]?.ngmsModel || '',
         maxTokens: extension_settings[extensionName]?.ngmsMaxTokens || 4000,
         temperature: extension_settings[extensionName]?.ngmsTemperature || 0.7,
-        tavernProfile: extension_settings[extensionName]?.ngmsTavernProfile || ''
+        tavernProfile: extension_settings[extensionName]?.ngmsTavernProfile || '',
+        useFakeStream: extension_settings[extensionName]?.ngmsFakeStreamEnabled || false
     };
 }
 
@@ -73,12 +74,22 @@ export async function callNgmsAI(messages, options = {}) {
         ...options
     };
 
+    // 确保 stream 标志位存在
+    finalOptions.stream = finalOptions.useFakeStream ?? apiSettings.useFakeStream ?? false;
+
     if (finalOptions.apiMode !== 'sillytavern_preset') {
         if (!finalOptions.apiUrl || !finalOptions.model || !finalOptions.apiKey) {
             console.warn("[Amily2-Ngms外交部] API配置不完整，无法调用AI");
             toastr.error("API配置不完整，请检查URL、Key和模型配置。", "Ngms-外交部");
             return null;
         }
+    } else {
+        // [限制] 预设模式暂不支持流式
+        if (finalOptions.stream) {
+            console.warn("[Amily2-Ngms] 预设模式目前尚不支持流式处理方案，已自动切换为标准模式。");
+            toastr.warning("SillyTavern预设模式目前暂不支持流式处理（假流式），已为您切换为标准请求模式。该功能将在后续版本中支持。", "Ngms-外交部");
+            finalOptions.stream = false;
+        }
     }
 
     console.groupCollapsed(`[Amily2号-Ngms统一API调用] ${new Date().toLocaleTimeString()}`);
@@ -87,6 +98,7 @@ export async function callNgmsAI(messages, options = {}) {
         model: finalOptions.model, 
         maxTokens: finalOptions.maxTokens, 
         temperature: finalOptions.temperature,
+        stream: finalOptions.stream,
         messagesCount: messages.length
     });
     console.log("【消息内容】:", messages);
@@ -139,6 +151,54 @@ export async function callNgmsAI(messages, options = {}) {
     }
 }
 
+async function fetchFakeStream(url, opts) {
+    const res = await fetch(url, opts);
+    if (!res.ok) {
+        const errorText = await res.text();
+        throw new Error(`Stream HTTP ${res.status}: ${errorText}`);
+    }
+    
+    const reader = res.body.getReader();
+    const decoder = new TextDecoder();
+    let fullContent = "";
+    let buffer = "";
+    
+    try {
+        while (true) {
+            const { done, value } = await reader.read();
+            if (done) break;
+            
+            buffer += decoder.decode(value, { stream: true });
+            const lines = buffer.split('\n');
+            buffer = lines.pop();
+
+            for (const line of lines) {
+                const trimmed = line.trim();
+                if (!trimmed || trimmed === 'data: [DONE]') continue;
+                if (trimmed.startsWith('data: ')) {
+                    try {
+                        const json = JSON.parse(trimmed.substring(6));
+                        const delta = json.choices?.[0]?.delta?.content;
+                        if (delta) fullContent += delta;
+                    } catch (e) {
+                        console.warn('[NgmsApi] SSE Parse Error:', e);
+                    }
+                }
+            }
+        }
+    } finally {
+        reader.releaseLock();
+    }
+    
+    if (!fullContent && buffer) {
+        try { 
+            const data = JSON.parse(buffer);
+            return data.choices?.[0]?.message?.content || data.content || buffer; 
+        } catch { return buffer; }
+    }
+    return fullContent;
+}
+
 async function callNgmsOpenAITest(messages, options) {
     const isGoogleApi = options.apiUrl.includes('googleapis.com');
 
@@ -148,7 +208,7 @@ async function callNgmsOpenAITest(messages, options) {
         model: options.model,
         reverse_proxy: options.apiUrl,
         proxy_password: options.apiKey,
-        stream: false,
+        stream: !!options.stream,
         max_tokens: options.maxTokens || 30000,
         temperature: options.temperature || 1,
         top_p: options.top_p || 1,
@@ -167,11 +227,17 @@ async function callNgmsOpenAITest(messages, options) {
         });
     }
 
-    const response = await fetch('/api/backends/chat-completions/generate', {
+    const fetchOpts = {
         method: 'POST',
         headers: { ...getRequestHeaders(), 'Content-Type': 'application/json' },
         body: JSON.stringify(body)
-    });
+    };
+
+    if (options.stream) {
+        return await fetchFakeStream('/api/backends/chat-completions/generate', fetchOpts);
+    }
+
+    const response = await fetch('/api/backends/chat-completions/generate', fetchOpts);
 
     if (!response.ok) {
         const errorText = await response.text();
diff --git a/ui/historiography-bindings.js b/ui/historiography-bindings.js
index ae4b2b7..a044c85 100644
--- a/ui/historiography-bindings.js
+++ b/ui/historiography-bindings.js
@@ -408,6 +408,7 @@ function bindNgmsApiEvents() {
 
     // Ngms API 开关控制
     const ngmsToggle = document.getElementById('amily2_ngms_enabled');
+    const ngmsFakeStreamToggle = document.getElementById('amily2_ngms_fakestream_enabled');
     const ngmsContent = document.getElementById('amily2_ngms_content');
     
     if (ngmsToggle && ngmsContent) {
@@ -421,6 +422,13 @@ function bindNgmsApiEvents() {
         });
     }
 
+    if (ngmsFakeStreamToggle) {
+        ngmsFakeStreamToggle.checked = extension_settings[extensionName].ngmsFakeStreamEnabled ?? false;
+        ngmsFakeStreamToggle.addEventListener('change', function() {
+            updateAndSaveSetting('ngmsFakeStreamEnabled', this.checked);
+        });
+    }
+
     // API模式切换
     const apiModeSelect = document.getElementById('amily2_ngms_api_mode');
     const compatibleConfig = document.getElementById('amily2_ngms_compatible_config');