phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

update testing key

Browse source
This commit is contained in:
ishaan-jaff 2023-09-12 11:50:14 -07:00
parent 8b60440b4b
commit 33f0da2d69
1 changed files with 3 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/tests/test_utils.py
View file

@ -10,7 +10,7 @@ sys.path.insert(
) # Adds the parent directory to the system path ) # Adds the parent directory to the system path
import pytest import pytest
import litellm import litellm
from litellm.utils import trim_messages, get_token_count, get_valid_models from litellm.utils import trim_messages, get_token_count, get_valid_models, check_valid_key
# Assuming your trim_messages, shorten_message_to_fit_limit, and get_token_count functions are all in a module named 'message_utils' # Assuming your trim_messages, shorten_message_to_fit_limit, and get_token_count functions are all in a module named 'message_utils'
@ -83,12 +83,9 @@ def test_get_valid_models():
# test_get_valid_models() # test_get_valid_models()
# def test_bad_key(): # def test_bad_key():
# key = "bad-key" # key = "bad-key"
# response = verify_access_key(key) # response = check_valid_key(key)
# if response == False: # if response == False:
# pass # pass
# else: # else:
@ -97,7 +94,7 @@ def test_get_valid_models():
# def test_good_key(): # def test_good_key():
# key = os.environ['OPENAI_API_KEY'] # key = os.environ['OPENAI_API_KEY']
# response = verify_access_key(key) # response = check_valid_key(key)
# if response == True: # if response == True:
# pass # pass
# else: # else: