test fix - cohere API Made a breaking change

2024-06-10 09:29:38 -07:00 · 2024-06-10 09:29:38 -07:00 · d5a1cc282e
commit d5a1cc282e
parent 3ec907059f
1 changed files with 1 additions and 0 deletions
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@ -530,6 +530,7 @@ def test_completion_cohere_command_r_plus_function_call():
            messages=messages,
            tools=tools,
            tool_choice="auto",
+            force_single_step=True,
        )
        print(second_response)
    except Exception as e: