From 1ed0b25910c61726b4198943cde920dce30836f3 Mon Sep 17 00:00:00 2001 From: Zhichang Yu Date: Mon, 9 Jun 2025 10:18:03 +0800 Subject: [PATCH] Fix task_limiter in raptor.py (#8124) ### What problem does this PR solve? Fix task_limiter in raptor.py ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- rag/raptor.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/rag/raptor.py b/rag/raptor.py index 007f2529a..db2c82d2e 100644 --- a/rag/raptor.py +++ b/rag/raptor.py @@ -151,8 +151,7 @@ class RecursiveAbstractiveProcessing4TreeOrganizedRetrieval: for c in range(n_clusters): ck_idx = [i + start for i in range(len(lbls)) if lbls[i] == c] assert len(ck_idx) > 0 - async with chat_limiter: - nursery.start_soon(summarize, ck_idx) + nursery.start_soon(summarize, ck_idx) assert len(chunks) - end == n_clusters, "{} vs. {}".format( len(chunks) - end, n_clusters