From 21b7757c147dd1b877a8f149229a7b8d377cf7e4 Mon Sep 17 00:00:00 2001 From: martsforever Date: Sun, 7 Sep 2025 20:25:59 +0800 Subject: [PATCH] feat: kd service after_delete, after_batch_delete --- app/model/KnowledgeDoc.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/app/model/KnowledgeDoc.py b/app/model/KnowledgeDoc.py index dd12853..d0b1211 100644 --- a/app/model/KnowledgeDoc.py +++ b/app/model/KnowledgeDoc.py @@ -25,14 +25,14 @@ class KnowledgeDocModel(BasicModel, table=True): error: Optional[str] = Field(default=None, description="文档处理失败时的错误信息") -async def handle_after_delete(doc_cls): - await milvus_service.async_delete(doc_cls.id) +async def handle_after_delete(delete_cls, row_dict, session): + await milvus_service.async_delete(delete_cls.id) -async def handle_after_batch_delete(doc_cls_list): +async def handle_after_batch_delete(delete_cls_list, row_dict_list, session): task_list = [ - asyncio.create_task(handle_after_delete(doc_cls)) - for doc_cls in doc_cls_list + asyncio.create_task(handle_after_delete(doc_cls, {}, session)) + for doc_cls in delete_cls_list ] await asyncio.gather(*task_list)