systemone

type APIError struct {
	Status  int
	Message string
	Kind    error
}

type Answer struct {
	Type          string             `json:"type"`
	Choice        string             `json:"choice"`
	Probabilities map[string]float64 `json:"probabilities"`
	Confidence    float64            `json:"confidence"`
	Score         *float64           `json:"score,omitempty"`
}

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

type Criterion struct {
	Key         string
	Description *string
}

type DecisionModel struct {
	ID            string   `json:"id"`
	Name          string   `json:"name"`
	ContextWindow int      `json:"contextWindow,omitempty"`
	Capabilities  []string `json:"capabilities,omitempty"`
}

type DecisionProvider interface {
	Kind() ProviderKind
	Client() *Client
	// ListDecisionModels lists router candidates. showAll returns every model
	// the backend exposes instead of only decision models.
	ListDecisionModels(ctx context.Context, showAll bool) ([]DecisionModel, ListStatus, error)
	// Health probes reachability, auth, version, model presence and latency.
	Health(ctx context.Context, model string) HealthReport
	// ContextBudget returns the router model's context budget in tokens.
	ContextBudget(ctx context.Context, model string) int
}

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

type HealthReport struct {
	OK         bool         `json:"ok"`
	Kind       ProviderKind `json:"kind"`
	Model      string       `json:"model"`
	Version    string       `json:"version,omitempty"`
	Reachable  bool         `json:"reachable"`
	Authorized bool         `json:"authorized"`
	VersionOK  bool         `json:"versionOK"`
	ModelFound bool         `json:"modelPresent"`
	IsDecision bool         `json:"isDecisionModel"`
	Remote     bool         `json:"remote"`
	LatencyMs  int64        `json:"latencyMs"`
	Problems   []string     `json:"problems"`
}

type ListStatus string

type Options struct {
	BaseURL string            // root URL, without /v1/systemone
	APIKey  string            // optional; sent as "Authorization: Bearer <key>"
	Headers map[string]string // extra headers sent on every request
	Timeout time.Duration     // per-call bound; DefaultTimeout when zero
	// HTTPClient overrides the transport (tests). Timeout is enforced through
	// the request context, not through HTTPClient.Timeout.
	HTTPClient *http.Client

	// Provider-level knobs (ignored by the raw client).
	KeepAlive     string // Ollama keep_alive for probes/warm-up; default "30m"
	ContextBudget int    // default context budget (tokens) for custom providers
}

type ProviderKind string

type Question struct {
	Type         string // "choice", "noul" or "score"
	Instructions any    // string | object | array; required by the server
	Criteria     []Criterion
	// Extra holds additional type-specific fields merged into the question
	// object (e.g. score bounds).
	Extra map[string]any
}

type Request struct {
	Model     string              `json:"model"`
	State     any                 `json:"state"`
	Questions map[string]Question `json:"questions"`
	KeepAlive string              `json:"keep_alive,omitempty"`
}

type Response struct {
	Model   string            `json:"model"`
	Answers map[string]Answer `json:"answers"`
	Usage   Usage             `json:"usage"`
}

type Usage struct {
	InputTokens  int      `json:"input_tokens"`
	OutputTokens int      `json:"output_tokens"`
	Cost         *float64 `json:"cost,omitempty"`
}