Force single step

cohere-ai · May 22, 2024 · dd970bc · dd970bc
1 parent d763bbf
commit dd970bc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/tests/test_async_client.py b/tests/test_async_client.py
@@ -367,6 +367,7 @@ async def test_tool_use(self) -> None:
             message="",
             tools=tools,
             tool_results=tool_results,
+            force_single_step=True,
             model="command-nightly",
         )
 

diff --git a/tests/test_client.py b/tests/test_client.py
@@ -381,6 +381,7 @@ def test_tool_use(self) -> None:
             message="",
             tools=tools,
             tool_results=tool_results,
+            force_single_step=True,
             model="command-nightly",
         )