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

Merge remote-tracking branch 'upstream/main'

Browse Source
This commit is contained in:
ckt1031 2023-07-16 13:12:45 +08:00
commit 203471d7a9
2 changed files with 22 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
controller/billing.go
View File

@ -8,7 +8,8 @@ import (
)
func GetSubscription(c *gin.Context) {
var quota int
var remainQuota int
var usedQuota int
var err error
var expirationDate int64
@ -18,10 +19,14 @@ func GetSubscription(c *gin.Context) {
expirationDate = token.ExpiredTime
if common.DisplayTokenStatEnabled {
quota = token.RemainQuota
tokenId := c.GetInt("token_id")
token, err = model.GetTokenById(tokenId)
remainQuota = token.RemainQuota
usedQuota = token.UsedQuota
} else {
userId := c.GetInt("id")
quota, err = model.GetUserQuota(userId)
remainQuota, err = model.GetUserQuota(userId)
usedQuota, err = model.GetUserUsedQuota(userId)
}
if err != nil {
openAIError := OpenAIError{
@ -33,6 +38,7 @@ func GetSubscription(c *gin.Context) {
})
return
}
quota := remainQuota + usedQuota
amount := float64(quota)
if common.DisplayInCurrencyEnabled {
amount /= common.QuotaPerUnit

26
web/src/pages/Channel/EditChannel.jsx
View File

@ -59,19 +59,19 @@ const EditChannel = () => {
data.models = [];
} else {
data.models = data.models.split(',');
// setTimeout(() => {
// let localModelOptions = [...modelOptions];
// data.models.forEach((model) => {
// if (!localModelOptions.find((option) => option.key === model)) {
// localModelOptions.push({
// key: model,
// text: model,
// value: model,
// });
// }
// });
// setModelOptions(localModelOptions);
// }, 1000);
setTimeout(() => {
let localModelOptions = [...data.models];
data.models.forEach((model) => {
if (!localModelOptions.find((option) => option.key === model)) {
localModelOptions.push({
key: model,
text: model,
value: model
});
}
});
setModelOptions(localModelOptions);
}, 1000);
}
if (data.group === '') {
data.groups = [];