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

feat: Anthropic allow users to set anthropic-beta in headers

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-16 14:40:31 -07:00
parent 176630bce9
commit 32815b06cb
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/llms/anthropic.py
View file

@ -93,6 +93,7 @@ class AnthropicConfig:
"max_tokens",
"tools",
"tool_choice",
"extra_headers",
]
def map_openai_params(self, non_default_params: dict, optional_params: dict):
@ -504,7 +505,9 @@ class AnthropicChatCompletion(BaseLLM):
## Handle Tool Calling
if "tools" in optional_params:
_is_function_call = True
headers["anthropic-beta"] = "tools-2024-04-04"
if "anthropic-beta" not in headers:
# default to v1 of "anthropic-beta"
headers["anthropic-beta"] = "tools-2024-04-04"
anthropic_tools = []
for tool in optional_params["tools"]: