// File generated from our OpenAPI spec by Stainless.

package com.openai.models.chat.completions

import com.fasterxml.jackson.annotation.JsonCreator
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 com.openai.core.BaseDeserializer
import com.openai.core.BaseSerializer
import com.openai.core.Enum
import com.openai.core.JsonField
import com.openai.core.JsonValue
import com.openai.core.allMaxBy
import com.openai.core.getOrThrow
import com.openai.errors.OpenAIInvalidDataException
import java.util.Objects
import java.util.Optional

/**
 * Controls which (if any) tool is called by the model. `none` means the model will not call any
 * tool and instead generates a message. `auto` means the model can pick between generating a
 * message or calling one or more tools. `required` means the model must call one or more tools.
 * Specifying a particular tool via `{"type": "function", "function": {"name": "my_function"}}`
 * forces the model to call that tool.
 *
 * `none` is the default when no tools are present. `auto` is the default if tools are present.
 */
@JsonDeserialize(using = ChatCompletionToolChoiceOption.Deserializer::class)
@JsonSerialize(using = ChatCompletionToolChoiceOption.Serializer::class)
class ChatCompletionToolChoiceOption
private constructor(
    private val auto: Auto? = null,
    private val allowedToolChoice: ChatCompletionAllowedToolChoice? = null,
    private val namedToolChoice: ChatCompletionNamedToolChoice? = null,
    private val namedToolChoiceCustom: ChatCompletionNamedToolChoiceCustom? = null,
    private val _json: JsonValue? = null,
) {

    /**
     * `none` means the model will not call any tool and instead generates a message. `auto` means
     * the model can pick between generating a message or calling one or more tools. `required`
     * means the model must call one or more tools.
     */
    fun auto(): Optional<Auto> = Optional.ofNullable(auto)

    /** Constrains the tools available to the model to a pre-defined set. */
    fun allowedToolChoice(): Optional<ChatCompletionAllowedToolChoice> =
        Optional.ofNullable(allowedToolChoice)

    /**
     * Specifies a tool the model should use. Use to force the model to call a specific function.
     */
    fun namedToolChoice(): Optional<ChatCompletionNamedToolChoice> =
        Optional.ofNullable(namedToolChoice)

    /**
     * Specifies a tool the model should use. Use to force the model to call a specific custom tool.
     */
    fun namedToolChoiceCustom(): Optional<ChatCompletionNamedToolChoiceCustom> =
        Optional.ofNullable(namedToolChoiceCustom)

    fun isAuto(): Boolean = auto != null

    fun isAllowedToolChoice(): Boolean = allowedToolChoice != null

    fun isNamedToolChoice(): Boolean = namedToolChoice != null

    fun isNamedToolChoiceCustom(): Boolean = namedToolChoiceCustom != null

    /**
     * `none` means the model will not call any tool and instead generates a message. `auto` means
     * the model can pick between generating a message or calling one or more tools. `required`
     * means the model must call one or more tools.
     */
    fun asAuto(): Auto = auto.getOrThrow("auto")

    /** Constrains the tools available to the model to a pre-defined set. */
    fun asAllowedToolChoice(): ChatCompletionAllowedToolChoice =
        allowedToolChoice.getOrThrow("allowedToolChoice")

    /**
     * Specifies a tool the model should use. Use to force the model to call a specific function.
     */
    fun asNamedToolChoice(): ChatCompletionNamedToolChoice =
        namedToolChoice.getOrThrow("namedToolChoice")

    /**
     * Specifies a tool the model should use. Use to force the model to call a specific custom tool.
     */
    fun asNamedToolChoiceCustom(): ChatCompletionNamedToolChoiceCustom =
        namedToolChoiceCustom.getOrThrow("namedToolChoiceCustom")

    fun _json(): Optional<JsonValue> = Optional.ofNullable(_json)

    fun <T> accept(visitor: Visitor<T>): T =
        when {
            auto != null -> visitor.visitAuto(auto)
            allowedToolChoice != null -> visitor.visitAllowedToolChoice(allowedToolChoice)
            namedToolChoice != null -> visitor.visitNamedToolChoice(namedToolChoice)
            namedToolChoiceCustom != null ->
                visitor.visitNamedToolChoiceCustom(namedToolChoiceCustom)
            else -> visitor.unknown(_json)
        }

    private var validated: Boolean = false

    fun validate(): ChatCompletionToolChoiceOption = apply {
        if (validated) {
            return@apply
        }

        accept(
            object : Visitor<Unit> {
                override fun visitAuto(auto: Auto) {
                    auto.validate()
                }

                override fun visitAllowedToolChoice(
                    allowedToolChoice: ChatCompletionAllowedToolChoice
                ) {
                    allowedToolChoice.validate()
                }

                override fun visitNamedToolChoice(namedToolChoice: ChatCompletionNamedToolChoice) {
                    namedToolChoice.validate()
                }

                override fun visitNamedToolChoiceCustom(
                    namedToolChoiceCustom: ChatCompletionNamedToolChoiceCustom
                ) {
                    namedToolChoiceCustom.validate()
                }
            }
        )
        validated = true
    }

    fun isValid(): Boolean =
        try {
            validate()
            true
        } catch (e: OpenAIInvalidDataException) {
            false
        }

    /**
     * Returns a score indicating how many valid values are contained in this object recursively.
     *
     * Used for best match union deserialization.
     */
    @JvmSynthetic
    internal fun validity(): Int =
        accept(
            object : Visitor<Int> {
                override fun visitAuto(auto: Auto) = auto.validity()

                override fun visitAllowedToolChoice(
                    allowedToolChoice: ChatCompletionAllowedToolChoice
                ) = allowedToolChoice.validity()

                override fun visitNamedToolChoice(namedToolChoice: ChatCompletionNamedToolChoice) =
                    namedToolChoice.validity()

                override fun visitNamedToolChoiceCustom(
                    namedToolChoiceCustom: ChatCompletionNamedToolChoiceCustom
                ) = namedToolChoiceCustom.validity()

                override fun unknown(json: JsonValue?) = 0
            }
        )

    override fun equals(other: Any?): Boolean {
        if (this === other) {
            return true
        }

        return other is ChatCompletionToolChoiceOption &&
            auto == other.auto &&
            allowedToolChoice == other.allowedToolChoice &&
            namedToolChoice == other.namedToolChoice &&
            namedToolChoiceCustom == other.namedToolChoiceCustom
    }

    override fun hashCode(): Int =
        Objects.hash(auto, allowedToolChoice, namedToolChoice, namedToolChoiceCustom)

    override fun toString(): String =
        when {
            auto != null -> "ChatCompletionToolChoiceOption{auto=$auto}"
            allowedToolChoice != null ->
                "ChatCompletionToolChoiceOption{allowedToolChoice=$allowedToolChoice}"
            namedToolChoice != null ->
                "ChatCompletionToolChoiceOption{namedToolChoice=$namedToolChoice}"
            namedToolChoiceCustom != null ->
                "ChatCompletionToolChoiceOption{namedToolChoiceCustom=$namedToolChoiceCustom}"
            _json != null -> "ChatCompletionToolChoiceOption{_unknown=$_json}"
            else -> throw IllegalStateException("Invalid ChatCompletionToolChoiceOption")
        }

    companion object {

        /**
         * `none` means the model will not call any tool and instead generates a message. `auto`
         * means the model can pick between generating a message or calling one or more tools.
         * `required` means the model must call one or more tools.
         */
        @JvmStatic fun ofAuto(auto: Auto) = ChatCompletionToolChoiceOption(auto = auto)

        /** Constrains the tools available to the model to a pre-defined set. */
        @JvmStatic
        fun ofAllowedToolChoice(allowedToolChoice: ChatCompletionAllowedToolChoice) =
            ChatCompletionToolChoiceOption(allowedToolChoice = allowedToolChoice)

        /**
         * Specifies a tool the model should use. Use to force the model to call a specific
         * function.
         */
        @JvmStatic
        fun ofNamedToolChoice(namedToolChoice: ChatCompletionNamedToolChoice) =
            ChatCompletionToolChoiceOption(namedToolChoice = namedToolChoice)

        /**
         * Specifies a tool the model should use. Use to force the model to call a specific custom
         * tool.
         */
        @JvmStatic
        fun ofNamedToolChoiceCustom(namedToolChoiceCustom: ChatCompletionNamedToolChoiceCustom) =
            ChatCompletionToolChoiceOption(namedToolChoiceCustom = namedToolChoiceCustom)
    }

    /**
     * An interface that defines how to map each variant of [ChatCompletionToolChoiceOption] to a
     * value of type [T].
     */
    interface Visitor<out T> {

        /**
         * `none` means the model will not call any tool and instead generates a message. `auto`
         * means the model can pick between generating a message or calling one or more tools.
         * `required` means the model must call one or more tools.
         */
        fun visitAuto(auto: Auto): T

        /** Constrains the tools available to the model to a pre-defined set. */
        fun visitAllowedToolChoice(allowedToolChoice: ChatCompletionAllowedToolChoice): T

        /**
         * Specifies a tool the model should use. Use to force the model to call a specific
         * function.
         */
        fun visitNamedToolChoice(namedToolChoice: ChatCompletionNamedToolChoice): T

        /**
         * Specifies a tool the model should use. Use to force the model to call a specific custom
         * tool.
         */
        fun visitNamedToolChoiceCustom(
            namedToolChoiceCustom: ChatCompletionNamedToolChoiceCustom
        ): T

        /**
         * Maps an unknown variant of [ChatCompletionToolChoiceOption] to a value of type [T].
         *
         * An instance of [ChatCompletionToolChoiceOption] can contain an unknown variant if it was
         * deserialized from data that doesn't match any known variant. For example, if the SDK is
         * on an older version than the API, then the API may respond with new variants that the SDK
         * is unaware of.
         *
         * @throws OpenAIInvalidDataException in the default implementation.
         */
        fun unknown(json: JsonValue?): T {
            throw OpenAIInvalidDataException("Unknown ChatCompletionToolChoiceOption: $json")
        }
    }

    internal class Deserializer :
        BaseDeserializer<ChatCompletionToolChoiceOption>(ChatCompletionToolChoiceOption::class) {

        override fun ObjectCodec.deserialize(node: JsonNode): ChatCompletionToolChoiceOption {
            val json = JsonValue.fromJsonNode(node)

            val bestMatches =
                sequenceOf(
                        tryDeserialize(node, jacksonTypeRef<Auto>())?.let {
                            ChatCompletionToolChoiceOption(auto = it, _json = json)
                        },
                        tryDeserialize(node, jacksonTypeRef<ChatCompletionAllowedToolChoice>())
                            ?.let {
                                ChatCompletionToolChoiceOption(allowedToolChoice = it, _json = json)
                            },
                        tryDeserialize(node, jacksonTypeRef<ChatCompletionNamedToolChoice>())?.let {
                            ChatCompletionToolChoiceOption(namedToolChoice = it, _json = json)
                        },
                        tryDeserialize(node, jacksonTypeRef<ChatCompletionNamedToolChoiceCustom>())
                            ?.let {
                                ChatCompletionToolChoiceOption(
                                    namedToolChoiceCustom = it,
                                    _json = json,
                                )
                            },
                    )
                    .filterNotNull()
                    .allMaxBy { it.validity() }
                    .toList()
            return when (bestMatches.size) {
                // This can happen if what we're deserializing is completely incompatible with all
                // the possible variants (e.g. deserializing from boolean).
                0 -> ChatCompletionToolChoiceOption(_json = json)
                1 -> bestMatches.single()
                // If there's more than one match with the highest validity, then use the first
                // completely valid match, or simply the first match if none are completely valid.
                else -> bestMatches.firstOrNull { it.isValid() } ?: bestMatches.first()
            }
        }
    }

    internal class Serializer :
        BaseSerializer<ChatCompletionToolChoiceOption>(ChatCompletionToolChoiceOption::class) {

        override fun serialize(
            value: ChatCompletionToolChoiceOption,
            generator: JsonGenerator,
            provider: SerializerProvider,
        ) {
            when {
                value.auto != null -> generator.writeObject(value.auto)
                value.allowedToolChoice != null -> generator.writeObject(value.allowedToolChoice)
                value.namedToolChoice != null -> generator.writeObject(value.namedToolChoice)
                value.namedToolChoiceCustom != null ->
                    generator.writeObject(value.namedToolChoiceCustom)
                value._json != null -> generator.writeObject(value._json)
                else -> throw IllegalStateException("Invalid ChatCompletionToolChoiceOption")
            }
        }
    }

    /**
     * `none` means the model will not call any tool and instead generates a message. `auto` means
     * the model can pick between generating a message or calling one or more tools. `required`
     * means the model must call one or more tools.
     */
    class Auto @JsonCreator private constructor(private val value: JsonField<String>) : Enum {

        /**
         * Returns this class instance's raw value.
         *
         * This is usually only useful if this instance was deserialized from data that doesn't
         * match any known member, and you want to know that value. For example, if the SDK is on an
         * older version than the API, then the API may respond with new members that the SDK is
         * unaware of.
         */
        @com.fasterxml.jackson.annotation.JsonValue fun _value(): JsonField<String> = value

        companion object {

            @JvmField val NONE = of("none")

            @JvmField val AUTO = of("auto")

            @JvmField val REQUIRED = of("required")

            @JvmStatic fun of(value: String) = Auto(JsonField.of(value))
        }

        /** An enum containing [Auto]'s known values. */
        enum class Known {
            NONE,
            AUTO,
            REQUIRED,
        }

        /**
         * An enum containing [Auto]'s known values, as well as an [_UNKNOWN] member.
         *
         * An instance of [Auto] can contain an unknown value in a couple of cases:
         * - It was deserialized from data that doesn't match any known member. For example, if the
         *   SDK is on an older version than the API, then the API may respond with new members that
         *   the SDK is unaware of.
         * - It was constructed with an arbitrary value using the [of] method.
         */
        enum class Value {
            NONE,
            AUTO,
            REQUIRED,
            /** An enum member indicating that [Auto] was instantiated with an unknown value. */
            _UNKNOWN,
        }

        /**
         * Returns an enum member corresponding to this class instance's value, or [Value._UNKNOWN]
         * if the class was instantiated with an unknown value.
         *
         * Use the [known] method instead if you're certain the value is always known or if you want
         * to throw for the unknown case.
         */
        fun value(): Value =
            when (this) {
                NONE -> Value.NONE
                AUTO -> Value.AUTO
                REQUIRED -> Value.REQUIRED
                else -> Value._UNKNOWN
            }

        /**
         * Returns an enum member corresponding to this class instance's value.
         *
         * Use the [value] method instead if you're uncertain the value is always known and don't
         * want to throw for the unknown case.
         *
         * @throws OpenAIInvalidDataException if this class instance's value is a not a known
         *   member.
         */
        fun known(): Known =
            when (this) {
                NONE -> Known.NONE
                AUTO -> Known.AUTO
                REQUIRED -> Known.REQUIRED
                else -> throw OpenAIInvalidDataException("Unknown Auto: $value")
            }

        /**
         * Returns this class instance's primitive wire representation.
         *
         * This differs from the [toString] method because that method is primarily for debugging
         * and generally doesn't throw.
         *
         * @throws OpenAIInvalidDataException if this class instance's value does not have the
         *   expected primitive type.
         */
        fun asString(): String =
            _value().asString().orElseThrow { OpenAIInvalidDataException("Value is not a String") }

        private var validated: Boolean = false

        fun validate(): Auto = apply {
            if (validated) {
                return@apply
            }

            known()
            validated = true
        }

        fun isValid(): Boolean =
            try {
                validate()
                true
            } catch (e: OpenAIInvalidDataException) {
                false
            }

        /**
         * Returns a score indicating how many valid values are contained in this object
         * recursively.
         *
         * Used for best match union deserialization.
         */
        @JvmSynthetic internal fun validity(): Int = if (value() == Value._UNKNOWN) 0 else 1

        override fun equals(other: Any?): Boolean {
            if (this === other) {
                return true
            }

            return other is Auto && value == other.value
        }

        override fun hashCode() = value.hashCode()

        override fun toString() = value.toString()
    }
}
