(ci/cd) fix olama hosted testing

This commit is contained in:
ishaan-jaff
2024-01-16 12:27:16 -08:00
parent 4929a6c95c
commit d6f0cb8756
2 changed files with 4 additions and 0 deletions
+2
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}")
+2
View File
@@ -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}")