datamodel

type Config struct {
	// Messages is a list of messages. The messages received from LaunchDarkly are uninterpolated.
	Messages []Message `json:"messages,omitempty"`

	// Meta is the config metadata.
	Meta Meta `json:"_ldMeta,omitempty"`

	// Model is the model.
	Model Model `json:"model,omitempty"`

	// Provider is the provider.
	Provider Provider `json:"provider,omitempty"`

	// Mode is the AI Config mode (e.g., "completion", "agent", "judge").
	Mode string `json:"mode,omitempty"`

	// EvaluationMetricKey is the evaluation metric key for judge mode configs.
	EvaluationMetricKey string `json:"evaluationMetricKey,omitempty"`

	// EvaluationMetricKeys is a deprecated array of evaluation metric keys.
	// Use EvaluationMetricKey instead.
	EvaluationMetricKeys []string `json:"evaluationMetricKeys,omitempty"`

	// JudgeConfiguration specifies judges attached to this config.
	JudgeConfiguration *JudgeConfiguration `json:"judgeConfiguration,omitempty"`
}

type EvalScore struct {
	// Score is the evaluation score between 0.0 and 1.0.
	Score float64 `json:"score"`

	// Reasoning is the explanation for the score.
	Reasoning string `json:"reasoning"`
}

type Judge struct {
	// Key is the judge config key.
	Key string `json:"key"`

	// SamplingRate is the probability (0.0-1.0) that the judge will evaluate.
	SamplingRate float64 `json:"samplingRate"`
}

type JudgeConfiguration struct {
	// Judges is a list of judges to evaluate this config's outputs.
	Judges []Judge `json:"judges,omitempty"`
}

type JudgeResponse struct {
	// Evals contains the evaluation results keyed by metric name.
	Evals map[string]EvalScore `json:"evals"`

	// Success indicates whether the evaluation completed successfully.
	Success bool `json:"success"`

	// JudgeConfigKey is the key of the judge config that produced this response.
	JudgeConfigKey string `json:"judgeConfigKey,omitempty"`

	// Error contains the error message if the evaluation failed.
	Error string `json:"error,omitempty"`
}

type Message struct {
	// Content is the message content.
	Content string `json:"content"`

	// Role is the role of the message.
	Role Role `json:"role"`
}

type Meta struct {
	// VariationKey is the variation key.
	VariationKey string `json:"variationKey,omitempty"`

	// Enabled is true if the config is enabled.
	Enabled bool `json:"enabled,omitempty"`

	// Version is the version of the Variation.
	Version *int `json:"version,omitempty"`
}

type Model struct {
	// Name identifies the model.
	Name string `json:"name"`

	// Parameters are the model parameters, generally provided by LaunchDarkly.
	Parameters map[string]ldvalue.Value `json:"parameters,omitempty"`

	// Custom are custom model parameters, generally provided by the user.
	Custom map[string]ldvalue.Value `json:"custom,omitempty"`
}

type Provider struct {
	// Name identifies the provider.
	Name string `json:"name"`
}

type Role string