From 39b96849a9904e05bea644e62bada7185372c1b8 Mon Sep 17 00:00:00 2001
From: Kevin Hu <kevinhu.sh@gmail.com>
Date: Thu, 20 Feb 2025 12:54:29 +0800
Subject: [PATCH] Fix window size issue of ES. (#5175)

### What problem does this PR solve?

#5152

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
---
 graphrag/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/graphrag/utils.py b/graphrag/utils.py
index 06e73bcb0..283156207 100644
--- a/graphrag/utils.py
+++ b/graphrag/utils.py
@@ -524,7 +524,7 @@ def rebuild_graph(tenant_id, kb_id):
     src_ids = []
     flds = ["entity_kwd", "entity_type_kwd", "from_entity_kwd", "to_entity_kwd", "weight_int", "knowledge_graph_kwd", "source_id"]
     bs = 256
-    for i in range(0, 10000, bs):
+    for i in range(0, 39*bs, bs):
         es_res = settings.docStoreConn.search(flds, [],
                                  {"kb_id": kb_id, "knowledge_graph_kwd": ["entity", "relation"]},
                                  [],