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

test_provider_budgets_e2e_test

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-19 13:57:56 -08:00
parent ba36ecc65a
commit 3d296926ae
1 changed files with 6 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/local_testing/test_provider_budgets.py
View file

@ -60,6 +60,9 @@ async def test_provider_budgets_e2e_test():
"model_info": {"id": "openai-model-id"}, "model_info": {"id": "openai-model-id"},
}, },
], ],
redis_host=os.getenv("REDIS_HOST"),
redis_port=int(os.getenv("REDIS_PORT")),
redis_password=os.getenv("REDIS_PASSWORD"),
) )
response = await router.acompletion( response = await router.acompletion(
@ -106,6 +109,9 @@ async def test_provider_budgets_e2e_test_expect_to_fail():
}, },
}, },
], ],
redis_host=os.getenv("REDIS_HOST"),
redis_port=int(os.getenv("REDIS_PORT")),
redis_password=os.getenv("REDIS_PASSWORD"),
) )
response = await router.acompletion( response = await router.acompletion(