Browse Source

测试修改

yulongyan_citu 10 months ago
parent
commit
a110405d9f
1 changed files with 3 additions and 3 deletions
  1. 3 3
      app.py

+ 3 - 3
app.py

@@ -86,6 +86,7 @@ async def ask(request: Request):
         res = failed({}, {"error": f"{e}"})
         return JSONResponse(res)
 
+# 纯rag问答
 @app.post('/ask/rag')
 async def ask_rag(request: Request):
     try:
@@ -154,10 +155,9 @@ async def upload_file(file: UploadFile = File(...)):
         data = []
         loader = TextLoader(f'{UPLOAD_FOLDER}/{file.filename}', encoding='utf8')
         data.extend(loader.load())
-        text_splitter = RecursiveCharacterTextSplitter(
-            chunk_size=500, chunk_overlap=150)
+        text_splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=150)
         split_docs = text_splitter.split_documents(data)
-        vector_db = get_vectordb(file_path="./knowledge_db", persist_path="./vector_db/chroma",embedding="m3e")
+        vector_db = get_vectordb(file_path=DEFAULT_DB_PATH, persist_path=DEFAULT_PERSIST_PATH,embedding="m3e")
         vector_db.add_documents(split_docs)
 
         return JSONResponse(success({"response": "File uploaded successfully"}, "success"))