From 2f37baa6902e940f038abe47249635d4acb0d635 Mon Sep 17 00:00:00 2001 From: Pratham Soni Date: Thu, 9 Nov 2023 20:47:46 -0800 Subject: [PATCH] add custom open ai models to asyncio call --- litellm/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/main.py b/litellm/main.py index 0b1147748277..0d05a2990458 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -139,7 +139,7 @@ async def acompletion(*args, **kwargs): _, custom_llm_provider, _, _ = get_llm_provider(model=model, api_base=kwargs.get("api_base", None)) - if (custom_llm_provider == "openai" or custom_llm_provider == "azure"): # currently implemented aiohttp calls for just azure and openai, soon all. + if (custom_llm_provider == "openai" or custom_llm_provider == "azure" or custom_llm_provider == "custom_openai"): # currently implemented aiohttp calls for just azure and openai, soon all. if kwargs.get("stream", False): response = completion(*args, **kwargs) else: