docs: add deprecation notice to readme for Generative AI submodules: vertexai.generative_models, vertexai.language_models, vertexai.vision_models, vertexai.tuning, vertexai.caching

PiperOrigin-RevId: 775007033

Author: sararob

README.rst

@@ -1,6 +1,15 @@
 Vertex AI SDK for Python
 =================================================
 
+.. note::
+
+   The following Generative AI modules in the Vertex AI SDK are deprecated as of June 24, 2025 and will be removed on June 24, 2026:
+   `vertexai.generative_models`, `vertexai.language_models`, `vertexai.vision_models`, `vertexai.tuning`, `vertexai.batch_prediction`, `vertexai.caching`. Please use the
+   [Google Gen AI SDK](https://pypi.org/project/google-genai/) to access these features. See
+   [the migration guide](https://cloud.google.com/vertex-ai/generative-ai/docs/deprecations/genai-vertexai-sdk) for details.
+   You can continue using all other Vertex AI SDK modules, as they are the recommended way to use the API.
+
+
 |GA| |pypi| |versions| |unit-tests| |system-tests| |sample-tests|
 
 `Vertex AI`_: Google Vertex AI is an integrated suite of machine learning tools and services for building and using ML models with AutoML or custom code. It offers both novices and experts the best workbench for the entire machine learning development lifecycle.

vertexai/_utils/deprecation_warning.py

@@ -0,0 +1,51 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import functools
+from typing import Any, Callable, Optional
+import warnings
+
+BASE_WARNING_MESSAGE = (
+    "This feature is deprecated as of June 24, 2025 and will be removed on June"
+    " 24, 2026. For details, see"
+    " https://cloud.google.com/vertex-ai/generative-ai/docs/deprecations/genai-vertexai-sdk."
+)
+
+
+class DeprecationWarning(Warning):
+    """A warning that a feature is deprecated."""
+
+
+def genai_class_deprecation_warning(
+    message: Optional[str] = BASE_WARNING_MESSAGE,
+) -> Callable[[Callable[..., Any]], Callable[..., Any]]:
+    """Deprecation warning decorator."""
+
+    def decorator(cls) -> Any:
+
+        original_class_init = cls.__init__
+
+        @functools.wraps(original_class_init)
+        def wrapper(self, *args: Any, **kwargs: Any) -> Any:
+            warnings.warn(
+                message=message,
+                category=DeprecationWarning,
+                stacklevel=2,
+            )
+            original_class_init(self, *args, **kwargs)
+
+        cls.__init__ = wrapper
+        return cls
+
+    return decorator

vertexai/caching/_caching.py

@@ -49,6 +49,7 @@
     ContentsType,
 )
 from google.protobuf import field_mask_pb2
+from vertexai._utils import deprecation_warning
 
 
 def _prepare_create_request(
@@ -127,6 +128,7 @@ def _prepare_get_cached_content_request(name: str) -> GetCachedContentRequest:
     return types_v1.GetCachedContentRequest(name=name)
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class CachedContent(aiplatform_base._VertexAiResourceNounPlus):
     """A cached content resource."""
 

vertexai/generative_models/_generative_models.py

@@ -63,6 +63,8 @@
 from google.protobuf import field_mask_pb2
 import warnings
 
+from vertexai._utils import deprecation_warning
+
 if TYPE_CHECKING:
     from vertexai.caching import CachedContent
 
@@ -3406,6 +3408,7 @@ def respond_to_model_response(
             return function_response_content
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class GenerativeModel(_GenerativeModel):
     __module__ = "vertexai.generative_models"
 

vertexai/language_models/_language_models.py

@@ -42,6 +42,8 @@
 from vertexai.language_models import (
     _evaluatable_language_models,
 )
+from vertexai._utils import deprecation_warning
+
 
 try:
     import pandas
@@ -1157,6 +1159,7 @@ def _to_grounding_source_dict(self) -> Dict[str, Any]:
         }
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 @dataclasses.dataclass
 class GroundingSource:
 
@@ -1236,6 +1239,7 @@ def __init__(self, response: Optional[Dict[str, Any]] = {}):
         self.search_queries = response.get("searchQueries", [])
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 @dataclasses.dataclass
 class TextGenerationResponse:
     """TextGenerationResponse represents a response of a language model.
@@ -1950,6 +1954,7 @@ def batch_predict(
         )
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class TextGenerationModel(
     _TextGenerationModel,
     _TunableTextModelMixin,
@@ -2070,6 +2075,7 @@ def send_message(
         return response_obj
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 @dataclasses.dataclass
 class TextEmbeddingInput:
     """Structural text embedding input.
@@ -2422,6 +2428,7 @@ class _TunableTextEmbeddingModelMixin(_PreviewTunableTextEmbeddingModelMixin):
     pass
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class TextEmbeddingModel(
     _TextEmbeddingModel,
     _ModelWithBatchPredict,
@@ -2451,6 +2458,7 @@ class TextEmbeddingStatistics:
     truncated: bool
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 @dataclasses.dataclass
 class TextEmbedding:
     """Text embedding vector and statistics."""
@@ -2492,6 +2500,7 @@ def _parse_text_embedding_response(
             return cls(values=prediction, _prediction_response=prediction_response)
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 @dataclasses.dataclass
 class InputOutputTextPair:
     """InputOutputTextPair represents a pair of input and output texts."""
@@ -2502,6 +2511,7 @@ class InputOutputTextPair:
     output_text: str
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 @dataclasses.dataclass
 class ChatMessage:
     """A chat message.
@@ -2562,6 +2572,7 @@ def start_chat(
         )
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class ChatModel(_ChatModelBase, _TunableChatModelMixin, _RlhfTunableModelMixin):
     """ChatModel represents a language model that is capable of chat.
 
@@ -2638,6 +2649,7 @@ def start_chat(
         )
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class CodeChatModel(_ChatModelBase, _TunableChatModelMixin):
     """CodeChatModel represents a model that is capable of completing code.
 
@@ -3255,6 +3267,7 @@ class _PreviewCodeChatSession(_ChatSessionBaseWithCountTokensMixin):
     __module__ = "vertexai.preview.language_models"
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class ChatSession(_ChatSessionBase):
     """ChatSession represents a chat session with a language model.
 
@@ -3288,6 +3301,7 @@ def __init__(
         )
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class CodeChatSession(_ChatSessionBase):
     """CodeChatSession represents a chat session with code chat language model.
 
@@ -3720,6 +3734,7 @@ def count_tokens(
         )
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class CodeGenerationModel(
     _CodeGenerationModel,
     _TunableTextModelMixin,

vertexai/tuning/_tuning.py

@@ -39,6 +39,8 @@
 
 from google.rpc import status_pb2  # type: ignore
 
+from vertexai._utils import deprecation_warning
+
 
 _LOGGER = aiplatform_base.Logger(__name__)
 
@@ -52,6 +54,7 @@ class TuningJobClientWithOverride(aiplatform_utils.ClientWithOverride):
     )
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class TuningJob(aiplatform_base._VertexAiResourceNounPlus):
     """Represents a TuningJob that runs with Google owned models."""
 

vertexai/vision_models/_vision_models.py

@@ -29,6 +29,7 @@
 
 from google.cloud.aiplatform import initializer as aiplatform_initializer
 from vertexai._model_garden import _model_garden_models
+from vertexai._utils import deprecation_warning
 
 # pylint: disable=g-import-not-at-top
 try:
@@ -129,6 +130,7 @@ class SubjectImageConfig:
     ]
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class Image:
     """Image."""
 
@@ -467,6 +469,7 @@ def __init__(
         )
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class Video:
     """Video."""
 
@@ -572,6 +575,7 @@ def _as_base64_string(self) -> str:
         return base64.b64encode(self._video_bytes).decode("ascii")
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class VideoSegmentConfig:
     """The specific video segments (in seconds) the embeddings are generated for."""
 
@@ -599,6 +603,7 @@ def __init__(
         self.interval_sec = interval_sec
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class VideoEmbedding:
     """Embeddings generated from video with offset times."""
 
@@ -623,6 +628,7 @@ def __init__(
         self.embedding = embedding
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class ImageGenerationModel(
     _model_garden_models._ModelGardenModel  # pylint: disable=protected-access
 ):
@@ -1377,6 +1383,7 @@ def upscale_image(
         )
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 @dataclasses.dataclass
 class ImageGenerationResponse:
     """Image generation response.
@@ -1404,6 +1411,7 @@ def __getitem__(self, idx: int) -> "GeneratedImage":
 )
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class GeneratedImage(Image):
     """Generated image."""
 
@@ -1475,6 +1483,7 @@ def save(self, location: str, include_generation_parameters: bool = True):
             super().save(location=location)
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class ImageCaptioningModel(
     _model_garden_models._ModelGardenModel  # pylint: disable=protected-access
 ):
@@ -1540,6 +1549,7 @@ def get_captions(
         return response.predictions
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class ImageQnAModel(
     _model_garden_models._ModelGardenModel  # pylint: disable=protected-access
 ):
@@ -1599,6 +1609,7 @@ def ask_question(
         return response.predictions
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class MultiModalEmbeddingModel(_model_garden_models._ModelGardenModel):
     """Generates embedding vectors from images and videos.
 
@@ -1724,6 +1735,7 @@ def get_embeddings(
         )
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 @dataclasses.dataclass
 class MultiModalEmbeddingResponse:
     """The multimodal embedding response.
@@ -1745,6 +1757,7 @@ class MultiModalEmbeddingResponse:
     text_embedding: Optional[List[float]] = None
 
 
+@deprecation_warning.genai_class_deprecation_warning()
 class ImageTextModel(ImageCaptioningModel, ImageQnAModel):
     """Generates text from images.