RecordUsageStatistics 记录用量统计（从日志记录中调用）
(tokenId int, tokenName, modelName string, promptTokens, completionTokens int, quota int, isSuccess bool)
| 240 | |
| 241 | // RecordUsageStatistics 记录用量统计（从日志记录中调用） |
| 242 | func RecordUsageStatistics(tokenId int, tokenName, modelName string, |
| 243 | promptTokens, completionTokens int, quota int, isSuccess bool) error { |
| 244 | |
| 245 | if tokenId <= 0 || modelName == "" { |
| 246 | return errors.New("invalid parameters for usage statistics") |
| 247 | } |
| 248 | |
| 249 | date := time.Now().Format("2006-01-02") |
| 250 | totalTokens := promptTokens + completionTokens |
| 251 | totalRequests := 1 |
| 252 | successfulRequests := 0 |
| 253 | failedRequests := 0 |
| 254 | |
| 255 | if isSuccess { |
| 256 | successfulRequests = 1 |
| 257 | } else { |
| 258 | failedRequests = 1 |
| 259 | } |
| 260 | |
| 261 | return UpsertUsageStatistics(date, tokenId, tokenName, modelName, |
| 262 | totalRequests, successfulRequests, failedRequests, |
| 263 | totalTokens, promptTokens, completionTokens, quota) |
| 264 | } |
| 265 | |
| 266 | // GetUsageStatisticsSummary 获取用量统计摘要信息 |
| 267 | func GetUsageStatisticsSummary(startDate, endDate string, tokenId int, modelName string) (map[string]interface{}, error) { |
no test coverage detected