diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index c90526a1405..72861fc2929 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -763,6 +763,8 @@ def test_completion_ollama_hosted(): litellm.request_timeout = None pass except Exception as e: + if "try pulling it first" in str(e): + return pytest.fail(f"Error occurred: {e}")