kanshan/graphrag
1
0
Fork 0
mirror of https://github.com/microsoft/graphrag.git synced 2026-02-08 12:12:59 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

optionally parse embeddings from df when creating CommunityReport

Browse Source
This commit is contained in:
Julian Whiting 2024-10-08 11:54:34 -04:00
parent 9c4619af43
commit f0d779f81a
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
graphrag/query/indexer_adapters.py
View File

@ -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,
)