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

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