metrics

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

type CallRecord struct {
	ID        string    `json:"id"`
	Timestamp time.Time `json:"timestamp"`
	Model     string    `json:"model"`
	Success   bool      `json:"success"`
	ErrorType string    `json:"error_type,omitempty"` // timeout, rate_limit, error

	// Latency
	LatencyMs        float64 `json:"latency_ms"`
	TimeToFirstToken float64 `json:"ttft_ms,omitempty"`

	// Tokens
	InputTokens      int64 `json:"input_tokens"`
	OutputTokens     int64 `json:"output_tokens"`
	CacheReadTokens  int64 `json:"cache_read_tokens,omitempty"`
	CacheWriteTokens int64 `json:"cache_write_tokens,omitempty"`

	// Speed
	TokensPerSecond float64 `json:"tokens_per_second,omitempty"`
}

type CallStats struct {
	// Call counts
	TotalCalls      int64 `json:"total_calls"`
	SuccessfulCalls int64 `json:"successful_calls"`
	FailedCalls     int64 `json:"failed_calls"`

	// By error type
	TimeoutCalls   int64 `json:"timeout_calls"`
	RateLimitCalls int64 `json:"rate_limit_calls"`
	ErrorCalls     int64 `json:"error_calls"`

	// Success rate
	SuccessRate float64 `json:"success_rate"`
}

type CallStatsResponse struct {
	Period string        `json:"period"`
	Stats  *CallStats    `json:"stats"`
	ByHour []HourlyStats `json:"by_hour,omitempty"`
}

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

type ContinuousQuery struct {
	Name          string
	Database      string
	Query         string
	ResampleEvery time.Duration
	ResampleFor   time.Duration
}

type GenerationSpeed struct {
	// Token generation rate
	TokensPerSecond    float64 `json:"tokens_per_second"`
	AvgTokensPerSecond float64 `json:"avg_tokens_per_second"`
	MaxTokensPerSecond float64 `json:"max_tokens_per_second"`

	// Prefill rate (time to first token)
	PrefillSpeed        float64 `json:"prefill_speed"`
	TimeToFirstToken    float64 `json:"time_to_first_token_ms"`
	AvgTimeToFirstToken float64 `json:"avg_ttft_ms"`

	// Decode rate
	DecodeSpeed float64 `json:"decode_speed"`
}

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

type HourlyStats struct {
	Hour        time.Time `json:"hour"`
	Calls       int64     `json:"calls"`
	SuccessRate float64   `json:"success_rate"`
}

type LatencySampleData struct {
	Model       string    `json:"model"`
	SampleType  string    `json:"sample_type"` // "latency", "tps", "ttft", "decode"
	Samples     []float64 `json:"samples"`
	TotalValue  float64   `json:"total_value"`
	MinValue    float64   `json:"min_value"`
	MaxValue    float64   `json:"max_value"`
	SampleCount int64     `json:"sample_count"`
}

type LatencySampleExport struct {
	Model       string
	SampleType  string // "latency", "tps", "ttft", "decode"
	Samples     []float64
	TotalValue  float64
	MinValue    float64
	MaxValue    float64
	SampleCount int64
}

type LatencyStats struct {
	// Basic statistics (ms)
	Min float64 `json:"min_ms"`
	Max float64 `json:"max_ms"`
	Avg float64 `json:"avg_ms"`

	// Percentiles (ms)
	P50 float64 `json:"p50_ms"`
	P90 float64 `json:"p90_ms"`
	P95 float64 `json:"p95_ms"`
	P99 float64 `json:"p99_ms"`

	// Sample count
	Samples int64 `json:"samples"`
}

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

type MetricsHistory struct {
	Metrics         []SystemMetrics `json:"metrics"`
	IntervalSeconds int             `json:"interval_seconds"`
}

type MetricsQuery struct {
	Period    string    `json:"period"`
	StartTime time.Time `json:"start_time,omitempty"`
	EndTime   time.Time `json:"end_time,omitempty"`
	Model     string    `json:"model,omitempty"`
	Limit     int       `json:"limit,omitempty"`
}

type MetricsStore interface {
	// Write writes a single point to the store.
	Write(ctx context.Context, point *Point) error

	// WriteBatch writes multiple points to the store.
	WriteBatch(ctx context.Context, points []*Point) error

	// Query executes a query and returns results.
	Query(ctx context.Context, query string) (*QueryResult, error)

	// QueryRange queries metrics within a time range.
	QueryRange(ctx context.Context, measurement string, start, end time.Time, aggregation string) (*QueryResult, error)

	// Close closes the store connection.
	Close() error
}

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

type ModelComparison struct {
	Period         string               `json:"period"`
	Models         []ModelStats         `json:"models"`
	Recommendation *ModelRecommendation `json:"recommendation,omitempty"`
	Insights       []string             `json:"insights,omitempty"`
}

type ModelDataPoint struct {
	Timestamp     time.Time `json:"timestamp"`
	Calls         int64     `json:"calls"`
	SuccessRate   float64   `json:"success_rate"`
	AvgLatency    float64   `json:"avg_latency_ms"`
	TotalTokens   int64     `json:"total_tokens"`
	EstimatedCost float64   `json:"estimated_cost"`
}

type ModelRecommendation struct {
	BestPerformance string `json:"best_performance"` // Best quality output
	BestCostValue   string `json:"best_cost_value"`  // Best cost-effectiveness
	MostReliable    string `json:"most_reliable"`    // Most stable
	Fastest         string `json:"fastest"`          // Fastest response
}

type ModelSpeedResponse struct {
	Model       string            `json:"model"`
	Current     *SpeedStats       `json:"current"`
	Average     *SpeedStats       `json:"average"`
	Percentiles *SpeedPercentiles `json:"percentiles"`
}

type ModelStats struct {
	Model           string  `json:"model"`
	Calls           int64   `json:"calls"`
	SuccessfulCalls int64   `json:"successful_calls"`
	FailedCalls     int64   `json:"failed_calls"`
	SuccessRate     float64 `json:"success_rate"`

	// Latency statistics (ms)
	AvgLatency float64 `json:"avg_latency_ms"`
	P50Latency float64 `json:"p50_latency_ms"`
	P95Latency float64 `json:"p95_latency_ms"`
	P99Latency float64 `json:"p99_latency_ms"`

	// Token statistics
	InputTokens      int64 `json:"input_tokens"`
	OutputTokens     int64 `json:"output_tokens"`
	TotalTokens      int64 `json:"total_tokens"`
	CacheReadTokens  int64 `json:"cache_read_tokens"`
	CacheWriteTokens int64 `json:"cache_write_tokens"`

	// Performance statistics
	AvgTokensPerSecond  float64 `json:"avg_tokens_per_second"`
	AvgTimeToFirstToken float64 `json:"avg_ttft_ms"`

	// Cost statistics
	EstimatedCost float64 `json:"estimated_cost"`
}

type ModelStatsHistory struct {
	Model      string           `json:"model"`
	Period     string           `json:"period"` // hourly, daily, monthly
	DataPoints []ModelDataPoint `json:"data_points"`
}

type ModelStatsResponse struct {
	Period  string        `json:"period"`
	Models  []ModelStats  `json:"models"`
	Summary *StatsSummary `json:"summary,omitempty"`
}

type ModelTokenUsage struct {
	Model            string  `json:"model"`
	InputTokens      int64   `json:"input_tokens"`
	OutputTokens     int64   `json:"output_tokens"`
	TotalTokens      int64   `json:"total_tokens"`
	CacheReadTokens  int64   `json:"cache_read_tokens"`
	CacheWriteTokens int64   `json:"cache_write_tokens"`
	EstimatedCost    float64 `json:"estimated_cost"`
}

type Point struct {
	// Measurement is the metric name (e.g., "api_calls", "token_usage")
	Measurement string

	// Tags are indexed metadata (e.g., model, status)
	Tags map[string]string

	// Fields are the actual metric values
	Fields map[string]interface{}

	// Timestamp is when the metric was recorded
	Timestamp time.Time
}

type ProcessMetrics struct {
	// Process info
	PID     int    `json:"pid"`
	Command string `json:"command"`
	State   string `json:"state"` // running, sleeping, etc.

	// CPU usage
	CPUPercent float64 `json:"cpu_percent"`
	CPUTime    float64 `json:"cpu_time_s"`

	// Memory usage
	MemoryRSS     int64   `json:"memory_rss_bytes"`
	MemoryVMS     int64   `json:"memory_vms_bytes"`
	MemoryPercent float64 `json:"memory_percent"`

	// I/O statistics
	IOReadBytes  int64 `json:"io_read_bytes"`
	IOWriteBytes int64 `json:"io_write_bytes"`

	// Threads
	NumThreads int `json:"num_threads"`

	// Runtime
	StartTime time.Time     `json:"start_time"`
	Uptime    time.Duration `json:"uptime"`
}

type ProcessMetricsResponse struct {
	Processes []ProcessMetrics `json:"processes"`
	Summary   *ProcessSummary  `json:"summary"`
}

type ProcessSummary struct {
	TotalProcesses   int     `json:"total_processes"`
	TotalCPUPercent  float64 `json:"total_cpu_percent"`
	TotalMemoryBytes int64   `json:"total_memory_bytes"`
}

type QueryResult struct {
	// Series contains the result series
	Series []Series
}

type ResourceHistory struct {
	Timestamp time.Time `json:"timestamp"`
	CPU       float64   `json:"cpu_percent"`
	Memory    float64   `json:"memory_percent"`
	Disk      float64   `json:"disk_percent"`
}

type RetentionPolicy struct {
	Name     string
	Duration time.Duration
	Default  bool
}

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

type Series struct {
	// Name is the measurement name
	Name string

	// Tags are the series tags
	Tags map[string]string

	// Columns are the column names
	Columns []string

	// Values are the data rows
	Values [][]interface{}
}

type SpeedPercentiles struct {
	TTFTP50Ms float64 `json:"ttft_p50_ms"`
	TTFTP95Ms float64 `json:"ttft_p95_ms"`
	TTFTP99Ms float64 `json:"ttft_p99_ms"`
	TPSP50    float64 `json:"tps_p50"`
	TPSP95    float64 `json:"tps_p95"`
	TPSP99    float64 `json:"tps_p99"`
}

type SpeedResponse struct {
	Current     *SpeedStats       `json:"current"`
	Average     *SpeedStats       `json:"average"`
	Percentiles *SpeedPercentiles `json:"percentiles"`
}

type SpeedStats struct {
	TokensPerSecond    float64 `json:"tokens_per_second"`
	TimeToFirstTokenMs float64 `json:"time_to_first_token_ms"`
	DecodeSpeed        float64 `json:"decode_speed"`
}

type StatsSummary struct {
	TotalCalls  int64   `json:"total_calls"`
	TotalTokens int64   `json:"total_tokens"`
	TotalCost   float64 `json:"total_cost"`
}

type StoreConfig struct {
	// Connection
	URL      string
	Database string
	Username string
	Password string

	// Retention policies
	RetentionPolicies []RetentionPolicy

	// Collection settings
	CollectionInterval time.Duration
	BatchSize          int
	FlushInterval      time.Duration
}

type SystemMetrics struct {
	Timestamp                      time.Time `json:"timestamp"`
	CPUPercent                     float64   `json:"cpu_percent"`
	MemoryUsedBytes                uint64    `json:"memory_used_bytes"`
	MemoryRSSBytes                 uint64    `json:"memory_rss_bytes"`
	MemoryTotalBytes               uint64    `json:"memory_total_bytes"`
	Goroutines                     int       `json:"goroutines"`
	GCPauseNs                      uint64    `json:"gc_pause_ns"`
	HeapAllocBytes                 uint64    `json:"heap_alloc_bytes"`
	HeapSysBytes                   uint64    `json:"heap_sys_bytes"`
	StackInuseBytes                uint64    `json:"stack_inuse_bytes"`
	ConversationCacheEntries       int       `json:"conversation_cache_entries,omitempty"`
	ConversationCacheBytes         uint64    `json:"conversation_cache_bytes,omitempty"`
	WarmupCacheEntries             int       `json:"warmup_cache_entries,omitempty"`
	WarmupCacheBytes               uint64    `json:"warmup_cache_bytes,omitempty"`
	PromptToolSurfaceRefs          int       `json:"prompt_tool_surface_refs,omitempty"`
	PromptToolSurfaceSharedEntries int       `json:"prompt_tool_surface_shared_entries,omitempty"`
	PromptToolSurfaceSharedBytes   uint64    `json:"prompt_tool_surface_shared_bytes,omitempty"`
	ProviderAffinityEntries        int       `json:"provider_affinity_entries,omitempty"`
}

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

type SystemResourceMetrics struct {
	// CPU
	CPUCount        int     `json:"cpu_count"`
	CPUUsagePercent float64 `json:"cpu_usage_percent"`
	LoadAvg1        float64 `json:"load_avg_1"`
	LoadAvg5        float64 `json:"load_avg_5"`
	LoadAvg15       float64 `json:"load_avg_15"`

	// Memory
	MemoryTotal   int64   `json:"memory_total_bytes"`
	MemoryUsed    int64   `json:"memory_used_bytes"`
	MemoryFree    int64   `json:"memory_free_bytes"`
	MemoryPercent float64 `json:"memory_percent"`

	// Disk
	DiskTotal   int64   `json:"disk_total_bytes"`
	DiskUsed    int64   `json:"disk_used_bytes"`
	DiskFree    int64   `json:"disk_free_bytes"`
	DiskPercent float64 `json:"disk_percent"`

	// Network
	NetworkBytesSent int64 `json:"network_bytes_sent"`
	NetworkBytesRecv int64 `json:"network_bytes_recv"`
}

type TimeBreakdown struct {
	QueueTime        time.Duration `json:"queue_time_ms"`
	TimeToFirstToken time.Duration `json:"ttft_ms"`
	GenerationTime   time.Duration `json:"generation_time_ms"`
	TotalTime        time.Duration `json:"total_time_ms"`
}

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

type TokenPricing struct {
	// Model matching pattern (supports wildcards)
	Pattern string `json:"pattern" yaml:"pattern"` // e.g., "opus*", "sonnet*", "haiku*"

	// Price per million tokens (USD)
	InputPrice  float64 `json:"input_price" yaml:"input"`
	OutputPrice float64 `json:"output_price" yaml:"output"`
	CacheRead   float64 `json:"cache_read" yaml:"cache_read"`
	CacheWrite  float64 `json:"cache_write" yaml:"cache_write"`
}

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

type TokenUsage struct {
	// Input/Output
	InputTokens  int64 `json:"input_tokens"`
	OutputTokens int64 `json:"output_tokens"`
	TotalTokens  int64 `json:"total_tokens"`

	// Cache related
	CacheReadTokens  int64 `json:"cache_read_tokens"`
	CacheWriteTokens int64 `json:"cache_write_tokens"`

	// Cost estimation (USD)
	EstimatedCost float64 `json:"estimated_cost"`
}

type TokenUsageResponse struct {
	Period  string            `json:"period"`
	Usage   *TokenUsage       `json:"usage"`
	ByModel []ModelTokenUsage `json:"by_model,omitempty"`
}

type UserTokenUsage struct {
	UserID           string  `json:"user_id"`
	InputTokens      int64   `json:"input_tokens"`
	OutputTokens     int64   `json:"output_tokens"`
	TotalTokens      int64   `json:"total_tokens"`
	CacheReadTokens  int64   `json:"cache_read_tokens"`
	CacheWriteTokens int64   `json:"cache_write_tokens"`
	EstimatedCost    float64 `json:"estimated_cost"`
	RequestCount     int64   `json:"request_count"`
}

type UserTokenUsageResponse struct {
	Period  string            `json:"period"`
	Users   []UserTokenUsage  `json:"users"`
	Summary *UserUsageSummary `json:"summary,omitempty"`
}

type UserUsageSummary struct {
	TotalUsers    int     `json:"total_users"`
	TotalTokens   int64   `json:"total_tokens"`
	TotalCost     float64 `json:"total_cost"`
	TotalRequests int64   `json:"total_requests"`
}

type WriterConfig struct {
	// How often to collect and write metrics
	CollectionInterval time.Duration

	// Maximum samples to keep in memory
	MaxSamples int

	// Enable system metrics collection
	EnableSystemMetrics bool

	// System metrics collection interval
	SystemMetricsInterval time.Duration

	// Disk path for disk usage monitoring
	DiskPath string

	// SQLite database path for persistence when using a dedicated metrics DB.
	SQLiteDBPath string

	// SharedSQLiteDB reuses an existing SQLite database instead of opening metrics.db.
	SharedSQLiteDB *sql.DB

	// SharedSQLiteReadDB optionally provides a separate reader when
	// SharedSQLiteDB points at a writer-only pool.
	SharedSQLiteReadDB *sql.DB

	// Persistence save interval
	PersistenceInterval time.Duration
}