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

testing fix - skip rate limit errors from anthropic api

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-01 12:12:50 -07:00
parent bae8be340e
commit dce8e3efb8
1 changed files with 25 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/tests/test_streaming.py
View file

@ -695,6 +695,7 @@ def test_completion_claude_2_stream():
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_acompletion_claude_2_stream(): async def test_acompletion_claude_2_stream():
try:
litellm.set_verbose = True litellm.set_verbose = True
response = await litellm.acompletion( response = await litellm.acompletion(
model="claude-2", model="claude-2",
@ -715,6 +716,10 @@ async def test_acompletion_claude_2_stream():
if complete_response.strip() == "": if complete_response.strip() == "":
raise Exception("Empty response received") raise Exception("Empty response received")
print(f"completion_response: {complete_response}") print(f"completion_response: {complete_response}")
except litellm.RateLimitError:
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")
def test_completion_palm_stream(): def test_completion_palm_stream():