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

fix test_in_memory_redis_sync_e2e

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-23 16:48:36 -08:00
parent 33a0744abe
commit e5c7189922
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/local_testing/test_router_provider_budgets.py
View file

@ -291,7 +291,7 @@ async def test_in_memory_redis_sync_e2e():
router = Router( router = Router(
model_list=[ model_list=[
{ {
"model_name": "gpt-3.5-turbo", "model_name": "gpt-3.5-turbo-very-new",
"litellm_params": { "litellm_params": {
"model": "openai/gpt-3.5-turbo", "model": "openai/gpt-3.5-turbo",
}, },
@ -317,7 +317,7 @@ async def test_in_memory_redis_sync_e2e():
# Make a completion call to trigger spend tracking # Make a completion call to trigger spend tracking
response = await router.acompletion( response = await router.acompletion(
model="gpt-3.5-turbo", model="gpt-3.5-turbo-very-new",
messages=[{"role": "user", "content": "Hello"}], messages=[{"role": "user", "content": "Hello"}],
mock_response="Hello there!", mock_response="Hello there!",
) )