diff --git a/litellm/llms/mistral/mistral_chat_transformation.py b/litellm/llms/mistral/mistral_chat_transformation.py index 3e7a97c92f2d..67d88868d350 100644 --- a/litellm/llms/mistral/mistral_chat_transformation.py +++ b/litellm/llms/mistral/mistral_chat_transformation.py @@ -80,6 +80,7 @@ def get_supported_openai_params(self, model: str) -> List[str]: "temperature", "top_p", "max_tokens", + "max_completion_tokens", "tools", "tool_choice", "seed", @@ -105,6 +106,10 @@ def map_openai_params( for param, value in non_default_params.items(): if param == "max_tokens": optional_params["max_tokens"] = value + if ( + param == "max_completion_tokens" + ): # max_completion_tokens should take priority + optional_params["max_tokens"] = value if param == "tools": optional_params["tools"] = value if param == "stream" and value is True: