From 383c640d98991f337a66de3b832f08db3c1f3d5c Mon Sep 17 00:00:00 2001 From: HuangHai <10402852@qq.com> Date: Tue, 15 Jul 2025 19:49:02 +0800 Subject: [PATCH] 'commit' --- dsLightRag/Start.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/dsLightRag/Start.py b/dsLightRag/Start.py index 3fd9b84c..277ac0d1 100644 --- a/dsLightRag/Start.py +++ b/dsLightRag/Start.py @@ -51,8 +51,7 @@ async def rag(request: fastapi.Request): if output_model == "txt": user_prompt = "\n 1、不要输出参考资料 或者 References !" - # user_prompt = "\n 1、不要输出参考资料 或者 References !" - user_prompt = "\n 2、资料中提供化学反应方程式的,一定要严格按提供的Latex公式输出,绝对不允许对Latex公式进行修改 !" + user_prompt = user_prompt+ "\n 2、资料中提供化学反应方程式的,一定要严格按提供的Latex公式输出,绝对不允许对Latex公式进行修改 !" user_prompt = user_prompt + "\n 3、如果资料中提供了图片的,一定要严格按照原文提供图片输出,绝对不能省略或不输出!" # user_prompt = user_prompt + "\n 4、资料中提到的知识内容,需要判断是否与本次问题相关,不相关的绝对不要输出!" user_prompt = user_prompt + "\n 4、根据资料回答问题,可以适当拓展一下内容进行回答!" @@ -99,9 +98,8 @@ async def rag(request: fastapi.Request): # 使用PG库后,这个是没有用的,但目前的项目代码要求必传,就写一个吧。 WORKING_DIR = workspace - if os.path.exists(WORKING_DIR): - shutil.rmtree(WORKING_DIR) - os.makedirs(WORKING_DIR) + if not os.path.exists(WORKING_DIR): + os.makedirs(WORKING_DIR) async def generate_response_stream(query: str): try: