Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: update name

Browse Source
This commit is contained in:
JustSong 2023-12-10 18:38:20 +08:00
parent 6c5ef2d7f7
commit b0d6fff643
2 changed files with 20 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
controller/relay-audio.go
View File

@ -155,24 +155,24 @@ func relayAudioHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode
var openAIErr TextResponse var openAIErr TextResponse
if err = json.Unmarshal(responseBody, &openAIErr); err == nil { if err = json.Unmarshal(responseBody, &openAIErr); err == nil {
if openAIErr.Error.Message != "" { if openAIErr.Error.Message != "" {
return errorWrapper(errors.New(fmt.Sprintf("type %s, code %v, message %s", openAIErr.Error.Type, openAIErr.Error.Code, openAIErr.Error.Message)), "request_error", http.StatusInternalServerError) return errorWrapper(fmt.Errorf("type %s, code %v, message %s", openAIErr.Error.Type, openAIErr.Error.Code, openAIErr.Error.Message), "request_error", http.StatusInternalServerError)
} }
} }
var text string var text string
switch responseFormat { switch responseFormat {
case "json": case "json":
text, err = getTextFromJson(responseBody) text, err = getTextFromJSON(responseBody)
case "text": case "text":
text, err = getTextFromText(responseBody) text, err = getTextFromText(responseBody)
case "srt": case "srt":
text, err = getTextFromSrt(responseBody) text, err = getTextFromSRT(responseBody)
case "verbose_json": case "verbose_json":
text, err = getTextFromVerboseJson(responseBody) text, err = getTextFromVerboseJSON(responseBody)
case "vtt": case "vtt":
text, err = getTextFromVtt(responseBody) text, err = getTextFromVTT(responseBody)
default: default:
return errorWrapper(errors.New("not_support_response_format"), "not_support_response_format", http.StatusInternalServerError) return errorWrapper(errors.New("unexpected_response_format"), "unexpected_response_format", http.StatusInternalServerError)
} }
if err != nil { if err != nil {
return errorWrapper(err, "get_text_from_body_err", http.StatusInternalServerError) return errorWrapper(err, "get_text_from_body_err", http.StatusInternalServerError)
@ -216,19 +216,19 @@ func relayAudioHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode
return nil return nil
} }
func getTextFromVtt(body []byte) (string, error) { func getTextFromVTT(body []byte) (string, error) {
return getTextFromSrt(body) return getTextFromSRT(body)
} }
func getTextFromVerboseJson(body []byte) (string, error) { func getTextFromVerboseJSON(body []byte) (string, error) {
var whisperResponse WhisperVerboseJsonResponse var whisperResponse WhisperVerboseJSONResponse
if err := json.Unmarshal(body, &whisperResponse); err != nil { if err := json.Unmarshal(body, &whisperResponse); err != nil {
return "", fmt.Errorf("unmarshal_response_body_failed err :%w", err) return "", fmt.Errorf("unmarshal_response_body_failed err :%w", err)
} }
return whisperResponse.Text, nil return whisperResponse.Text, nil
} }
func getTextFromSrt(body []byte) (string, error) { func getTextFromSRT(body []byte) (string, error) {
scanner := bufio.NewScanner(strings.NewReader(string(body))) scanner := bufio.NewScanner(strings.NewReader(string(body)))
var builder strings.Builder var builder strings.Builder
var textLine bool var textLine bool
@ -253,8 +253,8 @@ func getTextFromText(body []byte) (string, error) {
return strings.TrimSuffix(string(body), "\n"), nil return strings.TrimSuffix(string(body), "\n"), nil
} }
func getTextFromJson(body []byte) (string, error) { func getTextFromJSON(body []byte) (string, error) {
var whisperResponse WhisperJsonResponse var whisperResponse WhisperJSONResponse
if err := json.Unmarshal(body, &whisperResponse); err != nil { if err := json.Unmarshal(body, &whisperResponse); err != nil {
return "", fmt.Errorf("unmarshal_response_body_failed err :%w", err) return "", fmt.Errorf("unmarshal_response_body_failed err :%w", err)
} }

14
controller/relay.go
View File

@ -141,16 +141,16 @@ type ImageRequest struct {
User string `json:"user,omitempty"` User string `json:"user,omitempty"`
} }
type WhisperJsonResponse struct { type WhisperJSONResponse struct {
Text string `json:"text,omitempty"` Text string `json:"text,omitempty"`
} }
type WhisperVerboseJsonResponse struct { type WhisperVerboseJSONResponse struct {
Task string `json:"task"` Task string `json:"task,omitempty"`
Language string `json:"language"` Language string `json:"language,omitempty"`
Duration float64 `json:"duration"` Duration float64 `json:"duration,omitempty"`
Text string `json:"text"` Text string `json:"text,omitempty"`
Segments []Segment `json:"segments"` Segments []Segment `json:"segments,omitempty"`
} }
type Segment struct { type Segment struct {