fix:model param type, add moderation Model Name const. (#270)

* add ImageEditRequest.ResponseFormat

* add ImageEditRequest/ImageVariRequest.ResponseFormat

* complete image_test

* delete var prompt param

* fix:model param type, add  moderation Model Name const.

* rename ModerationText001

---------

Co-authored-by: Aceld <liudanbing@tal.com>
This commit is contained in:
刘丹冰
2023-04-20 21:10:19 +08:00
committed by GitHub
parent c2b58e77ed
commit 2f008f7054
2 changed files with 17 additions and 4 deletions

View File

@@ -5,10 +5,23 @@ import (
"net/http"
)
// The moderation endpoint is a tool you can use to check whether content complies with OpenAI's usage policies.
// Developers can thus identify content that our usage policies prohibits and take action, for instance by filtering it.
// The default is text-moderation-latest which will be automatically upgraded over time.
// This ensures you are always using our most accurate model.
// If you use text-moderation-stable, we will provide advanced notice before updating the model.
// Accuracy of text-moderation-stable may be slightly lower than for text-moderation-latest.
const (
ModerationTextStable = "text-moderation-stable"
ModerationTextLatest = "text-moderation-latest"
ModerationText001 = "text-moderation-001"
)
// ModerationRequest represents a request structure for moderation API.
type ModerationRequest struct {
Input string `json:"input,omitempty"`
Model *string `json:"model,omitempty"`
Model string `json:"model,omitempty"`
}
// Result represents one of possible moderation results.

View File

@@ -34,7 +34,7 @@ func TestModerations(t *testing.T) {
// create an edit request
model := "text-moderation-stable"
moderationReq := ModerationRequest{
Model: &model,
Model: model,
Input: "I want to kill them.",
}
_, err = client.Moderations(ctx, moderationReq)
@@ -77,7 +77,7 @@ func handleModerationEndpoint(w http.ResponseWriter, r *http.Request) {
res := ModerationResponse{
ID: strconv.Itoa(int(time.Now().Unix())),
Model: *moderationReq.Model,
Model: moderationReq.Model,
}
res.Results = append(res.Results, result)