fix: double brace issue (#3328)

### What problem does this PR solve? #3299 ### Type of change - [x] Performance Improvement
infiniflow · Nov 11, 2024 · 88072b1 · 88072b1
1 parent 34d1daa
commit 88072b1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/graphrag/community_reports_extractor.py b/graphrag/community_reports_extractor.py
@@ -80,6 +80,8 @@ def __call__(self, graph: nx.Graph, callback: Callable | None = None):
                     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, [