From 8806438ede60f83d9b47adaf73da007da16677b9 Mon Sep 17 00:00:00 2001 From: billytrend-cohere <144115527+billytrend-cohere@users.noreply.github.com> Date: Fri, 22 Nov 2024 09:31:21 -0500 Subject: [PATCH] Update test_client_v2.py --- tests/test_client_v2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_client_v2.py b/tests/test_client_v2.py index 235fc9436..fcee54981 100644 --- 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)