ai

type CommitSummarizer interface {
	// GenerateSuggestions generates a list of commit message suggestions.
	GenerateSuggestions(ctx context.Context, info models.CommitInfo, count int) ([]models.CommitSuggestion, error)
}

type ConfirmationCallback func(result ConfirmationResult) (choice string, proceed bool)

type ConfirmationResult struct {
	EstimatedCost  float64
	InputTokens    int
	OutputTokens   int
	SuggestedModel string
	CurrentModel   string
	RationaleKey   string
}

type CostAwareAIProvider interface {
	// CountTokens counts the tokens of a prompt without making the actual model call.
	// This allows estimating the cost before executing the generation.
	CountTokens(ctx context.Context, prompt string) (int, error)

	// GetModelName returns the name of the current model (e.g.: "gemini-2.5-flash")
	GetModelName() string

	// GetProviderName returns the name of the provider (e.g.: "gemini", "openai", "anthropic")
	GetProviderName() string
}

type CostAwareWrapper struct {
	// contains filtered or unexported fields
}

type GenerateFunc func(ctx context.Context, model string, prompt string) (interface{}, *models.TokenUsage, error)

type IssueContentGenerator interface {
	// GenerateIssueContent generates the title, description, and labels of an issue
	// based on the context provided in the request.
	GenerateIssueContent(ctx context.Context, request models.IssueGenerationRequest) (*models.IssueGenerationResult, error)
}

type PRSummarizer interface {
	// GeneratePRSummary generates a summary of a Pull Request given a prompt.
	GeneratePRSummary(ctx context.Context, prompt string, availableLabels []string) (models.PRSummary, error)
}

type PromptData struct {
	Count           int
	Files           string
	Diff            string
	Ticket          string
	History         string
	Instructions    string
	IssueNumber     int
	RelatedIssues   string
	IssueInfo       string
	RepoOwner       string
	RepoName        string
	PreviousVersion string
	CurrentVersion  string
	LatestVersion   string
	ReleaseDate     string
	Changelog       string
	PRContent       string
	TechnicalInfo   string
}

type ReleaseNotesGenerator interface {
	GenerateNotes(ctx context.Context, release *models.Release) (*models.ReleaseNotes, error)
}

type TokenCounter interface {
	CountTokens(ctx context.Context, content string) (int, error)
}

type WrapperConfig struct {
	Provider              CostAwareAIProvider
	BudgetDaily           float64
	EstimatedOutputTokens int
	SkipConfirmation      bool
	OnConfirmation        ConfirmationCallback
}