diff --git a/providers/azure/base.go b/providers/azure/base.go
index 6e560e6b..597dd2a8 100644
--- a/providers/azure/base.go
+++ b/providers/azure/base.go
@@ -26,7 +26,8 @@ func (f AzureProviderFactory) Create(c *gin.Context) base.ProviderInterface {
 				Context: c,
 				// AudioSpeech:         "/audio/speech",
 			},
-			IsAzure: true,
+			IsAzure:       true,
+			BalanceAction: false,
 		},
 	}
 }
diff --git a/providers/openai/base.go b/providers/openai/base.go
index f37546f7..5536871e 100644
--- a/providers/openai/base.go
+++ b/providers/openai/base.go
@@ -54,7 +54,7 @@ func CreateOpenAIProvider(c *gin.Context, baseURL string) *OpenAIProvider {
 			Context:             c,
 		},
 		IsAzure:       false,
-		BalanceAction: false,
+		BalanceAction: true,
 	}
 }
 
diff --git a/providers/providers.go b/providers/providers.go
index 01043ce3..c5de8094 100644
--- a/providers/providers.go
+++ b/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 = factory.Create(c)
 	provider.SetChannel(channel)
 
 	return provider