Fix

cohere-ai · Aug 9, 2024 · b1feebc · b1feebc
1 parent 9a64d58
commit b1feebc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_client_v2.py b/tests/test_client_v2.py
@@ -52,9 +52,9 @@ def test_chat_stream(self) -> None:
 
     def test_chat_documents(self) -> None:
         documents = [
-            'widget sales 2019: 1 million',
-            'widget sales 2020: 2 million',
-            'widget sales 2021: 4 million'
+            {'title': 'widget sales 2019', 'text': '1 million'},
+            {'title': 'widget sales 2020', 'text': '2 million'},
+            {'title': 'widget sales 2021', 'text': '4 million'}
         ]
         content: typing.List[typing.Union[TextContent, DocumentContent]] = [cohere.v2.TextContent(text="how many widges were sold in 2020?")]
         for doc in documents: