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"`
	Synopsis                  string               `json:"synopsis,omitempty"`
	CompressedTimelineSummary string               `json:"compressed_timeline_summary,omitempty"`
	CompressedTimelineCount   int                  `json:"compressed_timeline_count,omitempty"`
	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 DecisionTimelineEntry struct {
	Index     int            `json:"index"`
	Type      string         `json:"type"`
	Summary   string         `json:"summary"`
	Metadata  map[string]any `json:"metadata,omitempty"`
	Timestamp time.Time      `json:"timestamp"`
}

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"`
	MaxTimelineEntries     int           `json:"max_timeline_entries"`
	PreserveRecentTimeline int           `json:"preserve_recent_timeline"`
	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"`
	Metadata   map[string]any `json:"metadata,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"`
	Timeline                  []DecisionTimelineEntry `json:"timeline,omitempty"`
	Synopsis                  string                  `json:"synopsis,omitempty"`
	CompressedTimelineSummary string                  `json:"compressed_timeline_summary,omitempty"`
	CompressedTimelineCount   int                     `json:"compressed_timeline_count,omitempty"`
	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
	TraceID   string
	Name      string
	StartTime time.Time
	EndTime   time.Time
	Duration  time.Duration

	// 父 Span
	ParentSpanID string

	// 属性
	Attributes map[string]any
	Error      *SpanError

	// 事件
	Events []SpanEvent
}

type SpanError struct {
	Message string         `json:"message"`
	Data    map[string]any `json:"data,omitempty"`
}

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
	WorkflowName string
	GroupID      string
	StartTime    time.Time
	EndTime      time.Time
	Duration     time.Duration
	Disabled     bool

	// 执行步骤
	Spans []*Span

	// 状态
	Status string
	Error  error

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

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