agent

type APIError struct {
	StatusCode int
	Body       string
}

type ArchConcern struct {
	Area             string         `json:"area"`
	Description      string         `json:"description"`
	AffectedPackages []string       `json:"affected_packages"`
	Metrics          map[string]any `json:"metrics"`
}

type ArchCoverageGap struct {
	Package      string  `json:"package"`
	CurrentScore float64 `json:"current_score"`
	Issue        string  `json:"issue"`
}

type ArchDataFlow struct {
	From        string `json:"from"`
	To          string `json:"to"`
	Description string `json:"description"`
}

type ArchDelta struct {
	Metric    string `json:"metric"`
	Current   string `json:"current"`
	Projected string `json:"projected"`
}

type ArchFinding struct {
	Package        string             `json:"package"`
	Issue          string             `json:"issue"`
	CurrentMetrics map[string]float64 `json:"current_metrics"`
	Severity       string             `json:"severity"`
}

type ArchLayer struct {
	Name        string   `json:"name"`
	Packages    []string `json:"packages"`
	Description string   `json:"description"`
}

type ArchPhase1Result struct {
	Layers               []ArchLayer    `json:"layers"`
	DataFlows            []ArchDataFlow `json:"data_flows"`
	DependencyAssessment string         `json:"dependency_assessment"`
}

type ArchPhase2Result struct {
	Findings []ArchFinding `json:"findings"`
}

type ArchPhase3Result struct {
	CoverageGaps       []ArchCoverageGap `json:"coverage_gaps"`
	StrategyAssessment string            `json:"strategy_assessment"`
}

type ArchPhase4Result struct {
	Concerns []ArchConcern `json:"concerns"`
}

type ArchPhase5Result struct {
	Recommendations []ArchRecommendation `json:"recommendations"`
}

type ArchPhase6Result struct {
	ExecutiveSummary string         `json:"executive_summary"`
	RiskMatrix       []ArchRisk     `json:"risk_matrix"`
	Roadmap          []ArchRoadItem `json:"roadmap"`
}

type ArchRecommendation struct {
	Title         string      `json:"title"`
	CurrentState  string      `json:"current_state"`
	ProposedState string      `json:"proposed_state"`
	Deltas        []ArchDelta `json:"deltas"`
	AffectedUnits []string    `json:"affected_units"`
	Effort        string      `json:"effort"`
	Justification string      `json:"justification"`
}

type ArchRisk struct {
	Risk              string `json:"risk"`
	Severity          string `json:"severity"`
	Likelihood        string `json:"likelihood"`
	RecommendationRef string `json:"recommendation_ref"`
}

type ArchRoadItem struct {
	Priority          int    `json:"priority"`
	Title             string `json:"title"`
	Effort            string `json:"effort"`
	Impact            string `json:"impact"`
	RecommendationRef string `json:"recommendation_ref"`
	DeltaSummary      string `json:"delta_summary"`
}

type ArchSnapshot struct {
	Packages        []PackageNode     // every package with metrics, sorted by path
	DependencyEdges []DepEdge         // package → package imports
	Layers          map[string]string // package → layer (cmd, internal, domain, external)
	Hotspots        []PackageNode     // top 10 by risk score (units × (1 - avg_score))
	CouplingPairs   []CouplingPair    // most cross-referenced package pairs
	Metrics         SnapshotMetrics   // aggregate metrics
}

type ArchitectResult struct {
	Snapshot       *ArchSnapshot
	Phase1         *ArchPhase1Result
	Phase2         *ArchPhase2Result
	Phase3         *ArchPhase3Result
	Phase4         *ArchPhase4Result
	Phase5         *ArchPhase5Result
	Phase6         *ArchPhase6Result
	RawOutputs     []string // raw LLM responses per phase
	Thinking       []string // chain-of-thought reasoning per phase (from <think> tags)
	TotalTokens    int
	Duration       time.Duration
	Model          string
	PhasesComplete int
	Errors         []string // errors per failed phase
}

type ArchitectReviewer struct {
	Provider     Provider
	Model        string
	Verbose      bool
	OnPhaseStart func(phase int, name string)
	OnPhaseDone  func(phase int, name string, tokens int)
}

type ChatRequest struct {
	Model       string    `json:"model"`
	Messages    []Message `json:"messages"`
	Temperature float64   `json:"temperature,omitempty"`
	MaxTokens   int       `json:"max_tokens,omitempty"`
	// ResponseFormat can request structured JSON output
	ResponseFormat *ResponseFormat `json:"response_format,omitempty"`
}

type ChatResponse struct {
	ID      string   `json:"id"`
	Model   string   `json:"model"`
	Choices []Choice `json:"choices"`
	Usage   Usage    `json:"usage"`
}

type Choice struct {
	Index        int     `json:"index"`
	Message      Message `json:"message"`
	FinishReason string  `json:"finish_reason"`
}

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

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

type CoordinatorConfig struct {
	Models      domain.ModelAssignments
	Strategy    Strategy
	TokenBudget int // max tokens to spend across all reviews (0=unlimited)
}

type CouplingPair struct {
	PkgA      string `json:"pkg_a"`
	PkgB      string `json:"pkg_b"`
	EdgeCount int    `json:"edge_count"` // total edges both directions
}

type DecisionResponse struct {
	Status    string   `json:"status"`
	Reasoning string   `json:"reasoning"`
	Actions   []string `json:"actions"`
}

type DeepReviewResponse struct {
	Summary      string   `json:"summary"`
	Review       string   `json:"review"`
	Risks        []string `json:"risks"`
	SystemImpact []string `json:"system_impact"`
	Suggestions  []string `json:"suggestions"`
	Strengths    []string `json:"strengths"`
}

type DepEdge struct {
	From   string `json:"from"`
	To     string `json:"to"`
	Weight int    `json:"weight"` // number of import sites
}

type DetectedProvider struct {
	Name    string   // "openrouter", "groq", "ollama", "lmstudio"
	BaseURL string   // API base URL
	APIKey  string   // API key (empty for local providers)
	Models  []string // Preferred models for this provider
	Local   bool     // True for local providers (no auth required)
}

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

type Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

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

type ModelConfig struct {
	ID                string `json:"id"`
	ContextWindow     int    `json:"context_window"`
	MaxOutput         int    `json:"max_output"`
	StructuredOutputs bool   `json:"structured_outputs"`
	ToolCalling       bool   `json:"tool_calling"`
	Reasoning         bool   `json:"reasoning"`
}

type ModelInfo struct {
	ID            string `json:"id"`
	OwnedBy       string `json:"owned_by,omitempty"`
	ContextWindow int    `json:"context_window,omitempty"`
	Created       int64  `json:"created,omitempty"`
}

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

type PackageNode struct {
	Path         string   `json:"path"`
	Units        int      `json:"units"`
	AvgScore     float64  `json:"avg_score"`
	Grade        string   `json:"grade"`
	Observations int      `json:"observations"`
	TopIssues    []string `json:"top_issues,omitempty"`
	Imports      []string `json:"imports,omitempty"`
	ImportedBy   []string `json:"imported_by,omitempty"`
	ExportedAPI  []string `json:"exported_api,omitempty"`
}

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

type PipelineConfig struct {
	Strategy Strategy
	Models   domain.ModelAssignments
}

type PrescreenResponse struct {
	NeedsReview bool     `json:"needs_review"`
	Reason      string   `json:"reason"`
	Confidence  float64  `json:"confidence"`
	Suggestions []string `json:"suggestions,omitempty"`
}

type ProjectContext struct {
	RepoName      string
	CommitSHA     string
	Languages     []string
	Snapshot      *ArchSnapshot
	LowestUnits   []UnitSummary     // bottom 20 by score
	Documentation map[string]string // filename → content (README, PRD, etc.)
	FileTree      string
	PolicyRules   []string
	GitSummary    string
}

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

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

type Provider interface {
	// Chat sends a chat completion request and returns the response.
	Chat(ctx context.Context, req ChatRequest) (ChatResponse, error)

	// Name returns the provider name.
	Name() string
}

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

type RemediationResponse struct {
	Steps []RemediationStep `json:"steps"`
}

type RemediationStep struct {
	Priority    int    `json:"priority"`
	Dimension   string `json:"dimension"`
	Description string `json:"description"`
	Effort      string `json:"effort"` // low, medium, high
}

type RepoSummary struct {
	Languages    []string // Detected languages (e.g., "go", "typescript")
	UnitCount    int      // Total code units discovered
	FilePatterns []string // Top-level directories or patterns found
	Policies     []string // Currently configured policy pack names
}

type ResponseFormat struct {
	Type   string `json:"type"` // "json_object" or "json_schema"
	Schema any    `json:"json_schema,omitempty"`
}

type ReviewInput struct {
	Unit       domain.Unit
	SourceCode string
	Evidence   []domain.Evidence
}

type ReviewResult struct {
	Reviewed        bool               `json:"reviewed"`
	Prescreened     bool               `json:"prescreened"`
	ReviewOutput    string             `json:"review_output,omitempty"`
	Scores          map[string]float64 `json:"scores,omitempty"`
	Status          string             `json:"status,omitempty"`
	Actions         []string           `json:"actions,omitempty"`
	Remediation     []RemediationStep  `json:"remediation,omitempty"`
	Confidence      float64            `json:"confidence"`
	TokensUsed      int                `json:"tokens_used"`
	ModelsUsed      []string           `json:"models_used,omitempty"`
	PrescreenReason string             `json:"prescreen_reason,omitempty"`
	Suggestions     []string           `json:"suggestions,omitempty"`
}

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

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

type ScanSuggestion struct {
	Suggestions string // Human-readable suggestion text
	TokensUsed  int    // Tokens consumed by this call
	Model       string // Model that generated the suggestion
}

type ScoringResponse struct {
	Scores     map[string]float64 `json:"scores"`
	Confidence float64            `json:"confidence"`
	Reasoning  string             `json:"reasoning"`
}

type SnapshotMetrics struct {
	TotalUnits        int            `json:"total_units"`
	TotalPackages     int            `json:"total_packages"`
	AvgScore          float64        `json:"avg_score"`
	GradeDistribution map[string]int `json:"grade_distribution"`
	TopObservations   map[string]int `json:"top_observations"`
	PolicyViolations  map[string]int `json:"policy_violations"`
}

type Stage interface {
	Execute(ctx context.Context, input StageInput) (StageResult, bool, error)
	Name() string
}

type StageInput struct {
	Unit            domain.Unit
	SourceCode      string
	Evidence        []domain.Evidence
	EvidenceSummary string
	ReviewOutput    string // set by ReviewStage, read by ScoringStage
}

type StageResult struct {
	Reviewed        bool
	ReviewOutput    string
	Scores          map[string]float64
	Status          string
	Actions         []string
	Remediation     []RemediationStep
	Confidence      float64
	TokensUsed      int
	ModelsUsed      []string // attribution: which models contributed
	PrescreenReason string   // brief quality assessment from prescreen
	Suggestions     []string // actionable improvement suggestions
}

type Strategy int

type TaskType int

type UnitSummary struct {
	ID           string
	Score        float64
	Grade        string
	Observations []string
}

type Usage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}