Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'songquanpeng/main'

Browse Source
This commit is contained in:
quzard 2023-06-20 21:07:18 +08:00
commit 5108f3c98f
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
controller/billing.go
View File

@ -48,6 +48,9 @@ func GetUsage(c *gin.Context) {
return return
} }
amount := float64(quota) amount := float64(quota)
if common.DisplayInCurrencyEnabled {
amount /= common.QuotaPerUnit
}
usage := OpenAIUsageResponse{ usage := OpenAIUsageResponse{
Object: "list", Object: "list",
TotalUsage: amount, TotalUsage: amount,

2
model/cache.go
View File

@ -33,6 +33,7 @@ func CacheGetTokenByKey(key string) (*Token, error) {
if err != nil { if err != nil {
common.SysError("Redis set token error: " + err.Error()) common.SysError("Redis set token error: " + err.Error())
} }
return &token, nil
} }
err = json.Unmarshal([]byte(tokenObjectString), &token) err = json.Unmarshal([]byte(tokenObjectString), &token)
return &token, err return &token, err
@ -94,6 +95,7 @@ func CacheGetRandomSatisfiedChannel(group string, model string) (*Channel, error
if !common.RedisEnabled { if !common.RedisEnabled {
return GetRandomSatisfiedChannel(group, model) return GetRandomSatisfiedChannel(group, model)
} }
return GetRandomSatisfiedChannel(group, model)
// TODO: implement this // TODO: implement this
return nil, nil return nil, nil
} }