From 88e8f14b69d891ab77a581b656e5063ea6ddb126 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 3 Apr 2024 07:59:17 -0700 Subject: [PATCH] fix(main.py): support async calls from azure_text --- litellm/main.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/main.py b/litellm/main.py index b79019f2d..fe1a354ed 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"