From 3f2b10d7678fe6a0f47c014a70e6b899fd788bf9 Mon Sep 17 00:00:00 2001 From: martsforever Date: Sun, 7 Sep 2025 20:16:55 +0800 Subject: [PATCH] feat: add route /knowledge/delete --- app/controller/add_knowledge_route.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/app/controller/add_knowledge_route.py b/app/controller/add_knowledge_route.py index 040f755..6b57b9c 100644 --- a/app/controller/add_knowledge_route.py +++ b/app/controller/add_knowledge_route.py @@ -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]}