chore: GenAI Client - remove experimental (non-GA) warnings from Python SDK

PiperOrigin-RevId: 947753708

Author: jsondai

agentplatform/_genai/client.py

@@ -21,7 +21,6 @@
 
 import google.auth
 from google.cloud.aiplatform import version as aip_version
-from google.genai import _common
 from google.genai import client as genai_client
 from google.genai import types
 from google.genai import version as genai_version
@@ -94,10 +93,6 @@ def __init__(self, api_client: genai_client.BaseApiClient):  # type: ignore[name
         self._feedback_entries: Optional[ModuleType] = None
 
     @property
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI live module is experimental, and may change in future "
-        "versions."
-    )
     def live(self) -> "live_module.AsyncLive":
         return self._live
 
@@ -152,10 +147,6 @@ def prompts(self) -> "prompts_module.AsyncPrompts":
         return self._prompts.AsyncPrompts(self._api_client)  # type: ignore[no-any-return]
 
     @property
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI async datasets module is experimental, "
-        "and may change in future versions."
-    )
     def datasets(self) -> "datasets_module.AsyncDatasets":
         if self._datasets is None:
             self._datasets = importlib.import_module(
@@ -183,10 +174,6 @@ def feedback_entries(self) -> "feedback_entries_module.AsyncFeedbackEntries":
         return self._feedback_entries.AsyncFeedbackEntries(self._api_client)  # type: ignore[no-any-return]
 
     @property
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI async rag module is experimental, "
-        "and may change in future versions."
-    )
     def rag(self) -> "rag_module.AsyncRag":
         if self._rag is None:
             self._rag = importlib.import_module(
@@ -196,9 +183,6 @@ def rag(self) -> "rag_module.AsyncRag":
         return self._rag.AsyncRag(self._api_client)  # type: ignore[no-any-return]
 
     @property
-    @_common.experimental_warning(
-        "The Model Garden module is experimental, and may change in future " "versions."
-    )
     def model_garden(self) -> "model_garden_module.AsyncModelGarden":
         if self._model_garden is None:
             self._model_garden = importlib.import_module(
@@ -399,10 +383,6 @@ def prompts(self) -> "prompts_module.Prompts":
         return self._prompts.Prompts(self._api_client)  # type: ignore[no-any-return]
 
     @property
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI datasets module is experimental, "
-        "and may change in future versions."
-    )
     def datasets(self) -> "datasets_module.Datasets":
         if self._datasets is None:
             self._datasets = importlib.import_module(
@@ -430,10 +410,6 @@ def feedback_entries(self) -> "feedback_entries_module.FeedbackEntries":
         return self._feedback_entries.FeedbackEntries(self._api_client)  # type: ignore[no-any-return]
 
     @property
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI rag module is experimental, "
-        "and may change in future versions."
-    )
     def rag(self) -> "rag_module.Rag":
         if self._rag is None:
             self._rag = importlib.import_module(
@@ -443,9 +419,6 @@ def rag(self) -> "rag_module.Rag":
         return self._rag.Rag(self._api_client)  # type: ignore[no-any-return]
 
     @property
-    @_common.experimental_warning(
-        "The Model Garden module is experimental, and may change in future " "versions."
-    )
     def model_garden(self) -> "model_garden_module.ModelGarden":
         if self._model_garden is None:
             self._model_garden = importlib.import_module(

agentplatform/_genai/evals.py

@@ -2592,10 +2592,6 @@ def generate_rubrics(
         )
         return types.EvaluationDataset(eval_dataset_df=prompts_with_rubrics)
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.get_evaluation_run module is experimental, "
-        "and may change in future versions."
-    )
     def get_evaluation_run(
         self,
         *,
@@ -2633,10 +2629,6 @@ def get_evaluation_run(
             object.__setattr__(result, "_eval_item_map", eval_item_map)
         return result
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.create_evaluation_run module is experimental, "
-        "and may change in future versions."
-    )
     def create_evaluation_run(
         self,
         *,
@@ -2815,10 +2807,6 @@ def create_evaluation_run(
             config=config,
         )
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.get_evaluation_set method is experimental, "
-        "and may change in future versions."
-    )
     def get_evaluation_set(
         self,
         *,
@@ -2843,10 +2831,6 @@ def get_evaluation_set(
             name = name.split("/")[-1]
         return self._get_evaluation_set(name=name, config=config)
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.get_evaluation_item method is experimental, "
-        "and may change in future versions."
-    )
     def get_evaluation_item(
         self,
         *,
@@ -2889,10 +2873,6 @@ def get_evaluation_item(
             )
         return result
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.create_evaluation_item module is experimental, "
-        "and may change in future versions."
-    )
     def create_evaluation_item(
         self,
         *,
@@ -2920,10 +2900,6 @@ def create_evaluation_item(
             config=config,
         )
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.create_evaluation_set module is experimental, "
-        "and may change in future versions."
-    )
     def create_evaluation_set(
         self,
         *,
@@ -2949,10 +2925,6 @@ def create_evaluation_set(
             config=config,
         )
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.generate_conversation_scenarios module is experimental, "
-        "and may change in future versions."
-    )
     def generate_conversation_scenarios(
         self,
         *,
@@ -2986,10 +2958,6 @@ def generate_conversation_scenarios(
         )
         return _evals_utils._postprocess_user_scenarios_response(response)
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.generate_loss_clusters module is experimental, "
-        "and may change in future versions."
-    )
     def generate_loss_clusters(
         self,
         *,
@@ -3067,10 +3035,6 @@ def generate_loss_clusters(
         )
         return completed.response
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.create_evaluation_metric method is experimental, "
-        "and may change in future versions."
-    )
     def create_evaluation_metric(
         self,
         *,
@@ -3104,10 +3068,6 @@ def create_evaluation_metric(
         # result.name is Optional[str], but we know it's always returned on creation
         return cast(str, result.name)
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.get_evaluation_metric module is experimental, "
-        "and may change in future versions."
-    )
     def get_evaluation_metric(
         self,
         *,
@@ -3120,10 +3080,6 @@ def get_evaluation_metric(
             config=config,
         )
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.list_evaluation_metrics module is experimental, "
-        "and may change in future versions."
-    )
     def list_evaluation_metrics(
         self,
         *,
@@ -3162,10 +3118,6 @@ def list_evaluation_metrics(
             config=config,
         )
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.delete_evaluation_metric method is experimental, "
-        "and may change in future versions."
-    )
     def delete_evaluation_metric(
         self,
         *,
@@ -4395,10 +4347,6 @@ async def evaluate_instances(
 
         return result
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.get_evaluation_run module is experimental, "
-        "and may change in future versions."
-    )
     async def get_evaluation_run(
         self,
         *,
@@ -4439,10 +4387,6 @@ async def get_evaluation_run(
 
         return result
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.create_evaluation_run module is experimental, "
-        "and may change in future versions."
-    )
     async def create_evaluation_run(
         self,
         *,
@@ -4624,10 +4568,6 @@ async def create_evaluation_run(
 
         return result
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.get_evaluation_set method is experimental, "
-        "and may change in future versions."
-    )
     async def get_evaluation_set(
         self,
         *,
@@ -4653,10 +4593,6 @@ async def get_evaluation_set(
 
         return result
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.get_evaluation_item method is experimental, "
-        "and may change in future versions."
-    )
     async def get_evaluation_item(
         self,
         *,
@@ -4700,10 +4636,6 @@ async def get_evaluation_item(
 
         return result
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.create_evaluation_item module is experimental, "
-        "and may change in future versions."
-    )
     async def create_evaluation_item(
         self,
         *,
@@ -4732,10 +4664,6 @@ async def create_evaluation_item(
         )
         return result
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.create_evaluation_set module is experimental, "
-        "and may change in future versions."
-    )
     async def create_evaluation_set(
         self,
         *,
@@ -4762,10 +4690,6 @@ async def create_evaluation_set(
         )
         return result
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.generate_conversation_scenarios module is experimental, "
-        "and may change in future versions."
-    )
     async def generate_conversation_scenarios(
         self,
         *,
@@ -4799,10 +4723,6 @@ async def generate_conversation_scenarios(
         )
         return _evals_utils._postprocess_user_scenarios_response(response)
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.generate_loss_clusters module is experimental, "
-        "and may change in future versions."
-    )
     async def generate_loss_clusters(
         self,
         *,
@@ -4880,10 +4800,6 @@ async def generate_loss_clusters(
         )
         return completed.response
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.create_evaluation_metric module is experimental, "
-        "and may change in future versions."
-    )
     async def create_evaluation_metric(
         self,
         *,
@@ -4914,10 +4830,6 @@ async def create_evaluation_metric(
         )
         return cast(str, result.name)
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.get_evaluation_metric module is experimental, "
-        "and may change in future versions."
-    )
     async def get_evaluation_metric(
         self,
         *,
@@ -4930,10 +4842,6 @@ async def get_evaluation_metric(
             config=config,
         )
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.list_evaluation_metrics module is experimental, "
-        "and may change in future versions."
-    )
     async def list_evaluation_metrics(
         self,
         *,
@@ -4972,10 +4880,6 @@ async def list_evaluation_metrics(
             config=config,
         )
 
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI evals.delete_evaluation_metric method is experimental, "
-        "and may change in future versions."
-    )
     async def delete_evaluation_metric(
         self,
         *,

agentplatform/_genai/live.py

@@ -22,7 +22,6 @@
 from types import ModuleType
 
 from google.genai import _api_module
-from google.genai import _common
 from google.genai._api_client import BaseApiClient
 
 logger = logging.getLogger("google_genai.live")
@@ -41,10 +40,6 @@ def __init__(self, api_client: BaseApiClient):
         self._agent_engines: Optional[ModuleType] = None
 
     @property
-    @_common.experimental_warning(
-        "The Vertex SDK GenAI agent engines module is experimental, "
-        "and may change in future versions."
-    )
     def agent_engines(self) -> "live_agent_engines_module.AsyncLiveAgentEngines":
         if self._agent_engines is None:
             try: