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

fix: create NewOpenAIError function for improved error handling in Relay

CaIon 7 месяцев назад
Родитель
Сommit
d19a6914f9
2 измененных файлов с 9 добавлено и 1 удалено
  1. 1 1
      relay/relay-text.go
  2. 8 0
      types/error.go

+ 1 - 1
relay/relay-text.go

@@ -210,7 +210,7 @@ func TextHelper(c *gin.Context) (newAPIError *types.NewAPIError) {
 	resp, err := adaptor.DoRequest(c, relayInfo, requestBody)
 
 	if err != nil {
-		return types.NewError(err, types.ErrorCodeDoRequestFailed)
+		return types.NewOpenAIError(err, types.ErrorCodeDoRequestFailed, http.StatusInternalServerError)
 	}
 
 	statusCodeMappingStr := c.GetString("status_code_mapping")

+ 8 - 0
types/error.go

@@ -152,6 +152,14 @@ func NewError(err error, errorCode ErrorCode) *NewAPIError {
 	}
 }
 
+func NewOpenAIError(err error, errorCode ErrorCode, statusCode int) *NewAPIError {
+	openaiError := OpenAIError{
+		Message: err.Error(),
+		Type:    string(errorCode),
+	}
+	return WithOpenAIError(openaiError, statusCode)
+}
+
 func NewErrorWithStatusCode(err error, errorCode ErrorCode, statusCode int) *NewAPIError {
 	return &NewAPIError{
 		Err:        err,