feat: kd service after_delete, after_batch_delete
This commit is contained in:
@@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user