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
from copy import deepcopy
import httpx
import litellm
from ._logging import verbose_logger
from litellm import ( # type: ignore

View file

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