aitriage

type AIConfigInfo struct {
	// Mode is the AI mode: platform, byok, agent, disabled
	Mode string `json:"mode"`
	// Provider is the LLM provider: claude, openai, gemini
	Provider string `json:"provider"`
	// Model is the model being used
	Model string `json:"model"`
	// IsEnabled indicates if AI triage is enabled for this tenant
	IsEnabled bool `json:"is_enabled"`
	// AutoTriageEnabled indicates if auto-triage is enabled
	AutoTriageEnabled bool `json:"auto_triage_enabled"`
	// AutoTriageSeverities is the list of severities for auto-triage
	AutoTriageSeverities []string `json:"auto_triage_severities,omitempty"`
	// MonthlyTokenLimit is the monthly token limit (0 = unlimited)
	MonthlyTokenLimit int `json:"monthly_token_limit"`
	// TokensUsedThisMonth is the number of tokens used this month
	TokensUsedThisMonth int `json:"tokens_used_this_month"`
}

type AITriageJobEnqueuer interface {
	EnqueueAITriage(ctx context.Context, resultID, tenantID, findingID string, delay time.Duration) error
}

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

type BudgetRepository interface {
	GetOrCreate(ctx context.Context, tenantID shared.ID, periodStart time.Time, periodEnd time.Time, defaultLimit int64) (*BudgetRow, error)
	IncrementUsed(ctx context.Context, tenantID shared.ID, periodStart time.Time, delta int64) (int64, error)
	UpdateLastWarnSent(ctx context.Context, tenantID shared.ID, periodStart time.Time, used int64) error
	UpdateLastBlockSent(ctx context.Context, tenantID shared.ID, periodStart time.Time, used int64) error
}

type BudgetRow struct {
	TenantID          shared.ID
	PeriodStart       time.Time
	PeriodEnd         time.Time
	TokenLimit        int64 // 0 == unlimited
	TokensUsed        int64
	WarnThresholdPct  int // service default when stored as 0
	BlockThresholdPct int // service default when stored as 0
	LastWarnSentUsed  int64
	LastBlockSentUsed int64
}

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

type BudgetServiceConfig struct {
	// Enabled gates the whole service. When false, Check always
	// returns nil and Record becomes a no-op. Phase 1 default.
	Enabled bool
	// Strict controls repo-failure behaviour. See ErrBudgetUnavailable.
	Strict bool
	// DefaultTokensPerMonth applied when a tenant's row has
	// token_limit = 0 AND the tenant has no plan override. 0 =
	// unlimited; see RFC-008 §3.5.
	DefaultTokensPerMonth int64
	// Threshold defaults when a row stores 0.
	DefaultWarnPct  int
	DefaultBlockPct int
}

type BudgetStatus struct {
	TenantID    shared.ID
	PeriodStart time.Time
	PeriodEnd   time.Time
	Limit       int64
	Used        int64
	Remaining   int64 // 0 when Limit == 0 (unlimited)
	UsedPct     int   // 0 when Limit == 0

	// NeedsWarn is true when the current run (or the pre-check) has
	// crossed warn_threshold_pct and no warn event has been emitted
	// yet for the period. The service queries this when deciding
	// whether to enqueue a notification.
	NeedsWarn bool
}

type BulkTriageJob struct {
	FindingID string `json:"finding_id"`
	JobID     string `json:"job_id,omitempty"`
	Status    string `json:"status"`
	Error     string `json:"error,omitempty"`
}

type BulkTriageRequest struct {
	TenantID   string   `validate:"required,uuid"`
	FindingIDs []string `validate:"required,min=1,max=100,dive,uuid"`
	UserID     *string  `validate:"omitempty,uuid"`
}

type BulkTriageResponse struct {
	Jobs       []BulkTriageJob `json:"jobs"`
	TotalCount int             `json:"total_count"`
	Queued     int             `json:"queued"`
	Failed     int             `json:"failed"`
}

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

type RecoverStuckJobsInput struct {
	// StuckDuration is how long a job must be stuck before being recovered.
	// Default: 15 minutes
	StuckDuration time.Duration
	// Limit is the maximum number of stuck jobs to recover in one batch.
	// Default: 50
	Limit int
}

type RecoverStuckJobsOutput struct {
	// Total is the number of stuck jobs found.
	Total int `json:"total"`
	// Recovered is the number of jobs successfully marked as failed.
	Recovered int `json:"recovered"`
	// Skipped is the number of jobs that were already in a terminal state.
	Skipped int `json:"skipped"`
	// Errors is the number of jobs that failed to update.
	Errors int `json:"errors"`
}

type TokenLimitError struct {
	Used  int
	Limit int
}

type TriageBroadcaster interface {
	// BroadcastTriage sends a triage event to subscribers.
	// channel: the channel to broadcast to (e.g., "triage:{finding_id}")
	// data: the triage event data
	// tenantID: tenant isolation for the broadcast
	BroadcastTriage(channel string, data any, tenantID string)
}

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

type TriageRequest struct {
	TenantID   string  `validate:"required,uuid"`
	FindingID  string  `validate:"required,uuid"`
	TriageType string  `validate:"required,oneof=auto manual bulk"`
	UserID     *string `validate:"omitempty,uuid"` // For manual requests
}

type TriageResponse struct {
	JobID  string `json:"job_id"`
	Status string `json:"status"`
}

type TriageResultResponse struct {
	ID                      string     `json:"id"`
	Status                  string     `json:"status"`
	SeverityAssessment      string     `json:"severity_assessment,omitempty"`
	SeverityJustification   string     `json:"severity_justification,omitempty"`
	RiskScore               float64    `json:"risk_score,omitempty"`
	Exploitability          string     `json:"exploitability,omitempty"`
	ExploitabilityDetails   string     `json:"exploitability_details,omitempty"`
	BusinessImpact          string     `json:"business_impact,omitempty"`
	PriorityRank            int        `json:"priority_rank,omitempty"`
	FalsePositiveLikelihood float64    `json:"false_positive_likelihood,omitempty"`
	FalsePositiveReason     string     `json:"false_positive_reason,omitempty"`
	Summary                 string     `json:"summary,omitempty"`
	CreatedAt               time.Time  `json:"created_at"`
	CompletedAt             *time.Time `json:"completed_at,omitempty"`
	ErrorMessage            string     `json:"error_message,omitempty"`
}

type WorkflowEventDispatcherInterface interface {
	DispatchAITriageCompleted(ctx context.Context, tenantID, findingID, triageID shared.ID, triageData map[string]any)
	DispatchAITriageFailed(ctx context.Context, tenantID, findingID, triageID shared.ID, errorMessage string)
}