From 3d1ce4c53f0d38163d1073a5a178a16b7ab7ffbb Mon Sep 17 00:00:00 2001
From: luckylhb90 <luckylhb90@gmail.com>
Date: Tue, 21 Jan 2025 14:09:36 +0300
Subject: [PATCH] bug: fixed bedrock rerank bug (#12774)

Co-authored-by: hobo.l <hobo.l@binance.com>
---
 api/core/model_runtime/model_providers/bedrock/rerank/rerank.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/api/core/model_runtime/model_providers/bedrock/rerank/rerank.py b/api/core/model_runtime/model_providers/bedrock/rerank/rerank.py
index 9da23ba1b0..fc08852c1a 100644
--- a/api/core/model_runtime/model_providers/bedrock/rerank/rerank.py
+++ b/api/core/model_runtime/model_providers/bedrock/rerank/rerank.py
@@ -70,7 +70,7 @@ class BedrockRerankModel(RerankModel):
         rerankingConfiguration = {
             "type": "BEDROCK_RERANKING_MODEL",
             "bedrockRerankingConfiguration": {
-                "numberOfResults": top_n,
+                "numberOfResults": min(top_n, len(text_sources)),
                 "modelConfiguration": {
                     "modelArn": model_package_arn,
                 },