From 2af35e87c8d8679933a8912f9107928906a97d1c 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 --- models.go | 9 +++++++++ types.go | 2 ++ 2 files changed, 11 insertions(+) diff --git a/models.go b/models.go index 754a5024..e66df2f8 100644 --- a/models.go +++ b/models.go @@ -4289,6 +4289,10 @@ func updateModelConfigToMldev(fromObject map[string]any, parentObject map[string InternalSetValueByPath(parentObject, []string{"defaultCheckpointId"}, fromDefaultCheckpointId) } + if InternalGetValueByPath(fromObject, []string{"updateMask"}) != nil { + return nil, fmt.Errorf("updateMask parameter is only supported in Gemini Enterprise Agent Platform mode, not in Gemini Developer API mode.") + } + return toObject, nil } @@ -4310,6 +4314,11 @@ func updateModelConfigToVertex(fromObject map[string]any, parentObject map[strin InternalSetValueByPath(parentObject, []string{"defaultCheckpointId"}, fromDefaultCheckpointId) } + fromUpdateMask := InternalGetValueByPath(fromObject, []string{"updateMask"}) + if fromUpdateMask != nil { + InternalSetValueByPath(parentObject, []string{"_query", "updateMask"}, fromUpdateMask) + } + return toObject, nil } diff --git a/types.go b/types.go index 186a0c6f..fd2c9247 100644 --- a/types.go +++ b/types.go @@ -4417,6 +4417,8 @@ type UpdateModelConfig struct { Description string `json:"description,omitempty"` // Optional. DefaultCheckpointID string `json:"defaultCheckpointId,omitempty"` + // Optional. + UpdateMask string `json:"updateMask,omitempty"` } // Configuration for deleting a tuned model.