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)