wuzhiqiang/ai-admin-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: kd service after_delete, after_batch_delete

Browse Source
This commit is contained in:
martsforever
2025-09-07 20:25:59 +08:00
parent fd8f1ddb6f
commit 21b7757c14
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/model/KnowledgeDoc.py
+5 -5
View File
@@ -25,14 +25,14 @@ class KnowledgeDocModel(BasicModel, table=True):
error: Optional[str] = Field(default=None, description="文档处理失败时的错误信息") error: Optional[str] = Field(default=None, description="文档处理失败时的错误信息")
async def handle_after_delete(doc_cls): async def handle_after_delete(delete_cls, row_dict, session):
await milvus_service.async_delete(doc_cls.id) 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 = [ task_list = [
asyncio.create_task(handle_after_delete(doc_cls)) asyncio.create_task(handle_after_delete(doc_cls, {}, session))
for doc_cls in doc_cls_list for doc_cls in delete_cls_list
] ]
await asyncio.gather(*task_list) await asyncio.gather(*task_list)