fix - show model group in exceptions

This commit is contained in:
Ishaan Jaff 2024-06-10 20:30:31 -07:00
parent fcd1d077f5
commit b4db464604
2 changed files with 7 additions and 0 deletions

View file

@ -55,8 +55,12 @@ async def test_content_policy_exception_azure():
except litellm.ContentPolicyViolationError as e:
print("caught a content policy violation error! Passed")
print("exception", e)
assert e.litellm_debug_info is not None
assert isinstance(e.litellm_debug_info, str)
assert len(e.litellm_debug_info) > 0
pass
except Exception as e:
print()
pytest.fail(f"An exception occurred - {str(e)}")