Change top_n if number of input nodes are less

run-llama · Dec 26, 2024 · 52dafb9 · 52dafb9
1 parent 2c85e1c
commit 52dafb9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/...llama-index-postprocessor-bedrock-rerank/llama_index/postprocessor/bedrock_rerank/base.py b/...llama-index-postprocessor-bedrock-rerank/llama_index/postprocessor/bedrock_rerank/base.py
@@ -208,6 +208,9 @@ def _postprocess_nodes(
                         },
                     }
                 )
+            # change top_n if the number of nodes is less than top_n
+            if len(nodes) < self.top_n:
+                self.top_n = len(nodes)
 
             queries = [
                 {