phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor(azure.py): working client init logic in azure image generation

Browse source
This commit is contained in:
Krrish Dholakia 2025-03-11 14:22:25 -07:00
parent 4f4507ccc0
commit 26fca929da
2 changed files with 10 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/main.py
View file

@ -4544,6 +4544,8 @@ def image_generation( # noqa: PLR0915
**non_default_params,
)
litellm_params_dict = get_litellm_params(**kwargs)
logging: Logging = litellm_logging_obj
logging.update_environment_variables(
model=model,
@ -4614,6 +4616,7 @@ def image_generation( # noqa: PLR0915
aimg_generation=aimg_generation,
client=client,
headers=headers,
litellm_params=litellm_params_dict,
)
elif (
custom_llm_provider == "openai"