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

fix testing for budget manager

Browse source
This commit is contained in:
Krrish Dholakia 2023-09-12 16:11:49 -07:00
parent 7b90ca4d00
commit 27f1051792
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_budget_manager.py
View file

@ -19,7 +19,7 @@ def test_user_budget_enough():
try: try:
user = "1234" user = "1234"
# create a budget for a user # create a budget for a user
budget_manager.create_budget(total_budget=10, user=user) budget_manager.create_budget(total_budget=10, user=user, duration="daily")
# check if a given call can be made # check if a given call can be made
data = { data = {
@ -41,7 +41,7 @@ def test_user_budget_not_enough():
try: try:
user = "12345" user = "12345"
# create a budget for a user # create a budget for a user
budget_manager.create_budget(total_budget=0, user=user) budget_manager.create_budget(total_budget=0, user=user, duration="daily")
# check if a given call can be made # check if a given call can be made
data = { data = {

2
pyproject.toml
View file

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "litellm" name = "litellm"
version = "0.1.606" version = "0.1.607"
description = "Library to easily interface with LLM API providers" description = "Library to easily interface with LLM API providers"
authors = ["BerriAI"] authors = ["BerriAI"]
license = "MIT License" license = "MIT License"