package main

import (
	"errors"
	"testing"

	"github.com/firebase/genkit/go/ai"
	"github.com/firebase/genkit/go/genkit"
	"github.com/firebase/genkit/go/plugins/googlegenai"
)

var SomethingErr = errors.New("something went wrong")

func TestHandleError(t *testing.T) {
	g := genkit.Init(t.Context(),
		genkit.WithPlugins(new(googlegenai.GoogleAI)),
		genkit.WithDefaultModel("googleai/gemini-2.5-pro"),
	)

	getWeather := genkit.DefineTool(g, "getWeather", `Fetches the real-time weather for a city`,
		func(ctx *ai.ToolContext, location struct {
			City string `json:"city"`
		},
		) (string, error) {
			return "", SomethingErr
		},
	)

	prompt := genkit.DefinePrompt(g, "weather",
		ai.WithTools(getWeather),
		ai.WithInputType(In{}),
		ai.WithOutputType(Out{}),
		ai.WithPrompt("{{text}}"),
	)

	_, err := prompt.Execute(t.Context(), ai.WithInput(In{Text: "What is the weather in Riga?"}))

	// fails with:
	// want 'something went wrong', got 'tool "getWeather" failed: error calling tool getWeather: something went wrong'
	// err is core.GenkitError without the error chain
	if !errors.Is(err, SomethingErr) {
		t.Fatalf("want '%s', got '%s'", SomethingErr, err)
	}

}