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, },