Bugfix(RAG):handle exceptions in aload_document_with_limit results (#2712)

This commit is contained in:
geebytes
2025-05-22 17:00:03 +08:00
committed by GitHub
parent f79f81ccc3
commit cbc28ea335
3 changed files with 16 additions and 4 deletions

View File

@@ -187,7 +187,11 @@ class IndexStoreBase(ABC):
ids = []
loaded_cnt = 0
for success_ids in results:
for idx, success_ids in enumerate(results):
if isinstance(success_ids, Exception):
raise RuntimeError(
f"Failed to load chunk group {idx + 1}: {str(success_ids)}"
) from success_ids
ids.extend(success_ids)
loaded_cnt += len(success_ids)
logger.info(f"Loaded {loaded_cnt} chunks, total {len(chunks)} chunks.")