models

type AgentType string

type Complexity string

type CorrectionEvent struct {
	ID           string       `json:"id"`
	IssueNumber  int          `json:"issue_number"`
	FailureClass FailureClass `json:"failure_class"`
	Action       string       `json:"action"`
	Scope        string       `json:"scope"`
	Reason       string       `json:"reason"`
	NewProvider  string       `json:"new_provider,omitempty"`
	Outcome      string       `json:"outcome"` // pending, applied, failed
	CreatedAt    time.Time    `json:"created_at"`
}

type CostRecord struct {
	ID           string    `json:"id"`
	SessionID    string    `json:"session_id"`
	Provider     string    `json:"provider"`
	Model        string    `json:"model"`
	InputTokens  int       `json:"input_tokens"`
	OutputTokens int       `json:"output_tokens"`
	CostUSD      float64   `json:"cost_usd"` // 0 for local models
	CreatedAt    time.Time `json:"created_at"`
}

type CostSummary struct {
	TotalInputTokens  int     `json:"total_input_tokens"`
	TotalOutputTokens int     `json:"total_output_tokens"`
	TotalCostUSD      float64 `json:"total_cost_usd"`
	RecordCount       int     `json:"record_count"`
}

type Dependency struct {
	Owner  string // empty for same-project
	Repo   string // empty for same-project
	Number int
}

type DependencyList []Dependency

type FailureClass string

type FailureEvent struct {
	ID            string        `json:"id"`
	IssueNumber   int           `json:"issue_number"`
	SessionID     string        `json:"session_id,omitempty"` // empty for dispatcher-level failures
	FailureClass  FailureClass  `json:"failure_class"`
	ErrorMessage  string        `json:"error_message"`
	AgentType     string        `json:"agent_type,omitempty"`
	Provider      string        `json:"provider,omitempty"`
	AttemptNumber int           `json:"attempt_number"` // which attempt (1st, 2nd, 3rd, etc.)
	Duration      time.Duration `json:"duration_ms,omitempty"`
	Timestamp     time.Time     `json:"timestamp"`

	// RCA fields — populated when a fix is recorded.
	RootCauseCategory string     `json:"root_cause_category,omitempty"`
	FixClassification string     `json:"fix_classification,omitempty"`
	PreventionMeasure string     `json:"prevention_measure,omitempty"`
	RecurrenceRisk    string     `json:"recurrence_risk,omitempty"`
	DetectionMethod   string     `json:"detection_method,omitempty"`
	TimeToDetectMin   *int       `json:"time_to_detect_min,omitempty"`
	LinkedIssues      string     `json:"linked_issues,omitempty"`
	Component         string     `json:"component,omitempty"`
	ResolvedAt        *time.Time `json:"resolved_at,omitempty"`
	// Status mirrors the session status for KPI queries (e.g. "resolved").
	Status string `json:"status,omitempty"`
}

type FailureSummary struct {
	TotalFailures  int                  `json:"total_failures"`
	ByClass        map[FailureClass]int `json:"by_class"`
	TopIssues      []IssueFailureCount  `json:"top_issues"`      // issues with most failures
	LoopingIssues  []IssueFailureCount  `json:"looping_issues"`  // issues with 5+ failures
	ProviderHealth map[string]int       `json:"provider_health"` // provider -> failure count
	Since          time.Time            `json:"since"`
}

type IssueFailureCount struct {
	IssueNumber int `json:"issue_number"`
	Count       int `json:"count"`
}

type IssueFrontmatter struct {
	SchemaVersion   string         `yaml:"schema_version"`
	Type            IssueType      `yaml:"type"`
	AgentType       AgentType      `yaml:"agent_type"`
	Priority        Priority       `yaml:"priority"`
	ParentIssue     int            `yaml:"parent_issue,omitempty"`
	DependsOn       DependencyList `yaml:"depends_on,omitempty"`
	SourceProject   string         `yaml:"source_project,omitempty"`
	EstimatedTokens int            `yaml:"estimated_tokens,omitempty"`
	ActualTokens    int            `yaml:"actual_tokens,omitempty"`
	ModelUsed       string         `yaml:"model_used,omitempty"`
	TimeoutMinutes  int            `yaml:"timeout_minutes,omitempty"`
}

type IssueType string

type KPIReport struct {
	WindowDays           int                       `json:"window_days"`
	FirstTimeFixRate     *float64                  `json:"first_time_fix_rate"`
	FixSuccessRate       *float64                  `json:"fix_success_rate"`
	RecurrenceRate       *float64                  `json:"recurrence_rate"`
	WorkaroundRate       *float64                  `json:"workaround_rate"`
	MeanTimeToFixHours   *float64                  `json:"mean_time_to_fix_hours"`
	PlanningGapRate      *float64                  `json:"planning_gap_rate"`
	UnknownRootCauseRate *float64                  `json:"unknown_root_cause_rate"`
	PreventionCoverage   *float64                  `json:"prevention_coverage"`
	RootCauseBreakdown   []RootCauseBreakdownEntry `json:"root_cause_breakdown"`
	SampleSize           int                       `json:"sample_size"`
}

type ParseResult struct {
	Frontmatter *IssueFrontmatter
	Body        string // markdown after the closing ---
}

type Priority string

type QualityRecommendation struct {
	ID            string    `json:"id"`
	Level         string    `json:"level"` // "info" | "warn" | "critical"
	Category      string    `json:"category"`
	Title         string    `json:"title"`
	Detail        string    `json:"detail"`
	AffectedCount int       `json:"affected_count"`
	WindowDays    int       `json:"window_days"`
	GeneratedAt   time.Time `json:"generated_at"`
}

type RootCauseBreakdownEntry struct {
	Category string  `json:"category"`
	Count    int     `json:"count"`
	Pct      float64 `json:"pct"`
}

type ScalingControl struct {
	Paused        bool
	ManualWorkers int
	SetAt         time.Time
	Note          string
}

type ScalingEvent struct {
	ID          string
	Timestamp   time.Time
	Action      string // "scale-up" | "scale-down" | "manual-override"
	FromWorkers int
	ToWorkers   int
	Reason      string
	Confidence  float64
}

type Session struct {
	ID               string        `json:"id"`
	IssueNumber      int           `json:"issue_number"`
	AgentType        string        `json:"agent_type"` // e.g., "code-gen", "qc", "research"
	Provider         string        `json:"provider"`   // e.g., "ollama", "claude", "openai"
	Model            string        `json:"model"`      // e.g., "qwen2.5-coder:7b"
	Status           SessionStatus `json:"status"`
	StartedAt        time.Time     `json:"started_at"`
	FinishedAt       *time.Time    `json:"finished_at,omitempty"`
	Error            string        `json:"error,omitempty"`
	EstimatedTimeout time.Duration `json:"estimated_timeout_ms,omitempty"` // timeout predicted at dispatch time
	PartialOutput    string        `json:"partial_output,omitempty"`       // last checkpoint of streaming output
	CheckpointHash   string        `json:"checkpoint_hash,omitempty"`      // SHA-256 of posted checkpoint for dedup
	CreatedAt        time.Time     `json:"created_at"`
	UpdatedAt        time.Time     `json:"updated_at"`
}

type SessionStatus string

type Status string

type TaskProfile struct {
	AgentType   string        `json:"agent_type"`
	Provider    string        `json:"provider"`
	AvgDuration time.Duration `json:"avg_duration"`
	P50Duration time.Duration `json:"p50_duration"`
	P90Duration time.Duration `json:"p90_duration"`
	SampleCount int           `json:"sample_count"`
	AvgTokens   int           `json:"avg_tokens"`
	UpdatedAt   time.Time     `json:"updated_at"`
}