jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

bug: fixed bedrock rerank bug (#12774)

Browse source 
Co-authored-by: hobo.l <hobo.l@binance.com>
This commit is contained in:
luckylhb90 2025-01-21 14:09:36 +03:00 • committed by GitHub
commit 3d1ce4c53f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/core/model_runtime/model_providers/bedrock/rerank/rerank.py
View file

@ -70,7 +70,7 @@ class BedrockRerankModel(RerankModel):
rerankingConfiguration = { rerankingConfiguration = {
"type": "BEDROCK_RERANKING_MODEL", "type": "BEDROCK_RERANKING_MODEL",
"bedrockRerankingConfiguration": { "bedrockRerankingConfiguration": {
"numberOfResults": top_n, "numberOfResults": min(top_n, len(text_sources)),
"modelConfiguration": { "modelConfiguration": {
"modelArn": model_package_arn, "modelArn": model_package_arn,
}, },