diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index c90526a140..72861fc292 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}") diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py index 81d0815d69..44da6817eb 100644 --- a/litellm/tests/test_streaming.py +++ b/litellm/tests/test_streaming.py @@ -327,6 +327,8 @@ def test_completion_ollama_hosted_stream(): raise Exception("Empty response received") print(f"complete_response: {complete_response}") except Exception as e: + if "try pulling it first" in str(e): + return pytest.fail(f"Error occurred: {e}")