mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-27 06:13:07 +00:00
test(logging): mirror the proxy metadata in the deployment-price regression test
This commit is contained in:
parent
d5b0cdfb3d
commit
ddcaaaa696
1 changed files with 9 additions and 3 deletions
|
|
@ -10385,7 +10385,13 @@ def test_standard_logging_payload_reports_the_deployment_entry_the_request_was_b
|
|||
litellm_call_id="lit9660",
|
||||
function_id="fn",
|
||||
)
|
||||
litellm_params: Final = {"metadata": {"model_group": "lit9660-group", "model_info": deployment_model_info}}
|
||||
litellm_params: Final = {
|
||||
"metadata": {
|
||||
"model_group": "lit9660-group",
|
||||
"deployment": "openai/gpt-5.4-nano",
|
||||
"model_info": deployment_model_info,
|
||||
}
|
||||
}
|
||||
logging_obj.update_environment_variables(
|
||||
model="gpt-5.4-nano", user="", optional_params={}, litellm_params=litellm_params, custom_llm_provider="openai"
|
||||
)
|
||||
|
|
@ -10413,8 +10419,8 @@ def test_standard_logging_payload_reports_the_deployment_entry_the_request_was_b
|
|||
assert response_cost == 10 * input_cost_per_token + 6 * output_cost_per_token
|
||||
assert payload is not None
|
||||
assert payload["model_map_information"] == {
|
||||
"model_map_key": deployment_id,
|
||||
"model_map_value": litellm.get_model_info(model=deployment_id, custom_llm_provider="openai"),
|
||||
"model_map_key": f"openai/{deployment_id}",
|
||||
"model_map_value": litellm.get_model_info(model=f"openai/{deployment_id}", custom_llm_provider="openai"),
|
||||
}
|
||||
model_map_value: Final = payload["model_map_information"]["model_map_value"]
|
||||
assert model_map_value is not None
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue