fix tests

harry-cohere · harry-cohere · commit d5546a735466 · 2024-03-15T12:30:12.000Z
diff --git a/tests/test_client.py b/tests/test_client.py
@@ -269,6 +269,7 @@ def test_tool_use(self) -> None:
 
         self.assertEqual(cited_response.documents, [
             {
+                "tool_name": "sales_database",
                 "average_sale_value": "404.17",
                 "date": "2023-09-29",
                 "id": "sales_database:0:0",

Original file line number	Diff line number	Diff line change
`@@ -269,6 +269,7 @@ def test_tool_use(self) -> None:`
`269`	`269`
`270`	`270`	`self.assertEqual(cited_response.documents, [`
`271`	`271`	`{`
	`272`	`+ "tool_name": "sales_database",`
`272`	`273`	`"average_sale_value": "404.17",`
`273`	`274`	`"date": "2023-09-29",`
`274`	`275`	`"id": "sales_database:0:0",`