AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 06:55:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

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
parent 6db3ae9b8e
commit 3d1ce4c53f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats 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,
}, },