sttp.openai.requests.images.Size.scala Maven / Gradle / Ivy

package sttp.openai.requests.images

import sttp.openai.json.SnakePickle

sealed abstract class Size(val value: String)

object Size {
  case object Small extends Size("256x256")

  case object Medium extends Size("512x512")

  case object Large extends Size("1024x1024")

  /** Use only as a workaround if API supports a format that's not yet predefined as a case object of Size. Otherwise, a custom format would
    * be rejected. See [[https://platform.openai.com/docs/api-reference/images/create-edit]] for current list of supported formats
    */
  case class Custom(customSize: String) extends Size(customSize)

  val values: Set[Size] = Set(Small, Medium, Large)

  implicit val sizeW: SnakePickle.Writer[Size] = SnakePickle
    .writer[ujson.Value]
    .comap[Size](_.value)
}