testutil

type MockProvider struct {
	// NameFunc allows customizing the Name() return value
	NameFunc func() string

	// SupportsImagesFunc allows customizing the SupportsImages() return value
	SupportsImagesFunc func() bool

	// ChatWithToolsFunc allows customizing the ChatWithTools() behavior
	ChatWithToolsFunc func(ctx context.Context, systemPrompt string, history []api.Message, model string) (string, error)

	// ChatWithImageFunc allows customizing the ChatWithImage() behavior
	ChatWithImageFunc func(ctx context.Context, systemPrompt string, history []api.Message, userMessage string, image *api.ImageData, model string) (string, error)

	// CallCount tracks how many times each method was called
	CallCount struct {
		Name           int
		SupportsImages int
		ChatWithTools  int
		ChatWithImage  int
	}

	// LastCall stores the arguments of the last call to each method
	LastCall struct {
		ChatWithTools struct {
			SystemPrompt string
			History      []api.Message
			Model        string
		}
		ChatWithImage struct {
			SystemPrompt string
			History      []api.Message
			UserMessage  string
			Image        *api.ImageData
			Model        string
		}
	}
}