shadow

type Budget interface {
	// Reserve books est under reservation id; false = over budget.
	Reserve(ctx context.Context, id string, est float64) (bool, error)
	// Settle swaps reservation id (of est) for charged.
	Settle(ctx context.Context, id string, est, charged float64) error
	// Spent is the settled spend.
	Spent(ctx context.Context) (float64, error)
}

type BudgetDB interface {
	Reserve(ctx context.Context, scope, id string, amount, limit float64, ttl time.Duration) (bool, error)
	Settle(ctx context.Context, scope, id string, amount, charged float64) error
	Spent(ctx context.Context, scope string) (float64, error)
}

type Config struct {
	Queue   int // bounded queue length (default 64)
	Workers int // default 4
	Token   string
	// Policy returns the current org (gateway.Snapshot.Get). Both sides of a
	// pair are resolved from it; jobs never name upstreams. Required.
	Policy func() (*policy.Org, error)
	// UpstreamHeaders are halo-shadow's own credentials, keyed by policy upstream
	// name; they are only ever sent to that upstream's URL.
	UpstreamHeaders map[string]map[string]string
	// BudgetUSD caps estimated spend (default DefaultBudgetUSD). Each job
	// reserves a worst-case estimate before anything is sent, so concurrency
	// can't overshoot; the reservation is settled to actual usage afterwards.
	BudgetUSD       float64
	PriceInPerMTok  float64
	PriceOutPerMTok float64
	HTTP            *http.Client // default: NewHTTPClient()
	Log             *slog.Logger
	// Budget, when set, replaces the per-process budget (BudgetUSD) with a
	// shared one (SharedBudget), so the cap holds across replicas.
	Budget Budget
	// Ping, when set, gates /readyz on the shared store being reachable.
	Ping func(context.Context) error
}

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

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

type Job struct {
	Experiment string            `json:"experiment"`
	Variant    string            `json:"variant"`
	Ring       string            `json:"ring,omitempty"`
	SessionID  string            `json:"session_id,omitempty"`
	Protocol   string            `json:"protocol"`
	Method     string            `json:"method"`
	Path       string            `json:"path"`
	Headers    map[string]string `json:"headers,omitempty"`
	Body       json.RawMessage   `json:"body"`
}

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

type Pair struct {
	ID         string          `json:"id"`
	Time       time.Time       `json:"time"`
	Experiment string          `json:"experiment"`
	Ring       string          `json:"ring,omitempty"`
	SessionID  string          `json:"session_id,omitempty"`
	Protocol   string          `json:"protocol"`
	Request    json.RawMessage `json:"request"`
	Control    Side            `json:"control"`
	Candidate  Side            `json:"candidate"`
	CostUSD    float64         `json:"cost_usd_estimate"`
}

type PairDB interface {
	AppendPair(ctx context.Context, id string, at time.Time, row []byte) error
	PrunePairs(ctx context.Context, cutoff time.Time) (int, error)
}

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

type SharedBudget struct {
	DB    BudgetDB
	Scope string  // budget name; change it to start a fresh budget
	Limit float64 // USD
}

type Side struct {
	Target       Target          `json:"target"`
	Status       int             `json:"status"`
	LatencyMS    int64           `json:"latency_ms"`
	InputTokens  int64           `json:"input_tokens"`
	OutputTokens int64           `json:"output_tokens"`
	Response     json.RawMessage `json:"response,omitempty"`
	Error        string          `json:"error,omitempty"`
}

type Store interface {
	Append(ctx context.Context, p Pair) error
}

type StoreOptions struct {
	// Key (32 bytes) enables AES-256-GCM at rest. Rows become
	// {"id","time","kid","enc"}; id/time stay clear for retention pruning.
	Key []byte
	// Retention: Prune drops rows older than this (0 = keep forever).
	Retention time.Duration
}

type Target struct {
	Variant  string `json:"variant,omitempty"`
	Upstream string `json:"upstream"` // policy upstream name
	Model    string `json:"model"`
}