Browse Source

fix: double brace issue (#3328)

### What problem does this PR solve?

#3299

### Type of change

- [x] Performance Improvement
tags/v0.14.0
Kevin Hu 11 months ago
parent
commit
88072b1e90
No account linked to committer's email address
1 changed files with 2 additions and 0 deletions
  1. 2
    0
      graphrag/community_reports_extractor.py

+ 2
- 0
graphrag/community_reports_extractor.py View File

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

Loading…
Cancel
Save