test(test_team.py): fix error string asserted

This commit is contained in:
Krrish Dholakia 2024-06-11 18:05:20 -07:00
parent a1ca70a8a7
commit 0ab0111d57
2 changed files with 1 additions and 2 deletions

View file

@ -14,7 +14,6 @@ from functools import partial
import dotenv, traceback, random, asyncio, time, contextvars import dotenv, traceback, random, asyncio, time, contextvars
from copy import deepcopy from copy import deepcopy
import httpx import httpx
import litellm import litellm
from ._logging import verbose_logger from ._logging import verbose_logger
from litellm import ( # type: ignore from litellm import ( # type: ignore

View file

@ -578,4 +578,4 @@ async def test_users_in_team_budget():
except Exception as e: except Exception as e:
print("got exception, this is expected") print("got exception, this is expected")
print(e) print(e)
assert "Crossed spend within team" in str(e) assert "Budget has been exceeded" in str(e)