diff --git a/builtin_tools/aws/tools/bedrock_retrieve.py b/builtin_tools/aws/tools/bedrock_retrieve.py index 19e7bfa..d4f847c 100644 --- a/builtin_tools/aws/tools/bedrock_retrieve.py +++ b/builtin_tools/aws/tools/bedrock_retrieve.py @@ -33,7 +33,8 @@ def _bedrock_retrieve( } if rerank_model_id != "default": - model_for_rerank_arn = f"arn:aws:bedrock:us-west-2::foundation-model/{rerank_model_id}" + region = self.bedrock_client.meta.region_name + model_for_rerank_arn = f"arn:aws:bedrock:{region}::foundation-model/{rerank_model_id}" rerankingConfiguration = { "bedrockRerankingConfiguration": { "numberOfRerankedResults": num_results,