Merge pull request #1065 from voyage-ai/main

Adding voyage-large-2-instruct model
weaviate · May 16, 2024 · a9369e2 · a9369e2
2 parents ef07b8a + 6725c91
commit a9369e2
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/weaviate/collections/classes/config_vectorizers.py b/weaviate/collections/classes/config_vectorizers.py
@@ -24,7 +24,13 @@
  "text-embedding-3-small", "text-embedding-3-large", "text-embedding-ada-002"
 ]
 JinaModel: TypeAlias = Literal["jina-embeddings-v2-base-en", "jina-embeddings-v2-small-en"]
-VoyageModel: TypeAlias = Literal["voyage-large-2, voyage-code-2, voyage-2"]
+VoyageModel: TypeAlias = Literal[
+ "voyage-large-2",
+ "voyage-code-2",
+ "voyage-2",
+ "voyage-law-2",
+ "voyage-large-2-instruct",
+]
 AWSModel: TypeAlias = Literal[
  "amazon.titan-embed-text-v1",
  "cohere.embed-english-v3",