commonTest.com.aallam.openai.client.TestModerations.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of openai-client Show documentation

OpenAI API Kotlin Client

There is a newer version: 4.0.0-beta01

package com.aallam.openai.client

import com.aallam.openai.api.moderation.ModerationRequest
import kotlinx.coroutines.test.runTest
import kotlin.test.Test
import kotlin.test.assertTrue

class TestModerations : TestOpenAI() {

    @Test
    fun moderations() = runTest {
        val response = openAI.moderations(
            request = ModerationRequest(
                input = "I want to kill them."
            )
        )

        val result = response.results.first()
        assertTrue(result.flagged)
    }
}