systemone

type APIError struct {
	// StatusCode is the HTTP status.
	StatusCode int
	// Body is a bounded, control-character-free excerpt of the response body.
	Body string
	// RetryAfter is the server's Retry-After hint, or zero when absent.
	RetryAfter time.Duration
}

type Answer interface {
	// Kind returns the wire kind: "noul", "choice", or "score".
	Kind() string
	// contains filtered or unexported methods
}

type Choice struct {
	// Instructions describes what to decide about the state.
	Instructions Content
	// Options maps each selectable label to an optional description. A nil
	// description leaves the label undescribed. At least two labels are
	// required.
	Options map[string]Content
}

type ChoiceAnswer struct {
	// Choice is the selected label, always one of the question's options.
	Choice string
	// Probabilities holds one probability per option label.
	Probabilities map[string]float64
	// Confidence is the model's overall confidence in the selection, in
	// [0, 1].
	Confidence float64
}

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

type Content any

type Noul struct {
	// Instructions is the question to answer about the state.
	Instructions Content
	// True and False optionally describe what a yes and a no mean. Either may
	// be nil.
	True  Content
	False Content
}

type NoulAnswer struct {
	// Probability is the calibrated probability, in [0, 1], that the answer
	// is yes.
	Probability float64
}

type Option func(*Client) error

type Question interface {
	json.Marshaler
	// Kind returns the wire kind: "noul", "choice", or "score".
	Kind() string
	// contains filtered or unexported methods
}

type Request struct {
	// Model is the model id or alias, for example [ModelJevLatest].
	Model string `json:"model"`
	// State is the content every question is evaluated against.
	State Content `json:"state"`
	// Questions maps caller-chosen ids to questions. Answers come back under
	// the same ids.
	Questions map[string]Question `json:"questions"`
}

type Response struct {
	// Model is the model that served the request, with aliases resolved as
	// the API reports them.
	Model string
	// Answers holds one answer per question id in the request.
	Answers map[string]Answer
	// Usage is the token accounting for the call.
	Usage Usage
}

type Score struct {
	// Instructions describes what to rate about the state.
	Instructions Content
	// Levels is the rubric, lowest level first. At least two levels are
	// required.
	Levels []Content
}

type ScoreAnswer struct {
	// Score is the probability-weighted position on the rubric, in
	// [0, len(levels)-1].
	Score float64
	// Legend maps each level key used in Probabilities to the level's
	// description as the API rendered it.
	Legend map[string]string
	// Probabilities holds one probability per rubric level, keyed as in
	// Legend.
	Probabilities map[string]float64
	// Confidence is the model's overall confidence in the rating, in [0, 1].
	Confidence float64
}

type Usage struct {
	InputTokens  int64 `json:"input_tokens"`
	OutputTokens int64 `json:"output_tokens"`
}