(ci/cd) fixes for ollama test

This commit is contained in:
ishaan-jaff 2024-01-16 15:46:19 -08:00
parent 87ff4ca55c
commit ee3c584b0d

View file

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