scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-20 00:11:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 134

(ci/cd) fix olama hosted testing

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_completion.py
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
litellm/tests/test_streaming.py
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}")