From 8363e24e58a728a897e2b3d79300df6c6a23a28f Mon Sep 17 00:00:00 2001
From: Mikhail Khludnev <mkhl@apache.org>
Date: Fri, 1 Mar 2024 21:46:57 +0300
Subject: [PATCH] fix #2260 #2261

---
 litellm/utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/litellm/utils.py b/litellm/utils.py
index 1106d2fec9..80fda3d43e 100644
--- a/litellm/utils.py
+++ b/litellm/utils.py
@@ -5499,7 +5499,7 @@ def validate_environment(model: Optional[str] = None) -> dict:
         }
     ## EXTRACT LLM PROVIDER - if model name provided
     try:
-        custom_llm_provider = get_llm_provider(model=model)
+        _, custom_llm_provider, _, _ = get_llm_provider(model=model)
     except:
         custom_llm_provider = None
     # # check if llm provider part of model name
@@ -5593,7 +5593,7 @@ def validate_environment(model: Optional[str] = None) -> dict:
         ## openai - chatcompletion + text completion
         if (
             model in litellm.open_ai_chat_completion_models
-            or litellm.open_ai_text_completion_models
+            or model in litellm.open_ai_text_completion_models
         ):
             if "OPENAI_API_KEY" in os.environ:
                 keys_in_environment = True