in openai-java-core/src/main/kotlin/com/openai/services/blocking/chat/ChatCompletionServiceImpl.kt [50:288]
override fun create(
params: ChatCompletionCreateParams,
requestOptions: RequestOptions,
): ChatCompletion =
// post /chat/completions
withRawResponse().create(params, requestOptions).parse()
override fun createStreaming(
params: ChatCompletionCreateParams,
requestOptions: RequestOptions,
): StreamResponse<ChatCompletionChunk> =
// post /chat/completions
withRawResponse().createStreaming(params, requestOptions).parse()
override fun retrieve(
params: ChatCompletionRetrieveParams,
requestOptions: RequestOptions,
): ChatCompletion =
// get /chat/completions/{completion_id}
withRawResponse().retrieve(params, requestOptions).parse()
override fun update(
params: ChatCompletionUpdateParams,
requestOptions: RequestOptions,
): ChatCompletion =
// post /chat/completions/{completion_id}
withRawResponse().update(params, requestOptions).parse()
override fun list(
params: ChatCompletionListParams,
requestOptions: RequestOptions,
): ChatCompletionListPage =
// get /chat/completions
withRawResponse().list(params, requestOptions).parse()
override fun delete(
params: ChatCompletionDeleteParams,
requestOptions: RequestOptions,
): ChatCompletionDeleted =
// delete /chat/completions/{completion_id}
withRawResponse().delete(params, requestOptions).parse()
class WithRawResponseImpl internal constructor(private val clientOptions: ClientOptions) :
ChatCompletionService.WithRawResponse {
private val errorHandler: Handler<ErrorObject?> = errorHandler(clientOptions.jsonMapper)
private val messages: MessageService.WithRawResponse by lazy {
MessageServiceImpl.WithRawResponseImpl(clientOptions)
}
override fun messages(): MessageService.WithRawResponse = messages
private val createHandler: Handler<ChatCompletion> =
jsonHandler<ChatCompletion>(clientOptions.jsonMapper).withErrorHandler(errorHandler)
override fun create(
params: ChatCompletionCreateParams,
requestOptions: RequestOptions,
): HttpResponseFor<ChatCompletion> {
val request =
HttpRequest.builder()
.method(HttpMethod.POST)
.addPathSegments("chat", "completions")
.body(json(clientOptions.jsonMapper, params._body()))
.build()
.prepare(clientOptions, params, params.model().toString())
val requestOptions = requestOptions.applyDefaults(RequestOptions.from(clientOptions))
val response = clientOptions.httpClient.execute(request, requestOptions)
return response.parseable {
response
.use { createHandler.handle(it) }
.also {
if (requestOptions.responseValidation!!) {
it.validate()
}
}
}
}
private val createStreamingHandler: Handler<StreamResponse<ChatCompletionChunk>> =
sseHandler(clientOptions.jsonMapper)
.mapJson<ChatCompletionChunk>()
.withErrorHandler(errorHandler)
override fun createStreaming(
params: ChatCompletionCreateParams,
requestOptions: RequestOptions,
): HttpResponseFor<StreamResponse<ChatCompletionChunk>> {
val request =
HttpRequest.builder()
.method(HttpMethod.POST)
.addPathSegments("chat", "completions")
.body(
json(
clientOptions.jsonMapper,
params
._body()
.toBuilder()
.putAdditionalProperty("stream", JsonValue.from(true))
.build(),
)
)
.build()
.prepare(clientOptions, params, params.model().toString())
val requestOptions = requestOptions.applyDefaults(RequestOptions.from(clientOptions))
val response = clientOptions.httpClient.execute(request, requestOptions)
return response.parseable {
response
.let { createStreamingHandler.handle(it) }
.let { streamResponse ->
if (requestOptions.responseValidation!!) {
streamResponse.map { it.validate() }
} else {
streamResponse
}
}
}
}
private val retrieveHandler: Handler<ChatCompletion> =
jsonHandler<ChatCompletion>(clientOptions.jsonMapper).withErrorHandler(errorHandler)
override fun retrieve(
params: ChatCompletionRetrieveParams,
requestOptions: RequestOptions,
): HttpResponseFor<ChatCompletion> {
val request =
HttpRequest.builder()
.method(HttpMethod.GET)
.addPathSegments("chat", "completions", params._pathParam(0))
.build()
.prepare(clientOptions, params, null)
val requestOptions = requestOptions.applyDefaults(RequestOptions.from(clientOptions))
val response = clientOptions.httpClient.execute(request, requestOptions)
return response.parseable {
response
.use { retrieveHandler.handle(it) }
.also {
if (requestOptions.responseValidation!!) {
it.validate()
}
}
}
}
private val updateHandler: Handler<ChatCompletion> =
jsonHandler<ChatCompletion>(clientOptions.jsonMapper).withErrorHandler(errorHandler)
override fun update(
params: ChatCompletionUpdateParams,
requestOptions: RequestOptions,
): HttpResponseFor<ChatCompletion> {
val request =
HttpRequest.builder()
.method(HttpMethod.POST)
.addPathSegments("chat", "completions", params._pathParam(0))
.body(json(clientOptions.jsonMapper, params._body()))
.build()
.prepare(clientOptions, params, null)
val requestOptions = requestOptions.applyDefaults(RequestOptions.from(clientOptions))
val response = clientOptions.httpClient.execute(request, requestOptions)
return response.parseable {
response
.use { updateHandler.handle(it) }
.also {
if (requestOptions.responseValidation!!) {
it.validate()
}
}
}
}
private val listHandler: Handler<ChatCompletionListPageResponse> =
jsonHandler<ChatCompletionListPageResponse>(clientOptions.jsonMapper)
.withErrorHandler(errorHandler)
override fun list(
params: ChatCompletionListParams,
requestOptions: RequestOptions,
): HttpResponseFor<ChatCompletionListPage> {
val request =
HttpRequest.builder()
.method(HttpMethod.GET)
.addPathSegments("chat", "completions")
.build()
.prepare(
clientOptions,
params,
deploymentModel = params.model().map { it.toString() }.getOrNull(),
)
val requestOptions = requestOptions.applyDefaults(RequestOptions.from(clientOptions))
val response = clientOptions.httpClient.execute(request, requestOptions)
return response.parseable {
response
.use { listHandler.handle(it) }
.also {
if (requestOptions.responseValidation!!) {
it.validate()
}
}
.let {
ChatCompletionListPage.builder()
.service(ChatCompletionServiceImpl(clientOptions))
.params(params)
.response(it)
.build()
}
}
}
private val deleteHandler: Handler<ChatCompletionDeleted> =
jsonHandler<ChatCompletionDeleted>(clientOptions.jsonMapper)
.withErrorHandler(errorHandler)
override fun delete(
params: ChatCompletionDeleteParams,
requestOptions: RequestOptions,
): HttpResponseFor<ChatCompletionDeleted> {
val request =
HttpRequest.builder()
.method(HttpMethod.DELETE)
.addPathSegments("chat", "completions", params._pathParam(0))
.apply { params._body().ifPresent { body(json(clientOptions.jsonMapper, it)) } }
.build()
.prepare(clientOptions, params, null)
val requestOptions = requestOptions.applyDefaults(RequestOptions.from(clientOptions))
val response = clientOptions.httpClient.execute(request, requestOptions)
return response.parseable {
response
.use { deleteHandler.handle(it) }
.also {
if (requestOptions.responseValidation!!) {
it.validate()
}
}
}
}
}