From a6bc673ffa11c3ea15beb690d98d94db5fea3199 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 6 Apr 2024 12:23:58 -0700 Subject: [PATCH] feat - re-use OpenAI client for azure text --- litellm/router.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/router.py b/litellm/router.py index 452251a7e..e0c124c0c 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -1767,6 +1767,7 @@ class Router: model_name in litellm.open_ai_chat_completion_models or custom_llm_provider in litellm.openai_compatible_providers or custom_llm_provider == "azure" + or custom_llm_provider == "azure_text" or custom_llm_provider == "custom_openai" or custom_llm_provider == "openai" or custom_llm_provider == "text-completion-openai"