diff --git a/test/collection/test_config.py b/test/collection/test_config.py index 35ca349a0..e5ef861d2 100644 --- 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"])], {