mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 03:04:13 +00:00
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
|
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()
|
|
@ -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
|
Loading…
Add table
Add a link
Reference in a new issue