From 9cab86b86c975bd0b3680f279a66199760a2d32b Mon Sep 17 00:00:00 2001 From: wood Date: Wed, 15 Nov 2023 23:30:04 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=97=A5=E5=BF=97=E8=AF=A6?= =?UTF-8?q?=E6=83=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- controller/relay-text.go | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/controller/relay-text.go b/controller/relay-text.go index cb79ccf4..62c0c41a 100644 --- a/controller/relay-text.go +++ b/controller/relay-text.go @@ -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)