'commit'

dsRag/static/ai.html

@@ -4,6 +4,7 @@
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <title>东师理想教学大模型</title>
+    <link rel="icon" href="data:,">
     <style>
         body {
             font-family: 'Microsoft YaHei', sans-serif;
@@ -299,30 +300,65 @@
         answerArea.innerHTML = '';
         loader.style.display = 'block';
     
+        const data = {
+            query: question,
+            tags: selectedDocs
+        };
+    
         fetch('/api/rag', {
             method: 'POST',
             headers: {
                 'Content-Type': 'application/json',
+                'Accept': 'text/event-stream'
             },
-            body: JSON.stringify({
-                query: question,
-                tags: selectedDocs
-            })
+            body: JSON.stringify(data)
         })
-        .then(response => response.json())
-        .then(data => {
+        .then(response => {
+            const reader = response.body.getReader();
+            const decoder = new TextDecoder();
+            let buffer = '';
+            let accumulatedContent = '';
+    
+            function processChunk() {
+                return reader.read().then(({done, value}) => {
+                    if (done) {
+                        loader.style.display = 'none';
+                        return;
+                    }
+    
+                    buffer += decoder.decode(value, {stream: true});
+                    const lines = buffer.split('\n');
+                    buffer = lines.pop();
+    
+                    for (const line of lines) {
+                        if (line.includes('data:')) {
+                            const jsonStr = line.replace(/^data:\s*/, '').replace(/^data:\s*/, '').trim();
+                            if (jsonStr) {
+                                try {
+                                    const data = JSON.parse(jsonStr);
                                     if (data.reply) {
-                answerArea.innerHTML = marked.parse(data.reply);
-                MathJax.typeset();
-                localStorage.setItem('lastMarkdownContent', data.reply);
+                                        accumulatedContent += data.reply;
+                                        answerArea.innerHTML = marked.parse(accumulatedContent);
+                                        MathJax.typesetPromise();
                                     }
-            loader.style.display = 'none';
+                                } catch (e) {
+                                    console.log('忽略解析错误:', e);
+                                }
+                            }
+                        }
+                    }
+    
+                    return processChunk();
+                });
+            }
+    
+            return processChunk();
         })
         .catch(error => {
             console.error('Error:', error);
             loader.style.display = 'none';
         });
-    }ai.html
+    }
 
     function clearAll() {
         document.getElementById('questionInput').value = '';