AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-04-21 05:29:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

More robust community report. (#5328)

Browse Source 
### What problem does this PR solve?

#5289
### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu 2025-02-25 12:58:10 +08:00 committed by GitHub
parent 58a8f1f1b0
commit 1a41b92f77
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
graphrag/general/community_reports_extractor.py
View File

@ -68,7 +68,7 @@ class CommunityReportsExtractor(Extractor):
weight = ents["weight"]
ents = ents["nodes"]
ent_df = pd.DataFrame(self._get_entity_(ents)).dropna()#[{"entity": n, **graph.nodes[n]} for n in ents])
if ent_df.empty:
if ent_df.empty or "entity_name" not in ent_df.columns:
continue
ent_df["entity"] = ent_df["entity_name"]
del ent_df["entity_name"]