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