From cda78a5da0b2d705fb5ac56193c7508dc1ba7c4f Mon Sep 17 00:00:00 2001 From: Lucca Zenobio Date: Mon, 25 Mar 2024 13:08:17 -0300 Subject: [PATCH] update --- litellm/llms/bedrock.py | 2 +- litellm/utils.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/llms/bedrock.py b/litellm/llms/bedrock.py index a32cee381ec..d13301910cf 100644 --- a/litellm/llms/bedrock.py +++ b/litellm/llms/bedrock.py @@ -128,7 +128,7 @@ class AmazonAnthropicClaude3Config: } def get_supported_openai_params(self): - return ["max_tokens", "tools", "tool_choice", "stream"] + return ["max_tokens", "tools", "tool_choice", "stream", "extra_headers"] def map_openai_params(self, non_default_params: dict, optional_params: dict): for param, value in non_default_params.items(): diff --git a/litellm/utils.py b/litellm/utils.py index 4124ea4370a..57327473d88 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -5146,7 +5146,7 @@ def get_supported_openai_params(model: str, custom_llm_provider: str): """ if custom_llm_provider == "bedrock": if model.startswith("anthropic.claude-3"): - return litellm.AmazonAnthropicClaude3Config().get_supported_openai_params() + ["extra_headers"] + return litellm.AmazonAnthropicClaude3Config().get_supported_openai_params() elif model.startswith("anthropic"): return litellm.AmazonAnthropicConfig().get_supported_openai_params() elif model.startswith("ai21"):