gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update generator.py

Browse source
This commit is contained in:
“hu 2025-12-09 09:09:44 +08:00
parent 09a3854ed8
commit 9bd82fa077
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
rag/prompts/generator.py
View file

@ -781,7 +781,10 @@ async def run_toc_from_text(chunks, chat_mdl, callback=None):
# Merge structure and content (by index) # Merge structure and content (by index)
prune = len(toc_with_levels) > 512 prune = len(toc_with_levels) > 512
max_lvl = sorted([t.get("level", "0") for t in toc_with_levels if isinstance(t, dict)])[-1] max_lvl = "0"
sorted_list = sorted([t.get("level", "0") for t in toc_with_levels if isinstance(t, dict)])
if sorted_list:
max_lvl = sorted_list[-1]
merged = [] merged = []
for _ , (toc_item, src_item) in enumerate(zip(toc_with_levels, filtered)): for _ , (toc_item, src_item) in enumerate(zip(toc_with_levels, filtered)):
if prune and toc_item.get("level", "0") >= max_lvl: if prune and toc_item.get("level", "0") >= max_lvl: