com.simiacryptus.jopenai.models.ImageModels.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jo-penai Show documentation
A Java client for OpenAI's API
The newest version!
package com.simiacryptus.jopenai.models
import org.slf4j.LoggerFactory

@Suppress("unused")
enum class ImageModels(
    override val modelName: String,
    val maxPrompt: Int
) : OpenAIModel {

    DallE2("dall-e-2", 1000) {
        override fun pricing(width: Int, height: Int): Double {
            logger.debug("Calculating pricing for DallE2 with dimensions: ${width}x${height}")
            return when {
                width == 1024 && height == 1024 -> 0.02
                width == 512 && height == 512 -> 0.018
                width == 256 && height == 256 -> 0.016
                else -> {
                    logger.warn("Unsupported image size for DallE2: ${width}x${height}")
                    throw IllegalArgumentException("Unsupported image size: $width x $height")
                }
            }
        }
    },
    DallE3("dall-e-3", 1000) {
        override fun pricing(width: Int, height: Int): Double {
            logger.debug("Calculating pricing for DallE3 with dimensions: ${width}x${height}")
            return when {
                width == 1024 && height == 1024 -> 0.04
                width == 1024 && height == 1792 -> 0.08
                width == 1792 && height == 1024 -> 0.08
                else -> {
                    logger.warn("Unsupported image size for DallE3: ${width}x${height}")
                    throw IllegalArgumentException("Unsupported image size: $width x $height")
                }
            }
        }
    },
    DallE3_HD("dall-e-3", 1000) {
        override val quality: String = "hd"
        override fun pricing(width: Int, height: Int): Double {
            logger.debug("Calculating pricing for DallE3_HD with dimensions: ${width}x${height}")
            return when {
                width == 1024 && height == 1024 -> 0.08
                width == 1024 && height == 1792 -> 0.12
                width == 1792 && height == 1024 -> 0.12
                else -> {
                    logger.warn("Unsupported image size for DallE3_HD: ${width}x${height}")
                    throw IllegalArgumentException("Unsupported image size: $width x $height")
                }
            }
        }
    };

    open val quality: String = "standard"
    abstract fun pricing(width: Int, height: Int): Double


    companion object {
        private val logger = LoggerFactory.getLogger(ImageModels::class.java)
    }
}