vial_llm

defmodule Vial.Evals.Suite do
  @moduledoc """
  Schema for evaluation test suites.

  A suite groups test cases together for a specific prompt, allowing
  systematic testing across multiple scenarios.
  """

  use Ecto.Schema
  import Ecto.Changeset

  @type t :: %__MODULE__{}

  @required_fields ~w(name prompt_id)a
  @optional_fields ~w()a

  @primary_key {:id, :binary_id, autogenerate: true}
  @foreign_key_type :binary_id

  schema "suites" do
    field :name, :string

    belongs_to :prompt, Vial.Prompts.Prompt
    has_many :test_cases, Vial.Evals.TestCase
    has_many :suite_runs, Vial.Evals.SuiteRun

    timestamps(type: :utc_datetime)
  end

  @doc """
  Changeset for creating or updating a suite.

  Validates that name and prompt_id are present.
  """
  @spec changeset(t(), map()) :: Ecto.Changeset.t()
  def changeset(suite, attrs) do
    suite
    |> cast(attrs, @required_fields ++ @optional_fields)
    |> validate_required(@required_fields)
  end
end