diff --git a/graphrag/search.py b/graphrag/search.py index 3fa962fcb..69f150f4d 100644 --- a/graphrag/search.py +++ b/graphrag/search.py @@ -67,7 +67,7 @@ class KGSearch(Dealer): 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: