From 4b1389501d3f1418077a63b1fac2151a37c8df89 Mon Sep 17 00:00:00 2001 From: Chesars Date: Tue, 3 Mar 2026 15:58:23 -0300 Subject: [PATCH] fix: move _set_usage_and_cost outside try/except in OpenRouter image edit Prevents discarding already-parsed images when usage calculation fails. --- litellm/llms/openrouter/image_edit/transformation.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/llms/openrouter/image_edit/transformation.py b/litellm/llms/openrouter/image_edit/transformation.py index 19931f00245..ed5e6ae67d5 100644 --- a/litellm/llms/openrouter/image_edit/transformation.py +++ b/litellm/llms/openrouter/image_edit/transformation.py @@ -242,9 +242,6 @@ class OpenRouterImageEditConfig(BaseImageEditConfig): ) ) - self._set_usage_and_cost(model_response, response_json, model) - return model_response - except Exception as e: raise OpenRouterException( message=f"Error transforming OpenRouter image edit response: {str(e)}", @@ -252,6 +249,9 @@ class OpenRouterImageEditConfig(BaseImageEditConfig): headers={}, ) + self._set_usage_and_cost(model_response, response_json, model) + return model_response + def get_error_class( self, error_message: str, status_code: int, headers: Union[dict, httpx.Headers] ) -> BaseLLMException: