From 9dd7b795ca66ce2bb3a9653e63ad1755c9d6c5bb Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Fri, 8 Nov 2024 14:18:23 -0800
Subject: [PATCH] fix: Handle NotFoundError in send chat test with single print
 call

---
 tests/basic/test_sendchat.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/basic/test_sendchat.py b/tests/basic/test_sendchat.py
index 7bc1751e3..6b2684a40 100644
--- a/tests/basic/test_sendchat.py
+++ b/tests/basic/test_sendchat.py
@@ -84,10 +84,10 @@ class TestSendChat(unittest.TestCase):
         mock = MagicMock()
         mock.status_code = 400
 
-        mock_completion.side_effect = litellm.InvalidRequestError(
+        mock_completion.side_effect = litellm.NotFoundError(
             message="Invalid request", llm_provider="test_provider", model="test_model"
         )
 
         result = simple_send_with_retries(self.mock_model, self.mock_messages)
         assert result is None
-        assert mock_print.call_count == 2  # Error message and description
+        assert mock_print.call_count == 1