phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix test_cost_azure_gpt_35

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-12 17:47:31 -08:00
parent 96ec2a5a19
commit 855761d041
1 changed files with 0 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/cost_calculator.py
View file

@ -459,8 +459,6 @@ def _select_model_name_for_cost_calc(
elif return_model is None and hasattr(completion_response, "get"): elif return_model is None and hasattr(completion_response, "get"):
return_model = completion_response.get("model", "") # type: ignore return_model = completion_response.get("model", "") # type: ignore
elif hasattr(completion_response, "model"):
return_model = completion_response.model # type: ignore
hidden_params = getattr(completion_response, "_hidden_params", None) hidden_params = getattr(completion_response, "_hidden_params", None)
if hidden_params is not None: if hidden_params is not None: