Skip to content

Commit 5bec137

Browse files
fix(indexer): correct batch progress reporting
Change progress yield to return batch size instead of total processed, providing more accurate progress tracking during batch processing. Co-authored-by: Bob <[email protected]>
1 parent 18ba7cb commit 5bec137

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

gptme_rag/indexing/indexer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -188,7 +188,7 @@ def add_documents_progress(
188188
batch = documents[processed : processed + batch_size]
189189
self._add_documents(batch)
190190
processed += len(batch)
191-
yield processed
191+
yield len(batch)
192192

193193
def _add_documents(self, documents: list[Document]) -> None:
194194
try:

0 commit comments

Comments
 (0)