Fix test

cohere-ai · Feb 28, 2025 · 9280b97 · 9280b97
1 parent af25f3b
commit 9280b97
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_client_v2.py b/tests/test_client_v2.py
@@ -79,7 +79,7 @@ def test_chat_tools(self) -> None:
 
         # call the get_weather tool
         tool_result = {"temperature": "30C"}
-        tool_content = [cohere.Content(
+        tool_content = [cohere.TextContent(
             output=tool_result, text="The weather in Toronto is 30C")]
         messages.append(cohere.AssistantChatMessageV2(content=res.message))
         if res.message.tool_calls is not None: