session

type Calibration struct {
	Time     string                        `json:"time"`
	Purposes map[string]CalibrationPurpose `json:"purposes"`
}

type CalibrationMetrics struct {
	Precision  float64 `json:"precision"`
	Recall     float64 `json:"recall"`
	Coverage   float64 `json:"coverage"`
	ReviewRate float64 `json:"review_rate"`
	N          int     `json:"n"`
}

type CalibrationPurpose struct {
	Contract       string             `json:"contract"`
	Model          string             `json:"model"`
	Banks          map[string]string  `json:"banks"`
	DevLabels      int                `json:"dev_labels"`
	HoldoutLabels  int                `json:"holdout_labels"`
	ImportedLabels int                `json:"imported_labels,omitempty"`
	Thresholds     map[string]float64 `json:"thresholds"`
	Dev            CalibrationMetrics `json:"dev"`
	Holdout        CalibrationMetrics `json:"holdout"`
}

type Flags struct {
	// BudgetUSD overrides [decisions].budget_usd when BudgetSet (or when
	// > 0). An explicit 0 is a cache-only run: no new call is made.
	BudgetUSD float64
	// BudgetSet reports that --budget was given, so an explicit 0 is
	// honored instead of read as "not set".
	BudgetSet bool
	// Decisions overrides the decision mode for this run: "", shadow or apply.
	Decisions string
}

type Options struct {
	Config    config.Config
	VaultPath string
	Topic     string
	// Command names the kb command for error messages ("kb classify").
	Command    string
	Flags      Flags
	HTTPClient *http.Client
	Now        func() time.Time
	Logger     *slog.Logger
}

type Session struct {
	Config    config.Config
	VaultPath string
	Topic     models.TopicInfo
	Settings  contract.Settings
	// Contract is the active contract, nil when none is accepted.
	Contract *contract.Contract
	Ref      decisions.TopicRef
	Engine   *decisions.Engine
	Gen      *generation.Client
	State    *corpus.StateStore
	Writer   *corpus.Writer
	Now      func() time.Time
	Logger   *slog.Logger
	Flags    Flags
	// contains filtered or unexported fields
}