defmodule ExOpenAi.Chat do
  @moduledoc """
  Represents a chat request to the OpenAI API.

  - [OpenAI API Docs](https://platform.openai.com/docs/api-reference/chat/create)

  ## Examples
  """

  defstruct id: nil,
            object: nil,
            created: nil,
            choices: nil,
            usage: nil

  use ExOpenAi.Resource, import: [:new, :create]

  @spec keep_it_simple(map, boolean()) :: list() | map()
  def keep_it_simple(response, true) do
    response
    |> Map.get(:choices)
    |> Enum.sort_by(fn choice -> choice.index end)
    |> Enum.map(fn choice -> choice.text end)
  end

  def keep_it_simple(response, _), do: response
end