bug: fixed bedrock rerank bug (#12774)

Co-authored-by: hobo.l <hobo.l@binance.com>
1 year ago · 3d1ce4c53f
parent 6db3ae9b8e
commit 3d1ce4c53f
1 changed files with 1 additions and 1 deletions
--- 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,
                },