update bedrock rerank

warren830 · Dec 4, 2024 · 23a8497 · 23a8497
1 parent 13d0816
commit 23a8497
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
@@ -111,7 +111,7 @@ def _invoke(
 
             # score threshold check
             if score_threshold is not None:
-                if result.relevance_score >= score_threshold:
+                if rerank_document.score >= score_threshold:
                     rerank_documents.append(rerank_document)
             else:
                 rerank_documents.append(rerank_document)