commonMain.com.aallam.openai.client.internal.api.FineTunesApi.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.core.DeleteResponse
import com.aallam.openai.api.core.ListResponse
import com.aallam.openai.api.finetune.FineTune
import com.aallam.openai.api.finetune.FineTuneEvent
import com.aallam.openai.api.finetune.FineTuneId
import com.aallam.openai.api.finetune.FineTuneRequest
import com.aallam.openai.api.model.ModelId
import com.aallam.openai.client.FineTunes
import com.aallam.openai.client.internal.extension.streamEventsFrom
import com.aallam.openai.client.internal.http.HttpRequester
import com.aallam.openai.client.internal.http.perform
import io.ktor.client.call.body
import io.ktor.client.request.*
import io.ktor.client.statement.HttpResponse
import io.ktor.http.*
import kotlinx.coroutines.flow.Flow
import kotlinx.coroutines.flow.flow

/**
 * Implementation of [FineTunes].
 */
internal class FineTunesApi(private val requester: HttpRequester) : FineTunes {

    override suspend fun fineTune(request: FineTuneRequest): FineTune {
        return requester.perform {
            it.post {
                url(path = ApiPath.FineTunes)
                setBody(request)
                contentType(ContentType.Application.Json)
            }
        }
    }

    override suspend fun fineTune(fineTuneId: FineTuneId): FineTune? {
        val response = requester.perform {
            it.get { url(path = "${ApiPath.FineTunes}/${fineTuneId.id}") }
        }
        return if (response.status == HttpStatusCode.NotFound) null else response.body()
    }

    override suspend fun fineTunes(): List {
        return requester.perform> {
            it.get { url(path = ApiPath.FineTunes) }
        }.data
    }

    override suspend fun cancel(fineTuneId: FineTuneId): FineTune? {
        val response = requester.perform {
            it.post { url(path = "${ApiPath.FineTunes}/${fineTuneId.id}/cancel") }
        }
        return if (response.status == HttpStatusCode.NotFound) null else response.body()
    }

    override suspend fun fineTuneEvents(fineTuneId: FineTuneId): List {
        return requester.perform> {
            it.get { url(path = "${ApiPath.FineTunes}/${fineTuneId.id}/events") }
        }.data
    }

    override fun fineTuneEventsFlow(fineTuneId: FineTuneId): Flow {
        val request = HttpRequestBuilder().apply {
            method = HttpMethod.Get
            url(path = "${ApiPath.FineTunes}/${fineTuneId.id}/events") {
                parameters.append("stream", "true")
            }
            accept(ContentType.Text.EventStream)
            headers {
                append(HttpHeaders.CacheControl, "no-cache")
                append(HttpHeaders.Connection, "keep-alive")
            }
        }
        return flow {
            requester.perform(request) { response -> streamEventsFrom(response) }
        }
    }

    override suspend fun delete(fineTuneModel: ModelId): Boolean {
        val response = requester.perform {
            it.delete {
                url(path = "${ApiPath.Models}/${fineTuneModel.id}")
            }
        }
        return when (response.status) {
            HttpStatusCode.NotFound -> false
            else -> response.body().deleted
        }
    }
}