mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-24 00:52:24 +00:00
test: narrow Azure parity exception assertion
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
parent
bfb4a8a2b3
commit
df89665919
1 changed files with 3 additions and 1 deletions
|
|
@ -89,7 +89,9 @@ def test_azure_key_vault_matches_rust_parity_fixture() -> None:
|
|||
value=secret,
|
||||
)
|
||||
if case.expected.missing or case.expected.error:
|
||||
with pytest.raises(Exception):
|
||||
with pytest.raises(
|
||||
AzureResourceNotFoundError if case.expected.missing else AzureHttpResponseError
|
||||
):
|
||||
get_secret_from_manager(
|
||||
secret_name=case.secret_name,
|
||||
key_manager=KeyManagementSystem.AZURE_KEY_VAULT.value,
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue