xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-11-04 21:33: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:47:30 +08:00
parent d4a5df7373
commit ab1d61d910
3 changed files with 6 additions and 4 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) string {
return fmt.Sprintf("%s (request id: %s)", message, id)
func MessageWithRequestId(message string, id string, userId int) string {
return fmt.Sprintf("%s (request id: %s, user id: %d)", message, id, userId)
}
func RandomSleep() {

3
controller/relay.go
View File

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

3
middleware/utils.go
View File

@@ -6,9 +6,10 @@ import (
)
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)),
"message": common.MessageWithRequestId(message, c.GetString(common.RequestIdKey), userId),
"type": "new_api_error",
},
})