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

test - raise correct Auth exception for anthropic API

Browse source
This commit is contained in:
Ishaan Jaff 2024-07-16 20:44:40 -07:00
parent dc968a7213
commit dbca6bbb99
1 changed files with 27 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

27
litellm/tests/test_exceptions.py
View file

@ -388,6 +388,33 @@ def test_completion_openai_exception():
# test_completion_openai_exception()
def test_anthropic_openai_exception():
# test if anthropic raises litellm.AuthenticationError
try:
litellm.set_verbose = True
## Test azure call
old_azure_key = os.environ["ANTHROPIC_API_KEY"]
os.environ.pop("ANTHROPIC_API_KEY")
response = completion(
model="anthropic/claude-3-sonnet-20240229",
messages=[{"role": "user", "content": "hello"}],
)
print(f"response: {response}")
print(response)
except litellm.AuthenticationError as e:
os.environ["ANTHROPIC_API_KEY"] = old_azure_key
print("Exception vars=", vars(e))
assert (
"Missing Anthropic API Key - A call is being made to anthropic but no key is set either in the environment variables or via params"
in e.message
)
print(
"ANTHROPIC_API_KEY: good job got the correct error for ANTHROPIC_API_KEY when key not set"
)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
def test_completion_mistral_exception():
# test if mistral/mistral-tiny raises openai.AuthenticationError
try: