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

fix - test budget exceeded prisma

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-29 20:31:55 -07:00
parent fd3057f3b8
commit 3e9410ffd6
2 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

2
docs/my-website/docs/proxy/users.md
View file

@ -223,7 +223,7 @@ curl --location 'http://0.0.0.0:4000/chat/completions' \
Error Error
```shell ```shell
{"error":{"message":"Authentication Error, ExceededBudget: User ishaan3 has exceeded their budget. Current spend: 0.0008869999999999999; Max Budget: 0.0001","type":"auth_error","param":"None","code":401}}% {"error":{"message":"Budget has been exceeded: User ishaan3 has exceeded their budget. Current spend: 0.0008869999999999999; Max Budget: 0.0001","type":"auth_error","param":"None","code":401}}%
``` ```
</TabItem> </TabItem>

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