openai4s.api.chat.ChatApi.scala Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of openai4s-api_sjs1_2.13 Show documentation

openai4s-api

The newest version!

package openai4s.api.chat

import openai4s.api.ApiCore
import openai4s.types.chat.{Chat, Response}
import org.http4s.client.dsl.Http4sClientDsl
import org.http4s.{Method, Request, Uri}

/** https://platform.openai.com/docs/api-reference/chat/create
  * @author Kevin Lee
  * @since 2023-04-07
  */
trait ChatApi[F[*]] {
  def completion(chat: Chat): F[Response]
}
object ChatApi {
  def apply[F[*]](chatCompletionsUri: Uri, apiCore: ApiCore[F]): ChatApi[F] =
    new ChatApiF[F](chatCompletionsUri, apiCore)

  private final class ChatApiF[F[*]](chatCompletionsUri: Uri, apiCore: ApiCore[F])
      extends ChatApi[F]
      with Http4sClientDsl[F] {

    import org.http4s.circe.CirceEntityCodec.*

    override def completion(chat: Chat): F[Response] = {
      val request       = Request[F](method = Method.POST, uri = chatCompletionsUri).withEntity(chat)
      val authedRequest = apiCore.preprocess(request)
      apiCore.httpClient.send[Response](authedRequest)
    }
  }
}