Skip to content

Commit 4df1318

Browse files
committed
Update frequency in incremental
1 parent 36edd03 commit 4df1318

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

graphrag/index/update/entities.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,9 @@ def _group_and_resolve_entities(
7272
.reset_index()
7373
)
7474

75+
# recompute frequency to include new text units
76+
aggregated["frequency"] = aggregated["text_unit_ids"].apply(len)
77+
7578
# Force the result into a DataFrame
7679
resolved: pd.DataFrame = pd.DataFrame(aggregated)
7780

@@ -85,6 +88,7 @@ def _group_and_resolve_entities(
8588
"type",
8689
"description",
8790
"text_unit_ids",
91+
"frequency",
8892
"degree",
8993
"x",
9094
"y",

0 commit comments

Comments
 (0)