mirrors2/opencatd-open
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refact:openteam

Browse Source
This commit is contained in:
Sakurasan
2025-01-01 23:10:01 +08:00
parent 1f5e1c221c
commit 65d6d12972
18 changed files with 1126 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
store/keydb.go
View File

@@ -36,11 +36,14 @@ type Key struct {
ID uint `gorm:"primarykey" json:"id,omitempty"`
Key string `gorm:"unique;not null" json:"key,omitempty"`
Name string `gorm:"unique;not null" json:"name,omitempty"`
ModelAlias string `gorm:"column:model_alias"`
UserId string `json:"-,omitempty"`
ApiType string `gorm:"column:api_type"`
EndPoint string `gorm:"column:endpoint"`
ResourceNmae string `gorm:"column:resource_name"`
DeploymentName string `gorm:"column:deployment_name"`
ModelPrefix string `gorm:"column:model_prefix"`
SupportModel string `gorm:"column:support_model"`
ApiSecret string `gorm:"column:api_secret"`
CreatedAt time.Time `json:"createdAt,omitempty"`
UpdatedAt time.Time `json:"updatedAt,omitempty"`

12
store/usage.go
View File

@@ -9,14 +9,15 @@ import (
)
type DailyUsage struct {
ID int `gorm:"column:id"`
UserID int `gorm:"column:user_id";primaryKey`
ID int64 `gorm:"column:id"`
UserID int64 `gorm:"column:user_id;primaryKey" json:"user_id"`
Date time.Time `gorm:"column:date"`
SKU string `gorm:"column:sku"`
PromptUnits int `gorm:"column:prompt_units"`
CompletionUnits int `gorm:"column:completion_units"`
TotalUnit int `gorm:"column:total_unit"`
Cost string `gorm:"column:cost"`
CreatedAt time.Time `json:"createdAt,omitempty"`
}
func (DailyUsage) TableName() string {
@@ -39,13 +40,6 @@ func (Usage) TableName() string {
return "usages"
}
type Summary struct {
UserId int `gorm:"column:user_id"`
SumPromptUnits int `gorm:"column:sum_prompt_units"`
SumCompletionUnits int `gorm:"column:sum_completion_units"`
SumTotalUnit int `gorm:"column:sum_total_unit"`
SumCost float64 `gorm:"column:sum_cost"`
}
type CalcUsage struct {
UserID int `json:"userId,omitempty"`
TotalUnit int `json:"totalUnit,omitempty"`

7
store/userdb.go
View File

@@ -2,16 +2,13 @@ package store
import (
"time"
"gorm.io/gorm"
)
type User struct {
gorm.Model
IsDelete bool `gorm:"default:false" json:"IsDelete"`
ID uint `gorm:"primarykey autoIncrement;" json:"id,omitempty"`
ID int64 `gorm:"primaryKey;autoIncrement" json:"id,omitempty"`
Name string `gorm:"unique;not null" json:"name,omitempty"`
Token string `gorm:"unique;not null" json:"token,omitempty"`
Token string `gorm:"unique;not null;index" json:"token,omitempty"`
CreatedAt time.Time `json:"createdAt,omitempty"`
UpdatedAt time.Time `json:"updatedAt,omitempty"`
// DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`