Update bedrock_client.py

cohere-ai · Dec 4, 2024 · a878da9 · a878da9
1 parent 9fe3df8
commit a878da9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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__(