deduplicate sources in local search context (#1468)

Co-authored-by: Alonso Guevara <alonsog@microsoft.com>
microsoft · Dec 6, 2024 · b1f2ca7 · b1f2ca7
1 parent 5ff2d3c
commit b1f2ca7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/graphrag/query/structured_search/local_search/mixed_context.py b/graphrag/query/structured_search/local_search/mixed_context.py
@@ -334,6 +334,7 @@ def _build_text_unit_context(
                     num_relationships = count_relationships(
                         entity_relationships, selected_unit
                     )
+                    text_unit_ids_set.add(text_id)
                     unit_info_list.append((selected_unit, index, num_relationships))
 
         # sort by entity_order and the number of relationships desc