Просмотр исходного кода

fix: mapping siliconflow 429 error status code (#434)

zijiren 1 месяц назад
Родитель
Сommit
6e56d18eec
1 измененных файлов с 5 добавлено и 0 удалено
  1. 5 0
      core/relay/adaptor/siliconflow/error.go

+ 5 - 0
core/relay/adaptor/siliconflow/error.go

@@ -3,6 +3,7 @@ package siliconflow
 import (
 	"net/http"
 	"strconv"
+	"strings"
 
 	"github.com/labring/aiproxy/core/common"
 	"github.com/labring/aiproxy/core/relay/adaptor"
@@ -30,6 +31,10 @@ func ErrorHandler(resp *http.Response) adaptor.Error {
 
 	statusCode := resp.StatusCode
 
+	if strings.Contains(er.Message, "System is really busy") {
+		statusCode = http.StatusTooManyRequests
+	}
+
 	return relaymodel.WrapperOpenAIErrorWithMessage(
 		er.Message,
 		strconv.Itoa(er.Code),