fix(graphrag): variable refernence error (#2136)

### What problem does this PR solve? fix: Use wrong variable in graph rag step. ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) Co-authored-by: 陈晓强 <[email protected]>
infiniflow · Aug 28, 2024 · 3abc9be · 3abc9be
1 parent e627ee9
commit 3abc9be
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/graphrag/search.py b/graphrag/search.py
@@ -93,9 +93,9 @@ def merge_into_first(sres, title=""):
         s = s.query(bqry)[0: 6]
         s = s.to_dict()
         txt_res = self.es.search(deepcopy(s), idxnm=idxnm, timeout="600s", src=src)
-        txt_ids = self.es.getDocIds(comm_res)
+        txt_ids = self.es.getDocIds(txt_res)
         if merge_into_first(txt_res, "-Original Content-"):
-            txt_ids = comm_ids[0:1]
+            txt_ids = txt_ids[0:1]
 
         return self.SearchResult(
             total=len(ent_ids) + len(comm_ids) + len(txt_ids),