test(test_acooldowns_router.py): fix tpm

This commit is contained in:
Krrish Dholakia 2024-04-03 21:24:21 -07:00
parent 6bc48d7e8d
commit ef2f6ef6a2

View file

@ -146,7 +146,7 @@ def test_cooldown_same_model_name():
"api_key": os.getenv("AZURE_API_KEY"), "api_key": os.getenv("AZURE_API_KEY"),
"api_version": os.getenv("AZURE_API_VERSION"), "api_version": os.getenv("AZURE_API_VERSION"),
"api_base": os.getenv("AZURE_API_BASE"), "api_base": os.getenv("AZURE_API_BASE"),
"tpm": 0.000001, "tpm": 1,
}, },
}, },
] ]
@ -181,4 +181,4 @@ def test_cooldown_same_model_name():
pytest.fail(f"Got unexpected exception on router! - {e}") pytest.fail(f"Got unexpected exception on router! - {e}")
test_cooldown_same_model_name() # test_cooldown_same_model_name()