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 from litellm import verify_access_key
import os
def bad_key(): def test_bad_key():
key = "bad-key" key = "bad-key"
response = verify_access_key(key) response = verify_access_key(key)
if response == False: if response == False:
pass pass
else: else:
raise Exception("Bad key was not detected") 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 max_tokens = 10
) )
return False
except:
return True return True
except:
return False