From 82fd95d550ffe8d9633bc87c5dd75e1fd7652b39 Mon Sep 17 00:00:00 2001 From: cf3901646 Date: Wed, 21 Oct 2026 05:35:07 +0000 Subject: [PATCH] fix(image_endpoints): apply key/team router_settings to /v1/images/generations endpoint Signed-off-by: cf3901646 --- litellm/proxy/image_endpoints/endpoints.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/litellm/proxy/image_endpoints/endpoints.py b/litellm/proxy/image_endpoints/endpoints.py index 28fb9baf2aa..0ab56d99fb6 100644 --- a/litellm/proxy/image_endpoints/endpoints.py +++ b/litellm/proxy/image_endpoints/endpoints.py @@ -150,6 +150,27 @@ async def image_generation( data["prompt"] = get_str_from_messages(messages) data.pop("messages", None) + ## APPLY KEY/TEAM ROUTER SETTINGS ## + # Read key/team router_settings and apply them to route_request + # This matches the behavior in base_process_llm_request + if proxy_config is not None: + try: + from litellm.proxy.common_request_processing import ProxyBaseLLMRequestProcessing + + # Create a temporary processor to get router_settings_override + temp_processor = ProxyBaseLLMRequestProcessing(data=data) + router_settings_override = temp_processor.common_processing_pre_call_logic( + user_api_key_dict=user_api_key_dict, + data=data, + route_type="aimage_generation", + proxy_config=proxy_config, + ) + if router_settings_override: + data["router_settings_override"] = router_settings_override + except Exception: + # If reading router_settings fails, continue without override + pass + ## ROUTE TO CORRECT ENDPOINT ## llm_call: Final = await route_request( data=data,