feat: kd service after_delete, after_batch_delete

This commit is contained in:
martsforever
2025-09-07 20:25:59 +08:00
parent fd8f1ddb6f
commit 21b7757c14
+5 -5
View File
@@ -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)