diff --git a/vertexai/_genai/a2a_task_events.py b/vertexai/_genai/a2a_task_events.py index 34e75338fe..7f026e7290 100644 --- a/vertexai/_genai/a2a_task_events.py +++ b/vertexai/_genai/a2a_task_events.py @@ -129,7 +129,9 @@ def append( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _AppendAgentEngineTaskEventRequestParameters_to_vertex( parameter_model @@ -216,7 +218,9 @@ def _list( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListAgentEngineTaskEventsRequestParameters_to_vertex( parameter_model @@ -328,7 +332,9 @@ async def append( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _AppendAgentEngineTaskEventRequestParameters_to_vertex( parameter_model @@ -417,7 +423,9 @@ async def _list( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListAgentEngineTaskEventsRequestParameters_to_vertex( parameter_model diff --git a/vertexai/_genai/a2a_tasks.py b/vertexai/_genai/a2a_tasks.py index 842513157e..aa92f6d9e6 100644 --- a/vertexai/_genai/a2a_tasks.py +++ b/vertexai/_genai/a2a_tasks.py @@ -166,7 +166,9 @@ def delete( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _DeleteAgentEngineTaskRequestParameters_to_vertex( parameter_model @@ -222,7 +224,9 @@ def get( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineTaskRequestParameters_to_vertex( parameter_model @@ -304,7 +308,9 @@ def _list( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListAgentEngineTasksRequestParameters_to_vertex( parameter_model @@ -390,7 +396,9 @@ def create( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateAgentEngineTaskRequestParameters_to_vertex( parameter_model @@ -517,7 +525,9 @@ async def delete( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _DeleteAgentEngineTaskRequestParameters_to_vertex( parameter_model @@ -573,7 +583,9 @@ async def get( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineTaskRequestParameters_to_vertex( parameter_model @@ -657,7 +669,9 @@ async def _list( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListAgentEngineTasksRequestParameters_to_vertex( parameter_model @@ -745,7 +759,9 @@ async def create( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateAgentEngineTaskRequestParameters_to_vertex( parameter_model diff --git a/vertexai/_genai/agent_engines.py b/vertexai/_genai/agent_engines.py index 6af7a6b59d..8aa7194e2e 100644 --- a/vertexai/_genai/agent_engines.py +++ b/vertexai/_genai/agent_engines.py @@ -776,7 +776,9 @@ def cancel_query_job( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CancelQueryJobAgentEngineRequestParameters_to_vertex( parameter_model @@ -848,7 +850,9 @@ def _check_query_job( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CheckQueryJobAgentEngineRequestParameters_to_vertex( parameter_model @@ -923,7 +927,9 @@ def _run_query_job( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _RunQueryJobAgentEngineRequestParameters_to_vertex( parameter_model @@ -994,7 +1000,9 @@ def _create( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateAgentEngineRequestParameters_to_vertex( parameter_model @@ -1084,7 +1092,9 @@ def _delete( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _DeleteAgentEngineRequestParameters_to_vertex( parameter_model @@ -1153,7 +1163,9 @@ def _get( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1222,7 +1234,9 @@ def _list( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListAgentEngineRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1291,7 +1305,9 @@ def _get_agent_operation( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineOperationParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1361,7 +1377,9 @@ def _query( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _QueryAgentEngineRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1428,7 +1446,9 @@ def _update( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _UpdateAgentEngineRequestParameters_to_vertex( parameter_model @@ -3125,7 +3145,9 @@ async def cancel_query_job( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CancelQueryJobAgentEngineRequestParameters_to_vertex( parameter_model @@ -3199,7 +3221,9 @@ async def _check_query_job( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CheckQueryJobAgentEngineRequestParameters_to_vertex( parameter_model @@ -3276,7 +3300,9 @@ async def _run_query_job( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _RunQueryJobAgentEngineRequestParameters_to_vertex( parameter_model @@ -3349,7 +3375,9 @@ async def _create( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateAgentEngineRequestParameters_to_vertex( parameter_model @@ -3441,7 +3469,9 @@ async def _delete( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _DeleteAgentEngineRequestParameters_to_vertex( parameter_model @@ -3512,7 +3542,9 @@ async def _get( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3583,7 +3615,9 @@ async def _list( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListAgentEngineRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3654,7 +3688,9 @@ async def _get_agent_operation( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineOperationParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3726,7 +3762,9 @@ async def _query( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _QueryAgentEngineRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3795,7 +3833,9 @@ async def _update( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _UpdateAgentEngineRequestParameters_to_vertex( parameter_model diff --git a/vertexai/_genai/datasets.py b/vertexai/_genai/datasets.py index e9febf01b0..c8a25b9501 100644 --- a/vertexai/_genai/datasets.py +++ b/vertexai/_genai/datasets.py @@ -268,7 +268,9 @@ def _assemble_multimodal_dataset( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _AssembleDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -358,7 +360,9 @@ def _assess_multimodal_dataset( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _AssessDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -438,7 +442,9 @@ def _create_multimodal_dataset( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateMultimodalDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -505,7 +511,9 @@ def _delete_multimodal_dataset( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _DeleteMultimodalDatasetRequestParameters_to_vertex( parameter_model @@ -577,7 +585,9 @@ def _get_multimodal_dataset( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetMultimodalDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -649,7 +659,9 @@ def _get_multimodal_dataset_operation( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetMultimodalDatasetOperationParameters_to_vertex( parameter_model @@ -719,7 +731,9 @@ def _list_multimodal_datasets( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListMultimodalDatasetsRequestParameters_to_vertex( parameter_model @@ -799,7 +813,9 @@ def _update_multimodal_dataset( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _UpdateMultimodalDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1476,7 +1492,9 @@ async def _assemble_multimodal_dataset( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _AssembleDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1568,7 +1586,9 @@ async def _assess_multimodal_dataset( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _AssessDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1650,7 +1670,9 @@ async def _create_multimodal_dataset( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateMultimodalDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1719,7 +1741,9 @@ async def _delete_multimodal_dataset( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _DeleteMultimodalDatasetRequestParameters_to_vertex( parameter_model @@ -1793,7 +1817,9 @@ async def _get_multimodal_dataset( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetMultimodalDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1867,7 +1893,9 @@ async def _get_multimodal_dataset_operation( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetMultimodalDatasetOperationParameters_to_vertex( parameter_model @@ -1939,7 +1967,9 @@ async def _list_multimodal_datasets( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListMultimodalDatasetsRequestParameters_to_vertex( parameter_model @@ -2021,7 +2051,9 @@ async def _update_multimodal_dataset( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _UpdateMultimodalDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") diff --git a/vertexai/_genai/evals.py b/vertexai/_genai/evals.py index 956e48cffc..6e0f1c7582 100644 --- a/vertexai/_genai/evals.py +++ b/vertexai/_genai/evals.py @@ -947,7 +947,9 @@ def _create_evaluation_item( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateEvaluationItemParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1021,7 +1023,9 @@ def _create_evaluation_metric( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateEvaluationMetricParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1106,7 +1110,9 @@ def _create_evaluation_run( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateEvaluationRunParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1181,7 +1187,9 @@ def _create_evaluation_set( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateEvaluationSetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1281,7 +1289,9 @@ def _evaluate_instances( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _EvaluateInstancesRequestParameters_to_vertex( parameter_model @@ -1361,7 +1371,9 @@ def _generate_user_scenarios( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GenerateUserScenariosParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1437,7 +1449,9 @@ def _generate_loss_clusters( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GenerateLossClustersParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1515,7 +1529,9 @@ def _generate_rubrics( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GenerateInstanceRubricsRequest_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1585,7 +1601,9 @@ def _get_evaluation_metric( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetEvaluationMetricParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1655,7 +1673,9 @@ def _get_evaluation_run( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetEvaluationRunParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1725,7 +1745,9 @@ def _get_evaluation_set( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetEvaluationSetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1792,7 +1814,9 @@ def _get_evaluation_item( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetEvaluationItemParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1858,7 +1882,9 @@ def _list_evaluation_metrics( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListEvaluationMetricsParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -2905,7 +2931,9 @@ async def _create_evaluation_item( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateEvaluationItemParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -2981,7 +3009,9 @@ async def _create_evaluation_metric( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateEvaluationMetricParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3068,7 +3098,9 @@ async def _create_evaluation_run( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateEvaluationRunParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3145,7 +3177,9 @@ async def _create_evaluation_set( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateEvaluationSetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3247,7 +3281,9 @@ async def _evaluate_instances( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _EvaluateInstancesRequestParameters_to_vertex( parameter_model @@ -3329,7 +3365,9 @@ async def _generate_user_scenarios( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GenerateUserScenariosParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3407,7 +3445,9 @@ async def _generate_loss_clusters( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GenerateLossClustersParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3487,7 +3527,9 @@ async def _generate_rubrics( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GenerateInstanceRubricsRequest_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3559,7 +3601,9 @@ async def _get_evaluation_metric( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetEvaluationMetricParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3631,7 +3675,9 @@ async def _get_evaluation_run( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetEvaluationRunParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3703,7 +3749,9 @@ async def _get_evaluation_set( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetEvaluationSetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3772,7 +3820,9 @@ async def _get_evaluation_item( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetEvaluationItemParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3840,7 +3890,9 @@ async def _list_evaluation_metrics( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListEvaluationMetricsParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") diff --git a/vertexai/_genai/memories.py b/vertexai/_genai/memories.py index f39a000160..f08758b365 100644 --- a/vertexai/_genai/memories.py +++ b/vertexai/_genai/memories.py @@ -552,7 +552,9 @@ def _create( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateAgentEngineMemoryRequestParameters_to_vertex( parameter_model @@ -632,7 +634,9 @@ def delete( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _DeleteAgentEngineMemoryRequestParameters_to_vertex( parameter_model @@ -718,7 +722,9 @@ def _generate( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GenerateAgentEngineMemoriesRequestParameters_to_vertex( parameter_model @@ -796,7 +802,9 @@ def get( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineMemoryRequestParameters_to_vertex( parameter_model @@ -880,7 +888,9 @@ def _ingest_events( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _IngestEventsRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -950,7 +960,9 @@ def _list( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListAgentEngineMemoryRequestParameters_to_vertex( parameter_model @@ -1018,7 +1030,9 @@ def _get_memory_operation( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineMemoryOperationParameters_to_vertex( parameter_model @@ -1086,7 +1100,9 @@ def _get_generate_memories_operation( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineGenerateMemoriesOperationParameters_to_vertex( parameter_model @@ -1168,7 +1184,9 @@ def _retrieve( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _RetrieveAgentEngineMemoriesRequestParameters_to_vertex( parameter_model @@ -1268,7 +1286,9 @@ def retrieve_profiles( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _RetrieveMemoryProfilesRequestParameters_to_vertex( parameter_model @@ -1342,7 +1362,9 @@ def _rollback( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _RollbackAgentEngineMemoryRequestParameters_to_vertex( parameter_model @@ -1418,7 +1440,9 @@ def _update( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _UpdateAgentEngineMemoryRequestParameters_to_vertex( parameter_model @@ -1496,7 +1520,9 @@ def _purge( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _PurgeAgentEngineMemoriesRequestParameters_to_vertex( parameter_model @@ -1962,7 +1988,9 @@ async def _create( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateAgentEngineMemoryRequestParameters_to_vertex( parameter_model @@ -2044,7 +2072,9 @@ async def delete( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _DeleteAgentEngineMemoryRequestParameters_to_vertex( parameter_model @@ -2132,7 +2162,9 @@ async def _generate( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GenerateAgentEngineMemoriesRequestParameters_to_vertex( parameter_model @@ -2212,7 +2244,9 @@ async def get( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineMemoryRequestParameters_to_vertex( parameter_model @@ -2298,7 +2332,9 @@ async def _ingest_events( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _IngestEventsRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -2370,7 +2406,9 @@ async def _list( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListAgentEngineMemoryRequestParameters_to_vertex( parameter_model @@ -2440,7 +2478,9 @@ async def _get_memory_operation( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineMemoryOperationParameters_to_vertex( parameter_model @@ -2510,7 +2550,9 @@ async def _get_generate_memories_operation( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineGenerateMemoriesOperationParameters_to_vertex( parameter_model @@ -2594,7 +2636,9 @@ async def _retrieve( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _RetrieveAgentEngineMemoriesRequestParameters_to_vertex( parameter_model @@ -2696,7 +2740,9 @@ async def retrieve_profiles( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _RetrieveMemoryProfilesRequestParameters_to_vertex( parameter_model @@ -2772,7 +2818,9 @@ async def _rollback( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _RollbackAgentEngineMemoryRequestParameters_to_vertex( parameter_model @@ -2850,7 +2898,9 @@ async def _update( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _UpdateAgentEngineMemoryRequestParameters_to_vertex( parameter_model @@ -2930,7 +2980,9 @@ async def _purge( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _PurgeAgentEngineMemoriesRequestParameters_to_vertex( parameter_model diff --git a/vertexai/_genai/memory_revisions.py b/vertexai/_genai/memory_revisions.py index c61e48acd3..caf93acf2e 100644 --- a/vertexai/_genai/memory_revisions.py +++ b/vertexai/_genai/memory_revisions.py @@ -108,7 +108,9 @@ def get( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineMemoryRevisionRequestParameters_to_vertex( parameter_model @@ -190,7 +192,9 @@ def _list( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListAgentEngineMemoryRevisionsRequestParameters_to_vertex( parameter_model @@ -300,7 +304,9 @@ async def get( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineMemoryRevisionRequestParameters_to_vertex( parameter_model @@ -384,7 +390,9 @@ async def _list( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListAgentEngineMemoryRevisionsRequestParameters_to_vertex( parameter_model diff --git a/vertexai/_genai/prompt_optimizer.py b/vertexai/_genai/prompt_optimizer.py index 565f45a061..d11abec3ae 100644 --- a/vertexai/_genai/prompt_optimizer.py +++ b/vertexai/_genai/prompt_optimizer.py @@ -221,7 +221,9 @@ def _optimize_prompt( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _OptimizeRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -291,7 +293,9 @@ def _create_custom_job_resource( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CustomJobParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -361,7 +365,9 @@ def _get_custom_job( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetCustomJobParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -625,7 +631,9 @@ async def _optimize_prompt( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _OptimizeRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -697,7 +705,9 @@ async def _create_custom_job_resource( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CustomJobParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -769,7 +779,9 @@ async def _get_custom_job( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetCustomJobParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") diff --git a/vertexai/_genai/prompts.py b/vertexai/_genai/prompts.py index 2bb549e44b..782aa2f3fa 100644 --- a/vertexai/_genai/prompts.py +++ b/vertexai/_genai/prompts.py @@ -482,7 +482,9 @@ def _create_dataset_resource( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -560,7 +562,9 @@ def _create_dataset_version_resource( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateDatasetVersionParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -630,7 +634,9 @@ def _get_dataset_resource( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -702,7 +708,9 @@ def _get_dataset_version_resource( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetDatasetVersionParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -776,7 +784,9 @@ def _get_dataset_operation( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetDatasetOperationParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -844,7 +854,9 @@ def _list_prompts( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListDatasetsRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -916,7 +928,9 @@ def _list_versions( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListDatasetVersionsRequestParameters_to_vertex( parameter_model @@ -983,7 +997,9 @@ def _delete_dataset( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _DeleteDatasetRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1051,7 +1067,9 @@ def _delete_dataset_version( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _DeletePromptVersionRequestParameters_to_vertex( parameter_model @@ -1127,7 +1145,9 @@ def _restore_version( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _RestoreVersionRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1211,7 +1231,9 @@ def _update_dataset_resource( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _UpdateDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1281,7 +1303,9 @@ def _create_custom_job_resource( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CustomJobParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1351,7 +1375,9 @@ def _get_custom_job( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetCustomJobParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -1424,7 +1450,9 @@ def _optimize( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _OptimizeRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -2426,7 +2454,9 @@ async def _create_dataset_resource( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -2506,7 +2536,9 @@ async def _create_dataset_version_resource( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateDatasetVersionParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -2578,7 +2610,9 @@ async def _get_dataset_resource( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -2652,7 +2686,9 @@ async def _get_dataset_version_resource( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetDatasetVersionParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -2728,7 +2764,9 @@ async def _get_dataset_operation( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetDatasetOperationParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -2798,7 +2836,9 @@ async def _list_prompts( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListDatasetsRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -2872,7 +2912,9 @@ async def _list_versions( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListDatasetVersionsRequestParameters_to_vertex( parameter_model @@ -2941,7 +2983,9 @@ async def _delete_dataset( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _DeleteDatasetRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3011,7 +3055,9 @@ async def _delete_dataset_version( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _DeletePromptVersionRequestParameters_to_vertex( parameter_model @@ -3089,7 +3135,9 @@ async def _restore_version( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _RestoreVersionRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3175,7 +3223,9 @@ async def _update_dataset_resource( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _UpdateDatasetParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3247,7 +3297,9 @@ async def _create_custom_job_resource( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CustomJobParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3319,7 +3371,9 @@ async def _get_custom_job( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetCustomJobParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") @@ -3394,7 +3448,9 @@ async def _optimize( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _OptimizeRequestParameters_to_vertex(parameter_model) request_url_dict = request_dict.get("_url") diff --git a/vertexai/_genai/sandboxes.py b/vertexai/_genai/sandboxes.py index e785da6726..565e1d80d8 100644 --- a/vertexai/_genai/sandboxes.py +++ b/vertexai/_genai/sandboxes.py @@ -182,7 +182,9 @@ def _create( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateAgentEngineSandboxRequestParameters_to_vertex( parameter_model @@ -260,7 +262,9 @@ def _delete( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _DeleteAgentEngineSandboxRequestParameters_to_vertex( parameter_model @@ -334,7 +338,9 @@ def _execute_code( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ExecuteCodeAgentEngineSandboxRequestParameters_to_vertex( parameter_model @@ -412,7 +418,9 @@ def _get( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineSandboxRequestParameters_to_vertex( parameter_model @@ -494,7 +502,9 @@ def _list( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListAgentEngineSandboxesRequestParameters_to_vertex( parameter_model @@ -562,7 +572,9 @@ def _get_sandbox_operation( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineSandboxOperationParameters_to_vertex( parameter_model @@ -982,7 +994,9 @@ async def _create( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateAgentEngineSandboxRequestParameters_to_vertex( parameter_model @@ -1062,7 +1076,9 @@ async def _delete( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _DeleteAgentEngineSandboxRequestParameters_to_vertex( parameter_model @@ -1138,7 +1154,9 @@ async def _execute_code( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ExecuteCodeAgentEngineSandboxRequestParameters_to_vertex( parameter_model @@ -1218,7 +1236,9 @@ async def _get( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineSandboxRequestParameters_to_vertex( parameter_model @@ -1302,7 +1322,9 @@ async def _list( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListAgentEngineSandboxesRequestParameters_to_vertex( parameter_model @@ -1372,7 +1394,9 @@ async def _get_sandbox_operation( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineSandboxOperationParameters_to_vertex( parameter_model diff --git a/vertexai/_genai/session_events.py b/vertexai/_genai/session_events.py index b36ef6e21c..a1f96e8762 100644 --- a/vertexai/_genai/session_events.py +++ b/vertexai/_genai/session_events.py @@ -159,7 +159,9 @@ def append( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _AppendAgentEngineSessionEventRequestParameters_to_vertex( parameter_model @@ -241,7 +243,9 @@ def _list( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListAgentEngineSessionEventsRequestParameters_to_vertex( parameter_model @@ -364,7 +368,9 @@ async def append( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _AppendAgentEngineSessionEventRequestParameters_to_vertex( parameter_model @@ -448,7 +454,9 @@ async def _list( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListAgentEngineSessionEventsRequestParameters_to_vertex( parameter_model diff --git a/vertexai/_genai/sessions.py b/vertexai/_genai/sessions.py index bc284fcd3f..43793c56ae 100644 --- a/vertexai/_genai/sessions.py +++ b/vertexai/_genai/sessions.py @@ -241,7 +241,9 @@ def _create( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateAgentEngineSessionRequestParameters_to_vertex( parameter_model @@ -323,7 +325,9 @@ def delete( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _DeleteAgentEngineSessionRequestParameters_to_vertex( parameter_model @@ -405,7 +409,9 @@ def get( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineSessionRequestParameters_to_vertex( parameter_model @@ -487,7 +493,9 @@ def _list( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListAgentEngineSessionsRequestParameters_to_vertex( parameter_model @@ -555,7 +563,9 @@ def _get_session_operation( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineSessionOperationParameters_to_vertex( parameter_model @@ -637,7 +647,9 @@ def _update( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _UpdateAgentEngineSessionRequestParameters_to_vertex( parameter_model @@ -816,7 +828,9 @@ async def _create( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _CreateAgentEngineSessionRequestParameters_to_vertex( parameter_model @@ -900,7 +914,9 @@ async def delete( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _DeleteAgentEngineSessionRequestParameters_to_vertex( parameter_model @@ -984,7 +1000,9 @@ async def get( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineSessionRequestParameters_to_vertex( parameter_model @@ -1068,7 +1086,9 @@ async def _list( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _ListAgentEngineSessionsRequestParameters_to_vertex( parameter_model @@ -1138,7 +1158,9 @@ async def _get_session_operation( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _GetAgentEngineSessionOperationParameters_to_vertex( parameter_model @@ -1222,7 +1244,9 @@ async def _update( request_url_dict: Optional[dict[str, str]] if not self._api_client.vertexai: - raise ValueError("This method is only supported in the Vertex AI client.") + raise ValueError( + "This method is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client." + ) else: request_dict = _UpdateAgentEngineSessionRequestParameters_to_vertex( parameter_model