From 259169b2ed474ba1fe0ebcc4d4ba194d6ff2e897 Mon Sep 17 00:00:00 2001 From: stainless-bot Date: Mon, 28 Oct 2024 14:30:59 +0000 Subject: [PATCH] chore(api): manual updates --- .../api/models/ChatCompletionContentPart.kt | 166 ++++++++++++++ .../api/models/FunctionInvokeParams.kt | 178 --------------- .../braintrustdata/api/models/PromptData.kt | 203 ------------------ 3 files changed, 166 insertions(+), 381 deletions(-) create mode 100644 braintrust-java-core/src/main/kotlin/com/braintrustdata/api/models/ChatCompletionContentPart.kt diff --git a/braintrust-java-core/src/main/kotlin/com/braintrustdata/api/models/ChatCompletionContentPart.kt b/braintrust-java-core/src/main/kotlin/com/braintrustdata/api/models/ChatCompletionContentPart.kt new file mode 100644 index 00000000..35a3f312 --- /dev/null +++ b/braintrust-java-core/src/main/kotlin/com/braintrustdata/api/models/ChatCompletionContentPart.kt @@ -0,0 +1,166 @@ +// File generated from our OpenAPI spec by Stainless. + +package com.braintrustdata.api.models + +import com.braintrustdata.api.core.BaseDeserializer +import com.braintrustdata.api.core.BaseSerializer +import com.braintrustdata.api.core.JsonValue +import com.braintrustdata.api.core.getOrThrow +import com.braintrustdata.api.errors.BraintrustInvalidDataException +import com.fasterxml.jackson.core.JsonGenerator +import com.fasterxml.jackson.core.ObjectCodec +import com.fasterxml.jackson.databind.JsonNode +import com.fasterxml.jackson.databind.SerializerProvider +import com.fasterxml.jackson.databind.annotation.JsonDeserialize +import com.fasterxml.jackson.databind.annotation.JsonSerialize +import com.fasterxml.jackson.module.kotlin.jacksonTypeRef +import java.util.Objects +import java.util.Optional + +@JsonDeserialize(using = ChatCompletionContentPart.Deserializer::class) +@JsonSerialize(using = ChatCompletionContentPart.Serializer::class) +class ChatCompletionContentPart +private constructor( + private val chatCompletionContentPartText: ChatCompletionContentPartText? = null, + private val chatCompletionContentPartImage: ChatCompletionContentPartImage? = null, + private val _json: JsonValue? = null, +) { + + private var validated: Boolean = false + + fun chatCompletionContentPartText(): Optional = + Optional.ofNullable(chatCompletionContentPartText) + + fun chatCompletionContentPartImage(): Optional = + Optional.ofNullable(chatCompletionContentPartImage) + + fun isChatCompletionContentPartText(): Boolean = chatCompletionContentPartText != null + + fun isChatCompletionContentPartImage(): Boolean = chatCompletionContentPartImage != null + + fun asChatCompletionContentPartText(): ChatCompletionContentPartText = + chatCompletionContentPartText.getOrThrow("chatCompletionContentPartText") + + fun asChatCompletionContentPartImage(): ChatCompletionContentPartImage = + chatCompletionContentPartImage.getOrThrow("chatCompletionContentPartImage") + + fun _json(): Optional = Optional.ofNullable(_json) + + fun accept(visitor: Visitor): T { + return when { + chatCompletionContentPartText != null -> + visitor.visitChatCompletionContentPartText(chatCompletionContentPartText) + chatCompletionContentPartImage != null -> + visitor.visitChatCompletionContentPartImage(chatCompletionContentPartImage) + else -> visitor.unknown(_json) + } + } + + fun validate(): ChatCompletionContentPart = apply { + if (!validated) { + if (chatCompletionContentPartText == null && chatCompletionContentPartImage == null) { + throw BraintrustInvalidDataException("Unknown ChatCompletionContentPart: $_json") + } + chatCompletionContentPartText?.validate() + chatCompletionContentPartImage?.validate() + validated = true + } + } + + override fun equals(other: Any?): Boolean { + if (this === other) { + return true + } + + return /* spotless:off */ other is ChatCompletionContentPart && this.chatCompletionContentPartText == other.chatCompletionContentPartText && this.chatCompletionContentPartImage == other.chatCompletionContentPartImage /* spotless:on */ + } + + override fun hashCode(): Int { + return /* spotless:off */ Objects.hash(chatCompletionContentPartText, chatCompletionContentPartImage) /* spotless:on */ + } + + override fun toString(): String { + return when { + chatCompletionContentPartText != null -> + "ChatCompletionContentPart{chatCompletionContentPartText=$chatCompletionContentPartText}" + chatCompletionContentPartImage != null -> + "ChatCompletionContentPart{chatCompletionContentPartImage=$chatCompletionContentPartImage}" + _json != null -> "ChatCompletionContentPart{_unknown=$_json}" + else -> throw IllegalStateException("Invalid ChatCompletionContentPart") + } + } + + companion object { + + @JvmStatic + fun ofChatCompletionContentPartText( + chatCompletionContentPartText: ChatCompletionContentPartText + ) = ChatCompletionContentPart(chatCompletionContentPartText = chatCompletionContentPartText) + + @JvmStatic + fun ofChatCompletionContentPartImage( + chatCompletionContentPartImage: ChatCompletionContentPartImage + ) = + ChatCompletionContentPart( + chatCompletionContentPartImage = chatCompletionContentPartImage + ) + } + + interface Visitor { + + fun visitChatCompletionContentPartText( + chatCompletionContentPartText: ChatCompletionContentPartText + ): T + + fun visitChatCompletionContentPartImage( + chatCompletionContentPartImage: ChatCompletionContentPartImage + ): T + + fun unknown(json: JsonValue?): T { + throw BraintrustInvalidDataException("Unknown ChatCompletionContentPart: $json") + } + } + + class Deserializer : + BaseDeserializer(ChatCompletionContentPart::class) { + + override fun ObjectCodec.deserialize(node: JsonNode): ChatCompletionContentPart { + val json = JsonValue.fromJsonNode(node) + + tryDeserialize(node, jacksonTypeRef()) { it.validate() } + ?.let { + return ChatCompletionContentPart( + chatCompletionContentPartText = it, + _json = json + ) + } + tryDeserialize(node, jacksonTypeRef()) { it.validate() } + ?.let { + return ChatCompletionContentPart( + chatCompletionContentPartImage = it, + _json = json + ) + } + + return ChatCompletionContentPart(_json = json) + } + } + + class Serializer : BaseSerializer(ChatCompletionContentPart::class) { + + override fun serialize( + value: ChatCompletionContentPart, + generator: JsonGenerator, + provider: SerializerProvider + ) { + when { + value.chatCompletionContentPartText != null -> + generator.writeObject(value.chatCompletionContentPartText) + value.chatCompletionContentPartImage != null -> + generator.writeObject(value.chatCompletionContentPartImage) + value._json != null -> generator.writeObject(value._json) + else -> throw IllegalStateException("Invalid ChatCompletionContentPart") + } + } + } +} diff --git a/braintrust-java-core/src/main/kotlin/com/braintrustdata/api/models/FunctionInvokeParams.kt b/braintrust-java-core/src/main/kotlin/com/braintrustdata/api/models/FunctionInvokeParams.kt index 1e78cd12..6c16f962 100644 --- a/braintrust-java-core/src/main/kotlin/com/braintrustdata/api/models/FunctionInvokeParams.kt +++ b/braintrust-java-core/src/main/kotlin/com/braintrustdata/api/models/FunctionInvokeParams.kt @@ -1018,184 +1018,6 @@ constructor( } } } - - @JsonDeserialize(using = ChatCompletionContentPart.Deserializer::class) - @JsonSerialize(using = ChatCompletionContentPart.Serializer::class) - class ChatCompletionContentPart - private constructor( - private val chatCompletionContentPartText: ChatCompletionContentPartText? = - null, - private val chatCompletionContentPartImage: ChatCompletionContentPartImage? = - null, - private val _json: JsonValue? = null, - ) { - - private var validated: Boolean = false - - fun chatCompletionContentPartText(): Optional = - Optional.ofNullable(chatCompletionContentPartText) - - fun chatCompletionContentPartImage(): Optional = - Optional.ofNullable(chatCompletionContentPartImage) - - fun isChatCompletionContentPartText(): Boolean = - chatCompletionContentPartText != null - - fun isChatCompletionContentPartImage(): Boolean = - chatCompletionContentPartImage != null - - fun asChatCompletionContentPartText(): ChatCompletionContentPartText = - chatCompletionContentPartText.getOrThrow("chatCompletionContentPartText") - - fun asChatCompletionContentPartImage(): ChatCompletionContentPartImage = - chatCompletionContentPartImage.getOrThrow("chatCompletionContentPartImage") - - fun _json(): Optional = Optional.ofNullable(_json) - - fun accept(visitor: Visitor): T { - return when { - chatCompletionContentPartText != null -> - visitor.visitChatCompletionContentPartText( - chatCompletionContentPartText - ) - chatCompletionContentPartImage != null -> - visitor.visitChatCompletionContentPartImage( - chatCompletionContentPartImage - ) - else -> visitor.unknown(_json) - } - } - - fun validate(): ChatCompletionContentPart = apply { - if (!validated) { - if ( - chatCompletionContentPartText == null && - chatCompletionContentPartImage == null - ) { - throw BraintrustInvalidDataException( - "Unknown ChatCompletionContentPart: $_json" - ) - } - chatCompletionContentPartText?.validate() - chatCompletionContentPartImage?.validate() - validated = true - } - } - - override fun equals(other: Any?): Boolean { - if (this === other) { - return true - } - - return /* spotless:off */ other is ChatCompletionContentPart && this.chatCompletionContentPartText == other.chatCompletionContentPartText && this.chatCompletionContentPartImage == other.chatCompletionContentPartImage /* spotless:on */ - } - - override fun hashCode(): Int { - return /* spotless:off */ Objects.hash(chatCompletionContentPartText, chatCompletionContentPartImage) /* spotless:on */ - } - - override fun toString(): String { - return when { - chatCompletionContentPartText != null -> - "ChatCompletionContentPart{chatCompletionContentPartText=$chatCompletionContentPartText}" - chatCompletionContentPartImage != null -> - "ChatCompletionContentPart{chatCompletionContentPartImage=$chatCompletionContentPartImage}" - _json != null -> "ChatCompletionContentPart{_unknown=$_json}" - else -> throw IllegalStateException("Invalid ChatCompletionContentPart") - } - } - - companion object { - - @JvmStatic - fun ofChatCompletionContentPartText( - chatCompletionContentPartText: ChatCompletionContentPartText - ) = - ChatCompletionContentPart( - chatCompletionContentPartText = chatCompletionContentPartText - ) - - @JvmStatic - fun ofChatCompletionContentPartImage( - chatCompletionContentPartImage: ChatCompletionContentPartImage - ) = - ChatCompletionContentPart( - chatCompletionContentPartImage = chatCompletionContentPartImage - ) - } - - interface Visitor { - - fun visitChatCompletionContentPartText( - chatCompletionContentPartText: ChatCompletionContentPartText - ): T - - fun visitChatCompletionContentPartImage( - chatCompletionContentPartImage: ChatCompletionContentPartImage - ): T - - fun unknown(json: JsonValue?): T { - throw BraintrustInvalidDataException( - "Unknown ChatCompletionContentPart: $json" - ) - } - } - - class Deserializer : - BaseDeserializer( - ChatCompletionContentPart::class - ) { - - override fun ObjectCodec.deserialize( - node: JsonNode - ): ChatCompletionContentPart { - val json = JsonValue.fromJsonNode(node) - - tryDeserialize(node, jacksonTypeRef()) { - it.validate() - } - ?.let { - return ChatCompletionContentPart( - chatCompletionContentPartText = it, - _json = json - ) - } - tryDeserialize(node, jacksonTypeRef()) { - it.validate() - } - ?.let { - return ChatCompletionContentPart( - chatCompletionContentPartImage = it, - _json = json - ) - } - - return ChatCompletionContentPart(_json = json) - } - } - - class Serializer : - BaseSerializer( - ChatCompletionContentPart::class - ) { - - override fun serialize( - value: ChatCompletionContentPart, - generator: JsonGenerator, - provider: SerializerProvider - ) { - when { - value.chatCompletionContentPartText != null -> - generator.writeObject(value.chatCompletionContentPartText) - value.chatCompletionContentPartImage != null -> - generator.writeObject(value.chatCompletionContentPartImage) - value._json != null -> generator.writeObject(value._json) - else -> - throw IllegalStateException("Invalid ChatCompletionContentPart") - } - } - } - } } override fun equals(other: Any?): Boolean { diff --git a/braintrust-java-core/src/main/kotlin/com/braintrustdata/api/models/PromptData.kt b/braintrust-java-core/src/main/kotlin/com/braintrustdata/api/models/PromptData.kt index a503fa7b..ef3c319f 100755 --- a/braintrust-java-core/src/main/kotlin/com/braintrustdata/api/models/PromptData.kt +++ b/braintrust-java-core/src/main/kotlin/com/braintrustdata/api/models/PromptData.kt @@ -3238,209 +3238,6 @@ private constructor( } } } - - @JsonDeserialize(using = ChatCompletionContentPart.Deserializer::class) - @JsonSerialize(using = ChatCompletionContentPart.Serializer::class) - class ChatCompletionContentPart - private constructor( - private val chatCompletionContentPartText: - ChatCompletionContentPartText? = - null, - private val chatCompletionContentPartImage: - ChatCompletionContentPartImage? = - null, - private val _json: JsonValue? = null, - ) { - - private var validated: Boolean = false - - fun chatCompletionContentPartText(): - Optional = - Optional.ofNullable(chatCompletionContentPartText) - - fun chatCompletionContentPartImage(): - Optional = - Optional.ofNullable(chatCompletionContentPartImage) - - fun isChatCompletionContentPartText(): Boolean = - chatCompletionContentPartText != null - - fun isChatCompletionContentPartImage(): Boolean = - chatCompletionContentPartImage != null - - fun asChatCompletionContentPartText(): ChatCompletionContentPartText = - chatCompletionContentPartText.getOrThrow( - "chatCompletionContentPartText" - ) - - fun asChatCompletionContentPartImage(): ChatCompletionContentPartImage = - chatCompletionContentPartImage.getOrThrow( - "chatCompletionContentPartImage" - ) - - fun _json(): Optional = Optional.ofNullable(_json) - - fun accept(visitor: Visitor): T { - return when { - chatCompletionContentPartText != null -> - visitor.visitChatCompletionContentPartText( - chatCompletionContentPartText - ) - chatCompletionContentPartImage != null -> - visitor.visitChatCompletionContentPartImage( - chatCompletionContentPartImage - ) - else -> visitor.unknown(_json) - } - } - - fun validate(): ChatCompletionContentPart = apply { - if (!validated) { - if ( - chatCompletionContentPartText == null && - chatCompletionContentPartImage == null - ) { - throw BraintrustInvalidDataException( - "Unknown ChatCompletionContentPart: $_json" - ) - } - chatCompletionContentPartText?.validate() - chatCompletionContentPartImage?.validate() - validated = true - } - } - - override fun equals(other: Any?): Boolean { - if (this === other) { - return true - } - - return /* spotless:off */ other is ChatCompletionContentPart && this.chatCompletionContentPartText == other.chatCompletionContentPartText && this.chatCompletionContentPartImage == other.chatCompletionContentPartImage /* spotless:on */ - } - - override fun hashCode(): Int { - return /* spotless:off */ Objects.hash(chatCompletionContentPartText, chatCompletionContentPartImage) /* spotless:on */ - } - - override fun toString(): String { - return when { - chatCompletionContentPartText != null -> - "ChatCompletionContentPart{chatCompletionContentPartText=$chatCompletionContentPartText}" - chatCompletionContentPartImage != null -> - "ChatCompletionContentPart{chatCompletionContentPartImage=$chatCompletionContentPartImage}" - _json != null -> "ChatCompletionContentPart{_unknown=$_json}" - else -> - throw IllegalStateException( - "Invalid ChatCompletionContentPart" - ) - } - } - - companion object { - - @JvmStatic - fun ofChatCompletionContentPartText( - chatCompletionContentPartText: ChatCompletionContentPartText - ) = - ChatCompletionContentPart( - chatCompletionContentPartText = - chatCompletionContentPartText - ) - - @JvmStatic - fun ofChatCompletionContentPartImage( - chatCompletionContentPartImage: ChatCompletionContentPartImage - ) = - ChatCompletionContentPart( - chatCompletionContentPartImage = - chatCompletionContentPartImage - ) - } - - interface Visitor { - - fun visitChatCompletionContentPartText( - chatCompletionContentPartText: ChatCompletionContentPartText - ): T - - fun visitChatCompletionContentPartImage( - chatCompletionContentPartImage: ChatCompletionContentPartImage - ): T - - fun unknown(json: JsonValue?): T { - throw BraintrustInvalidDataException( - "Unknown ChatCompletionContentPart: $json" - ) - } - } - - class Deserializer : - BaseDeserializer( - ChatCompletionContentPart::class - ) { - - override fun ObjectCodec.deserialize( - node: JsonNode - ): ChatCompletionContentPart { - val json = JsonValue.fromJsonNode(node) - - tryDeserialize( - node, - jacksonTypeRef() - ) { - it.validate() - } - ?.let { - return ChatCompletionContentPart( - chatCompletionContentPartText = it, - _json = json - ) - } - tryDeserialize( - node, - jacksonTypeRef() - ) { - it.validate() - } - ?.let { - return ChatCompletionContentPart( - chatCompletionContentPartImage = it, - _json = json - ) - } - - return ChatCompletionContentPart(_json = json) - } - } - - class Serializer : - BaseSerializer( - ChatCompletionContentPart::class - ) { - - override fun serialize( - value: ChatCompletionContentPart, - generator: JsonGenerator, - provider: SerializerProvider - ) { - when { - value.chatCompletionContentPartText != null -> - generator.writeObject( - value.chatCompletionContentPartText - ) - value.chatCompletionContentPartImage != null -> - generator.writeObject( - value.chatCompletionContentPartImage - ) - value._json != null -> generator.writeObject(value._json) - else -> - throw IllegalStateException( - "Invalid ChatCompletionContentPart" - ) - } - } - } - } } override fun equals(other: Any?): Boolean {