diff --git a/litellm/main.py b/litellm/main.py
index b79019f2d4..fe1a354edf 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -2938,6 +2938,7 @@ async def atext_completion(*args, **kwargs):
         if (
             custom_llm_provider == "openai"
             or custom_llm_provider == "azure"
+            or custom_llm_provider == "azure_text"
             or custom_llm_provider == "custom_openai"
             or custom_llm_provider == "anyscale"
             or custom_llm_provider == "mistral"