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 8117fd6364
commit bc1ee5c838

View file

@ -2938,6 +2938,7 @@ async def atext_completion(*args, **kwargs):
if ( if (
custom_llm_provider == "openai" custom_llm_provider == "openai"
or custom_llm_provider == "azure" or custom_llm_provider == "azure"
or custom_llm_provider == "azure_text"
or custom_llm_provider == "custom_openai" or custom_llm_provider == "custom_openai"
or custom_llm_provider == "anyscale" or custom_llm_provider == "anyscale"
or custom_llm_provider == "mistral" or custom_llm_provider == "mistral"