blob: 7dfb39759ebc6556689d7cc4aa9334a6762a6c08 [file] [edit]
// Copyright 2026 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// Code generated by the Google Gen AI SDK generator DO NOT EDIT.
package agentplatform
import (
"context"
"fmt"
"net/http"
"reflect"
"cloud.google.com/go/agentplatform/types"
"google.golang.org/genai"
)
func createAgentEngineSandboxConfigToVertex(fromObject map[string]any, parentObject map[string]any, rootObject map[string]any) (toObject map[string]any, err error) {
toObject = make(map[string]any)
fromDisplayName := genai.InternalGetValueByPath(fromObject, []string{"displayName"})
if fromDisplayName != nil {
genai.InternalSetValueByPath(parentObject, []string{"displayName"}, fromDisplayName)
}
fromDescription := genai.InternalGetValueByPath(fromObject, []string{"description"})
if fromDescription != nil {
genai.InternalSetValueByPath(parentObject, []string{"description"}, fromDescription)
}
fromTtl := genai.InternalGetValueByPath(fromObject, []string{"ttl"})
if fromTtl != nil {
genai.InternalSetValueByPath(parentObject, []string{"ttl"}, fromTtl)
}
fromSandboxEnvironmentTemplate := genai.InternalGetValueByPath(fromObject, []string{"sandboxEnvironmentTemplate"})
if fromSandboxEnvironmentTemplate != nil {
genai.InternalSetValueByPath(parentObject, []string{"sandboxEnvironmentTemplate"}, fromSandboxEnvironmentTemplate)
}
fromSandboxEnvironmentSnapshot := genai.InternalGetValueByPath(fromObject, []string{"sandboxEnvironmentSnapshot"})
if fromSandboxEnvironmentSnapshot != nil {
genai.InternalSetValueByPath(parentObject, []string{"sandboxEnvironmentSnapshot"}, fromSandboxEnvironmentSnapshot)
}
fromOwner := genai.InternalGetValueByPath(fromObject, []string{"owner"})
if fromOwner != nil {
genai.InternalSetValueByPath(parentObject, []string{"owner"}, fromOwner)
}
return toObject, nil
}
func createAgentEngineSandboxRequestParametersToVertex(fromObject map[string]any, parentObject map[string]any, rootObject map[string]any) (toObject map[string]any, err error) {
toObject = make(map[string]any)
fromName := genai.InternalGetValueByPath(fromObject, []string{"name"})
if fromName != nil {
genai.InternalSetValueByPath(toObject, []string{"_url", "name"}, fromName)
}
fromSpec := genai.InternalGetValueByPath(fromObject, []string{"spec"})
if fromSpec != nil {
genai.InternalSetValueByPath(toObject, []string{"spec"}, fromSpec)
}
fromConfig := genai.InternalGetValueByPath(fromObject, []string{"config"})
if fromConfig != nil {
_, err = createAgentEngineSandboxConfigToVertex(fromConfig.(map[string]any), toObject, rootObject)
if err != nil {
return nil, err
}
}
return toObject, nil
}
func deleteAgentEngineSandboxRequestParametersToVertex(fromObject map[string]any, parentObject map[string]any, rootObject map[string]any) (toObject map[string]any, err error) {
toObject = make(map[string]any)
fromName := genai.InternalGetValueByPath(fromObject, []string{"name"})
if fromName != nil {
genai.InternalSetValueByPath(toObject, []string{"_url", "name"}, fromName)
}
return toObject, nil
}
func executeCodeAgentEngineSandboxRequestParametersToVertex(fromObject map[string]any, parentObject map[string]any, rootObject map[string]any) (toObject map[string]any, err error) {
toObject = make(map[string]any)
fromName := genai.InternalGetValueByPath(fromObject, []string{"name"})
if fromName != nil {
genai.InternalSetValueByPath(toObject, []string{"_url", "name"}, fromName)
}
fromInputs := genai.InternalGetValueByPath(fromObject, []string{"inputs"})
if fromInputs != nil {
genai.InternalSetValueByPath(toObject, []string{"inputs"}, fromInputs)
}
return toObject, nil
}
func getAgentEngineSandboxOperationParametersToVertex(fromObject map[string]any, parentObject map[string]any, rootObject map[string]any) (toObject map[string]any, err error) {
toObject = make(map[string]any)
fromOperationName := genai.InternalGetValueByPath(fromObject, []string{"operationName"})
if fromOperationName != nil {
genai.InternalSetValueByPath(toObject, []string{"_url", "operationName"}, fromOperationName)
}
return toObject, nil
}
func getAgentEngineSandboxRequestParametersToVertex(fromObject map[string]any, parentObject map[string]any, rootObject map[string]any) (toObject map[string]any, err error) {
toObject = make(map[string]any)
fromName := genai.InternalGetValueByPath(fromObject, []string{"name"})
if fromName != nil {
genai.InternalSetValueByPath(toObject, []string{"_url", "name"}, fromName)
}
return toObject, nil
}
func listAgentEngineSandboxesConfigToVertex(fromObject map[string]any, parentObject map[string]any, rootObject map[string]any) (toObject map[string]any, err error) {
toObject = make(map[string]any)
fromPageSize := genai.InternalGetValueByPath(fromObject, []string{"pageSize"})
if fromPageSize != nil {
genai.InternalSetValueByPath(parentObject, []string{"_query", "pageSize"}, fromPageSize)
}
fromPageToken := genai.InternalGetValueByPath(fromObject, []string{"pageToken"})
if fromPageToken != nil {
genai.InternalSetValueByPath(parentObject, []string{"_query", "pageToken"}, fromPageToken)
}
fromFilter := genai.InternalGetValueByPath(fromObject, []string{"filter"})
if fromFilter != nil {
genai.InternalSetValueByPath(parentObject, []string{"_query", "filter"}, fromFilter)
}
return toObject, nil
}
func listAgentEngineSandboxesRequestParametersToVertex(fromObject map[string]any, parentObject map[string]any, rootObject map[string]any) (toObject map[string]any, err error) {
toObject = make(map[string]any)
fromName := genai.InternalGetValueByPath(fromObject, []string{"name"})
if fromName != nil {
genai.InternalSetValueByPath(toObject, []string{"_url", "name"}, fromName)
}
fromConfig := genai.InternalGetValueByPath(fromObject, []string{"config"})
if fromConfig != nil {
_, err = listAgentEngineSandboxesConfigToVertex(fromConfig.(map[string]any), toObject, rootObject)
if err != nil {
return nil, err
}
}
return toObject, nil
}
type Sandboxes struct {
apiClient *genai.InternalAPIClient
}
func (m Sandboxes) create(ctx context.Context, name string, spec *types.SandboxEnvironmentSpec, config *types.CreateAgentEngineSandboxConfig) (*types.AgentEngineSandboxOperation, error) {
parameterMap := make(map[string]any)
kwargs := map[string]any{"name": name, "spec": spec, "config": config}
genai.InternalDeepMarshal(kwargs, &parameterMap)
var httpOptions *genai.HTTPOptions
if config == nil || config.HTTPOptions == nil {
httpOptions = &genai.HTTPOptions{}
} else {
httpOptions = config.HTTPOptions
}
if httpOptions.Headers == nil {
httpOptions.Headers = http.Header{}
}
var response = new(types.AgentEngineSandboxOperation)
var responseMap map[string]any
var toConverter func(map[string]any, map[string]any, map[string]any) (map[string]any, error)
if m.apiClient.ClientConfig().Backend == genai.BackendVertexAI {
toConverter = createAgentEngineSandboxRequestParametersToVertex
} else {
return nil, fmt.Errorf("method Create is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client. You can choose to use Gemini Enterprise Agent Platform by setting ClientConfig.Backend to BackendEnterprise.")
}
body, err := toConverter(parameterMap, nil, parameterMap)
if err != nil {
return nil, err
}
delete(body, "config")
var path string
var urlParams map[string]any
if _, ok := body["_url"]; ok {
urlParams = body["_url"].(map[string]any)
delete(body, "_url")
}
if m.apiClient.ClientConfig().Backend == genai.BackendVertexAI {
path, err = genai.InternalFormatMap("{name}/sandboxEnvironments", urlParams)
} else {
path, err = genai.InternalFormatMap("None", urlParams)
}
if err != nil {
return nil, fmt.Errorf("invalid url params: %#v.\n%w", urlParams, err)
}
if _, ok := body["_query"]; ok {
query, err := genai.InternalCreateURLQuery(body["_query"].(map[string]any))
if err != nil {
return nil, err
}
path += "?" + query
delete(body, "_query")
}
responseMap, err = genai.SendRequest(ctx, m.apiClient, path, http.MethodPost, body, httpOptions)
if err != nil {
return nil, err
}
err = genai.InternalMapToStruct(responseMap, response)
if err != nil {
return nil, err
}
if field, ok := reflect.TypeOf(response).Elem().FieldByName("SDKHTTPResponse"); ok {
{
if reflect.ValueOf(response).Elem().FieldByName("SDKHTTPResponse").IsValid() {
{
reflect.ValueOf(response).Elem().FieldByName("SDKHTTPResponse").Set(reflect.Zero(field.Type))
}
}
}
}
return response, nil
}
func (m Sandboxes) delete(ctx context.Context, name string, config *types.DeleteAgentEngineSandboxConfig) (*types.DeleteAgentEngineSandboxOperation, error) {
parameterMap := make(map[string]any)
kwargs := map[string]any{"name": name, "config": config}
genai.InternalDeepMarshal(kwargs, &parameterMap)
var httpOptions *genai.HTTPOptions
if config == nil || config.HTTPOptions == nil {
httpOptions = &genai.HTTPOptions{}
} else {
httpOptions = config.HTTPOptions
}
if httpOptions.Headers == nil {
httpOptions.Headers = http.Header{}
}
var response = new(types.DeleteAgentEngineSandboxOperation)
var responseMap map[string]any
var toConverter func(map[string]any, map[string]any, map[string]any) (map[string]any, error)
if m.apiClient.ClientConfig().Backend == genai.BackendVertexAI {
toConverter = deleteAgentEngineSandboxRequestParametersToVertex
} else {
return nil, fmt.Errorf("method Delete is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client. You can choose to use Gemini Enterprise Agent Platform by setting ClientConfig.Backend to BackendEnterprise.")
}
body, err := toConverter(parameterMap, nil, parameterMap)
if err != nil {
return nil, err
}
delete(body, "config")
var path string
var urlParams map[string]any
if _, ok := body["_url"]; ok {
urlParams = body["_url"].(map[string]any)
delete(body, "_url")
}
if m.apiClient.ClientConfig().Backend == genai.BackendVertexAI {
path, err = genai.InternalFormatMap("{name}", urlParams)
} else {
path, err = genai.InternalFormatMap("None", urlParams)
}
if err != nil {
return nil, fmt.Errorf("invalid url params: %#v.\n%w", urlParams, err)
}
if _, ok := body["_query"]; ok {
query, err := genai.InternalCreateURLQuery(body["_query"].(map[string]any))
if err != nil {
return nil, err
}
path += "?" + query
delete(body, "_query")
}
responseMap, err = genai.SendRequest(ctx, m.apiClient, path, http.MethodDelete, body, httpOptions)
if err != nil {
return nil, err
}
err = genai.InternalMapToStruct(responseMap, response)
if err != nil {
return nil, err
}
if field, ok := reflect.TypeOf(response).Elem().FieldByName("SDKHTTPResponse"); ok {
{
if reflect.ValueOf(response).Elem().FieldByName("SDKHTTPResponse").IsValid() {
{
reflect.ValueOf(response).Elem().FieldByName("SDKHTTPResponse").Set(reflect.Zero(field.Type))
}
}
}
}
return response, nil
}
func (m Sandboxes) executeCode(ctx context.Context, name string, inputs []*types.Chunk, config *types.ExecuteCodeAgentEngineSandboxConfig) (*types.ExecuteSandboxEnvironmentResponse, error) {
parameterMap := make(map[string]any)
kwargs := map[string]any{"name": name, "inputs": inputs, "config": config}
genai.InternalDeepMarshal(kwargs, &parameterMap)
var httpOptions *genai.HTTPOptions
if config == nil || config.HTTPOptions == nil {
httpOptions = &genai.HTTPOptions{}
} else {
httpOptions = config.HTTPOptions
}
if httpOptions.Headers == nil {
httpOptions.Headers = http.Header{}
}
var response = new(types.ExecuteSandboxEnvironmentResponse)
var responseMap map[string]any
var toConverter func(map[string]any, map[string]any, map[string]any) (map[string]any, error)
if m.apiClient.ClientConfig().Backend == genai.BackendVertexAI {
toConverter = executeCodeAgentEngineSandboxRequestParametersToVertex
} else {
return nil, fmt.Errorf("method ExecuteCode is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client. You can choose to use Gemini Enterprise Agent Platform by setting ClientConfig.Backend to BackendEnterprise.")
}
body, err := toConverter(parameterMap, nil, parameterMap)
if err != nil {
return nil, err
}
delete(body, "config")
var path string
var urlParams map[string]any
if _, ok := body["_url"]; ok {
urlParams = body["_url"].(map[string]any)
delete(body, "_url")
}
if m.apiClient.ClientConfig().Backend == genai.BackendVertexAI {
path, err = genai.InternalFormatMap("{name}/:execute", urlParams)
} else {
path, err = genai.InternalFormatMap("None", urlParams)
}
if err != nil {
return nil, fmt.Errorf("invalid url params: %#v.\n%w", urlParams, err)
}
if _, ok := body["_query"]; ok {
query, err := genai.InternalCreateURLQuery(body["_query"].(map[string]any))
if err != nil {
return nil, err
}
path += "?" + query
delete(body, "_query")
}
responseMap, err = genai.SendRequest(ctx, m.apiClient, path, http.MethodPost, body, httpOptions)
if err != nil {
return nil, err
}
err = genai.InternalMapToStruct(responseMap, response)
if err != nil {
return nil, err
}
if field, ok := reflect.TypeOf(response).Elem().FieldByName("SDKHTTPResponse"); ok {
{
if reflect.ValueOf(response).Elem().FieldByName("SDKHTTPResponse").IsValid() {
{
reflect.ValueOf(response).Elem().FieldByName("SDKHTTPResponse").Set(reflect.Zero(field.Type))
}
}
}
}
return response, nil
}
func (m Sandboxes) get(ctx context.Context, name string, config *types.GetAgentEngineSandboxConfig) (*types.SandboxEnvironment, error) {
parameterMap := make(map[string]any)
kwargs := map[string]any{"name": name, "config": config}
genai.InternalDeepMarshal(kwargs, &parameterMap)
var httpOptions *genai.HTTPOptions
if config == nil || config.HTTPOptions == nil {
httpOptions = &genai.HTTPOptions{}
} else {
httpOptions = config.HTTPOptions
}
if httpOptions.Headers == nil {
httpOptions.Headers = http.Header{}
}
var response = new(types.SandboxEnvironment)
var responseMap map[string]any
var toConverter func(map[string]any, map[string]any, map[string]any) (map[string]any, error)
if m.apiClient.ClientConfig().Backend == genai.BackendVertexAI {
toConverter = getAgentEngineSandboxRequestParametersToVertex
} else {
return nil, fmt.Errorf("method Get is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client. You can choose to use Gemini Enterprise Agent Platform by setting ClientConfig.Backend to BackendEnterprise.")
}
body, err := toConverter(parameterMap, nil, parameterMap)
if err != nil {
return nil, err
}
delete(body, "config")
var path string
var urlParams map[string]any
if _, ok := body["_url"]; ok {
urlParams = body["_url"].(map[string]any)
delete(body, "_url")
}
if m.apiClient.ClientConfig().Backend == genai.BackendVertexAI {
path, err = genai.InternalFormatMap("{name}", urlParams)
} else {
path, err = genai.InternalFormatMap("None", urlParams)
}
if err != nil {
return nil, fmt.Errorf("invalid url params: %#v.\n%w", urlParams, err)
}
if _, ok := body["_query"]; ok {
query, err := genai.InternalCreateURLQuery(body["_query"].(map[string]any))
if err != nil {
return nil, err
}
path += "?" + query
delete(body, "_query")
}
responseMap, err = genai.SendRequest(ctx, m.apiClient, path, http.MethodGet, body, httpOptions)
if err != nil {
return nil, err
}
err = genai.InternalMapToStruct(responseMap, response)
if err != nil {
return nil, err
}
if field, ok := reflect.TypeOf(response).Elem().FieldByName("SDKHTTPResponse"); ok {
{
if reflect.ValueOf(response).Elem().FieldByName("SDKHTTPResponse").IsValid() {
{
reflect.ValueOf(response).Elem().FieldByName("SDKHTTPResponse").Set(reflect.Zero(field.Type))
}
}
}
}
return response, nil
}
func (m Sandboxes) list(ctx context.Context, name string, config *types.ListAgentEngineSandboxesConfig) (*types.ListAgentEngineSandboxesResponse, error) {
parameterMap := make(map[string]any)
kwargs := map[string]any{"name": name, "config": config}
genai.InternalDeepMarshal(kwargs, &parameterMap)
var httpOptions *genai.HTTPOptions
if config == nil || config.HTTPOptions == nil {
httpOptions = &genai.HTTPOptions{}
} else {
httpOptions = config.HTTPOptions
}
if httpOptions.Headers == nil {
httpOptions.Headers = http.Header{}
}
var response = new(types.ListAgentEngineSandboxesResponse)
var responseMap map[string]any
var toConverter func(map[string]any, map[string]any, map[string]any) (map[string]any, error)
if m.apiClient.ClientConfig().Backend == genai.BackendVertexAI {
toConverter = listAgentEngineSandboxesRequestParametersToVertex
} else {
return nil, fmt.Errorf("method List is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client. You can choose to use Gemini Enterprise Agent Platform by setting ClientConfig.Backend to BackendEnterprise.")
}
body, err := toConverter(parameterMap, nil, parameterMap)
if err != nil {
return nil, err
}
delete(body, "config")
var path string
var urlParams map[string]any
if _, ok := body["_url"]; ok {
urlParams = body["_url"].(map[string]any)
delete(body, "_url")
}
if m.apiClient.ClientConfig().Backend == genai.BackendVertexAI {
path, err = genai.InternalFormatMap("{name}/sandboxEnvironments", urlParams)
} else {
path, err = genai.InternalFormatMap("None", urlParams)
}
if err != nil {
return nil, fmt.Errorf("invalid url params: %#v.\n%w", urlParams, err)
}
if _, ok := body["_query"]; ok {
query, err := genai.InternalCreateURLQuery(body["_query"].(map[string]any))
if err != nil {
return nil, err
}
path += "?" + query
delete(body, "_query")
}
responseMap, err = genai.SendRequest(ctx, m.apiClient, path, http.MethodGet, body, httpOptions)
if err != nil {
return nil, err
}
err = genai.InternalMapToStruct(responseMap, response)
if err != nil {
return nil, err
}
if field, ok := reflect.TypeOf(response).Elem().FieldByName("SDKHTTPResponse"); ok {
{
if reflect.ValueOf(response).Elem().FieldByName("SDKHTTPResponse").IsValid() {
{
reflect.ValueOf(response).Elem().FieldByName("SDKHTTPResponse").Set(reflect.Zero(field.Type))
}
}
}
}
return response, nil
}
func (m Sandboxes) getSandboxOperation(ctx context.Context, operationName string, config *types.GetAgentEngineOperationConfig) (*types.AgentEngineSandboxOperation, error) {
parameterMap := make(map[string]any)
kwargs := map[string]any{"operationName": operationName, "config": config}
genai.InternalDeepMarshal(kwargs, &parameterMap)
var httpOptions *genai.HTTPOptions
if config == nil || config.HTTPOptions == nil {
httpOptions = &genai.HTTPOptions{}
} else {
httpOptions = config.HTTPOptions
}
if httpOptions.Headers == nil {
httpOptions.Headers = http.Header{}
}
var response = new(types.AgentEngineSandboxOperation)
var responseMap map[string]any
var toConverter func(map[string]any, map[string]any, map[string]any) (map[string]any, error)
if m.apiClient.ClientConfig().Backend == genai.BackendVertexAI {
toConverter = getAgentEngineSandboxOperationParametersToVertex
} else {
return nil, fmt.Errorf("method GetSandboxOperation is only supported in the Gemini Enterprise Agent Platform (previously known as Vertex AI) client. You can choose to use Gemini Enterprise Agent Platform by setting ClientConfig.Backend to BackendEnterprise.")
}
body, err := toConverter(parameterMap, nil, parameterMap)
if err != nil {
return nil, err
}
delete(body, "config")
var path string
var urlParams map[string]any
if _, ok := body["_url"]; ok {
urlParams = body["_url"].(map[string]any)
delete(body, "_url")
}
if m.apiClient.ClientConfig().Backend == genai.BackendVertexAI {
path, err = genai.InternalFormatMap("{operationName}", urlParams)
} else {
path, err = genai.InternalFormatMap("None", urlParams)
}
if err != nil {
return nil, fmt.Errorf("invalid url params: %#v.\n%w", urlParams, err)
}
if _, ok := body["_query"]; ok {
query, err := genai.InternalCreateURLQuery(body["_query"].(map[string]any))
if err != nil {
return nil, err
}
path += "?" + query
delete(body, "_query")
}
responseMap, err = genai.SendRequest(ctx, m.apiClient, path, http.MethodGet, body, httpOptions)
if err != nil {
return nil, err
}
err = genai.InternalMapToStruct(responseMap, response)
if err != nil {
return nil, err
}
if field, ok := reflect.TypeOf(response).Elem().FieldByName("SDKHTTPResponse"); ok {
{
if reflect.ValueOf(response).Elem().FieldByName("SDKHTTPResponse").IsValid() {
{
reflect.ValueOf(response).Elem().FieldByName("SDKHTTPResponse").Set(reflect.Zero(field.Type))
}
}
}
}
return response, nil
}
// Create creates a sandbox for the given name, spec, and config, and returns the sandbox operation.
func (m Sandboxes) Create(ctx context.Context, name string, spec *types.SandboxEnvironmentSpec, config *types.CreateAgentEngineSandboxConfig) (*types.AgentEngineSandboxOperation, error) {
return m.create(ctx, name, spec, config)
}
// Delete deletes the sandbox with the specified name and returns the sandbox operation representing
// the deletion process.
func (m Sandboxes) Delete(ctx context.Context, name string, config *types.DeleteAgentEngineSandboxConfig) (*types.DeleteAgentEngineSandboxOperation, error) {
return m.delete(ctx, name, config)
}
// ExecuteCode executes the code for the given name, inputs, and config, and returns the response.
func (m Sandboxes) ExecuteCode(ctx context.Context, name string, inputs []*types.Chunk, config *types.ExecuteCodeAgentEngineSandboxConfig) (*types.ExecuteSandboxEnvironmentResponse, error) {
return m.executeCode(ctx, name, inputs, config)
}
// Get returns the sandbox with the specified name and returns the sandbox object representing the
// get sandbox process.
func (m Sandboxes) Get(ctx context.Context, name string, config *types.GetAgentEngineSandboxConfig) (*types.SandboxEnvironment, error) {
return m.get(ctx, name, config)
}
// List lists the sandboxes for the given name and config, and returns the response.
func (m Sandboxes) List(ctx context.Context, name string, config *types.ListAgentEngineSandboxesConfig) (*types.ListAgentEngineSandboxesResponse, error) {
return m.list(ctx, name, config)
}
// GetSandboxOperation returns the sandbox operation for the given operation name and config, and
// returns the sandbox operation.
func (m Sandboxes) GetSandboxOperation(ctx context.Context, operationName string, config *types.GetAgentEngineOperationConfig) (*types.AgentEngineSandboxOperation, error) {
return m.getSandboxOperation(ctx, operationName, config)
}