feat: add route /knowledge/delete

This commit is contained in:
martsforever
2025-09-07 20:16:55 +08:00
parent e24e3d04ba
commit 3f2b10d767
+9 -6
View File
@@ -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]}