-
Notifications
You must be signed in to change notification settings - Fork 6
Expand file tree
/
Copy pathExampleModeration.java
More file actions
62 lines (49 loc) · 2.09 KB
/
Copy pathExampleModeration.java
File metadata and controls
62 lines (49 loc) · 2.09 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
package examples;
import io.github.jetkai.openai.api.CreateModeration;
import io.github.jetkai.openai.api.data.moderation.ModerationData;
import io.github.jetkai.openai.api.data.moderation.response.results.ModerationResultsData;
import io.github.jetkai.openai.openai.OpenAI;
import java.util.Optional;
/**
* ExampleModeration
*
* @author <a href="https://github.com/jetkai">Kai</a>
* @version 1.1.3
* {@code - 13/03/2023}
* @since 1.1.3
* {@code - 13/03/2023}
*
* <p>
* Note - This is just a test class, it is recommended to import this project as a library
* <p>
* You can get a free API key from <a href="https://platform.openai.com/account/api-keys">here</a>
*/
final class ExampleModeration {
public static void main(String[] args) {
//Initialize ExampleModeration class
ExampleModeration moderation = new ExampleModeration();
String model = "text-moderation-latest";
String threat = "I will kill you and I hate you!";
//Send request to API - response as JSON string
ModerationResultsData resultsData = moderation.communicate(model, threat);
//Print out true or false if the message is flagged by the moderation AI
System.out.println("Threat: " + threat);
System.out.println("isFlagged: " + resultsData.isFlagged());
}
private ModerationResultsData communicate(String model, String threat) {
ModerationData moderationData = ModerationData.builder()
.setModel(model)
.setInput(threat)
.build();
OpenAI openAI = OpenAI.builder()
.setApiKey(System.getenv("OPEN_AI_API_KEY"))
.createModeration(moderationData)
.build()
//Sends the request to OpenAI's endpoint & parses the response data
.sendRequest();
//Call the CreateModeration API from OpenAI & create instance
Optional<CreateModeration> optionalModeration = openAI.moderation();
//Return model as a JSON string
return optionalModeration.map(CreateModeration::asResults).orElse(null);
}
}