mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-10 13:29:35 +00:00
Bugfix(RAG):handle exceptions in aload_document_with_limit results (#2712)
This commit is contained in:
@@ -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.")
|
||||
|
Reference in New Issue
Block a user