fix(main.py): support async calls from azure_text

This commit is contained in:
Krrish Dholakia 2024-04-03 07:59:17 -07:00
parent d7fda60889
commit 88e8f14b69

View file

@ -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"