Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions controller/relay.go
Original file line number Diff line number Diff line change
Expand Up @@ -293,6 +293,10 @@ func shouldRetry(c *gin.Context, openaiErr *dto.OpenAIErrorWithStatusCode, retry
if openaiErr.StatusCode == 307 {
return true
}
if strings.Contains(openaiErr.Error.Message, "deadline exceeded") || strings.Contains(openaiErr.Error.Message, "request canceled") {
common.LogInfo(c, "客户端请求下游超时,不再重试")
return false
}
if openaiErr.StatusCode/100 == 5 {
// 超时不重试
if openaiErr.StatusCode == 504 || openaiErr.StatusCode == 524 {
Expand All @@ -314,10 +318,6 @@ func shouldRetry(c *gin.Context, openaiErr *dto.OpenAIErrorWithStatusCode, retry
if openaiErr.StatusCode/100 == 2 {
return false
}
if strings.Contains(openaiErr.Error.Message, "deadline exceeded") || strings.Contains(openaiErr.Error.Message, "request canceled") {
common.LogInfo(c, "客户端请求下游超时,不再重试")
return false
}
return true
}

Expand Down
3 changes: 1 addition & 2 deletions service/error.go
Original file line number Diff line number Diff line change
Expand Up @@ -30,8 +30,7 @@ func OpenAIErrorWrapper(err error, code string, statusCode int) *dto.OpenAIError
text := err.Error()
lowerText := strings.ToLower(text)
if strings.Contains(lowerText, "post") || strings.Contains(lowerText, "dial") || strings.Contains(lowerText, "http") {
common.SysLog(fmt.Sprintf("error: %s", text))
text = "请求上游地址失败"
text = fmt.Sprintf("请求上游地址失败,错误信息:%s", text)
}
openAIError := dto.OpenAIError{
Message: text,
Expand Down