Add tests

weaviate · Aug 19, 2024 · e0c4df9 · e0c4df9
1 parent 7affa78
commit e0c4df9
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 1 deletion.
diff --git a/test/collection/test_config.py b/test/collection/test_config.py
@@ -250,6 +250,18 @@ def test_basic_config():
             }
         },
     ),
+    (
+        Configure.Vectorizer.text2vec_mistral(
+            vectorize_collection_name=False,
+            model="cool-model",
+        ),
+        {
+            "text2vec-mistral": {
+                "vectorizeClassName": False,
+                "model": "cool-model",
+            }
+        },
+    ),
     (
         Configure.Vectorizer.text2vec_palm(
             project_id="project",
@@ -1185,6 +1197,20 @@ def test_vector_config_flat_pq() -> None:
             }
         },
     ),
+    (
+        [Configure.NamedVectors.text2vec_mistral(name="test", source_properties=["prop"])],
+        {
+            "test": {
+                "vectorizer": {
+                    "text2vec-mistral": {
+                        "vectorizeClassName": True,
+                        "properties": ["prop"],
+                    }
+                },
+                "vectorIndexType": "hnsw",
+            }
+        },
+    ),
     (
         [
             Configure.NamedVectors.text2vec_palm(

diff --git a/test/collection/test_vectorizer.py b/test/collection/test_vectorizer.py
@@ -0,0 +1,5 @@
+from weaviate.collections.classes.config import Configure
+
+
+def test_multi2vec_clip() -> None:
+    Configure.Vectorizer.multi2vec_clip(image_fields=["test"])
diff --git a/weaviate/collections/classes/config_vectorizers.py b/weaviate/collections/classes/config_vectorizers.py
@@ -232,7 +232,7 @@ class _Text2VecHuggingFaceConfigCreate(_Text2VecHuggingFaceConfig, _VectorizerCo
 
 
 class _Text2VecMistralConfig(_VectorizerConfigCreate):
-    vectorizer: Vectorizers = Field(default=Vectorizers.TEXT2VEC_MISTRAL, frozen=True, exclude=True)
+    vectorizer: Union[Vectorizers, _EnumLikeStr] = Field(default=Vectorizers.TEXT2VEC_MISTRAL, frozen=True, exclude=True)
     model: Optional[str]
     vectorizeClassName: bool