phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix test_all_model_config

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-16 17:47:31 -07:00
parent 4dcb092d12
commit ce7a937a53
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/llms/bedrock/chat/converse_transformation.py
View file

@ -72,6 +72,7 @@ class AmazonConverseConfig:
def get_supported_openai_params(self, model: str) -> List[str]: def get_supported_openai_params(self, model: str) -> List[str]:
supported_params = [ supported_params = [
"max_tokens", "max_tokens",
"max_completion_tokens",
"stream", "stream",
"stream_options", "stream_options",
"stop", "stop",
@ -179,7 +180,7 @@ class AmazonConverseConfig:
), ),
status_code=400, status_code=400,
) )
if param == "max_tokens": if param == "max_tokens" or param == "max_completion_tokens":
optional_params["maxTokens"] = value optional_params["maxTokens"] = value
if param == "stream": if param == "stream":
optional_params["stream"] = value optional_params["stream"] = value