tokenusage

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

type Alert struct {
	Type    string    `json:"type"`  // usage_quota/error_rate/latency
	Level   string    `json:"level"` // warning/critical
	Message string    `json:"message"`
	Value   float64   `json:"value"`
	Time    time.Time `json:"time"`
}

type AlertResult struct {
	Alerts []*Alert `json:"alerts"`
}

type Config struct {
	StorageType     string        `json:"storage_type"` // memory/file/database
	StoragePath     string        `json:"storage_path"`
	BackupInterval  time.Duration `json:"backup_interval"`
	MaxSessions     int           `json:"max_sessions"`
	AlertThreshold  int64         `json:"alert_threshold"`
	EnableRealtime  bool          `json:"enable_realtime"`
	RetentionDays   int           `json:"retention_days"`
	MonitorInterval time.Duration `json:"monitor_interval,omitempty"`
	BatchSize       int           `json:"batch_size,omitempty"`
	DatabaseDriver  string        `json:"database_driver,omitempty"`
}

type DailyStats struct {
	Date         string           `json:"date"`
	InputTokens  int64            `json:"input_tokens"`
	OutputTokens int64            `json:"output_tokens"`
	TotalTokens  int64            `json:"total_tokens"`
	RequestCount int64            `json:"request_count"`
	SuccessCount int64            `json:"success_count"`
	ErrorCount   int64            `json:"error_count"`
	AvgLatencyMs float64          `json:"avg_latency_ms"`
	ModelTokens  map[string]int64 `json:"model_tokens,omitempty"`
}

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

type ErrorCode string

type ExportRequest struct {
	Format string         `json:"format"` // json/csv/excel
	Report *ReportRequest `json:"report"`
}

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

type HistoricalData struct {
	DailyStats   map[string]*DailyStats   `json:"daily_stats"`   // 按日期
	WeeklyStats  map[string]*WeeklyStats  `json:"weekly_stats"`  // 按周
	MonthlyStats map[string]*MonthlyStats `json:"monthly_stats"` // 按月
	ModelStats   map[string]*ModelStats   `json:"model_stats"`   // 按模型
	Records      []*UsageRecord           `json:"records,omitempty"`
}

type LLMProvider interface {
	Complete(ctx context.Context, content string) (*Response, error)
	ModelName() string
}

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

type Message struct {
	Content string
}

type Metrics struct {
	CurrentTPS     float64       `json:"current_tps"` // 每秒token数
	PeakTPS        float64       `json:"peak_tps"`
	ActiveSessions int           `json:"active_sessions"`
	ErrorRate      float64       `json:"error_rate"`
	Latency        time.Duration `json:"avg_latency"`
}

type ModelStats struct {
	ModelName    string    `json:"model_name"`
	InputTokens  int64     `json:"input_tokens"`
	OutputTokens int64     `json:"output_tokens"`
	TotalTokens  int64     `json:"total_tokens"`
	RequestCount int64     `json:"request_count"`
	SuccessCount int64     `json:"success_count"`
	ErrorCount   int64     `json:"error_count"`
	AvgLatencyMs float64   `json:"avg_latency_ms"`
	LastUpdated  time.Time `json:"last_updated"`
}

type ModelUsageReport struct {
	ModelName         string        `json:"model_name"`
	Period            string        `json:"period"`
	TotalInputTokens  int64         `json:"total_input_tokens"`
	TotalOutputTokens int64         `json:"total_output_tokens"`
	TotalTokens       int64         `json:"total_tokens"`
	RequestCount      int64         `json:"request_count"`
	SuccessRate       float64       `json:"success_rate"`
	AverageTokens     float64       `json:"average_tokens"`
	AvgLatency        time.Duration `json:"avg_latency"`
	StartDate         time.Time     `json:"start_date"`
	EndDate           time.Time     `json:"end_date"`
}

type MonthlyStats struct {
	Month        string           `json:"month"`
	InputTokens  int64            `json:"input_tokens"`
	OutputTokens int64            `json:"output_tokens"`
	TotalTokens  int64            `json:"total_tokens"`
	RequestCount int64            `json:"request_count"`
	SuccessCount int64            `json:"success_count"`
	ErrorCount   int64            `json:"error_count"`
	AvgLatencyMs float64          `json:"avg_latency_ms"`
	ModelTokens  map[string]int64 `json:"model_tokens,omitempty"`
}

type Notifier interface {
	Notify(ctx context.Context, alert *Alert) error
}

type RealtimeStats struct {
	Metrics     Metrics         `json:"metrics"`
	Sessions    []*SessionStats `json:"sessions"`
	GeneratedAt time.Time       `json:"generated_at"`
	TotalTokens int64           `json:"total_tokens"`
}

type ReportRequest struct {
	Type      string       `json:"type"` // daily/weekly/monthly/custom
	StartDate time.Time    `json:"start_date"`
	EndDate   time.Time    `json:"end_date"`
	Format    string       `json:"format"` // json/csv/excel
	Filters   *UsageFilter `json:"filters"`
}

type Response struct {
	Content string
}

type SessionStats struct {
	SessionID     string            `json:"session_id"`
	StartTime     time.Time         `json:"start_time"`
	LastUpdate    time.Time         `json:"last_update"`
	InputTokens   int64             `json:"input_tokens"`
	OutputTokens  int64             `json:"output_tokens"`
	TotalTokens   int64             `json:"total_tokens"`
	RequestCount  int               `json:"request_count"`
	AverageTokens float64           `json:"average_tokens"`
	Metadata      map[string]string `json:"metadata"`
}

type Storage interface {
	SaveSession(sessionID string, stats *SessionStats) error
	LoadSession(sessionID string) (*SessionStats, error)
	SaveHistorical(data *HistoricalData) error
	LoadHistorical() (*HistoricalData, error)
	ListSessions(start, end time.Time) ([]*SessionStats, error)
	DeleteSession(sessionID string) error
	Cleanup() error
}

type TokenRecordRequest struct {
	SessionID    string            `json:"session_id"`
	ModelName    string            `json:"model_name"`
	InputTokens  int64             `json:"input_tokens"`
	OutputTokens int64             `json:"output_tokens"`
	RequestID    string            `json:"request_id"`
	Latency      time.Duration     `json:"latency"`
	Success      bool              `json:"success"`
	Metadata     map[string]string `json:"metadata"`
}

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

type UsageError struct {
	Code    ErrorCode
	Message string
	Err     error
}

type UsageFilter struct {
	SessionIDs []string          `json:"session_ids,omitempty"`
	ModelNames []string          `json:"model_names,omitempty"`
	Success    *bool             `json:"success,omitempty"`
	Metadata   map[string]string `json:"metadata,omitempty"`
}

type UsageRecord struct {
	Time         time.Time         `json:"time"`
	SessionID    string            `json:"session_id"`
	ModelName    string            `json:"model_name"`
	InputTokens  int64             `json:"input_tokens"`
	OutputTokens int64             `json:"output_tokens"`
	TotalTokens  int64             `json:"total_tokens"`
	RequestID    string            `json:"request_id"`
	Latency      time.Duration     `json:"latency"`
	Success      bool              `json:"success"`
	Metadata     map[string]string `json:"metadata,omitempty"`
}

type UsageReport struct {
	Type              string                  `json:"type"`
	StartDate         time.Time               `json:"start_date"`
	EndDate           time.Time               `json:"end_date"`
	TotalInputTokens  int64                   `json:"total_input_tokens"`
	TotalOutputTokens int64                   `json:"total_output_tokens"`
	TotalTokens       int64                   `json:"total_tokens"`
	RequestCount      int64                   `json:"request_count"`
	SuccessCount      int64                   `json:"success_count"`
	ErrorCount        int64                   `json:"error_count"`
	AvgLatency        time.Duration           `json:"avg_latency"`
	SessionStats      map[string]SessionStats `json:"session_stats,omitempty"`
	ModelStats        map[string]ModelStats   `json:"model_stats,omitempty"`
	DailyStats        map[string]DailyStats   `json:"daily_stats,omitempty"`
	GeneratedAt       time.Time               `json:"generated_at"`
}

type WeeklyStats struct {
	Week         string           `json:"week"`
	InputTokens  int64            `json:"input_tokens"`
	OutputTokens int64            `json:"output_tokens"`
	TotalTokens  int64            `json:"total_tokens"`
	RequestCount int64            `json:"request_count"`
	SuccessCount int64            `json:"success_count"`
	ErrorCount   int64            `json:"error_count"`
	AvgLatencyMs float64          `json:"avg_latency_ms"`
	ModelTokens  map[string]int64 `json:"model_tokens,omitempty"`
}