diff --git a/libs/vertexai/tests/integration_tests/test_embeddings.py b/libs/vertexai/tests/integration_tests/test_embeddings.py index ed1abc79..210da903 100644 --- a/libs/vertexai/tests/integration_tests/test_embeddings.py +++ b/libs/vertexai/tests/integration_tests/test_embeddings.py @@ -81,18 +81,21 @@ def test_warning(caplog: pytest.LogCaptureFixture) -> None: assert record.message == expected_message +@pytest.mark.release def test_langchain_google_vertexai_image_embeddings(tmp_image) -> None: model = VertexAIEmbeddings(model_name="multimodalembedding") output = model.embed_image(tmp_image) assert len(output) == 1408 +@pytest.mark.release def test_langchain_google_vertexai_text_model() -> None: embeddings_model = VertexAIEmbeddings(model_name="textembedding-gecko@001") assert isinstance(embeddings_model.client, TextEmbeddingModel) assert embeddings_model.model_type == GoogleEmbeddingModelType.TEXT +@pytest.mark.release def test_langchain_google_vertexai_multimodal_model() -> None: embeddings_model = VertexAIEmbeddings(model_name="multimodalembedding@001") assert isinstance(embeddings_model.client, MultiModalEmbeddingModel)