Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 16 additions & 0 deletions openai-java-core/src/main/kotlin/com/openai/core/Params.kt
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
package com.openai.core

import com.openai.core.http.Headers
import com.openai.core.http.QueryParams

/** An interface representing parameters passed to a service method. */
interface Params {
/** The full set of headers in the parameters, including both fixed and additional headers. */
fun _headers(): Headers

/**
* The full set of query params in the parameters, including both fixed and additional query
* params.
*/
fun _queryParams(): QueryParams
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
@file:JvmName("PrepareRequest")

package com.openai.core

import com.openai.core.http.HttpRequest
import java.util.concurrent.CompletableFuture

@JvmSynthetic
internal fun HttpRequest.prepare(clientOptions: ClientOptions, params: Params): HttpRequest =
toBuilder()
.putAllQueryParams(clientOptions.queryParams)
.replaceAllQueryParams(params._queryParams())
.putAllHeaders(clientOptions.headers)
.replaceAllHeaders(params._headers())
.build()

@JvmSynthetic
internal fun HttpRequest.prepareAsync(
clientOptions: ClientOptions,
params: Params
): CompletableFuture<HttpRequest> =
// This async version exists to make it easier to add async specific preparation logic in the
// future.
CompletableFuture.completedFuture(prepare(clientOptions, params))
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ package com.openai.models

import com.openai.core.JsonValue
import com.openai.core.NoAutoDetect
import com.openai.core.Params
import com.openai.core.checkRequired
import com.openai.core.http.Headers
import com.openai.core.http.QueryParams
Expand All @@ -22,7 +23,7 @@ private constructor(
private val additionalHeaders: Headers,
private val additionalQueryParams: QueryParams,
private val additionalBodyProperties: Map<String, JsonValue>,
) {
) : Params {

fun batchId(): String = batchId

Expand All @@ -33,12 +34,12 @@ private constructor(
fun _additionalBodyProperties(): Map<String, JsonValue> = additionalBodyProperties

@JvmSynthetic
internal fun getBody(): Optional<Map<String, JsonValue>> =
internal fun _body(): Optional<Map<String, JsonValue>> =
Optional.ofNullable(additionalBodyProperties.ifEmpty { null })

@JvmSynthetic internal fun getHeaders(): Headers = additionalHeaders
override fun _headers(): Headers = additionalHeaders

@JvmSynthetic internal fun getQueryParams(): QueryParams = additionalQueryParams
override fun _queryParams(): QueryParams = additionalQueryParams

fun getPathParam(index: Int): String {
return when (index) {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@ import com.openai.core.JsonField
import com.openai.core.JsonMissing
import com.openai.core.JsonValue
import com.openai.core.NoAutoDetect
import com.openai.core.Params
import com.openai.core.checkRequired
import com.openai.core.http.Headers
import com.openai.core.http.QueryParams
Expand All @@ -27,7 +28,7 @@ private constructor(
private val body: BatchCreateBody,
private val additionalHeaders: Headers,
private val additionalQueryParams: QueryParams,
) {
) : Params {

/**
* The time frame within which the batch should be processed. Currently only `24h` is supported.
Expand Down Expand Up @@ -91,11 +92,11 @@ private constructor(

fun _additionalQueryParams(): QueryParams = additionalQueryParams

@JvmSynthetic internal fun getBody(): BatchCreateBody = body
@JvmSynthetic internal fun _body(): BatchCreateBody = body

@JvmSynthetic internal fun getHeaders(): Headers = additionalHeaders
override fun _headers(): Headers = additionalHeaders

@JvmSynthetic internal fun getQueryParams(): QueryParams = additionalQueryParams
override fun _queryParams(): QueryParams = additionalQueryParams

@NoAutoDetect
class BatchCreateBody
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
package com.openai.models

import com.openai.core.NoAutoDetect
import com.openai.core.Params
import com.openai.core.http.Headers
import com.openai.core.http.QueryParams
import java.util.Objects
Expand All @@ -15,7 +16,7 @@ private constructor(
private val limit: Long?,
private val additionalHeaders: Headers,
private val additionalQueryParams: QueryParams,
) {
) : Params {

/**
* A cursor for use in pagination. `after` is an object ID that defines your place in the list.
Expand All @@ -34,10 +35,9 @@ private constructor(

fun _additionalQueryParams(): QueryParams = additionalQueryParams

@JvmSynthetic internal fun getHeaders(): Headers = additionalHeaders
override fun _headers(): Headers = additionalHeaders

@JvmSynthetic
internal fun getQueryParams(): QueryParams {
override fun _queryParams(): QueryParams {
val queryParams = QueryParams.builder()
this.after?.let { queryParams.put("after", listOf(it.toString())) }
this.limit?.let { queryParams.put("limit", listOf(it.toString())) }
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
package com.openai.models

import com.openai.core.NoAutoDetect
import com.openai.core.Params
import com.openai.core.checkRequired
import com.openai.core.http.Headers
import com.openai.core.http.QueryParams
Expand All @@ -14,17 +15,17 @@ private constructor(
private val batchId: String,
private val additionalHeaders: Headers,
private val additionalQueryParams: QueryParams,
) {
) : Params {

fun batchId(): String = batchId

fun _additionalHeaders(): Headers = additionalHeaders

fun _additionalQueryParams(): QueryParams = additionalQueryParams

@JvmSynthetic internal fun getHeaders(): Headers = additionalHeaders
override fun _headers(): Headers = additionalHeaders

@JvmSynthetic internal fun getQueryParams(): QueryParams = additionalQueryParams
override fun _queryParams(): QueryParams = additionalQueryParams

fun getPathParam(index: Int): String {
return when (index) {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ import com.openai.core.JsonField
import com.openai.core.JsonMissing
import com.openai.core.JsonValue
import com.openai.core.NoAutoDetect
import com.openai.core.Params
import com.openai.core.checkRequired
import com.openai.core.http.Headers
import com.openai.core.http.QueryParams
Expand All @@ -25,7 +26,7 @@ private constructor(
private val body: BetaAssistantCreateBody,
private val additionalHeaders: Headers,
private val additionalQueryParams: QueryParams,
) {
) : Params {

/**
* ID of the model to use. You can use the
Expand Down Expand Up @@ -178,11 +179,11 @@ private constructor(

fun _additionalQueryParams(): QueryParams = additionalQueryParams

@JvmSynthetic internal fun getBody(): BetaAssistantCreateBody = body
@JvmSynthetic internal fun _body(): BetaAssistantCreateBody = body

@JvmSynthetic internal fun getHeaders(): Headers = additionalHeaders
override fun _headers(): Headers = additionalHeaders

@JvmSynthetic internal fun getQueryParams(): QueryParams = additionalQueryParams
override fun _queryParams(): QueryParams = additionalQueryParams

@NoAutoDetect
class BetaAssistantCreateBody
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@ package com.openai.models

import com.openai.core.JsonValue
import com.openai.core.NoAutoDetect
import com.openai.core.Params
import com.openai.core.checkRequired
import com.openai.core.http.Headers
import com.openai.core.http.QueryParams
Expand All @@ -18,7 +19,7 @@ private constructor(
private val additionalHeaders: Headers,
private val additionalQueryParams: QueryParams,
private val additionalBodyProperties: Map<String, JsonValue>,
) {
) : Params {

fun assistantId(): String = assistantId

Expand All @@ -29,12 +30,12 @@ private constructor(
fun _additionalBodyProperties(): Map<String, JsonValue> = additionalBodyProperties

@JvmSynthetic
internal fun getBody(): Optional<Map<String, JsonValue>> =
internal fun _body(): Optional<Map<String, JsonValue>> =
Optional.ofNullable(additionalBodyProperties.ifEmpty { null })

@JvmSynthetic internal fun getHeaders(): Headers = additionalHeaders
override fun _headers(): Headers = additionalHeaders

@JvmSynthetic internal fun getQueryParams(): QueryParams = additionalQueryParams
override fun _queryParams(): QueryParams = additionalQueryParams

fun getPathParam(index: Int): String {
return when (index) {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ import com.fasterxml.jackson.annotation.JsonCreator
import com.openai.core.Enum
import com.openai.core.JsonField
import com.openai.core.NoAutoDetect
import com.openai.core.Params
import com.openai.core.http.Headers
import com.openai.core.http.QueryParams
import com.openai.errors.OpenAIInvalidDataException
Expand All @@ -21,7 +22,7 @@ private constructor(
private val order: Order?,
private val additionalHeaders: Headers,
private val additionalQueryParams: QueryParams,
) {
) : Params {

/**
* A cursor for use in pagination. `after` is an object ID that defines your place in the list.
Expand Down Expand Up @@ -53,10 +54,9 @@ private constructor(

fun _additionalQueryParams(): QueryParams = additionalQueryParams

@JvmSynthetic internal fun getHeaders(): Headers = additionalHeaders
override fun _headers(): Headers = additionalHeaders

@JvmSynthetic
internal fun getQueryParams(): QueryParams {
override fun _queryParams(): QueryParams {
val queryParams = QueryParams.builder()
this.after?.let { queryParams.put("after", listOf(it.toString())) }
this.before?.let { queryParams.put("before", listOf(it.toString())) }
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
package com.openai.models

import com.openai.core.NoAutoDetect
import com.openai.core.Params
import com.openai.core.checkRequired
import com.openai.core.http.Headers
import com.openai.core.http.QueryParams
Expand All @@ -14,17 +15,17 @@ private constructor(
private val assistantId: String,
private val additionalHeaders: Headers,
private val additionalQueryParams: QueryParams,
) {
) : Params {

fun assistantId(): String = assistantId

fun _additionalHeaders(): Headers = additionalHeaders

fun _additionalQueryParams(): QueryParams = additionalQueryParams

@JvmSynthetic internal fun getHeaders(): Headers = additionalHeaders
override fun _headers(): Headers = additionalHeaders

@JvmSynthetic internal fun getQueryParams(): QueryParams = additionalQueryParams
override fun _queryParams(): QueryParams = additionalQueryParams

fun getPathParam(index: Int): String {
return when (index) {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ import com.openai.core.JsonField
import com.openai.core.JsonMissing
import com.openai.core.JsonValue
import com.openai.core.NoAutoDetect
import com.openai.core.Params
import com.openai.core.checkRequired
import com.openai.core.http.Headers
import com.openai.core.http.QueryParams
Expand All @@ -26,7 +27,7 @@ private constructor(
private val body: BetaAssistantUpdateBody,
private val additionalHeaders: Headers,
private val additionalQueryParams: QueryParams,
) {
) : Params {

fun assistantId(): String = assistantId

Expand Down Expand Up @@ -181,11 +182,11 @@ private constructor(

fun _additionalQueryParams(): QueryParams = additionalQueryParams

@JvmSynthetic internal fun getBody(): BetaAssistantUpdateBody = body
@JvmSynthetic internal fun _body(): BetaAssistantUpdateBody = body

@JvmSynthetic internal fun getHeaders(): Headers = additionalHeaders
override fun _headers(): Headers = additionalHeaders

@JvmSynthetic internal fun getQueryParams(): QueryParams = additionalQueryParams
override fun _queryParams(): QueryParams = additionalQueryParams

fun getPathParam(index: Int): String {
return when (index) {
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ import com.openai.core.JsonField
import com.openai.core.JsonMissing
import com.openai.core.JsonValue
import com.openai.core.NoAutoDetect
import com.openai.core.Params
import com.openai.core.checkRequired
import com.openai.core.getOrThrow
import com.openai.core.http.Headers
Expand All @@ -38,7 +39,7 @@ private constructor(
private val body: BetaThreadCreateAndRunBody,
private val additionalHeaders: Headers,
private val additionalQueryParams: QueryParams,
) {
) : Params {

/**
* The ID of the [assistant](https://platform.openai.com/docs/api-reference/assistants) to use
Expand Down Expand Up @@ -275,11 +276,11 @@ private constructor(

fun _additionalQueryParams(): QueryParams = additionalQueryParams

@JvmSynthetic internal fun getBody(): BetaThreadCreateAndRunBody = body
@JvmSynthetic internal fun _body(): BetaThreadCreateAndRunBody = body

@JvmSynthetic internal fun getHeaders(): Headers = additionalHeaders
override fun _headers(): Headers = additionalHeaders

@JvmSynthetic internal fun getQueryParams(): QueryParams = additionalQueryParams
override fun _queryParams(): QueryParams = additionalQueryParams

@NoAutoDetect
class BetaThreadCreateAndRunBody
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ import com.openai.core.JsonField
import com.openai.core.JsonMissing
import com.openai.core.JsonValue
import com.openai.core.NoAutoDetect
import com.openai.core.Params
import com.openai.core.checkRequired
import com.openai.core.getOrThrow
import com.openai.core.http.Headers
Expand All @@ -38,7 +39,7 @@ private constructor(
private val body: BetaThreadCreateBody,
private val additionalHeaders: Headers,
private val additionalQueryParams: QueryParams,
) {
) : Params {

/**
* A list of [messages](https://platform.openai.com/docs/api-reference/messages) to start the
Expand Down Expand Up @@ -79,11 +80,11 @@ private constructor(

fun _additionalQueryParams(): QueryParams = additionalQueryParams

@JvmSynthetic internal fun getBody(): BetaThreadCreateBody = body
@JvmSynthetic internal fun _body(): BetaThreadCreateBody = body

@JvmSynthetic internal fun getHeaders(): Headers = additionalHeaders
override fun _headers(): Headers = additionalHeaders

@JvmSynthetic internal fun getQueryParams(): QueryParams = additionalQueryParams
override fun _queryParams(): QueryParams = additionalQueryParams

@NoAutoDetect
class BetaThreadCreateBody
Expand Down
Loading