From ffbdaf868fb004a9f63608a46c6627f79917f231 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Thu, 14 Nov 2024 22:31:22 +0530
Subject: [PATCH] test: fix test

---
 tests/local_testing/test_completion.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/local_testing/test_completion.py b/tests/local_testing/test_completion.py
index a0e961070..5b9a534fa 100644
--- a/tests/local_testing/test_completion.py
+++ b/tests/local_testing/test_completion.py
@@ -917,6 +917,9 @@ def test_completion_base64(model):
     except litellm.ServiceUnavailableError as e:
         print("got service unavailable error: ", e)
         pass
+    except litellm.InternalServerError as e:
+        print("got internal server error: ", e)
+        pass
     except Exception as e:
         if "500 Internal error encountered.'" in str(e):
             pass