diff --git a/dsLightRag/Start.py b/dsLightRag/Start.py index 48dd6608..dadd8cbd 100644 --- a/dsLightRag/Start.py +++ b/dsLightRag/Start.py @@ -1,14 +1,11 @@ +import asyncio import json -import logging import os.path -import shutil import subprocess import tempfile import urllib import uuid -import asyncio from io import BytesIO -from urllib import request import fastapi import uvicorn @@ -58,7 +55,6 @@ async def rag(request: fastapi.Request): logger.info("工作空间:" + workspace) # 查询的问题 query = data.get("query") - # 用户提示词 output_model = data.get("output_model", "txt") @@ -109,10 +105,7 @@ async def rag(request: fastapi.Request): """ # 使用PG库后,这个是没有用的,但目前的项目代码要求必传,就写一个吧。 - WORKING_DIR = 'WorkingPath/' + workspace - if not os.path.exists(WORKING_DIR): - os.makedirs(WORKING_DIR) - + WORKING_DIR = './output/' async def generate_response_stream(query: str): try: logger.info("workspace=" + workspace)