test - setting extra headers for anthropic tool use

2026-08-23 18:25:22 +00:00 · 2024-05-16 14:41:26 -07:00
parent 23bcd03904
commit e19e475c9f
1 changed files with 2 additions and 1 deletions
@@ -278,7 +278,8 @@ def test_completion_claude_3_function_call():
            model="anthropic/claude-3-opus-20240229",
            messages=messages,
            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
        print(response)