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

反代 不查询余额

Browse Source
This commit is contained in:
quzard 2023-06-20 15:00:02 +08:00
parent 5ce638a4d4
commit aeb4685f2f
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
controller/channel-billing.go
View File

@ -9,6 +9,7 @@ import (
"one-api/common" "one-api/common"
"one-api/model" "one-api/model"
"strconv" "strconv"
"strings"
"time" "time"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
@ -160,6 +161,9 @@ func updateChannelBalance(channel *model.Channel) (float64, error) {
case common.ChannelTypeAzure: case common.ChannelTypeAzure:
return 0, errors.New("尚未实现") return 0, errors.New("尚未实现")
case common.ChannelTypeCustom: case common.ChannelTypeCustom:
if strings.Contains(channel.Name, "反代") {
return 0, errors.New("尚未实现")
}
baseURL = channel.BaseURL baseURL = channel.BaseURL
case common.ChannelTypeOpenAISB: case common.ChannelTypeOpenAISB:
return updateChannelOpenAISBBalance(channel) return updateChannelOpenAISBBalance(channel)