From a878da9b2b9c7c873565eea71b90bccf174b7e92 Mon Sep 17 00:00:00 2001
From: billytrend-cohere <144115527+billytrend-cohere@users.noreply.github.com>
Date: Wed, 4 Dec 2024 12:26:47 -0500
Subject: [PATCH] Update bedrock_client.py

---
 src/cohere/bedrock_client.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/cohere/bedrock_client.py b/src/cohere/bedrock_client.py
index 3c1d33512..16f47b031 100644
--- a/src/cohere/bedrock_client.py
+++ b/src/cohere/bedrock_client.py
@@ -26,7 +26,7 @@ def __init__(
         )
 
     def rerank(self, *, query, documents, model = ..., top_n = ..., rank_fields = ..., return_documents = ..., max_chunks_per_doc = ..., request_options = None):
-        raise NotImplementedError("Please use cohere.BedrockClientV2 instead: method rerank is not available in the v1 client, cohere.BedrockClient.")
+        raise NotImplementedError("Please use cohere.BedrockClientV2 instead: Rerank API on Bedrock is not supported with cohere.BedrockClient for this model.")
 
 class BedrockClientV2(AwsClientV2):
     def __init__(