sttp.openai.requests.embeddings.EmbeddingsResponseBody.scala Maven / Gradle / Ivy

package sttp.openai.requests.embeddings

import sttp.openai.json.SnakePickle
import sttp.openai.requests.embeddings.EmbeddingsRequestBody.EmbeddingsModel

object EmbeddingsResponseBody {
  case class EmbeddingData(
      `object`: String,
      index: Int,
      embedding: Seq[Double]
  )
  object EmbeddingData {
    implicit val embeddingDataReader: SnakePickle.Reader[EmbeddingData] = SnakePickle.macroR
  }

  case class EmbeddingResponse(
      `object`: String,
      data: Seq[EmbeddingData],
      model: EmbeddingsModel,
      usage: Usage
  )

  object EmbeddingResponse {
    implicit val embeddingResponseDataReader: SnakePickle.Reader[EmbeddingResponse] = SnakePickle.macroR
  }
  case class Usage(
      promptTokens: Int,
      totalTokens: Int
  )

  object Usage {
    implicit val usageDataReader: SnakePickle.Reader[Usage] = SnakePickle.macroR
  }
}