61 lines
2.2 KiB
Go
61 lines
2.2 KiB
Go
package openai
|
|
|
|
import (
|
|
"bytes"
|
|
"net/http"
|
|
"one-api/common"
|
|
"one-api/types"
|
|
)
|
|
|
|
func (p *OpenAIProvider) TranslationAction(request *types.AudioRequest, isModelMapped bool, promptTokens int) (usage *types.Usage, errWithCode *types.OpenAIErrorWithStatusCode) {
|
|
fullRequestURL := p.GetFullRequestURL(p.AudioTranslations, request.Model)
|
|
headers := p.GetRequestHeaders()
|
|
|
|
client := common.NewClient()
|
|
|
|
var formBody bytes.Buffer
|
|
var req *http.Request
|
|
var err error
|
|
if isModelMapped {
|
|
builder := client.CreateFormBuilder(&formBody)
|
|
if err := audioMultipartForm(request, builder); err != nil {
|
|
return nil, common.ErrorWrapper(err, "create_form_builder_failed", http.StatusInternalServerError)
|
|
}
|
|
req, err = client.NewRequest(p.Context.Request.Method, fullRequestURL, common.WithBody(&formBody), common.WithHeader(headers), common.WithContentType(builder.FormDataContentType()))
|
|
req.ContentLength = int64(formBody.Len())
|
|
|
|
} else {
|
|
req, err = client.NewRequest(p.Context.Request.Method, fullRequestURL, common.WithBody(p.Context.Request.Body), common.WithHeader(headers), common.WithContentType(p.Context.Request.Header.Get("Content-Type")))
|
|
req.ContentLength = p.Context.Request.ContentLength
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, common.ErrorWrapper(err, "new_request_failed", http.StatusInternalServerError)
|
|
}
|
|
|
|
var textResponse string
|
|
if hasJSONResponse(request) {
|
|
openAIProviderTranscriptionsResponse := &OpenAIProviderTranscriptionsResponse{}
|
|
errWithCode = p.SendRequest(req, openAIProviderTranscriptionsResponse, true)
|
|
if errWithCode != nil {
|
|
return
|
|
}
|
|
textResponse = openAIProviderTranscriptionsResponse.Text
|
|
} else {
|
|
openAIProviderTranscriptionsTextResponse := new(OpenAIProviderTranscriptionsTextResponse)
|
|
errWithCode = p.SendRequest(req, openAIProviderTranscriptionsTextResponse, true)
|
|
if errWithCode != nil {
|
|
return
|
|
}
|
|
textResponse = getTextContent(*openAIProviderTranscriptionsTextResponse.GetString(), request.ResponseFormat)
|
|
}
|
|
|
|
completionTokens := common.CountTokenText(textResponse, request.Model)
|
|
usage = &types.Usage{
|
|
PromptTokens: promptTokens,
|
|
CompletionTokens: completionTokens,
|
|
TotalTokens: promptTokens + completionTokens,
|
|
}
|
|
return
|
|
}
|