Update test_client_v2.py

cohere-ai · Nov 22, 2024 · 8806438 · 8806438
1 parent 756515a
commit 8806438
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_client_v2.py b/tests/test_client_v2.py
@@ -14,7 +14,7 @@
 class TestClientV2(unittest.TestCase):
 
     def test_chat(self) -> None:
-        response = co.chat(model="command-r-plus", messages=[cohere.UserChatMessageV2(content="hello world!")])
+        response = co.chat(model="command-r-plus", documents=["yoyo"], messages=[cohere.UserChatMessageV2(content="hello world!")])
 
         print(response.message)