snapshot

type AnalysisReport struct {
	Traces      int                `json:"traces"`
	Fallbacks   FallbackBreakdown  `json:"fallbacks"`
	Topics      TypeStats          `json:"topics"`
	People      TypeStats          `json:"people"`
	Artifacts   TypeStats          `json:"artifacts"`
	CostUSD     Numeric            `json:"cost_usd"`
	LLMCalls    Numeric            `json:"llm_calls"`
	ToolCalls   Numeric            `json:"tool_calls"`
	DurationMs  Numeric            `json:"duration_ms"`
	HallucTopic Numeric            `json:"halluc_rate_topics"` // (raw-kept)/raw, per-trace
	Per         []PerTraceSnapshot `json:"per_trace,omitempty"`
}

type ArtifactCandidateLine struct {
	ID         int
	Similarity float64
	FileType   string
	FileName   string
	Keywords   string // empty if absent
	Entities   string // empty if absent
	Summary    string
}

type CandidateLine struct {
	ID         int
	Similarity float64
	Date       string // free-form string from the producer (yyyy-mm-dd)
	MsgCount   int
	Size       string // "12K", "371", etc — kept verbatim
	Summary    string
}

type FallbackBreakdown struct {
	Total    int            `json:"total"`
	Success  int            `json:"success"`
	ByReason map[string]int `json:"by_reason"`
}

type Label struct {
	Score  int    `json:"score"`
	Reason string `json:"reason,omitempty"`
}

type LabelsFile map[string]TraceLabels

type Manifest struct {
	Version     string               `yaml:"version"`
	Agent       string               `yaml:"agent"`     // free-form label, e.g. "reranker"
	SpanName    string               `yaml:"span_name"` // OTel span name driving the search
	TempoURL    string               `yaml:"tempo_url"`
	Filter      string               `yaml:"filter"` // TraceQL query used
	WindowStart time.Time            `yaml:"window_start"`
	WindowEnd   time.Time            `yaml:"window_end"`
	CapturedAt  time.Time            `yaml:"captured_at"`
	TraceCount  int                  `yaml:"trace_count"`
	DBPath      string               `yaml:"db_path"` // relative to snapshot dir
	Traces      []TraceManifestEntry `yaml:"traces"`
}

type Numeric struct {
	N      int     `json:"n"`
	Mean   float64 `json:"mean"`
	Median float64 `json:"median"`
	P95    float64 `json:"p95"`
	Max    float64 `json:"max"`
	Min    float64 `json:"min"`
	Sum    float64 `json:"sum"`
}

type PerTraceSnapshot struct {
	TraceID    string  `json:"trace_id"`
	UserID     int64   `json:"user_id"`
	Fallback   string  `json:"fallback,omitempty"`
	TopicsIn   int     `json:"topics_in"`
	TopicsKept int     `json:"topics_kept"`
	PeopleIn   int     `json:"people_in"`
	PeopleKept int     `json:"people_kept"`
	ArtIn      int     `json:"artifacts_in"`
	ArtKept    int     `json:"artifacts_kept"`
	CostUSD    float64 `json:"cost_usd"`
	DurationMs int64   `json:"duration_ms"`
	ToolCalls  int     `json:"tool_calls"`
}

type PersonCandidateLine struct {
	ID     int
	Name   string
	Circle string
	Bio    string
}

type RerankerSpanData struct {
	TraceID    string
	UserID     int64
	DurationMs int64

	CandidatesIn struct {
		Topics, People, Artifacts int
	}
	ModelKept struct {
		Topics, People, Artifacts int
	}
	ModelRawCount struct {
		Topics, People, Artifacts int
	}

	FallbackReason string
	CostUSD        float64
	LLMCalls       int
	ToolCalls      int

	RawQuery                string
	EnrichedQuery           string
	CandidatesInput         string // raw multi-line body, see ParseCandidates
	PeopleCandidatesInput   string // raw body, see ParsePersonCandidates
	ArtifactCandidatesInput string // raw body, see ParseArtifactCandidates
	UserProfile             string // raw <user_profile> XML at trace time
	RecentTopics            string // raw <recent_topics> XML at trace time

	// ToolCallRequests is one slice per `reranker.tool_call` event, each
	// holding the requested topic IDs for that iteration.
	ToolCallRequests [][]int

	// Selected* are the IDs that ended up in the final selection (post-fallback
	// if any). The *Reasons maps hold the model-supplied reason per ID; entries
	// are empty when the trace ended via a fallback path.
	SelectedTopics    []int
	SelectedPeople    []int
	SelectedArtifacts []int
	SelectedReasons   map[int]string // topic-only, kept for back-compat
	SelectedReasonsP  map[int]string
	SelectedReasonsA  map[int]string
}

type SelectionResult struct {
	Topics    []int
	People    []int
	Artifacts []int

	TopicReasons    map[int]string
	PersonReasons   map[int]string
	ArtifactReasons map[int]string
}

type TempoClient struct {
	BaseURL string
	HTTP    *http.Client
}

type TraceLabels struct {
	Topics    map[string]Label `json:"topics"`
	People    map[string]Label `json:"people"`
	Artifacts map[string]Label `json:"artifacts"`
}

type TraceManifestEntry struct {
	TraceID        string `yaml:"trace_id"`
	File           string `yaml:"file"`
	DurationMs     int64  `yaml:"duration_ms"`
	StartTimeUnix  int64  `yaml:"start_time_unix"`
	ServiceVersion string `yaml:"service_version,omitempty"`
}

type TraceMeta struct {
	TraceID           string `json:"traceID"`
	RootServiceName   string `json:"rootServiceName"`
	RootTraceName     string `json:"rootTraceName"`
	StartTimeUnixNano string `json:"startTimeUnixNano"`
	DurationMs        int64  `json:"durationMs"`
}

type TypeStats struct {
	CandidatesIn  Numeric `json:"candidates_in"`
	ModelRawCount Numeric `json:"model_raw_count"`
	ModelKept     Numeric `json:"model_kept"`
	SelectionRate Numeric `json:"selection_rate"` // model_kept / candidates_in (per-trace)
}