scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-13 08:52:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 353

fix(image_endpoints): apply key/team router_settings to /v1/images/generations endpoint

Browse source 
Signed-off-by: cf3901646 <cf3901646@users.noreply.github.com>
This commit is contained in:
cf3901646 2026-10-21 05:35:07 +00:00
parent 9fbeebc988
commit 82fd95d550
1 changed files with 21 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
litellm/proxy/image_endpoints/endpoints.py
View file

@ -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,