test - setting extra headers for anthropic tool use

This commit is contained in:
Ishaan Jaff 2024-05-16 14:41:26 -07:00
parent 23bcd03904
commit e19e475c9f

View file

@ -278,7 +278,8 @@ def test_completion_claude_3_function_call():
model="anthropic/claude-3-opus-20240229", model="anthropic/claude-3-opus-20240229",
messages=messages, messages=messages,
tools=tools, tools=tools,
tool_choice="auto", tool_choice={"type": "tool", "name": "get_weather"},
extra_headers={"anthropic-beta": "tools-2024-05-16"},
) )
# Add any assertions, here to check response args # Add any assertions, here to check response args
print(response) print(response)