phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix llama index validate environment error

Browse source
This commit is contained in:
Krrish Dholakia 2023-09-26 17:53:22 -07:00
parent 3cd8ab85db
commit 7f62b5b6e0
4 changed files with 15 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/tests/test_utils.py
View file

@ -10,7 +10,7 @@ sys.path.insert(
) # Adds the parent directory to the system path
import pytest
import litellm
from litellm.utils import trim_messages, get_token_count, get_valid_models, check_valid_key
from litellm.utils import trim_messages, get_token_count, get_valid_models, check_valid_key, validate_environment
# Assuming your trim_messages, shorten_message_to_fit_limit, and get_token_count functions are all in a module named 'message_utils'
@ -93,3 +93,12 @@ def test_good_key():
key = os.environ['OPENAI_API_KEY']
response = check_valid_key(model="gpt-3.5-turbo", api_key=key)
assert(response == True)
# test validate environment
def test_validate_environment_empty_model():
api_key = validate_environment()
if api_key is None:
raise Exception()
# test_validate_environment_empty_model()