1ada6cf768
在使用记录中添加 user_agent 字段，用于记录 API 请求的 User-Agent 头信息， 便于分析客户端类型和调试。 变更内容： - 新增数据库迁移 028_add_usage_logs_user_agent.sql - 更新 UsageLog 模型和 Ent Schema 添加 user_agent 字段 - 更新 Repository 层的 Create 和 scanUsageLog 方法 - 更新 RecordUsageInput 结构体支持传入 UserAgent - 更新 Claude/OpenAI/Gemini 三个网关 Handler 传递 UserAgent 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
59 lines
1.1 KiB
Go
59 lines
1.1 KiB
Go
package service
|
|
|
|
import "time"
|
|
|
|
const (
|
|
BillingTypeBalance int8 = 0 // 钱包余额
|
|
BillingTypeSubscription int8 = 1 // 订阅套餐
|
|
)
|
|
|
|
type UsageLog struct {
|
|
ID int64
|
|
UserID int64
|
|
APIKeyID int64
|
|
AccountID int64
|
|
RequestID string
|
|
Model string
|
|
|
|
GroupID *int64
|
|
SubscriptionID *int64
|
|
|
|
InputTokens int
|
|
OutputTokens int
|
|
CacheCreationTokens int
|
|
CacheReadTokens int
|
|
|
|
CacheCreation5mTokens int
|
|
CacheCreation1hTokens int
|
|
|
|
InputCost float64
|
|
OutputCost float64
|
|
CacheCreationCost float64
|
|
CacheReadCost float64
|
|
TotalCost float64
|
|
ActualCost float64
|
|
RateMultiplier float64
|
|
|
|
BillingType int8
|
|
Stream bool
|
|
DurationMs *int
|
|
FirstTokenMs *int
|
|
UserAgent *string
|
|
|
|
// 图片生成字段
|
|
ImageCount int
|
|
ImageSize *string
|
|
|
|
CreatedAt time.Time
|
|
|
|
User *User
|
|
APIKey *APIKey
|
|
Account *Account
|
|
Group *Group
|
|
Subscription *UserSubscription
|
|
}
|
|
|
|
func (u *UsageLog) TotalTokens() int {
|
|
return u.InputTokens + u.OutputTokens + u.CacheCreationTokens + u.CacheReadTokens
|
|
}
|