phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(ci/cd) retry cloudflare request 3 times

Browse source
This commit is contained in:
ishaan-jaff 2024-01-05 12:40:53 +05:30
parent 83b31141c6
commit 113b5e7284
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_completion.py
View file

@ -1933,6 +1933,7 @@ def test_completion_cloudflare():
model="cloudflare/@cf/meta/llama-2-7b-chat-int8",
messages=[{"content": "what llm are you", "role": "user"}],
max_tokens=15,
num_retries=3,
)
print(response)
@ -1940,7 +1941,7 @@ def test_completion_cloudflare():
pytest.fail(f"Error occurred: {e}")
# test_completion_cloudflare()
test_completion_cloudflare()
def test_moderation():