Skip to content

Commit 7903698

Browse files
author
Chao Huang
committed
style: format files with ruff
1 parent 408980b commit 7903698

3 files changed

Lines changed: 10 additions & 15 deletions

File tree

backend/alembic/versions/h9i0j1k2l3m4_add_teams.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,8 +64,7 @@ def upgrade() -> None:
6464

6565
def downgrade() -> None:
6666
op.execute(
67-
"ALTER TABLE team_members "
68-
"DROP CONSTRAINT IF EXISTS ck_allocated_non_negative"
67+
"ALTER TABLE team_members DROP CONSTRAINT IF EXISTS ck_allocated_non_negative"
6968
)
7069
op.drop_index("ix_team_members_token_id", "team_members")
7170
op.drop_index("ix_team_members_team_id", "team_members")

backend/app/api/admin/endpoints/teams.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -371,7 +371,11 @@ async def update_team(
371371
monthly_reset_policy=request.monthly_reset_policy,
372372
daily_limit_enabled=request.daily_limit_enabled,
373373
)
374-
return {"id": str(team.id), "name": team.name, "monthly_budget_usd": str(team.monthly_budget_usd)}
374+
return {
375+
"id": str(team.id),
376+
"name": team.name,
377+
"monthly_budget_usd": str(team.monthly_budget_usd),
378+
}
375379

376380

377381
@router.delete("/{team_id}", status_code=status.HTTP_204_NO_CONTENT)
@@ -437,9 +441,7 @@ async def add_member(
437441
)
438442

439443

440-
@router.delete(
441-
"/{team_id}/members/{token_id}", status_code=status.HTTP_204_NO_CONTENT
442-
)
444+
@router.delete("/{team_id}/members/{token_id}", status_code=status.HTTP_204_NO_CONTENT)
443445
async def remove_member(
444446
team_id: str,
445447
token_id: str,
@@ -534,9 +536,7 @@ async def transfer_allocation(
534536

535537
# Invalidate caches for both tokens
536538
for tid in (from_uuid, to_uuid):
537-
result = await db.execute(
538-
select(APIToken.token_hash).where(APIToken.id == tid)
539-
)
539+
result = await db.execute(select(APIToken.token_hash).where(APIToken.id == tid))
540540
token_hash = result.scalar_one_or_none()
541541
if token_hash:
542542
await _invalidate_token_cache(token_hash)

backend/app/services/team.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -204,9 +204,7 @@ async def add_member(
204204
await self.db.refresh(member)
205205
return member
206206

207-
async def remove_member(
208-
self, team_id: UUID, token_id: UUID, user_id: UUID
209-
) -> None:
207+
async def remove_member(self, team_id: UUID, token_id: UUID, user_id: UUID) -> None:
210208
team = await self.get_team(team_id, user_id)
211209

212210
result = await self.db.execute(
@@ -260,9 +258,7 @@ async def adjust_member_allocation(
260258
pool = team.monthly_budget_usd - total_others
261259
raise HTTPException(
262260
status_code=400,
263-
detail=(
264-
f"Exceeds budget. Max allocation for this member: ${pool:.2f}"
265-
),
261+
detail=(f"Exceeds budget. Max allocation for this member: ${pool:.2f}"),
266262
)
267263

268264
target.allocated_usd = new_allocated_usd

0 commit comments

Comments
 (0)