diff --git a/litellm/tests/test_verify_openai_key.py b/litellm/tests/test_verify_openai_key.py index a8429d32e..e671641ac 100644 --- a/litellm/tests/test_verify_openai_key.py +++ b/litellm/tests/test_verify_openai_key.py @@ -1,10 +1,20 @@ from litellm import verify_access_key +import os -def bad_key(): +def test_bad_key(): key = "bad-key" response = verify_access_key(key) if response == False: pass else: raise Exception("Bad key was not detected") -bad_key() \ No newline at end of file +test_bad_key() + +def test_good_key(): + key = os.environ['OPENAI_API_KEY'] + response = verify_access_key(key) + if response == True: + pass + else: + raise Exception("Good key did not pass") +test_good_key() \ No newline at end of file diff --git a/litellm/utils.py b/litellm/utils.py index db6cd1e4f..772cb2fe2 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -2566,6 +2566,6 @@ def verify_access_key(access_key:str): ], max_tokens = 10 ) - return False + return True except: - return True \ No newline at end of file + return False \ No newline at end of file