From d5546a735466021e7a1d23e333cc85860004c6ea Mon Sep 17 00:00:00 2001
From: Harry M <127103098+harry-cohere@users.noreply.github.com>
Date: Fri, 15 Mar 2024 12:30:12 +0000
Subject: [PATCH] fix tests

---
 tests/test_client.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/test_client.py b/tests/test_client.py
index 2ab148bdc..58b3d5b66 100644
--- 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",