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

fix response_cost_calculator

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

2
litellm/cost_calculator.py
View file

@ -854,6 +854,8 @@ def response_cost_calculator(
if isinstance(response_object, BaseModel):
response_object._hidden_params["optional_params"] = optional_params
if isinstance(response_object, ImageResponse):
if base_model is not None:
model = base_model
response_cost = completion_cost(
completion_response=response_object,
model=model,