xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-10 10:33:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Enhance logging and error handling in processing logic

Browse Source 
- Enhance logging by including requested image counts for improved tracking.
- Implement user quota validation before processing image requests.
- Improve error handling for model validation and JSON unmarshalling in billing processes.
This commit is contained in:
Laisky.Cai
2025-03-16 07:47:59 +00:00
parent fa794e7bd5
commit 400ee3aa24
2 changed files with 52 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
relay/controller/image.go
View File

@@ -226,7 +226,8 @@ func RelayImageHelper(c *gin.Context, relayMode int) *relaymodel.ErrorWithStatus
}
if quota >= 0 {
tokenName := c.GetString(ctxkey.TokenName)
logContent := fmt.Sprintf("model rate %.2f, group rate %.2f", modelRatio, groupRatio)
logContent := fmt.Sprintf("model rate %.2f, group rate %.2f, num %d",
modelRatio, groupRatio, imageRequest.N)
model.RecordConsumeLog(ctx, &model.Log{
UserId: meta.UserId,
ChannelId: meta.ChannelId,