Throw when trying to rerank with v1

cohere-ai · Dec 4, 2024 · d18390f · d18390f
1 parent 36d0e18
commit d18390f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/cohere/bedrock_client.py b/src/cohere/bedrock_client.py
@@ -25,6 +25,8 @@ def __init__(
             timeout=timeout,
         )
 
+    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.")
 
 class BedrockClientV2(AwsClientV2):
     def __init__(