bug: fixed bedrock rerank bug (#12774)

Co-authored-by: hobo.l <hobo.l@binance.com>
langgenius · Jan 21, 2025 · 3d1ce4c · 3d1ce4c
1 parent 6db3ae9
commit 3d1ce4c
Showing 1 changed file with 1 addition and 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
@@ -70,7 +70,7 @@ def _invoke(
         rerankingConfiguration = {
             "type": "BEDROCK_RERANKING_MODEL",
             "bedrockRerankingConfiguration": {
-                "numberOfResults": top_n,
+                "numberOfResults": min(top_n, len(text_sources)),
                 "modelConfiguration": {
                     "modelArn": model_package_arn,
                 },