From d5a1cc282e21ecffc2e05fca1e7e57009aa4eaa2 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 10 Jun 2024 09:29:38 -0700
Subject: [PATCH] test fix - cohere API Made a breaking change

---
 litellm/tests/test_completion.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index a669dccfec..2428cbf48d 100644
--- 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: