From c8cedbed202adca233c797fb46fb336e193d8c1f Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sun, 12 Jan 2025 16:31:04 -0800 Subject: [PATCH] fix img gen cost --- litellm/main.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/main.py b/litellm/main.py index 26eebfb170..f094d74877 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -4434,6 +4434,8 @@ def image_generation( # noqa: PLR0915 }, 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: return mock_image_generation(model=model, mock_response=mock_response)