small fix
This commit is contained in:
@@ -94,9 +94,7 @@ func (w *DatabaseWriter) Print(level slog.LogLevel, prefix string, traceback []*
|
|||||||
Level: level.GetName(),
|
Level: level.GetName(),
|
||||||
Prefix: prefix,
|
Prefix: prefix,
|
||||||
Traceback: slog.FormatFullTraceback(traceback),
|
Traceback: slog.FormatFullTraceback(traceback),
|
||||||
Message: strings.Join(utils.Map(messages, func(el any) string {
|
Message: strings.Join(utils.Map(messages, utils.AnyToString), "\n"),
|
||||||
return fmt.Sprintf("%v", el)
|
|
||||||
}), "\n"),
|
|
||||||
Time: t,
|
Time: t,
|
||||||
TimeStamp: t.Unix(),
|
TimeStamp: t.Unix(),
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -147,27 +147,31 @@ func (o *OpenAIAPI) DoRequest(url string, params any, retries int) ([]byte, erro
|
|||||||
return responseBody, err
|
return responseBody, err
|
||||||
}
|
}
|
||||||
|
|
||||||
tempData := make(map[string]any)
|
var tempData any
|
||||||
err = json.Unmarshal(responseBody, &tempData)
|
err = json.Unmarshal(responseBody, &tempData)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Println("json unmarshal failed:", err)
|
log.Println("json unmarshal failed:", err)
|
||||||
return responseBody, err
|
return responseBody, err
|
||||||
}
|
}
|
||||||
// {"error":{"message":"openai_error","type":"bad_response_status_code","param":"","code":"bad_response_status_code"}}
|
// {"error":{"message":"openai_error","type":"bad_response_status_code","param":"","code":"bad_response_status_code"}}
|
||||||
if errorData, ok := tempData["error"]; ok {
|
if eData, ok := tempData.(map[string]any); ok {
|
||||||
if errorPayload, ok := errorData.(map[string]any); ok {
|
if errorData, ok := eData["error"]; ok {
|
||||||
code := errorPayload["code"].(string)
|
if errorPayload, ok := errorData.(map[string]any); ok {
|
||||||
if code == "bad_response_status_code" {
|
code := errorPayload["code"].(string)
|
||||||
if retries >= 3 {
|
if code == "bad_response_status_code" {
|
||||||
return responseBody, BadResponseErr
|
if retries >= 3 {
|
||||||
|
return responseBody, BadResponseErr
|
||||||
|
}
|
||||||
|
o.Logger.Warnln("Retrying because of bad response status code")
|
||||||
|
return o.DoRequest(url, params, retries+1)
|
||||||
}
|
}
|
||||||
o.Logger.Warnln("Retrying because of bad response status code")
|
return nil, errors.New(code)
|
||||||
return o.DoRequest(url, params, retries+1)
|
|
||||||
}
|
}
|
||||||
return nil, errors.New(code)
|
o.Logger.Errorln("Unknown error", errorData)
|
||||||
|
return nil, errors.New(string(responseBody))
|
||||||
}
|
}
|
||||||
o.Logger.Errorln("Unknown error", errorData)
|
} else if eData, ok := tempData.(string); ok {
|
||||||
return nil, errors.New(string(responseBody))
|
return responseBody, errors.New(eData)
|
||||||
}
|
}
|
||||||
|
|
||||||
return responseBody, err
|
return responseBody, err
|
||||||
|
|||||||
Reference in New Issue
Block a user