From cfc900c25a10e6ffe95ddc4da393caa7c48a4338 Mon Sep 17 00:00:00 2001
From: fzowl <zoltan@voyageai.com>
Date: Wed, 25 Sep 2024 12:40:37 +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 e0b7700e3..0718eccf3 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", "rerank-1"]
+RerankerVoyageAIModel = Literal["rerank-2", "rerank-2-lite", "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 f4263bea4..2680e2071 100644
--- a/weaviate/collections/classes/config_vectorizers.py
+++ b/weaviate/collections/classes/config_vectorizers.py
@@ -25,6 +25,8 @@
 ]
 JinaModel: TypeAlias = Literal["jina-embeddings-v2-base-en", "jina-embeddings-v2-small-en"]
 VoyageModel: TypeAlias = Literal[
+    "voyage-3",
+    "voyage-3-lite",
     "voyage-large-2",
     "voyage-code-2",
     "voyage-2",