phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

extra headers

Browse source
This commit is contained in:
Lucca Zenobio 2024-03-21 10:43:27 -03:00
parent 872ff6176d
commit 0c0780be83
4 changed files with 21 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/utils.py
View file

@ -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()
return litellm.AmazonAnthropicClaude3Config().get_supported_openai_params() + ["extra_headers"]
elif model.startswith("anthropic"):
return litellm.AmazonAnthropicConfig().get_supported_openai_params()
elif model.startswith("ai21"):