observability

type AgentMetric struct {
	Name         string
	InputTokens  int64
	OutputTokens int64
	ToolCalls    int64
}

type MetricsCollector struct {

	// MaxSessions caps the session map size; 0 means DefaultMaxSessions.
	MaxSessions int
	// contains filtered or unexported fields
}

type PolicyMetrics struct {
	Blocks   int64            `json:"blocks"`
	Observes int64            `json:"observes"`
	ByReason map[string]int64 `json:"byReason"`
}

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

type SessionMetric struct {
	SessionKey   string
	InputTokens  int64
	OutputTokens int64
	TotalTokens  int64
	RequestCount int64
	LastUpdated  time.Time
}

type SystemSnapshot struct {
	StartedAt        time.Time
	Uptime           time.Duration
	TokenUsageTotal  TokenUsageSummary
	ToolExecutions   int64
	ToolBreakdown    map[string]ToolMetric
	AgentBreakdown   map[string]AgentMetric
	SessionBreakdown map[string]SessionMetric
	Policy           PolicyMetrics
}

type TokenUsage struct {
	Provider     string
	Model        string
	SessionKey   string
	AgentName    string
	InputTokens  int64
	OutputTokens int64
	TotalTokens  int64
	CacheTokens  int64
	Timestamp    time.Time
}

type TokenUsageSummary struct {
	InputTokens  int64
	OutputTokens int64
	TotalTokens  int64
	CacheTokens  int64
}

type ToolMetric struct {
	Name          string
	Count         int64
	Errors        int64
	TotalDuration time.Duration
	AvgDuration   time.Duration
}