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

test: handle claude instability

Browse source
This commit is contained in:
Krrish Dholakia 2024-11-15 01:11:59 +05:30
parent 2995f3adc2
commit fbba9b464c
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/local_testing/test_completion.py
View file

@ -406,8 +406,13 @@ def test_completion_claude_3_empty_response():
"content": "I was hoping we could chat a bit",
},
]
response = litellm.completion(model="claude-3-opus-20240229", messages=messages)
print(response)
try:
response = litellm.completion(model="claude-3-opus-20240229", messages=messages)
print(response)
except litellm.InternalServerError as e:
pytest.skip(f"InternalServerError - {str(e)}")
except Exception as e:
pytest.fail(f"Error occurred: {e}")
def test_completion_claude_3():