Merge pull request #789 from PrathamSoni/psoni2002/fix_async

add custom open ai models to asyncio call
BerriAI · Nov 10, 2023 · 8f10d1a · 8f10d1a
2 parents 86f9bb0 + 2f37baa
commit 8f10d1a
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: