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

(test) router: raises exception

Browse source
This commit is contained in:
ishaan-jaff 2023-11-23 17:09:31 -08:00
parent 3ee141e1e2
commit 3118101e2c
1 changed files with 13 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

16
litellm/tests/test_router.py
View file

@ -120,6 +120,7 @@ Who among the mentioned figures from Ancient Greece contributed to the domain of
def test_exception_raising():
# this tests if the router raises an exception when invalid params are set
# in this test both deployments have bad keys - Keep this test. It validates if the router raises the most recent exception
litellm.set_verbose=True
import openai
try:
@ -137,9 +138,17 @@ def test_exception_raising():
},
"tpm": 240000,
"rpm": 1800
},
{
"model_name": "gpt-3.5-turbo", # openai model name
"litellm_params": { #
"model": "gpt-3.5-turbo",
"api_key": "bad-key",
},
"tpm": 240000,
"rpm": 1800
}
]
router = Router(model_list=model_list,
redis_host=os.getenv("REDIS_HOST"),
redis_password=os.getenv("REDIS_PASSWORD"),
@ -160,8 +169,9 @@ def test_exception_raising():
print("Test Passed: Caught an OPENAI AUTH Error, Good job. This is what we needed!")
os.environ["AZURE_API_KEY"] = old_api_key
except Exception as e:
os.environ["AZURE_API_KEY"] = old_api_key
print("Got unexpected exception on router!", e)
# test_exception_raising()
test_exception_raising()
def test_reading_key_from_model_list():
@ -207,7 +217,7 @@ def test_reading_key_from_model_list():
print("Test Passed: Caught an OPENAI AUTH Error, Good job. This is what we needed!")
except Exception as e:
print("Got unexpected exception on router!", e)
test_reading_key_from_model_list()
# test_reading_key_from_model_list()
### FUNCTION CALLING