-
Notifications
You must be signed in to change notification settings - Fork 1
/
moderations.go
49 lines (44 loc) · 1.89 KB
/
moderations.go
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
package openai_gosdk
const createModerationURL = "https://api.openai.com/v1/moderations"
type RequestCreateModeration struct {
// string or array Required
// The input text to classify
Input StrongOrArray `json:"input"`
// string Optional Defaults to text-moderation-latest
// Two content moderations models are available: and .text-moderation-stabletext-moderation-latest
//
//The default is which will be automatically upgraded over time. This ensures you are always using our most accurate model. If you use , we will provide advanced notice before updating the model. Accuracy of may be slightly lower than for .text-moderation-latest text-moderation-stable text-moderation-stable text-moderation-latest
Model string `json:"model,omitempty"`
}
type ResponseCreateModeration struct {
Id string `json:"id"`
Model string `json:"model"`
Results []struct {
Categories struct {
Hate bool `json:"hate"`
HateThreatening bool `json:"hate/threatening"`
SelfHarm bool `json:"self-harm"`
Sexual bool `json:"sexual"`
SexualMinors bool `json:"sexual/minors"`
Violence bool `json:"violence"`
ViolenceGraphic bool `json:"violence/graphic"`
} `json:"categories"`
CategoryScores struct {
Hate float64 `json:"hate"`
HateThreatening float64 `json:"hate/threatening"`
SelfHarm float64 `json:"self-harm"`
Sexual float64 `json:"sexual"`
SexualMinors float64 `json:"sexual/minors"`
Violence float64 `json:"violence"`
ViolenceGraphic float64 `json:"violence/graphic"`
} `json:"category_scores"`
Flagged bool `json:"flagged"`
} `json:"results"`
}
func NewCreateModeration(baseOpenAI BaseOpenAI) OpenAI[RequestCreateModeration, ResponseCreateModeration] {
return OpenAI[RequestCreateModeration, ResponseCreateModeration]{
BaseOpenAI: baseOpenAI,
TargetURL: createModerationURL,
Method: POST,
}
}