ai

type Analysis struct {
	Model     string    `json:"model"` // e.g. "openai:gpt-4o-mini"
	CreatedAt time.Time `json:"created_at"`
	// Rating is the model's high-level grade of the shot. Parsed out of
	// the first line of the LLM response and stripped from Summary before
	// markdown rendering. nil when the model didn't emit one in the
	// expected format — the UI just hides the badge.
	Rating *Rating `json:"rating,omitempty"`
	// Markdown summary (2-5 short paragraphs) suitable for rendering directly.
	Summary string `json:"summary"`
	// Extracted numeric metrics the UI can render as stat tiles.
	Metrics Metrics `json:"metrics"`
	// Usage is rough input/output byte accounting for the usage ledger.
	// Not part of the cached analysis payload the UI cares about, so we
	// keep it out of the JSON.
	Usage CallUsage `json:"-"`
}

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

type AnthropicConfig struct {
	APIKey   string // required
	Model    string // default "claude-3-5-haiku-latest"
	Endpoint string
	Version  string // x-api-key header version, default "2023-06-01"
	Timeout  time.Duration
}

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

type BeanInfo struct {
	Name       string
	Roaster    string
	Origin     string
	Process    string
	RoastLevel string
	RoastDate  string // ISO yyyy-mm-dd; empty if unknown
	Notes      string
}

type CallUsage struct {
	InputTokens  int64
	OutputTokens int64
	DurationMs   int64
}

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

type CoachInput struct {
	Shot       ShotInput
	ShotRating *int          // 1..5 or nil (user's own rating)
	ShotNote   string        // user's own note
	Siblings   []ShotSummary // recent shots with the same profile, newest first
}

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

type CompareInput struct {
	A       ShotInput
	B       ShotInput
	ARating *int
	BRating *int
	ANote   string
	BNote   string
}

type Comparison struct {
	Model     string    `json:"model"`
	CreatedAt time.Time `json:"created_at"`
	Markdown  string    `json:"markdown"`
	Usage     CallUsage `json:"-"`
}

type CostBreak struct {
	Calls        int     `json:"calls"`
	InputTokens  int64   `json:"input_tokens"`
	OutputTokens int64   `json:"output_tokens"`
	CostUSD      float64 `json:"cost_usd"`
	LastUsedUnix int64   `json:"last_used_unix,omitempty"`
}

type ExtractBeanRequest struct {
	APIKey string
	Model  string // e.g. "gpt-4o-mini" — must be a vision-capable OpenAI model
	Image  []byte
	MIME   string // e.g. "image/jpeg", "image/png", "image/webp"
}

type ExtractBeanRequestGemini struct {
	APIKey   string
	Model    string // e.g. "gemini-2.5-flash" — any multimodal Gemini works
	Image    []byte
	MIME     string
	Endpoint string // optional override; defaults to the public v1beta endpoint
}

type ExtractBeanResponse struct {
	Bean  ExtractedBean
	Usage TokenUsage
}

type ExtractedBean struct {
	Name       string `json:"name"`
	Roaster    string `json:"roaster"`
	Origin     string `json:"origin"`
	Process    string `json:"process"`
	RoastLevel string `json:"roast_level"`
	RoastDate  string `json:"roast_date"` // ISO yyyy-mm-dd
	Notes      string `json:"notes"`
	// Confidence is the model's self-reported confidence on a 0..1
	// scale. Handy for the UI to flag low-quality reads ("we couldn't
	// read much — please double-check").
	Confidence float64 `json:"confidence"`
}

type GeminiConfig struct {
	APIKey   string // required
	Model    string // default "gemini-1.5-flash"
	Endpoint string // v1beta base URL; default official endpoint
	Timeout  time.Duration
}

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

type GenerateImageRequest struct {
	// APIKey is required.
	APIKey string
	// Model defaults to "gemini-2.5-flash-image-preview" (Nano Banana).
	// The preview family returns one or more inline_data parts.
	Model string
	// Prompt is the human-readable instruction.
	Prompt string
	// Endpoint lets tests inject a fake server. Defaults to the official
	// v1beta base URL.
	Endpoint string
	// Timeout bounds the entire HTTP round-trip (including image encoding
	// on the server). Image generation is slower than text; 90s is safe.
	Timeout time.Duration
}

type GeneratedImage struct {
	MimeType string
	Data     []byte
}

type Metrics struct {
	Duration       float64 `json:"duration_s"`
	PreinfusionEnd float64 `json:"preinfusion_end_s,omitempty"`
	PeakPressure   float64 `json:"peak_pressure_bar"`
	AvgPressure    float64 `json:"avg_pressure_bar"`
	PeakFlow       float64 `json:"peak_flow_mls"`
	AvgFlow        float64 `json:"avg_flow_mls"`
	FinalWeight    float64 `json:"final_weight_g"`
	FirstDripAt    float64 `json:"first_drip_s,omitempty"`
}

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

type OpenAIConfig struct {
	APIKey   string // required
	Model    string // default "gpt-4o-mini"
	Endpoint string // default official OpenAI chat endpoint
	Timeout  time.Duration
}

type OpenAIImageRequest struct {
	APIKey  string
	Model   string // default "gpt-image-1"
	Prompt  string
	Size    string // e.g. "1024x1024" (default), "1024x1536", "1536x1024"
	Base    string // default https://api.openai.com/v1
	Timeout time.Duration
}

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

type ProfileNameInput struct {
	Profile     json.RawMessage
	CurrentName string
}

type ProfileNameSuggestion struct {
	Model     string    `json:"model"`
	CreatedAt time.Time `json:"created_at"`
	Name      string    `json:"name"`
	Reason    string    `json:"reason"`
	Usage     CallUsage `json:"-"`
}

type Provider interface {
	// Complete sends a system+user prompt pair and returns the assistant
	// text along with real token usage parsed from the provider response.
	// Implementations MUST return usage whenever the API gives it to them;
	// a zero-valued TokenUsage is only acceptable when the upstream
	// response omitted the counts (fall back to zeros, never estimate).
	Complete(ctx context.Context, system, user string) (string, TokenUsage, error)
	// Name returns a short identifier (e.g. "openai:gpt-4o-mini") used for
	// cache keying so a change of model invalidates old cached analyses.
	Name() string
}

type Rating struct {
	Score int    `json:"score"` // 0..10 inclusive
	Label string `json:"label,omitempty"`
}

type Record struct {
	Time         time.Time
	Provider     string // openai, anthropic, gemini
	Model        string // gpt-4o-mini, claude-haiku-4-5, ...
	Feature      string // analyze, coach, compare, digest, ask, name, transcribe, image
	InputTokens  int64
	OutputTokens int64
	DurationMs   int64
	ShotID       string
	OK           bool
	Err          string
}

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

type ShotInput struct {
	Name        string
	ProfileName string
	Samples     json.RawMessage
	Profile     json.RawMessage
	// Bean describes the bag the shot was pulled with (optional). When
	// set, the analyzer surfaces it in the user prompt so the LLM can
	// factor origin / roast age / process into its critique instead of
	// guessing from numbers alone.
	Bean *BeanInfo
	// Grind is the user's grinder setting for this shot (free-form
	// label, e.g. "2.8" or "12 clicks"). Empty = not recorded.
	Grind string
	// GrindRPM is the variable-speed grinder RPM for this shot. Nil =
	// not recorded / not applicable to this grinder.
	GrindRPM *float64
}

type ShotSummary struct {
	Name         string  `json:"name"`
	TimeISO      string  `json:"time_iso"`
	Duration     float64 `json:"duration_s"`
	PeakPressure float64 `json:"peak_pressure_bar"`
	AvgPressure  float64 `json:"avg_pressure_bar"`
	PeakFlow     float64 `json:"peak_flow_mls"`
	FinalWeight  float64 `json:"final_weight_g"`
	FirstDripAt  float64 `json:"first_drip_s,omitempty"`
	Rating       *int    `json:"user_rating,omitempty"`
	Note         string  `json:"user_note,omitempty"`
}

type Suggestion struct {
	Model      string    `json:"model"`
	CreatedAt  time.Time `json:"created_at"`
	Change     string    `json:"change"`            // short imperative, e.g. "Grind 2 notches finer"
	Rationale  string    `json:"rationale"`         // 1-2 sentences citing the numbers
	VarKey     string    `json:"var_key,omitempty"` // profile variable key or ""
	Before     *float64  `json:"before,omitempty"`  // current profile value
	After      *float64  `json:"after,omitempty"`   // proposed new value
	Confidence string    `json:"confidence"`        // "low"|"medium"|"high"
	Usage      CallUsage `json:"-"`
}

type TokenUsage struct {
	InputTokens  int64 `json:"input_tokens"`
	OutputTokens int64 `json:"output_tokens"`
}

type TranscribeGeminiRequest struct {
	APIKey   string
	Model    string        // default "gemini-2.5-flash"
	Audio    []byte        // raw audio bytes (20 MB inline cap)
	MIME     string        // e.g. "audio/webm", "audio/mp4", "audio/ogg"
	Endpoint string        // default https://generativelanguage.googleapis.com/v1beta
	Timeout  time.Duration // default 2m
	// Prompt overrides the default "transcribe this audio" instruction.
	Prompt string
}

type TranscribeOpenAIRequest struct {
	APIKey  string
	Model   string        // default "whisper-1"
	Audio   []byte        // raw audio bytes
	MIME    string        // e.g. "audio/webm", "audio/mp4"
	Base    string        // default https://api.openai.com/v1
	Timeout time.Duration // default 2m
	// Language is an optional ISO-639-1 hint (e.g. "en"). Leave empty to
	// let Whisper auto-detect.
	Language string
	// Prompt is an optional short hint that nudges the decoder (helpful
	// for domain words like "profile", "preinfusion"). Leave empty for
	// generic transcription.
	Prompt string
}

type Transcription struct {
	Text     string `json:"text"`
	Language string `json:"language,omitempty"`
}

type UsageSummary struct {
	Since        time.Time            `json:"since"`
	TotalCalls   int                  `json:"total_calls"`
	TotalCostUSD float64              `json:"total_cost_usd"`
	ByProvider   map[string]CostBreak `json:"by_provider"`
	ByFeature    map[string]CostBreak `json:"by_feature"`
	ByModel      map[string]CostBreak `json:"by_model"`
	Recent       []Record             `json:"recent"` // newest first
}