Fix model names

cohere-ai · Nov 4, 2024 · 68ecc72 · 68ecc72
1 parent b637cf8
commit 68ecc72
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/test/bedrock-tests.test.ts b/src/test/bedrock-tests.test.ts
@@ -17,11 +17,13 @@ const models: Record<AwsPlatform, Record<AwsEndpoint, string>> = {
         generate: "cohere.command-text-v14",
         embed: "cohere.embed-multilingual-v3",
         chat: "cohere.command-r-plus-v1:0",
+        rerank: "cohere.rerank-v1",
     },
     sagemaker: {
         generate: "cohere-command-light",
         embed: "cohere-embed-multilingual-v3",
         chat: "cohere-command-plus",
+        rerank: "cohere.rerank-v1",
     },
 };