mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-26 03:04:13 +00:00
fix(watsonx): use correct parameter name for tool choice
Ref: BerriAI/litellm#9979
This commit is contained in:
parent
64bb89c70f
commit
9ae0df4272
3 changed files with 4 additions and 4 deletions
|
@ -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):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue