fix: fix not using proxy when update balance

d6931d05 · JustSong · a04afb22 · d6931d05
Commit d6931d05 authored Jun 09, 2023 by JustSong
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletions

controller/channel-billing.go
+3 -1

No files found.
--- a/controller/channel-billing.go
+++ b/controller/channel-billing.go
@@ -41,7 +41,9 @@ func updateChannelBalance(channel *model.Channel) (float64, error) {
 	baseURL := common.ChannelBaseURLs[channel.Type]
 	switch channel.Type {
 	case common.ChannelTypeOpenAI:
-		// do nothing
+		if channel.BaseURL != "" {
+			baseURL = channel.BaseURL
+		}
 	case common.ChannelTypeAzure:
 		return 0, errors.New("尚未实现")
 	case common.ChannelTypeCustom: