From 4e9aa0d3381fd7a180b4b442c416d3d2cd6fea88 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 28 Nov 2023 17:42:50 -0800 Subject: [PATCH] fix(utils.py): fix bedrock/cohere supported params --- litellm/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/utils.py b/litellm/utils.py index b56890cd2c..7d60480eac 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -2213,7 +2213,7 @@ def get_optional_params( # use the openai defaults if stream: optional_params["stream"] = stream elif "cohere" in model: # cohere models on bedrock - supported_params = ["stream", "temperature", "max_tokens", "logit_bias", "top_p", "frequency_penalty", "presence_penalty", "stop"] + supported_params = ["stream", "temperature", "max_tokens"] _check_valid_arg(supported_params=supported_params) # handle cohere params if stream: