chore: remove prompt_optimizer property from GenAI client

PiperOrigin-RevId: 773126369

Author: sararob

tests/unit/vertexai/genai/test_prompt_optimizer.py

@@ -28,8 +28,9 @@
 from google.cloud.aiplatform.compat.types import (
     job_state as gca_job_state_compat,
 )
-from google.cloud.aiplatform.utils import gcs_utils
-from google.genai import client
+
+# from google.cloud.aiplatform.utils import gcs_utils
+# from google.genai import client
 import pytest
 
 
@@ -76,29 +77,29 @@ def setup_method(self):
             location=_TEST_LOCATION,
         )
 
-    @pytest.mark.usefixtures("google_auth_mock")
-    def test_prompt_optimizer_client(self):
-        test_client = vertexai.Client(project=_TEST_PROJECT, location=_TEST_LOCATION)
-        assert test_client is not None
-        assert test_client._api_client.vertexai
-        assert test_client._api_client.project == _TEST_PROJECT
-        assert test_client._api_client.location == _TEST_LOCATION
+    # @pytest.mark.usefixtures("google_auth_mock")
+    # def test_prompt_optimizer_client(self):
+    #     test_client = vertexai.Client(project=_TEST_PROJECT, location=_TEST_LOCATION)
+    #     assert test_client is not None
+    #     assert test_client._api_client.vertexai
+    #     assert test_client._api_client.project == _TEST_PROJECT
+    #     assert test_client._api_client.location == _TEST_LOCATION
 
-    @mock.patch.object(client.Client, "_get_api_client")
-    @mock.patch.object(
-        gcs_utils.resource_manager_utils, "get_project_number", return_value=12345
-    )
-    def test_prompt_optimizer_optimize(
-        self, mock_get_project_number, mock_client, mock_create_custom_job
-    ):
-        """Test that prompt_optimizer.optimize method creates a custom job."""
-        test_client = vertexai.Client(project=_TEST_PROJECT, location=_TEST_LOCATION)
-        test_client.prompt_optimizer.optimize(
-            method="vapo",
-            config={
-                "config_path": "gs://ssusie-vapo-sdk-test/config.json",
-                "wait_for_completion": False,
-            },
-        )
-        mock_create_custom_job.assert_called_once()
-        mock_get_project_number.assert_called_once()
+    # @mock.patch.object(client.Client, "_get_api_client")
+    # @mock.patch.object(
+    #     gcs_utils.resource_manager_utils, "get_project_number", return_value=12345
+    # )
+    # def test_prompt_optimizer_optimize(
+    #     self, mock_get_project_number, mock_client, mock_create_custom_job
+    # ):
+    #     """Test that prompt_optimizer.optimize method creates a custom job."""
+    #     test_client = vertexai.Client(project=_TEST_PROJECT, location=_TEST_LOCATION)
+    #     test_client.prompt_optimizer.optimize(
+    #         method="vapo",
+    #         config={
+    #             "config_path": "gs://ssusie-vapo-sdk-test/config.json",
+    #             "wait_for_completion": False,
+    #         },
+    #     )
+    #     mock_create_custom_job.assert_called_once()
+    #     mock_get_project_number.assert_called_once()

vertexai/_genai/client.py

@@ -124,17 +124,6 @@ def evals(self):
                 ) from e
         return self._evals.Evals(self._api_client)
 
-    @property
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI prompt optimizer module is experimental, "
-        "and may change in future versions."
-    )
-    def prompt_optimizer(self):
-        self._prompt_optimizer = importlib.import_module(
-            ".prompt_optimizer", __package__
-        )
-        return self._prompt_optimizer.PromptOptimizer(self._api_client)
-
     @property
     @_common.experimental_warning(
         "The Vertex SDK GenAI async client is experimental, "