From e6cd8963e82bfb72af0de7ff5b1ae2f9e09de86c Mon Sep 17 00:00:00 2001 From: William Espegren Date: Tue, 12 Sep 2023 19:02:55 +0200 Subject: [PATCH] renamed and added test_good_key() --- litellm/tests/test_verify_openai_key.py | 14 ++++++++++++-- litellm/utils.py | 4 ++-- 2 files changed, 14 insertions(+), 4 deletions(-) diff --git a/litellm/tests/test_verify_openai_key.py b/litellm/tests/test_verify_openai_key.py index a8429d32e8..e671641ac0 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 db6cd1e4f0..772cb2fe24 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