From d85bfa367b59ef1584409cc676714cd7f36e104f Mon Sep 17 00:00:00 2001
From: fzowl <zoltan@voyageai.com>
Date: Sun, 9 Jun 2024 15:16:55 +0200
Subject: [PATCH] VoyageAI new embedding and rerank models

---
 weaviate/collections/classes/config.py             | 2 +-
 weaviate/collections/classes/config_vectorizers.py | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/weaviate/collections/classes/config.py b/weaviate/collections/classes/config.py
index 34e5ea4bc..e0b7700e3 100644
--- a/weaviate/collections/classes/config.py
+++ b/weaviate/collections/classes/config.py
@@ -488,7 +488,7 @@ class _RerankerTransformersConfig(_RerankerConfigCreate):
     reranker: Rerankers = Field(default=Rerankers.TRANSFORMERS, frozen=True, exclude=True)
 
 
-RerankerVoyageAIModel = Literal["rerank-lite-1"]
+RerankerVoyageAIModel = Literal["rerank-lite-1", "rerank-1"]
 
 
 class _RerankerVoyageAIConfig(_RerankerConfigCreate):
diff --git a/weaviate/collections/classes/config_vectorizers.py b/weaviate/collections/classes/config_vectorizers.py
index e9ec6ff4e..f4263bea4 100644
--- a/weaviate/collections/classes/config_vectorizers.py
+++ b/weaviate/collections/classes/config_vectorizers.py
@@ -30,6 +30,8 @@
     "voyage-2",
     "voyage-law-2",
     "voyage-large-2-instruct",
+    "voyage-finance-2",
+    "voyage-multilingual-2",
 ]
 AWSModel: TypeAlias = Literal[
     "amazon.titan-embed-text-v1",