'commit'

dsLightRag/Start.py

@@ -53,6 +53,7 @@ async def rag(request: fastapi.Request):
     user_prompt = user_prompt + "\n 3、如果资料中提供了图片的，一定要严格按照原文提供图片输出，不允许省略或不输出！"
     user_prompt = user_prompt + "\n 4、资料中提到的知识内容，需要判断是否与本次问题相关，不相关的绝对不要输出！"
     user_prompt = user_prompt + "\n 5、如果问题与提供的知识库内容不符，则明确告诉未在知识库范围内提到！"
+    user_prompt = user_prompt + "\n 6、发现输出内容中包含Latex公式的，一定要检查是不是包含了$$或$的包含符号,不能让Latex无包含符号出现！"
 
     async def generate_response_stream(query: str):
         try:
@@ -67,6 +68,7 @@ async def rag(request: fastapi.Request):
             resp = await rag.aquery(
                 query=query,
                 param=QueryParam(mode="hybrid", stream=True, user_prompt=user_prompt))
+            # hybrid naive
 
             async for chunk in resp:
                 if not chunk:

dsLightRag/T2_Query.py

@@ -26,6 +26,7 @@ async def main():
         resp = await rag.aquery(
             data[idx]["Q"],
             param=QueryParam(mode="hybrid", stream=True, user_prompt=user_prompt),
+            # hybrid naive
         )
         if inspect.isasyncgen(resp):
             await print_stream(resp)