forked from phoenix/litellm-mirror
(test) hosted ollama - retry 3 times
This commit is contained in:
parent
90973d92bf
commit
0eb899c087
1 changed files with 4 additions and 0 deletions
|
@ -749,10 +749,14 @@ def test_completion_ollama_hosted():
|
|||
model="ollama/phi",
|
||||
messages=messages,
|
||||
max_tokens=10,
|
||||
num_retries=3,
|
||||
timeout=90,
|
||||
api_base="https://test-ollama-endpoint.onrender.com",
|
||||
)
|
||||
# Add any assertions here to check the response
|
||||
print(response)
|
||||
except Timeout as e:
|
||||
pass
|
||||
except Exception as e:
|
||||
pytest.fail(f"Error occurred: {e}")
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue