xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-11-04 05:13:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: print user id when error

Browse Source
This commit is contained in:
CalciumIon
2024-07-27 17:55:36 +08:00
parent 88cc88c5d0
commit b7bc205b73
3 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
common/utils.go
View File

@@ -179,8 +179,8 @@ func Max(a int, b int) int {
}
}
func MessageWithRequestId(message string, id string, userId int) string {
return fmt.Sprintf("%s (request id: %s, user id: %d)", message, id, userId)
func MessageWithRequestId(message string, id string) string {
return fmt.Sprintf("%s (request id: %s)", message, id)
}
func RandomSleep() {

3
controller/relay.go
View File

@@ -82,8 +82,7 @@ func Relay(c *gin.Context) {
if openaiErr.StatusCode == http.StatusTooManyRequests {
openaiErr.Error.Message = "当前分组上游负载已饱和，请稍后再试"
}
userId := c.GetInt("id")
openaiErr.Error.Message = common.MessageWithRequestId(openaiErr.Error.Message, requestId, userId)
openaiErr.Error.Message = common.MessageWithRequestId(openaiErr.Error.Message, requestId)
c.JSON(openaiErr.StatusCode, gin.H{
"error": openaiErr.Error,
})

5
middleware/utils.go
View File

@@ -1,6 +1,7 @@
package middleware
import (
"fmt"
"github.com/gin-gonic/gin"
"one-api/common"
)
@@ -9,12 +10,12 @@ func abortWithOpenAiMessage(c *gin.Context, statusCode int, message string) {
userId := c.GetInt("id")
c.JSON(statusCode, gin.H{
"error": gin.H{
"message": common.MessageWithRequestId(message, c.GetString(common.RequestIdKey), userId),
"message": common.MessageWithRequestId(message, c.GetString(common.RequestIdKey)),
"type": "new_api_error",
},
})
c.Abort()
common.LogError(c.Request.Context(), message)
common.LogError(c.Request.Context(), fmt.Sprintf("user %d | %s", userId, message))
}
func abortWithMidjourneyMessage(c *gin.Context, statusCode int, code int, description string) {