func JsonControlledGeneration()

in go/controlled_generation.go [12:55]

39 lines of code
3 McCabe index (conditional complexity)


func JsonControlledGeneration() (*genai.GenerateContentResponse, error) {
	// [START json_controlled_generation]
	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)
	}

	schema := &genai.Schema{
		Type: genai.TypeArray,
		Items: &genai.Schema{
			Type: genai.TypeObject,
			Properties: map[string]*genai.Schema{
				"recipe_name": {Type: genai.TypeString},
				"ingredients": {
					Type:  genai.TypeArray,
					Items: &genai.Schema{Type: genai.TypeString},
				},
			},
			Required: []string{"recipe_name"},
		},
	}

	config := &genai.GenerateContentConfig{
		ResponseMIMEType: "application/json",
		ResponseSchema:   schema,
	}

	response, err := client.Models.GenerateContent(
		ctx,
		"gemini-2.0-flash",
		genai.Text("List a few popular cookie recipes."),
		config,
	)
	if err != nil {
		log.Fatal(err)
	}
	printResponse(response)
	// [END json_controlled_generation]
	return response, err
}