From 340a63bace76d4f833355cc3a5b5e4996f99740f Mon Sep 17 00:00:00 2001 From: Krish Dholakia Date: Thu, 27 Mar 2025 22:16:21 -0700 Subject: [PATCH] fix(mistral_chat_transformation.py): add missing comma (#9606) --- litellm/llms/mistral/mistral_chat_transformation.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/litellm/llms/mistral/mistral_chat_transformation.py b/litellm/llms/mistral/mistral_chat_transformation.py index 3e7a97c92f2..67d88868d35 100644 --- a/litellm/llms/mistral/mistral_chat_transformation.py +++ b/litellm/llms/mistral/mistral_chat_transformation.py @@ -80,6 +80,7 @@ class MistralConfig(OpenAIGPTConfig): "temperature", "top_p", "max_tokens", + "max_completion_tokens", "tools", "tool_choice", "seed", @@ -105,6 +106,10 @@ class MistralConfig(OpenAIGPTConfig): 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: