rev: added a file to support moderations endpoint (#21)
* Added support for Moderations API * gofmt moderation.go
This commit is contained in:
69
moderation.go
Normal file
69
moderation.go
Normal file
@@ -0,0 +1,69 @@
|
||||
package gogpt
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
// ModerationRequest represents a request structure for moderation API
|
||||
type ModerationRequest struct {
|
||||
Input string `json:"input,omitempty"`
|
||||
Model *string `json:"model,omitempty"`
|
||||
}
|
||||
|
||||
// Result represents one of possible moderation results
|
||||
type Result struct {
|
||||
Categories []ResultCategories `json:"categories"`
|
||||
CategoryScores []ResultCategoryScores `json:"category_scores"`
|
||||
Flagged int `json:"flagged"`
|
||||
}
|
||||
|
||||
// ResultCategories represents Categories of Result
|
||||
type ResultCategories struct {
|
||||
Hate int `json:"hate"`
|
||||
HateThreatening int `json:"hate/threatening"`
|
||||
SelfHarm int `json:"self-harm"`
|
||||
Sexual int `json:"sexual"`
|
||||
SexualMinors int `json:"sexual/minors"`
|
||||
Violence int `json:"violence"`
|
||||
ViolenceGraphic int `json:"violence/graphic"`
|
||||
}
|
||||
|
||||
// ResultCategoryScores represents CategoryScores of Result
|
||||
type ResultCategoryScores struct {
|
||||
Hate float32 `json:"hate"`
|
||||
HateThreatening float32 `json:"hate/threatening"`
|
||||
SelfHarm float32 `json:"self-harm"`
|
||||
Sexual float32 `json:"sexual"`
|
||||
SexualMinors float32 `json:"sexual/minors"`
|
||||
Violence float32 `json:"violence"`
|
||||
ViolenceGraphic float32 `json:"violence/graphic"`
|
||||
}
|
||||
|
||||
// ModerationResponse represents a response structure for moderation API
|
||||
type ModerationResponse struct {
|
||||
ID string `json:"id"`
|
||||
Model string `json:"model"`
|
||||
Results []Result `json:"results"`
|
||||
}
|
||||
|
||||
// Moderations — perform a moderation api call over a string.
|
||||
// Input can be an array or slice but a string will reduce the complexity.
|
||||
func (c *Client) Moderations(ctx context.Context, request ModerationRequest) (response ModerationResponse, err error) {
|
||||
var reqBytes []byte
|
||||
reqBytes, err = json.Marshal(request)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
req, err := http.NewRequest("POST", c.fullURL("/moderations"), bytes.NewBuffer(reqBytes))
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
req = req.WithContext(ctx)
|
||||
err = c.sendRequest(req, &response)
|
||||
return
|
||||
}
|
||||
Reference in New Issue
Block a user