diff --git a/packages/graphrag/graphrag/prompt_tune/template/community_report_summarization.py b/packages/graphrag/graphrag/prompt_tune/template/community_report_summarization.py index b0d037ab6b..32888fb137 100644 --- a/packages/graphrag/graphrag/prompt_tune/template/community_report_summarization.py +++ b/packages/graphrag/graphrag/prompt_tune/template/community_report_summarization.py @@ -56,13 +56,13 @@ Entities -id,entity,description +human_readable_id,title,description 5,VERDANT OASIS PLAZA,Verdant Oasis Plaza is the location of the Unity March 6,HARMONY ASSEMBLY,Harmony Assembly is an organization that is holding a march at Verdant Oasis Plaza Relationships -id,source,target,description +human_readable_id,source,target,description 37,VERDANT OASIS PLAZA,UNITY MARCH,Verdant Oasis Plaza is the location of the Unity March 38,VERDANT OASIS PLAZA,HARMONY ASSEMBLY,Harmony Assembly is holding a march at Verdant Oasis Plaza 39,VERDANT OASIS PLAZA,UNITY MARCH,The Unity March is taking place at Verdant Oasis Plaza diff --git a/packages/graphrag/graphrag/prompts/index/community_report.py b/packages/graphrag/graphrag/prompts/index/community_report.py index c3a7702ba0..b1f0f8abe2 100644 --- a/packages/graphrag/graphrag/prompts/index/community_report.py +++ b/packages/graphrag/graphrag/prompts/index/community_report.py @@ -59,13 +59,13 @@ Entities -id,entity,description +human_readable_id,title,description 5,VERDANT OASIS PLAZA,Verdant Oasis Plaza is the location of the Unity March 6,HARMONY ASSEMBLY,Harmony Assembly is an organization that is holding a march at Verdant Oasis Plaza Relationships -id,source,target,description +human_readable_id,source,target,description 37,VERDANT OASIS PLAZA,UNITY MARCH,Verdant Oasis Plaza is the location of the Unity March 38,VERDANT OASIS PLAZA,HARMONY ASSEMBLY,Harmony Assembly is holding a march at Verdant Oasis Plaza 39,VERDANT OASIS PLAZA,UNITY MARCH,The Unity March is taking place at Verdant Oasis Plaza diff --git a/tests/fixtures/text/prompts/community_report.txt b/tests/fixtures/text/prompts/community_report.txt index f939001c5b..4d78837774 100644 --- a/tests/fixtures/text/prompts/community_report.txt +++ b/tests/fixtures/text/prompts/community_report.txt @@ -35,12 +35,12 @@ Text: Entities -id,entity,description +human_readable_id,title,description 5,ABILA CITY PARK,Abila City Park is the location of the POK rally Relationships -id,source,target,description +human_readable_id,source,target,description 37,ABILA CITY PARK,POK RALLY,Abila City Park is the location of the POK rally 38,ABILA CITY PARK,POK,POK is holding a rally in Abila City Park 39,ABILA CITY PARK,POKRALLY,The POKRally is taking place at Abila City Park