commonMain.com.aallam.openai.client.internal.api.ImagesApi.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of openai-client Show documentation
OpenAI API Kotlin Client
There is a newer version: 4.0.0-beta01
package com.aallam.openai.client.internal.api

import com.aallam.openai.api.ExperimentalOpenAI
import com.aallam.openai.api.core.ListResponse
import com.aallam.openai.api.image.*
import com.aallam.openai.client.Images
import com.aallam.openai.client.internal.extension.appendBinaryFile
import com.aallam.openai.client.internal.http.HttpRequester
import com.aallam.openai.client.internal.http.perform
import io.ktor.client.request.forms.FormBuilder
import io.ktor.client.request.forms.formData
import io.ktor.client.request.forms.submitFormWithBinaryData
import io.ktor.client.request.post
import io.ktor.client.request.setBody
import io.ktor.client.request.url
import io.ktor.http.ContentType
import io.ktor.http.contentType
import okio.FileSystem

internal class ImagesApi(
    private val requester: HttpRequester,
    private val fileSystem: FileSystem,
) : Images {

    @ExperimentalOpenAI
    override suspend fun image(creation: ImageCreationURL): List {
        return requester.perform> {
            it.post {
                url(path = ImagesGenerationV1)
                setBody(creation)
                contentType(ContentType.Application.Json)
            }
        }.data
    }

    @ExperimentalOpenAI
    override suspend fun image(creation: ImageCreationJSON): List {
        return requester.perform> {
            it.post {
                url(path = ImagesGenerationV1)
                setBody(creation)
                contentType(ContentType.Application.Json)
            }
        }.data
    }

    @ExperimentalOpenAI
    override suspend fun image(edit: ImageEditURL): List {
        return requester.perform> {
            it.submitFormWithBinaryData(url = ImagesEditsV1, formData = imageEditRequest(edit, ResponseFormat.url))
        }.data
    }

    @ExperimentalOpenAI
    override suspend fun image(edit: ImageEditJSON): List {
        return requester.perform> {
            it.submitFormWithBinaryData(
                url = ImagesEditsV1,
                formData = imageEditRequest(edit, ResponseFormat.base64Json)
            )
        }.data
    }

    /**
     * Build image edit request.
     */
    @ExperimentalOpenAI
    private fun imageEditRequest(edit: ImageEdit, responseFormat: ResponseFormat) = formData {
        appendBinaryFile(fileSystem, "image", edit.image)
        appendBinaryFile(fileSystem, "mask", edit.mask)
        append(key = "prompt", value = edit.prompt)
        imageRequest(responseFormat, edit)
    }

    @ExperimentalOpenAI
    override suspend fun image(variation: ImageVariationURL): List {
        return requester.perform> {
            it.submitFormWithBinaryData(
                url = ImagesVariantsV1,
                formData = imageVariantRequest(variation, ResponseFormat.url)
            )
        }.data
    }

    @ExperimentalOpenAI
    override suspend fun image(variation: ImageVariationJSON): List {
        return requester.perform> {
            it.submitFormWithBinaryData(
                url = ImagesVariantsV1,
                formData = imageVariantRequest(variation, ResponseFormat.base64Json)
            )
        }.data
    }

    /**
     * Build image variant request.
     */
    @ExperimentalOpenAI
    private fun imageVariantRequest(edit: ImageVariation, responseFormat: ResponseFormat) = formData {
        appendBinaryFile(fileSystem, "image", edit.image)
        imageRequest(responseFormat, edit)
    }

    /**
     * Append image request fields.
     */
    @ExperimentalOpenAI
    private fun FormBuilder.imageRequest(responseFormat: ResponseFormat, edit: ImageRequest) {
        append(key = "response_format", value = responseFormat.format)
        edit.n?.let { n -> append(key = "n", value = n) }
        edit.size?.let { dim -> append(key = "size", value = dim.size) }
        edit.user?.let { user -> append(key = "user", value = user) }
    }

    companion object {
        private const val ImagesGenerationV1 = "v1/images/generations"
        private const val ImagesEditsV1 = "v1/images/edits"
        private const val ImagesVariantsV1 = "v1/images/variations"
    }
}