phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(ci/cd) retry hosted ollama + stream test 3 times

Browse source
This commit is contained in:
ishaan-jaff 2024-01-05 18:02:20 +05:30
parent 0eb899c087
commit 6f9d3fc3bc
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_streaming.py
View file

@ -306,6 +306,8 @@ def test_completion_ollama_hosted_stream():
model="ollama/phi", model="ollama/phi",
messages=messages, messages=messages,
max_tokens=10, max_tokens=10,
num_retries=3,
timeout=90,
api_base="https://test-ollama-endpoint.onrender.com", api_base="https://test-ollama-endpoint.onrender.com",
stream=True, stream=True,
) )