renamed and added test_good_key()

This commit is contained in:
William Espegren 2023-09-12 19:02:55 +02:00
parent 1b51016680
commit b26933301d
No known key found for this signature in database
GPG key ID: A77EDBFC7CD4D2AE
2 changed files with 14 additions and 4 deletions

View file

@ -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()
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()

View file

@ -2566,6 +2566,6 @@ def verify_access_key(access_key:str):
],
max_tokens = 10
)
return False
return True
except:
return True
return False