fix(watsonx): use correct parameter name for tool choice

Ref: BerriAI/litellm#9979
This commit is contained in:
Tomas Dvorak 2025-04-14 14:38:27 +02:00
parent 64bb89c70f
commit 9ae0df4272
3 changed files with 4 additions and 4 deletions

View file

@ -791,7 +791,7 @@ async def test_watsonx_tool_choice(sync_mode):
mock_completion.assert_called_once()
print(mock_completion.call_args.kwargs)
json_data = json.loads(mock_completion.call_args.kwargs["data"])
json_data["tool_choice_options"] == "auto"
json_data["tool_choice_option"] == "auto"
except Exception as e:
print(e)
if "The read operation timed out" in str(e):