phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(base_aws_llm.py): remove region name before sending in args (#8998)

Browse source 
* fix(base_aws_llm.py): remove region name before sending in args

* fix(base_aws_llm.py): fix optional param pop position

* fix: fix linting error
This commit is contained in:
Krish Dholakia 2025-03-04 23:05:28 -08:00 • committed by GitHub
parent 4cd35205ae
commit ecef915ac9
5 changed files with 39 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/main.py
View file

@ -4521,6 +4521,7 @@ def image_generation( # noqa: PLR0915
non_default_params = {
k: v for k, v in kwargs.items() if k not in default_params
} # model-specific params - pass them straight to the model/provider
optional_params = get_optional_params_image_gen(
model=model,
n=n,
@ -4532,6 +4533,7 @@ def image_generation( # noqa: PLR0915
custom_llm_provider=custom_llm_provider,
**non_default_params,
)
logging: Logging = litellm_logging_obj
logging.update_environment_variables(
model=model,
@ -4630,6 +4632,7 @@ def image_generation( # noqa: PLR0915
optional_params=optional_params,
model_response=model_response,
aimg_generation=aimg_generation,
client=client,
)
elif custom_llm_provider == "vertex_ai":
vertex_ai_project = (