phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix img gen cost

Browse source
This commit is contained in:
Ishaan Jaff 2025-01-12 16:31:04 -08:00
parent b71021f1bf
commit c8cedbed20
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

@ -4434,6 +4434,8 @@ def image_generation( # noqa: PLR0915
}, },
custom_llm_provider=custom_llm_provider, custom_llm_provider=custom_llm_provider,
) )
if "custom_llm_provider" not in logging.model_call_details:
logging.model_call_details["custom_llm_provider"] = custom_llm_provider
if mock_response is not None: if mock_response is not None:
return mock_image_generation(model=model, mock_response=mock_response) return mock_image_generation(model=model, mock_response=mock_response)