diff --git a/litellm/main.py b/litellm/main.py index af0dc1f5b6..8ebf66fd06 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -1097,8 +1097,10 @@ def completion( # type: ignore # noqa: PLR0915 logit_bias=logit_bias, user=user, # params to identify the model - model=model, - base_model=base_model, + model=LitellmCoreRequestUtils.select_model_for_request_transformation( + model=model, + base_model=base_model, + ), custom_llm_provider=custom_llm_provider, response_format=response_format, seed=seed, diff --git a/litellm/utils.py b/litellm/utils.py index bcf22e902c..45fa9fa1fd 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -2847,18 +2847,12 @@ def get_optional_params( # noqa: PLR0915 additional_drop_params=None, messages: Optional[List[AllMessageValues]] = None, thinking: Optional[AnthropicThinkingParam] = None, - base_model: Optional[str] = None, **kwargs, ): # retrieve all parameters passed to the function passed_params = locals().copy() special_params = passed_params.pop("kwargs") - # Use `base_model` for paramter mapping if passed in by user - model = LitellmCoreRequestUtils.select_model_for_request_transformation( - model=model, - base_model=base_model, - ) for k, v in special_params.items(): if k.startswith("aws_") and ( custom_llm_provider != "bedrock" and custom_llm_provider != "sagemaker"