fix linting

This commit is contained in:
Michael Riad Zaky
2026-04-23 12:08:10 -07:00
parent 0bd49ecb8b
commit 46336b1ac3
2 changed files with 8 additions and 5 deletions
+7 -2
View File
@@ -3290,10 +3290,15 @@ async def _check_team_member_budget(
# Per-member override wins; otherwise fall back to the team-level
# default configured via team.metadata["team_member_budget_id"].
team_member_budget: Optional[float] = None
if team_membership is not None and team_membership.litellm_budget_table is not None:
if (
team_membership is not None
and team_membership.litellm_budget_table is not None
):
team_member_budget = team_membership.litellm_budget_table.max_budget
else:
default_budget_id = (team_object.metadata or {}).get("team_member_budget_id")
default_budget_id = (team_object.metadata or {}).get(
"team_member_budget_id"
)
if isinstance(default_budget_id, str):
default_budget = await get_team_member_default_budget(
budget_id=default_budget_id,
+1 -3
View File
@@ -2009,9 +2009,7 @@ async def _init_and_increment_spend_counter(
key=counter_key, value=base_spend
)
await spend_counter_cache.async_increment_cache(
key=counter_key, value=increment
)
await spend_counter_cache.async_increment_cache(key=counter_key, value=increment)
async def update_cache( # noqa: PLR0915