usage

type Budget struct {
	ID              string
	Name            string
	BudgetType      string
	LimitAmount     float64
	LimitUnit       string
	ScopeType       string
	ScopeTarget     string
	AlertThresholds []int
	IsActive        bool
	WindowStart     *time.Time
	WindowEnd       *time.Time
	ResetsAt        *time.Time
}

type BudgetAlert struct {
	ID               int64
	BudgetID         string
	AlertType        string
	ThresholdPercent int
	TokensAtAlert    int
	SentAt           time.Time
	Channel          string
}

type CompletionRecord struct {
	SessionKey       string
	ProviderID       string
	ModelName        string
	RequestType      string
	PromptTokens     int
	CompletionTokens int
	TotalTokens      int
	DurationMS       int
	Status           string
	UsageSource      string
	RecordedAt       time.Time
}

type CookieLoader interface {
	Load() ([]*http.Cookie, error)
}

type HistoricalUsageSample struct {
	ID            int64
	ProviderID    string
	ModelName     string
	WindowType    string
	Source        string
	SampledAt     time.Time
	UsedPercent   float64
	ResetsAt      time.Time
	WindowMinutes int
	TotalTokens   int
}

type OllamaMeIdentity struct {
	IdentityState            QuotaIdentityState
	Source                   QuotaSource
	AccountName              string
	AccountEmail             string
	PlanName                 string
	NotifyUsageLimits        bool
	AccountMetadataPopulated bool
}

type OllamaMeSigner interface {
	Sign(ctx context.Context, challenge []byte) (string, error)
}

type OllamaQuotaFetcher struct {
	BaseURL      string
	Client       *http.Client
	Clock        func() time.Time
	Signer       OllamaMeSigner
	CookieLoader CookieLoader
}

type ProviderSummary struct {
	ProviderID      string
	ModelName       string
	SessionKey      string
	SessionTokens   int
	TodayTokens     int
	WeeklyTokens    int
	SessionRequests int
	TodayRequests   int
	WeeklyRequests  int
	BudgetStatus    string
	WarningLevel    string
	Quota           *QuotaSummary
}

type QuotaIdentityState string

type QuotaSource string

type QuotaState string

type QuotaSummary struct {
	ProviderID           string
	AccountName          string
	AccountEmail         string
	PlanName             string
	SessionUsedPercent   float64
	SessionResetsAt      *time.Time
	WeeklyUsedPercent    float64
	WeeklyResetsAt       *time.Time
	NotifyUsageLimits    bool
	State                QuotaState
	Source               QuotaSource
	FetchedAt            time.Time
	ExpiresAt            time.Time
	IdentityState        QuotaIdentityState
	IdentitySource       QuotaSource
	IdentityAccountName  string
	IdentityAccountEmail string
}

type Recorder interface {
	RecordCompletion(ctx context.Context, record CompletionRecord) error
}

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

type Summary struct {
	TotalRequests         int
	TotalPromptTokens     int
	TotalCompletionTokens int
	TotalTokens           int
}

type UsageRecord struct {
	ID               int64
	SessionKey       string
	ProviderID       string
	ModelName        string
	RequestType      string
	PromptTokens     int
	CompletionTokens int
	TotalTokens      int
	DurationMS       int
	Status           string
	UsageSource      string
	CreatedAt        time.Time
}