add custom open ai models to asyncio call

BerriAI · Nov 10, 2023 · 2f37baa · 2f37baa
1 parent 86f9bb0
commit 2f37baa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: