Fix: search issue for graphrag

KevinHuSh · Nov 21, 2024 · 7822d09 · 7822d09
1 parent e10b0e6
commit 7822d09
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/graphrag/search.py b/graphrag/search.py
@@ -67,7 +67,7 @@ def merge_into_first(sres, title="") -> dict[str, str]:
 
         ent_res = self.dataStore.search(src, list(), condition, [matchText, matchDense, fusionExpr], OrderByExpr(), 0, 32, idxnm, kb_ids)
         ent_res_fields = self.dataStore.getFields(ent_res, src)
-        entities = [d.get["name_kwd"] for d in ent_res_fields.values() if d.get("name_kwd")]
+        entities = [d["name_kwd"] for d in ent_res_fields.values() if d.get("name_kwd")]
         ent_ids = self.dataStore.getChunkIds(ent_res)
         ent_content = merge_into_first(ent_res_fields, "-Entities-")
         if ent_content: