tokenaccounting

type CountCallService interface {
	CountCall(context.Context, app.CountCallInput) (app.CountResult, error)
}

type CountRequest struct {
	Backend string      `json:"backend"`
	Model   string      `json:"model"`
	Mode    app.Mode    `json:"mode,omitempty"`
	Call    lipapi.Call `json:"call"`
}

type CountResponse struct {
	Planes []PlaneCount `json:"planes"`
}

type Options struct {
	Enabled      bool
	MaxBodyBytes int64
	Service      Service
}

type PlaneCount struct {
	Plane             lipapi.UsagePlane     `json:"plane"`
	Tokens            TokenDimensions       `json:"tokens"`
	Source            lipapi.UsageSource    `json:"source"`
	Authority         lipapi.UsageAuthority `json:"authority"`
	Tokenizer         lipapi.TokenizerRef   `json:"tokenizer"`
	UnavailableReason string                `json:"unavailable_reason,omitempty"`
	FallbackReason    app.FallbackReason    `json:"fallback_reason,omitempty"`
}

type Service interface {
	Count(context.Context, CountRequest) (CountResponse, error)
}

type TokenDimensions struct {
	Input      int `json:"input,omitempty"`
	Output     int `json:"output,omitempty"`
	CacheRead  int `json:"cache_read,omitempty"`
	CacheWrite int `json:"cache_write,omitempty"`
	Reasoning  int `json:"reasoning,omitempty"`
	Total      int `json:"total,omitempty"`
}