Documentation
¶
Overview ¶
Package cost tracks per-turn token usage and dollar cost across providers. The Tracker is process-local, thread-safe, and consumed by both the TUI status bar (for live session total) and the cost-monitor pane (for the historical breakdown view).
Index ¶
- func AddLifetime(in, out int)
- func FormatLifetimeTokens(total int64) string
- func LifetimeTotals() (in, out int64)
- func ResetLifetimeForTest()
- func SortedKeys[V any](m map[string]V) []string
- type Event
- type Price
- type Summary
- type Tracker
- func (t *Tracker) ByModel() map[string]Summary
- func (t *Tracker) ByProvider() map[string]Summary
- func (t *Tracker) Events() []Event
- func (t *Tracker) Filter(provider, model string) []Event
- func (t *Tracker) LastInput() int
- func (t *Tracker) Record(provider, model string, in, out int) Event
- func (t *Tracker) Reset()
- func (t *Tracker) SetEstimatedInput(n int)
- func (t *Tracker) Total() Summary
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func AddLifetime ¶
func AddLifetime(in, out int)
AddLifetime bumps persisted totals and writes the file atomically. I/O errors are swallowed: a missing or unwritable home should never crash a turn, and on next start the counter just stays at the last successful sum.
func FormatLifetimeTokens ¶
FormatLifetimeTokens compacts a token count for banner display: thousands as K, millions as M, billions as B. Two significant digits below 10, one digit above so the column doesn't grow as totals climb.
func LifetimeTotals ¶
func LifetimeTotals() (in, out int64)
LifetimeTotals returns persisted input/output token totals across every bee session that ever called AddLifetime.
func ResetLifetimeForTest ¶
func ResetLifetimeForTest()
ResetLifetimeForTest clears the cached load so the next call re-reads from disk. Test-only; production never hot-reloads.
func SortedKeys ¶
SortedKeys returns map keys sorted alphabetically — convenience for deterministic rendering in the TUI pane.
Types ¶
type Price ¶
Price is per-million-token rates for one model.
type Tracker ¶
type Tracker struct {
// contains filtered or unexported fields
}
Tracker is the in-memory store. Append-only; never persisted.
func (*Tracker) ByProvider ¶
ByProvider groups events by provider name.
func (*Tracker) Filter ¶
Filter narrows events by provider and/or model. Empty string = no filter on that dimension.
func (*Tracker) LastInput ¶
LastInput returns the InputTokens of the most recent event, or 0 when the tracker is empty. Used by the TUI to estimate current context fill — each turn re-sends the full conversation so the latest input count approximates live context usage.
func (*Tracker) Record ¶
Record computes cost from the active pricing table and appends an event. Unknown models fall back to zero-cost so unpriced local models still log token totals without polluting the dollar figure.
func (*Tracker) Reset ¶
func (t *Tracker) Reset()
Reset drops every recorded event so a fresh session starts at zero. Used by /new and /clear to bring the context-fill indicator back to 0%.
func (*Tracker) SetEstimatedInput ¶
SetEstimatedInput stores an override that LastInput returns until the next Record call overwrites it. /compact uses this so the context-fill indicator drops to the post-compact estimate immediately, instead of staying frozen at the prior turn's input until the next assistant reply.
Source Files