'commit'

dsLightRag/Config/Config.py

@@ -10,9 +10,9 @@ ALY_SK = 'oizcTOZ8izbGUouboC00RcmGE8vBQ1'
 # 阿里云提供的大模型服务 【阿里云在处理文字材料时，容易引发绿网拦截，导致数据上报异常】
 LLM_API_KEY = "sk-f6da0c787eff4b0389e4ad03a35a911f"
 LLM_BASE_URL = "https://dashscope.aliyuncs.com/compatible-mode/v1"
-# LLM_MODEL_NAME = "qwen-plus" # 不要使用通义千问，会导致化学方程式不正确！
-# LLM_MODEL_NAME = "deepseek-v3"
-LLM_MODEL_NAME = "deepseek-r1"  # 使用更牛B的r1模型
+#LLM_MODEL_NAME = "qwen-plus" # 不要使用通义千问，会导致化学方程式不正确！
+LLM_MODEL_NAME = "deepseek-v3"
+#LLM_MODEL_NAME = "deepseek-r1"  # 使用更牛B的r1模型
 
 EMBED_MODEL_NAME = "BAAI/bge-m3"
 EMBED_API_KEY = "sk-pbqibyjwhrgmnlsmdygplahextfaclgnedetybccknxojlyl"

dsLightRag/Start.py

@@ -68,7 +68,11 @@ async def rag(request: fastapi.Request):
         user_prompt = user_prompt + "\n 6、发现输出内容中包含Latex公式的，一定要检查是不是包含了$$或$的包含符号,不能让Latex无包含符号出现！"
     elif output_model=='html':
         user_prompt="""
-        我需要一个专业的交互式数据可视化，数据资料我将提供，你也可以根据自己了解的信息进行补充，注意：（1）不要与用户进行二次交互，直接生成即可。（2）直接输出html代码，不要其它信息出现。（3）、正面负面信息都要。
+        我需要一个专业的交互式数据可视化，数据资料我将提供，你也可以根据自己了解的信息进行补充，
+        注意：
+        (1)直接输出html代码,以```html 开头， ``` 结尾。
+        (2)不要与用户进行二次交互，直接生成即可。
+        (3)正面负面信息都要。
         绘制可视化具体要求如下：
         1. **主题类型**：
            [明确主题，如历史人物关系/公司组织架构/生物基因网络等]

dsLightRag/static/ShiJi.html

@@ -149,8 +149,6 @@
 
         #answerArea {
             min-height: 240px;
-            max-height: 500px;  /* 设置最大高度 */
-            overflow-y: auto;   /* 添加垂直滚动条 */
             height: auto;
             overflow-y: auto;
         }
@@ -209,7 +207,8 @@
     </div>
     <button id="submitBtn" onclick="submitQuestion()"><span class="icon">💡</span>提问</button>
     <button id="clearBtn" onclick="clearAll()"><span class="icon">🗑️</span>清空</button>
-    <button id="relationBtn" onclick="generateRelation()" style="background-color:#9c27b0"><span class="icon">👥</span>生成刘帮集团的人物关系图</button>
+    <button id="relationBtn" onclick="generateRelation()" style="background-color:#9c27b0"><span class="icon">👥</span>生成刘帮集团的人物关系图
+    </button>
 </div>
 
 <script>
@@ -346,8 +345,8 @@
         document.getElementById('answerArea').innerHTML = '';
     }
 
-// 在script部分添加新函数
-function generateRelation() {
+    // 在script部分添加新函数
+    function generateRelation() {
         const answerArea = document.getElementById('answerArea');
         answerArea.innerHTML = '<div class="loading-animation"><div class="spinner"></div><div>正在生成人物关系图...</div></div>';
 
@@ -372,18 +371,7 @@ function generateRelation() {
 
                 function processChunk() {
                     return reader.read().then(({done, value}) => {
-                if (done) {
-                    // 在HTML内容完全接收后执行其中的脚本
-                    const scripts = answerArea.getElementsByTagName('script');
-                    for (let i = 0; i < scripts.length; i++) {
-                        try {
-                            eval(scripts[i].innerHTML);
-                        } catch (e) {
-                            console.error('脚本执行错误:', e);
-                        }
-                    }
-                    return;
-                }
+                        if (done) return;
 
                         buffer += decoder.decode(value, {stream: true});
                         const lines = buffer.split('\n');
@@ -399,7 +387,6 @@ function generateRelation() {
                                             accumulatedContent += data.reply;
                                             answerArea.innerHTML = marked.parse(accumulatedContent, markedOptions);
                                             MathJax.typesetPromise();
-                                    answerArea.scrollTop = answerArea.scrollHeight;
                                         }
                                     } catch (e) {
                                         console.log('忽略解析错误:', e);
@@ -407,15 +394,25 @@ function generateRelation() {
                                 }
                             }
                         }
-                return processChunk();
+
+                        return processChunk().then(() => {
+                            // 在流处理完成后保存完整的markdown内容
+                            localStorage.setItem('lastMarkdownContent', accumulatedContent);
+                        });
                     });
                 }
+
                 return processChunk();
             })
             .catch(error => {
-        answerArea.innerHTML = '<div style="color:red">生成关系图出错，请重试</div>';
+                // 移除加载动画
+                answerArea.innerHTML = '';
+
+                console.error('Error:', error);
+                // 出错时也移除加载动画
+                answerArea.innerHTML = '<div style="color:red">请求出错，请重试</div>';
             });
-}
+    }
 </script>
 <script src="/static/mathjax/es5/tex-mml-chtml.js" id="MathJax-script" async></script>
 <script src="/static/js/marked.min.js"></script>