AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-12 23:38:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix index of range (#3279)

Browse Source 
### What problem does this PR solve?

#3273

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu 2024-11-08 09:58:11 +08:00 committed by GitHub
parent 416efbe7e8
commit ff2bbb487f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
graphrag/mind_map_extractor.py
View File

@ -39,7 +39,6 @@ class MindMapResult:
class MindMapExtractor: class MindMapExtractor:
_llm: CompletionLLM _llm: CompletionLLM
_input_text_key: str _input_text_key: str
_mind_map_prompt: str _mind_map_prompt: str
@ -93,7 +92,7 @@ class MindMapExtractor:
max_workers = int(os.environ.get('MINDMAP_EXTRACTOR_MAX_WORKERS', 12)) max_workers = int(os.environ.get('MINDMAP_EXTRACTOR_MAX_WORKERS', 12))
exe = ThreadPoolExecutor(max_workers=max_workers) exe = ThreadPoolExecutor(max_workers=max_workers)
threads = [] threads = []
token_count = max(self._llm.max_length * 0.8, self._llm.max_length-512) token_count = max(self._llm.max_length * 0.8, self._llm.max_length - 512)
texts = [] texts = []
res = [] res = []
cnt = 0 cnt = 0
@ -163,14 +162,14 @@ class MindMapExtractor:
elif isinstance(value, list): elif isinstance(value, list):
new_value = {} new_value = {}
for i in range(len(value)): for i in range(len(value)):
if isinstance(value[i], list): if isinstance(value[i], list) and i > 0:
new_value[value[i - 1]] = value[i][0] new_value[value[i - 1]] = value[i][0]
data[key] = new_value data[key] = new_value
else: else:
continue continue
return data return data
def _todict(self, layer:collections.OrderedDict): def _todict(self, layer: collections.OrderedDict):
to_ret = layer to_ret = layer
if isinstance(layer, collections.OrderedDict): if isinstance(layer, collections.OrderedDict):
to_ret = dict(layer) to_ret = dict(layer)