store

Documentation

Index

• func DefaultDBPath() (string, error)
• func Today() string
• type UsageRecord
• type UsageStore
  • func NewUsageStore(dbPath string) (*UsageStore, error)
  • func (s *UsageStore) AsyncRecord(record UsageRecord)
  • func (s *UsageStore) Close() error
  • func (s *UsageStore) QueryUsage(provider, model, startDate, endDate string) ([]UsageRecord, error)

Functions

func DefaultDBPath

func DefaultDBPath() (string, error)

DefaultDBPath returns the default path for the usage database.

func Today

func Today() string

Today returns today's date in YYYY-MM-DD format.

Types

type UsageRecord

type UsageRecord struct {
	Provider            string `json:"provider"`
	Model               string `json:"model"`
	Date                string `json:"date"`
	Requests            int64  `json:"requests"`
	SuccessRequests     int64  `json:"success_requests"`
	ErrorRequests       int64  `json:"error_requests"`
	InputTokens         int64  `json:"input_tokens"`
	OutputTokens        int64  `json:"output_tokens"`
	CacheCreationTokens int64  `json:"cache_creation_tokens"`
	CacheReadTokens     int64  `json:"cache_read_tokens"`
	TotalTokens         int64  `json:"total_tokens"`
}

UsageRecord represents a single usage data point.

type UsageStore

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

UsageStore provides async SQLite storage for usage statistics.

func NewUsageStore

func NewUsageStore(dbPath string) (*UsageStore, error)

NewUsageStore creates a new UsageStore with the database at the given path.

func (*UsageStore) AsyncRecord

func (s *UsageStore) AsyncRecord(record UsageRecord)

AsyncRecord queues a usage record for async writing.

func (*UsageStore) Close

func (s *UsageStore) Close() error

Close stops the processing loop and closes the database.

func (*UsageStore) QueryUsage

func (s *UsageStore) QueryUsage(provider, model, startDate, endDate string) ([]UsageRecord, error)

QueryUsage retrieves usage records with optional filters.