From 32a9bcc8c58621f6de921947faa872b489c1d2fa Mon Sep 17 00:00:00 2001 From: Sara Robinson Date: Mon, 13 Jul 2026 07:35:38 -0700 Subject: [PATCH] feat: Add Models.get, Models.delete, Models.list and Models.UpdateTunedModel methods PiperOrigin-RevId: 947032107 --- google/genai/models.py | 13 +++++++++++++ google/genai/types.py | 4 ++++ 2 files changed, 17 insertions(+) diff --git a/google/genai/models.py b/google/genai/models.py index 7a35a8b6c..8c7509c9f 100644 --- a/google/genai/models.py +++ b/google/genai/models.py @@ -4548,6 +4548,12 @@ def _UpdateModelConfig_to_mldev( getv(from_object, ['default_checkpoint_id']), ) + if getv(from_object, ['update_mask']) is not None: + raise ValueError( + 'update_mask parameter is only supported in Gemini Enterprise Agent' + ' Platform mode, not in Gemini Developer API mode.' + ) + return to_object @@ -4571,6 +4577,13 @@ def _UpdateModelConfig_to_vertex( getv(from_object, ['default_checkpoint_id']), ) + if getv(from_object, ['update_mask']) is not None: + setv( + parent_object, + ['_query', 'updateMask'], + getv(from_object, ['update_mask']), + ) + return to_object diff --git a/google/genai/types.py b/google/genai/types.py index feffb5756..05872a344 100644 --- a/google/genai/types.py +++ b/google/genai/types.py @@ -10703,6 +10703,7 @@ class UpdateModelConfig(_common.BaseModel): display_name: Optional[str] = Field(default=None, description="""""") description: Optional[str] = Field(default=None, description="""""") default_checkpoint_id: Optional[str] = Field(default=None, description="""""") + update_mask: Optional[str] = Field(default=None, description="""""") class UpdateModelConfigDict(TypedDict, total=False): @@ -10720,6 +10721,9 @@ class UpdateModelConfigDict(TypedDict, total=False): default_checkpoint_id: Optional[str] """""" + update_mask: Optional[str] + """""" + UpdateModelConfigOrDict = Union[UpdateModelConfig, UpdateModelConfigDict]