From 55f7a169e3f8b3d26f45d2a2708145f2125dcde1 Mon Sep 17 00:00:00 2001
From: Augustas <augustas.skaburskas@gmail.com>
Date: Mon, 20 Jan 2025 15:28:39 +0200
Subject: [PATCH] add Snowflake/snowflake-arctic-embed-l-v2.0 model to weaviate
 vectorizer

---
 test/collection/test_config.py                   | 16 ++++++++++++++++
 .../collections/classes/config_vectorizers.py    |  4 +++-
 2 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/test/collection/test_config.py b/test/collection/test_config.py
index da4cd0188..5b9349dcc 100644
--- a/test/collection/test_config.py
+++ b/test/collection/test_config.py
@@ -428,6 +428,22 @@ def test_basic_config():
             }
         },
     ),
+    (
+        Configure.Vectorizer.text2vec_weaviate(
+            vectorize_collection_name=False,
+            model="Snowflake/snowflake-arctic-embed-l-v2.0",
+            base_url="https://api.embedding.weaviate.io",
+            dimensions=1024,
+        ),
+        {
+            "text2vec-weaviate": {
+                "vectorizeClassName": False,
+                "model": "Snowflake/snowflake-arctic-embed-l-v2.0",
+                "baseURL": "https://api.embedding.weaviate.io",
+                "dimensions": 1024,
+            }
+        },
+    ),
     (
         Configure.Vectorizer.text2vec_weaviate(
             vectorize_collection_name=False,
diff --git a/weaviate/collections/classes/config_vectorizers.py b/weaviate/collections/classes/config_vectorizers.py
index 96074d969..f83b9be89 100644
--- a/weaviate/collections/classes/config_vectorizers.py
+++ b/weaviate/collections/classes/config_vectorizers.py
@@ -65,7 +65,9 @@
     "bedrock",
     "sagemaker",
 ]
-WeaviateModel: TypeAlias = Literal["Snowflake/snowflake-arctic-embed-m-v1.5"]
+WeaviateModel: TypeAlias = Literal[
+    "Snowflake/snowflake-arctic-embed-l-v2.0", "Snowflake/snowflake-arctic-embed-m-v1.5"
+]
 
 
 class Vectorizers(str, Enum):