Merge pull request #1318 from voyage-ai/main

Introducing VoyageAI's new embedding and rerank models
weaviate · Sep 25, 2024 · e75cb29 · e75cb29
2 parents de26e2d + 28b7a3b
commit e75cb29
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/weaviate/collections/classes/config.py b/weaviate/collections/classes/config.py
@@ -605,7 +605,7 @@ class _RerankerJinaAIConfig(_RerankerConfigCreate):
     model: Optional[Union[RerankerJinaAIModel, str]] = Field(default=None)
 
 
-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
@@ -33,6 +33,8 @@
     "jina-embeddings-v3",
 ]
 VoyageModel: TypeAlias = Literal[
+    "voyage-3",
+    "voyage-3-lite",
     "voyage-large-2",
     "voyage-code-2",
     "voyage-2",