Kevin Hu
commited on
Commit
·
140db5b
1
Parent(s):
d102a8b
fix index of range (#3279)
Browse files### What problem does this PR solve?
#3273
### Type of change
- [x] Bug Fix (non-breaking change which fixes an issue)
graphrag/mind_map_extractor.py
CHANGED
@@ -39,7 +39,6 @@ class MindMapResult:
|
|
39 |
|
40 |
|
41 |
class MindMapExtractor:
|
42 |
-
|
43 |
_llm: CompletionLLM
|
44 |
_input_text_key: str
|
45 |
_mind_map_prompt: str
|
@@ -93,7 +92,7 @@ class MindMapExtractor:
|
|
93 |
max_workers = int(os.environ.get('MINDMAP_EXTRACTOR_MAX_WORKERS', 12))
|
94 |
exe = ThreadPoolExecutor(max_workers=max_workers)
|
95 |
threads = []
|
96 |
-
token_count = max(self._llm.max_length * 0.8, self._llm.max_length-512)
|
97 |
texts = []
|
98 |
res = []
|
99 |
cnt = 0
|
@@ -163,14 +162,14 @@ class MindMapExtractor:
|
|
163 |
elif isinstance(value, list):
|
164 |
new_value = {}
|
165 |
for i in range(len(value)):
|
166 |
-
if isinstance(value[i], list):
|
167 |
new_value[value[i - 1]] = value[i][0]
|
168 |
data[key] = new_value
|
169 |
else:
|
170 |
continue
|
171 |
return data
|
172 |
|
173 |
-
def _todict(self, layer:collections.OrderedDict):
|
174 |
to_ret = layer
|
175 |
if isinstance(layer, collections.OrderedDict):
|
176 |
to_ret = dict(layer)
|
|
|
39 |
|
40 |
|
41 |
class MindMapExtractor:
|
|
|
42 |
_llm: CompletionLLM
|
43 |
_input_text_key: str
|
44 |
_mind_map_prompt: str
|
|
|
92 |
max_workers = int(os.environ.get('MINDMAP_EXTRACTOR_MAX_WORKERS', 12))
|
93 |
exe = ThreadPoolExecutor(max_workers=max_workers)
|
94 |
threads = []
|
95 |
+
token_count = max(self._llm.max_length * 0.8, self._llm.max_length - 512)
|
96 |
texts = []
|
97 |
res = []
|
98 |
cnt = 0
|
|
|
162 |
elif isinstance(value, list):
|
163 |
new_value = {}
|
164 |
for i in range(len(value)):
|
165 |
+
if isinstance(value[i], list) and i > 0:
|
166 |
new_value[value[i - 1]] = value[i][0]
|
167 |
data[key] = new_value
|
168 |
else:
|
169 |
continue
|
170 |
return data
|
171 |
|
172 |
+
def _todict(self, layer: collections.OrderedDict):
|
173 |
to_ret = layer
|
174 |
if isinstance(layer, collections.OrderedDict):
|
175 |
to_ret = dict(layer)
|