From d2fe2b30a9debdea3d421e726ba2b35aa77b8290 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Sun, 23 Jun 2024 00:06:15 -0700
Subject: [PATCH] test(test_completion.py): handle replicate api error

---
 litellm/tests/test_completion.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index 31ac792d8e..830b3acd38 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -2580,6 +2580,8 @@ async def test_completion_replicate_llama3(sync_mode):
         # Add any assertions here to check the response
         assert isinstance(response, litellm.ModelResponse)
         response_format_tests(response=response)
+    except litellm.APIError as e:
+        pass
     except Exception as e:
         pytest.fail(f"Error occurred: {e}")