Merge pull request #43 from aws-samples/feat/monthly-auto-recharge

feat: teams, monthly quota, Opus 4.7, UI polish

Author: koljahuang

.gitignore

@@ -101,3 +101,4 @@ npm-debug.log*
 yarn-debug.log*
 yarn-error.log*
 .gitnexus
+kolya-br-proxy-arch/graphify-out/

backend/alembic/versions/f7g8h9i0j1k2_add_monthly_daily_quota.py

@@ -0,0 +1,36 @@
+"""Add monthly quota, daily limit, reset policy, and quota start to api_tokens.
+
+Revision ID: f7g8h9i0j1k2
+Revises: e6f7g8h9i0j1
+Create Date: 2026-04-30
+"""
+
+from alembic import op
+import sqlalchemy as sa
+
+
+revision = "f7g8h9i0j1k2"
+down_revision = "e6f7g8h9i0j1"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    op.add_column(
+        "api_tokens",
+        sa.Column("monthly_quota_usd", sa.Numeric(10, 2), nullable=True),
+    )
+    op.add_column(
+        "api_tokens",
+        sa.Column("monthly_reset_policy", sa.String(20), nullable=True),
+    )
+    op.add_column(
+        "api_tokens",
+        sa.Column("monthly_quota_start", sa.DateTime, nullable=True),
+    )
+
+
+def downgrade() -> None:
+    op.drop_column("api_tokens", "monthly_quota_start")
+    op.drop_column("api_tokens", "monthly_reset_policy")
+    op.drop_column("api_tokens", "monthly_quota_usd")

backend/alembic/versions/g8h9i0j1k2l3_add_record_type_to_usage.py

@@ -0,0 +1,36 @@
+"""Add record_type to usage_records for distinguishing adjustments.
+
+Revision ID: g8h9i0j1k2l3
+Revises: f7g8h9i0j1k2
+Create Date: 2026-04-30
+"""
+
+from alembic import op
+import sqlalchemy as sa
+
+
+revision = "g8h9i0j1k2l3"
+down_revision = "f7g8h9i0j1k2"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    op.add_column(
+        "usage_records",
+        sa.Column(
+            "record_type",
+            sa.String(20),
+            nullable=False,
+            server_default="usage",
+        ),
+    )
+    op.add_column(
+        "usage_records",
+        sa.Column("note", sa.String(500), nullable=True),
+    )
+
+
+def downgrade() -> None:
+    op.drop_column("usage_records", "note")
+    op.drop_column("usage_records", "record_type")

backend/alembic/versions/h9i0j1k2l3m4_add_teams.py

@@ -0,0 +1,73 @@
+"""Add teams and team_members tables.
+
+Revision ID: h9i0j1k2l3m4
+Revises: g8h9i0j1k2l3
+Create Date: 2026-05-01
+"""
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects.postgresql import UUID
+
+
+revision = "h9i0j1k2l3m4"
+down_revision = "g8h9i0j1k2l3"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    op.create_table(
+        "teams",
+        sa.Column("id", UUID(as_uuid=True), nullable=False),
+        sa.Column("user_id", UUID(as_uuid=True), nullable=False),
+        sa.Column("name", sa.String(255), nullable=False),
+        sa.Column("monthly_budget_usd", sa.Numeric(10, 2), nullable=False),
+        sa.Column(
+            "monthly_reset_policy",
+            sa.String(20),
+            nullable=False,
+            server_default="reset",
+        ),
+        sa.Column("monthly_budget_start", sa.DateTime(), nullable=False),
+        sa.Column("is_active", sa.Boolean(), nullable=False, server_default="true"),
+        sa.Column("created_at", sa.DateTime(), nullable=False),
+        sa.Column("updated_at", sa.DateTime(), nullable=False),
+        sa.ForeignKeyConstraint(["user_id"], ["users.id"]),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index("ix_teams_user_id", "teams", ["user_id"])
+
+    op.create_table(
+        "team_members",
+        sa.Column("id", UUID(as_uuid=True), nullable=False),
+        sa.Column("team_id", UUID(as_uuid=True), nullable=False),
+        sa.Column("token_id", UUID(as_uuid=True), nullable=False),
+        sa.Column(
+            "allocated_usd", sa.Numeric(10, 2), nullable=False, server_default="0.00"
+        ),
+        sa.Column("created_at", sa.DateTime(), nullable=False),
+        sa.Column("updated_at", sa.DateTime(), nullable=False),
+        sa.ForeignKeyConstraint(["team_id"], ["teams.id"], ondelete="CASCADE"),
+        sa.ForeignKeyConstraint(["token_id"], ["api_tokens.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+    )
+    op.create_index("ix_team_members_team_id", "team_members", ["team_id"])
+    op.create_index(
+        "ix_team_members_token_id", "team_members", ["token_id"], unique=True
+    )
+    op.execute(
+        "ALTER TABLE team_members "
+        "ADD CONSTRAINT ck_allocated_non_negative CHECK (allocated_usd >= 0)"
+    )
+
+
+def downgrade() -> None:
+    op.execute(
+        "ALTER TABLE team_members DROP CONSTRAINT IF EXISTS ck_allocated_non_negative"
+    )
+    op.drop_index("ix_team_members_token_id", "team_members")
+    op.drop_index("ix_team_members_team_id", "team_members")
+    op.drop_table("team_members")
+    op.drop_index("ix_teams_user_id", "teams")
+    op.drop_table("teams")

backend/alembic/versions/i0j1k2l3m4n5_add_daily_limit_enabled.py

@@ -0,0 +1,31 @@
+"""Add daily_limit_enabled to teams.
+
+Revision ID: i0j1k2l3m4n5
+Revises: h9i0j1k2l3m4
+Create Date: 2026-05-01
+"""
+
+from alembic import op
+import sqlalchemy as sa
+
+
+revision = "i0j1k2l3m4n5"
+down_revision = "h9i0j1k2l3m4"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    op.add_column(
+        "teams",
+        sa.Column(
+            "daily_limit_enabled",
+            sa.Boolean(),
+            nullable=False,
+            server_default="true",
+        ),
+    )
+
+
+def downgrade() -> None:
+    op.drop_column("teams", "daily_limit_enabled")