Update frequency in incremental

microsoft · Feb 13, 2025 · 4df1318 · 4df1318
1 parent 36edd03
commit 4df1318
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/graphrag/index/update/entities.py b/graphrag/index/update/entities.py
@@ -72,6 +72,9 @@ def _group_and_resolve_entities(
         .reset_index()
     )
 
+    # recompute frequency to include new text units
+    aggregated["frequency"] = aggregated["text_unit_ids"].apply(len)
+
     # Force the result into a DataFrame
     resolved: pd.DataFrame = pd.DataFrame(aggregated)
 
@@ -85,6 +88,7 @@ def _group_and_resolve_entities(
             "type",
             "description",
             "text_unit_ids",
+            "frequency",
             "degree",
             "x",
             "y",