'commit'

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)