feat: add route /knowledge/delete
This commit is contained in:
@@ -6,6 +6,7 @@ from fastapi import UploadFile, File, Form
|
||||
|
||||
from app.utils.db_utils import AsyncSessionDep
|
||||
from app.utils.knowledge_utils import knowledge_service
|
||||
from app.utils.milvus_utils import milvus_service
|
||||
|
||||
|
||||
def add_knowledge_route(app):
|
||||
@@ -34,11 +35,13 @@ def add_knowledge_route(app):
|
||||
# "search_documents": search_result,
|
||||
# }
|
||||
#
|
||||
# @app.post("/knowledge/delete")
|
||||
# async def knowledge_embed_text(body: dict):
|
||||
# await milvus_service.async_delete(body.get('id'))
|
||||
# return {
|
||||
# "result": "删除成功",
|
||||
@app.post("/knowledge/delete")
|
||||
async def knowledge_embed_text(body: dict):
|
||||
await milvus_service.async_delete(body.get('id'))
|
||||
return {
|
||||
"result": "删除成功",
|
||||
}
|
||||
|
||||
@app.post("/knowledge/upload_files")
|
||||
async def knowledge_search(
|
||||
session: AsyncSessionDep,
|
||||
@@ -64,7 +67,7 @@ def add_knowledge_route(app):
|
||||
doc_cls_list = await knowledge_service.save_knowledge_doc_list(session=session, file_dict_list=file_dict_list, kb_code=kb_code)
|
||||
|
||||
# 异步处理嵌入文档,不再等待
|
||||
[asyncio.create_task(knowledge_service.process_file_dict(file_dict=file_dict, kb_code=kb_code)) for file_dict in file_dict_list]
|
||||
[asyncio.create_task(knowledge_service.process_doc_cls(doc_cls=doc_cls)) for doc_cls in doc_cls_list]
|
||||
|
||||
# 直接返回文档对象
|
||||
return {"message": f"正在处理 {len(doc_cls_list)} 个文档,请刷新列表查看文档状态。", "result": [item.model_dump() for item in doc_cls_list]}
|
||||
|
||||
Reference in New Issue
Block a user