25 lines
672 B
Java
25 lines
672 B
Java
|
package com.ossez.openai;
|
||
|
|
|
||
|
|
import org.junit.jupiter.api.Test;
|
||
|
|
|
||
|
|
import static org.junit.jupiter.api.Assertions.assertTrue;
|
||
|
|
|
||
|
|
|
||
|
|
public class ModerationTest {
|
||
|
|
|
||
|
|
String token = System.getenv("OPENAI_TOKEN");
|
||
|
|
OpenAiService service = new OpenAiService(token);
|
||
|
|
|
||
|
|
@Test
|
||
|
|
void createModeration() {
|
||
|
|
// ModerationRequest moderationRequest = ModerationRequest.builder()
|
||
|
|
// .input("I want to kill them")
|
||
|
|
// .model("text-moderation-latest")
|
||
|
|
// .build();
|
||
|
|
//
|
||
|
|
// Moderation moderationScore = service.createModeration(moderationRequest).getResults().get(0);
|
||
|
|
//
|
||
|
|
// assertTrue(moderationScore.isFlagged());
|
||
|
|
}
|
||
|
|
}
|