|
@@ -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"))
|