From f41396ca95646a9d7b690d6124d146accee42fcc Mon Sep 17 00:00:00 2001 From: Billy Trend Date: Thu, 19 Sep 2024 14:55:15 +0100 Subject: [PATCH] Fixes --- tests/test_client_v2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_client_v2.py b/tests/test_client_v2.py index ffb5825b7..235fc9436 100644 --- a/tests/test_client_v2.py +++ b/tests/test_client_v2.py @@ -14,12 +14,12 @@ class TestClientV2(unittest.TestCase): def test_chat(self) -> None: - response = co.chat(model="command-r-plus", messages=[cohere.UserMessage(message="hello world!")]) + response = co.chat(model="command-r-plus", messages=[cohere.UserChatMessageV2(content="hello world!")]) print(response.message) def test_chat_stream(self) -> None: - stream = co.chat_stream(model="command-r-plus", messages=[cohere.UserMessage(message="hello world!")]) + stream = co.chat_stream(model="command-r-plus", messages=[cohere.UserChatMessageV2(content="hello world!")]) events = set()