Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改日志详情

Browse Source
This commit is contained in:
wood 2023-11-15 23:30:04 +08:00
parent bff5456313
commit 9cab86b86c
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
controller/relay-text.go
View File

@ -435,9 +435,15 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
common.LogError(ctx, "error update user quota cache: "+err.Error())
}
if quota != 0 {
// logContent := fmt.Sprintf("模型倍率 %.4f", modelRatio)
ratio = ratio * 0.002
logContent := fmt.Sprintf("输入: $%.10g/1k tokens", ratio)
inputPrice := ratio * 0.002
var logContent string
if completionRatio == 1 {
logContent = fmt.Sprintf("单价: $%.6g/1k tokens", inputPrice)
} else {
outputPrice := inputPrice * completionRatio
logContent = fmt.Sprintf("输入: $%.6g/1k tokens, 输出: $%.6g/1k tokens", inputPrice, outputPrice)
}
model.RecordConsumeLog(ctx, userId, channelId, promptTokens, completionTokens, textRequest.Model, tokenName, quota, logContent)
model.UpdateUserUsedQuotaAndRequestCount(userId, quota)
model.UpdateChannelUsedQuota(channelId, quota)