diff --git a/dsLightRag/Start.py b/dsLightRag/Start.py index e21c049c..8923fe23 100644 --- a/dsLightRag/Start.py +++ b/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: diff --git a/dsLightRag/T2_Query.py b/dsLightRag/T2_Query.py index 9dab03af..be89e0c8 100644 --- a/dsLightRag/T2_Query.py +++ b/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)