youngkingdom/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-05-03 00:37:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: remove useless variable in raptor (#13648)

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

remove useless variable in raptor

### Type of change


- [x] Refactoring
This commit is contained in:
Stephen Hu
2026-03-17 15:56:51 +08:00
committed by GitHub
parent 986dcf1cc8
commit 77483b1e58
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
rag/raptor.py
View File

@ -161,7 +161,6 @@ class RecursiveAbstractiveProcessing4TreeOrganizedRetrieval:
if self._error_count >= self._max_errors:
raise RuntimeError(f"RAPTOR aborted after {self._error_count} errors. Last error: {exc}") from exc
labels = []
while end - start > 1:
self._check_task_canceled(task_id, "layer processing")
@ -170,7 +169,6 @@ class RecursiveAbstractiveProcessing4TreeOrganizedRetrieval:
await summarize([start, start + 1])
if callback:
callback(msg="Cluster one layer: {} -> {}".format(end - start, len(chunks) - end))
labels.extend([0, 0])
layers.append((end, len(chunks)))
start = end
end = len(chunks)
@ -208,7 +206,6 @@ class RecursiveAbstractiveProcessing4TreeOrganizedRetrieval:
raise
assert len(chunks) - end == n_clusters, "{} vs. {}".format(len(chunks) - end, n_clusters)
labels.extend(lbls)
layers.append((end, len(chunks)))
if callback:
callback(msg="Cluster one layer: {} -> {}".format(end - start, len(chunks) - end))