warm

type APIError struct {
	Status int
	Msg    string
}

type Config struct {
	Ollama    OllamaControl
	Shard     ShardControl
	Gauge     LoadGauge
	UsableMem func() uint64
	APIPort   int // shard API port (0 → 11434)

	// Desired seeds the persisted desired warm set (FileConfig.WarmDesired).
	Desired []string
	// PersistDesired persists the updated desired set (threaded through
	// admincore so writes serialize against the config mutex). nil → the
	// desired set is in-memory only (still survives within a daemon run).
	PersistDesired func([]string) error

	Interval time.Duration // supervisor tick (0 → 30s)
	Logger   *slog.Logger
}

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

type LoadGauge interface {
	Busy() bool
	WarmBudgetBytes(usableMem uint64) int64
}

type OllamaControl interface {
	// EnsureResident makes the model resident with a persistent
	// keep-alive (keep_alive: -1). When pull is true a missing model is
	// pulled first; when false a missing model is a no-op error (the
	// supervisor never blocks a tick on a multi-GB download).
	EnsureResident(ctx context.Context, model string, pull bool) error
	// Release unloads the model (keep_alive: 0). Idempotent.
	Release(ctx context.Context, model string) error
	// ModelSizeBytes returns the model's on-disk size (0 when unknown).
	ModelSizeBytes(ctx context.Context, model string) (uint64, error)
	// LoadedModels returns the names currently resident (/api/ps).
	LoadedModels(ctx context.Context) ([]string, error)
	// OnDisk reports whether the model is already downloaded (/api/tags).
	OnDisk(ctx context.Context, model string) bool
}

type ShardControl interface {
	ActiveModel() string
	Orchestrate(ctx context.Context, model string, apiPort int, extra []string) error
	Stop()
}

type WarmRequest struct {
	Model string `json:"model"`
	Mode  string `json:"mode"`
}

type WarmResponse struct {
	Status          string `json:"status"`
	ActiveModel     string `json:"active_model,omitempty"`
	Busy            bool   `json:"busy"`
	WarmBudgetBytes int64  `json:"warm_budget_bytes"`
}