observability

type AgentMetrics struct {
	AgentID string

	// 性能指标
	TotalTasks      int64
	SuccessfulTasks int64
	FailedTasks     int64
	TaskSuccessRate float64
	AvgLatency      time.Duration
	P50Latency      time.Duration
	P95Latency      time.Duration
	P99Latency      time.Duration

	// Token 指标
	TotalTokens      int64
	PromptTokens     int64
	CompletionTokens int64
	TokenEfficiency  float64 // tokens per task

	// 质量指标
	AvgOutputQuality float64
	HumanSimilarity  float64

	// 成本指标
	TotalCost   float64
	CostPerTask float64

	// 时间统计
	FirstTaskAt time.Time
	LastTaskAt  time.Time

	// 详细记录
	LatencyHistory []time.Duration
	QualityHistory []float64
}

type Alternative struct {
	Option    string  `json:"option"`
	Score     float64 `json:"score"`
	Reason    string  `json:"reason"`
	WasChosen bool    `json:"was_chosen"`
}

type AuditReport struct {
	TraceID         string               `json:"trace_id"`
	SessionID       string               `json:"session_id"`
	AgentID         string               `json:"agent_id"`
	StartTime       time.Time            `json:"start_time"`
	EndTime         time.Time            `json:"end_time"`
	Duration        time.Duration        `json:"duration"`
	Success         bool                 `json:"success"`
	TotalSteps      int                  `json:"total_steps"`
	TotalDecisions  int                  `json:"total_decisions"`
	DecisionSummary map[DecisionType]int `json:"decision_summary"`
	Timeline        []TimelineEvent      `json:"timeline"`
}

type Benchmark struct {
	Name        string
	Description string
	Dataset     []BenchmarkCase
	Results     map[string]*BenchmarkResult // agentID -> result
}

type BenchmarkCase struct {
	ID             string
	Input          *agent.Input
	ExpectedOutput string
	Metadata       map[string]any
}

type BenchmarkResult struct {
	AgentID     string
	TotalCases  int
	PassedCases int
	FailedCases int
	SuccessRate float64
	AvgScore    float64
	AvgLatency  time.Duration
	TotalCost   float64
	Timestamp   time.Time
}

type Decision struct {
	ID           string            `json:"id"`
	Type         DecisionType      `json:"type"`
	Description  string            `json:"description"`
	Input        any               `json:"input,omitempty"`
	Output       any               `json:"output,omitempty"`
	Reasoning    string            `json:"reasoning"`
	Confidence   float64           `json:"confidence,omitempty"`
	Alternatives []Alternative     `json:"alternatives,omitempty"`
	Factors      []Factor          `json:"factors,omitempty"`
	Timestamp    time.Time         `json:"timestamp"`
	Duration     time.Duration     `json:"duration,omitempty"`
	Metadata     map[string]string `json:"metadata,omitempty"`
}

type DecisionType string

type EvaluationResult struct {
	Score      float64
	Dimensions map[string]float64 // 各维度分数
	Feedback   string
	Timestamp  time.Time
}

type EvaluationStrategy interface {
	Evaluate(ctx context.Context, input *agent.Input, output *agent.Output) (*EvaluationResult, error)
}

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

type ExplainabilityConfig struct {
	Enabled            bool          `json:"enabled"`
	DetailLevel        string        `json:"detail_level"` // minimal, standard, verbose
	MaxTraceAge        time.Duration `json:"max_trace_age"`
	MaxTracesPerAgent  int           `json:"max_traces_per_agent"`
	RecordAlternatives bool          `json:"record_alternatives"`
	RecordFactors      bool          `json:"record_factors"`
}

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

type Factor struct {
	Name        string  `json:"name"`
	Value       float64 `json:"value"`
	Weight      float64 `json:"weight"`
	Impact      string  `json:"impact"` // positive, negative, neutral
	Explanation string  `json:"explanation"`
}

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

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

type ReasoningStep struct {
	StepNumber int           `json:"step_number"`
	Type       string        `json:"type"` // thought, action, observation, decision
	Content    string        `json:"content"`
	Decisions  []Decision    `json:"decisions,omitempty"`
	Timestamp  time.Time     `json:"timestamp"`
	Duration   time.Duration `json:"duration,omitempty"`
}

type ReasoningTrace struct {
	ID          string          `json:"id"`
	SessionID   string          `json:"session_id"`
	AgentID     string          `json:"agent_id"`
	TaskID      string          `json:"task_id,omitempty"`
	Steps       []ReasoningStep `json:"steps"`
	Decisions   []Decision      `json:"decisions"`
	StartTime   time.Time       `json:"start_time"`
	EndTime     time.Time       `json:"end_time,omitempty"`
	Duration    time.Duration   `json:"duration,omitempty"`
	Success     bool            `json:"success"`
	FinalOutput string          `json:"final_output,omitempty"`
	Error       string          `json:"error,omitempty"`
	Metadata    map[string]any  `json:"metadata,omitempty"`
}

type SimpleEvaluationStrategy struct{}

type Span struct {
	SpanID    string
	Name      string
	StartTime time.Time
	EndTime   time.Time
	Duration  time.Duration

	// 父 Span
	ParentSpanID string

	// 属性
	Attributes map[string]any

	// 事件
	Events []SpanEvent
}

type SpanEvent struct {
	Name       string
	Timestamp  time.Time
	Attributes map[string]any
}

type TimelineEvent struct {
	Timestamp   time.Time `json:"timestamp"`
	Type        string    `json:"type"`
	Description string    `json:"description"`
}

type Trace struct {
	TraceID   string
	AgentID   string
	StartTime time.Time
	EndTime   time.Time
	Duration  time.Duration

	// 执行步骤
	Spans []*Span

	// 状态
	Status string
	Error  error

	// 元数据
	Metadata map[string]any
}

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