adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #603 from SuanmoSuanyangTechnology/fix/extraction-complete

Browse Source 
[add] Add the completion status of knowledge extraction to the stream…
This commit is contained in:
Ke Sun
2026-03-18 17:35:59 +08:00
committed by GitHub
parent e8c3744f5e 9d9250954b
commit a0f2f738df
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
api/app/core/memory/storage_services/extraction_engine/extraction_orchestrator.py
View File

@@ -384,6 +384,14 @@ class ExtractionOrchestrator:
logger.info(f"陈述句提取完成，共提取 {len(all_statements)} 条陈述句") logger.info(f"陈述句提取完成，共提取 {len(all_statements)} 条陈述句")
# 试运行模式下，所有分块提取完成后发送完成事件
if self.progress_callback and self.is_pilot_run:
await self.progress_callback(
"knowledge_extraction_complete",
f"陈述句提取完成，共提取 {len(all_statements)} 条",
{"total_statements": len(all_statements), "total_chunks": total_chunks}
)
return dialog_data_list return dialog_data_list
async def _extract_triplets( async def _extract_triplets(