From 01fef1a9f872551e882c45deca46ecbc313b4d15 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 6 Apr 2024 11:28:25 -0700 Subject: [PATCH] feat - re-use openai client for text completion --- litellm/router.py | 1 + 1 file changed, 1 insertion(+) diff --git a/litellm/router.py b/litellm/router.py index 8c7c1be58..452251a7e 100644 --- a/litellm/router.py +++ b/litellm/router.py @@ -1769,6 +1769,7 @@ class Router: or custom_llm_provider == "azure" or custom_llm_provider == "custom_openai" or custom_llm_provider == "openai" + or custom_llm_provider == "text-completion-openai" or "ft:gpt-3.5-turbo" in model_name or model_name in litellm.open_ai_embedding_models ):