From ce7a937a53db5cbb4c294214f0737ba39ae37c1d Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 16 Sep 2024 17:47:31 -0700 Subject: [PATCH] fix test_all_model_config --- litellm/llms/bedrock/chat/converse_transformation.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm/llms/bedrock/chat/converse_transformation.py b/litellm/llms/bedrock/chat/converse_transformation.py index 5f6141cb4..86f1c102d 100644 --- a/litellm/llms/bedrock/chat/converse_transformation.py +++ b/litellm/llms/bedrock/chat/converse_transformation.py @@ -72,6 +72,7 @@ class AmazonConverseConfig: def get_supported_openai_params(self, model: str) -> List[str]: supported_params = [ "max_tokens", + "max_completion_tokens", "stream", "stream_options", "stop", @@ -179,7 +180,7 @@ class AmazonConverseConfig: ), status_code=400, ) - if param == "max_tokens": + if param == "max_tokens" or param == "max_completion_tokens": optional_params["maxTokens"] = value if param == "stream": optional_params["stream"] = value