io.cequence.openaiscala.service.impl.RunBodyMaker.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of openai-scala-client_2.12 Show documentation

Scala client for OpenAI API implemented using Play WS lib.

There is a newer version: 1.1.0.RC.2

Show newest version

package io.cequence.openaiscala.service.impl

import io.cequence.openaiscala.JsonFormats._
import io.cequence.openaiscala.domain.settings.CreateRunSettings
import io.cequence.wsclient.service.WSClient
import play.api.libs.json.{JsValue, Json}

trait RunBodyMaker {

  this: WSClient =>

  protected def createBodyParamsForRun(
    settings: CreateRunSettings,
    stream: Boolean
  ): Seq[(Param, Option[JsValue])] = {

    jsonBodyParams(
      Param.model -> settings.model,
      Param.temperature -> settings.temperature,
      Param.top_p -> settings.topP,
      Param.stream -> Some(stream),
      Param.max_prompt_tokens -> settings.maxPromptTokens,
      Param.response_format -> settings.responseFormat.map { format =>
        Json.toJson(format)
      }
    )
  }
}