package main

import (
	"context"
	"fmt"
	"log"

	"google.golang.org/genai"
)

func main() {
	ctx := context.Background()

	client, err := genai.NewClient(ctx, &genai.ClientConfig{
		Backend: genai.BackendGeminiAPI,
	})
	if err != nil {
		log.Fatalf("Error creating client: %v\n", err)
	}

	fridgeContentFunc := &genai.FunctionDeclaration{
		Name:        "fridgeContent",
		Description: "Get the current content of the fridge.",
	}

	modelCfg := &genai.GenerateContentConfig{
		Tools: []*genai.Tool{{
			FunctionDeclarations: []*genai.FunctionDeclaration{
				fridgeContentFunc,
			}},
		},
		ToolConfig: &genai.ToolConfig{
			FunctionCallingConfig: &genai.FunctionCallingConfig{
				Mode: genai.FunctionCallingConfigModeAuto,
			},
		},
	}

	chat, err := client.Chats.Create(ctx, "gemini-2.5-flash", modelCfg, []*genai.Content{})
	if err != nil {
		log.Fatalf("Error creating chat: %v\n", err)
	}

	nextParts := []*genai.Part{genai.NewPartFromText("What's in the fridge?")}
	for {
		resp, err := chat.Send(ctx, nextParts...)
		if err != nil {
			log.Fatalf("Error sending message: %v\n", err)
		}

		if len(resp.FunctionCalls()) == 0 {
			fmt.Println(resp.Text())
			break
		}

		nextParts = nextParts[:0] // reset
		for _, fc := range resp.FunctionCalls() {
			nextParts = append(nextParts, genai.NewPartFromFunctionResponse(
				fc.Name,
				map[string]any{"content": "pizza"},
			))
		}
	}
}