瀏覽代碼

chore: only check OpenAI channel & custom channel

JustSong 2 年之前
父節點
當前提交
f9f42997b2
共有 1 個文件被更改,包括 4 次插入0 次删除
  1. 4 0
      controller/channel-billing.go

+ 4 - 0
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