test(proxy): keep the utils test diff to the new send_email case

This commit is contained in:
Rameshwaran Mohan 2026-09-15 04:49:37 +05:30
parent 7712f4f0ae
commit a03be776ea

View file

@ -2110,7 +2110,9 @@ def test_create_model_info_response_resolves_alias_to_deployment_model():
]
)
response = create_model_info_response(model_id="bedrock-claude-opus-5", provider="openai", llm_router=router)
response = create_model_info_response(
model_id="bedrock-claude-opus-5", provider="openai", llm_router=router
)
finally:
litellm.model_cost.clear()
litellm.model_cost.update(saved_model_cost)
@ -2139,7 +2141,9 @@ def test_create_model_info_response_keeps_exact_alias_over_generalized_deploymen
]
)
response = create_model_info_response(model_id="claude-opus-5", provider="openai", llm_router=router)
response = create_model_info_response(
model_id="claude-opus-5", provider="openai", llm_router=router
)
finally:
litellm.model_cost.clear()
litellm.model_cost.update(saved_model_cost)
@ -2163,7 +2167,9 @@ def test_create_model_info_response_falls_back_to_alias_for_opaque_deployment_na
]
)
response = create_model_info_response(model_id="gpt-4o", provider="openai", llm_router=router)
response = create_model_info_response(
model_id="gpt-4o", provider="openai", llm_router=router
)
finally:
litellm.model_cost.clear()
litellm.model_cost.update(saved_model_cost)
@ -2188,7 +2194,9 @@ def test_create_model_info_response_resolves_mode_through_deployment_model():
]
)
response = create_model_info_response(model_id="my-embeddings", provider="openai", llm_router=router)
response = create_model_info_response(
model_id="my-embeddings", provider="openai", llm_router=router
)
finally:
litellm.model_cost.clear()
litellm.model_cost.update(saved_model_cost)
@ -2266,9 +2274,7 @@ async def test_post_call_failure_hook_redacts_traceback_before_callbacks(monkeyp
with patch.object(proxy_logging_obj, "update_request_status", new=AsyncMock()):
await proxy_logging_obj.post_call_failure_hook(
request_data={"metadata": {}},
original_exception=HTTPException(
status_code=400, detail="Upstream passthrough request failed with status 400"
),
original_exception=HTTPException(status_code=400, detail="Upstream passthrough request failed with status 400"),
user_api_key_dict=UserAPIKeyAuth(),
traceback_str=upstream_traceback,
)