diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py index 5a0cbdf6ce..7465e98487 100644 --- a/litellm/proxy/_types.py +++ b/litellm/proxy/_types.py @@ -642,6 +642,7 @@ class ResetTeamBudgetRequest(LiteLLMBase): team_id: str spend: float budget_reset_at: datetime + updated_at: datetime class DeleteTeamRequest(LiteLLMBase): diff --git a/litellm/proxy/utils.py b/litellm/proxy/utils.py index 70aba55809..563496d66b 100644 --- a/litellm/proxy/utils.py +++ b/litellm/proxy/utils.py @@ -50,6 +50,13 @@ from typing_extensions import overload def print_verbose(print_statement): + """ + Prints the given `print_statement` to the console if `litellm.set_verbose` is True. + Also logs the `print_statement` at the debug level using `verbose_proxy_logger`. + + :param print_statement: The statement to be printed and logged. + :type print_statement: Any + """ verbose_proxy_logger.debug(print_statement) if litellm.set_verbose: print(f"LiteLLM Proxy: {print_statement}") # noqa @@ -194,7 +201,7 @@ class ProxyLogging: 2. /embeddings 3. /image/generation """ - print_verbose(f"Inside Proxy Logging Pre-call hook!") + print_verbose("Inside Proxy Logging Pre-call hook!") ### ALERTING ### asyncio.create_task( self.slack_alerting_instance.response_taking_too_long(request_data=data) @@ -2098,6 +2105,7 @@ async def reset_budget(prisma_client: PrismaClient): team_id=team.team_id, spend=0.0, budget_reset_at=now + timedelta(seconds=duration_s), + updated_at=now, ) team_reset_requests.append(reset_team_budget_request) await prisma_client.update_data(