oadmin/new-api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: restructure token usage routes and enhance token retrieval logic

Browse Source
This commit is contained in:
CaIon
2025-08-23 15:45:43 +08:00
parent df1ec4832c
commit 93ce48aca8
2 changed files with 22 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
controller/token.go
View File

@@ -103,7 +103,7 @@ func GetTokenUsage(c *gin.Context) {
}
tokenKey := parts[1]
token, err := model.GetTokenByKey(tokenKey, true)
token, err := model.GetTokenByKey(strings.TrimPrefix(tokenKey, "sk-"), false)
if err != nil {
c.JSON(http.StatusOK, gin.H{
"success": false,
@@ -118,17 +118,18 @@ func GetTokenUsage(c *gin.Context) {
}
c.JSON(http.StatusOK, gin.H{
"code": true,
"code": true,
"message": "ok",
"data": gin.H{
"object": "token_usage",
"id": token.Id,
"name": token.Name,
"total_granted": token.RemainQuota + token.UsedQuota,
"total_used": token.UsedQuota,
"total_available": token.RemainQuota,
"unlimited_quota": token.UnlimitedQuota,
"expires_at": expiredAt,
"object": "token_usage",
"name": token.Name,
"total_granted": token.RemainQuota + token.UsedQuota,
"total_used": token.UsedQuota,
"total_available": token.RemainQuota,
"unlimited_quota": token.UnlimitedQuota,
"model_limits": token.GetModelLimitsMap(),
"model_limits_enabled": token.ModelLimitsEnabled,
"expires_at": expiredAt,
},
})
}