From 88072b1e900cf69716038bba77087e19f38ee45e Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Mon, 11 Nov 2024 12:07:02 +0800 Subject: [PATCH] fix: double brace issue (#3328) ### What problem does this PR solve? #3299 ### Type of change - [x] Performance Improvement --- graphrag/community_reports_extractor.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/graphrag/community_reports_extractor.py b/graphrag/community_reports_extractor.py index 1412555e4..5efc120af 100644 --- a/graphrag/community_reports_extractor.py +++ b/graphrag/community_reports_extractor.py @@ -80,6 +80,8 @@ class CommunityReportsExtractor: token_count += num_tokens_from_string(text + response) response = re.sub(r"^[^\{]*", "", response) response = re.sub(r"[^\}]*$", "", response) + response = re.sub(r"\{\{", "{", response) + response = re.sub(r"\}\}", "}", response) print(response) response = json.loads(response) if not dict_has_keys_with_types(response, [