98952198bb
This commit refactors the application's error handling mechanism by introducing a new standardized error type, `types.NewAPIError`. It also renames common JSON utility functions for better clarity. Previously, internal error handling was tightly coupled to the `dto.OpenAIError` format. This change decouples the internal logic from the external API representation. Key changes: - A new `types.NewAPIError` struct is introduced to serve as a canonical internal representation for all API errors. - All relay adapters (OpenAI, Claude, Gemini, etc.) are updated to return `*types.NewAPIError`. - Controllers now convert the internal `NewAPIError` to the client-facing `OpenAIError` format at the API boundary, ensuring backward compatibility. - Channel auto-disable/enable logic is updated to use the new standardized error type. - JSON utility functions are renamed to align with Go's standard library conventions (e.g., `UnmarshalJson` -> `Unmarshal`, `EncodeJson` -> `Marshal`).
85 lines
2.1 KiB
Go
85 lines
2.1 KiB
Go
package controller
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"one-api/common"
|
|
"one-api/constant"
|
|
"one-api/dto"
|
|
"one-api/middleware"
|
|
"one-api/model"
|
|
"one-api/setting"
|
|
"one-api/types"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func Playground(c *gin.Context) {
|
|
var newAPIError *types.NewAPIError
|
|
|
|
defer func() {
|
|
if newAPIError != nil {
|
|
c.JSON(newAPIError.StatusCode, gin.H{
|
|
"error": newAPIError.ToOpenAIError(),
|
|
})
|
|
}
|
|
}()
|
|
|
|
useAccessToken := c.GetBool("use_access_token")
|
|
if useAccessToken {
|
|
newAPIError = types.NewError(errors.New("暂不支持使用 access token"), types.ErrorCodeAccessDenied)
|
|
return
|
|
}
|
|
|
|
playgroundRequest := &dto.PlayGroundRequest{}
|
|
err := common.UnmarshalBodyReusable(c, playgroundRequest)
|
|
if err != nil {
|
|
newAPIError = types.NewError(err, types.ErrorCodeInvalidRequest)
|
|
return
|
|
}
|
|
|
|
if playgroundRequest.Model == "" {
|
|
newAPIError = types.NewError(errors.New("请选择模型"), types.ErrorCodeInvalidRequest)
|
|
return
|
|
}
|
|
c.Set("original_model", playgroundRequest.Model)
|
|
group := playgroundRequest.Group
|
|
userGroup := c.GetString("group")
|
|
|
|
if group == "" {
|
|
group = userGroup
|
|
} else {
|
|
if !setting.GroupInUserUsableGroups(group) && group != userGroup {
|
|
newAPIError = types.NewError(errors.New("无权访问该分组"), types.ErrorCodeAccessDenied)
|
|
return
|
|
}
|
|
c.Set("group", group)
|
|
}
|
|
|
|
userId := c.GetInt("id")
|
|
//c.Set("token_name", "playground-"+group)
|
|
tempToken := &model.Token{
|
|
UserId: userId,
|
|
Name: fmt.Sprintf("playground-%s", group),
|
|
Group: group,
|
|
}
|
|
_ = middleware.SetupContextForToken(c, tempToken)
|
|
_, err = getChannel(c, group, playgroundRequest.Model, 0)
|
|
if err != nil {
|
|
newAPIError = types.NewError(err, types.ErrorCodeGetChannelFailed)
|
|
return
|
|
}
|
|
//middleware.SetupContextForSelectedChannel(c, channel, playgroundRequest.Model)
|
|
common.SetContextKey(c, constant.ContextKeyRequestStartTime, time.Now())
|
|
|
|
// Write user context to ensure acceptUnsetRatio is available
|
|
userCache, err := model.GetUserCache(userId)
|
|
if err != nil {
|
|
newAPIError = types.NewError(err, types.ErrorCodeQueryDataError)
|
|
return
|
|
}
|
|
userCache.WriteContext(c)
|
|
Relay(c)
|
|
}
|