diff --git a/graphrag/mind_map_extractor.py b/graphrag/mind_map_extractor.py index 9a5560daf..e33650ba5 100644 --- a/graphrag/mind_map_extractor.py +++ b/graphrag/mind_map_extractor.py @@ -66,7 +66,7 @@ class MindMapExtractor: obj = [obj] if isinstance(obj, list): for i in obj: keyset.add(i) - return [{"id": re.sub(r"\*+", "", i), "children": []} for i in obj] + return [{"id": re.sub(r"\*+", "", i), "children": []} for i in obj if re.sub(r"\*+", "", i)] arr = [] for k, v in obj.items(): k = self._key(k)