chore: only check OpenAI channel & custom channel

c658ce0a · JustSong · bc4c572b · c658ce0a
Commit c658ce0a authored May 23, 2023 by JustSong
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

controller/channel-billing.go
+4 -0

No files found.
--- a/controller/channel-billing.go
+++ b/controller/channel-billing.go
@@ -144,6 +144,10 @@ func updateAllChannelsBalance() error {
 		if channel.Status != common.ChannelStatusEnabled {
 			continue
 		}
+		// TODO: support Azure
+		if channel.Type != common.ChannelTypeOpenAI && channel.Type != common.ChannelTypeCustom {
+			continue
+		}
 		balance, err := updateChannelBalance(channel)
 		if err != nil {
 			continue