From f0d779f81a217a271b4db819e3d08a8a064853d7 Mon Sep 17 00:00:00 2001 From: Julian Whiting Date: Tue, 8 Oct 2024 11:54:34 -0400 Subject: [PATCH] optionally parse embeddings from df when creating CommunityReport --- graphrag/query/indexer_adapters.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/graphrag/query/indexer_adapters.py b/graphrag/query/indexer_adapters.py index dba4712b..812a3c23 100644 --- a/graphrag/query/indexer_adapters.py +++ b/graphrag/query/indexer_adapters.py @@ -63,6 +63,7 @@ def read_indexer_reports( final_community_reports: pd.DataFrame, final_nodes: pd.DataFrame, community_level: int, + content_embedding_col: str | None = None, ) -> list[CommunityReport]: """Read in the Community Reports from the raw indexing outputs.""" report_df = final_community_reports @@ -83,7 +84,7 @@ def read_indexer_reports( id_col="community", short_id_col="community", summary_embedding_col=None, - content_embedding_col=None, + content_embedding_col=content_embedding_col, )