From d6f0cb87566f2e331c75670fb8e0b05a1ff61caf Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 16 Jan 2024 12:27:16 -0800 Subject: [PATCH] (ci/cd) fix olama hosted testing --- litellm/tests/test_completion.py | 2 ++ litellm/tests/test_streaming.py | 2 ++ 2 files changed, 4 insertions(+) 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}")