Function RecordConsumeLog

model/log.go:163–221

(c *gin.Context, userId int, params RecordConsumeLogParams)

}

func RecordConsumeLog(c *gin.Context, userId int, params RecordConsumeLogParams) {
	log.Println("=======,", c)
	common.LogInfo(c, fmt.Sprintf("record consume log: userId=%d, params=%s", userId, common.GetJsonString(params)))
	if !common.LogConsumeEnabled {
		return
	}
	username := c.GetString("username")
	otherStr := common.MapToJsonStr(params.Other)
	// 判断是否需要记录 IP
	needRecordIp := false
	if settingMap, err := GetUserSetting(userId, false); err == nil {
		if settingMap.RecordIpLog {
			needRecordIp = true
		}
	}
	log := &Log{
		UserId:           userId,
		Username:         username,
		CreatedAt:        common.GetTimestamp(),
		Type:             LogTypeConsume,
		Content:          params.Content,
		UserInput:        params.UserInput,
		PromptTokens:     params.PromptTokens,
		CompletionTokens: params.CompletionTokens,
		TokenName:        params.TokenName,
		ModelName:        params.ModelName,
		Quota:            params.Quota,
		ChannelId:        params.ChannelId,
		TokenId:          params.TokenId,
		UseTime:          params.UseTimeSeconds,
		IsStream:         params.IsStream,
		Group:            params.Group,
		Ip: func() string {
			if needRecordIp {
				return c.ClientIP()
			}
			return ""
		}(),
		Other: otherStr,
	}
	err := LOG_DB.Create(log).Error
	if err != nil {
		common.LogError(c, "failed to record log: "+err.Error())
	}

	// 异步记录用量统计
	gopool.Go(func() {
		if recordErr := RecordUsageStatistics(params.TokenId, params.TokenName, params.ModelName,
			params.PromptTokens, params.CompletionTokens, params.Quota, true); recordErr != nil {
			common.SysError("failed to record usage statistics: " + recordErr.Error())
		}
	})

	if common.DataExportEnabled {
		gopool.Go(func() {
			LogQuotaData(userId, username, params.ModelName, params.Quota, common.GetTimestamp(), params.PromptTokens+params.CompletionTokens)
		})
	}