decisions

type Answer struct {
	Type       string             `json:"type"`
	Status     Status             `json:"status"`
	Reason     string             `json:"reason,omitempty"`
	Band       Band               `json:"band,omitempty"`
	Noul       *float64           `json:"noul,omitempty"`
	Choice     string             `json:"choice,omitempty"`
	Score      *float64           `json:"score,omitempty"`
	Probs      map[string]float64 `json:"probabilities,omitempty"`
	Confidence *float64           `json:"confidence,omitempty"`
	// Receipt is the receipts key of the request that produced the answer
	// (requests over 48 questions are split, so answers of one Result may
	// come from different receipts).
	Receipt string `json:"receipt,omitempty"`
}

type Band string

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

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

type GenerationSummary struct {
	Calls       int
	CacheHits   int
	Fallbacks   int
	Failures    int
	CostUSD     float64
	CostUnknown int
}

type Options struct {
	Config     config.DecisionsConfig
	APIKey     string
	APIURL     string
	HTTPClient *http.Client
	// Budget is the run budget; nil creates one from Config.BudgetUSD.
	Budget *Budget
	// Receipts is the receipts store; nil creates one. Share it with the
	// generation client through Engine.Receipts.
	Receipts *Receipts
	Now      func() time.Time
	// Sleep waits between retries; it must return ctx.Err() when ctx ends.
	Sleep  func(ctx context.Context, d time.Duration) error
	Logger *slog.Logger
}

type Purpose string

type PurposeSummary struct {
	Decided    int
	Bands      map[Band]int
	Undecided  map[string]int // by reason
	NotChecked map[string]int // by reason
}

type Receipt struct {
	Key           string                     `json:"key"`
	Time          string                     `json:"time"`
	Purpose       string                     `json:"purpose"`
	Subject       string                     `json:"subject"`
	Bank          string                     `json:"bank"`
	BankVersion   string                     `json:"bank_version"`
	BankHash      string                     `json:"bank_hash"`
	Contract      string                     `json:"contract"`
	Model         string                     `json:"model"`
	ReportedModel string                     `json:"reported_model"`
	Route         string                     `json:"route"`
	ResponseID    string                     `json:"response_id,omitempty"`
	Attempts      int                        `json:"attempts"`
	LatencyMS     int64                      `json:"latency_ms"`
	InputTokens   *int64                     `json:"input_tokens"`
	Cost          *float64                   `json:"cost"`
	CostUnknown   bool                       `json:"cost_unknown"`
	Status        Status                     `json:"status"`
	Reason        string                     `json:"reason,omitempty"`
	Questions     []string                   `json:"questions"`
	Answers       map[string]json.RawMessage `json:"answers"`
}

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

type Request struct {
	Topic     TopicRef
	Purpose   Purpose
	Subject   string
	Bank      *questions.Bank
	State     any
	Questions []questions.Q
}

type Result struct {
	Answers   map[string]Answer
	ReceiptID string
	CostUSD   float64
	CacheHit  bool
}

type Status string

type Summary struct {
	// Calls counts decision requests that reached the network.
	Calls int
	// CacheHits counts decision requests served from receipts.
	CacheHits int
	// CostUSD sums the reported cost of decision calls.
	CostUSD float64
	// CostUnknown counts decision calls with no reported cost (failed calls
	// included); their spend is never assumed to be zero.
	CostUnknown int
	// Purposes holds per-purpose answer counts.
	Purposes map[Purpose]PurposeSummary
	// Generation is the generation client's part of the run.
	Generation GenerationSummary
}

type Thresholds map[string]float64

type TopicRef struct {
	Slug       string
	Root       string
	Contract   string
	Thresholds Thresholds
	// Exclude are the topic.yaml `decisions.exclude` globs (topic-relative,
	// `**`-aware): a request whose Subject matches one is never sent to any
	// model (spec §14).
	Exclude []string
}