Add tests for google ai studio method

weaviate · Jan 13, 2025 · e7af01f · e7af01f
1 parent 8482f4b
commit e7af01f
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/test/collection/test_config.py b/test/collection/test_config.py
@@ -318,6 +318,15 @@ def test_basic_config():
             }
         },
     ),
+    (
+        Configure.Vectorizer.text2vec_google_aistudio(),
+        {
+            "text2vec-palm": {
+                "apiEndpoint": "generativelanguage.googleapis.com",
+                "vectorizeClassName": True,
+            }
+        },
+    ),
     (
         Configure.Vectorizer.text2vec_palm(
             project_id="project",
@@ -1509,6 +1518,26 @@ def test_vector_config_flat_pq() -> None:
             }
         },
     ),
+    (
+        [
+            Configure.NamedVectors.text2vec_google_aistudio(
+                name="test",
+                source_properties=["prop"],
+            )
+        ],
+        {
+            "test": {
+                "vectorizer": {
+                    "text2vec-palm": {
+                        "apiEndpoint": "generativelanguage.googleapis.com",
+                        "properties": ["prop"],
+                        "vectorizeClassName": True,
+                    }
+                },
+                "vectorIndexType": "hnsw",
+            }
+        },
+    ),
     (
         [Configure.NamedVectors.text2vec_transformers(name="test", source_properties=["prop"])],
         {