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

fix test_call_with_end_user_over_budget

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-29 20:48:08 -07:00
parent 88c5b3e000
commit 6f50e7681d
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/tests/test_key_generate_prisma.py
View file

@ -467,7 +467,7 @@ def test_call_with_user_over_budget(prisma_client):
asyncio.run(test())
except Exception as e:
error_detail = e.message
assert "Budget has been exceeded" in error_detail
assert "Authentication Error, ExceededBudget:" in error_detail
print(vars(e))
@ -559,7 +559,7 @@ def test_call_with_end_user_over_budget(prisma_client):
asyncio.run(test())
except Exception as e:
error_detail = e.message
assert "Budget has been exceeded" in error_detail
assert "Budget has been exceeded! Current" in error_detail
print(vars(e))
@ -648,7 +648,7 @@ def test_call_with_proxy_over_budget(prisma_client):
error_detail = e.message
else:
error_detail = traceback.format_exc()
assert "Budget has been exceeded" in error_detail
assert "Authentication Error, ExceededBudget:" in error_detail
print(vars(e))
@ -726,7 +726,7 @@ def test_call_with_user_over_budget_stream(prisma_client):
asyncio.run(test())
except Exception as e:
error_detail = e.message
assert "Budget has been exceeded" in error_detail
assert "Authentication Error, ExceededBudget:" in error_detail
print(vars(e))
@ -823,7 +823,7 @@ def test_call_with_proxy_over_budget_stream(prisma_client):
asyncio.run(test())
except Exception as e:
error_detail = e.message
assert "Budget has been exceeded" in error_detail
assert "Authentication Error, ExceededBudget:" in error_detail
print(vars(e))