'commit'

dsLightRag/Start.py

@@ -1,4 +1,5 @@
 import json
+import logging
 import os.path
 import shutil
 import subprocess
@@ -21,7 +22,7 @@ from Util.PostgreSQLUtil import init_postgres_pool
 
 # 想更详细地控制日志输出
 logger = logging.getLogger('lightrag')
-logger.setLevel(logging.INFO)
+logger.setLevel(logging.DEBUG)
 handler = logging.StreamHandler()
 handler.setFormatter(logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s'))
 logger.addHandler(handler)
@@ -55,11 +56,12 @@ async def rag(request: fastapi.Request):
     output_model = data.get("output_model", "txt")
 
     if output_model == "txt":
-        user_prompt = "1、资料中提供化学反应方程式的，一定要严格按提供的Latex公式输出，绝对不允许对Latex公式进行修改 ！"
-        user_prompt = user_prompt + "\n 2、如果资料中提供了图片的，一定要严格按照原文提供图片输出，绝对不能省略或不输出！"
-        user_prompt = user_prompt + "\n 3、知识库中存在的问题，严格按知识库中的内容回答，不允许扩展！"
-        user_prompt = user_prompt + "\n 4、如果问题与提供的知识库内容不符，则明确告诉未在知识库范围内提到！"
-        user_prompt = user_prompt + "\n 5、发现输出内容中包含Latex公式的，一定要检查是不是包含了$$或$的包含符号,不能让Latex无包含符号出现！"
+        user_prompt = "1、如果资料中提供了图片的，一定要严格按照原文提供图片输出，绝对不能省略或不输出！"
+        user_prompt = user_prompt + "\n 2、不要提供引用信息！"
+        user_prompt = user_prompt + "\n 3、提供给你的材料中，与问题完全相关的需要完整保留！"
+        user_prompt = user_prompt + "\n 4、提供给你的材料中，与问题不完全相关的一定不要输出！"
+        user_prompt = user_prompt + "\n 5、资料中提供化学反应方程式的，一定要严格按提供的Latex公式输出，绝对不允许对Latex公式进行修改 ！"
+        user_prompt = user_prompt + "\n 6、发现输出内容中包含Latex公式的，一定要检查是不是包含了$$或$的包含符号,不能让Latex无包含符号出现！"
     elif output_model == 'html':
         user_prompt = """
         我需要一个专业的交互式数据可视化，数据资料我将提供，你也可以根据自己了解的信息进行补充，