test(test_completion.py): handle replicate api error

2024-06-23 00:06:15 -07:00 · 2024-06-23 00:06:15 -07:00 · d2fe2b30a9
commit d2fe2b30a9
parent b577f65798
1 changed files with 2 additions and 0 deletions
--- 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}")