go/configure_model_parameters.go

package examples

import (
	"context"
	"os"
	"log"
	
	"google.golang.org/genai"
)

func ConfigureModelParameters() (*genai.GenerateContentResponse, error) {
	// [START configure_model_parameters]
	ctx := context.Background()
	client, err := genai.NewClient(ctx, &genai.ClientConfig{
		APIKey:  os.Getenv("GEMINI_API_KEY"),
		Backend: genai.BackendGeminiAPI,
	})
	if err != nil {
		log.Fatal(err)
	}

	// Create local variables for parameters.
	candidateCount := int32(1)
	maxOutputTokens := int32(20)
	temperature := float32(1.0)

	response, err := client.Models.GenerateContent(
		ctx,
		"gemini-2.0-flash",
		genai.Text("Tell me a story about a magic backpack."),
		&genai.GenerateContentConfig{
			CandidateCount:  candidateCount,
			StopSequences:   []string{"x"},
			MaxOutputTokens: maxOutputTokens,
			Temperature:     &temperature,
		},
	)
	if err != nil {
		log.Fatal(err)
	}

	printResponse(response)
	// [END configure_model_parameters]
	return response, err
}