🐛 fix: custom channel error

providers/azure/base.go

@@ -27,6 +27,7 @@ func (f AzureProviderFactory) Create(c *gin.Context) base.ProviderInterface {
 				// AudioSpeech:         "/audio/speech",
 			},
 			IsAzure:       true,
+			BalanceAction: false,
 		},
 	}
 }

providers/openai/base.go

@@ -54,7 +54,7 @@ func CreateOpenAIProvider(c *gin.Context, baseURL string) *OpenAIProvider {
 			Context:             c,
 		},
 		IsAzure:       false,
-		BalanceAction: false,
+		BalanceAction: true,
 	}
 }
 

providers/providers.go

@@ -72,8 +72,9 @@ func GetProvider(channel *model.Channel, c *gin.Context) base.ProviderInterface
 		}
 
 		provider = openai.CreateOpenAIProvider(c, baseURL)
-	}
+	} else {
 		provider = factory.Create(c)
+	}
 	provider.SetChannel(channel)
 
 	return provider