50 lines
1.4 KiB
Go
50 lines
1.4 KiB
Go
|
package openai
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/http"
|
||
|
|
"one-api/common"
|
||
|
|
"one-api/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (c *OpenAIProviderModerationResponse) responseHandler(resp *http.Response) (errWithCode *types.OpenAIErrorWithStatusCode) {
|
||
|
|
if c.Error.Type != "" {
|
||
|
|
errWithCode = &types.OpenAIErrorWithStatusCode{
|
||
|
|
OpenAIError: c.Error,
|
||
|
|
StatusCode: resp.StatusCode,
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *OpenAIProvider) ModerationAction(request *types.ModerationRequest, isModelMapped bool, promptTokens int) (usage *types.Usage, errWithCode *types.OpenAIErrorWithStatusCode) {
|
||
|
|
|
||
|
|
requestBody, err := p.getRequestBody(&request, isModelMapped)
|
||
|
|
if err != nil {
|
||
|
|
return nil, types.ErrorWrapper(err, "json_marshal_failed", http.StatusInternalServerError)
|
||
|
|
}
|
||
|
|
|
||
|
|
fullRequestURL := p.GetFullRequestURL(p.Moderation, request.Model)
|
||
|
|
headers := p.GetRequestHeaders()
|
||
|
|
|
||
|
|
client := common.NewClient()
|
||
|
|
req, err := client.NewRequest(p.Context.Request.Method, fullRequestURL, common.WithBody(requestBody), common.WithHeader(headers))
|
||
|
|
if err != nil {
|
||
|
|
return nil, types.ErrorWrapper(err, "new_request_failed", http.StatusInternalServerError)
|
||
|
|
}
|
||
|
|
|
||
|
|
openAIProviderModerationResponse := &OpenAIProviderModerationResponse{}
|
||
|
|
errWithCode = p.sendRequest(req, openAIProviderModerationResponse)
|
||
|
|
if errWithCode != nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
usage = &types.Usage{
|
||
|
|
PromptTokens: promptTokens,
|
||
|
|
CompletionTokens: 0,
|
||
|
|
TotalTokens: promptTokens,
|
||
|
|
}
|
||
|
|
|
||
|
|
return
|
||
|
|
}
|