// File generated from our OpenAPI spec by Stainless.

package com.openai.models.evals.runs

import com.fasterxml.jackson.module.kotlin.jacksonTypeRef
import com.openai.core.JsonValue
import com.openai.core.jsonMapper
import com.openai.models.FunctionDefinition
import com.openai.models.FunctionParameters
import com.openai.models.ReasoningEffort
import com.openai.models.ResponseFormatText
import com.openai.models.chat.completions.ChatCompletionFunctionTool
import com.openai.models.responses.EasyInputMessage
import org.assertj.core.api.Assertions.assertThat
import org.junit.jupiter.api.Test

internal class CreateEvalCompletionsRunDataSourceTest {

    @Test
    fun create() {
        val createEvalCompletionsRunDataSource =
            CreateEvalCompletionsRunDataSource.builder()
                .fileContentSource(
                    listOf(
                        CreateEvalCompletionsRunDataSource.Source.FileContent.Content.builder()
                            .item(
                                CreateEvalCompletionsRunDataSource.Source.FileContent.Content.Item
                                    .builder()
                                    .putAdditionalProperty("foo", JsonValue.from("bar"))
                                    .build()
                            )
                            .sample(
                                CreateEvalCompletionsRunDataSource.Source.FileContent.Content.Sample
                                    .builder()
                                    .putAdditionalProperty("foo", JsonValue.from("bar"))
                                    .build()
                            )
                            .build()
                    )
                )
                .type(CreateEvalCompletionsRunDataSource.Type.COMPLETIONS)
                .templateInputMessages(
                    listOf(
                        CreateEvalCompletionsRunDataSource.InputMessages.Template.InnerTemplate
                            .ofEasyInputMessage(
                                EasyInputMessage.builder()
                                    .content("string")
                                    .role(EasyInputMessage.Role.USER)
                                    .type(EasyInputMessage.Type.MESSAGE)
                                    .build()
                            )
                    )
                )
                .model("model")
                .samplingParams(
                    CreateEvalCompletionsRunDataSource.SamplingParams.builder()
                        .maxCompletionTokens(0L)
                        .reasoningEffort(ReasoningEffort.NONE)
                        .responseFormat(ResponseFormatText.builder().build())
                        .seed(0L)
                        .temperature(0.0)
                        .addTool(
                            ChatCompletionFunctionTool.builder()
                                .function(
                                    FunctionDefinition.builder()
                                        .name("name")
                                        .description("description")
                                        .parameters(
                                            FunctionParameters.builder()
                                                .putAdditionalProperty("foo", JsonValue.from("bar"))
                                                .build()
                                        )
                                        .strict(true)
                                        .build()
                                )
                                .build()
                        )
                        .topP(0.0)
                        .build()
                )
                .build()

        assertThat(createEvalCompletionsRunDataSource.source())
            .isEqualTo(
                CreateEvalCompletionsRunDataSource.Source.ofFileContent(
                    CreateEvalCompletionsRunDataSource.Source.FileContent.builder()
                        .addContent(
                            CreateEvalCompletionsRunDataSource.Source.FileContent.Content.builder()
                                .item(
                                    CreateEvalCompletionsRunDataSource.Source.FileContent.Content
                                        .Item
                                        .builder()
                                        .putAdditionalProperty("foo", JsonValue.from("bar"))
                                        .build()
                                )
                                .sample(
                                    CreateEvalCompletionsRunDataSource.Source.FileContent.Content
                                        .Sample
                                        .builder()
                                        .putAdditionalProperty("foo", JsonValue.from("bar"))
                                        .build()
                                )
                                .build()
                        )
                        .build()
                )
            )
        assertThat(createEvalCompletionsRunDataSource.type())
            .isEqualTo(CreateEvalCompletionsRunDataSource.Type.COMPLETIONS)
        assertThat(createEvalCompletionsRunDataSource.inputMessages())
            .contains(
                CreateEvalCompletionsRunDataSource.InputMessages.ofTemplate(
                    CreateEvalCompletionsRunDataSource.InputMessages.Template.builder()
                        .addTemplate(
                            EasyInputMessage.builder()
                                .content("string")
                                .role(EasyInputMessage.Role.USER)
                                .type(EasyInputMessage.Type.MESSAGE)
                                .build()
                        )
                        .build()
                )
            )
        assertThat(createEvalCompletionsRunDataSource.model()).contains("model")
        assertThat(createEvalCompletionsRunDataSource.samplingParams())
            .contains(
                CreateEvalCompletionsRunDataSource.SamplingParams.builder()
                    .maxCompletionTokens(0L)
                    .reasoningEffort(ReasoningEffort.NONE)
                    .responseFormat(ResponseFormatText.builder().build())
                    .seed(0L)
                    .temperature(0.0)
                    .addTool(
                        ChatCompletionFunctionTool.builder()
                            .function(
                                FunctionDefinition.builder()
                                    .name("name")
                                    .description("description")
                                    .parameters(
                                        FunctionParameters.builder()
                                            .putAdditionalProperty("foo", JsonValue.from("bar"))
                                            .build()
                                    )
                                    .strict(true)
                                    .build()
                            )
                            .build()
                    )
                    .topP(0.0)
                    .build()
            )
    }

    @Test
    fun roundtrip() {
        val jsonMapper = jsonMapper()
        val createEvalCompletionsRunDataSource =
            CreateEvalCompletionsRunDataSource.builder()
                .fileContentSource(
                    listOf(
                        CreateEvalCompletionsRunDataSource.Source.FileContent.Content.builder()
                            .item(
                                CreateEvalCompletionsRunDataSource.Source.FileContent.Content.Item
                                    .builder()
                                    .putAdditionalProperty("foo", JsonValue.from("bar"))
                                    .build()
                            )
                            .sample(
                                CreateEvalCompletionsRunDataSource.Source.FileContent.Content.Sample
                                    .builder()
                                    .putAdditionalProperty("foo", JsonValue.from("bar"))
                                    .build()
                            )
                            .build()
                    )
                )
                .type(CreateEvalCompletionsRunDataSource.Type.COMPLETIONS)
                .templateInputMessages(
                    listOf(
                        CreateEvalCompletionsRunDataSource.InputMessages.Template.InnerTemplate
                            .ofEasyInputMessage(
                                EasyInputMessage.builder()
                                    .content("string")
                                    .role(EasyInputMessage.Role.USER)
                                    .type(EasyInputMessage.Type.MESSAGE)
                                    .build()
                            )
                    )
                )
                .model("model")
                .samplingParams(
                    CreateEvalCompletionsRunDataSource.SamplingParams.builder()
                        .maxCompletionTokens(0L)
                        .reasoningEffort(ReasoningEffort.NONE)
                        .responseFormat(ResponseFormatText.builder().build())
                        .seed(0L)
                        .temperature(0.0)
                        .addTool(
                            ChatCompletionFunctionTool.builder()
                                .function(
                                    FunctionDefinition.builder()
                                        .name("name")
                                        .description("description")
                                        .parameters(
                                            FunctionParameters.builder()
                                                .putAdditionalProperty("foo", JsonValue.from("bar"))
                                                .build()
                                        )
                                        .strict(true)
                                        .build()
                                )
                                .build()
                        )
                        .topP(0.0)
                        .build()
                )
                .build()

        val roundtrippedCreateEvalCompletionsRunDataSource =
            jsonMapper.readValue(
                jsonMapper.writeValueAsString(createEvalCompletionsRunDataSource),
                jacksonTypeRef<CreateEvalCompletionsRunDataSource>(),
            )

        assertThat(roundtrippedCreateEvalCompletionsRunDataSource)
            .isEqualTo(createEvalCompletionsRunDataSource)
    }
}
