scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-12 08:42:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 207

fix(proxy): annotate function_setup rebind and update test_img_gen mock expectation

Browse source
This commit is contained in:
Pyhroff 2026-10-23 12:25:05 +05:30
parent 11493298f2
commit 264fb2eb1a
2 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -135,7 +135,7 @@ async def image_generation(
data["model"] = litellm.model_alias_map[data["model"]]
### LOGGING OBJECT SETUP ### - ensures router rejected requests are dispatched to failure callbacks
logging_obj, data = litellm.utils.function_setup(
logging_obj, data = litellm.utils.function_setup( # rebind-ok: function_setup returns updated data dict
original_function="aimage_generation",
rules_obj=litellm.utils.Rules(),
start_time=start_time,

1
tests/unit/proxy/test_proxy_server.py
View file

@ -1004,6 +1004,7 @@ def test_img_gen(mock_aimage_generation, client_no_auth):
size="1024x1024",
imageConfig={"aspectRatio": "9:16", "imageSize": "1K"},
litellm_call_id=mock.ANY,
litellm_logging_obj=mock.ANY,
metadata=mock.ANY,
proxy_server_request=mock.ANY,
secret_fields=mock.ANY,