io.cequence.openaiscala.service.HandleOpenAIErrorCodes.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

import io.cequence.openaiscala._
import io.cequence.wsclient.service.WSClient

/**
 * Core WS stuff for OpenAI services.
 *
 * @since March
 *   2024
 */
trait HandleOpenAIErrorCodes extends WSClient {

  override protected def handleErrorCodes(
    httpCode: Int,
    message: String
  ): Nothing = {
    val errorMessage = s"Code ${httpCode} : ${message}"
    httpCode match {
      case 401 => throw new OpenAIScalaUnauthorizedException(errorMessage)
      case 429 => throw new OpenAIScalaRateLimitException(errorMessage)
      case 500 => throw new OpenAIScalaServerErrorException(errorMessage)
      case 503 => throw new OpenAIScalaEngineOverloadedException(errorMessage)
      case 400 =>
        if (
          message.contains("Please reduce your prompt; or completion length") ||
          message.contains("Please reduce the length of the messages")
        )
          throw new OpenAIScalaTokenCountExceededException(errorMessage)
        else
          throw new OpenAIScalaClientException(errorMessage)

      case _ => throw new OpenAIScalaClientException(errorMessage)
    }
  }
}