recorder

Documentation

Index

• type Cache
• type CostEstimator
• type EventsStore
• type Recorder
  • func NewRecorder(s, us Store, c, uc Cache, ce CostEstimator, es EventsStore) *Recorder
  • func (r *Recorder) RecordEvent(e *event.Event) error
  • func (r *Recorder) RecordKeySpend(keyId string, micros int64, costLimitUnit key.TimeUnit) error
  • func (r *Recorder) RecordUserSpend(userId string, micros int64, costLimitUnit key.TimeUnit) error
• type Store

Types

type Cache

type Cache interface {
	IncrementCounter(keyId string, rateLimitUnit key.TimeUnit, incr int64) error
}

type CostEstimator

type CostEstimator interface {
	EstimatePromptCost(model string, tks int) (float64, error)
	EstimateCompletionCost(model string, tks int) (float64, error)
}

type EventsStore

type EventsStore interface {
	InsertEvent(e *event.Event) error
}

type Recorder

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

func NewRecorder

func NewRecorder(s, us Store, c, uc Cache, ce CostEstimator, es EventsStore) *Recorder

func (*Recorder) RecordEvent

func (r *Recorder) RecordEvent(e *event.Event) error

func (*Recorder) RecordKeySpend

func (r *Recorder) RecordKeySpend(keyId string, micros int64, costLimitUnit key.TimeUnit) error

func (*Recorder) RecordUserSpend

func (r *Recorder) RecordUserSpend(userId string, micros int64, costLimitUnit key.TimeUnit) error

type Store

type Store interface {
	IncrementCounter(keyId string, incr int64) error
}