forked from phoenix/litellm-mirror
renamed and added test_good_key()
This commit is contained in:
parent
1b51016680
commit
b26933301d
2 changed files with 14 additions and 4 deletions
|
@ -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()
|
|
@ -2566,6 +2566,6 @@ def verify_access_key(access_key:str):
|
|||
],
|
||||
max_tokens = 10
|
||||
)
|
||||
return False
|
||||
return True
|
||||
except:
|
||||
return True
|
||||
return False
|
Loading…
Add table
Add a link
Reference in a new issue