From ee3c584b0d69bf5189c5dc4806ebec19c783e17b Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 16 Jan 2024 15:46:19 -0800
Subject: [PATCH] (ci/cd) fixes for ollama test

---
 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 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}")