test(test_router.py): fix testing

This commit is contained in:
Krrish Dholakia 2024-05-21 17:31:31 -07:00
parent 988970f4c2
commit cc41db018f

View file

@ -762,7 +762,7 @@ def test_router_region_pre_call_check(allowed_model_region):
model="gpt-3.5-turbo", model="gpt-3.5-turbo",
healthy_deployments=model_list, healthy_deployments=model_list,
messages=[{"role": "user", "content": "Hey!"}], messages=[{"role": "user", "content": "Hey!"}],
allowed_model_region=allowed_model_region, request_kwargs={"allowed_model_region": allowed_model_region},
) )
if allowed_model_region is None: if allowed_model_region is None: