fix tests

cohere-ai · Mar 15, 2024 · d5546a7 · d5546a7
1 parent 490a01b
commit d5546a7
Showing 1 changed file with 1 addition and 0 deletions.
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",