rm sixty nuts mention

.github/workflows/test.yml

@@ -62,18 +62,14 @@ jobs:
       - name: Setup Node.js
         uses: actions/setup-node@v4
         with:
-          node-version: "18"
-          cache: "npm"
+          node-version: '18'
+          cache: 'npm'
           cache-dependency-path: ui/package-lock.json
 
       - name: Install UI dependencies
         working-directory: ./ui
         run: npm ci
 
-      - name: Run UI format check
-        working-directory: ./ui
-        run: npm run format-check
-
       - name: Run UI linting
         working-directory: ./ui
         run: npm run lint

README.md

@@ -34,7 +34,7 @@ sequenceDiagram
 - **Model-Based Pricing** – Convert USD prices in `models.json` to sats using live BTC/USD rates
 - **Admin Dashboard** – Simple HTML interface at `/admin/` to view balances and API keys
 - **Discovery** – Fetch available providers from Nostr relays using NIP-91 protocol
-- **NIP-91 Auto-Announcement** – Automatically announce this provider to Nostr relays when NSEC is provided
+- **Auto-Listing** – Automatically announce this provider to Nostr relays when NSEC is provided
 - **Docker Support** – Provided `Dockerfile` and `compose.yml` for running with an optional Tor hidden service
 
 ## Getting Started
@@ -99,7 +99,7 @@ The most common settings are shown below. See `.env.example` for the full list.
 - `NPUB` – Nostr public key of the proxy
 - `HTTP_URL` – Public-facing URL of the proxy
 - `ONION_URL` – Tor hidden service URL of the proxy
-- `NEXT_PUBLIC_API_URL` - UI Configuration for Next.js frontend (proxy URL, default: 'http://127.0.0.1:8000' )
+- `NEXT_PUBLIC_API_URL` - UI Configuration for Next.js frontend (proxy URL, default: '<http://127.0.0.1:8000>' )
 
 ## Database Migrations
 

migrations/versions/5f6ac1e4fa9f_the_refactor.py

@@ -0,0 +1,60 @@
+"""the refactor
+
+Revision ID: 5f6ac1e4fa9f
+Revises: a1a1a1a1a1a1
+Create Date: 2025-11-28 13:31:49.796461
+"""
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "5f6ac1e4fa9f"
+down_revision = "a1a1a1a1a1a1"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # Rename the table
+    op.rename_table("api_keys", "temporary_credit")
+
+    op.add_column(
+        "temporary_credit", sa.Column("created", sa.DateTime(), nullable=True)
+    )
+    op.add_column(
+        "temporary_credit",
+        sa.Column("refund_expiration_time", sa.Integer(), nullable=True),
+    )
+    op.drop_column("temporary_credit", "key_expiry_time")
+    op.drop_column("temporary_credit", "total_spent")
+    op.drop_column("temporary_credit", "total_requests")
+
+
+def downgrade() -> None:
+    op.add_column(
+        "temporary_credit",
+        sa.Column(
+            "total_requests",
+            sa.Integer(),
+            nullable=False,
+            server_default=sa.text("0"),
+        ),
+    )
+    op.add_column(
+        "temporary_credit",
+        sa.Column(
+            "total_spent",
+            sa.Integer(),
+            nullable=False,
+            server_default=sa.text("0"),
+        ),
+    )
+    op.add_column(
+        "temporary_credit", sa.Column("key_expiry_time", sa.Integer(), nullable=True)
+    )
+    op.drop_column("temporary_credit", "refund_expiration_time")
+    op.drop_column("temporary_credit", "created")
+
+    # Revert table rename
+    op.rename_table("temporary_credit", "api_keys")

pyproject.toml

@@ -20,7 +20,6 @@ dependencies = [
     "nostr>=0.0.2",
     "mdurl==0.1.2",
     "pillow>=10",
-    "openai>=1.98.0",
 ]
 
 [dependency-groups]

routstr/auth.py

@@ -1,661 +1,98 @@
 import hashlib
-import math
-from typing import Optional
+from typing import Annotated
 
-from fastapi import HTTPException
+from fastapi import Depends, Header, HTTPException
 from sqlalchemy.exc import IntegrityError
-from sqlmodel import col, update
 
 from .core import get_logger
-from .core.db import ApiKey, AsyncSession
+from .core.db import AsyncSession, TemporaryCredit, get_session
 from .core.settings import settings
-from .payment.cost_calculation import (
-    CostData,
-    CostDataError,
-    MaxCostData,
-    calculate_cost,
-)
-from .wallet import credit_balance, deserialize_token_from_string
+from .payment.wallet import credit_balance, deserialize_token_from_string
 
 logger = get_logger(__name__)
 
-# TODO: implement prepaid api key (not like it was before)
-# PREPAID_API_KEY = os.environ.get("PREPAID_API_KEY", None)
-# PREPAID_BALANCE = int(os.environ.get("PREPAID_BALANCE", "0")) * 1000  # Convert to msats
+
+async def api_key_to_credit(key: str, db_session: AsyncSession) -> TemporaryCredit:
+    key = key[3:] if key.startswith("sk-") else key
+    if existing_credit := await db_session.get(TemporaryCredit, key):
+        return existing_credit
+
+    raise HTTPException(status_code=401, detail="Invalid API-KEY")
 
 
-async def validate_bearer_key(
-    bearer_key: str,
-    session: AsyncSession,
-    refund_address: Optional[str] = None,
-    key_expiry_time: Optional[int] = None,
-) -> ApiKey:
-    """
-    Validates the provided API key using SQLModel.
-    If it's a cashu key, it redeems it and stores its hash and balance.
-    Otherwise checks if the hash of the key exists.
-    """
-    logger.debug(
-        "Starting bearer key validation",
-        extra={
-            "key_preview": bearer_key[:20] + "..."
-            if len(bearer_key) > 20
-            else bearer_key,
-            "has_refund_address": bool(refund_address),
-            "has_expiry_time": bool(key_expiry_time),
-        },
+async def cashu_token_to_credit(
+    cashu_token: str, db_session: AsyncSession
+) -> TemporaryCredit:
+    try:
+        token_hash = hashlib.sha256(cashu_token.encode()).hexdigest()
+        token_obj = deserialize_token_from_string(cashu_token)
+    except Exception:
+        raise HTTPException(status_code=400, detail="Invalid token format")
+
+    if existing_credit := await db_session.get(TemporaryCredit, token_hash):
+        return existing_credit
+
+    # create new TemporaryCredit
+    if token_obj.mint in settings.cashu_mints:
+        refund_currency = token_obj.unit
+        refund_mint_url = token_obj.mint
+    else:
+        refund_mint_url = settings.primary_mint
+        refund_currency = "sat"
+
+    new_credit = TemporaryCredit(
+        hashed_key=token_hash,
+        refund_currency=refund_currency,
+        refund_mint_url=refund_mint_url,
     )
+    db_session.add(new_credit)
 
-    if not bearer_key:
-        logger.error("Empty bearer key provided")
-        raise HTTPException(
-            status_code=401,
-            detail={
-                "error": {
-                    "message": "API key or Cashu token required",
-                    "type": "invalid_request_error",
-                    "code": "missing_api_key",
-                }
-            },
+    try:
+        await db_session.flush()
+    except IntegrityError:  # fallback to api key in case of race condition
+        await db_session.rollback()
+        return await api_key_to_credit(f"sk-{token_hash}", db_session)
+
+    try:
+        msats = await credit_balance(cashu_token, new_credit, db_session)
+    except Exception as e:
+        await db_session.rollback()
+        logger.error(
+            "Token redemption failed",
+            extra={"error": str(e)},
         )
+        raise HTTPException(status_code=400, detail="Token redemption failed")
 
-    if bearer_key.startswith("sk-"):
-        logger.debug(
-            "Processing sk- prefixed API key",
-            extra={"key_preview": bearer_key[:10] + "..."},
-        )
-
-        if existing_key := await session.get(ApiKey, bearer_key[3:]):
-            logger.info(
-                "Existing sk- API key found",
-                extra={
-                    "key_hash": existing_key.hashed_key[:8] + "...",
-                    "balance": existing_key.balance,
-                    "total_requests": existing_key.total_requests,
-                },
-            )
-
-            if key_expiry_time is not None:
-                existing_key.key_expiry_time = key_expiry_time
-                logger.debug(
-                    "Updated key expiry time",
-                    extra={
-                        "key_hash": existing_key.hashed_key[:8] + "...",
-                        "expiry_time": key_expiry_time,
-                    },
-                )
-
-            if refund_address is not None:
-                existing_key.refund_address = refund_address
-                logger.debug(
-                    "Updated refund address",
-                    extra={
-                        "key_hash": existing_key.hashed_key[:8] + "...",
-                        "refund_address_preview": refund_address[:20] + "..."
-                        if len(refund_address) > 20
-                        else refund_address,
-                    },
-                )
-
-            return existing_key
-        else:
-            logger.warning(
-                "sk- API key not found in database",
-                extra={"key_preview": bearer_key[:10] + "..."},
-            )
-
-    if bearer_key.startswith("cashu"):
-        logger.debug(
-            "Processing Cashu token",
-            extra={
-                "token_preview": bearer_key[:20] + "...",
-                "token_type": bearer_key[:6] if len(bearer_key) >= 6 else bearer_key,
-            },
-        )
-
-        try:
-            hashed_key = hashlib.sha256(bearer_key.encode()).hexdigest()
-            token_obj = deserialize_token_from_string(bearer_key)
-            logger.debug(
-                "Generated token hash", extra={"hash_preview": hashed_key[:16] + "..."}
-            )
-
-            if existing_key := await session.get(ApiKey, hashed_key):
-                logger.info(
-                    "Existing Cashu token found",
-                    extra={
-                        "key_hash": existing_key.hashed_key[:8] + "...",
-                        "balance": existing_key.balance,
-                        "total_requests": existing_key.total_requests,
-                    },
-                )
-
-                if key_expiry_time is not None:
-                    existing_key.key_expiry_time = key_expiry_time
-                    logger.debug(
-                        "Updated key expiry time for existing Cashu key",
-                        extra={
-                            "key_hash": existing_key.hashed_key[:8] + "...",
-                            "expiry_time": key_expiry_time,
-                        },
-                    )
-
-                if refund_address is not None:
-                    existing_key.refund_address = refund_address
-                    logger.debug(
-                        "Updated refund address for existing Cashu key",
-                        extra={
-                            "key_hash": existing_key.hashed_key[:8] + "...",
-                            "refund_address_preview": refund_address[:20] + "..."
-                            if len(refund_address) > 20
-                            else refund_address,
-                        },
-                    )
-
-                return existing_key
-
-            logger.info(
-                "Creating new Cashu token entry",
-                extra={
-                    "hash_preview": hashed_key[:16] + "...",
-                    "has_refund_address": bool(refund_address),
-                    "has_expiry_time": bool(key_expiry_time),
-                },
-            )
-            if token_obj.mint in settings.cashu_mints:
-                refund_currency = token_obj.unit
-                refund_mint_url = token_obj.mint
-            else:
-                refund_currency = "sat"
-                refund_mint_url = settings.primary_mint
-
-            new_key = ApiKey(
-                hashed_key=hashed_key,
-                balance=0,
-                refund_address=refund_address,
-                key_expiry_time=key_expiry_time,
-                refund_currency=refund_currency,
-                refund_mint_url=refund_mint_url,
-            )
-            session.add(new_key)
-
-            try:
-                await session.flush()
-            except IntegrityError:
-                await session.rollback()
-                logger.info(
-                    "Concurrent key creation detected, fetching existing key",
-                    extra={"key_hash": hashed_key[:8] + "..."},
-                )
-                existing_key = await session.get(ApiKey, hashed_key)
-                if not existing_key:
-                    raise Exception("Failed to fetch existing key after IntegrityError")
-
-                if key_expiry_time is not None:
-                    existing_key.key_expiry_time = key_expiry_time
-                if refund_address is not None:
-                    existing_key.refund_address = refund_address
-
-                return existing_key
-
-            logger.debug(
-                "New key created, starting token redemption",
-                extra={"key_hash": hashed_key[:8] + "..."},
-            )
-
-            logger.info(
-                "AUTH: About to call credit_balance",
-                extra={"token_preview": bearer_key[:50]},
-            )
-            try:
-                msats = await credit_balance(bearer_key, new_key, session)
-                logger.info(
-                    "AUTH: credit_balance returned successfully", extra={"msats": msats}
-                )
-            except Exception as credit_error:
-                logger.error(
-                    "AUTH: credit_balance failed",
-                    extra={
-                        "error": str(credit_error),
-                        "error_type": type(credit_error).__name__,
-                    },
-                )
-                raise credit_error
-
-            if msats <= 0:
-                logger.error(
-                    "Token redemption returned zero or negative amount",
-                    extra={"msats": msats, "key_hash": hashed_key[:8] + "..."},
-                )
-                raise Exception("Token redemption failed")
-
-            await session.refresh(new_key)
-            await session.commit()
-
-            logger.info(
-                "New Cashu token successfully redeemed and stored",
-                extra={
-                    "key_hash": hashed_key[:8] + "...",
-                    "redeemed_msats": msats,
-                    "final_balance": new_key.balance,
-                },
-            )
-
-            return new_key
-        except Exception as e:
-            logger.error(
-                "Cashu token redemption failed",
-                extra={
-                    "error": str(e),
-                    "error_type": type(e).__name__,
-                    "token_preview": bearer_key[:20] + "..."
-                    if len(bearer_key) > 20
-                    else bearer_key,
-                },
-            )
-            raise HTTPException(
-                status_code=401,
-                detail={
-                    "error": {
-                        "message": f"Invalid or expired Cashu key: {str(e)}",
-                        "type": "invalid_request_error",
-                        "code": "invalid_api_key",
-                    }
-                },
-            )
-
-    logger.error(
-        "Invalid API key format",
-        extra={
-            "key_preview": bearer_key[:10] + "..."
-            if len(bearer_key) > 10
-            else bearer_key,
-            "key_length": len(bearer_key),
-        },
-    )
-
-    raise HTTPException(
-        status_code=401,
-        detail={
-            "error": {
-                "message": "Invalid API key",
-                "type": "invalid_request_error",
-                "code": "invalid_api_key",
-            }
-        },
-    )
-
-
-async def pay_for_request(
-    key: ApiKey, cost_per_request: int, session: AsyncSession
-) -> int:
-    """Process payment for a request."""
+    await db_session.refresh(new_credit)
+    await db_session.commit()
 
     logger.info(
-        "Processing payment for request",
-        extra={
-            "key_hash": key.hashed_key[:8] + "...",
-            "current_balance": key.balance,
-            "required_cost": cost_per_request,
-            "sufficient_balance": key.balance >= cost_per_request,
-        },
+        "New TemporaryCredit created from CashuToken",
+        extra={"key_hash": token_hash[:8] + "...", "amount_msats": msats},
     )
 
-    if key.total_balance < cost_per_request:
-        logger.warning(
-            "Insufficient balance for request",
-            extra={
-                "key_hash": key.hashed_key[:8] + "...",
-                "balance": key.balance,
-                "reserved_balance": key.reserved_balance,
-                "required": cost_per_request,
-                "shortfall": cost_per_request - key.total_balance,
-            },
-        )
+    return new_credit
 
-        raise HTTPException(
-            status_code=402,
-            detail={
-                "error": {
-                    "message": f"Insufficient balance: {cost_per_request} mSats required. {key.total_balance} available. (reserved: {key.reserved_balance})",
-                    "type": "insufficient_quota",
-                    "code": "insufficient_balance",
-                }
-            },
-        )
 
-    logger.debug(
-        "Charging base cost for request",
-        extra={
-            "key_hash": key.hashed_key[:8] + "...",
-            "cost": cost_per_request,
-            "balance_before": key.balance,
-        },
+async def nwc_to_credit(
+    connection_string: str, db_session: AsyncSession
+) -> TemporaryCredit:
+    raise NotImplementedError
+
+
+async def get_credit(
+    authorization: Annotated[str, Header(...)],
+    db_session: AsyncSession = Depends(get_session),
+) -> TemporaryCredit:
+    authorization = (
+        authorization[7:] if authorization.startswith("Bearer ") else authorization
     )
 
-    # Charge the base cost for the request atomically to avoid race conditions
-    stmt = (
-        update(ApiKey)
-        .where(col(ApiKey.hashed_key) == key.hashed_key)
-        .where(col(ApiKey.balance) - col(ApiKey.reserved_balance) >= cost_per_request)
-        .values(
-            reserved_balance=col(ApiKey.reserved_balance) + cost_per_request,
-            total_requests=col(ApiKey.total_requests) + 1,
-        )
-    )
-    result = await session.exec(stmt)  # type: ignore[call-overload]
-    await session.commit()
-
-    if result.rowcount == 0:
-        logger.error(
-            "Concurrent request depleted balance",
-            extra={
-                "key_hash": key.hashed_key[:8] + "...",
-                "required_cost": cost_per_request,
-                "current_balance": key.balance,
-            },
-        )
-
-        # Another concurrent request spent the balance first
-        raise HTTPException(
-            status_code=402,
-            detail={
-                "error": {
-                    "message": f"Insufficient balance: {cost_per_request} mSats required. {key.balance} available.",
-                    "type": "insufficient_quota",
-                    "code": "insufficient_balance",
-                }
-            },
-        )
-
-    await session.refresh(key)
-
-    logger.info(
-        "Payment processed successfully",
-        extra={
-            "key_hash": key.hashed_key[:8] + "...",
-            "charged_amount": cost_per_request,
-            "new_balance": key.balance,
-            "total_spent": key.total_spent,
-            "total_requests": key.total_requests,
-        },
-    )
-
-    return cost_per_request
-
-
-async def revert_pay_for_request(
-    key: ApiKey, session: AsyncSession, cost_per_request: int
-) -> None:
-    stmt = (
-        update(ApiKey)
-        .where(col(ApiKey.hashed_key) == key.hashed_key)
-        .values(
-            reserved_balance=col(ApiKey.reserved_balance) - cost_per_request,
-            total_requests=col(ApiKey.total_requests) - 1,
-        )
-    )
-
-    result = await session.exec(stmt)  # type: ignore[call-overload]
-    await session.commit()
-    if result.rowcount == 0:
-        logger.error(
-            "Failed to revert payment - insufficient reserved balance",
-            extra={
-                "key_hash": key.hashed_key[:8] + "...",
-                "cost_to_revert": cost_per_request,
-                "current_reserved_balance": key.reserved_balance,
-            },
-        )
-        raise HTTPException(
-            status_code=402,
-            detail={
-                "error": {
-                    "message": f"failed to revert request payment: {cost_per_request} mSats required. {key.balance} available.",
-                    "type": "payment_error",
-                    "code": "payment_error",
-                }
-            },
-        )
-    await session.refresh(key)
-
-
-async def adjust_payment_for_tokens(
-    key: ApiKey, response_data: dict, session: AsyncSession, deducted_max_cost: int
-) -> dict:
-    """
-    Adjusts the payment based on token usage in the response.
-    This is called after the initial payment and the upstream request is complete.
-    Returns cost data to be included in the response.
-    """
-    model = response_data.get("model", "unknown")
-
-    logger.debug(
-        "Starting payment adjustment for tokens",
-        extra={
-            "key_hash": key.hashed_key[:8] + "...",
-            "model": model,
-            "deducted_max_cost": deducted_max_cost,
-            "current_balance": key.balance,
-            "has_usage": "usage" in response_data,
-        },
-    )
-
-    match await calculate_cost(response_data, deducted_max_cost, session):
-        case MaxCostData() as cost:
-            logger.debug(
-                "Using max cost data (no token adjustment)",
-                extra={
-                    "key_hash": key.hashed_key[:8] + "...",
-                    "model": model,
-                    "max_cost": cost.total_msats,
-                },
-            )
-            # Finalize by releasing reservation and charging max cost
-            finalize_stmt = (
-                update(ApiKey)
-                .where(col(ApiKey.hashed_key) == key.hashed_key)
-                .values(
-                    reserved_balance=col(ApiKey.reserved_balance) - deducted_max_cost,
-                    balance=col(ApiKey.balance) - cost.total_msats,
-                    total_spent=col(ApiKey.total_spent) + cost.total_msats,
-                )
-            )
-            result = await session.exec(finalize_stmt)  # type: ignore[call-overload]
-            await session.commit()
-            if result.rowcount == 0:
-                logger.error(
-                    "Failed to finalize max-cost payment - insufficient reserved balance",
-                    extra={
-                        "key_hash": key.hashed_key[:8] + "...",
-                        "deducted_max_cost": deducted_max_cost,
-                        "current_reserved_balance": key.reserved_balance,
-                        "total_cost": cost.total_msats,
-                        "model": model,
-                    },
-                )
-            else:
-                await session.refresh(key)
-                logger.info(
-                    "Max cost payment finalized",
-                    extra={
-                        "key_hash": key.hashed_key[:8] + "...",
-                        "charged_amount": cost.total_msats,
-                        "new_balance": key.balance,
-                        "model": model,
-                    },
-                )
-            return cost.dict()
-
-        case CostData() as cost:
-            # If token-based pricing is enabled and base cost is 0, use token-based cost
-            # Otherwise, token cost is additional to the base cost
-            cost_difference = cost.total_msats - deducted_max_cost
-            total_cost_msats: int = math.ceil(cost.total_msats)
-
-            logger.info(
-                "Calculated token-based cost",
-                extra={
-                    "key_hash": key.hashed_key[:8] + "...",
-                    "model": model,
-                    "token_cost": cost.total_msats,
-                    "deducted_max_cost": deducted_max_cost,
-                    "cost_difference": cost_difference,
-                    "input_msats": cost.input_msats,
-                    "output_msats": cost.output_msats,
-                },
-            )
-
-            if cost_difference == 0:
-                logger.debug(
-                    "Finalizing with exact reserved cost",
-                    extra={"key_hash": key.hashed_key[:8] + "...", "model": model},
-                )
-                finalize_stmt = (
-                    update(ApiKey)
-                    .where(col(ApiKey.hashed_key) == key.hashed_key)
-                    .values(
-                        reserved_balance=col(ApiKey.reserved_balance)
-                        - deducted_max_cost,
-                        balance=col(ApiKey.balance) - total_cost_msats,
-                        total_spent=col(ApiKey.total_spent) + total_cost_msats,
-                    )
-                )
-                await session.exec(finalize_stmt)  # type: ignore[call-overload]
-                await session.commit()
-                await session.refresh(key)
-                return cost.dict()
-
-            # this should never happen why do we handle this???
-            if cost_difference > 0:
-                # Need to charge more than reserved, finalize by releasing reservation and charging total
-                logger.info(
-                    "Additional charge required for token usage",
-                    extra={
-                        "key_hash": key.hashed_key[:8] + "...",
-                        "additional_charge": cost_difference,
-                        "current_balance": key.balance,
-                        "sufficient_balance": key.balance >= cost_difference,
-                        "model": model,
-                    },
-                )
-
-                finalize_stmt = (
-                    update(ApiKey)
-                    .where(col(ApiKey.hashed_key) == key.hashed_key)
-                    .values(
-                        reserved_balance=col(ApiKey.reserved_balance)
-                        - deducted_max_cost,
-                        balance=col(ApiKey.balance) - total_cost_msats,
-                        total_spent=col(ApiKey.total_spent) + total_cost_msats,
-                    )
-                )
-                result = await session.exec(finalize_stmt)  # type: ignore[call-overload]
-                await session.commit()
-
-                if result.rowcount:
-                    cost.total_msats = total_cost_msats
-                    await session.refresh(key)
-
-                    logger.info(
-                        "Finalized payment with additional charge",
-                        extra={
-                            "key_hash": key.hashed_key[:8] + "...",
-                            "charged_amount": total_cost_msats,
-                            "new_balance": key.balance,
-                            "model": model,
-                        },
-                    )
-                else:
-                    logger.warning(
-                        "Failed to finalize additional charge (concurrent operation)",
-                        extra={
-                            "key_hash": key.hashed_key[:8] + "...",
-                            "attempted_charge": total_cost_msats,
-                            "model": model,
-                        },
-                    )
-            else:
-                # Refund some of the base cost
-                refund = abs(cost_difference)
-                logger.info(
-                    "Refunding excess payment",
-                    extra={
-                        "key_hash": key.hashed_key[:8] + "...",
-                        "refund_amount": refund,
-                        "current_balance": key.balance,
-                        "model": model,
-                    },
-                )
-
-                refund_stmt = (
-                    update(ApiKey)
-                    .where(col(ApiKey.hashed_key) == key.hashed_key)
-                    .values(
-                        reserved_balance=col(ApiKey.reserved_balance)
-                        - deducted_max_cost,
-                        balance=col(ApiKey.balance) - total_cost_msats,
-                        total_spent=col(ApiKey.total_spent) + total_cost_msats,
-                    )
-                )
-                result = await session.exec(refund_stmt)  # type: ignore[call-overload]
-                await session.commit()
-
-                if result.rowcount == 0:
-                    logger.error(
-                        "Failed to finalize payment - insufficient reserved balance",
-                        extra={
-                            "key_hash": key.hashed_key[:8] + "...",
-                            "deducted_max_cost": deducted_max_cost,
-                            "current_reserved_balance": key.reserved_balance,
-                            "total_cost": total_cost_msats,
-                            "model": model,
-                        },
-                    )
-                    # Still return the cost data even if we couldn't properly finalize
-                    # The reservation was already made, so the user has paid
-
-                cost.total_msats = total_cost_msats
-                await session.refresh(key)
-
-                logger.info(
-                    "Refund processed successfully",
-                    extra={
-                        "key_hash": key.hashed_key[:8] + "...",
-                        "refunded_amount": refund,
-                        "new_balance": key.balance,
-                        "final_cost": cost.total_msats,
-                        "model": model,
-                    },
-                )
-
-            return cost.dict()
-
-        case CostDataError() as error:
-            logger.error(
-                "Cost calculation error during payment adjustment",
-                extra={
-                    "key_hash": key.hashed_key[:8] + "...",
-                    "model": model,
-                    "error_message": error.message,
-                    "error_code": error.code,
-                },
-            )
-
-            raise HTTPException(
-                status_code=400,
-                detail={
-                    "error": {
-                        "message": error.message,
-                        "type": "invalid_request_error",
-                        "code": error.code,
-                    }
-                },
-            )
-    # Fallback return to satisfy type checker; execution should not reach here
-    return {
-        "base_msats": deducted_max_cost,
-        "input_msats": 0,
-        "output_msats": 0,
-        "total_msats": deducted_max_cost,
-    }
+    if authorization.startswith("cashu"):
+        return await cashu_token_to_credit(authorization, db_session)
+    elif authorization.startswith("sk-"):
+        return await api_key_to_credit(authorization, db_session)
+    elif authorization.startswith("nwc"):
+        return await nwc_to_credit(authorization, db_session)
+    else:
+        raise HTTPException(status_code=401, detail="Unable to parse bearer token")

routstr/balance.py

@@ -6,11 +6,12 @@ from typing import Annotated, NoReturn
 from fastapi import APIRouter, Depends, Header, HTTPException
 from pydantic import BaseModel
 
-from .auth import validate_bearer_key
-from .core.db import ApiKey, AsyncSession, get_session
+from .auth import get_credit
+from .core.db import AsyncSession, TemporaryCredit, get_session
 from .core.logging import get_logger
 from .core.settings import settings
-from .wallet import credit_balance, recieve_token, send_to_lnurl, send_token
+from .payment.lnurl import send_to_lnurl
+from .payment.wallet import credit_balance, recieve_token, send_token
 
 router = APIRouter()
 balance_router = APIRouter(prefix="/v1/balance")
@@ -18,26 +19,15 @@ balance_router = APIRouter(prefix="/v1/balance")
 logger = get_logger(__name__)
 
 
-async def get_key_from_header(
-    authorization: Annotated[str, Header(...)],
-    session: AsyncSession = Depends(get_session),
-) -> ApiKey:
-    if authorization.startswith("Bearer "):
-        return await validate_bearer_key(authorization[7:], session)
-
-    raise HTTPException(
-        status_code=401,
-        detail="Invalid authorization. Use 'Bearer <cashu-token>' or 'Bearer <api-key>'",
-    )
-
-
 # TODO: remove this endpoint when frontend is updated
 @router.get("/", include_in_schema=False)
-async def account_info(key: ApiKey = Depends(get_key_from_header)) -> dict:
+async def account_info(
+    credit: TemporaryCredit = Depends(get_credit),
+) -> dict:
     return {
-        "api_key": "sk-" + key.hashed_key,
-        "balance": key.balance,
-        "reserved": key.reserved_balance,
+        "api_key": credit.api_key,
+        "balance": credit.balance,
+        "reserved": credit.reserved_balance,
     }
 
 
@@ -57,19 +47,18 @@ async def account_info(key: ApiKey = Depends(get_key_from_header)) -> dict:
 async def create_balance(
     initial_balance_token: str, session: AsyncSession = Depends(get_session)
 ) -> dict:
-    key = await validate_bearer_key(initial_balance_token, session)
-    return {
-        "api_key": "sk-" + key.hashed_key,
-        "balance": key.balance,
-    }
+    credit = await get_credit(initial_balance_token, session)
+    return {"api_key": credit.api_key, "balance": credit.balance}
 
 
 @router.get("/info")
-async def wallet_info(key: ApiKey = Depends(get_key_from_header)) -> dict:
+async def wallet_info(
+    credit: TemporaryCredit = Depends(get_credit),
+) -> dict:
     return {
-        "api_key": "sk-" + key.hashed_key,
-        "balance": key.balance,
-        "reserved": key.reserved_balance,
+        "api_key": credit.api_key,
+        "balance": credit.balance,
+        "reserved": credit.reserved_balance,
     }
 
 
@@ -81,7 +70,7 @@ class TopupRequest(BaseModel):
 async def topup_wallet_endpoint(
     cashu_token: str | None = None,
     topup_request: TopupRequest | None = None,
-    key: ApiKey = Depends(get_key_from_header),
+    credit: TemporaryCredit = Depends(get_credit),
     session: AsyncSession = Depends(get_session),
 ) -> dict[str, int]:
     if topup_request is not None:
@@ -93,7 +82,7 @@ async def topup_wallet_endpoint(
     if len(cashu_token) < 10 or "cashu" not in cashu_token:
         raise HTTPException(status_code=400, detail="Invalid token format")
     try:
-        amount_msats = await credit_balance(cashu_token, key, session)
+        amount_msats = await credit_balance(cashu_token, credit, session)
     except ValueError as e:
         error_msg = str(e)
         if "already spent" in error_msg.lower():
@@ -152,10 +141,11 @@ async def refund_wallet_endpoint(
     if cached := await _refund_cache_get(bearer_value):
         return cached
 
-    key: ApiKey = await validate_bearer_key(bearer_value, session)
+    key: TemporaryCredit = await get_credit(bearer_value, session)
     remaining_balance_msats: int = key.balance
+    refund_currency = key.refund_currency or "sat"
 
-    if key.refund_currency == "sat":
+    if refund_currency == "sat":
         remaining_balance = remaining_balance_msats // 1000
     else:
         remaining_balance = remaining_balance_msats
@@ -172,19 +162,18 @@ async def refund_wallet_endpoint(
 
             await send_to_lnurl(
                 remaining_balance,
-                key.refund_currency or "sat",
+                refund_currency,
                 key.refund_mint_url or global_settings.primary_mint,
                 key.refund_address,
             )
             result = {"recipient": key.refund_address}
         else:
-            refund_currency = key.refund_currency or "sat"
             token = await send_token(
                 remaining_balance, refund_currency, key.refund_mint_url
             )
             result = {"token": token}
 
-        if key.refund_currency == "sat":
+        if refund_currency == "sat":
             result["sats"] = str(remaining_balance_msats // 1000)
         else:
             result["msats"] = str(remaining_balance_msats)

routstr/core/admin.py

@@ -8,16 +8,16 @@ from fastapi.responses import HTMLResponse, RedirectResponse
 from pydantic import BaseModel
 from sqlmodel import select
 
-from ..payment.models import _row_to_model, list_models
-from ..proxy import refresh_model_maps, reinitialize_upstreams
-from ..wallet import (
+from ..models.crud import row_to_model, list_models
+from ..payment.wallet import (
     fetch_all_balances,
     get_proofs_per_mint_and_unit,
     get_wallet,
     send_token,
     slow_filter_spend_proofs,
 )
-from .db import ApiKey, ModelRow, UpstreamProviderRow, create_session
+from ..proxy import refresh_model_maps, reinitialize_upstreams
+from .db import ModelRow, TemporaryCredit, UpstreamProviderRow, create_session
 from .log_manager import log_manager
 from .logging import get_logger
 from .settings import SettingsService, settings
@@ -107,13 +107,13 @@ async def partial_balances(request: Request) -> str:
 
 
 @admin_router.get(
-    "/partials/apikeys",
+    "/partials/TemporaryCredits",
     dependencies=[Depends(require_admin_api)],
     response_class=HTMLResponse,
 )
-async def partial_apikeys(request: Request) -> str:
+async def partial_TemporaryCredits(request: Request) -> str:
     async with create_session() as session:
-        result = await session.exec(select(ApiKey))
+        result = await session.exec(select(TemporaryCredit))
         api_keys = result.all()
 
     def fmt_time(ts: int | None) -> str:
@@ -124,7 +124,7 @@ async def partial_apikeys(request: Request) -> str:
 
     rows = "".join(
         [
-            f"<tr><td>{key.hashed_key}</td><td>{key.balance}</td><td>{key.total_spent}</td><td>{key.total_requests}</td><td>{key.refund_address}</td><td>{fmt_time(key.key_expiry_time)}</td></tr>"
+            f"<tr><td>{key.hashed_key}</td><td>{key.balance}</td><td>{key.refund_address}</td><td>{key.refund_mint_url}</td><td>{key.refund_currency}</td><td>{fmt_time(key.refund_expiration_time)}</td></tr>"
             for key in api_keys
         ]
     )
@@ -134,9 +134,9 @@ async def partial_apikeys(request: Request) -> str:
             <tr>
                 <th>Hashed Key</th>
                 <th>Balance (mSats)</th>
-                <th>Total Spent (mSats)</th>
-                <th>Total Requests</th>
                 <th>Refund Address</th>
+                <th>Refund Mint</th>
+                <th>Currency</th>
                 <th>Refund Time</th>
             </tr>
             {rows}
@@ -147,17 +147,17 @@ async def partial_apikeys(request: Request) -> str:
 @admin_router.get("/api/temporary-balances", dependencies=[Depends(require_admin_api)])
 async def get_temporary_balances_api(request: Request) -> list[dict[str, object]]:
     async with create_session() as session:
-        result = await session.exec(select(ApiKey))
+        result = await session.exec(select(TemporaryCredit))
         api_keys = result.all()
 
     return [
         {
             "hashed_key": key.hashed_key,
             "balance": key.balance,
-            "total_spent": key.total_spent,
-            "total_requests": key.total_requests,
             "refund_address": key.refund_address,
-            "key_expiry_time": key.key_expiry_time,
+            "refund_mint_url": key.refund_mint_url,
+            "refund_currency": key.refund_currency,
+            "refund_expiration_time": key.refund_expiration_time,
         }
         for key in api_keys
     ]
@@ -777,8 +777,8 @@ async def dashboard(request: Request) -> str:
                 <p><em>Save this token! It represents your withdrawn balance.</em></p>
             </div>
 
-            <div id="apikeys-table"
-                 hx-get="/admin/partials/apikeys"
+            <div id="TemporaryCredits-table"
+                 hx-get="/admin/partials/TemporaryCredits"
                  hx-trigger="load"
                  hx-swap="outerHTML">
                 <h2>Temporary Balances</h2>
@@ -1763,9 +1763,9 @@ UPSTREAM_PROVIDERS_JS: str = """<!--html-->
             provider_fee: feeValue ? parseFloat(feeValue) : defaultFee,
         };
 
-        const apiKey = document.getElementById('provider-api-key').value;
-        if (apiKey) {
-            payload.api_key = apiKey;
+        const TemporaryCredit = document.getElementById('provider-api-key').value;
+        if (TemporaryCredit) {
+            payload.api_key = TemporaryCredit;
         }
 
         const saveBtn = document.getElementById('provider-save-btn');
@@ -1783,7 +1783,7 @@ UPSTREAM_PROVIDERS_JS: str = """<!--html-->
                     body: JSON.stringify(payload)
                 });
             } else {
-                if (!apiKey) {
+                if (!TemporaryCredit) {
                     throw new Error('API Key is required for new providers');
                 }
                 resp = await fetch('/admin/api/upstream-providers', {
@@ -2456,7 +2456,7 @@ async def create_provider_model(
         await session.refresh(row)
 
     await refresh_model_maps()
-    return _row_to_model(
+    return row_to_model(
         row, apply_provider_fee=True, provider_fee=provider.provider_fee
     ).dict()  # type: ignore
 
@@ -2476,7 +2476,7 @@ async def get_provider_model(provider_id: int, model_id: str) -> dict[str, objec
             raise HTTPException(
                 status_code=404, detail="Model not found for this provider"
             )
-        return _row_to_model(
+        return row_to_model(
             row, apply_provider_fee=True, provider_fee=provider.provider_fee
         ).dict()  # type: ignore
 
@@ -2525,7 +2525,7 @@ async def update_provider_model(
         await session.refresh(row)
 
     if was_disabled and payload.enabled:
-        from ..payment.models import _cleanup_enabled_models_once
+        from ..models.models import _cleanup_enabled_models_once
 
         try:
             await _cleanup_enabled_models_once()
@@ -2536,7 +2536,7 @@ async def update_provider_model(
             )
 
     await refresh_model_maps()
-    return _row_to_model(
+    return row_to_model(
         row, apply_provider_fee=True, provider_fee=provider.provider_fee
     ).dict()  # type: ignore
 
@@ -2791,192 +2791,12 @@ async def get_provider_models(provider_id: int) -> dict[str, object]:
         }
 
 
-class CreateAccountRequest(BaseModel):
-    provider_type: str
-
-
-@admin_router.post(
-    "/api/upstream-providers/create-account",
-    dependencies=[Depends(require_admin_api)],
-)
-async def create_provider_account_by_type(
-    payload: CreateAccountRequest,
-) -> dict[str, object]:
-    """Create a new account with a provider by provider type (before provider exists in DB)."""
-    from ..upstream import upstream_provider_classes
-
-    provider_class = next(
-        (
-            cls
-            for cls in upstream_provider_classes
-            if cls.provider_type == payload.provider_type
-        ),
-        None,
-    )
-    if not provider_class:
-        raise HTTPException(status_code=404, detail="Provider type not found")
-
-    try:
-        account_data = await provider_class.create_account_static()
-
-        return {
-            "ok": True,
-            "account_data": account_data,
-            "message": "Account created successfully",
-        }
-    except NotImplementedError as e:
-        raise HTTPException(
-            status_code=400,
-            detail=f"Provider does not support account creation: {str(e)}",
-        )
-    except Exception as e:
-        logger.error(
-            f"Failed to create account for provider type {payload.provider_type}: {e}"
-        )
-        raise HTTPException(status_code=500, detail=str(e))
-
-
-class TopupRequest(BaseModel):
-    amount: int
-
-
-@admin_router.post(
-    "/api/upstream-providers/{provider_id}/topup",
-    dependencies=[Depends(require_admin_api)],
-)
-async def initiate_provider_topup(
-    provider_id: int, payload: TopupRequest
-) -> dict[str, object]:
-    """Initiate a Lightning Network top-up for the upstream provider account."""
-    from ..upstream.helpers import _instantiate_provider
-
-    async with create_session() as session:
-        provider = await session.get(UpstreamProviderRow, provider_id)
-        if not provider:
-            raise HTTPException(status_code=404, detail="Provider not found")
-
-        upstream_instance = _instantiate_provider(provider)
-        if not upstream_instance:
-            raise HTTPException(
-                status_code=400, detail="Could not instantiate provider"
-            )
-
-        try:
-            logger.info(
-                f"Initiating top-up for provider {provider_id}",
-                extra={"amount": payload.amount},
-            )
-            topup_data = await upstream_instance.initiate_topup(payload.amount)
-            logger.info(
-                "Top-up initiated successfully",
-                extra={
-                    "provider_id": provider_id,
-                    "invoice_id": topup_data.invoice_id,
-                    "amount": topup_data.amount,
-                },
-            )
-
-            response_data = {
-                "ok": True,
-                "topup_data": {
-                    "invoice_id": topup_data.invoice_id,
-                    "payment_request": topup_data.payment_request,
-                    "amount": topup_data.amount,
-                    "currency": topup_data.currency,
-                    "expires_at": topup_data.expires_at,
-                    "checkout_url": topup_data.checkout_url,
-                },
-                "message": "Top-up initiated successfully",
-            }
-            logger.info("Returning response", extra={"response": response_data})
-            return response_data
-        except NotImplementedError as e:
-            logger.error(f"Provider does not support top-up: {e}")
-            raise HTTPException(
-                status_code=400, detail=f"Provider does not support top-up: {str(e)}"
-            )
-        except Exception as e:
-            logger.error(
-                f"Failed to initiate top-up for provider {provider_id}: {e}",
-                extra={"error_type": type(e).__name__, "error": str(e)},
-            )
-            raise HTTPException(status_code=500, detail=str(e))
-
-
-@admin_router.get(
-    "/api/upstream-providers/{provider_id}/topup/{invoice_id}/status",
-    dependencies=[Depends(require_admin_api)],
-)
-async def check_topup_status(provider_id: int, invoice_id: str) -> dict[str, object]:
-    """Check the status of a Lightning Network top-up invoice."""
-    from ..upstream.helpers import _instantiate_provider
-    from ..upstream.ppqai import PPQAIUpstreamProvider
-
-    async with create_session() as session:
-        provider = await session.get(UpstreamProviderRow, provider_id)
-        if not provider:
-            raise HTTPException(status_code=404, detail="Provider not found")
-
-        upstream_instance = _instantiate_provider(provider)
-        if not upstream_instance:
-            raise HTTPException(
-                status_code=400, detail="Could not instantiate provider"
-            )
-
-        if not isinstance(upstream_instance, PPQAIUpstreamProvider):
-            raise HTTPException(
-                status_code=400,
-                detail="Provider does not support top-up status checking",
-            )
-
-        try:
-            paid = await upstream_instance.check_topup_status(invoice_id)
-            return {"ok": True, "paid": paid}
-        except Exception as e:
-            logger.error(
-                f"Failed to check top-up status for provider {provider_id}: {e}"
-            )
-            raise HTTPException(status_code=500, detail=str(e))
-
-
-@admin_router.get(
-    "/api/upstream-providers/{provider_id}/balance",
-    dependencies=[Depends(require_admin_api)],
-)
-async def get_provider_balance(provider_id: int) -> dict[str, object]:
-    """Get the current account balance for the upstream provider."""
-    from ..upstream.helpers import _instantiate_provider
-
-    async with create_session() as session:
-        provider = await session.get(UpstreamProviderRow, provider_id)
-        if not provider:
-            raise HTTPException(status_code=404, detail="Provider not found")
-
-        upstream_instance = _instantiate_provider(provider)
-        if not upstream_instance:
-            raise HTTPException(
-                status_code=400, detail="Could not instantiate provider"
-            )
-
-        try:
-            balance_data = await upstream_instance.get_balance()
-            return {"ok": True, "balance_data": balance_data}
-        except NotImplementedError as e:
-            raise HTTPException(
-                status_code=400,
-                detail=f"Provider does not support balance checking: {str(e)}",
-            )
-        except Exception as e:
-            logger.error(f"Failed to fetch balance for provider {provider_id}: {e}")
-            raise HTTPException(status_code=500, detail=str(e))
-
-
 @admin_router.get(
     "/api/openrouter-presets",
     dependencies=[Depends(require_admin_api)],
 )
 async def get_openrouter_presets() -> list[dict[str, object]]:
-    from ..payment.models import async_fetch_openrouter_models
+    from ..models import async_fetch_openrouter_models
 
     models_data = await async_fetch_openrouter_models()
     return models_data

routstr/core/db.py

@@ -1,5 +1,6 @@
 import os
 from contextlib import asynccontextmanager
+from datetime import datetime
 from typing import AsyncGenerator
 
 from alembic import command
@@ -18,39 +19,40 @@ DATABASE_URL = os.environ.get("DATABASE_URL", "sqlite+aiosqlite:///keys.db")
 engine = create_async_engine(DATABASE_URL, echo=False)  # echo=True for debugging SQL
 
 
-class ApiKey(SQLModel, table=True):  # type: ignore
-    __tablename__ = "api_keys"
+class TemporaryCredit(SQLModel, table=True):  # type: ignore
+    __tablename__ = "temporary_credit"
 
     hashed_key: str = Field(primary_key=True)
-    balance: int = Field(default=0, description="Balance in millisatoshis (msats)")
-    reserved_balance: int = Field(
-        default=0, description="Reserved balance in millisatoshis (msats)"
-    )
+    balance: int = Field(default=0, description="Balance in msats")
+    reserved_balance: int = Field(default=0, description="Blocked balance in msats")
+    created: datetime | None = Field(None, description="Timestamp of creation")
     refund_address: str | None = Field(
-        default=None,
-        description="Lightning address to refund remaining balance after key expires",
+        None, description="Address to refund on expiration"
     )
-    key_expiry_time: int | None = Field(
-        default=None,
-        description="Unix-timestamp after which the cashu-token's balance gets refunded to the refund_address",
-    )
-    total_spent: int = Field(
-        default=0, description="Total spent in millisatoshis (msats)"
-    )
-    total_requests: int = Field(default=0)
     refund_mint_url: str | None = Field(
-        default=None,
-        description="URL of the mint used to create the cashu-token",
+        default=None, description="Mint used to issue the refund token"
     )
-    refund_currency: str | None = Field(
-        default=None,
-        description="Currency of the cashu-token",
+    refund_currency: str | None = Field(None, description="Currency of the cashu-token")
+    refund_expiration_time: int | None = Field(
+        None, description="Refund not allowed after timeout"
     )
 
     @property
-    def total_balance(self) -> int:
+    def total_balance_msat(self) -> int:
         return self.balance - self.reserved_balance
 
+    @property
+    def total_balance_sat(self) -> int:
+        return self.total_balance_msat // 1000
+
+    @property
+    def total_balance(self) -> int:
+        return self.total_balance_msat
+
+    @property
+    def api_key(self) -> str:
+        return "sk-" + self.hashed_key
+
 
 class ModelRow(SQLModel, table=True):  # type: ignore
     __tablename__ = "models"
@@ -95,8 +97,9 @@ class UpstreamProviderRow(SQLModel, table=True):  # type: ignore
 async def balances_for_mint_and_unit(
     db_session: AsyncSession, mint_url: str, unit: str
 ) -> int:
-    query = select(func.sum(ApiKey.balance)).where(
-        ApiKey.refund_mint_url == mint_url, ApiKey.refund_currency == unit
+    query = select(func.sum(TemporaryCredit.balance)).where(
+        TemporaryCredit.refund_mint_url == mint_url,
+        TemporaryCredit.refund_currency == unit,
     )
     result = await db_session.exec(query)
     return result.one() or 0
@@ -126,6 +129,8 @@ def run_migrations() -> None:
     import pathlib
 
     try:
+        logger.info("Starting database migrations")
+
         # Get the path to the alembic.ini file
         project_root = pathlib.Path(__file__).resolve().parents[2]
         alembic_ini_path = project_root / "alembic.ini"
@@ -142,6 +147,7 @@ def run_migrations() -> None:
         alembic_cfg.set_main_option("sqlalchemy.url", DATABASE_URL)
 
         # Run migrations to the latest revision
+        logger.info("Running migrations to latest revision")
         command.upgrade(alembic_cfg, "head")
 
         logger.info("Database migrations completed successfully")

routstr/core/logging.py

@@ -192,24 +192,21 @@ class SecurityFilter(logging.Filter):
         """Filter out sensitive information from log records."""
         try:
             message = record.getMessage()
-            standalone_patterns = [
-                r"Bearer\s+([a-zA-Z0-9_\-\.]{10,})",  # Bearer token (must be 10 characters or more to reduce false-positives)
-                r"cashu[A-Z]+([a-zA-Z0-9_\-\.=/+]+)",  # Cashu tokens
-                r"nsec[a-z0-9]+",  # Nostr Public / Private Key
-            ]
-            for pattern in standalone_patterns:
-                message = re.sub(pattern, "[REDACTED]", message, flags=re.IGNORECASE)
 
             for key in self.SENSITIVE_KEYS:
                 if key in message.lower():
-                    key_patterns = [
-                        rf"{key}\s*[:=]\s*([a-zA-Z0-9_\-\.=/+]+)",  # key:value or key=value (including any variant with spaces)
-                        rf'{key}\s*[:=]\s*["\']([^"\']+)["\']',  # key:"value" or key='value' (including any variant with spaces)
+                    patterns = [
+                        rf"{key}[:\s=]+([a-zA-Z0-9_\-\.]+)",  # key: value or key=value
+                        rf'{key}[:\s=]+["\']([^"\']+)["\']',  # key: "value" or key='value'
+                        r"Bearer\s+([a-zA-Z0-9_\-\.]+)",  # Bearer token
+                        r"cashu[A-Z]+([a-zA-Z0-9_\-\.=/+]+)",  # Cashu tokens
                     ]
-                    for pattern in key_patterns:
+
+                    for pattern in patterns:
                         message = re.sub(
                             pattern, f"{key}: [REDACTED]", message, flags=re.IGNORECASE
                         )
+
             record.msg = message
             record.args = ()
 
@@ -338,11 +335,6 @@ def setup_logging() -> None:
                 "handlers": ["console"] if console_enabled else [],
                 "propagate": False,
             },
-            "openai": {
-                "level": "WARNING",
-                "handlers": ["console"] if console_enabled else [],
-                "propagate": False,
-            },
             "httpcore": {
                 "level": "WARNING",
                 "handlers": ["console"] if console_enabled else [],
@@ -365,11 +357,6 @@ def setup_logging() -> None:
             },
             "watchfiles.main": {"level": "WARNING", "handlers": [], "propagate": False},
             "aiosqlite": {"level": "ERROR", "handlers": [], "propagate": False},
-            "alembic": {
-                "level": "WARNING",
-                "handlers": ["console"] if console_enabled else [],
-                "propagate": False,
-            },
         },
         "root": {
             "level": log_level,

routstr/core/main.py

@@ -1,33 +1,21 @@
-import asyncio
 import os
-from contextlib import asynccontextmanager
-from pathlib import Path
-from typing import AsyncGenerator
 
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import FileResponse, RedirectResponse
-from fastapi.staticfiles import StaticFiles
+from fastapi.responses import RedirectResponse
 from starlette.exceptions import HTTPException
 
 from ..balance import balance_router, deprecated_wallet_router
-from ..discovery import providers_cache_refresher, providers_router
-from ..nip91 import announce_provider
-from ..payment.models import (
-    cleanup_enabled_models_periodically,
-    models_router,
-    update_sats_pricing,
-)
-from ..payment.price import update_prices_periodically
-from ..proxy import initialize_upstreams, proxy_router, refresh_model_maps_periodically
-from ..wallet import periodic_payout
+from ..models.models import models_router
+from ..nostr.discovery import providers_router
+from ..proxy import proxy_router
 from .admin import admin_router
-from .db import create_session, init_db, run_migrations
 from .exceptions import general_exception_handler, http_exception_handler
 from .logging import get_logger, setup_logging
 from .middleware import LoggingMiddleware
-from .settings import SettingsService
 from .settings import settings as global_settings
+from .tasks import lifespan
+from .ui import setup_ui_routes
 
 # Initialize logging first
 setup_logging()
@@ -39,126 +27,6 @@ else:
     __version__ = "0.2.1"
 
 
-@asynccontextmanager
-async def lifespan(_: FastAPI) -> AsyncGenerator[None, None]:
-    logger.info("Application startup initiated", extra={"version": __version__})
-
-    btc_price_task = None
-    pricing_task = None
-    payout_task = None
-    nip91_task = None
-    providers_task = None
-    models_refresh_task = None
-    models_cleanup_task = None
-    model_maps_refresh_task = None
-
-    try:
-        # Run database migrations on startup
-        run_migrations()
-
-        # Initialize database connection pools
-        # This creates any tables that might not be tracked by migrations yet
-        await init_db()
-
-        # Initialize application settings (env -> computed -> DB precedence)
-        async with create_session() as session:
-            s = await SettingsService.initialize(session)
-
-        # Apply app metadata from settings
-        try:
-            app.title = s.name
-            app.description = s.description
-        except Exception:
-            pass
-
-        # await ensure_models_bootstrapped()
-
-        from ..payment.price import _update_prices
-        from ..proxy import get_upstreams
-        from ..upstream.helpers import refresh_upstreams_models_periodically
-
-        _update_prices_task = asyncio.create_task(_update_prices())
-        _initialize_upstreams_task = asyncio.create_task(initialize_upstreams())
-
-        btc_price_task = asyncio.create_task(update_prices_periodically())
-        pricing_task = asyncio.create_task(update_sats_pricing())
-        if global_settings.models_refresh_interval_seconds > 0:
-            models_refresh_task = asyncio.create_task(
-                refresh_upstreams_models_periodically(get_upstreams())
-            )
-        models_cleanup_task = asyncio.create_task(cleanup_enabled_models_periodically())
-        model_maps_refresh_task = asyncio.create_task(refresh_model_maps_periodically())
-        payout_task = asyncio.create_task(periodic_payout())
-        if global_settings.nsec:
-            nip91_task = asyncio.create_task(announce_provider())
-        if global_settings.providers_refresh_interval_seconds > 0:
-            providers_task = asyncio.create_task(providers_cache_refresher())
-
-        # ensure both setup tasks complete
-        await asyncio.gather(
-            _update_prices_task, _initialize_upstreams_task, return_exceptions=True
-        )
-
-        yield
-
-    except asyncio.CancelledError:
-        # Expected during shutdown
-        pass
-    except Exception as e:
-        logger.error(
-            "Application startup failed",
-            extra={"error": str(e), "error_type": type(e).__name__},
-        )
-        raise
-    finally:
-        logger.info("Application shutdown initiated")
-
-        if btc_price_task is not None:
-            btc_price_task.cancel()
-        if pricing_task is not None:
-            pricing_task.cancel()
-        if payout_task is not None:
-            payout_task.cancel()
-        if nip91_task is not None:
-            nip91_task.cancel()
-        if providers_task is not None:
-            providers_task.cancel()
-        if models_refresh_task is not None:
-            models_refresh_task.cancel()
-        if models_cleanup_task is not None:
-            models_cleanup_task.cancel()
-        if model_maps_refresh_task is not None:
-            model_maps_refresh_task.cancel()
-
-        try:
-            tasks_to_wait = []
-            if btc_price_task is not None:
-                tasks_to_wait.append(btc_price_task)
-            if pricing_task is not None:
-                tasks_to_wait.append(pricing_task)
-            if payout_task is not None:
-                tasks_to_wait.append(payout_task)
-            if nip91_task is not None:
-                tasks_to_wait.append(nip91_task)
-            if providers_task is not None:
-                tasks_to_wait.append(providers_task)
-            if models_refresh_task is not None:
-                tasks_to_wait.append(models_refresh_task)
-            if models_cleanup_task is not None:
-                tasks_to_wait.append(models_cleanup_task)
-            if model_maps_refresh_task is not None:
-                tasks_to_wait.append(model_maps_refresh_task)
-
-            if tasks_to_wait:
-                await asyncio.gather(*tasks_to_wait, return_exceptions=True)
-            logger.info("Background tasks stopped successfully")
-        except Exception as e:
-            logger.error(
-                "Error stopping background tasks",
-                extra={"error": str(e), "error_type": type(e).__name__},
-            )
-
-
 app = FastAPI(version=__version__, lifespan=lifespan)
 
 
@@ -197,144 +65,7 @@ async def providers() -> RedirectResponse:
     return RedirectResponse("/v1/providers/")
 
 
-UI_DIST_PATH = Path(__file__).parent.parent.parent / "ui_out"
-
-if UI_DIST_PATH.exists() and UI_DIST_PATH.is_dir():
-    logger.info(f"Serving static UI from {UI_DIST_PATH}")
-
-    app.mount(
-        "/_next",
-        StaticFiles(directory=UI_DIST_PATH / "_next", check_dir=True),
-        name="next-static",
-    )
-
-    @app.get("/", include_in_schema=False)
-    async def serve_root_ui() -> FileResponse:
-        return FileResponse(UI_DIST_PATH / "index.html")
-
-    # Add explicit route for /index.txt to redirect to /
-    @app.get("/index.txt", include_in_schema=False)
-    async def redirect_index_txt() -> RedirectResponse:
-        return RedirectResponse("/")
-
-    @app.get("/admin")
-    async def admin_redirect() -> FileResponse:
-        return FileResponse(UI_DIST_PATH / "index.html")
-
-    @app.get("/dashboard", include_in_schema=False)
-    async def serve_dashboard_ui() -> FileResponse:
-        return FileResponse(UI_DIST_PATH / "index.html")
-
-    @app.get("/login", include_in_schema=False)
-    async def serve_login_ui() -> FileResponse:
-        return FileResponse(UI_DIST_PATH / "login" / "index.html")
-
-    # Add explicit route for /login/index.txt to redirect to /login
-    @app.get("/login/index.txt", include_in_schema=False)
-    async def redirect_login_index_txt() -> RedirectResponse:
-        return RedirectResponse("/login")
-
-    @app.get("/model", include_in_schema=False)
-    async def serve_models_ui() -> FileResponse:
-        return FileResponse(UI_DIST_PATH / "model" / "index.html")
-
-    # Add explicit route for /model/index.txt to redirect to /model
-    @app.get("/model/index.txt", include_in_schema=False)
-    async def redirect_model_index_txt() -> RedirectResponse:
-        return RedirectResponse("/model")
-
-    @app.get("/providers", include_in_schema=False)
-    async def serve_providers_ui() -> FileResponse:
-        return FileResponse(UI_DIST_PATH / "providers" / "index.html")
-
-    # Add explicit route for /providers/index.txt to redirect to /providers
-    @app.get("/providers/index.txt", include_in_schema=False)
-    async def redirect_providers_index_txt() -> RedirectResponse:
-        return RedirectResponse("/providers")
-
-    @app.get("/settings", include_in_schema=False)
-    async def serve_settings_ui() -> FileResponse:
-        return FileResponse(UI_DIST_PATH / "settings" / "index.html")
-
-    # Add explicit route for /settings/index.txt to redirect to /settings
-    @app.get("/settings/index.txt", include_in_schema=False)
-    async def redirect_settings_index_txt() -> RedirectResponse:
-        return RedirectResponse("/settings")
-
-    @app.get("/transactions", include_in_schema=False)
-    async def serve_transactions_ui() -> FileResponse:
-        return FileResponse(UI_DIST_PATH / "transactions" / "index.html")
-
-    # Add explicit route for /transactions/index.txt to redirect to /transactions
-    @app.get("/transactions/index.txt", include_in_schema=False)
-    async def redirect_transactions_index_txt() -> RedirectResponse:
-        return RedirectResponse("/transactions")
-
-    @app.get("/balances", include_in_schema=False)
-    async def serve_balances_ui() -> FileResponse:
-        return FileResponse(UI_DIST_PATH / "balances" / "index.html")
-
-    # Add explicit route for /balances/index.txt to redirect to /balances
-    @app.get("/balances/index.txt", include_in_schema=False)
-    async def redirect_balances_index_txt() -> RedirectResponse:
-        return RedirectResponse("/balances")
-
-    @app.get("/logs", include_in_schema=False)
-    async def serve_logs_ui() -> FileResponse:
-        return FileResponse(UI_DIST_PATH / "logs" / "index.html")
-
-    # Add explicit route for /logs/index.txt to redirect to /logs
-    @app.get("/logs/index.txt", include_in_schema=False)
-    async def redirect_logs_index_txt() -> RedirectResponse:
-        return RedirectResponse("/logs")
-
-    @app.get("/usage", include_in_schema=False)
-    async def serve_usage_ui() -> FileResponse:
-        return FileResponse(UI_DIST_PATH / "usage" / "index.html")
-
-    # Add explicit route for /usage/index.txt to redirect to /usage
-    @app.get("/usage/index.txt", include_in_schema=False)
-    async def redirect_usage_index_txt() -> RedirectResponse:
-        return RedirectResponse("/usage")
-
-    @app.get("/unauthorized", include_in_schema=False)
-    async def serve_unauthorized_ui() -> FileResponse:
-        return FileResponse(UI_DIST_PATH / "unauthorized" / "index.html")
-
-    # Add explicit route for /unauthorized/index.txt to redirect to /unauthorized
-    @app.get("/unauthorized/index.txt", include_in_schema=False)
-    async def redirect_unauthorized_index_txt() -> RedirectResponse:
-        return RedirectResponse("/unauthorized")
-
-    @app.get("/favicon.ico", include_in_schema=False)
-    async def serve_favicon() -> FileResponse:
-        icon_path = UI_DIST_PATH / "icon.ico"
-        if icon_path.exists():
-            return FileResponse(icon_path)
-        return FileResponse(UI_DIST_PATH / "favicon.ico")
-
-    @app.get("/icon.ico", include_in_schema=False)
-    async def serve_icon() -> FileResponse:
-        return FileResponse(UI_DIST_PATH / "icon.ico")
-
-    app.mount(
-        "/static", StaticFiles(directory=UI_DIST_PATH, check_dir=True), name="ui-static"
-    )
-else:
-    logger.warning(
-        f"UI dist directory not found at {UI_DIST_PATH}, skipping static file serving"
-    )
-
-    @app.get("/", include_in_schema=False)
-    async def root_fallback() -> dict:
-        return {
-            "name": global_settings.name,
-            "description": global_settings.description,
-            "version": __version__,
-            "status": "running",
-            "ui": "not available",
-        }
-
+setup_ui_routes(app)
 
 app.include_router(models_router)
 app.include_router(admin_router)

routstr/core/settings.py

@@ -139,7 +139,7 @@ def resolve_bootstrap() -> Settings:
         pass
     if not base.onion_url:
         try:
-            from ..nip91 import discover_onion_url_from_tor  # type: ignore
+            from ..nostr.listing import discover_onion_url_from_tor
 
             discovered = discover_onion_url_from_tor()
             if discovered:

routstr/core/tasks.py

@@ -0,0 +1,132 @@
+import asyncio
+from contextlib import asynccontextmanager
+from typing import AsyncGenerator
+
+from fastapi import FastAPI
+
+from ..models.models import (
+    cleanup_enabled_models_periodically,
+    update_sats_pricing,
+)
+from ..nostr import announce_provider, providers_cache_refresher
+from ..payment.price import update_prices_periodically
+from ..payment.wallet import periodic_payout
+from ..proxy import get_upstreams, initialize_upstreams, refresh_model_maps_periodically
+from ..upstream.helpers import refresh_upstreams_models_periodically
+from .db import create_session, init_db, run_migrations
+from .logging import get_logger
+from .settings import SettingsService
+from .settings import settings as global_settings
+
+logger = get_logger(__name__)
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI) -> AsyncGenerator[None, None]:
+    # Extract version from app if available, or use default/log without it
+    version = getattr(app, "version", "unknown")
+
+    logger.info("Application startup initiated", extra={"version": version})
+
+    btc_price_task = None
+    pricing_task = None
+    payout_task = None
+    listing_task = None
+    providers_task = None
+    models_refresh_task = None
+    models_cleanup_task = None
+    model_maps_refresh_task = None
+
+    try:
+        # Run database migrations on startup
+        # This ensures the database schema is always up-to-date in production
+        # Migrations are idempotent - running them multiple times is safe
+        logger.info("Running database migrations")
+        run_migrations()
+
+        # Initialize database connection pools
+        # This creates any tables that might not be tracked by migrations yet
+        await init_db()
+
+        # Initialize application settings (env -> computed -> DB precedence)
+        async with create_session() as session:
+            s = await SettingsService.initialize(session)
+
+        # Apply app metadata from settings
+        try:
+            app.title = s.name
+            app.description = s.description
+        except Exception:
+            pass
+
+        _initialize_upstreams_task = asyncio.create_task(initialize_upstreams())
+
+        btc_price_task = asyncio.create_task(update_prices_periodically())
+        pricing_task = asyncio.create_task(update_sats_pricing())
+        if global_settings.models_refresh_interval_seconds > 0:
+            models_refresh_task = asyncio.create_task(
+                refresh_upstreams_models_periodically(get_upstreams())
+            )
+        models_cleanup_task = asyncio.create_task(cleanup_enabled_models_periodically())
+        model_maps_refresh_task = asyncio.create_task(refresh_model_maps_periodically())
+        payout_task = asyncio.create_task(periodic_payout())
+        listing_task = asyncio.create_task(announce_provider())
+        providers_task = asyncio.create_task(providers_cache_refresher())
+
+        await _initialize_upstreams_task
+
+        yield
+
+    except Exception as e:
+        logger.error(
+            "Application startup failed",
+            extra={"error": str(e), "error_type": type(e).__name__},
+        )
+        raise
+    finally:
+        logger.info("Application shutdown initiated")
+
+        if btc_price_task is not None:
+            btc_price_task.cancel()
+        if pricing_task is not None:
+            pricing_task.cancel()
+        if payout_task is not None:
+            payout_task.cancel()
+        if listing_task is not None:
+            listing_task.cancel()
+        if providers_task is not None:
+            providers_task.cancel()
+        if models_refresh_task is not None:
+            models_refresh_task.cancel()
+        if models_cleanup_task is not None:
+            models_cleanup_task.cancel()
+        if model_maps_refresh_task is not None:
+            model_maps_refresh_task.cancel()
+
+        try:
+            tasks_to_wait = []
+            if btc_price_task is not None:
+                tasks_to_wait.append(btc_price_task)
+            if pricing_task is not None:
+                tasks_to_wait.append(pricing_task)
+            if payout_task is not None:
+                tasks_to_wait.append(payout_task)
+            if listing_task is not None:
+                tasks_to_wait.append(listing_task)
+            if providers_task is not None:
+                tasks_to_wait.append(providers_task)
+            if models_refresh_task is not None:
+                tasks_to_wait.append(models_refresh_task)
+            if models_cleanup_task is not None:
+                tasks_to_wait.append(models_cleanup_task)
+            if model_maps_refresh_task is not None:
+                tasks_to_wait.append(model_maps_refresh_task)
+
+            if tasks_to_wait:
+                await asyncio.gather(*tasks_to_wait, return_exceptions=True)
+            logger.info("Background tasks stopped successfully")
+        except Exception as e:
+            logger.error(
+                "Error stopping background tasks",
+                extra={"error": str(e), "error_type": type(e).__name__},
+            )

routstr/core/ui.py

@@ -0,0 +1,160 @@
+from pathlib import Path
+
+from fastapi import APIRouter, FastAPI
+from fastapi.responses import FileResponse, RedirectResponse
+from fastapi.staticfiles import StaticFiles
+
+from .logging import get_logger
+from .settings import settings as global_settings
+
+logger = get_logger(__name__)
+
+
+def setup_ui_routes(app: FastAPI) -> None:
+    UI_DIST_PATH = Path(__file__).parent.parent.parent / "ui_out"
+
+    if UI_DIST_PATH.exists() and UI_DIST_PATH.is_dir():
+        logger.info(f"Serving static UI from {UI_DIST_PATH}")
+
+        app.mount(
+            "/_next",
+            StaticFiles(directory=UI_DIST_PATH / "_next", check_dir=True),
+            name="next-static",
+        )
+
+        router = APIRouter()
+
+        @router.get("/", include_in_schema=False)
+        async def serve_root_ui() -> FileResponse:
+            return FileResponse(UI_DIST_PATH / "index.html")
+
+        # Add explicit route for /index.txt to redirect to /
+        @router.get("/index.txt", include_in_schema=False)
+        async def redirect_index_txt() -> RedirectResponse:
+            return RedirectResponse("/")
+
+        @router.get("/admin")
+        async def admin_redirect() -> FileResponse:
+            return FileResponse(UI_DIST_PATH / "index.html")
+
+        @router.get("/dashboard", include_in_schema=False)
+        async def serve_dashboard_ui() -> FileResponse:
+            return FileResponse(UI_DIST_PATH / "index.html")
+
+        @router.get("/login", include_in_schema=False)
+        async def serve_login_ui() -> FileResponse:
+            return FileResponse(UI_DIST_PATH / "login" / "index.html")
+
+        # Add explicit route for /login/index.txt to redirect to /login
+        @router.get("/login/index.txt", include_in_schema=False)
+        async def redirect_login_index_txt() -> RedirectResponse:
+            return RedirectResponse("/login")
+
+        @router.get("/model", include_in_schema=False)
+        async def serve_models_ui() -> FileResponse:
+            return FileResponse(UI_DIST_PATH / "model" / "index.html")
+
+        # Add explicit route for /model/index.txt to redirect to /model
+        @router.get("/model/index.txt", include_in_schema=False)
+        async def redirect_model_index_txt() -> RedirectResponse:
+            return RedirectResponse("/model")
+
+        @router.get("/providers", include_in_schema=False)
+        async def serve_providers_ui() -> FileResponse:
+            return FileResponse(UI_DIST_PATH / "providers" / "index.html")
+
+        # Add explicit route for /providers/index.txt to redirect to /providers
+        @router.get("/providers/index.txt", include_in_schema=False)
+        async def redirect_providers_index_txt() -> RedirectResponse:
+            return RedirectResponse("/providers")
+
+        @router.get("/settings", include_in_schema=False)
+        async def serve_settings_ui() -> FileResponse:
+            return FileResponse(UI_DIST_PATH / "settings" / "index.html")
+
+        # Add explicit route for /settings/index.txt to redirect to /settings
+        @router.get("/settings/index.txt", include_in_schema=False)
+        async def redirect_settings_index_txt() -> RedirectResponse:
+            return RedirectResponse("/settings")
+
+        @router.get("/transactions", include_in_schema=False)
+        async def serve_transactions_ui() -> FileResponse:
+            return FileResponse(UI_DIST_PATH / "transactions" / "index.html")
+
+        # Add explicit route for /transactions/index.txt to redirect to /transactions
+        @router.get("/transactions/index.txt", include_in_schema=False)
+        async def redirect_transactions_index_txt() -> RedirectResponse:
+            return RedirectResponse("/transactions")
+
+        @router.get("/balances", include_in_schema=False)
+        async def serve_balances_ui() -> FileResponse:
+            return FileResponse(UI_DIST_PATH / "balances" / "index.html")
+
+        # Add explicit route for /balances/index.txt to redirect to /balances
+        @router.get("/balances/index.txt", include_in_schema=False)
+        async def redirect_balances_index_txt() -> RedirectResponse:
+            return RedirectResponse("/balances")
+
+        @router.get("/logs", include_in_schema=False)
+        async def serve_logs_ui() -> FileResponse:
+            return FileResponse(UI_DIST_PATH / "logs" / "index.html")
+
+        # Add explicit route for /logs/index.txt to redirect to /logs
+        @router.get("/logs/index.txt", include_in_schema=False)
+        async def redirect_logs_index_txt() -> RedirectResponse:
+            return RedirectResponse("/logs")
+
+        @router.get("/usage", include_in_schema=False)
+        async def serve_usage_ui() -> FileResponse:
+            return FileResponse(UI_DIST_PATH / "usage" / "index.html")
+
+        # Add explicit route for /usage/index.txt to redirect to /usage
+        @router.get("/usage/index.txt", include_in_schema=False)
+        async def redirect_usage_index_txt() -> RedirectResponse:
+            return RedirectResponse("/usage")
+
+        @router.get("/unauthorized", include_in_schema=False)
+        async def serve_unauthorized_ui() -> FileResponse:
+            return FileResponse(UI_DIST_PATH / "unauthorized" / "index.html")
+
+        # Add explicit route for /unauthorized/index.txt to redirect to /unauthorized
+        @router.get("/unauthorized/index.txt", include_in_schema=False)
+        async def redirect_unauthorized_index_txt() -> RedirectResponse:
+            return RedirectResponse("/unauthorized")
+
+        @router.get("/favicon.ico", include_in_schema=False)
+        async def serve_favicon() -> FileResponse:
+            icon_path = UI_DIST_PATH / "icon.ico"
+            if icon_path.exists():
+                return FileResponse(icon_path)
+            return FileResponse(UI_DIST_PATH / "favicon.ico")
+
+        @router.get("/icon.ico", include_in_schema=False)
+        async def serve_icon() -> FileResponse:
+            return FileResponse(UI_DIST_PATH / "icon.ico")
+
+        app.include_router(router)
+
+        app.mount(
+            "/static",
+            StaticFiles(directory=UI_DIST_PATH, check_dir=True),
+            name="ui-static",
+        )
+    else:
+        logger.warning(
+            f"UI dist directory not found at {UI_DIST_PATH}, skipping static file serving"
+        )
+
+        router = APIRouter()
+
+        @router.get("/", include_in_schema=False)
+        async def root_fallback() -> dict:
+            return {
+                "name": global_settings.name,
+                "description": global_settings.description,
+                "version": app.version,
+                "status": "running",
+                "ui": "not available",
+            }
+
+        app.include_router(router)

routstr/models/__init__.py

@@ -0,0 +1,9 @@
+from .metadata import async_fetch_openrouter_models
+from .models import Model
+
+__all__ = [
+    "Model",
+    "async_fetch_openrouter_models",
+]
+
+# specifically ai models that the node sells and not meaning database models

routstr/models/algorithm.py

@@ -2,11 +2,13 @@
 
 from typing import TYPE_CHECKING
 
-from .core.logging import get_logger
+from ..core import get_logger
+from ..models.crud import row_to_model
+from ..upstream.helpers import resolve_model_alias
 
 if TYPE_CHECKING:
-    from .payment.models import Model
-    from .upstream import BaseUpstreamProvider
+    from ..upstream import BaseUpstreamProvider
+    from .models import Model
 
 logger = get_logger(__name__)
 
@@ -150,6 +152,18 @@ def should_prefer_model(
     # Prefer lower adjusted cost
     should_replace = candidate_adjusted < current_adjusted
 
+    # Log provider changes when candidate wins
+    if should_replace:
+        candidate_provider_name = getattr(
+            candidate_provider, "upstream_name", "unknown"
+        )
+        current_provider_name = getattr(current_provider, "upstream_name", "unknown")
+        logger.debug(
+            f"Model selection for alias '{alias}': choosing {candidate_provider_name} "
+            f"(cost: ${candidate_adjusted:.6f}) over {current_provider_name} "
+            f"(cost: ${current_adjusted:.6f})"
+        )
+
     return should_replace
 
 
@@ -180,8 +194,6 @@ def create_model_mappings(
     Returns:
         Tuple of (model_instances, provider_map, unique_models)
     """
-    from .payment.models import _row_to_model
-    from .upstream.helpers import resolve_model_alias
 
     model_instances: dict[str, "Model"] = {}
     provider_map: dict[str, "BaseUpstreamProvider"] = {}
@@ -233,7 +245,7 @@ def create_model_mappings(
             # Apply overrides if present
             if model.id in overrides_by_id:
                 override_row, provider_fee = overrides_by_id[model.id]
-                model_to_use = _row_to_model(
+                model_to_use = row_to_model(
                     override_row, apply_provider_fee=True, provider_fee=provider_fee
                 )
             else:
@@ -242,12 +254,7 @@ def create_model_mappings(
             # Add to unique models
             base_id = get_base_model_id(model_to_use.id)
             if not is_openrouter or base_id not in unique_models:
-                unique_model = model_to_use.copy(
-                    update={
-                        "id": base_id,
-                        "upstream_provider_id": upstream.provider_type,
-                    }
-                )
+                unique_model = model_to_use.copy(update={"id": base_id})
                 unique_models[base_id] = unique_model
 
             # Get all aliases for this model

routstr/models/crud.py

@@ -0,0 +1,105 @@
+import json
+
+from ..core import get_logger
+from ..core.db import AsyncSession, ModelRow, UpstreamProviderRow
+from ..payment.price import sats_usd_price
+from .models import (
+    Architecture,
+    Model,
+    Pricing,
+    TopProvider,
+    _update_model_sats_pricing,
+    calculate_usd_max_costs,
+)
+
+logger = get_logger(__name__)
+
+
+def row_to_model(
+    row: ModelRow, apply_provider_fee: bool = False, provider_fee: float = 1.01
+) -> Model:
+    architecture = json.loads(row.architecture)
+    pricing = json.loads(row.pricing)
+    per_request_limits = (
+        json.loads(row.per_request_limits) if row.per_request_limits else None
+    )
+    top_provider_dict = json.loads(row.top_provider) if row.top_provider else None
+
+    if apply_provider_fee and isinstance(pricing, dict):
+        pricing = {k: float(v) * provider_fee for k, v in pricing.items()}
+
+    if isinstance(pricing, dict) and float(pricing.get("request", 0.0)) <= 0.0:
+        pricing["request"] = max(pricing.get("request", 0.0), 0.0)
+
+    parsed_pricing = Pricing.parse_obj(pricing)
+    model = Model(
+        id=row.id,
+        name=row.name,
+        created=row.created,
+        description=row.description,
+        context_length=row.context_length,
+        architecture=Architecture.parse_obj(architecture),
+        pricing=parsed_pricing,
+        sats_pricing=None,
+        per_request_limits=per_request_limits,
+        top_provider=TopProvider.parse_obj(top_provider_dict)
+        if top_provider_dict
+        else None,
+        enabled=row.enabled,
+        upstream_provider_id=row.upstream_provider_id,
+        canonical_slug=getattr(row, "canonical_slug", None),
+    )
+
+    if apply_provider_fee:
+        (
+            parsed_pricing.max_prompt_cost,
+            parsed_pricing.max_completion_cost,
+            parsed_pricing.max_cost,
+        ) = calculate_usd_max_costs(model)
+
+    try:
+        sats_to_usd = sats_usd_price()
+        model = _update_model_sats_pricing(model, sats_to_usd)
+    except Exception as e:
+        logger.warning(f"Could not calculate sats pricing: {e}")
+
+    return model
+
+
+async def list_models(
+    session: AsyncSession,
+    upstream_id: int,
+    include_disabled: bool = False,
+) -> list[Model]:
+    from sqlmodel import select
+
+    query = select(ModelRow)
+    if upstream_id is not None:
+        query = query.where(ModelRow.upstream_provider_id == upstream_id)
+    if not include_disabled:
+        query = query.where(ModelRow.enabled)
+
+    rows = (await session.exec(query)).all()  # type: ignore
+    provider_result = await session.exec(select(UpstreamProviderRow))
+    providers_by_id = {p.id: p for p in provider_result.all()}
+    return [
+        row_to_model(
+            r,
+            apply_provider_fee=True,
+            provider_fee=providers_by_id[r.upstream_provider_id].provider_fee
+            if r.upstream_provider_id in providers_by_id
+            else 1.01,
+        )
+        for r in rows
+    ]
+
+
+async def get_model_by_id(
+    model_id: str, provider_id: int, session: AsyncSession
+) -> Model | None:
+    row = await session.get(ModelRow, (model_id, provider_id))
+    if not row or not row.enabled:
+        return None
+    provider = await session.get(UpstreamProviderRow, provider_id)
+    provider_fee = provider.provider_fee if provider else 1.01
+    return row_to_model(row, apply_provider_fee=True, provider_fee=provider_fee)

routstr/models/metadata.py

@@ -0,0 +1,52 @@
+from typing import Final
+
+import httpx
+
+from ..core import get_logger
+
+logger = get_logger(__name__)
+
+DEFAULT_EXCLUDED_MODEL_IDS: Final[set[str]] = {
+    "openrouter/auto",
+    "google/gemini-2.5-pro-exp-03-25",
+    "opengvlab/internvl3-78b",
+    "openrouter/sonoma-dusk-alpha",
+    "openrouter/sonoma-sky-alpha",
+}
+
+
+async def async_fetch_openrouter_models(source_filter: str | None = None) -> list[dict]:
+    """Asynchronously fetch model information from OpenRouter API."""
+    base_url = "https://openrouter.ai/api/v1"
+
+    try:
+        async with httpx.AsyncClient() as client:
+            response = await client.get(f"{base_url}/models", timeout=30)
+            response.raise_for_status()
+            data = response.json()
+
+            models_data: list[dict] = []
+            for model in data.get("data", []):
+                model_id = model.get("id", "")
+
+                if source_filter:
+                    source_prefix = f"{source_filter}/"
+                    if not model_id.startswith(source_prefix):
+                        continue
+
+                    model = dict(model)
+                    model["id"] = model_id[len(source_prefix) :]
+                    model_id = model["id"]
+
+                if (
+                    "(free)" in model.get("name", "")
+                    or model_id in DEFAULT_EXCLUDED_MODEL_IDS
+                ):
+                    continue
+
+                models_data.append(model)
+
+            return models_data
+    except Exception as e:
+        logger.error(f"Error (async) fetching models from OpenRouter API: {e}")
+        return []

routstr/models/models.py

@@ -0,0 +1,376 @@
+import asyncio
+import json
+import random
+
+from fastapi import APIRouter, Depends
+from pydantic.v1 import BaseModel
+from sqlmodel import select
+from sqlmodel.ext.asyncio.session import AsyncSession
+
+from ..core.db import ModelRow, create_session, get_session
+from ..core.logging import get_logger
+from ..core.settings import settings
+from ..payment.price import sats_usd_price
+
+logger = get_logger(__name__)
+
+models_router = APIRouter()
+
+
+class Architecture(BaseModel):
+    modality: str
+    input_modalities: list[str]
+    output_modalities: list[str]
+    tokenizer: str
+    instruct_type: str | None
+
+
+class Pricing(BaseModel):
+    prompt: float
+    completion: float
+    request: float
+    image: float
+    web_search: float
+    internal_reasoning: float
+    max_prompt_cost: float = 0.0  # in sats not msats
+    max_completion_cost: float = 0.0  # in sats not msats
+    max_cost: float = 0.0  # in sats not msats
+
+
+class TopProvider(BaseModel):
+    context_length: int | None = None
+    max_completion_tokens: int | None = None
+    is_moderated: bool | None = None
+
+
+class Model(BaseModel):
+    id: str
+    name: str
+    created: int
+    description: str
+    context_length: int
+    architecture: Architecture
+    pricing: Pricing
+    sats_pricing: Pricing | None = None
+    per_request_limits: dict | None = None
+    top_provider: TopProvider | None = None
+    enabled: bool = True
+    upstream_provider_id: int | None = None
+    canonical_slug: str | None = None
+    alias_ids: list[str] | None = None
+
+    def __hash__(self) -> int:
+        return hash(self.id)
+
+
+def is_openrouter_upstream() -> bool:
+    try:
+        base = (settings.upstream_base_url or "").strip().rstrip("/")
+    except Exception:
+        return False
+    return base.lower() == "https://openrouter.ai/api/v1"
+
+
+def calculate_usd_max_costs(model: Model) -> tuple[float, float, float]:
+    """Calculate max costs in USD based on model context/token limits.
+
+    Args:
+        model: Model object
+
+    Returns:
+        Tuple of (max_prompt_cost, max_completion_cost, max_cost) in USD
+    """
+    min_req_msat = max(1, int(getattr(settings, "min_request_msat", 1)))
+    min_req_usd = float(min_req_msat) / 1_000_000.0
+
+    prompt_price = model.pricing.prompt
+    completion_price = model.pricing.completion
+
+    if model.top_provider and (
+        model.top_provider.context_length or model.top_provider.max_completion_tokens
+    ):
+        if (cl := model.top_provider.context_length) and (
+            mct := model.top_provider.max_completion_tokens
+        ):
+            if cl <= mct:
+                return (
+                    cl * prompt_price,
+                    cl * completion_price,
+                    cl * max(completion_price, prompt_price),
+                )
+            return (
+                cl * prompt_price,
+                mct * completion_price,
+                (cl - mct) * prompt_price + mct * completion_price,
+            )
+        elif cl := model.top_provider.context_length:
+            return (
+                cl * prompt_price,
+                cl * completion_price,
+                cl * max(completion_price, prompt_price),
+            )
+        elif mct := model.top_provider.max_completion_tokens:
+            return (
+                mct * prompt_price,
+                mct * completion_price,
+                mct * completion_price,
+            )
+    elif model.context_length:
+        return (
+            model.context_length * prompt_price,
+            model.context_length * completion_price,
+            model.context_length * max(completion_price, prompt_price),
+        )
+
+    p = prompt_price * 1_000_000
+    c = completion_price * 32_000
+    r = model.pricing.request * 100_000
+    i = model.pricing.image * 100
+    w = model.pricing.web_search * 1000
+    ir = model.pricing.internal_reasoning * 100
+    return (p, c, max(p + c + r + i + w + ir, min_req_usd))
+
+
+def _update_model_sats_pricing(model: Model, sats_to_usd: float) -> Model:
+    """Update a model's sats_pricing based on USD pricing and exchange rate.
+
+    Args:
+        model: Model object to update
+        sats_to_usd: Current sats to USD exchange rate
+
+    Returns:
+        Updated Model object with new sats_pricing
+    """
+    try:
+        min_req_msat = max(1, int(getattr(settings, "min_request_msat", 1)))
+        min_req_sats = float(min_req_msat) / 1000.0
+
+        sats = Pricing.parse_obj(
+            {k: v / sats_to_usd for k, v in model.pricing.dict().items()}
+        )
+
+        if sats.request <= 0.0:
+            sats.request = min_req_sats
+        if (sats.max_cost or 0.0) < min_req_sats:
+            sats.max_cost = min_req_sats
+
+        return Model(
+            id=model.id,
+            name=model.name,
+            created=model.created,
+            description=model.description,
+            context_length=model.context_length,
+            architecture=model.architecture,
+            pricing=model.pricing,
+            sats_pricing=sats,
+            per_request_limits=model.per_request_limits,
+            top_provider=model.top_provider,
+            enabled=model.enabled,
+            upstream_provider_id=model.upstream_provider_id,
+            canonical_slug=model.canonical_slug,
+            alias_ids=model.alias_ids,
+        )
+    except Exception as e:
+        logger.error(
+            "Failed to update sats pricing for model",
+            extra={
+                "model_id": model.id,
+                "error": str(e),
+                "error_type": type(e).__name__,
+            },
+        )
+        return model
+
+
+async def _update_sats_pricing_once() -> None:
+    """Update sats pricing once for all provider models (in-memory only)."""
+    from ..proxy import get_upstreams
+
+    upstreams = get_upstreams()
+    sats_to_usd = sats_usd_price()
+
+    updated_count = 0
+    for upstream in upstreams:
+        updated_models = [
+            _update_model_sats_pricing(m, sats_to_usd)
+            for m in upstream.get_cached_models()
+        ]
+        upstream._models_cache = updated_models
+        upstream._models_by_id = {m.id: m for m in updated_models}
+        updated_count += len(updated_models)
+
+    if updated_count > 0:
+        logger.info("Updated sats pricing", extra={"models_updated": updated_count})
+
+
+async def update_sats_pricing() -> None:
+    """Periodically update sats pricing for all provider models and database overrides."""
+    try:
+        if not settings.enable_pricing_refresh:
+            return
+    except Exception:
+        pass
+
+    await _update_sats_pricing_once()
+
+    while True:
+        try:
+            interval = getattr(settings, "pricing_refresh_interval_seconds", 120)
+            jitter = max(0.0, float(interval) * 0.1)
+            await asyncio.sleep(interval + random.uniform(0, jitter))
+        except asyncio.CancelledError:
+            break
+
+        try:
+            try:
+                if not settings.enable_pricing_refresh:
+                    return
+            except Exception:
+                pass
+
+            await _update_sats_pricing_once()
+        except asyncio.CancelledError:
+            break
+        except Exception as e:
+            logger.error(f"Error updating sats pricing: {e}")
+
+
+async def cleanup_enabled_models_periodically() -> None:
+    """Background task to clean up enabled models that match upstream pricing.
+
+    When model is enabled (enabled=True), remove it from DB if it matches upstream pricing.
+    Keep it in DB only if pricing differs from upstream or if it's disabled.
+    """
+    interval = getattr(
+        settings, "models_cleanup_interval_seconds", 300
+    )  # 5 minutes default
+    if not interval or interval <= 0:
+        return
+
+    while True:
+        try:
+            await _cleanup_enabled_models_once()
+        except asyncio.CancelledError:
+            break
+        except Exception as e:
+            logger.error(
+                "Error during enabled models cleanup",
+                extra={"error": str(e), "error_type": type(e).__name__},
+            )
+
+        try:
+            jitter = max(0.0, float(interval) * 0.1)
+            await asyncio.sleep(interval + random.uniform(0, jitter))
+        except asyncio.CancelledError:
+            break
+
+
+async def _cleanup_enabled_models_once() -> None:
+    """Clean up enabled models that match upstream pricing."""
+    from ..proxy import get_upstreams
+
+    async with create_session() as session:
+        # Get all enabled models from DB
+        result = await session.exec(
+            select(ModelRow).where(
+                ModelRow.enabled,  # Only enabled models
+            )
+        )
+        db_models = result.all()
+
+        if not db_models:
+            return
+
+        upstreams = get_upstreams()
+        models_to_remove = []
+
+        for db_model in db_models:
+            # Find corresponding upstream model
+            upstream_model = None
+            for upstream in upstreams:
+                upstream_model = upstream.get_cached_model_by_id(db_model.id)
+                if upstream_model:
+                    break
+
+            if not upstream_model:
+                continue
+
+            # Compare pricing to see if they match
+            db_pricing = json.loads(db_model.pricing)
+            upstream_pricing = upstream_model.pricing.dict()
+
+            # Check if pricing matches (with small tolerance for float comparison)
+            pricing_matches = _pricing_matches(db_pricing, upstream_pricing)
+
+            if pricing_matches:
+                models_to_remove.append(db_model)
+                logger.info(
+                    f"Removing enabled model {db_model.id} - matches upstream pricing",
+                    extra={"model_id": db_model.id},
+                )
+
+        # Remove models that match upstream pricing
+        for model in models_to_remove:
+            await session.delete(model)
+
+        if models_to_remove:
+            await session.commit()
+            logger.info(
+                f"Cleaned up {len(models_to_remove)} enabled models that match upstream pricing"
+            )
+
+
+def _pricing_matches(
+    db_pricing: dict, upstream_pricing: dict, tolerance: float = 0.0
+) -> bool:
+    """Check if pricing dictionaries match within tolerance."""
+    keys_to_compare = [
+        "prompt",
+        "completion",
+        "request",
+        "image",
+        "web_search",
+        "internal_reasoning",
+    ]
+
+    for key in keys_to_compare:
+        db_val = int(float(db_pricing.get(key, 0.0)) * 1000000)
+        upstream_val = int(float(upstream_pricing.get(key, 0.0)) * 1000000)
+
+        if abs(db_val - upstream_val) > tolerance:
+            return False
+
+    return True
+
+
+def _model_to_row_payload(model: Model) -> dict[str, str | int | bool | None]:
+    return {
+        "id": model.id,
+        "name": model.name,
+        "created": model.created,
+        "description": model.description,
+        "context_length": model.context_length,
+        "architecture": json.dumps(model.architecture.dict()),
+        "pricing": json.dumps(model.pricing.dict()),
+        "sats_pricing": json.dumps(model.sats_pricing.dict())
+        if model.sats_pricing
+        else None,
+        "per_request_limits": json.dumps(model.per_request_limits)
+        if model.per_request_limits is not None
+        else None,
+        "top_provider": json.dumps(model.top_provider.dict())
+        if model.top_provider is not None
+        else None,
+        "enabled": model.enabled,
+        "upstream_provider_id": model.upstream_provider_id,
+    }
+
+
+@models_router.get("/v1/models")
+@models_router.get("/models", include_in_schema=False)
+async def models(session: AsyncSession = Depends(get_session)) -> dict:
+    """Get all available models from all providers with database overrides applied."""
+    from ..proxy import get_unique_models
+
+    items = get_unique_models()
+    return {"data": items}

routstr/nostr/__init__.py

@@ -0,0 +1,7 @@
+from .discovery import providers_cache_refresher
+from .listing import announce_provider
+
+__all__ = [
+    "providers_cache_refresher",
+    "announce_provider",
+]

routstr/nostr/discovery.py

@@ -8,8 +8,8 @@ import httpx
 import websockets
 from fastapi import APIRouter
 
-from .core.logging import get_logger
-from .core.settings import settings
+from ..core.logging import get_logger
+from ..core.settings import settings
 
 logger = get_logger(__name__)
 

routstr/nostr/listing.py

@@ -18,15 +18,15 @@ from nostr.key import PrivateKey
 from nostr.message_type import ClientMessageType
 from nostr.relay_manager import RelayManager
 
-from .core import get_logger
-from .core.settings import settings
+from ..core import get_logger
+from ..core.settings import settings
 
 logger = get_logger(__name__)
 
 
 def get_app_version() -> str | None:
     try:
-        from .core.main import __version__ as imported_version
+        from ..core.main import __version__ as imported_version
 
         return imported_version
     except Exception:
@@ -71,7 +71,7 @@ def nsec_to_keypair(nsec: str) -> tuple[str, str] | None:
         return None
 
 
-def create_nip91_event(
+def create_listing_event(
     private_key_hex: str,
     provider_id: str,
     endpoint_urls: list[str],
@@ -164,7 +164,7 @@ def events_semantically_equal(a: dict[str, Any], b: dict[str, Any]) -> bool:
     return True
 
 
-async def query_nip91_events(
+async def query_listing_events(
     relay_url: str,
     pubkey: str,
     provider_id: str | None = None,
@@ -188,7 +188,7 @@ async def query_nip91_events(
 
             flt = Filter(kinds=[38421], authors=[pubkey], limit=10)
             filters = Filters([flt])
-            sub_id = f"nip91_{int(time.time())}"
+            sub_id = f"routstr_listing_{int(time.time())}"
             rm.add_subscription(sub_id, filters)
             req: list[Any] = [ClientMessageType.REQUEST, sub_id]
             req.extend(filters.to_json_array())
@@ -294,7 +294,7 @@ async def _determine_provider_id(public_key_hex: str, relay_urls: list[str]) ->
 
     async def query_single_relay(relay_url: str) -> list[dict[str, Any]]:
         try:
-            events, _ok = await query_nip91_events(relay_url, public_key_hex, None)
+            events, _ok = await query_listing_events(relay_url, public_key_hex, None)
             return events
         except Exception:
             return []
@@ -434,7 +434,7 @@ async def announce_provider() -> None:
 
     # Create the candidate event that we would publish
     version_str = get_app_version()
-    candidate_event = create_nip91_event(
+    candidate_event = create_listing_event(
         private_key_hex=private_key_hex,
         provider_id=provider_id,
         endpoint_urls=endpoint_urls,
@@ -474,7 +474,7 @@ async def announce_provider() -> None:
         if _should_skip(relay_url):
             logger.debug(f"Skipping {relay_url} due to backoff")
             continue
-        events, ok = await query_nip91_events(relay_url, public_key_hex, provider_id)
+        events, ok = await query_listing_events(relay_url, public_key_hex, provider_id)
         if ok:
             _register_success(relay_url)
             existing_events.extend(events)
@@ -518,7 +518,7 @@ async def announce_provider() -> None:
 
             # Build fresh candidate event for comparison
             version_str = get_app_version()
-            candidate_event = create_nip91_event(
+            candidate_event = create_listing_event(
                 private_key_hex=private_key_hex,
                 provider_id=provider_id,
                 endpoint_urls=endpoint_urls,
@@ -533,7 +533,7 @@ async def announce_provider() -> None:
                 if _should_skip(relay_url):
                     logger.debug(f"Skipping {relay_url} due to backoff")
                     continue
-                events, ok = await query_nip91_events(
+                events, ok = await query_listing_events(
                     relay_url, public_key_hex, provider_id
                 )
                 if ok:

routstr/payment/__init__.py

@@ -1,4 +1,4 @@
-from .cost_calculation import CostData, CostDataError, MaxCostData, calculate_cost
+from .cost import CostData, CostDataError, MaxCostData, calculate_cost
 
 __all__ = [
     "CostData",

routstr/payment/cashu.py

@@ -0,0 +1,48 @@
+from cashu.wallet.helpers import deserialize_token_from_string
+from fastapi import HTTPException
+
+from ..core import get_logger
+
+logger = get_logger(__name__)
+
+
+def get_token_balance_msat(cashu_token: str) -> int:
+    try:
+        token = deserialize_token_from_string(cashu_token)
+    except Exception:
+        raise HTTPException(401, "Invalid authentication token format")
+
+    if token.unit == "sat":
+        return token.amount * 1000
+    if token.unit == "msat":
+        return token.amount
+
+    raise HTTPException(401, f"Unit {token.unit} not supported yet")
+
+
+def pre_check_header_token_balance(
+    headers: dict, body: dict, max_cost_for_model: int
+) -> None:
+    if x_cashu := headers.get("x-cashu", None):
+        cashu_token = x_cashu
+    elif auth := headers.get("authorization", None):
+        cashu_token = auth.split(" ")[1] if len(auth.split(" ")) > 1 else ""
+    else:
+        raise HTTPException(status_code=401, detail="Unauthorized")
+
+    # Handle regular API keys (sk-*)
+    if cashu_token.startswith("sk-"):
+        return
+
+    amount_msat = get_token_balance_msat(cashu_token)
+
+    if max_cost_for_model > amount_msat:
+        raise HTTPException(
+            status_code=413,
+            detail={
+                "reason": "Insufficient balance",
+                "amount_required_msat": max_cost_for_model,
+                "model": body.get("model", "unknown"),
+                "type": "minimum_balance_required",
+            },
+        )

routstr/payment/cost.py

@@ -0,0 +1,425 @@
+import base64
+import math
+from io import BytesIO
+from typing import Any
+
+import httpx
+from PIL import Image
+from pydantic.v1 import BaseModel
+
+from ..core import get_logger
+from ..core.settings import settings
+from ..models.models import Model
+
+logger = get_logger(__name__)
+
+
+class CostData(BaseModel):
+    base_msats: int
+    input_msats: int
+    output_msats: int
+    total_msats: int
+
+
+class MaxCostData(CostData):
+    pass
+
+
+class CostDataError(BaseModel):
+    message: str
+    code: str
+
+
+def calculate_cost(
+    response_data: dict, max_cost: int
+) -> CostData | MaxCostData | CostDataError:
+    """
+    Calculate the cost of an API request based on token usage.
+
+    Args:
+        response_data: Response data containing usage information
+        max_cost: Maximum cost in millisats
+
+    Returns:
+        Cost data or error information
+    """
+    logger.debug(
+        "Starting cost calculation",
+        extra={
+            "max_cost_msats": max_cost,
+            "has_usage_data": "usage" in response_data,
+            "response_model": response_data.get("model", "unknown"),
+        },
+    )
+
+    cost_data = MaxCostData(
+        base_msats=max_cost,
+        input_msats=0,
+        output_msats=0,
+        total_msats=max_cost,
+    )
+
+    if "usage" not in response_data or response_data["usage"] is None:
+        logger.warning(
+            "No usage data in response, using base cost only",
+            extra={
+                "max_cost_msats": max_cost,
+                "model": response_data.get("model", "unknown"),
+            },
+        )
+        return cost_data
+
+    MSATS_PER_1K_INPUT_TOKENS: float = (
+        float(settings.fixed_per_1k_input_tokens) * 1000.0
+    )
+    MSATS_PER_1K_OUTPUT_TOKENS: float = (
+        float(settings.fixed_per_1k_output_tokens) * 1000.0
+    )
+
+    if not settings.fixed_pricing:
+        response_model = response_data.get("model", "")
+        logger.debug(
+            "Using model-based pricing",
+            extra={"model": response_model},
+        )
+
+        from ..proxy import get_model_instance
+
+        model_obj = get_model_instance(response_model)
+
+        if not model_obj:
+            logger.error(
+                "Invalid model in response",
+                extra={"response_model": response_model},
+            )
+            return CostDataError(
+                message=f"Invalid model in response: {response_model}",
+                code="model_not_found",
+            )
+
+        if not model_obj.sats_pricing:
+            logger.error(
+                "Model pricing not defined",
+                extra={"model": response_model, "model_id": response_model},
+            )
+            return CostDataError(
+                message="Model pricing not defined", code="pricing_not_found"
+            )
+
+        try:
+            mspp = float(model_obj.sats_pricing.prompt)
+            mspc = float(model_obj.sats_pricing.completion)
+        except Exception:
+            return CostDataError(message="Invalid pricing data", code="pricing_invalid")
+
+        MSATS_PER_1K_INPUT_TOKENS = mspp * 1_000_000.0
+        MSATS_PER_1K_OUTPUT_TOKENS = mspc * 1_000_000.0
+
+        logger.info(
+            "Applied model-specific pricing",
+            extra={
+                "model": response_model,
+                "input_price_msats_per_1k": MSATS_PER_1K_INPUT_TOKENS,
+                "output_price_msats_per_1k": MSATS_PER_1K_OUTPUT_TOKENS,
+            },
+        )
+
+    if not (MSATS_PER_1K_OUTPUT_TOKENS and MSATS_PER_1K_INPUT_TOKENS):
+        logger.warning(
+            "No token pricing configured, using base cost",
+            extra={
+                "base_cost_msats": max_cost,
+                "model": response_data.get("model", "unknown"),
+            },
+        )
+        return cost_data
+
+    input_tokens = response_data.get("usage", {}).get("prompt_tokens", 0)
+    output_tokens = response_data.get("usage", {}).get("completion_tokens", 0)
+
+    input_msats = round(input_tokens / 1000 * MSATS_PER_1K_INPUT_TOKENS, 3)
+    output_msats = round(output_tokens / 1000 * MSATS_PER_1K_OUTPUT_TOKENS, 3)
+    token_based_cost = math.ceil(input_msats + output_msats)
+
+    logger.info(
+        "Calculated token-based cost",
+        extra={
+            "input_tokens": input_tokens,
+            "output_tokens": output_tokens,
+            "input_cost_msats": input_msats,
+            "output_cost_msats": output_msats,
+            "total_cost_msats": token_based_cost,
+            "model": response_data.get("model", "unknown"),
+        },
+    )
+
+    return CostData(
+        base_msats=0,
+        input_msats=int(input_msats),
+        output_msats=int(output_msats),
+        total_msats=token_based_cost,
+    )
+
+
+def get_max_cost_for_model(model_obj: Model) -> int:
+    """Get the maximum cost for a specific model from providers with overrides."""
+    if settings.fixed_pricing:
+        default_cost_msats = settings.fixed_cost_per_request * 1000
+        return max(settings.min_request_msat, default_cost_msats)
+
+    if model_obj.sats_pricing:
+        max_cost = (
+            model_obj.sats_pricing.max_cost
+            * 1000
+            * (1 - settings.tolerance_percentage / 100)
+        )
+        calculated_msats = int(max_cost)
+        return max(settings.min_request_msat, calculated_msats)
+
+    logger.warning(
+        "Model pricing not found, using fixed cost",
+        extra={
+            "model": model_obj.id,
+            "default_cost_msats": settings.fixed_cost_per_request * 1000,
+        },
+    )
+    return max(settings.min_request_msat, settings.fixed_cost_per_request * 1000)
+
+
+async def calculate_discounted_max_cost(
+    max_cost_for_model: int,
+    body: dict,
+    model_obj: Any | None = None,
+) -> int:
+    """Calculate the discounted max cost for a request using model pricing when available."""
+    if settings.fixed_pricing:
+        return max_cost_for_model
+
+    model = body.get("model", "unknown")
+
+    model_pricing = model_obj.sats_pricing if model_obj else None
+    if not model_pricing:
+        return max_cost_for_model
+
+    tol = settings.tolerance_percentage
+    tol_factor = max(0.0, 1 - float(tol) / 100.0)
+    max_prompt_allowed_sats = model_pricing.max_prompt_cost * tol_factor
+    max_completion_allowed_sats = model_pricing.max_completion_cost * tol_factor
+
+    adjusted = max_cost_for_model
+
+    if messages := body.get("messages"):
+        prompt_tokens = _estimate_tokens(messages)
+
+        image_tokens = await _estimate_image_tokens_in_messages(messages)
+        if image_tokens > 0:
+            logger.debug(
+                "Found images in request",
+                extra={
+                    "model": model,
+                    "image_tokens": image_tokens,
+                },
+            )
+            prompt_tokens += image_tokens
+
+        estimated_prompt_delta_sats = (
+            max_prompt_allowed_sats - prompt_tokens * model_pricing.prompt
+        )
+        if estimated_prompt_delta_sats > 0:
+            adjusted = adjusted - math.floor(estimated_prompt_delta_sats * 1000)
+
+    max_tokens_raw = body.get("max_tokens", None)
+    if max_tokens_raw is not None:
+        try:
+            max_tokens_int = int(max_tokens_raw)
+        except (TypeError, ValueError):
+            logger.warning(
+                "Invalid max_tokens; ignoring in cost adjustment",
+                extra={"max_tokens": str(max_tokens_raw)[:64], "model": model},
+            )
+        else:
+            estimated_completion_delta_sats = (
+                max_completion_allowed_sats - max_tokens_int * model_pricing.completion
+            )
+            if estimated_completion_delta_sats > 0:
+                adjusted = adjusted - math.floor(estimated_completion_delta_sats * 1000)
+
+    logger.debug(
+        "Discounted max cost computed",
+        extra={
+            "model": model,
+            "original_msats": max_cost_for_model,
+            "adjusted_msats": adjusted,
+            "tolerance_pct": tol,
+        },
+    )
+
+    return max(0, adjusted)
+
+
+def _estimate_tokens(messages: list) -> int:
+    """Estimate tokens for text content, excluding image_url fields."""
+    total = 0
+    for msg in messages:
+        if isinstance(msg, dict):
+            content = msg.get("content")
+            if isinstance(content, str):
+                total += len(content)
+            elif isinstance(content, list):
+                total += sum(
+                    len(item.get("text", ""))
+                    for item in content
+                    if isinstance(item, dict) and item.get("type") == "text"
+                )
+    return total // 3
+
+
+def _get_image_dimensions(image_data: bytes) -> tuple[int, int]:
+    """Extract image dimensions from image bytes."""
+    try:
+        img = Image.open(BytesIO(image_data))
+        return img.size
+    except Exception as e:
+        logger.warning(
+            "Failed to get image dimensions, using default",
+            extra={"error": str(e)},
+        )
+        return (512, 512)
+
+
+async def _fetch_image_from_url(url: str) -> bytes | None:
+    """Fetch image from URL."""
+    try:
+        async with httpx.AsyncClient(timeout=10.0) as client:
+            response = await client.get(url)
+            response.raise_for_status()
+            return response.content
+    except Exception as e:
+        logger.warning(
+            "Failed to fetch image from URL",
+            extra={"error": str(e), "url": url[:100]},
+        )
+        return None
+
+
+def _calculate_image_tokens(width: int, height: int, detail: str = "auto") -> int:
+    """Calculate image tokens based on OpenAI's vision pricing.
+
+    For low detail: 85 tokens
+    For high detail/auto: 85 base tokens + 170 tokens per 512px tile
+    """
+    if detail == "low":
+        return 85
+
+    if width > 2048 or height > 2048:
+        aspect_ratio = width / height
+        if width > height:
+            width = 2048
+            height = int(width / aspect_ratio)
+        else:
+            height = 2048
+            width = int(height * aspect_ratio)
+
+    if width > 768 or height > 768:
+        aspect_ratio = width / height
+        if width > height:
+            width = 768
+            height = int(width / aspect_ratio)
+        else:
+            height = 768
+            width = int(height * aspect_ratio)
+
+    tiles_width = (width + 511) // 512
+    tiles_height = (height + 511) // 512
+    num_tiles = tiles_width * tiles_height
+
+    return 85 + (170 * num_tiles)
+
+
+async def _estimate_image_tokens_in_messages(messages: list) -> int:
+    """Estimate total tokens for all images in messages.
+
+    Supports both base64 encoded images and image URLs.
+    """
+    total_image_tokens = 0
+
+    for message in messages:
+        if not isinstance(message, dict):
+            continue
+
+        content = message.get("content")
+        if not content:
+            continue
+
+        if isinstance(content, str):
+            continue
+
+        if not isinstance(content, list):
+            continue
+
+        for content_item in content:
+            if not isinstance(content_item, dict):
+                continue
+
+            content_type = content_item.get("type")
+            if content_type not in ("image_url", "input_image"):
+                continue
+
+            image_url_data = content_item.get("image_url")
+            if not image_url_data:
+                continue
+
+            if isinstance(image_url_data, str):
+                url = image_url_data
+                detail = "auto"
+            elif isinstance(image_url_data, dict):
+                url = image_url_data.get("url", "")
+                detail = image_url_data.get("detail", "auto")
+            else:
+                continue
+
+            if not url:
+                continue
+
+            if url.startswith("data:image/"):
+                try:
+                    header, base64_data = url.split(",", 1)
+                    image_bytes = base64.b64decode(base64_data)
+                    width, height = _get_image_dimensions(image_bytes)
+                    tokens = _calculate_image_tokens(width, height, detail)
+                    total_image_tokens += tokens
+                    logger.debug(
+                        "Calculated tokens for base64 image",
+                        extra={
+                            "width": width,
+                            "height": height,
+                            "detail": detail,
+                            "tokens": tokens,
+                        },
+                    )
+                except Exception as e:
+                    logger.warning(
+                        "Failed to process base64 image",
+                        extra={"error": str(e)},
+                    )
+                    total_image_tokens += 85
+            else:
+                image_bytes_or_none = await _fetch_image_from_url(url)
+                if image_bytes_or_none:
+                    width, height = _get_image_dimensions(image_bytes_or_none)
+                    tokens = _calculate_image_tokens(width, height, detail)
+                    total_image_tokens += tokens
+                    logger.debug(
+                        "Calculated tokens for URL image",
+                        extra={
+                            "url": url[:100],
+                            "width": width,
+                            "height": height,
+                            "detail": detail,
+                            "tokens": tokens,
+                        },
+                    )
+                else:
+                    total_image_tokens += 85
+
+    return total_image_tokens

routstr/payment/cost_calculation.py

@@ -1,156 +0,0 @@
-import math
-
-from pydantic.v1 import BaseModel
-
-from ..core import get_logger
-from ..core.db import AsyncSession
-from ..core.settings import settings
-
-logger = get_logger(__name__)
-
-
-class CostData(BaseModel):
-    base_msats: int
-    input_msats: int
-    output_msats: int
-    total_msats: int
-
-
-class MaxCostData(CostData):
-    pass
-
-
-class CostDataError(BaseModel):
-    message: str
-    code: str
-
-
-async def calculate_cost(  # todo: can be sync
-    response_data: dict, max_cost: int, session: AsyncSession
-) -> CostData | MaxCostData | CostDataError:
-    """
-    Calculate the cost of an API request based on token usage.
-
-    Args:
-        response_data: Response data containing usage information
-        max_cost: Maximum cost in millisats
-
-    Returns:
-        Cost data or error information
-    """
-    logger.debug(
-        "Starting cost calculation",
-        extra={
-            "max_cost_msats": max_cost,
-            "has_usage_data": "usage" in response_data,
-            "response_model": response_data.get("model", "unknown"),
-        },
-    )
-
-    cost_data = MaxCostData(
-        base_msats=max_cost,
-        input_msats=0,
-        output_msats=0,
-        total_msats=max_cost,
-    )
-
-    if "usage" not in response_data or response_data["usage"] is None:
-        logger.warning(
-            "No usage data in response, using base cost only",
-            extra={
-                "max_cost_msats": max_cost,
-                "model": response_data.get("model", "unknown"),
-            },
-        )
-        return cost_data
-
-    MSATS_PER_1K_INPUT_TOKENS: float = (
-        float(settings.fixed_per_1k_input_tokens) * 1000.0
-    )
-    MSATS_PER_1K_OUTPUT_TOKENS: float = (
-        float(settings.fixed_per_1k_output_tokens) * 1000.0
-    )
-
-    if not settings.fixed_pricing:
-        response_model = response_data.get("model", "")
-        logger.debug(
-            "Using model-based pricing",
-            extra={"model": response_model},
-        )
-
-        from ..proxy import get_model_instance
-
-        model_obj = get_model_instance(response_model)
-
-        if not model_obj:
-            logger.error(
-                "Invalid model in response",
-                extra={"response_model": response_model},
-            )
-            return CostDataError(
-                message=f"Invalid model in response: {response_model}",
-                code="model_not_found",
-            )
-
-        if not model_obj.sats_pricing:
-            logger.error(
-                "Model pricing not defined",
-                extra={"model": response_model, "model_id": response_model},
-            )
-            return CostDataError(
-                message="Model pricing not defined", code="pricing_not_found"
-            )
-
-        try:
-            mspp = float(model_obj.sats_pricing.prompt)
-            mspc = float(model_obj.sats_pricing.completion)
-        except Exception:
-            return CostDataError(message="Invalid pricing data", code="pricing_invalid")
-
-        MSATS_PER_1K_INPUT_TOKENS = mspp * 1_000_000.0
-        MSATS_PER_1K_OUTPUT_TOKENS = mspc * 1_000_000.0
-
-        logger.info(
-            "Applied model-specific pricing",
-            extra={
-                "model": response_model,
-                "input_price_msats_per_1k": MSATS_PER_1K_INPUT_TOKENS,
-                "output_price_msats_per_1k": MSATS_PER_1K_OUTPUT_TOKENS,
-            },
-        )
-
-    if not (MSATS_PER_1K_OUTPUT_TOKENS and MSATS_PER_1K_INPUT_TOKENS):
-        logger.warning(
-            "No token pricing configured, using base cost",
-            extra={
-                "base_cost_msats": max_cost,
-                "model": response_data.get("model", "unknown"),
-            },
-        )
-        return cost_data
-
-    input_tokens = response_data.get("usage", {}).get("prompt_tokens", 0)
-    output_tokens = response_data.get("usage", {}).get("completion_tokens", 0)
-
-    input_msats = round(input_tokens / 1000 * MSATS_PER_1K_INPUT_TOKENS, 3)
-    output_msats = round(output_tokens / 1000 * MSATS_PER_1K_OUTPUT_TOKENS, 3)
-    token_based_cost = math.ceil(input_msats + output_msats)
-
-    logger.info(
-        "Calculated token-based cost",
-        extra={
-            "input_tokens": input_tokens,
-            "output_tokens": output_tokens,
-            "input_cost_msats": input_msats,
-            "output_cost_msats": output_msats,
-            "total_cost_msats": token_based_cost,
-            "model": response_data.get("model", "unknown"),
-        },
-    )
-
-    return CostData(
-        base_msats=0,
-        input_msats=int(input_msats),
-        output_msats=int(output_msats),
-        total_msats=token_based_cost,
-    )

routstr/payment/helpers.py

@@ -1,396 +1,18 @@
-import base64
 import json
 import math
-from io import BytesIO
-from typing import Any
 
-import httpx
 from fastapi import HTTPException, Response
 from fastapi.requests import Request
-from PIL import Image
-from sqlmodel.ext.asyncio.session import AsyncSession
+from sqlmodel import col, update
 
 from ..core import get_logger
-from ..core.settings import settings
-from ..wallet import deserialize_token_from_string
+from ..core.db import AsyncSession, TemporaryCredit
+from ..payment.cost import CostData, CostDataError, MaxCostData, calculate_cost
 
 logger = get_logger(__name__)
 
 
-def check_token_balance(headers: dict, body: dict, max_cost_for_model: int) -> None:
-    if x_cashu := headers.get("x-cashu", None):
-        cashu_token = x_cashu
-        logger.debug(
-            "Using X-Cashu token",
-            extra={
-                "token_preview": cashu_token[:20] + "..."
-                if len(cashu_token) > 20
-                else cashu_token
-            },
-        )
-    elif auth := headers.get("authorization", None):
-        cashu_token = auth.split(" ")[1] if len(auth.split(" ")) > 1 else ""
-        logger.debug(
-            "Using Authorization header token",
-            extra={
-                "token_preview": cashu_token[:20] + "..."
-                if len(cashu_token) > 20
-                else cashu_token
-            },
-        )
-    else:
-        logger.error("No authentication token provided")
-        raise HTTPException(status_code=401, detail="Unauthorized")
-
-    # Handle empty token
-    if not cashu_token:
-        logger.error("Empty token provided")
-        raise HTTPException(
-            status_code=401,
-            detail={
-                "error": {
-                    "message": "API key or Cashu token required",
-                    "type": "invalid_request_error",
-                    "code": "missing_api_key",
-                }
-            },
-        )
-
-    # Handle regular API keys (sk-*)
-    if cashu_token.startswith("sk-"):
-        return
-
-    try:
-        token_obj = deserialize_token_from_string(cashu_token)
-    except Exception:
-        # Invalid token format - let the auth system handle it
-        raise HTTPException(
-            status_code=401,
-            detail="Invalid authentication token format",
-        )
-
-    amount_msat = (
-        token_obj.amount if token_obj.unit == "msat" else token_obj.amount * 1000
-    )
-
-    if max_cost_for_model > amount_msat:
-        raise HTTPException(
-            status_code=413,
-            detail={
-                "reason": "Insufficient balance",
-                "amount_required_msat": max_cost_for_model,
-                "model": body.get("model", "unknown"),
-                "type": "minimum_balance_required",
-            },
-        )
-
-
-async def get_max_cost_for_model(
-    model: str,
-    session: AsyncSession,
-    model_obj: Any | None = None,
-) -> int:
-    """Get the maximum cost for a specific model from providers with overrides."""
-    logger.debug(
-        "Getting max cost for model",
-        extra={
-            "model": model,
-            "fixed_pricing": settings.fixed_pricing,
-        },
-    )
-
-    if settings.fixed_pricing:
-        default_cost_msats = settings.fixed_cost_per_request * 1000
-        logger.debug(
-            "Using fixed cost pricing",
-            extra={"cost_msats": default_cost_msats, "model": model},
-        )
-        return max(settings.min_request_msat, default_cost_msats)
-
-    if not model_obj:
-        from ..proxy import get_model_instance
-
-        model_obj = get_model_instance(model)
-
-    if not model_obj:
-        fallback_msats = settings.fixed_cost_per_request * 1000
-        logger.warning(
-            "Model not found in providers or overrides",
-            extra={
-                "requested_model": model,
-                "using_default_cost": fallback_msats,
-            },
-        )
-        return max(settings.min_request_msat, fallback_msats)
-
-    if model_obj.sats_pricing:
-        try:
-            max_cost = (
-                model_obj.sats_pricing.max_cost
-                * 1000
-                * (1 - settings.tolerance_percentage / 100)
-            )
-            logger.debug(
-                "Found model-specific max cost",
-                extra={"model": model, "max_cost_msats": max_cost},
-            )
-            calculated_msats = int(max_cost)
-            return max(settings.min_request_msat, calculated_msats)
-        except Exception as e:
-            logger.error(
-                "Error calculating max cost from model pricing",
-                extra={"model": model, "error": str(e)},
-            )
-
-    logger.warning(
-        "Model pricing not found, using fixed cost",
-        extra={
-            "model": model,
-            "default_cost_msats": settings.fixed_cost_per_request * 1000,
-        },
-    )
-    return max(settings.min_request_msat, settings.fixed_cost_per_request * 1000)
-
-
-async def calculate_discounted_max_cost(
-    max_cost_for_model: int,
-    body: dict,
-    model_obj: Any | None = None,
-) -> int:
-    """Calculate the discounted max cost for a request using model pricing when available."""
-    if settings.fixed_pricing:
-        return max_cost_for_model
-
-    model = body.get("model", "unknown")
-
-    model_pricing = model_obj.sats_pricing if model_obj else None
-    if not model_pricing:
-        return max_cost_for_model
-
-    tol = settings.tolerance_percentage
-    tol_factor = max(0.0, 1 - float(tol) / 100.0)
-    max_prompt_allowed_sats = model_pricing.max_prompt_cost * tol_factor
-    max_completion_allowed_sats = model_pricing.max_completion_cost * tol_factor
-
-    adjusted = max_cost_for_model
-
-    if messages := body.get("messages"):
-        prompt_tokens = estimate_tokens(messages)
-
-        image_tokens = await estimate_image_tokens_in_messages(messages)
-        if image_tokens > 0:
-            logger.debug(
-                "Found images in request",
-                extra={
-                    "model": model,
-                    "image_tokens": image_tokens,
-                },
-            )
-            prompt_tokens += image_tokens
-
-        estimated_prompt_delta_sats = (
-            max_prompt_allowed_sats - prompt_tokens * model_pricing.prompt
-        )
-        if estimated_prompt_delta_sats > 0:
-            adjusted = adjusted - math.floor(estimated_prompt_delta_sats * 1000)
-
-    max_tokens_raw = body.get("max_tokens", None)
-    if max_tokens_raw is not None:
-        try:
-            max_tokens_int = int(max_tokens_raw)
-        except (TypeError, ValueError):
-            logger.warning(
-                "Invalid max_tokens; ignoring in cost adjustment",
-                extra={"max_tokens": str(max_tokens_raw)[:64], "model": model},
-            )
-        else:
-            estimated_completion_delta_sats = (
-                max_completion_allowed_sats - max_tokens_int * model_pricing.completion
-            )
-            if estimated_completion_delta_sats > 0:
-                adjusted = adjusted - math.floor(estimated_completion_delta_sats * 1000)
-
-    logger.debug(
-        "Discounted max cost computed",
-        extra={
-            "model": model,
-            "original_msats": max_cost_for_model,
-            "adjusted_msats": adjusted,
-            "tolerance_pct": tol,
-        },
-    )
-
-    return max(0, adjusted)
-
-
-def estimate_tokens(messages: list) -> int:
-    """Estimate tokens for text content, excluding image_url fields."""
-    total = 0
-    for msg in messages:
-        if isinstance(msg, dict):
-            content = msg.get("content")
-            if isinstance(content, str):
-                total += len(content)
-            elif isinstance(content, list):
-                total += sum(
-                    len(item.get("text", ""))
-                    for item in content
-                    if isinstance(item, dict) and item.get("type") == "text"
-                )
-    return total // 3
-
-
-def _get_image_dimensions(image_data: bytes) -> tuple[int, int]:
-    """Extract image dimensions from image bytes."""
-    try:
-        img = Image.open(BytesIO(image_data))
-        return img.size
-    except Exception as e:
-        logger.warning(
-            "Failed to get image dimensions, using default",
-            extra={"error": str(e)},
-        )
-        return (512, 512)
-
-
-async def _fetch_image_from_url(url: str) -> bytes | None:
-    """Fetch image from URL."""
-    try:
-        async with httpx.AsyncClient(timeout=10.0) as client:
-            response = await client.get(url)
-            response.raise_for_status()
-            return response.content
-    except Exception as e:
-        logger.warning(
-            "Failed to fetch image from URL",
-            extra={"error": str(e), "url": url[:100]},
-        )
-        return None
-
-
-def _calculate_image_tokens(width: int, height: int, detail: str = "auto") -> int:
-    """Calculate image tokens based on OpenAI's vision pricing.
-
-    For low detail: 85 tokens
-    For high detail/auto: 85 base tokens + 170 tokens per 512px tile
-    """
-    if detail == "low":
-        return 85
-
-    if width > 2048 or height > 2048:
-        aspect_ratio = width / height
-        if width > height:
-            width = 2048
-            height = int(width / aspect_ratio)
-        else:
-            height = 2048
-            width = int(height * aspect_ratio)
-
-    if width > 768 or height > 768:
-        aspect_ratio = width / height
-        if width > height:
-            width = 768
-            height = int(width / aspect_ratio)
-        else:
-            height = 768
-            width = int(height * aspect_ratio)
-
-    tiles_width = (width + 511) // 512
-    tiles_height = (height + 511) // 512
-    num_tiles = tiles_width * tiles_height
-
-    return 85 + (170 * num_tiles)
-
-
-async def estimate_image_tokens_in_messages(messages: list) -> int:
-    """Estimate total tokens for all images in messages.
-
-    Supports both base64 encoded images and image URLs.
-    """
-    total_image_tokens = 0
-
-    for message in messages:
-        if not isinstance(message, dict):
-            continue
-
-        content = message.get("content")
-        if not content:
-            continue
-
-        if isinstance(content, str):
-            continue
-
-        if not isinstance(content, list):
-            continue
-
-        for content_item in content:
-            if not isinstance(content_item, dict):
-                continue
-
-            content_type = content_item.get("type")
-            if content_type not in ("image_url", "input_image"):
-                continue
-
-            image_url_data = content_item.get("image_url")
-            if not image_url_data:
-                continue
-
-            if isinstance(image_url_data, str):
-                url = image_url_data
-                detail = "auto"
-            elif isinstance(image_url_data, dict):
-                url = image_url_data.get("url", "")
-                detail = image_url_data.get("detail", "auto")
-            else:
-                continue
-
-            if not url:
-                continue
-
-            if url.startswith("data:image/"):
-                try:
-                    header, base64_data = url.split(",", 1)
-                    image_bytes = base64.b64decode(base64_data)
-                    width, height = _get_image_dimensions(image_bytes)
-                    tokens = _calculate_image_tokens(width, height, detail)
-                    total_image_tokens += tokens
-                    logger.debug(
-                        "Calculated tokens for base64 image",
-                        extra={
-                            "width": width,
-                            "height": height,
-                            "detail": detail,
-                            "tokens": tokens,
-                        },
-                    )
-                except Exception as e:
-                    logger.warning(
-                        "Failed to process base64 image",
-                        extra={"error": str(e)},
-                    )
-                    total_image_tokens += 85
-            else:
-                image_bytes_or_none = await _fetch_image_from_url(url)
-                if image_bytes_or_none:
-                    width, height = _get_image_dimensions(image_bytes_or_none)
-                    tokens = _calculate_image_tokens(width, height, detail)
-                    total_image_tokens += tokens
-                    logger.debug(
-                        "Calculated tokens for URL image",
-                        extra={
-                            "url": url[:100],
-                            "width": width,
-                            "height": height,
-                            "detail": detail,
-                            "tokens": tokens,
-                        },
-                    )
-                else:
-                    total_image_tokens += 85
-
-    return total_image_tokens
-
-
+# TODO: remove and replace with custom HTTPException
 def create_error_response(
     error_type: str,
     message: str,
@@ -414,3 +36,380 @@ def create_error_response(
         media_type="application/json",
         headers={"X-Cashu": token} if token else {},
     )
+
+
+# Request payment handlers, todo: maybe mode somewhere else...
+
+
+async def pay_for_request(
+    key: TemporaryCredit, cost_per_request: int, session: AsyncSession
+) -> int:
+    """Process payment for a request."""
+
+    logger.info(
+        "Processing payment for request",
+        extra={
+            "key_hash": key.hashed_key[:8] + "...",
+            "current_balance": key.balance,
+            "required_cost": cost_per_request,
+            "sufficient_balance": key.balance >= cost_per_request,
+        },
+    )
+
+    if key.total_balance_msat < cost_per_request:
+        logger.warning(
+            "Insufficient balance for request",
+            extra={
+                "key_hash": key.hashed_key[:8] + "...",
+                "balance": key.balance,
+                "reserved_balance": key.reserved_balance,
+                "required": cost_per_request,
+                "shortfall": cost_per_request - key.total_balance_msat,
+            },
+        )
+
+        raise HTTPException(
+            status_code=402,
+            detail={
+                "error": {
+                    "message": f"Insufficient balance: {cost_per_request} mSats required. {key.total_balance} available. (reserved: {key.reserved_balance})",
+                    "type": "insufficient_quota",
+                    "code": "insufficient_balance",
+                }
+            },
+        )
+
+    logger.debug(
+        "Charging base cost for request",
+        extra={
+            "key_hash": key.hashed_key[:8] + "...",
+            "cost": cost_per_request,
+            "balance_before": key.balance,
+        },
+    )
+
+    # Charge the base cost for the request atomically to avoid race conditions
+    stmt = (
+        update(TemporaryCredit)
+        .where(col(TemporaryCredit.hashed_key) == key.hashed_key)
+        .where(
+            col(TemporaryCredit.balance) - col(TemporaryCredit.reserved_balance)
+            >= cost_per_request
+        )
+        .values(
+            reserved_balance=col(TemporaryCredit.reserved_balance) + cost_per_request
+        )
+    )
+    result = await session.exec(stmt)  # type: ignore[call-overload]
+    await session.commit()
+
+    if result.rowcount == 0:
+        logger.error(
+            "Concurrent request depleted balance",
+            extra={
+                "key_hash": key.hashed_key[:8] + "...",
+                "required_cost": cost_per_request,
+                "current_balance": key.balance,
+            },
+        )
+
+        # Another concurrent request spent the balance first
+        raise HTTPException(
+            status_code=402,
+            detail={
+                "error": {
+                    "message": f"Insufficient balance: {cost_per_request} mSats required. {key.balance} available.",
+                    "type": "insufficient_quota",
+                    "code": "insufficient_balance",
+                }
+            },
+        )
+
+    await session.refresh(key)
+
+    logger.info(
+        "Payment processed successfully",
+        extra={
+            "key_hash": key.hashed_key[:8] + "...",
+            "charged_amount": cost_per_request,
+            "new_balance": key.balance,
+        },
+    )
+
+    return cost_per_request
+
+
+async def revert_pay_for_request(
+    key: TemporaryCredit, session: AsyncSession, cost_per_request: int
+) -> None:
+    stmt = (
+        update(TemporaryCredit)
+        .where(col(TemporaryCredit.hashed_key) == key.hashed_key)
+        .values(
+            reserved_balance=col(TemporaryCredit.reserved_balance) - cost_per_request
+        )
+    )
+
+    result = await session.exec(stmt)  # type: ignore[call-overload]
+    await session.commit()
+    if result.rowcount == 0:
+        logger.error(
+            "Failed to revert payment - insufficient reserved balance",
+            extra={
+                "key_hash": key.hashed_key[:8] + "...",
+                "cost_to_revert": cost_per_request,
+                "current_reserved_balance": key.reserved_balance,
+            },
+        )
+        raise HTTPException(
+            status_code=402,
+            detail={
+                "error": {
+                    "message": f"failed to revert request payment: {cost_per_request} mSats required. {key.balance} available.",
+                    "type": "payment_error",
+                    "code": "payment_error",
+                }
+            },
+        )
+    await session.refresh(key)
+
+
+async def adjust_payment_for_tokens(
+    key: TemporaryCredit,
+    response_data: dict,
+    session: AsyncSession,
+    deducted_max_cost: int,
+) -> dict:
+    """
+    Adjusts the payment based on token usage in the response.
+    This is called after the initial payment and the upstream request is complete.
+    Returns cost data to be included in the response.
+    """
+    model = response_data.get("model", "unknown")
+
+    logger.debug(
+        "Starting payment adjustment for tokens",
+        extra={
+            "key_hash": key.hashed_key[:8] + "...",
+            "model": model,
+            "deducted_max_cost": deducted_max_cost,
+            "current_balance": key.balance,
+            "has_usage": "usage" in response_data,
+        },
+    )
+
+    match calculate_cost(response_data, deducted_max_cost):
+        case MaxCostData() as cost:
+            logger.debug(
+                "Using max cost data (no token adjustment)",
+                extra={
+                    "key_hash": key.hashed_key[:8] + "...",
+                    "model": model,
+                    "max_cost": cost.total_msats,
+                },
+            )
+            # Finalize by releasing reservation and charging max cost
+            finalize_stmt = (
+                update(TemporaryCredit)
+                .where(col(TemporaryCredit.hashed_key) == key.hashed_key)
+                .values(
+                    reserved_balance=col(TemporaryCredit.reserved_balance)
+                    - deducted_max_cost,
+                    balance=col(TemporaryCredit.balance) - cost.total_msats,
+                )
+            )
+            result = await session.exec(finalize_stmt)  # type: ignore[call-overload]
+            await session.commit()
+            if result.rowcount == 0:
+                logger.error(
+                    "Failed to finalize max-cost payment - insufficient reserved balance",
+                    extra={
+                        "key_hash": key.hashed_key[:8] + "...",
+                        "deducted_max_cost": deducted_max_cost,
+                        "current_reserved_balance": key.reserved_balance,
+                        "total_cost": cost.total_msats,
+                        "model": model,
+                    },
+                )
+            else:
+                await session.refresh(key)
+                logger.info(
+                    "Max cost payment finalized",
+                    extra={
+                        "key_hash": key.hashed_key[:8] + "...",
+                        "charged_amount": cost.total_msats,
+                        "new_balance": key.balance,
+                        "model": model,
+                    },
+                )
+            return cost.dict()
+
+        case CostData() as cost:
+            # If token-based pricing is enabled and base cost is 0, use token-based cost
+            # Otherwise, token cost is additional to the base cost
+            cost_difference = cost.total_msats - deducted_max_cost
+            total_cost_msats: int = math.ceil(cost.total_msats)
+
+            logger.info(
+                "Calculated token-based cost",
+                extra={
+                    "key_hash": key.hashed_key[:8] + "...",
+                    "model": model,
+                    "token_cost": cost.total_msats,
+                    "deducted_max_cost": deducted_max_cost,
+                    "cost_difference": cost_difference,
+                    "input_msats": cost.input_msats,
+                    "output_msats": cost.output_msats,
+                },
+            )
+
+            if cost_difference == 0:
+                logger.debug(
+                    "Finalizing with exact reserved cost",
+                    extra={"key_hash": key.hashed_key[:8] + "...", "model": model},
+                )
+                finalize_stmt = (
+                    update(TemporaryCredit)
+                    .where(col(TemporaryCredit.hashed_key) == key.hashed_key)
+                    .values(
+                        reserved_balance=col(TemporaryCredit.reserved_balance)
+                        - deducted_max_cost,
+                        balance=col(TemporaryCredit.balance) - total_cost_msats,
+                    )
+                )
+                await session.exec(finalize_stmt)  # type: ignore[call-overload]
+                await session.commit()
+                await session.refresh(key)
+                return cost.dict()
+
+            # this should never happen why do we handle this???
+            if cost_difference > 0:
+                # Need to charge more than reserved, finalize by releasing reservation and charging total
+                logger.info(
+                    "Additional charge required for token usage",
+                    extra={
+                        "key_hash": key.hashed_key[:8] + "...",
+                        "additional_charge": cost_difference,
+                        "current_balance": key.balance,
+                        "sufficient_balance": key.balance >= cost_difference,
+                        "model": model,
+                    },
+                )
+
+                finalize_stmt = (
+                    update(TemporaryCredit)
+                    .where(col(TemporaryCredit.hashed_key) == key.hashed_key)
+                    .values(
+                        reserved_balance=col(TemporaryCredit.reserved_balance)
+                        - deducted_max_cost,
+                        balance=col(TemporaryCredit.balance) - total_cost_msats,
+                    )
+                )
+                result = await session.exec(finalize_stmt)  # type: ignore[call-overload]
+                await session.commit()
+
+                if result.rowcount:
+                    cost.total_msats = total_cost_msats
+                    await session.refresh(key)
+
+                    logger.info(
+                        "Finalized payment with additional charge",
+                        extra={
+                            "key_hash": key.hashed_key[:8] + "...",
+                            "charged_amount": total_cost_msats,
+                            "new_balance": key.balance,
+                            "model": model,
+                        },
+                    )
+                else:
+                    logger.warning(
+                        "Failed to finalize additional charge (concurrent operation)",
+                        extra={
+                            "key_hash": key.hashed_key[:8] + "...",
+                            "attempted_charge": total_cost_msats,
+                            "model": model,
+                        },
+                    )
+            else:
+                # Refund some of the base cost
+                refund = abs(cost_difference)
+                logger.info(
+                    "Refunding excess payment",
+                    extra={
+                        "key_hash": key.hashed_key[:8] + "...",
+                        "refund_amount": refund,
+                        "current_balance": key.balance,
+                        "model": model,
+                    },
+                )
+
+                refund_stmt = (
+                    update(TemporaryCredit)
+                    .where(col(TemporaryCredit.hashed_key) == key.hashed_key)
+                    .values(
+                        reserved_balance=col(TemporaryCredit.reserved_balance)
+                        - deducted_max_cost,
+                        balance=col(TemporaryCredit.balance) - total_cost_msats,
+                    )
+                )
+                result = await session.exec(refund_stmt)  # type: ignore[call-overload]
+                await session.commit()
+
+                if result.rowcount == 0:
+                    logger.error(
+                        "Failed to finalize payment - insufficient reserved balance",
+                        extra={
+                            "key_hash": key.hashed_key[:8] + "...",
+                            "deducted_max_cost": deducted_max_cost,
+                            "current_reserved_balance": key.reserved_balance,
+                            "total_cost": total_cost_msats,
+                            "model": model,
+                        },
+                    )
+                    # Still return the cost data even if we couldn't properly finalize
+                    # The reservation was already made, so the user has paid
+
+                cost.total_msats = total_cost_msats
+                await session.refresh(key)
+
+                logger.info(
+                    "Refund processed successfully",
+                    extra={
+                        "key_hash": key.hashed_key[:8] + "...",
+                        "refunded_amount": refund,
+                        "new_balance": key.balance,
+                        "final_cost": cost.total_msats,
+                        "model": model,
+                    },
+                )
+
+            return cost.dict()
+
+        case CostDataError() as error:
+            logger.error(
+                "Cost calculation error during payment adjustment",
+                extra={
+                    "key_hash": key.hashed_key[:8] + "...",
+                    "model": model,
+                    "error_message": error.message,
+                    "error_code": error.code,
+                },
+            )
+
+            raise HTTPException(
+                status_code=400,
+                detail={
+                    "error": {
+                        "message": error.message,
+                        "type": "invalid_request_error",
+                        "code": error.code,
+                    }
+                },
+            )
+    # Fallback return to satisfy type checker; execution should not reach here
+    return {
+        "base_msats": deducted_max_cost,
+        "input_msats": 0,
+        "output_msats": 0,
+        "total_msats": deducted_max_cost,
+    }

routstr/payment/lnurl.py

@@ -305,3 +305,12 @@ async def raw_send_to_lnurl(
         quote_id=melt_quote_resp.quote,
     )
     return final_amount
+
+
+async def send_to_lnurl(amount: int, unit: str, mint: str, address: str) -> int:
+    from .wallet import get_wallet
+
+    wallet = await get_wallet(mint, unit)
+    proofs = wallet._get_proofs_per_keyset(wallet.proofs)[wallet.keyset_id]
+    proofs, _ = await wallet.select_to_send(proofs, amount, set_reserved=True)
+    return await raw_send_to_lnurl(wallet, proofs, address, unit)

routstr/payment/models.py

@@ -1,724 +0,0 @@
-import asyncio
-import json
-import random
-from pathlib import Path
-from typing import Final
-from urllib.request import urlopen
-
-import httpx
-from fastapi import APIRouter, Depends
-from pydantic.v1 import BaseModel
-from sqlmodel import select
-from sqlmodel.ext.asyncio.session import AsyncSession
-
-from ..core.db import ModelRow, create_session, get_session
-from ..core.logging import get_logger
-from ..core.settings import settings
-from .price import sats_usd_price
-
-logger = get_logger(__name__)
-
-models_router = APIRouter()
-
-DEFAULT_EXCLUDED_MODEL_IDS: Final[set[str]] = {
-    "openrouter/auto",
-    "openrouter/bodybuilder",
-    "google/gemini-2.5-pro-exp-03-25",
-    "opengvlab/internvl3-78b",
-    "openrouter/sonoma-dusk-alpha",
-    "openrouter/sonoma-sky-alpha",
-}
-
-
-class Architecture(BaseModel):
-    modality: str
-    input_modalities: list[str]
-    output_modalities: list[str]
-    tokenizer: str
-    instruct_type: str | None
-
-
-class Pricing(BaseModel):
-    prompt: float
-    completion: float
-    request: float = 0.0
-    image: float = 0.0
-    web_search: float = 0.0
-    internal_reasoning: float = 0.0
-    max_prompt_cost: float = 0.0  # in sats not msats
-    max_completion_cost: float = 0.0  # in sats not msats
-    max_cost: float = 0.0  # in sats not msats
-
-
-class TopProvider(BaseModel):
-    context_length: int | None = None
-    max_completion_tokens: int | None = None
-    is_moderated: bool | None = None
-
-
-class Model(BaseModel):
-    id: str
-    name: str
-    created: int
-    description: str
-    context_length: int
-    architecture: Architecture
-    pricing: Pricing
-    sats_pricing: Pricing | None = None
-    per_request_limits: dict | None = None
-    top_provider: TopProvider | None = None
-    enabled: bool = True
-    upstream_provider_id: int | str | None = None
-    canonical_slug: str | None = None
-    alias_ids: list[str] | None = None
-
-    def __hash__(self) -> int:
-        return hash(self.id)
-
-
-def fetch_openrouter_models(source_filter: str | None = None) -> list[dict]:
-    """Fetches model information from OpenRouter API."""
-    base_url = "https://openrouter.ai/api/v1"
-
-    try:
-        with urlopen(f"{base_url}/models") as response:
-            data = json.loads(response.read().decode("utf-8"))
-
-            models_data: list[dict] = []
-            for model in data.get("data", []):
-                model_id = model.get("id", "")
-
-                if source_filter:
-                    source_prefix = f"{source_filter}/"
-                    if not model_id.startswith(source_prefix):
-                        continue
-
-                    model = dict(model)
-                    model["id"] = model_id[len(source_prefix) :]
-                    model_id = model["id"]
-
-                if (
-                    "(free)" in model.get("name", "")
-                    or model_id in DEFAULT_EXCLUDED_MODEL_IDS
-                ):
-                    continue
-
-                models_data.append(model)
-
-            return models_data
-    except Exception as e:
-        logger.error(f"Error fetching models from OpenRouter API: {e}")
-        return []
-
-
-async def async_fetch_openrouter_models(source_filter: str | None = None) -> list[dict]:
-    """Asynchronously fetch model information from OpenRouter API."""
-    base_url = "https://openrouter.ai/api/v1"
-
-    try:
-        async with httpx.AsyncClient() as client:
-            response = await client.get(f"{base_url}/models", timeout=30)
-            response.raise_for_status()
-            data = response.json()
-
-            models_data: list[dict] = []
-            for model in data.get("data", []):
-                model_id = model.get("id", "")
-
-                if source_filter:
-                    source_prefix = f"{source_filter}/"
-                    if not model_id.startswith(source_prefix):
-                        continue
-
-                    model = dict(model)
-                    model["id"] = model_id[len(source_prefix) :]
-                    model_id = model["id"]
-
-                if (
-                    "(free)" in model.get("name", "")
-                    or model_id in DEFAULT_EXCLUDED_MODEL_IDS
-                ):
-                    continue
-
-                models_data.append(model)
-
-            return models_data
-    except Exception as e:
-        logger.error(f"Error (async) fetching models from OpenRouter API: {e}")
-        return []
-
-
-def is_openrouter_upstream() -> bool:
-    try:
-        base = (settings.upstream_base_url or "").strip().rstrip("/")
-    except Exception:
-        return False
-    return base.lower() == "https://openrouter.ai/api/v1"
-
-
-def load_models() -> list[Model]:
-    """Load model definitions from a JSON file or auto-generate from OpenRouter API.
-
-    The file path can be specified via the ``MODELS_PATH`` environment variable.
-    If a user-provided models.json exists, it will be used. Otherwise, models are
-    automatically fetched from OpenRouter API in memory. If the example file exists
-    and no user file is provided, it will be used as a fallback.
-    """
-
-    try:
-        models_path = Path(settings.models_path)
-    except Exception:
-        models_path = Path("models.json")
-
-    # Check if user has actively provided a models.json file
-    if models_path.exists():
-        logger.info(f"Loading models from user-provided file: {models_path}")
-        try:
-            with models_path.open("r") as f:
-                data = json.load(f)
-            return [Model(**model) for model in data.get("models", [])]  # type: ignore
-        except Exception as e:
-            logger.error(f"Error loading models from {models_path}: {e}")
-            # Fall through to auto-generation
-
-    # Only auto-generate from OpenRouter when upstream is OpenRouter
-    if not is_openrouter_upstream():
-        logger.info(
-            "Skipping auto-generation from OpenRouter because upstream_base_url is not https://openrouter.ai/api/v1"
-        )
-        return []
-
-    logger.info("Auto-generating models from OpenRouter API")
-    try:
-        source_filter = settings.source or None
-    except Exception:
-        source_filter = None
-    source_filter = source_filter if source_filter and source_filter.strip() else None
-
-    models_data = fetch_openrouter_models(source_filter=source_filter)
-    if not models_data:
-        logger.error("Failed to fetch models from OpenRouter API")
-        return []
-
-    logger.info(f"Successfully fetched {len(models_data)} models from OpenRouter API")
-    return [Model(**model) for model in models_data]  # type: ignore
-
-
-def _row_to_model(
-    row: ModelRow, apply_provider_fee: bool = False, provider_fee: float = 1.01
-) -> Model:
-    architecture = json.loads(row.architecture)
-    pricing = json.loads(row.pricing)
-    per_request_limits = (
-        json.loads(row.per_request_limits) if row.per_request_limits else None
-    )
-    top_provider_dict = json.loads(row.top_provider) if row.top_provider else None
-
-    if apply_provider_fee and isinstance(pricing, dict):
-        pricing = {k: float(v) * provider_fee for k, v in pricing.items()}
-
-    if isinstance(pricing, dict) and float(pricing.get("request", 0.0)) <= 0.0:
-        pricing["request"] = max(pricing.get("request", 0.0), 0.0)
-
-    parsed_pricing = Pricing.parse_obj(pricing)
-    model = Model(
-        id=row.id,
-        name=row.name,
-        created=row.created,
-        description=row.description,
-        context_length=row.context_length,
-        architecture=Architecture.parse_obj(architecture),
-        pricing=parsed_pricing,
-        sats_pricing=None,
-        per_request_limits=per_request_limits,
-        top_provider=TopProvider.parse_obj(top_provider_dict)
-        if top_provider_dict
-        else None,
-        enabled=row.enabled,
-        upstream_provider_id=row.upstream_provider_id,
-        canonical_slug=getattr(row, "canonical_slug", None),
-    )
-
-    if apply_provider_fee:
-        (
-            parsed_pricing.max_prompt_cost,
-            parsed_pricing.max_completion_cost,
-            parsed_pricing.max_cost,
-        ) = _calculate_usd_max_costs(model)
-
-    try:
-        sats_to_usd = sats_usd_price()
-        model = _update_model_sats_pricing(model, sats_to_usd)
-    except Exception as e:
-        logger.warning(f"Could not calculate sats pricing: {e}")
-
-    return model
-
-
-def _model_to_row_payload(model: Model) -> dict[str, str | int | bool | None]:
-    return {
-        "id": model.id,
-        "name": model.name,
-        "created": model.created,
-        "description": model.description,
-        "context_length": model.context_length,
-        "architecture": json.dumps(model.architecture.dict()),
-        "pricing": json.dumps(model.pricing.dict()),
-        "sats_pricing": json.dumps(model.sats_pricing.dict())
-        if model.sats_pricing
-        else None,
-        "per_request_limits": json.dumps(model.per_request_limits)
-        if model.per_request_limits is not None
-        else None,
-        "top_provider": json.dumps(model.top_provider.dict())
-        if model.top_provider is not None
-        else None,
-        "enabled": model.enabled,
-        "upstream_provider_id": model.upstream_provider_id,
-    }
-
-
-async def list_models(
-    session: AsyncSession,
-    upstream_id: int,
-    include_disabled: bool = False,
-) -> list[Model]:
-    from sqlmodel import select
-
-    from ..core.db import UpstreamProviderRow
-
-    query = select(ModelRow)
-    if upstream_id is not None:
-        query = query.where(ModelRow.upstream_provider_id == upstream_id)
-    if not include_disabled:
-        query = query.where(ModelRow.enabled)
-
-    rows = (await session.exec(query)).all()  # type: ignore
-    provider_result = await session.exec(select(UpstreamProviderRow))
-    providers_by_id = {p.id: p for p in provider_result.all()}
-    return [
-        _row_to_model(
-            r,
-            apply_provider_fee=True,
-            provider_fee=providers_by_id[r.upstream_provider_id].provider_fee
-            if r.upstream_provider_id in providers_by_id
-            else 1.01,
-        )
-        for r in rows
-    ]
-
-
-async def get_model_by_id(
-    model_id: str, provider_id: int, session: AsyncSession
-) -> Model | None:
-    from ..core.db import UpstreamProviderRow
-
-    row = await session.get(ModelRow, (model_id, provider_id))
-    if not row or not row.enabled:
-        return None
-    provider = await session.get(UpstreamProviderRow, provider_id)
-    provider_fee = provider.provider_fee if provider else 1.01
-    return _row_to_model(row, apply_provider_fee=True, provider_fee=provider_fee)
-
-
-def _calculate_usd_max_costs(model: Model) -> tuple[float, float, float]:
-    """Calculate max costs in USD based on model context/token limits.
-
-    Args:
-        model: Model object
-
-    Returns:
-        Tuple of (max_prompt_cost, max_completion_cost, max_cost) in USD
-    """
-    min_req_msat = max(1, int(getattr(settings, "min_request_msat", 1)))
-    min_req_usd = float(min_req_msat) / 1_000_000.0
-
-    prompt_price = model.pricing.prompt
-    completion_price = model.pricing.completion
-
-    if model.top_provider and (
-        model.top_provider.context_length or model.top_provider.max_completion_tokens
-    ):
-        if (cl := model.top_provider.context_length) and (
-            mct := model.top_provider.max_completion_tokens
-        ):
-            if cl <= mct:
-                return (
-                    cl * prompt_price,
-                    cl * completion_price,
-                    cl * max(completion_price, prompt_price),
-                )
-            return (
-                cl * prompt_price,
-                mct * completion_price,
-                (cl - mct) * prompt_price + mct * completion_price,
-            )
-        elif cl := model.top_provider.context_length:
-            return (
-                cl * prompt_price,
-                cl * completion_price,
-                cl * max(completion_price, prompt_price),
-            )
-        elif mct := model.top_provider.max_completion_tokens:
-            return (
-                mct * prompt_price,
-                mct * completion_price,
-                mct * completion_price,
-            )
-    elif model.context_length:
-        return (
-            model.context_length * prompt_price,
-            model.context_length * completion_price,
-            model.context_length * max(completion_price, prompt_price),
-        )
-
-    p = prompt_price * 1_000_000
-    c = completion_price * 32_000
-    r = model.pricing.request * 100_000
-    i = model.pricing.image * 100
-    w = model.pricing.web_search * 1000
-    ir = model.pricing.internal_reasoning * 100
-    return (p, c, max(p + c + r + i + w + ir, min_req_usd))
-
-
-def _update_model_sats_pricing(model: Model, sats_to_usd: float) -> Model:
-    """Update a model's sats_pricing based on USD pricing and exchange rate.
-
-    Args:
-        model: Model object to update
-        sats_to_usd: Current sats to USD exchange rate
-
-    Returns:
-        Updated Model object with new sats_pricing
-    """
-    try:
-        min_req_msat = max(1, int(getattr(settings, "min_request_msat", 1)))
-        min_req_sats = float(min_req_msat) / 1000.0
-
-        sats = Pricing.parse_obj(
-            {k: v / sats_to_usd for k, v in model.pricing.dict().items()}
-        )
-
-        if sats.request <= 0.0:
-            sats.request = min_req_sats
-        if (sats.max_cost or 0.0) < min_req_sats:
-            sats.max_cost = min_req_sats
-
-        return Model(
-            id=model.id,
-            name=model.name,
-            created=model.created,
-            description=model.description,
-            context_length=model.context_length,
-            architecture=model.architecture,
-            pricing=model.pricing,
-            sats_pricing=sats,
-            per_request_limits=model.per_request_limits,
-            top_provider=model.top_provider,
-            enabled=model.enabled,
-            upstream_provider_id=model.upstream_provider_id,
-            canonical_slug=model.canonical_slug,
-            alias_ids=model.alias_ids,
-        )
-    except Exception as e:
-        logger.error(
-            "Failed to update sats pricing for model",
-            extra={
-                "model_id": model.id,
-                "error": str(e),
-                "error_type": type(e).__name__,
-            },
-        )
-        return model
-
-
-async def ensure_models_bootstrapped() -> None:
-    async with create_session() as s:
-        existing = (await s.exec(select(ModelRow.id).limit(1))).all()  # type: ignore
-        if existing:
-            return
-
-        try:
-            models_path = Path(settings.models_path)
-        except Exception:
-            models_path = Path("models.json")
-
-        models_to_insert: list[dict] = []
-        if models_path.exists():
-            try:
-                with models_path.open("r") as f:
-                    data = json.load(f)
-                models_to_insert = data.get("models", [])
-                logger.info(
-                    f"Bootstrapping {len(models_to_insert)} models from {models_path}"
-                )
-            except Exception as e:
-                logger.error(f"Error loading models from {models_path}: {e}")
-
-        if not models_to_insert and is_openrouter_upstream():
-            logger.info("Bootstrapping models from OpenRouter API")
-            source_filter = None
-            try:
-                src = settings.source or None
-                source_filter = src if src and src.strip() else None
-            except Exception:
-                pass
-            models_to_insert = fetch_openrouter_models(source_filter=source_filter)
-        elif not models_to_insert:
-            logger.info(
-                "No models.json found and upstream is not OpenRouter; skipping bootstrap"
-            )
-
-        for m in models_to_insert:
-            try:
-                model = Model(**m)  # type: ignore
-            except Exception:
-                # Some OpenRouter models include extra fields; only map required ones
-                continue
-            exists = await s.get(ModelRow, model.id)
-            if exists:
-                continue
-            payload = _model_to_row_payload(model)
-            s.add(ModelRow(**payload))  # type: ignore
-        await s.commit()
-
-
-async def _update_sats_pricing_once() -> None:
-    """Update sats pricing once for all provider models (in-memory only)."""
-    from ..proxy import get_upstreams
-
-    upstreams = get_upstreams()
-    sats_to_usd = sats_usd_price()
-
-    updated_count = 0
-    for upstream in upstreams:
-        updated_models = [
-            _update_model_sats_pricing(m, sats_to_usd)
-            for m in upstream.get_cached_models()
-        ]
-        upstream._models_cache = updated_models
-        upstream._models_by_id = {m.id: m for m in updated_models}
-        updated_count += len(updated_models)
-
-    if updated_count > 0:
-        logger.info("Updated sats pricing", extra={"models_updated": updated_count})
-
-
-async def update_sats_pricing() -> None:
-    """Periodically update sats pricing for all provider models and database overrides."""
-    try:
-        if not settings.enable_pricing_refresh:
-            return
-    except Exception:
-        pass
-
-    await _update_sats_pricing_once()
-
-    while True:
-        try:
-            interval = getattr(settings, "pricing_refresh_interval_seconds", 120)
-            jitter = max(0.0, float(interval) * 0.1)
-            await asyncio.sleep(interval + random.uniform(0, jitter))
-        except asyncio.CancelledError:
-            break
-
-        try:
-            try:
-                if not settings.enable_pricing_refresh:
-                    return
-            except Exception:
-                pass
-
-            await _update_sats_pricing_once()
-        except asyncio.CancelledError:
-            break
-        except Exception as e:
-            logger.error(f"Error updating sats pricing: {e}")
-
-
-async def cleanup_enabled_models_periodically() -> None:
-    """Background task to clean up enabled models that match upstream pricing.
-
-    When model is enabled (enabled=True), remove it from DB if it matches upstream pricing.
-    Keep it in DB only if pricing differs from upstream or if it's disabled.
-    """
-    interval = getattr(
-        settings, "models_cleanup_interval_seconds", 300
-    )  # 5 minutes default
-    if not interval or interval <= 0:
-        return
-
-    while True:
-        try:
-            await _cleanup_enabled_models_once()
-        except asyncio.CancelledError:
-            break
-        except Exception as e:
-            logger.error(
-                "Error during enabled models cleanup",
-                extra={"error": str(e), "error_type": type(e).__name__},
-            )
-
-        try:
-            jitter = max(0.0, float(interval) * 0.1)
-            await asyncio.sleep(interval + random.uniform(0, jitter))
-        except asyncio.CancelledError:
-            break
-
-
-async def _cleanup_enabled_models_once() -> None:
-    """Clean up enabled models that match upstream pricing."""
-    from ..proxy import get_upstreams
-
-    async with create_session() as session:
-        # Get all enabled models from DB
-        result = await session.exec(
-            select(ModelRow).where(
-                ModelRow.enabled,  # Only enabled models
-            )
-        )
-        db_models = result.all()
-
-        if not db_models:
-            return
-
-        upstreams = get_upstreams()
-        models_to_remove = []
-
-        for db_model in db_models:
-            # Find corresponding upstream model
-            upstream_model = None
-            for upstream in upstreams:
-                upstream_model = upstream.get_cached_model_by_id(db_model.id)
-                if upstream_model:
-                    break
-
-            if not upstream_model:
-                continue
-
-            # Compare pricing to see if they match
-            db_pricing = json.loads(db_model.pricing)
-            upstream_pricing = upstream_model.pricing.dict()
-
-            # Check if pricing matches (with small tolerance for float comparison)
-            pricing_matches = _pricing_matches(db_pricing, upstream_pricing)
-
-            if pricing_matches:
-                models_to_remove.append(db_model)
-                logger.info(
-                    f"Removing enabled model {db_model.id} - matches upstream pricing",
-                    extra={"model_id": db_model.id},
-                )
-
-        # Remove models that match upstream pricing
-        for model in models_to_remove:
-            await session.delete(model)
-
-        if models_to_remove:
-            await session.commit()
-            logger.info(
-                f"Cleaned up {len(models_to_remove)} enabled models that match upstream pricing"
-            )
-
-
-def _pricing_matches(
-    db_pricing: dict, upstream_pricing: dict, tolerance: float = 0.0
-) -> bool:
-    """Check if pricing dictionaries match within tolerance."""
-    keys_to_compare = [
-        "prompt",
-        "completion",
-        "request",
-        "image",
-        "web_search",
-        "internal_reasoning",
-    ]
-
-    for key in keys_to_compare:
-        db_val = int(float(db_pricing.get(key, 0.0)) * 1000000)
-        upstream_val = int(float(upstream_pricing.get(key, 0.0)) * 1000000)
-
-        if abs(db_val - upstream_val) > tolerance:
-            return False
-
-    return True
-
-
-async def refresh_models_periodically() -> None:
-    """Background task: periodically fetch OpenRouter models and insert new ones.
-
-    - Respects optional SOURCE filter from settings
-    - Does not overwrite existing rows
-    - Sleeps according to settings.models_refresh_interval_seconds; disabled when 0
-    """
-    interval = getattr(settings, "models_refresh_interval_seconds", 0)
-    if not interval or interval <= 0:
-        return
-
-    # Only refresh from OpenRouter when upstream is OpenRouter
-    if not is_openrouter_upstream():
-        logger.info("Skipping models refresh: upstream_base_url is not OpenRouter")
-        return
-
-    while True:
-        try:
-            try:
-                if not settings.enable_models_refresh:
-                    return
-            except Exception:
-                pass
-            try:
-                src = settings.source or None
-                source_filter = src if src and src.strip() else None
-            except Exception:
-                source_filter = None
-
-            models = fetch_openrouter_models(source_filter=source_filter)
-            if not models:
-                await asyncio.sleep(interval)
-                continue
-
-            async with create_session() as s:
-                result = await s.exec(select(ModelRow.id))  # type: ignore
-                existing_ids = {
-                    row[0] if isinstance(row, tuple) else row for row in result.all()
-                }
-                inserted = 0
-                for m in models:
-                    try:
-                        model = Model(**m)  # type: ignore
-                    except Exception:
-                        continue
-                    if model.id in existing_ids:
-                        continue
-                    payload = _model_to_row_payload(model)
-                    try:
-                        s.add(ModelRow(**payload))  # type: ignore
-                    except Exception:
-                        pass
-                    inserted += 1
-                if inserted:
-                    await s.commit()
-                    logger.info(f"Inserted {inserted} new models from OpenRouter")
-        except asyncio.CancelledError:
-            break
-        except Exception as e:
-            logger.error(
-                "Error during models refresh",
-                extra={"error": str(e), "error_type": type(e).__name__},
-            )
-        try:
-            jitter = max(0.0, float(interval) * 0.1)
-            await asyncio.sleep(interval + random.uniform(0, jitter))
-        except asyncio.CancelledError:
-            break
-
-
-@models_router.get("/v1/models")
-@models_router.get("/models", include_in_schema=False)
-async def models(session: AsyncSession = Depends(get_session)) -> dict:
-    """Get all available models from all providers with database overrides applied."""
-    from ..proxy import get_unique_models
-
-    items = get_unique_models()
-    return {"data": items}

routstr/payment/price.py

@@ -110,6 +110,10 @@ async def _update_prices() -> None:
         return
     BTC_USD_PRICE = btc_price
     SATS_USD_PRICE = btc_price / 100_000_000
+    logger.info(
+        "Updated BTC/USD price",
+        extra={"btc_usd": btc_price, "sats_usd": SATS_USD_PRICE},
+    )
 
 
 def btc_usd_price() -> float:

routstr/payment/wallet.py

@@ -7,9 +7,9 @@ from cashu.wallet.helpers import deserialize_token_from_string
 from cashu.wallet.wallet import Wallet
 from sqlmodel import col, update
 
-from .core import db, get_logger
-from .core.settings import settings
-from .payment.lnurl import raw_send_to_lnurl
+from ..core import db, get_logger
+from ..core.settings import settings
+from ..payment.lnurl import raw_send_to_lnurl
 
 logger = get_logger(__name__)
 
@@ -19,9 +19,7 @@ async def get_balance(unit: str) -> int:
     return wallet.available_balance.amount
 
 
-async def recieve_token(
-    token: str,
-) -> tuple[int, str, str]:  # amount, unit, mint_url
+async def recieve_token(token: str) -> tuple[int, str, str]:  # amount, unit, mint_url
     token_obj = deserialize_token_from_string(token)
     if len(token_obj.keysets) > 1:
         raise ValueError("Multiple keysets per token currently not supported")
@@ -104,7 +102,7 @@ async def swap_to_primary_mint(
 
 
 async def credit_balance(
-    cashu_token: str, key: db.ApiKey, session: db.AsyncSession
+    cashu_token: str, credit: db.TemporaryCredit, session: db.AsyncSession
 ) -> int:
     logger.info(
         "credit_balance: Starting token redemption",
@@ -126,22 +124,22 @@ async def credit_balance(
 
         logger.info(
             "credit_balance: Updating balance",
-            extra={"old_balance": key.balance, "credit_amount": amount},
+            extra={"old_balance": credit.balance, "credit_amount": amount},
         )
 
         # Use atomic SQL UPDATE to prevent race conditions during concurrent topups
         stmt = (
-            update(db.ApiKey)
-            .where(col(db.ApiKey.hashed_key) == key.hashed_key)
-            .values(balance=(db.ApiKey.balance) + amount)
+            update(db.TemporaryCredit)
+            .where(col(db.TemporaryCredit.hashed_key) == credit.hashed_key)
+            .values(balance=(db.TemporaryCredit.balance) + amount)
         )
         await session.exec(stmt)  # type: ignore[call-overload]
         await session.commit()
-        await session.refresh(key)
+        await session.refresh(credit)
 
         logger.info(
             "credit_balance: Balance updated successfully",
-            extra={"new_balance": key.balance},
+            extra={"new_balance": credit.balance},
         )
 
         logger.info(
@@ -351,34 +349,3 @@ async def periodic_payout() -> None:
                 f"Error sending payout: {type(e).__name__}",
                 extra={"error": str(e)},
             )
-
-
-async def send_to_lnurl(amount: int, unit: str, mint: str, address: str) -> int:
-    wallet = await get_wallet(mint, unit)
-    proofs = wallet._get_proofs_per_keyset(wallet.proofs)[wallet.keyset_id]
-    proofs, _ = await wallet.select_to_send(proofs, amount, set_reserved=True)
-    return await raw_send_to_lnurl(wallet, proofs, address, unit)
-
-
-# class Payment:
-#     """
-#     Stores all cashu payment related data
-#     """
-
-#     def __init__(self, token: str) -> None:
-#         self.initial_token = token
-#         amount, unit, mint_url = self.parse_token(token)
-#         self.amount = amount
-#         self.unit = unit
-#         self.mint_url = mint_url
-
-#         self.claimed_proofs = redeem_to_proofs(token)
-
-#     def parse_token(self, token: str) -> tuple[int, CurrencyUnit, str]:
-#         raise NotImplementedError
-
-#     def refund_full(self) -> None:
-#         raise NotImplementedError
-
-#     def refund_partial(self, amount: int) -> None:
-#         raise NotImplementedError

routstr/proxy.py

@@ -5,24 +5,20 @@ from fastapi import APIRouter, Depends, HTTPException, Request
 from fastapi.responses import Response, StreamingResponse
 from sqlmodel import select
 
-from .algorithm import create_model_mappings
-from .auth import pay_for_request, revert_pay_for_request, validate_bearer_key
+from .auth import get_credit
 from .core import get_logger
 from .core.db import (
-    ApiKey,
     AsyncSession,
     ModelRow,
     UpstreamProviderRow,
     create_session,
     get_session,
 )
-from .payment.helpers import (
-    calculate_discounted_max_cost,
-    check_token_balance,
-    create_error_response,
-    get_max_cost_for_model,
-)
-from .payment.models import Model
+from .models.algorithm import create_model_mappings
+from .models.models import Model
+from .payment.cashu import pre_check_header_token_balance
+from .payment.cost import calculate_discounted_max_cost, get_max_cost_for_model
+from .payment.helpers import pay_for_request, revert_pay_for_request
 from .upstream import BaseUpstreamProvider
 from .upstream.helpers import init_upstreams
 
@@ -130,18 +126,21 @@ async def refresh_model_maps_periodically() -> None:
             )
 
 
-@proxy_router.api_route("/{path:path}", methods=["GET", "POST"], response_model=None)
-async def proxy(
-    request: Request, path: str, session: AsyncSession = Depends(get_session)
-) -> Response | StreamingResponse:
+async def parse_request(request: Request) -> tuple[dict, dict, bytes]:
     headers = dict(request.headers)
+    body_bytes = await request.body()
+    body_dict = parse_request_body_json(body_bytes)
+    return headers, body_dict, body_bytes
 
+
+def ensure_authorization(headers: dict) -> None:
     if "x-cashu" not in headers and "authorization" not in headers.keys():
-        return create_error_response(
-            "unauthorized", "Unauthorized", 401, request=request
-        )
+        # return create_error_response("unauthorized", "Unauthorized", 401, request=request)
+        raise HTTPException(status_code=401, detail="Unauthorized")
 
-    logger.info(  # TODO: move to middleware, async
+
+def log_request(request: Request, path: str) -> None:
+    logger.info(
         "Received proxy request",
         extra={
             "method": request.method,
@@ -151,58 +150,49 @@ async def proxy(
         },
     )
 
-    request_body = await request.body()
-    request_body_dict = parse_request_body_json(request_body, path)
 
-    model_id = request_body_dict.get("model", "unknown")
+def get_model_and_upstream(request_body: dict) -> tuple[Model, BaseUpstreamProvider]:
+    model_id = request_body.get("model", "unknown")
 
-    model_obj = get_model_instance(model_id)
-    if not model_obj:
-        return create_error_response(
-            "invalid_model", f"Model '{model_id}' not found", 400, request=request
-        )
+    if not (model := get_model_instance(model_id)):
+        raise HTTPException(status_code=400, detail=f"Model '{model_id}' not found")
 
-    upstream = get_provider_for_model(model_id)
-    if not upstream:
-        return create_error_response(
-            "invalid_model",
-            f"No provider found for model '{model_id}'",
-            400,
-            request=request,
-        )
+    if not (upstream := get_provider_for_model(model_id)):
+        raise HTTPException(status_code=400, detail=f"No provider with '{model_id}'")
 
-    _max_cost_for_model = await get_max_cost_for_model(
-        model=model_id, session=session, model_obj=model_obj
-    )
+    return model, upstream
+
+
+@proxy_router.api_route("/{path:path}", methods=["GET", "POST"], response_model=None)
+async def proxy(
+    request: Request,
+    path: str,
+    session: AsyncSession = Depends(get_session),
+) -> Response | StreamingResponse:
+    headers, body_dict, body_bytes = await parse_request(request)
+
+    ensure_authorization(headers)
+
+    log_request(request, path)
+
+    model, upstream = get_model_and_upstream(body_dict)
+
+    _max_cost_for_model = get_max_cost_for_model(model_obj=model)
     max_cost_for_model = await calculate_discounted_max_cost(
-        _max_cost_for_model, request_body_dict, model_obj=model_obj
+        _max_cost_for_model, body_dict, model_obj=model
     )
-    check_token_balance(headers, request_body_dict, max_cost_for_model)
+    pre_check_header_token_balance(headers, body_dict, max_cost_for_model)
 
     if x_cashu := headers.get("x-cashu", None):
         return await upstream.handle_x_cashu(
-            request, x_cashu, path, max_cost_for_model, model_obj
+            request, x_cashu, path, max_cost_for_model, model
         )
 
     elif auth := headers.get("authorization", None):
-        key = await get_bearer_token_key(headers, path, session, auth)
-
-    else:
-        if request.method not in ["GET"]:
-            raise HTTPException(
-                status_code=401,
-                detail={
-                    "error": {"type": "invalid_request_error", "code": "unauthorized"}
-                },
-            )
-
-        logger.debug("Processing unauthenticated GET request", extra={"path": path})
-        # TODO: why is this needed? can we remove it?
-        headers = upstream.prepare_headers(dict(request.headers))
-        return await upstream.forward_get_request(request, path, headers)
+        key = await get_credit(auth, session)
 
     # Only pay for request if we have request body data (for completions endpoints)
-    if request_body_dict:
+    if body_dict:
         await pay_for_request(key, max_cost_for_model, session)
 
     # Prepare headers for upstream
@@ -213,11 +203,11 @@ async def proxy(
         request,
         path,
         headers,
-        request_body,
+        body_bytes,
         key,
         max_cost_for_model,
         session,
-        model_obj,
+        model,
     )
 
     if response.status_code != 200:
@@ -241,87 +231,7 @@ async def proxy(
     return response
 
 
-async def get_bearer_token_key(
-    headers: dict, path: str, session: AsyncSession, auth: str
-) -> ApiKey:
-    """Handle bearer token authentication proxy requests."""
-    bearer_key = auth.replace("Bearer ", "") if auth.startswith("Bearer ") else ""
-    refund_address = headers.get("Refund-LNURL", None)
-    key_expiry_time = headers.get("Key-Expiry-Time", None)
-
-    logger.debug(
-        "Processing bearer token",
-        extra={
-            "path": path,
-            "has_refund_address": bool(refund_address),
-            "has_expiry_time": bool(key_expiry_time),
-            "bearer_key_preview": bearer_key[:20] + "..."
-            if len(bearer_key) > 20
-            else bearer_key,
-        },
-    )
-
-    # Validate key_expiry_time header
-    if key_expiry_time:
-        try:
-            key_expiry_time = int(key_expiry_time)  # type: ignore
-            logger.debug(
-                "Key expiry time validated",
-                extra={"expiry_time": key_expiry_time, "path": path},
-            )
-        except ValueError:
-            logger.error(
-                "Invalid Key-Expiry-Time header",
-                extra={"key_expiry_time": key_expiry_time, "path": path},
-            )
-            raise HTTPException(
-                status_code=400,
-                detail="Invalid Key-Expiry-Time: must be a valid Unix timestamp",
-            )
-        if not refund_address:
-            logger.error(
-                "Missing Refund-LNURL header with Key-Expiry-Time",
-                extra={"path": path, "expiry_time": key_expiry_time},
-            )
-            raise HTTPException(
-                status_code=400,
-                detail="Error: Refund-LNURL header required when using Key-Expiry-Time",
-            )
-    else:
-        key_expiry_time = None
-
-    try:
-        key = await validate_bearer_key(
-            bearer_key,
-            session,
-            refund_address,
-            key_expiry_time,  # type: ignore
-        )
-        logger.info(
-            "Bearer token validated successfully",
-            extra={
-                "path": path,
-                "key_hash": key.hashed_key[:8] + "...",
-                "key_balance": key.balance,
-            },
-        )
-        return key
-    except Exception as e:
-        logger.error(
-            "Bearer token validation failed",
-            extra={
-                "error": str(e),
-                "error_type": type(e).__name__,
-                "path": path,
-                "bearer_key_preview": bearer_key[:20] + "..."
-                if len(bearer_key) > 20
-                else bearer_key,
-            },
-        )
-        raise
-
-
-def parse_request_body_json(request_body: bytes, path: str) -> dict[str, Any]:
+def parse_request_body_json(request_body: bytes) -> dict[str, Any]:
     request_body_dict = {}
     if request_body:
         try:
@@ -341,7 +251,6 @@ def parse_request_body_json(request_body: bytes, path: str) -> dict[str, Any]:
             logger.debug(
                 "Request body parsed",
                 extra={
-                    "path": path,
                     "body_keys": list(request_body_dict.keys()),
                     "model": request_body_dict.get("model", "not_specified"),
                 },
@@ -351,7 +260,6 @@ def parse_request_body_json(request_body: bytes, path: str) -> dict[str, Any]:
                 "Invalid JSON in request body",
                 extra={
                     "error": str(e),
-                    "path": path,
                     "body_preview": request_body[:200].decode(errors="ignore")
                     if request_body
                     else "empty",

routstr/upstream/__init__.py

@@ -2,28 +2,24 @@ from .anthropic import AnthropicUpstreamProvider
 from .azure import AzureUpstreamProvider
 from .base import BaseUpstreamProvider
 from .fireworks import FireworksUpstreamProvider
-from .gemini import GeminiUpstreamProvider
 from .generic import GenericUpstreamProvider
 from .groq import GroqUpstreamProvider
 from .ollama import OllamaUpstreamProvider
 from .openai import OpenAIUpstreamProvider
 from .openrouter import OpenRouterUpstreamProvider
 from .perplexity import PerplexityUpstreamProvider
-from .ppqai import PPQAIUpstreamProvider
 from .xai import XAIUpstreamProvider
 
 upstream_provider_classes: list[type[BaseUpstreamProvider]] = [
     AnthropicUpstreamProvider,
     AzureUpstreamProvider,
     FireworksUpstreamProvider,
-    GeminiUpstreamProvider,
     GenericUpstreamProvider,
     GroqUpstreamProvider,
     OllamaUpstreamProvider,
     OpenAIUpstreamProvider,
     OpenRouterUpstreamProvider,
     PerplexityUpstreamProvider,
-    PPQAIUpstreamProvider,
     XAIUpstreamProvider,
 ]
 """List of all upstream classes"""

routstr/upstream/anthropic.py

@@ -1,6 +1,6 @@
 from typing import TYPE_CHECKING
 
-from ..payment.models import Model, async_fetch_openrouter_models
+from ..models import Model, async_fetch_openrouter_models
 from .base import BaseUpstreamProvider
 
 if TYPE_CHECKING:

routstr/upstream/base.py

@@ -10,45 +10,28 @@ from typing import TYPE_CHECKING, Mapping
 import httpx
 from fastapi import BackgroundTasks, HTTPException, Request
 from fastapi.responses import Response, StreamingResponse
-from pydantic import BaseModel
 
-from ..auth import adjust_payment_for_tokens
 from ..core import get_logger
-from ..core.db import ApiKey, AsyncSession, create_session
+from ..core.db import AsyncSession, TemporaryCredit, create_session
+from ..payment.helpers import adjust_payment_for_tokens
 
 if TYPE_CHECKING:
     from ..core.db import UpstreamProviderRow
 
-from ..payment.cost_calculation import (
-    CostData,
-    CostDataError,
-    MaxCostData,
-    calculate_cost,
-)
-from ..payment.helpers import create_error_response
-from ..payment.models import (
+from ..models.models import (
     Model,
     Pricing,
-    _calculate_usd_max_costs,
     _update_model_sats_pricing,
+    calculate_usd_max_costs,
 )
+from ..payment.cost import CostData, CostDataError, MaxCostData, calculate_cost
+from ..payment.helpers import create_error_response
 from ..payment.price import sats_usd_price
-from ..wallet import recieve_token, send_token
+from ..payment.wallet import recieve_token, send_token
 
 logger = get_logger(__name__)
 
 
-class TopupData(BaseModel):
-    """Universal top-up data schema for Lightning Network invoices."""
-
-    invoice_id: str
-    payment_request: str
-    amount: int
-    currency: str
-    expires_at: int | None = None
-    checkout_url: str | None = None
-
-
 class BaseUpstreamProvider:
     """Provider for forwarding requests to an upstream AI service API."""
 
@@ -99,7 +82,7 @@ class BaseUpstreamProvider:
         """Get metadata about this provider type for API responses.
 
         Returns:
-            Dict with provider type metadata including id, name, default_base_url, fixed_base_url, platform_url, can_create_account, can_topup, can_show_balance
+            Dict with provider type metadata including id, name, default_base_url, fixed_base_url, platform_url
         """
         return {
             "id": cls.provider_type,
@@ -107,9 +90,6 @@ class BaseUpstreamProvider:
             "default_base_url": cls.default_base_url or "",
             "fixed_base_url": bool(cls.default_base_url),
             "platform_url": cls.platform_url,
-            "can_create_account": False,
-            "can_topup": False,
-            "can_show_balance": False,
         }
 
     def prepare_headers(self, request_headers: dict) -> dict:
@@ -352,7 +332,7 @@ class BaseUpstreamProvider:
         )
 
     async def handle_streaming_chat_completion(
-        self, response: httpx.Response, key: ApiKey, max_cost_for_model: int
+        self, response: httpx.Response, key: TemporaryCredit, max_cost_for_model: int
     ) -> StreamingResponse:
         """Handle streaming chat completion responses with token usage tracking and cost adjustment.
 
@@ -541,7 +521,7 @@ class BaseUpstreamProvider:
     async def handle_non_streaming_chat_completion(
         self,
         response: httpx.Response,
-        key: ApiKey,
+        key: TemporaryCredit,
         session: AsyncSession,
         deducted_max_cost: int,
     ) -> Response:
@@ -647,7 +627,7 @@ class BaseUpstreamProvider:
         path: str,
         headers: dict,
         request_body: bytes | None,
-        key: ApiKey,
+        key: TemporaryCredit,
         max_cost_for_model: int,
         session: AsyncSession,
         model_obj: Model,
@@ -961,45 +941,43 @@ class BaseUpstreamProvider:
             extra={"model": model, "has_usage": "usage" in response_data},
         )
 
-        async with create_session() as session:
-            match await calculate_cost(response_data, max_cost_for_model, session):
-                case MaxCostData() as cost:
-                    logger.debug(
-                        "Using max cost pricing",
-                        extra={"model": model, "max_cost_msats": cost.total_msats},
-                    )
-                    return cost
-                case CostData() as cost:
-                    logger.debug(
-                        "Using token-based pricing",
-                        extra={
-                            "model": model,
-                            "total_cost_msats": cost.total_msats,
-                            "input_msats": cost.input_msats,
-                            "output_msats": cost.output_msats,
-                        },
-                    )
-                    return cost
-                case CostDataError() as error:
-                    logger.error(
-                        "Cost calculation error",
-                        extra={
-                            "model": model,
-                            "error_message": error.message,
-                            "error_code": error.code,
-                        },
-                    )
-                    raise HTTPException(
-                        status_code=400,
-                        detail={
-                            "error": {
-                                "message": error.message,
-                                "type": "invalid_request_error",
-                                "code": error.code,
-                            }
-                        },
-                    )
-        return None
+        match calculate_cost(response_data, max_cost_for_model):
+            case MaxCostData() as cost:
+                logger.debug(
+                    "Using max cost pricing",
+                    extra={"model": model, "max_cost_msats": cost.total_msats},
+                )
+                return cost
+            case CostData() as cost:
+                logger.debug(
+                    "Using token-based pricing",
+                    extra={
+                        "model": model,
+                        "total_cost_msats": cost.total_msats,
+                        "input_msats": cost.input_msats,
+                        "output_msats": cost.output_msats,
+                    },
+                )
+                return cost
+            case CostDataError() as error:
+                logger.error(
+                    "Cost calculation error",
+                    extra={
+                        "model": model,
+                        "error_message": error.message,
+                        "error_code": error.code,
+                    },
+                )
+                raise HTTPException(
+                    status_code=400,
+                    detail={
+                        "error": {
+                            "message": error.message,
+                            "type": "invalid_request_error",
+                            "code": error.code,
+                        }
+                    },
+                )
 
     async def send_refund(self, amount: int, unit: str, mint: str | None = None) -> str:
         """Create and send a refund token to the user.
@@ -1701,7 +1679,7 @@ class BaseUpstreamProvider:
             adjusted_pricing.max_prompt_cost,
             adjusted_pricing.max_completion_cost,
             adjusted_pricing.max_cost,
-        ) = _calculate_usd_max_costs(temp_model)
+        ) = calculate_usd_max_costs(temp_model)
 
         return Model(
             id=model.id,
@@ -1726,6 +1704,7 @@ class BaseUpstreamProvider:
         Returns:
             List of Model objects with pricing
         """
+        logger.debug(f"Fetching models for {self.provider_type or self.base_url}")
 
         try:
             or_models, provider_models_response = await asyncio.gather(
@@ -1752,9 +1731,18 @@ class BaseUpstreamProvider:
                 else:
                     not_found_models.append(model_id)
 
+            logger.info(
+                "Fetched models for provider",
+                extra={
+                    "provider": self.provider_type or self.base_url,
+                    "found_count": len(found_models),
+                    "not_found_count": len(not_found_models),
+                },
+            )
+
             if not_found_models:
                 logger.debug(
-                    f"({len(not_found_models)}/{len(provider_model_ids)}) unmatched models for {self.provider_type or self.base_url}",
+                    "Models not found in OpenRouter",
                     extra={"not_found_models": not_found_models},
                 )
 
@@ -1822,7 +1810,10 @@ class BaseUpstreamProvider:
                 self._models_cache = models_with_fees
 
             self._models_by_id = {m.id: m for m in self._models_cache}
-
+            logger.info(
+                f"Refreshed models cache for {self.provider_type or self.base_url}",
+                extra={"model_count": len(models)},
+            )
         except Exception as e:
             logger.error(
                 f"Failed to refresh models cache for {self.provider_type or self.base_url}",
@@ -1847,60 +1838,3 @@ class BaseUpstreamProvider:
             Model object or None if not found
         """
         return self._models_by_id.get(model_id)
-
-    @classmethod
-    async def create_account_static(cls) -> dict[str, object]:
-        """Create a new account with the provider (class method, no instance needed).
-
-        Returns:
-            Dict with account creation details including api_key
-
-        Raises:
-            NotImplementedError: If provider does not support account creation
-        """
-        raise NotImplementedError(
-            f"Provider {cls.provider_type} does not support account creation"
-        )
-
-    async def create_account(self) -> dict[str, object]:
-        """Create a new account with the provider.
-
-        Returns:
-            Dict with account creation details including api_key
-
-        Raises:
-            NotImplementedError: If provider does not support account creation
-        """
-        raise NotImplementedError(
-            f"Provider {self.provider_type} does not support account creation"
-        )
-
-    async def initiate_topup(self, amount: int) -> TopupData:
-        """Initiate a Lightning Network top-up for the provider account.
-
-        Args:
-            amount: Amount in currency units to top up
-
-        Returns:
-            TopupData with standardized invoice information
-
-        Raises:
-            NotImplementedError: If provider does not support top-up
-        """
-        raise NotImplementedError(
-            f"Provider {self.provider_type} does not support top-up"
-        )
-
-    async def get_balance(self) -> float | None:
-        """Get the current account balance from the provider.
-
-        Returns:
-            Float representing the balance amount, or None if not supported/available.
-            Typically in USD or the provider's credit unit.
-
-        Raises:
-            NotImplementedError: If provider does not support balance checking (default behavior)
-        """
-        raise NotImplementedError(
-            f"Provider {self.provider_type} does not support balance checking"
-        )

routstr/upstream/clients/__init__.py

@@ -1,3 +0,0 @@
-from .gemini import GeminiClient
-
-__all__ = ["GeminiClient"]

routstr/upstream/clients/base.py

@@ -1,40 +0,0 @@
-from __future__ import annotations
-
-from abc import ABC, abstractmethod
-from typing import Any, AsyncGenerator
-
-
-class BaseAPIClient(ABC):
-    """Base class for AI provider API clients."""
-
-    def __init__(self, api_key: str, base_url: str | None = None):
-        self.api_key = api_key
-        self.base_url = base_url
-
-    @abstractmethod
-    async def generate_content(
-        self,
-        model: str,
-        messages: list[dict[str, Any]],
-        temperature: float | None = None,
-        max_tokens: int | None = None,
-        **kwargs: Any,
-    ) -> dict[str, Any]:
-        """Generate content non-streaming."""
-        pass
-
-    @abstractmethod
-    def generate_content_stream(
-        self,
-        model: str,
-        messages: list[dict[str, Any]],
-        temperature: float | None = None,
-        max_tokens: int | None = None,
-        **kwargs: Any,
-    ) -> AsyncGenerator[dict[str, Any], None]:
-        pass
-
-    @abstractmethod
-    async def list_models(self) -> list[dict[str, Any]]:
-        """List available models."""
-        pass

routstr/upstream/clients/gemini.py

@@ -1,88 +0,0 @@
-from __future__ import annotations
-
-from typing import Any, AsyncGenerator
-
-from openai import AsyncOpenAI
-
-from .base import BaseAPIClient
-
-
-class GeminiClient(BaseAPIClient):
-    """Gemini API client using OpenAI compatibility layer."""
-
-    def __init__(self, api_key: str, base_url: str | None = None):
-        super().__init__(api_key, base_url)
-        self.client = AsyncOpenAI(
-            api_key=api_key,
-            base_url=base_url
-            or "https://generativelanguage.googleapis.com/v1beta/openai/",
-        )
-
-    async def generate_content(
-        self,
-        model: str,
-        messages: list[dict[str, Any]],
-        temperature: float | None = None,
-        max_tokens: int | None = None,
-        **kwargs: Any,
-    ) -> dict[str, Any]:
-        from openai import NOT_GIVEN
-
-        response = await self.client.chat.completions.create(
-            model=model,
-            messages=messages,  # type: ignore
-            temperature=temperature if temperature is not None else NOT_GIVEN,
-            max_tokens=max_tokens if max_tokens is not None else NOT_GIVEN,
-            top_p=kwargs.get("top_p", NOT_GIVEN),
-        )
-        return response.model_dump()
-
-    async def generate_content_stream(
-        self,
-        model: str,
-        messages: list[dict[str, Any]],
-        temperature: float | None = None,
-        max_tokens: int | None = None,
-        **kwargs: Any,
-    ) -> AsyncGenerator[dict[str, Any], None]:
-        from openai import NOT_GIVEN
-
-        usage_callback = kwargs.get("usage_callback")
-        completion_callback = kwargs.get("completion_callback")
-
-        stream = await self.client.chat.completions.create(
-            model=model,
-            messages=messages,  # type: ignore
-            stream=True,
-            stream_options={"include_usage": True},
-            temperature=temperature if temperature is not None else NOT_GIVEN,
-            max_tokens=max_tokens if max_tokens is not None else NOT_GIVEN,
-            top_p=kwargs.get("top_p", NOT_GIVEN),
-        )
-
-        final_usage = None
-
-        async for chunk in stream:
-            chunk_data = chunk.model_dump()
-
-            if chunk.usage:
-                final_usage = chunk.usage.model_dump()
-                if usage_callback:
-                    usage_callback(final_usage)
-
-            yield chunk_data
-
-        if completion_callback:
-            await completion_callback(model, final_usage)
-
-    async def list_models(self) -> list[dict[str, Any]]:
-        """List available Gemini models."""
-        try:
-            response = await self.client.models.list()
-            return [model.model_dump() for model in response.data]
-        except Exception as e:
-            from ...core.logging import get_logger
-
-            logger = get_logger(__name__)
-            logger.error(f"Failed to list Gemini models: {e}")
-            return []

routstr/upstream/gemini.py

@@ -1,283 +0,0 @@
-from __future__ import annotations
-
-import json
-from collections.abc import AsyncGenerator
-from typing import TYPE_CHECKING, Any
-
-from fastapi import Request
-from fastapi.responses import Response, StreamingResponse
-
-from .base import BaseUpstreamProvider
-from .clients.gemini import GeminiClient
-
-if TYPE_CHECKING:
-    from ..core.db import ApiKey, AsyncSession, UpstreamProviderRow
-    from ..payment.models import Model
-
-from ..core.logging import get_logger
-
-logger = get_logger(__name__)
-
-
-class GeminiUpstreamProvider(BaseUpstreamProvider):
-    provider_type = "gemini"
-    default_base_url = "https://generativelanguage.googleapis.com/v1beta"
-    platform_url = "https://aistudio.google.com/app/apikey"
-
-    def __init__(
-        self,
-        base_url: str = "https://generativelanguage.googleapis.com/v1beta",
-        api_key: str = "",
-        provider_fee: float = 1.01,
-    ):
-        super().__init__(
-            api_key=api_key,
-            provider_fee=provider_fee,
-            base_url=base_url,
-        )
-        self._client: GeminiClient | None = None
-
-    @property
-    def client(self) -> GeminiClient:
-        """Get or create the Gemini API client."""
-        if self._client is None:
-            self._client = GeminiClient(api_key=self.api_key)
-        return self._client
-
-    @classmethod
-    def from_db_row(
-        cls, provider_row: "UpstreamProviderRow"
-    ) -> "GeminiUpstreamProvider":
-        return cls(
-            base_url=provider_row.base_url,
-            api_key=provider_row.api_key,
-            provider_fee=provider_row.provider_fee,
-        )
-
-    @classmethod
-    def get_provider_metadata(cls) -> dict[str, object]:
-        return {
-            "id": cls.provider_type,
-            "name": "Google Gemini",
-            "default_base_url": cls.default_base_url,
-            "fixed_base_url": True,
-            "platform_url": cls.platform_url,
-        }
-
-    def transform_model_name(self, model_id: str) -> str:
-        return model_id.removeprefix("gemini/")
-
-    async def forward_request(
-        self,
-        request: Request,
-        path: str,
-        headers: dict,
-        request_body: bytes | None,
-        key: ApiKey,
-        max_cost_for_model: int,
-        session: AsyncSession,
-        model_obj: Model,
-    ) -> Response | StreamingResponse:
-        # Remove provider prefix from model ID for Gemini API
-        if "/" in model_obj.id:
-            model_obj.id = model_obj.id.split("/", 1)[1]
-
-        if not path.startswith("chat/completions"):
-            return await super().forward_request(
-                request,
-                path,
-                headers,
-                request_body,
-                key,
-                max_cost_for_model,
-                session,
-                model_obj,
-            )
-
-        if not request_body:
-            return await super().forward_request(
-                request,
-                path,
-                headers,
-                request_body,
-                key,
-                max_cost_for_model,
-                session,
-                model_obj,
-            )
-
-        try:
-            openai_data = json.loads(request_body)
-            messages = openai_data.get("messages", [])
-            temperature = openai_data.get("temperature")
-            max_tokens = openai_data.get("max_tokens")
-            top_p = openai_data.get("top_p")
-            is_streaming = openai_data.get("stream", False)
-
-            logger.info(
-                "Processing Gemini request with client abstraction",
-                extra={
-                    "model": model_obj.id,
-                    "is_streaming": is_streaming,
-                    "message_count": len(messages),
-                    "key_hash": key.hashed_key[:8] + "...",
-                },
-            )
-
-            if is_streaming:
-                final_usage_data: dict | None = None
-
-                def usage_callback(usage_data: dict[str, Any]) -> None:
-                    """Callback to capture usage data during streaming"""
-                    nonlocal final_usage_data
-                    final_usage_data = usage_data
-
-                async def completion_callback(
-                    model: str, usage_data: dict[str, Any] | None
-                ) -> None:
-                    """Callback to handle payment when streaming completes"""
-                    nonlocal final_usage_data
-                    if usage_data:
-                        final_usage_data = usage_data
-
-                    payment_data = {
-                        "model": model,
-                        "usage": final_usage_data,
-                    }
-
-                    from ..auth import adjust_payment_for_tokens
-                    from ..core.db import create_session
-
-                    async with create_session() as new_session:
-                        fresh_key = await new_session.get(key.__class__, key.hashed_key)
-                        if fresh_key:
-                            try:
-                                cost_data = await adjust_payment_for_tokens(
-                                    fresh_key,
-                                    payment_data,
-                                    new_session,
-                                    max_cost_for_model,
-                                )
-
-                                logger.info(
-                                    "Gemini streaming payment finalized",
-                                    extra={
-                                        "cost_data": cost_data,
-                                        "usage_data": final_usage_data,
-                                        "key_hash": key.hashed_key[:8] + "...",
-                                    },
-                                )
-                            except Exception as cost_error:
-                                logger.error(
-                                    "Error finalizing Gemini streaming payment",
-                                    extra={
-                                        "error": str(cost_error),
-                                        "key_hash": key.hashed_key[:8] + "...",
-                                    },
-                                )
-
-                response_generator = self.client.generate_content_stream(
-                    model=model_obj.id,
-                    messages=messages,
-                    temperature=temperature,
-                    max_tokens=max_tokens,
-                    top_p=top_p,
-                    usage_callback=usage_callback,
-                    completion_callback=completion_callback,
-                )
-
-                async def stream_with_cost() -> AsyncGenerator[bytes, None]:
-                    try:
-                        async for chunk in response_generator:
-                            sse_data = f"data: {json.dumps(chunk)}\n\n"
-                            yield sse_data.encode()
-
-                    except Exception as e:
-                        logger.error(
-                            "Error in Gemini streaming response",
-                            extra={
-                                "error": str(e),
-                                "error_type": type(e).__name__,
-                                "key_hash": key.hashed_key[:8] + "...",
-                            },
-                        )
-                        raise
-
-                return StreamingResponse(
-                    stream_with_cost(),
-                    media_type="text/event-stream",
-                    headers={"Cache-Control": "no-cache", "Connection": "keep-alive"},
-                )
-
-            else:
-                openai_format_response = await self.client.generate_content(
-                    model=model_obj.id,
-                    messages=messages,
-                    temperature=temperature,
-                    max_tokens=max_tokens,
-                    top_p=top_p,
-                )
-
-                from ..auth import adjust_payment_for_tokens
-
-                cost_data = await adjust_payment_for_tokens(
-                    key, openai_format_response, session, max_cost_for_model
-                )
-                openai_format_response["cost"] = cost_data
-
-                logger.info(
-                    "Gemini non-streaming payment completed",
-                    extra={
-                        "cost_data": cost_data,
-                        "model": model_obj.id,
-                        "key_hash": key.hashed_key[:8] + "...",
-                    },
-                )
-
-                return Response(
-                    content=json.dumps(openai_format_response),
-                    media_type="application/json",
-                    headers={"Cache-Control": "no-cache"},
-                )
-
-        except Exception as e:
-            logger.error(
-                "Error in Gemini forward_request",
-                extra={
-                    "error": str(e),
-                    "error_type": type(e).__name__,
-                    "path": path,
-                    "key_hash": key.hashed_key[:8] + "...",
-                },
-            )
-            return await super().forward_request(
-                request,
-                path,
-                headers,
-                request_body,
-                key,
-                max_cost_for_model,
-                session,
-                model_obj,
-            )
-
-    async def _fetch_provider_models(self) -> dict:
-        """Fetch models from Gemini API."""
-        try:
-            models_data = await self.client.list_models()
-
-            for model in models_data:
-                if "id" in model and model["id"].startswith("models/"):
-                    model["id"] = model["id"].removeprefix("models/")
-
-            return {"data": models_data}
-        except Exception as e:
-            logger.error(
-                f"Failed to fetch models from Gemini API: {e}",
-                extra={
-                    "error": str(e),
-                    "error_type": type(e).__name__,
-                    "base_url": self.base_url,
-                },
-            )
-            return {"data": []}

routstr/upstream/helpers.py

@@ -10,7 +10,7 @@ if TYPE_CHECKING:
 
 from ..core import get_logger
 from ..core.db import AsyncSession, ModelRow, UpstreamProviderRow, create_session
-from ..payment.models import Model
+from ..models import Model
 from .base import BaseUpstreamProvider
 
 logger = get_logger(__name__)
@@ -82,7 +82,7 @@ async def get_all_models_with_overrides(
     """
     from sqlmodel import select
 
-    from ..payment.models import _row_to_model
+    from ..models.crud import row_to_model
 
     async with create_session() as session:
         result = await session.exec(select(ModelRow).where(ModelRow.enabled))
@@ -108,7 +108,7 @@ async def get_all_models_with_overrides(
         for model in upstream.get_cached_models():
             if model.id in overrides_by_id:
                 override_row, provider_fee = overrides_by_id[model.id]
-                all_models[model.id] = _row_to_model(
+                all_models[model.id] = row_to_model(
                     override_row, apply_provider_fee=True, provider_fee=provider_fee
                 )
             elif model.enabled:
@@ -193,7 +193,7 @@ async def init_upstreams() -> list[BaseUpstreamProvider]:
             if provider:
                 await provider.refresh_models_cache()
                 upstreams.append(provider)
-                logger.debug(
+                logger.info(
                     f"Initialized {provider_row.provider_type} provider",
                     extra={
                         "base_url": provider_row.base_url,

routstr/upstream/ollama.py

@@ -9,8 +9,8 @@ from fastapi.responses import Response, StreamingResponse
 from .base import BaseUpstreamProvider
 
 if TYPE_CHECKING:
-    from ..core.db import ApiKey, AsyncSession, UpstreamProviderRow
-    from ..payment.models import Model
+    from ..core.db import AsyncSession, TemporaryCredit, UpstreamProviderRow
+    from ..models import Model
 
 from ..core.logging import get_logger
 
@@ -73,7 +73,7 @@ class OllamaUpstreamProvider(BaseUpstreamProvider):
         path: str,
         headers: dict,
         request_body: bytes | None,
-        key: ApiKey,
+        key: TemporaryCredit,
         max_cost_for_model: int,
         session: AsyncSession,
         model_obj: Model,
@@ -102,7 +102,7 @@ class OllamaUpstreamProvider(BaseUpstreamProvider):
 
     async def fetch_models(self) -> list[Model]:
         """Fetch models from Ollama API using /api/tags endpoint."""
-        from ..payment.models import Architecture, Model, Pricing, TopProvider
+        from ..models.models import Architecture, Model, Pricing, TopProvider
 
         try:
             async with httpx.AsyncClient(timeout=30.0) as client:
@@ -199,7 +199,7 @@ class OllamaUpstreamProvider(BaseUpstreamProvider):
     async def refresh_models_cache(self) -> None:
         """Refresh the in-memory models cache from upstream API."""
         try:
-            from ..payment.models import _update_model_sats_pricing
+            from ..models.models import _update_model_sats_pricing
             from ..payment.price import sats_usd_price
 
             models = await self.fetch_models()
@@ -252,7 +252,7 @@ class OllamaUpstreamProvider(BaseUpstreamProvider):
         Returns:
             Model with provider fee applied to pricing and max costs calculated
         """
-        from ..payment.models import Model, Pricing, _calculate_usd_max_costs
+        from ..models.models import Model, Pricing, calculate_usd_max_costs
 
         adjusted_pricing = Pricing.parse_obj(
             {k: v * self.provider_fee for k, v in model.pricing.dict().items()}
@@ -278,7 +278,7 @@ class OllamaUpstreamProvider(BaseUpstreamProvider):
             adjusted_pricing.max_prompt_cost,
             adjusted_pricing.max_completion_cost,
             adjusted_pricing.max_cost,
-        ) = _calculate_usd_max_costs(temp_model)
+        ) = calculate_usd_max_costs(temp_model)
 
         return Model(
             id=model.id,

routstr/upstream/openai.py

@@ -1,6 +1,6 @@
 from typing import TYPE_CHECKING
 
-from ..payment.models import Model, async_fetch_openrouter_models
+from ..models import Model, async_fetch_openrouter_models
 from .base import BaseUpstreamProvider
 
 if TYPE_CHECKING:

routstr/upstream/openrouter.py

@@ -1,8 +1,6 @@
 from typing import TYPE_CHECKING
 
-import httpx
-
-from ..payment.models import Model, async_fetch_openrouter_models
+from ..models import Model, async_fetch_openrouter_models
 from .base import BaseUpstreamProvider
 
 if TYPE_CHECKING:
@@ -44,33 +42,9 @@ class OpenRouterUpstreamProvider(BaseUpstreamProvider):
             "default_base_url": cls.default_base_url,
             "fixed_base_url": True,
             "platform_url": cls.platform_url,
-            "can_show_balance": True,
         }
 
     async def fetch_models(self) -> list[Model]:
         """Fetch all OpenRouter models."""
         models_data = await async_fetch_openrouter_models()
         return [Model(**model) for model in models_data]  # type: ignore
-
-    async def get_balance(self) -> float | None:
-        """Get the current account balance from OpenRouter.
-
-        Returns:
-            Float representing the balance amount (in credits/USD), or None if unavailable.
-        """
-        url = f"{self.base_url}/credits"
-        headers = {"Authorization": f"Bearer {self.api_key}"}
-
-        try:
-            async with httpx.AsyncClient(timeout=30.0) as client:
-                response = await client.get(url, headers=headers)
-                response.raise_for_status()
-                data = response.json()
-
-                credits_data = data.get("data", {})
-                total_credits = float(credits_data.get("total_credits", 0.0))
-                total_usage = float(credits_data.get("total_usage", 0.0))
-
-                return total_credits - total_usage
-        except Exception:
-            return None

routstr/upstream/perplexity.py

@@ -1,6 +1,6 @@
 from typing import TYPE_CHECKING
 
-from ..payment.models import Model, async_fetch_openrouter_models
+from ..models import Model, async_fetch_openrouter_models
 from .base import BaseUpstreamProvider
 
 if TYPE_CHECKING:

routstr/upstream/ppqai.py

@@ -1,401 +0,0 @@
-from __future__ import annotations
-
-from typing import TYPE_CHECKING
-
-import httpx
-from pydantic import BaseModel
-
-from ..core.logging import get_logger
-from ..payment.models import Architecture, Model, Pricing, async_fetch_openrouter_models
-from .base import BaseUpstreamProvider, TopupData
-
-if TYPE_CHECKING:
-    from ..core.db import UpstreamProviderRow
-
-logger = get_logger(__name__)
-
-
-class PPQAIModelPricing(BaseModel):
-    ui: dict[str, float]
-    api: dict[str, float]
-
-
-class PPQAIModel(BaseModel):
-    id: str
-    provider: str
-    name: str
-    created_at: int
-    context_length: int
-    pricing: PPQAIModelPricing
-    popular: bool
-
-
-class PPQAIUpstreamProvider(BaseUpstreamProvider):
-    """Upstream provider for PPQ.AI API with Lightning Network top-up support."""
-
-    provider_type = "ppqai"
-    default_base_url = "https://api.ppq.ai"
-    platform_url = "https://ppq.ai/api-docs"
-    IGNORED_MODEL_IDS: list[str] = ["auto"]
-
-    def __init__(self, api_key: str, provider_fee: float = 1.0):
-        super().__init__(
-            base_url=self.default_base_url, api_key=api_key, provider_fee=provider_fee
-        )
-
-    @classmethod
-    def from_db_row(
-        cls, provider_row: "UpstreamProviderRow"
-    ) -> "PPQAIUpstreamProvider":
-        return cls(
-            api_key=provider_row.api_key,
-            provider_fee=provider_row.provider_fee,
-        )
-
-    @classmethod
-    def get_provider_metadata(cls) -> dict[str, object]:
-        return {
-            "id": cls.provider_type,
-            "name": "PPQ.AI",
-            "default_base_url": cls.default_base_url,
-            "fixed_base_url": True,
-            "platform_url": cls.platform_url,
-            "can_create_account": True,
-            "can_topup": True,
-            "can_show_balance": True,
-        }
-
-    def transform_model_name(self, model_id: str) -> str:
-        return model_id
-
-    @classmethod
-    async def create_account_static(cls) -> dict[str, object]:
-        """Create a new PPQ.AI account without requiring an instance.
-
-        Returns:
-            Dict containing 'credit_id' and 'api_key' for the new account.
-
-        Raises:
-            httpx.HTTPStatusError: If the API request fails.
-        """
-        url = f"{cls.default_base_url}/accounts/create"
-
-        logger.info("Creating new PPQ.AI account", extra={"url": url})
-
-        async with httpx.AsyncClient(timeout=30.0) as client:
-            response = await client.post(url)
-            response.raise_for_status()
-            account_data = response.json()
-
-            logger.info(
-                "Successfully created PPQ.AI account",
-                extra={
-                    "credit_id": account_data.get("credit_id"),
-                    "has_api_key": bool(account_data.get("api_key")),
-                },
-            )
-
-            return account_data
-
-    async def fetch_models(self) -> list[Model]:
-        """Fetch models from PPQ.AI API."""
-        url = f"{self.base_url}/models"
-        headers = {"Authorization": f"Bearer {self.api_key}"}
-
-        try:
-            async with httpx.AsyncClient(timeout=30.0) as client:
-                response = await client.get(url, headers=headers)
-                response.raise_for_status()
-                data = response.json()
-
-                models_data = data.get("data", [])
-
-                or_models = [
-                    Model(**model)  # type: ignore
-                    for model in await async_fetch_openrouter_models()
-                ]
-
-                models = []
-                for model_data in models_data:
-                    try:
-                        ppqai_model = PPQAIModel.parse_obj(model_data)
-                        if ppqai_model.id in self.IGNORED_MODEL_IDS:
-                            continue
-
-                        or_model = next(
-                            (
-                                model
-                                for model in or_models
-                                if (model.id == ppqai_model.id)
-                                or (model.id.split("/")[-1] == ppqai_model.id)
-                                or (model.id == ppqai_model.id.split("/")[-1])
-                            ),
-                            None,
-                        )
-
-                        if or_model:
-                            if input_price := ppqai_model.pricing.api.get(
-                                "input_per_1M"
-                            ):
-                                or_model.pricing.prompt = input_price / 1_000_000
-                            if output_price := ppqai_model.pricing.api.get(
-                                "output_per_1M"
-                            ):
-                                or_model.pricing.completion = output_price / 1_000_000
-                            if cl := ppqai_model.context_length:
-                                or_model.context_length = cl
-                            models.append(or_model)
-                        else:
-                            input_price = ppqai_model.pricing.api.get(
-                                "input_per_1M", 0.0
-                            )
-                            output_price = ppqai_model.pricing.api.get(
-                                "output_per_1M", 0.0
-                            )
-
-                            models.append(
-                                Model(
-                                    id=ppqai_model.id,
-                                    name=ppqai_model.name,
-                                    created=ppqai_model.created_at // 1000,
-                                    description=f"{ppqai_model.provider} model",
-                                    context_length=ppqai_model.context_length,
-                                    architecture=Architecture(
-                                        modality="text->text",
-                                        input_modalities=["text"],
-                                        output_modalities=["text"],
-                                        tokenizer="Unknown",
-                                        instruct_type=None,
-                                    ),
-                                    pricing=Pricing(
-                                        prompt=input_price / 1_000_000,
-                                        completion=output_price / 1_000_000,
-                                        request=0.0,
-                                        image=0.0,
-                                        web_search=0.0,
-                                        internal_reasoning=0.0,
-                                    ),
-                                )
-                            )
-                    except Exception as e:
-                        logger.warning(
-                            "Failed to parse PPQ.AI model",
-                            extra={
-                                "model_id": model_data.get("id", "unknown"),
-                                "error": str(e),
-                                "error_type": type(e).__name__,
-                            },
-                        )
-
-                return models
-
-        except Exception as e:
-            logger.error(
-                "Error fetching models from PPQ.AI",
-                extra={"error": str(e), "error_type": type(e).__name__},
-            )
-            return []
-
-    async def create_account(self) -> dict[str, object]:
-        """Create a new PPQ.AI account.
-
-        Returns:
-            Dict containing 'credit_id' and 'api_key' for the new account.
-
-        Raises:
-            httpx.HTTPStatusError: If the API request fails.
-        """
-        url = f"{self.base_url}/accounts/create"
-
-        logger.info("Creating new PPQ.AI account", extra={"url": url})
-
-        async with httpx.AsyncClient(timeout=30.0) as client:
-            response = await client.post(url)
-            response.raise_for_status()
-            account_data = response.json()
-
-            logger.info(
-                "Successfully created PPQ.AI account",
-                extra={
-                    "credit_id": account_data.get("credit_id"),
-                    "has_api_key": bool(account_data.get("api_key")),
-                },
-            )
-
-            return account_data
-
-    async def create_lightning_topup(
-        self, amount: int, currency: str
-    ) -> dict[str, object]:
-        """Create a Lightning Network top-up invoice for this account.
-
-        Args:
-            amount: Amount to top up (in the specified currency)
-            currency: Currency for the top-up (default: "USD")
-
-        Returns:
-            Dict containing invoice details including 'invoice_id', 'payment_request', etc.
-
-        Raises:
-            httpx.HTTPStatusError: If the API request fails.
-        """
-        url = f"{self.base_url}/topup/create/btc-lightning"
-        headers = {
-            "Authorization": f"Bearer {self.api_key}",
-            "Content-Type": "application/json",
-        }
-        payload = {"amount": amount, "currency": currency}
-
-        logger.info(
-            "Creating Lightning top-up invoice",
-            extra={"url": url, "amount": amount, "currency": currency},
-        )
-
-        async with httpx.AsyncClient(timeout=30.0) as client:
-            print(f"Payload: {payload}", "sending to", url)
-            response = await client.post(url, headers=headers, json=payload)
-            response.raise_for_status()
-            invoice_data = response.json()
-
-            logger.info(
-                "Successfully created Lightning top-up invoice",
-                extra={
-                    "invoice_id": invoice_data.get("invoice_id"),
-                    "amount": amount,
-                    "currency": currency,
-                },
-            )
-
-            return invoice_data
-
-    async def check_topup_status(self, invoice_id: str) -> bool:
-        """Check the status of a Lightning top-up invoice.
-
-        Args:
-            invoice_id: The invoice ID to check
-
-        Returns:
-            True if the invoice is paid (status == "Settled"), False otherwise
-
-        Raises:
-            httpx.HTTPStatusError: If the API request fails.
-        """
-        url = f"{self.base_url}/topup/status/{invoice_id}"
-        headers = {"Authorization": f"Bearer {self.api_key}"}
-
-        logger.debug(
-            "Checking Lightning top-up status",
-            extra={"url": url, "invoice_id": invoice_id},
-        )
-
-        async with httpx.AsyncClient(timeout=30.0) as client:
-            response = await client.get(url, headers=headers)
-            response.raise_for_status()
-            status_data = response.json()
-
-            is_paid = status_data.get("status") == "Settled"
-
-            logger.debug(
-                "Retrieved Lightning top-up status",
-                extra={
-                    "invoice_id": invoice_id,
-                    "status": status_data.get("status"),
-                    "is_paid": is_paid,
-                },
-            )
-
-            return is_paid
-
-    async def initiate_topup(self, amount: int) -> TopupData:
-        """Initiate a Lightning Network top-up for the PPQ.AI account.
-
-        Args:
-            amount: Amount in currency units to top up (will be sent to PPQ.AI API)
-
-        Returns:
-            TopupData with standardized invoice information
-
-        Raises:
-            httpx.HTTPStatusError: If the API request fails
-        """
-        ppq_response = await self.create_lightning_topup(amount, "USD")
-
-        logger.info(
-            "PPQ.AI top-up response",
-            extra={
-                "ppq_response": ppq_response,
-                "invoice_id": ppq_response.get("invoice_id"),
-                "has_lightning_invoice": "lightning_invoice" in ppq_response,
-            },
-        )
-
-        expires_at_value = ppq_response.get("expires_at")
-        checkout_url_value = ppq_response.get("checkout_url")
-
-        topup_data = TopupData(
-            invoice_id=str(ppq_response["invoice_id"]),
-            payment_request=str(ppq_response["lightning_invoice"]),
-            amount=int(ppq_response["amount"])
-            if isinstance(ppq_response["amount"], (int, float, str))
-            else 0,
-            currency=str(ppq_response["currency"]),
-            expires_at=int(expires_at_value)
-            if isinstance(expires_at_value, (int, float, str))
-            and expires_at_value is not None
-            else None,
-            checkout_url=str(checkout_url_value)
-            if checkout_url_value is not None
-            else None,
-        )
-
-        logger.info(
-            "Created TopupData",
-            extra={
-                "invoice_id": topup_data.invoice_id,
-                "payment_request_length": len(topup_data.payment_request),
-                "amount": topup_data.amount,
-            },
-        )
-
-        return topup_data
-
-    async def get_balance(self) -> float | None:
-        """Get the current account balance from PPQ.AI.
-
-        Returns:
-            Float representing the balance amount (in USD), or None if unavailable.
-
-        Raises:
-            httpx.HTTPStatusError: If the API request fails
-        """
-        data = await self.check_balance()
-        balance = data.get("balance")
-        if isinstance(balance, (int, float)):
-            return float(balance)
-        return None
-
-    async def check_balance(self) -> dict[str, object]:
-        """Check the account balance for this PPQ.AI account.
-
-        Returns:
-            Dict containing balance information
-
-        Raises:
-            httpx.HTTPStatusError: If the API request fails.
-        """
-        url = f"{self.base_url}/credits/balance"
-        headers = {"Authorization": f"Bearer {self.api_key}"}
-
-        logger.debug("Checking PPQ.AI account balance", extra={"url": url})
-
-        async with httpx.AsyncClient(timeout=30.0) as client:
-            response = await client.post(url, headers=headers, json={})
-            response.raise_for_status()
-            balance_data = response.json()
-
-            logger.debug(
-                "Retrieved PPQ.AI account balance",
-                extra={"balance": balance_data.get("balance")},
-            )
-
-            return balance_data

routstr/upstream/xai.py

@@ -1,6 +1,6 @@
 from typing import TYPE_CHECKING
 
-from ..payment.models import Model, async_fetch_openrouter_models
+from ..models import Model, async_fetch_openrouter_models
 from .base import BaseUpstreamProvider
 
 if TYPE_CHECKING:

scripts/crontab.example

@@ -1,5 +0,0 @@
-# Example crontab entries for Routstr tasks
-
-# Update models.json daily at 03:15 (optional)
-# OUTPUT_FILE=/app/models.json SOURCE=openrouter
-15 3 * * * /usr/local/bin/python /app/scripts/models_meta.py >> /var/log/cron.log 2>&1

scripts/models_meta.py

@@ -1,90 +0,0 @@
-#!/usr/bin/env python3
-
-import json
-import os
-from typing import TypedDict
-from urllib.request import urlopen
-
-
-class ModelArchitecture(TypedDict):
-    modality: str
-    input_modalities: list[str]
-    output_modalities: list[str]
-    tokenizer: str
-    instruct_type: str | None
-
-
-class ModelPricing(TypedDict):
-    prompt: str
-    completion: str
-    request: str
-    image: str
-    web_search: str
-    internal_reasoning: str
-
-
-class ModelProvider(TypedDict):
-    context_length: int
-    max_completion_tokens: int | None
-    is_moderated: bool
-
-
-class Model(TypedDict):
-    id: str
-    name: str
-    created: int
-    description: str
-    context_length: int
-    architecture: ModelArchitecture
-    pricing: ModelPricing
-    top_provider: ModelProvider
-    per_request_limits: dict | None
-
-
-OUTPUT_FILE = os.getenv("OUTPUT_FILE", "models.json")
-SOURCE = os.getenv("SOURCE")
-
-
-def fetch_openrouter_models(source_filter: str | None = None) -> list[Model]:
-    """Fetches model information from OpenRouter API."""
-    base_url = "https://openrouter.ai/api/v1"
-    with urlopen(f"{base_url}/models") as response:
-        data = json.loads(response.read().decode("utf-8"))
-
-        models_data: list[Model] = []
-        for model in data.get("data", []):
-            model_id = model.get("id", "")
-
-            if source_filter:
-                source_prefix = f"{source_filter}/"
-                if not model_id.startswith(source_prefix):
-                    continue
-
-                model = dict(model)
-                model["id"] = model_id[len(source_prefix) :]
-                model_id = model["id"]
-
-            if (
-                "(free)" in model.get("name", "")
-                or model_id == "openrouter/auto"
-                or model_id == "google/gemini-2.5-pro-exp-03-25"
-            ):
-                continue
-
-            models_data.append(model)
-
-        return models_data
-
-
-def main() -> None:
-    source_filter = SOURCE if SOURCE and SOURCE.strip() else None
-    models = fetch_openrouter_models(source_filter=source_filter)
-
-    print(f"Writing {len(models)} models to {OUTPUT_FILE}")
-
-    with open(OUTPUT_FILE, "w") as f:
-        json.dump({"models": models}, f, indent=4)
-
-
-if __name__ == "__main__":
-    main()

tests/integration/conftest.py

@@ -66,7 +66,7 @@ os.environ.update(test_env)
 os.environ.pop("ADMIN_PASSWORD", None)
 
 
-from routstr.core.db import ApiKey, get_session  # noqa: E402
+from routstr.core.db import TemporaryCredit, get_session  # noqa: E402
 from routstr.core.main import app, lifespan  # noqa: E402
 
 
@@ -113,7 +113,7 @@ class TestmintWallet:
         self.refund_history: List[Dict[str, Any]] = []
 
     async def init(self) -> None:
-        """Initialize the sixty_nuts wallet"""
+        """Initialize the cashu wallet"""
         # In mock mode, we don't actually create a real wallet
         # This is just a placeholder for the mock implementation
         self.wallet = None
@@ -280,7 +280,7 @@ class TestmintWallet:
         return 100000  # 100k sats
 
     async def credit_balance(
-        self, cashu_token: str, key: ApiKey, session: AsyncSession
+        self, cashu_token: str, key: TemporaryCredit, session: AsyncSession
     ) -> int:
         """Credit balance to API key - test implementation"""
         try:
@@ -301,9 +301,9 @@ class TestmintWallet:
 
             # Use atomic update to avoid lost update problem in concurrent scenarios
             stmt = (
-                update(ApiKey)
-                .where(col(ApiKey.hashed_key) == key.hashed_key)
-                .values(balance=ApiKey.balance + amount_msat)
+                update(TemporaryCredit)
+                .where(col(TemporaryCredit.hashed_key) == key.hashed_key)
+                .values(balance=TemporaryCredit.balance + amount_msat)
             )
             await session.execute(stmt)
             await session.commit()
@@ -390,7 +390,7 @@ class DatabaseSnapshot:
     async def capture(self) -> Dict[str, List[Dict]]:
         """Capture current database state"""
         # Get all API keys with their data
-        result = await self.session.execute(select(ApiKey))
+        result = await self.session.execute(select(TemporaryCredit))
         api_keys = result.scalars().all()
 
         snapshot = {
@@ -398,10 +398,8 @@ class DatabaseSnapshot:
                 {
                     "hashed_key": key.hashed_key,
                     "balance": key.balance,
-                    "total_spent": key.total_spent,
-                    "total_requests": key.total_requests,
                     "refund_address": key.refund_address,
-                    "key_expiry_time": key.key_expiry_time,
+                    "refund_expiration_time": key.refund_expiration_time,
                 }
                 for key in api_keys
             ]
@@ -447,10 +445,8 @@ class DatabaseSnapshot:
 
                 for field in [
                     "balance",
-                    "total_spent",
-                    "total_requests",
                     "refund_address",
-                    "key_expiry_time",
+                    "refund_expiration_time",
                 ]:
                     if old[field] != new[field]:
                         changes[field] = {
@@ -522,18 +518,19 @@ async def integration_app(
         with (
             patch("routstr.core.db.engine", integration_engine),
             patch.object(_settings, "cashu_mints", [mint_url]),
-            patch("routstr.wallet.credit_balance", testmint_wallet.credit_balance),
-            patch("routstr.wallet.send_token", testmint_wallet.send_token),
-            patch("routstr.wallet.send_to_lnurl", testmint_wallet.send_to_lnurl),
-            patch("routstr.wallet.recieve_token", testmint_wallet.redeem_token),
-            patch("routstr.wallet.get_balance", testmint_wallet.get_balance),
+            patch(
+                "routstr.payment.wallet.credit_balance", testmint_wallet.credit_balance
+            ),
+            patch("routstr.payment.wallet.send_token", testmint_wallet.send_token),
+            patch("routstr.payment.wallet.recieve_token", testmint_wallet.redeem_token),
+            patch("routstr.payment.wallet.get_balance", testmint_wallet.get_balance),
             patch("routstr.balance.send_token", testmint_wallet.send_token),
             patch("routstr.balance.send_to_lnurl", testmint_wallet.send_to_lnurl),
             patch("websockets.connect") as mock_websockets,
             patch("routstr.payment.price.btc_usd_price", return_value=50000.0),
             patch("routstr.payment.price.sats_usd_price", return_value=0.0005),
             patch(
-                "routstr.payment.helpers.calculate_discounted_max_cost",
+                "routstr.payment.cost.calculate_discounted_max_cost",
                 side_effect=_passthrough_discount,
             ),
         ):
@@ -705,8 +702,8 @@ async def background_tasks_controller() -> AsyncGenerator[Any, None]:
     original_periodic_payout: Optional[Callable] = None
 
     try:
-        from routstr.payment.models import update_sats_pricing
-        from routstr.wallet import periodic_payout
+        from routstr.models.models import update_sats_pricing
+        from routstr.payment.wallet import periodic_payout
 
         async def controlled_update_pricing() -> None:
             while not controller.cancelled:

tests/integration/real_testmint.py

@@ -1,18 +1,18 @@
 """
 Real Cashu mint integration for integration tests.
 
-This module provides a real sixty_nuts Wallet implementation that can be used
+This module provides a real cashu Wallet implementation that can be used
 with an actual Cashu mint instance for more thorough integration testing.
 """
 
 import os
 from typing import Optional, Tuple
 
-from sixty_nuts import Wallet
+from cashu.wallet.wallet import Wallet
 
 
 class RealMintWallet:
-    """Real Cashu mint wallet using sixty_nuts library"""
+    """Real Cashu mint wallet using cashu library"""
 
     def __init__(self, mint_url: str, nsec: str):
         self.mint_url = mint_url
@@ -22,7 +22,12 @@ class RealMintWallet:
     async def init(self) -> None:
         """Initialize the wallet connection"""
         if not self._wallet:
-            self._wallet = await Wallet.create(nsec=self.nsec)
+            self._wallet = await Wallet.with_db(
+                self.mint_url,
+                db="sqlite+aiosqlite:///:memory:",
+                load_all_keysets=True,
+                unit="sat",
+            )
 
     @property
     def wallet(self) -> Wallet:

tests/integration/test_background_tasks.py

@@ -9,9 +9,9 @@ from unittest.mock import AsyncMock, patch
 
 import pytest
 
-from routstr.core.db import ApiKey
-from routstr.payment.models import Model, Pricing, update_sats_pricing
-from routstr.wallet import periodic_payout
+from routstr.core.db import TemporaryCredit
+from routstr.models.models import Model, Pricing, update_sats_pricing
+from routstr.payment.wallet import periodic_payout
 
 
 @pytest.mark.asyncio
@@ -201,12 +201,12 @@ class TestRefundCheckTask:
     ) -> None:
         """Test that expired keys with balance and refund address are refunded"""
         # Create an expired API key with balance
-        expired_key = ApiKey(
+        expired_key = TemporaryCredit(
             hashed_key="expired_test_key",
             balance=5000,  # 5 sats in msats
             refund_address="lnurl1test",
-            key_expiry_time=int(time.time()) - 3600,  # Expired 1 hour ago
-            created_at=datetime.utcnow() - timedelta(days=1),
+            refund_expiration_time=int(time.time()) - 3600,  # Expired 1 hour ago
+            created=datetime.utcnow() - timedelta(days=1),
         )
         integration_session.add(expired_key)
         await integration_session.commit()
@@ -214,7 +214,7 @@ class TestRefundCheckTask:
         # Mock the wallet send_to_lnurl method and get_session
         with (
             patch(
-                "routstr.wallet.send_to_lnurl", AsyncMock(return_value=5)
+                "routstr.payment.lnurl.send_to_lnurl", AsyncMock(return_value=5)
             ) as mock_send_to_lnurl,
             patch("routstr.core.db.get_session") as mock_get_session,
         ):
@@ -233,8 +233,8 @@ class TestRefundCheckTask:
             if (
                 expired_key.balance > 0
                 and expired_key.refund_address
-                and expired_key.key_expiry_time
-                and expired_key.key_expiry_time < current_time
+                and expired_key.refund_expiration_time
+                and expired_key.refund_expiration_time < current_time
             ):
                 # Call wallet send_to_lnurl to trigger the refund
                 amount_sats = expired_key.balance // 1000
@@ -258,12 +258,12 @@ class TestRefundCheckTask:
         """Test that refund check continues after mint errors"""
         # Create multiple expired keys
         for i in range(3):
-            key = ApiKey(
+            key = TemporaryCredit(
                 hashed_key=f"expired_key_{i}",
                 balance=1000 * (i + 1),
                 refund_address=f"lnurl{i}",
-                key_expiry_time=int(time.time()) - 3600,
-                created_at=datetime.utcnow(),
+                refund_expiration_time=int(time.time()) - 3600,
+                created=datetime.utcnow(),
             )
             integration_session.add(key)
         await integration_session.commit()
@@ -279,7 +279,7 @@ class TestRefundCheckTask:
 
         with (
             patch(
-                "routstr.wallet.send_to_lnurl", mock_send_to_lnurl
+                "routstr.payment.lnurl.send_to_lnurl", mock_send_to_lnurl
             ) as mock_send_to_lnurl_patch,
             patch("routstr.core.db.get_session") as mock_get_session,
         ):
@@ -293,15 +293,15 @@ class TestRefundCheckTask:
             current_time = int(time.time())
             from sqlalchemy import select as sa_select
 
-            result = await integration_session.execute(sa_select(ApiKey))
+            result = await integration_session.execute(sa_select(TemporaryCredit))
             keys = result.scalars().all()
 
             for key in keys:
                 if (
                     key.balance > 0
                     and key.refund_address
-                    and key.key_expiry_time
-                    and key.key_expiry_time < current_time
+                    and key.refund_expiration_time
+                    and key.refund_expiration_time < current_time
                 ):
                     amount_sats = key.balance // 1000
                     try:
@@ -352,21 +352,21 @@ class TestRefundCheckTask:
 
         current_time = int(time.time())
         for data in keys_data:
-            key = ApiKey(
+            key = TemporaryCredit(
                 hashed_key=data["hashed_key"],
                 balance=data["balance"],
                 refund_address=data["refund_address"],
-                key_expiry_time=current_time - 3600
+                refund_expiration_time=current_time - 3600
                 if data["expired"]
                 else current_time + 3600,
-                created_at=datetime.utcnow(),
+                created=datetime.utcnow(),
             )
             integration_session.add(key)
         await integration_session.commit()
 
         with (
             patch(
-                "routstr.wallet.send_to_lnurl", AsyncMock(return_value=1)
+                "routstr.payment.lnurl.send_to_lnurl", AsyncMock(return_value=1)
             ) as mock_send_to_lnurl,
             patch("routstr.core.db.get_session") as mock_get_session,
         ):
@@ -382,15 +382,15 @@ class TestRefundCheckTask:
             current_time = int(time.time())
             from sqlalchemy import select as sa_select
 
-            result = await integration_session.execute(sa_select(ApiKey))
+            result = await integration_session.execute(sa_select(TemporaryCredit))
             keys = result.scalars().all()
 
             for key in keys:
                 if (
                     key.balance > 0
                     and key.refund_address
-                    and key.key_expiry_time
-                    and key.key_expiry_time < current_time
+                    and key.refund_expiration_time
+                    and key.refund_expiration_time < current_time
                 ):
                     amount_sats = key.balance // 1000
                     await mock_send_to_lnurl(key.refund_address, amount=amount_sats)
@@ -410,7 +410,7 @@ class TestRefundCheckTask:
             # Check final state
             from sqlalchemy import select as sa_select
 
-            result = await integration_session.execute(sa_select(ApiKey))
+            result = await integration_session.execute(sa_select(TemporaryCredit))
             remaining_keys_list = result.scalars().all()
             remaining_ids = [k.hashed_key for k in remaining_keys_list]
 
@@ -454,10 +454,10 @@ class TestPeriodicPayoutTask:
         for i in range(5):
             balance = 10000 * (i + 1)  # 10, 20, 30, 40, 50 sats
             total_user_balance += balance
-            key = ApiKey(
+            key = TemporaryCredit(
                 hashed_key=f"user_key_{i}",
                 balance=balance,
-                created_at=datetime.utcnow(),
+                created=datetime.utcnow(),
             )
             integration_session.add(key)
         await integration_session.commit()
@@ -466,9 +466,12 @@ class TestPeriodicPayoutTask:
         wallet_balance = 200000  # 200 sats total
 
         with (
-            patch("routstr.wallet.get_balance", AsyncMock(return_value=wallet_balance)),
             patch(
-                "routstr.wallet.send_to_lnurl", AsyncMock(return_value=None)
+                "routstr.payment.wallet.get_balance",
+                AsyncMock(return_value=wallet_balance),
+            ),
+            patch(
+                "routstr.payment.lnurl.send_to_lnurl", AsyncMock(return_value=None)
             ) as mock_send_to_lnurl,
         ):
             # Mock environment variables
@@ -481,7 +484,7 @@ class TestPeriodicPayoutTask:
                 },
             ):
                 # Call periodic_payout directly (pay_out was renamed/refactored)
-                from routstr.wallet import periodic_payout
+                from routstr.payment.wallet import periodic_payout
 
                 await periodic_payout()
 
@@ -501,7 +504,7 @@ class TestPeriodicPayoutTask:
     #     key = ApiKey(
     #         hashed_key="single_user",
     #         balance=50000,  # 50 sats
-    #         created_at=datetime.utcnow(),
+    #         created=datetime.utcnow(),
     #     )
     #     integration_session.add(key)
     #     await integration_session.commit()
@@ -538,7 +541,7 @@ class TestPeriodicPayoutTask:
     #     key = ApiKey(
     #         hashed_key="low_revenue_user",
     #         balance=95000,  # 95 sats
-    #         created_at=datetime.utcnow(),
+    #         created=datetime.utcnow(),
     #     )
     #     integration_session.add(key)
     #     await integration_session.commit()
@@ -648,14 +651,14 @@ class TestTaskInteractions:
         """Test that database operations don't deadlock during concurrent task execution"""
         # Create test data
         for i in range(10):
-            key = ApiKey(
+            key = TemporaryCredit(
                 hashed_key=f"concurrent_key_{i}",
                 balance=1000 * i,
                 refund_address=f"lnurl{i}" if i % 2 == 0 else None,
-                key_expiry_time=int(time.time()) - 3600
+                refund_expiration_time=int(time.time()) - 3600
                 if i % 3 == 0
                 else int(time.time()) + 3600,
-                created_at=datetime.utcnow(),
+                created=datetime.utcnow(),
             )
             integration_session.add(key)
         await integration_session.commit()
@@ -664,14 +667,14 @@ class TestTaskInteractions:
         async def read_operation() -> int:
             from sqlalchemy import select as sa_select
 
-            result = await integration_session.execute(sa_select(ApiKey))
+            result = await integration_session.execute(sa_select(TemporaryCredit))
             return len(result.scalars().all())
 
         async def write_operation(key_id: int) -> None:
             from sqlalchemy import select as sa_select
 
-            stmt = sa_select(ApiKey).where(
-                ApiKey.hashed_key == f"concurrent_key_{key_id}"  # type: ignore[arg-type]
+            stmt = sa_select(TemporaryCredit).where(
+                TemporaryCredit.hashed_key == f"concurrent_key_{key_id}"  # type: ignore[arg-type]
             )
             result = await integration_session.execute(stmt)
             key = result.scalar_one_or_none()
@@ -708,14 +711,16 @@ class TestTaskInteractions:
         # Patch the actual task functions
         with (
             patch(
-                "routstr.payment.models.update_sats_pricing",
+                "routstr.models.models.update_sats_pricing",
                 lambda: task_with_cleanup("pricing"),
             ),
             patch(
-                "routstr.wallet.periodic_payout", lambda: task_with_cleanup("refund")
+                "routstr.payment.wallet.periodic_payout",
+                lambda: task_with_cleanup("refund"),
             ),
             patch(
-                "routstr.wallet.periodic_payout", lambda: task_with_cleanup("payout")
+                "routstr.payment.wallet.periodic_payout",
+                lambda: task_with_cleanup("payout"),
             ),
         ):
             # Start all tasks

tests/integration/test_database_consistency.py

@@ -11,7 +11,7 @@ from sqlalchemy.exc import IntegrityError
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlmodel import select
 
-from routstr.core.db import ApiKey
+from routstr.core.db import TemporaryCredit
 
 
 class TestTransactionAtomicity:
@@ -34,7 +34,7 @@ class TestTransactionAtomicity:
             api_key_header[3:] if api_key_header.startswith("sk-") else api_key_header
         )
 
-        stmt = select(ApiKey).where(ApiKey.hashed_key == api_key_hash)  # type: ignore[arg-type]
+        stmt = select(TemporaryCredit).where(TemporaryCredit.hashed_key == api_key_hash)  # type: ignore[arg-type]
         result = await integration_session.execute(stmt)
         api_key = result.scalar_one()
         initial_balance = api_key.balance
@@ -47,7 +47,9 @@ class TestTransactionAtomicity:
             try:
                 # Get api key in new session
                 result = await test_session.execute(
-                    select(ApiKey).where(ApiKey.hashed_key == api_key_hash)  # type: ignore[arg-type]
+                    select(TemporaryCredit).where(
+                        TemporaryCredit.hashed_key == api_key_hash
+                    )  # type: ignore[arg-type]
                 )
                 test_api_key = result.scalar_one()
 
@@ -72,14 +74,14 @@ class TestTransactionAtomicity:
 
         try:
             await integration_session.execute(
-                update(ApiKey)
-                .where(ApiKey.hashed_key == api_key_hash)  # type: ignore[arg-type]
-                .values(balance=ApiKey.balance - 1000)
+                update(TemporaryCredit)
+                .where(TemporaryCredit.hashed_key == api_key_hash)  # type: ignore[arg-type]
+                .values(balance=TemporaryCredit.balance - 1000)
             )
             # Force a constraint violation or error
             await integration_session.execute(
-                update(ApiKey)
-                .where(ApiKey.hashed_key == "non_existent_key")  # type: ignore[arg-type]
+                update(TemporaryCredit)
+                .where(TemporaryCredit.hashed_key == "non_existent_key")  # type: ignore[arg-type]
                 .values(balance=-1)  # This should fail
             )
             await integration_session.commit()
@@ -108,13 +110,13 @@ class TestTransactionAtomicity:
             api_key_header[3:] if api_key_header.startswith("sk-") else api_key_header
         )
 
-        stmt = select(ApiKey).where(ApiKey.hashed_key == api_key_hash)  # type: ignore[arg-type]
+        stmt = select(TemporaryCredit).where(TemporaryCredit.hashed_key == api_key_hash)  # type: ignore[arg-type]
         result = await integration_session.execute(stmt)
         api_key = result.scalar_one()
         initial_balance = api_key.balance
 
         # Mock wallet to fail after token validation
-        with patch("routstr.wallet.send_token") as mock_wallet_func:
+        with patch("routstr.payment.wallet.send_token") as mock_wallet_func:
             mock_proof = MagicMock()
             mock_proof.amount = 1000
             mock_wallet = AsyncMock()
@@ -158,7 +160,7 @@ class TestTransactionAtomicity:
         )
 
         # Set a known balance
-        stmt = select(ApiKey).where(ApiKey.hashed_key == api_key_hash)  # type: ignore[arg-type]
+        stmt = select(TemporaryCredit).where(TemporaryCredit.hashed_key == api_key_hash)  # type: ignore[arg-type]
         result = await integration_session.execute(stmt)
         api_key = result.scalar_one()
         api_key.balance = 10000
@@ -166,12 +168,12 @@ class TestTransactionAtomicity:
 
         # Simulate concurrent balance updates through direct database operations
         async def update_balance(session: AsyncSession, amount: int) -> bool:
-            stmt = select(ApiKey).where(ApiKey.hashed_key == api_key_hash)  # type: ignore[arg-type]
+            stmt = select(TemporaryCredit).where(
+                TemporaryCredit.hashed_key == api_key_hash
+            )  # type: ignore[arg-type]
             result = await session.execute(stmt)
             key = result.scalar_one()
             key.balance -= amount
-            key.total_spent += amount
-            key.total_requests += 1
             try:
                 await session.commit()
                 return True
@@ -248,7 +250,7 @@ class TestConcurrentOperations:
         )
 
         # Set initial balance
-        stmt = select(ApiKey).where(ApiKey.hashed_key == api_key_hash)  # type: ignore[arg-type]
+        stmt = select(TemporaryCredit).where(TemporaryCredit.hashed_key == api_key_hash)  # type: ignore[arg-type]
         result = await integration_session.execute(stmt)
         api_key = result.scalar_one()
         initial_balance = 5000
@@ -256,7 +258,7 @@ class TestConcurrentOperations:
         await integration_session.commit()
 
         # Mock wallet for topup
-        with patch("routstr.wallet.send_token") as mock_wallet_func:
+        with patch("routstr.payment.wallet.send_token") as mock_wallet_func:
             mock_proof = MagicMock()
             mock_proof.amount = 2000
             mock_wallet = AsyncMock()
@@ -323,12 +325,10 @@ class TestConcurrentOperations:
         )
 
         # Set a specific balance
-        stmt = select(ApiKey).where(ApiKey.hashed_key == api_key_hash)  # type: ignore[arg-type]
+        stmt = select(TemporaryCredit).where(TemporaryCredit.hashed_key == api_key_hash)  # type: ignore[arg-type]
         result = await integration_session.execute(stmt)
         api_key = result.scalar_one()
         api_key.balance = 1000
-        api_key.total_spent = 0
-        api_key.total_requests = 0
         await integration_session.commit()
 
         # Create a controlled race condition scenario
@@ -336,7 +336,9 @@ class TestConcurrentOperations:
 
         async def check_and_update_balance() -> bool:
             # Read current balance
-            stmt = select(ApiKey).where(ApiKey.hashed_key == api_key_hash)  # type: ignore[arg-type]
+            stmt = select(TemporaryCredit).where(
+                TemporaryCredit.hashed_key == api_key_hash
+            )  # type: ignore[arg-type]
             result = await integration_session.execute(stmt)
             current_api_key = result.scalar_one()
             current_balance = current_api_key.balance
@@ -347,8 +349,6 @@ class TestConcurrentOperations:
 
             # Try to update based on read value
             current_api_key.balance = current_balance - 100
-            current_api_key.total_spent += 100
-            current_api_key.total_requests += 1
 
             try:
                 await integration_session.commit()
@@ -371,8 +371,6 @@ class TestConcurrentOperations:
         # Final balance should reflect successful updates
         expected_balance = 1000 - (successful_updates * 100)
         assert api_key.balance == expected_balance
-        assert api_key.total_spent == successful_updates * 100
-        assert api_key.total_requests == successful_updates
 
 
 class TestDataIntegrity:
@@ -396,7 +394,7 @@ class TestDataIntegrity:
         )
 
         # Set low balance
-        stmt = select(ApiKey).where(ApiKey.hashed_key == api_key_hash)  # type: ignore[arg-type]
+        stmt = select(TemporaryCredit).where(TemporaryCredit.hashed_key == api_key_hash)  # type: ignore[arg-type]
         result = await integration_session.execute(stmt)
         api_key = result.scalar_one()
         api_key.balance = 0
@@ -431,9 +429,7 @@ class TestDataIntegrity:
         )
 
         # Try to manually insert duplicate key with same hash
-        duplicate_key = ApiKey(
-            hashed_key=api_key_hash, balance=5000, total_spent=0, total_requests=0
-        )
+        duplicate_key = TemporaryCredit(hashed_key=api_key_hash, balance=5000)
 
         integration_session.add(duplicate_key)
 
@@ -481,20 +477,13 @@ class TestDataIntegrity:
             api_key_header[3:] if api_key_header.startswith("sk-") else api_key_header
         )
 
-        stmt = select(ApiKey).where(ApiKey.hashed_key == api_key_hash)  # type: ignore[arg-type]
+        stmt = select(TemporaryCredit).where(TemporaryCredit.hashed_key == api_key_hash)  # type: ignore[arg-type]
         result = await integration_session.execute(stmt)
         api_key = result.scalar_one()
 
         # Test setting invalid values directly
         # These should maintain integrity
         assert api_key.balance >= 0
-        assert api_key.total_spent >= 0
-        assert api_key.total_requests >= 0
-
-        # Verify calculations are consistent
-        if api_key.total_requests > 0:
-            average_cost = api_key.total_spent / api_key.total_requests
-            assert average_cost >= 0
 
 
 class TestPerformance:
@@ -522,7 +511,7 @@ class TestPerformance:
             operation_times["select"].append((end - start) * 1000)  # Convert to ms
 
         # Test UPDATE performance (via topup)
-        with patch("routstr.wallet.send_token") as mock_wallet_func:
+        with patch("routstr.payment.wallet.send_token") as mock_wallet_func:
             mock_proof = MagicMock()
             mock_proof.amount = 100
             mock_wallet = AsyncMock()
@@ -607,7 +596,7 @@ class TestPerformance:
 
         # Primary key lookup should be fast
         start = time.time()
-        stmt = select(ApiKey).where(ApiKey.hashed_key == api_key_hash)  # type: ignore[arg-type]
+        stmt = select(TemporaryCredit).where(TemporaryCredit.hashed_key == api_key_hash)  # type: ignore[arg-type]
         result = await integration_session.execute(stmt)
         api_key = result.scalar_one()
         end = time.time()

tests/integration/test_error_handling_edge_cases.py

@@ -11,7 +11,7 @@ from httpx import ASGITransport, AsyncClient, ConnectError
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlmodel import select
 
-from routstr.core.db import ApiKey
+from routstr.core.db import TemporaryCredit
 
 
 class TestNetworkFailureScenarios:
@@ -27,7 +27,7 @@ class TestNetworkFailureScenarios:
         # Patch the wallet send function to simulate failure across all modules
         with (
             patch(
-                "routstr.wallet.send_token",
+                "routstr.payment.wallet.send_token",
                 AsyncMock(side_effect=ConnectError("Mint service unavailable")),
             ),
             patch(
@@ -422,7 +422,7 @@ class TestRecoveryScenarios:
             api_key_header[3:] if api_key_header.startswith("sk-") else api_key_header
         )
 
-        stmt = select(ApiKey).where(ApiKey.hashed_key == api_key_hash)  # type: ignore[arg-type]
+        stmt = select(TemporaryCredit).where(TemporaryCredit.hashed_key == api_key_hash)  # type: ignore[arg-type]
         result = await integration_session.execute(stmt)
         initial_key = result.scalar_one()
         initial_balance = initial_key.balance
@@ -458,17 +458,15 @@ class TestRecoveryScenarios:
             api_key_header[3:] if api_key_header.startswith("sk-") else api_key_header
         )
 
-        stmt = select(ApiKey).where(ApiKey.hashed_key == api_key_hash)  # type: ignore[arg-type]
+        stmt = select(TemporaryCredit).where(TemporaryCredit.hashed_key == api_key_hash)  # type: ignore[arg-type]
         result = await integration_session.execute(stmt)
         api_key = result.scalar_one()
         initial_balance = api_key.balance
-        initial_requests = api_key.total_requests
 
         # Simulate operations that might be interrupted
         try:
             # Start a transaction
             api_key.reserved_balance += 1000
-            api_key.total_requests += 1
             # Don't commit - simulate crash
             raise Exception("Simulated database crash")
         except Exception:
@@ -478,7 +476,6 @@ class TestRecoveryScenarios:
         # Verify state is consistent after "recovery"
         await integration_session.refresh(api_key)
         assert api_key.balance == initial_balance
-        assert api_key.total_requests == initial_requests
 
     @pytest.mark.asyncio
     async def test_state_consistency_after_failures(
@@ -527,10 +524,7 @@ class TestRecoveryScenarios:
             for mod in diff["api_keys"]["modified"]:
                 # Only acceptable changes are request counts
                 for field, change in mod["changes"].items():
-                    if field == "total_requests":
-                        # Request count might increase
-                        assert change["delta"] >= 0
-                    elif field == "balance":
+                    if field == "balance":
                         # Balance should not decrease from failed operations
                         assert change["delta"] >= 0
                     else:
@@ -641,12 +635,10 @@ class TestEdgeCaseCombinations:
         api_key_hash = test_key[3:]  # Remove sk- prefix
 
         # Create the API key with only 500 msats (less than one request cost)
-        new_key = ApiKey(
+        new_key = TemporaryCredit(
             hashed_key=api_key_hash,
             balance=500,  # Less than fixed cost per request (1000 msats)
             reserved_balance=0,
-            total_spent=0,
-            total_requests=0,
         )
         integration_session.add(new_key)
         await integration_session.commit()
@@ -687,7 +679,7 @@ class TestEdgeCaseCombinations:
         assert insufficient_funds_count > 0
 
         # Balance should never go negative
-        stmt = select(ApiKey).where(ApiKey.hashed_key == api_key_hash)  # type: ignore[arg-type]
+        stmt = select(TemporaryCredit).where(TemporaryCredit.hashed_key == api_key_hash)  # type: ignore[arg-type]
         result = await integration_session.execute(stmt)
         final_key = result.scalar_one()
         assert final_key.balance >= 0

tests/integration/test_performance_load.py

@@ -142,6 +142,44 @@ class TestPerformanceBaseline:
                 f"  P99: {sorted(response_times)[int(len(response_times) * 0.99)]:.2f}ms"
             )
 
+    @pytest.mark.asyncio
+    async def test_database_query_performance(
+        self, integration_session: Any, db_snapshot: Any
+    ) -> None:
+        """Test database operation performance"""
+        from sqlmodel import select
+
+        from routstr.core.db import TemporaryCredit
+
+        # Create test data
+        for i in range(100):
+            key = TemporaryCredit(
+                hashed_key=f"test_key_{i}",
+                balance=1000000,
+            )
+            integration_session.add(key)
+        await integration_session.commit()
+
+        # Test query performance
+        query_times = []
+
+        for _ in range(100):
+            start = time.time()
+            result = await integration_session.execute(
+                select(TemporaryCredit).where(TemporaryCredit.balance > 0)  # type: ignore[arg-type]
+            )
+            _ = result.all()
+            duration = (time.time() - start) * 1000
+            query_times.append(duration)
+
+        # All queries should complete < 100ms
+        assert max(query_times) < 100, (
+            f"Max query time {max(query_times)}ms exceeds 100ms limit"
+        )
+        print("\nDatabase query performance:")
+        print(f"  Mean: {statistics.mean(query_times):.2f}ms")
+        print(f"  Max: {max(query_times):.2f}ms")
+
 
 @pytest.mark.integration
 @pytest.mark.slow

tests/integration/test_provider_management.py

@@ -9,7 +9,7 @@ from unittest.mock import patch
 import pytest
 from httpx import AsyncClient
 
-from routstr.discovery import _PROVIDERS_CACHE
+from routstr.nostr.discovery import _PROVIDERS_CACHE
 
 from .utils import PerformanceValidator, ResponseValidator
 
@@ -62,9 +62,9 @@ async def test_providers_endpoint_default_response(
     }
 
     with patch(
-        "routstr.discovery.query_nostr_relay_for_providers", return_value=mock_events
+        "routstr.nostr.discovery.query_nostr_relay_for_providers", return_value=mock_events
     ):
-        with patch("routstr.discovery.fetch_provider_health") as mock_fetch:
+        with patch("routstr.nostr.discovery.fetch_provider_health") as mock_fetch:
             # Configure mock to return appropriate responses
             mock_fetch.side_effect = lambda url: mock_fetch_responses.get(
                 url, {"status_code": 500, "json": {"error": "Unknown provider"}}
@@ -126,9 +126,9 @@ async def test_providers_endpoint_with_include_json(
     }
 
     with patch(
-        "routstr.discovery.query_nostr_relay_for_providers", return_value=mock_events
+        "routstr.nostr.discovery.query_nostr_relay_for_providers", return_value=mock_events
     ):
-        with patch("routstr.discovery.fetch_provider_health") as mock_fetch:
+        with patch("routstr.nostr.discovery.fetch_provider_health") as mock_fetch:
             mock_fetch.return_value = {
                 "status_code": 200,
                 "json": mock_provider_response,
@@ -200,9 +200,9 @@ async def test_providers_data_structure_validation(
     }
 
     with patch(
-        "routstr.discovery.query_nostr_relay_for_providers", return_value=mock_events
+        "routstr.nostr.discovery.query_nostr_relay_for_providers", return_value=mock_events
     ):
-        with patch("routstr.discovery.fetch_provider_health") as mock_fetch:
+        with patch("routstr.nostr.discovery.fetch_provider_health") as mock_fetch:
             mock_fetch.return_value = mock_health_response
 
             response = await integration_client.get("/v1/providers/?include_json=true")
@@ -247,7 +247,7 @@ async def test_providers_endpoint_no_providers_found(
     ]
 
     with patch(
-        "routstr.discovery.query_nostr_relay_for_providers", return_value=mock_events
+        "routstr.nostr.discovery.query_nostr_relay_for_providers", return_value=mock_events
     ):
         response = await integration_client.get("/v1/providers/")
 
@@ -308,10 +308,10 @@ async def test_providers_endpoint_offline_providers(
             }
 
     with patch(
-        "routstr.discovery.query_nostr_relay_for_providers", return_value=mock_events
+        "routstr.nostr.discovery.query_nostr_relay_for_providers", return_value=mock_events
     ):
         with patch(
-            "routstr.discovery.fetch_provider_health",
+            "routstr.nostr.discovery.fetch_provider_health",
             side_effect=mock_fetch_provider_health,
         ):
             response = await integration_client.get("/v1/providers/?include_json=true")
@@ -377,9 +377,9 @@ async def test_providers_endpoint_duplicate_urls(
     ]
 
     with patch(
-        "routstr.discovery.query_nostr_relay_for_providers", return_value=mock_events
+        "routstr.nostr.discovery.query_nostr_relay_for_providers", return_value=mock_events
     ):
-        with patch("routstr.discovery.fetch_provider_health") as mock_fetch:
+        with patch("routstr.nostr.discovery.fetch_provider_health") as mock_fetch:
             mock_fetch.return_value = {
                 "status_code": 200,
                 "endpoint": "root",
@@ -416,7 +416,7 @@ async def test_providers_endpoint_nostr_relay_failures(
         raise Exception("Connection to relay failed")
 
     with patch(
-        "routstr.discovery.query_nostr_relay_for_providers", side_effect=failing_query
+        "routstr.nostr.discovery.query_nostr_relay_for_providers", side_effect=failing_query
     ):
         response = await integration_client.get("/v1/providers/")
 
@@ -454,9 +454,9 @@ async def test_providers_endpoint_malformed_urls(
     ]
 
     with patch(
-        "routstr.discovery.query_nostr_relay_for_providers", return_value=mock_events
+        "routstr.nostr.discovery.query_nostr_relay_for_providers", return_value=mock_events
     ):
-        with patch("routstr.discovery.fetch_provider_health") as mock_fetch:
+        with patch("routstr.nostr.discovery.fetch_provider_health") as mock_fetch:
             mock_fetch.return_value = {"status_code": 200, "json": {"status": "online"}}
 
             response = await integration_client.get("/v1/providers/")
@@ -486,9 +486,9 @@ async def test_providers_endpoint_response_format(
     ]
 
     with patch(
-        "routstr.discovery.query_nostr_relay_for_providers", return_value=mock_events
+        "routstr.nostr.discovery.query_nostr_relay_for_providers", return_value=mock_events
     ):
-        with patch("routstr.discovery.fetch_provider_health") as mock_fetch:
+        with patch("routstr.nostr.discovery.fetch_provider_health") as mock_fetch:
             mock_fetch.return_value = {"status_code": 200, "json": {"status": "online"}}
 
             # Test default format
@@ -536,9 +536,9 @@ async def test_providers_endpoint_performance(integration_client: AsyncClient) -
     validator = PerformanceValidator()
 
     with patch(
-        "routstr.discovery.query_nostr_relay_for_providers", return_value=mock_events
+        "routstr.nostr.discovery.query_nostr_relay_for_providers", return_value=mock_events
     ):
-        with patch("routstr.discovery.fetch_provider_health") as mock_fetch:
+        with patch("routstr.nostr.discovery.fetch_provider_health") as mock_fetch:
             mock_fetch.return_value = {"status_code": 200, "json": {"status": "online"}}
 
             # Test multiple requests
@@ -576,9 +576,9 @@ async def test_providers_endpoint_concurrent_requests(
     ]
 
     with patch(
-        "routstr.discovery.query_nostr_relay_for_providers", return_value=mock_events
+        "routstr.nostr.discovery.query_nostr_relay_for_providers", return_value=mock_events
     ):
-        with patch("routstr.discovery.fetch_provider_health") as mock_fetch:
+        with patch("routstr.nostr.discovery.fetch_provider_health") as mock_fetch:
             mock_fetch.return_value = {"status_code": 200, "json": {"status": "online"}}
 
             # Create concurrent requests
@@ -618,9 +618,9 @@ async def test_providers_endpoint_parameter_validation(
     ]
 
     with patch(
-        "routstr.discovery.query_nostr_relay_for_providers", return_value=mock_events
+        "routstr.nostr.discovery.query_nostr_relay_for_providers", return_value=mock_events
     ):
-        with patch("routstr.discovery.fetch_provider_health") as mock_fetch:
+        with patch("routstr.nostr.discovery.fetch_provider_health") as mock_fetch:
             mock_fetch.return_value = {"status_code": 200, "json": {"status": "online"}}
 
             # Test various parameter values
@@ -670,9 +670,9 @@ async def test_no_database_changes_during_provider_operations(
     ]
 
     with patch(
-        "routstr.discovery.query_nostr_relay_for_providers", return_value=mock_events
+        "routstr.nostr.discovery.query_nostr_relay_for_providers", return_value=mock_events
     ):
-        with patch("routstr.discovery.fetch_provider_health") as mock_fetch:
+        with patch("routstr.nostr.discovery.fetch_provider_health") as mock_fetch:
             mock_fetch.return_value = {"status_code": 200, "json": {"status": "online"}}
 
             # Make multiple requests with different parameters

tests/integration/test_proxy_get_endpoints.py

@@ -14,7 +14,7 @@ import pytest
 from httpx import AsyncClient
 from sqlmodel import select
 
-from routstr.core.db import ApiKey
+from routstr.core.db import TemporaryCredit
 
 from .utils import (
     ConcurrencyTester,
@@ -297,8 +297,8 @@ async def test_proxy_get_insufficient_balance(
     from sqlmodel import update
 
     await integration_session.execute(
-        update(ApiKey)
-        .where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+        update(TemporaryCredit)
+        .where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
         .values(balance=100)  # Only 0.1 sats
     )
     await integration_session.commit()
@@ -382,7 +382,7 @@ async def test_proxy_get_database_state_verification(
 
     # Get initial key state
     result = await integration_session.execute(
-        select(ApiKey).where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+        select(TemporaryCredit).where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
     )
     initial_key = result.scalar_one()
     initial_balance = initial_key.balance

tests/integration/test_real_mint.py

@@ -9,7 +9,7 @@ import os
 try:
     from .real_testmint import create_real_mint_wallet
 except ImportError:
-    # sixty_nuts not available, tests will be skipped
+    # cashu not available, tests will be skipped
     create_real_mint_wallet = None  # type: ignore
 
 
@@ -17,9 +17,9 @@ async def test_real_wallet() -> None:
     """Test basic operations with a real Cashu mint wallet"""
     print("Testing real Cashu mint wallet...")
 
-    # Check if sixty_nuts dependency is available
+    # Check if cashu dependency is available
     if create_real_mint_wallet is None:
-        print("sixty_nuts not available. Skipping real mint tests.")
+        print("cashu not available. Skipping real mint tests.")
         return
 
     # Check if real mint is enabled

tests/integration/test_reserved_balance_negative.py

@@ -7,7 +7,7 @@ import pytest
 from httpx import AsyncClient
 from sqlmodel.ext.asyncio.session import AsyncSession
 
-from routstr.core.db import ApiKey, create_session
+from routstr.core.db import TemporaryCredit, create_session
 
 
 @pytest.mark.asyncio
@@ -16,7 +16,7 @@ async def test_reserved_balance_never_negative(integration_client: AsyncClient)
 
     # Create a test API key with limited balance
     async with create_session() as session:
-        test_key = ApiKey(
+        test_key = TemporaryCredit(
             hashed_key="test_reserved_balance_key",
             balance=1000,  # 1 sat
             reserved_balance=0,
@@ -40,7 +40,7 @@ async def test_reserved_balance_never_negative(integration_client: AsyncClient)
 
     # Check reserved balance after failed request
     async with create_session() as session:
-        key = await session.get(ApiKey, "test_reserved_balance_key")
+        key = await session.get(TemporaryCredit, "test_reserved_balance_key")
         assert key is not None
         assert key.reserved_balance >= 0, (
             f"Reserved balance went negative: {key.reserved_balance}"
@@ -69,7 +69,7 @@ async def test_reserved_balance_never_negative(integration_client: AsyncClient)
 
     # Check final state
     async with create_session() as session:
-        key = await session.get(ApiKey, "test_reserved_balance_key")
+        key = await session.get(TemporaryCredit, "test_reserved_balance_key")
         assert key is not None
         assert key.reserved_balance >= 0, (
             f"Reserved balance went negative after concurrent requests: {key.reserved_balance}"
@@ -86,7 +86,7 @@ async def test_reserved_balance_with_successful_requests(
     # Create a test API key with more balance
     async with create_session() as session:
         unique_key = f"test_successful_key_{uuid.uuid4().hex[:8]}"
-        test_key = ApiKey(
+        test_key = TemporaryCredit(
             hashed_key=unique_key,
             balance=100000,  # 100 sats
             reserved_balance=0,
@@ -111,24 +111,23 @@ async def test_reserved_balance_with_successful_requests(
 
     # Check that reserved balance was properly adjusted
     async with create_session() as session:
-        key = await session.get(ApiKey, unique_key)
+        key = await session.get(TemporaryCredit, unique_key)
         assert key is not None
         assert key.reserved_balance >= 0, (
             f"Reserved balance went negative: {key.reserved_balance}"
         )
         # Check if the request was processed (might fail due to model pricing in test env)
         # The important part is that reserved_balance doesn't go negative
-        if key.total_spent > 0:
-            assert key.balance < 100000, (
-                "Balance should decrease after successful request"
-            )
+        if key.balance < 100000:
+            # Balance decreased
+            pass
         else:
             # Request failed, but reserved balance should still be non-negative
             assert key.balance == 100000, (
                 "Balance should remain unchanged if request failed"
             )
         print(
-            f"After successful request - Balance: {key.balance}, Reserved: {key.reserved_balance}, Spent: {key.total_spent}"
+            f"After successful request - Balance: {key.balance}, Reserved: {key.reserved_balance}"
         )
 
 
@@ -137,11 +136,11 @@ async def test_insufficient_reserved_balance_for_revert(
     integration_session: AsyncSession,
 ) -> None:
     """Test revert_pay_for_request behavior with insufficient reserved balance."""
-    from routstr.auth import revert_pay_for_request
+    from routstr.payment.helpers import revert_pay_for_request
 
     # Create key with zero reserved balance
     unique_key = f"test_revert_key_{uuid.uuid4().hex[:8]}"
-    test_key = ApiKey(
+    test_key = TemporaryCredit(
         hashed_key=unique_key,
         balance=1000,
         reserved_balance=0,
@@ -160,6 +159,3 @@ async def test_insufficient_reserved_balance_for_revert(
     assert test_key.reserved_balance == -100, (
         f"Expected reserved_balance to be -100, got: {test_key.reserved_balance}"
     )
-    assert test_key.total_requests == -1, (
-        f"Expected total_requests to be -1, got: {test_key.total_requests}"
-    )

tests/integration/test_wallet_authentication.py

@@ -11,7 +11,7 @@ import pytest
 from httpx import AsyncClient
 from sqlmodel import select
 
-from routstr.core.db import ApiKey
+from routstr.core.db import TemporaryCredit
 
 from .utils import (
     CashuTokenGenerator,
@@ -55,13 +55,11 @@ async def test_api_key_generation_valid_token(
     # Verify database state directly
     hashed_key = api_key[3:]  # Remove "sk-" prefix
     result = await integration_session.execute(
-        select(ApiKey).where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+        select(TemporaryCredit).where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
     )
     db_key = result.scalar_one()
 
     assert db_key.balance == amount * 1000
-    assert db_key.total_spent == 0
-    assert db_key.total_requests == 0
 
     # Verify the API key can be used for authentication
     integration_client.headers["Authorization"] = f"Bearer {api_key}"
@@ -93,15 +91,15 @@ async def test_api_key_generation_invalid_token(
         integration_client.headers["Authorization"] = f"Bearer {invalid_token}"
         response = await integration_client.get("/v1/wallet/info")
 
-        # Should fail with 401
-        assert response.status_code == 401, (
-            f"Token {invalid_token[:20]}... should be invalid"
+        # Should fail with 401 or 400
+        assert response.status_code in [400, 401], (
+            f"Token {invalid_token[:20]}... should be invalid (400 or 401)"
         )
 
         # Validate error response
         validator = ResponseValidator()
         error_validation = validator.validate_error_response(
-            response, expected_status=401, expected_error_key="detail"
+            response, expected_status=response.status_code, expected_error_key="detail"
         )
         assert error_validation["valid"]
 
@@ -169,7 +167,7 @@ async def test_authorization_header_validation(
     valid_api_key = response.json()["api_key"]
 
     # Test scenarios
-    test_cases = [
+    test_cases: list[tuple[dict[str, str], int, str]] = [
         # (headers, expected_status, description)
         (
             {},
@@ -193,7 +191,7 @@ async def test_authorization_header_validation(
         integration_client.headers.pop("authorization", None)
 
         # Set test headers
-        integration_client.headers.update(headers)
+        integration_client.headers.update(headers)  # type: ignore[arg-type]
 
         # Make request to protected endpoint
         response = await integration_client.get("/v1/wallet/")
@@ -256,16 +254,14 @@ async def test_database_state_api_key_creation(
         # Verify database record
         hashed_key = api_key[3:]  # Remove "sk-" prefix
         result = await integration_session.execute(
-            select(ApiKey).where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+            select(TemporaryCredit).where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
         )
         db_key = result.scalar_one()
 
         # Validate stored data
         assert db_key.balance == amount * 1000  # msats
-        assert db_key.total_spent == 0
-        assert db_key.total_requests == 0
         assert db_key.refund_address is None
-        assert db_key.key_expiry_time is None
+        assert db_key.refund_expiration_time is None
 
         # Creation timestamp should be recent (within last minute)
         # Note: The model doesn't have a creation timestamp field,
@@ -446,7 +442,7 @@ async def test_concurrent_token_submissions(
     for api_key in api_keys:
         hashed_key = api_key[3:]  # Remove "sk-" prefix
         result = await integration_session.execute(
-            select(ApiKey).where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+            select(TemporaryCredit).where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
         )
         db_key = result.scalar_one()
         assert db_key.balance == expected_balances[hashed_key]
@@ -568,7 +564,7 @@ async def test_database_timestamp_accuracy(
     # Verify key exists in database
     hashed_key = api_key[3:]  # Remove "sk-" prefix
     result = await integration_session.execute(
-        select(ApiKey).where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+        select(TemporaryCredit).where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
     )
     db_key = result.scalar_one()
 

tests/integration/test_wallet_information.py

@@ -11,7 +11,7 @@ import pytest
 from httpx import AsyncClient
 from sqlmodel import select, update
 
-from routstr.core.db import ApiKey
+from routstr.core.db import TemporaryCredit
 
 from .utils import ConcurrencyTester, ResponseValidator
 
@@ -49,7 +49,7 @@ async def test_wallet_endpoint_with_valid_api_key(
     hashed_key = api_key[3:]  # Remove "sk-" prefix
 
     result = await integration_session.execute(
-        select(ApiKey).where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+        select(TemporaryCredit).where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
     )
     db_key = result.scalar_one()
 
@@ -138,7 +138,9 @@ async def test_wallet_with_zero_balance(
     # Manually set balance to zero in database
     hashed_key = api_key[3:]  # Remove "sk-" prefix
     await integration_session.execute(
-        update(ApiKey).where(ApiKey.hashed_key == hashed_key).values(balance=0)  # type: ignore[arg-type]
+        update(TemporaryCredit)
+        .where(TemporaryCredit.hashed_key == hashed_key)
+        .values(balance=0)  # type: ignore[arg-type]
     )
     await integration_session.commit()
 
@@ -181,9 +183,11 @@ async def test_expired_api_key_behavior(
     from sqlmodel import update
 
     await integration_session.execute(
-        update(ApiKey)
-        .where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
-        .values(key_expiry_time=past_expiry, refund_address="test@lightning.address")
+        update(TemporaryCredit)
+        .where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
+        .values(
+            refund_expiration_time=past_expiry, refund_address="test@lightning.address"
+        )
     )
     await integration_session.commit()
 
@@ -197,10 +201,10 @@ async def test_expired_api_key_behavior(
     # Verify expiry time was stored
     hashed_key = api_key[3:]  # Remove "sk-" prefix
     result = await integration_session.execute(
-        select(ApiKey).where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+        select(TemporaryCredit).where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
     )
     db_key = result.scalar_one()
-    assert db_key.key_expiry_time == past_expiry
+    assert db_key.refund_expiration_time == past_expiry
     assert db_key.refund_address == "test@lightning.address"
 
 
@@ -270,7 +274,7 @@ async def test_wallet_info_data_consistency(
     # Verify against database
     hashed_key = api_key[3:]  # Remove "sk-" prefix
     result = await integration_session.execute(
-        select(ApiKey).where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+        select(TemporaryCredit).where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
     )
     db_key = result.scalar_one()
 
@@ -360,12 +364,10 @@ async def test_wallet_after_partial_spending(
     hashed_key = api_key[3:]  # Remove "sk-" prefix
 
     await integration_session.execute(
-        update(ApiKey)
-        .where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+        update(TemporaryCredit)
+        .where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
         .values(
             balance=initial_balance - spent_amount,
-            total_spent=spent_amount,
-            total_requests=5,  # Simulate 5 requests
         )
     )
     await integration_session.commit()

tests/integration/test_wallet_refund.py

@@ -13,7 +13,7 @@ import pytest
 from httpx import AsyncClient
 from sqlmodel import select
 
-from routstr.core.db import ApiKey
+from routstr.core.db import TemporaryCredit
 
 
 @pytest.mark.integration
@@ -123,7 +123,9 @@ async def test_zero_balance_refund_handling(
     from sqlmodel import update
 
     await integration_session.execute(
-        update(ApiKey).where(ApiKey.hashed_key == hashed_key).values(balance=0)  # type: ignore[arg-type]
+        update(TemporaryCredit)
+        .where(TemporaryCredit.hashed_key == hashed_key)
+        .values(balance=0)  # type: ignore[arg-type]
     )
     await integration_session.commit()
 
@@ -136,7 +138,7 @@ async def test_zero_balance_refund_handling(
 
     # Key should still exist
     result = await integration_session.execute(
-        select(ApiKey).where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+        select(TemporaryCredit).where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
     )
     assert result.scalar_one_or_none() is not None
 
@@ -159,7 +161,7 @@ async def test_refund_amount_validation(
 
     # Verify the key has no refund address (needed for the "too small" check)
     result = await integration_session.execute(
-        select(ApiKey).where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+        select(TemporaryCredit).where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
     )
     key = result.scalar_one()
     assert key.refund_address is None
@@ -192,8 +194,8 @@ async def test_refund_with_lightning_address(
     from sqlmodel import update
 
     await integration_session.execute(
-        update(ApiKey)
-        .where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+        update(TemporaryCredit)
+        .where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
         .values(refund_address=refund_address)
     )
     await integration_session.commit()
@@ -252,7 +254,7 @@ async def test_database_state_after_refund(
 
     # Verify key exists before refund
     result = await integration_session.execute(
-        select(ApiKey).where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+        select(TemporaryCredit).where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
     )
     key_before = result.scalar_one()
     assert key_before.balance == 10_000_000
@@ -263,12 +265,12 @@ async def test_database_state_after_refund(
 
     # Verify key is deleted after refund
     result = await integration_session.execute(
-        select(ApiKey).where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+        select(TemporaryCredit).where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
     )
     assert result.scalar_one_or_none() is None
 
     # Count total keys to ensure only the specific one was deleted
-    result = await integration_session.execute(select(ApiKey))
+    result = await integration_session.execute(select(TemporaryCredit))
     remaining_keys = result.scalars().all()
     # Should have no keys left (assuming clean test environment)
     assert len(remaining_keys) == 0
@@ -480,8 +482,8 @@ async def test_refund_error_handling(
     from sqlmodel import update
 
     await integration_session.execute(
-        update(ApiKey)
-        .where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+        update(TemporaryCredit)
+        .where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
         .values(balance=-1000)  # Invalid negative balance
     )
     await integration_session.commit()
@@ -517,9 +519,9 @@ async def test_refund_with_expired_key(
     from sqlmodel import update
 
     await integration_session.execute(
-        update(ApiKey)
-        .where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
-        .values(key_expiry_time=past_expiry, refund_address="expired@ln.address")
+        update(TemporaryCredit)
+        .where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
+        .values(refund_expiration_time=past_expiry, refund_address="expired@ln.address")
     )
     await integration_session.commit()
 

tests/integration/test_wallet_topup.py

@@ -11,7 +11,7 @@ import pytest
 from httpx import AsyncClient
 from sqlmodel import select
 
-from routstr.core.db import ApiKey
+from routstr.core.db import TemporaryCredit
 
 from .utils import (
     CashuTokenGenerator,
@@ -64,7 +64,7 @@ async def test_topup_with_valid_token(  # type: ignore[no-untyped-def]
     # Get the hashed key from the API key
     hashed_key = api_key[3:]  # Remove "sk-" prefix
     result = await integration_session.execute(
-        select(ApiKey).where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+        select(TemporaryCredit).where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
     )
     db_key = result.scalar_one()
 

tests/integration/utils.py

@@ -9,7 +9,7 @@ import httpx
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlmodel import select
 
-from routstr.core.db import ApiKey
+from routstr.core.db import TemporaryCredit
 
 
 class CashuTokenGenerator:
@@ -97,11 +97,11 @@ class DatabaseStateValidator:
     def __init__(self, session: AsyncSession) -> None:
         self.session = session
 
-    async def get_api_key(self, api_key: str) -> Optional[ApiKey]:
+    async def get_api_key(self, api_key: str) -> Optional[TemporaryCredit]:
         """Get API key from database"""
         hashed_key = hashlib.sha256(api_key.encode()).hexdigest()
         result = await self.session.execute(
-            select(ApiKey).where(ApiKey.hashed_key == hashed_key)  # type: ignore[arg-type]
+            select(TemporaryCredit).where(TemporaryCredit.hashed_key == hashed_key)  # type: ignore[arg-type]
         )
         return result.scalar_one_or_none()
 
@@ -125,20 +125,6 @@ class DatabaseStateValidator:
             "current_balance": key_obj.balance,
         }
 
-    async def validate_request_count(
-        self, api_key: str, expected_count: int
-    ) -> Dict[str, Any]:
-        """Validate request count for an API key"""
-        key_obj = await self.get_api_key(api_key)
-        if not key_obj:
-            return {"valid": False, "error": "API key not found"}
-
-        return {
-            "valid": key_obj.total_requests == expected_count,
-            "expected": expected_count,
-            "actual": key_obj.total_requests,
-        }
-
     async def validate_atomic_update(
         self, api_key: str, field: str, expected_value: Any
     ) -> bool:
@@ -315,7 +301,7 @@ class ConcurrencyTester:
                 )
 
         tasks = [make_request(req) for req in requests]
-        return await asyncio.gather(*tasks, return_exceptions=False)
+        return await asyncio.gather(*tasks, return_exceptions=False)  # type: ignore
 
     @staticmethod
     async def test_race_condition(
@@ -463,8 +449,6 @@ class TestDataBuilder:
         """Create test API key data"""
         data: Dict[str, Any] = {
             "balance": balance,
-            "total_spent": 0,
-            "total_requests": 0,
         }
 
         if refund_address:

tests/integration/verify_setup.py

@@ -44,9 +44,9 @@ def check_imports() -> bool:
         print("Conftest fixtures imported successfully")
 
         # Check routstr modules - imports are for verification only
-        from routstr.core.db import ApiKey
+        from routstr.core.db import TemporaryCredit
 
-        del ApiKey
+        del TemporaryCredit
 
         print("Router modules imported successfully")
 

tests/unit/test_algorithm.py

@@ -7,12 +7,12 @@ from unittest.mock import Mock
 os.environ["UPSTREAM_BASE_URL"] = "http://test"
 os.environ["UPSTREAM_API_KEY"] = "test"
 
-from routstr.algorithm import (  # noqa: E402
+from routstr.models.algorithm import (  # noqa: E402
     calculate_model_cost_score,
     get_provider_penalty,
     should_prefer_model,
 )
-from routstr.payment.models import Architecture, Model, Pricing  # noqa: E402
+from routstr.models.models import Architecture, Model, Pricing  # noqa: E402
 
 
 def create_test_model(

tests/unit/test_fee_consistency.py

@@ -21,11 +21,11 @@ import pytest
 os.environ["UPSTREAM_BASE_URL"] = "http://test"
 os.environ["UPSTREAM_API_KEY"] = "test"
 
-from routstr.payment.models import (  # noqa: E402
+from routstr.models.crud import _model_to_row_payload
+from routstr.models.models import (  # noqa: E402
     Architecture,
     Model,
     Pricing,
-    _model_to_row_payload,
 )
 
 

tests/unit/test_image_tokens.py

@@ -4,10 +4,10 @@ from io import BytesIO
 import pytest
 from PIL import Image
 
-from routstr.payment.helpers import (
+from routstr.payment.cost import (
     _calculate_image_tokens,
+    _estimate_image_tokens_in_messages,
     _get_image_dimensions,
-    estimate_image_tokens_in_messages,
 )
 
 
@@ -81,7 +81,7 @@ async def test_estimate_image_tokens_base64() -> None:
         }
     ]
 
-    tokens = await estimate_image_tokens_in_messages(messages)
+    tokens = await _estimate_image_tokens_in_messages(messages)
     assert tokens > 0
 
 
@@ -108,7 +108,7 @@ async def test_estimate_image_tokens_multiple_images() -> None:
         }
     ]
 
-    tokens = await estimate_image_tokens_in_messages(messages)
+    tokens = await _estimate_image_tokens_in_messages(messages)
     assert tokens > 0
 
 
@@ -148,8 +148,8 @@ async def test_estimate_image_tokens_with_detail() -> None:
         }
     ]
 
-    tokens_low = await estimate_image_tokens_in_messages(messages_low)
-    tokens_high = await estimate_image_tokens_in_messages(messages_high)
+    tokens_low = await _estimate_image_tokens_in_messages(messages_low)
+    tokens_high = await _estimate_image_tokens_in_messages(messages_high)
 
     assert tokens_low == 85
     assert tokens_high > tokens_low
@@ -163,7 +163,7 @@ async def test_estimate_image_tokens_no_images() -> None:
         {"role": "assistant", "content": "I'm doing well, thank you!"},
     ]
 
-    tokens = await estimate_image_tokens_in_messages(messages)
+    tokens = await _estimate_image_tokens_in_messages(messages)
     assert tokens == 0
 
 
@@ -185,5 +185,5 @@ async def test_estimate_image_tokens_input_image_type() -> None:
         }
     ]
 
-    tokens = await estimate_image_tokens_in_messages(messages)
+    tokens = await _estimate_image_tokens_in_messages(messages)
     assert tokens > 0

tests/unit/test_logging_securityfilter.py

@@ -1,118 +0,0 @@
-"""Unit tests for the logging SecurityFilter.
-
-This module tests that the SecurityFilter correctly identifies and redacts
-sensitive information from log messages without causing false positives.
-
-"""
-
-import logging
-from collections.abc import Callable
-
-import pytest
-
-from routstr.core.logging import SecurityFilter
-
-
-@pytest.fixture
-def security_filter() -> SecurityFilter:
-    """Provide an instance of the SecurityFilter for testing."""
-    return SecurityFilter()
-
-
-@pytest.fixture
-def filter_message(security_filter: SecurityFilter) -> Callable[[str], str]:
-    """A helper fixture to apply the filter to a message string."""
-
-    def _filter(msg: str) -> str:
-        record = logging.LogRecord(
-            name="test_logger",
-            level=logging.INFO,
-            pathname="",
-            lineno=0,
-            msg=msg,
-            args=(),
-            exc_info=None,
-        )
-        security_filter.filter(record)
-        return record.getMessage()
-
-    return _filter
-
-
-def test_redacts_unquoted_key_value_pairs(filter_message: Callable[[str], str]) -> None:
-    """Test that an unquoted key-value pair is correctly redacted."""
-    original = "Processing request with api_key=sk-12345abcdef"
-    expected = "Processing request with api_key: [REDACTED]"
-    assert filter_message(original) == expected
-
-
-def test_redacts_quoted_key_value_pairs(filter_message: Callable[[str], str]) -> None:
-    """Test that a quoted token is correctly redacted."""
-    original = 'User authenticated with token="cashuA123abc"'
-    expected = "User authenticated with token: [REDACTED]"
-    assert filter_message(original) == expected
-
-
-def test_redacts_bearer_token(filter_message: Callable[[str], str]) -> None:
-    """Test that a Bearer token of sufficient length is redacted."""
-    original = "Authorization: Bearer abc1234567890xyzabcdefg"
-    expected = "Authorization: [REDACTED]"
-    assert filter_message(original) == expected
-
-
-def test_redacts_cashu_token(filter_message: Callable[[str], str]) -> None:
-    """Test that a Cashu token is redacted."""
-    original = "Received cashuTOKENeyJ0b2tlbiI6W3siaWQiOiI"
-    expected = "Received [REDACTED]"
-    assert filter_message(original) == expected
-
-
-def test_redacts_nsec_key(filter_message: Callable[[str], str]) -> None:
-    """Test that a full-length Nostr private key is redacted."""
-    original = "Private key is nsec1a8d9f8s7d9f8a7s6d5f4a3s2d1f9a8s7d6f5a4s3d2f1a9s8d7f6a5s4d3f"
-    expected = "Private key is [REDACTED]"
-    assert filter_message(original) == expected
-
-
-def test_ignores_non_sensitive_message(filter_message: Callable[[str], str]) -> None:
-    """Test that a message with no sensitive data is left untouched."""
-    original = "No token pricing configured, using base cost"
-    expected = "No token pricing configured, using base cost"
-    assert filter_message(original) == expected
-
-
-def test_multiple_secrets_in_one_message(filter_message: Callable[[str], str]) -> None:
-    """Test that multiple different secrets in one message are all redacted."""
-    original = 'Auth with Bearer abcdefghijklmnopqrstuvwxyz and api_key="sk-12345"'
-    expected = "Auth with [REDACTED] and api_key: [REDACTED]"
-    assert filter_message(original) == expected
-
-
-def test_redacts_key_with_no_value(filter_message: Callable[[str], str]) -> None:
-    """Test that a key with no value is not redacted."""
-    original = "Request contains api_key and secret."
-    expected = "Request contains api_key and secret."
-    assert filter_message(original) == expected
-
-
-def test_redacts_key_value_with_spaces(filter_message: Callable[[str], str]) -> None:
-    """Test that key-value pairs with extra spaces are correctly redacted."""
-    original = "Auth info:   api_key   =   'sk-12345'"
-    expected = "Auth info:   api_key: [REDACTED]"
-    assert filter_message(original) == expected
-
-
-def test_is_case_insensitive_for_keys(filter_message: Callable[[str], str]) -> None:
-    """Test that key matching is case-insensitive."""
-    original = "TOKEN=sk-abcdef12345"
-    expected = "token: [REDACTED]"
-    assert filter_message(original) == expected
-
-
-def test_is_case_insensitive_for_standalone(
-    filter_message: Callable[[str], str],
-) -> None:
-    """Test that standalone matching is case-insensitive."""
-    original = "Using NSEC1a8d9f8s7d9f8a7s6d5f4a3s2d1f9a8s7d6f5a4s3d2f1a9s8d7f6a5s4d3f and CaShuA123abc"
-    expected = "Using [REDACTED] and [REDACTED]"
-    assert filter_message(original) == expected

tests/unit/test_payment_helpers.py

@@ -7,11 +7,11 @@ os.environ["UPSTREAM_BASE_URL"] = "http://test"
 os.environ["UPSTREAM_API_KEY"] = "test"
 
 from routstr.core.settings import settings  # noqa: E402
-from routstr.payment.helpers import get_max_cost_for_model  # noqa: E402
+from routstr.payment.cost import get_max_cost_for_model  # noqa: E402
 
 
 async def test_get_max_cost_for_model_known() -> None:
-    from routstr.payment.models import Pricing
+    from routstr.models.models import Pricing
 
     # Mock DB session behavior
     mock_session = AsyncMock()
@@ -63,48 +63,32 @@ async def test_get_max_cost_for_model_known() -> None:
 
     with patch.object(settings, "fixed_pricing", False):
         with patch.object(settings, "tolerance_percentage", 0):
-            cost = await get_max_cost_for_model(
-                "gpt-4", session=mock_session, model_obj=mock_model
-            )
+            cost = get_max_cost_for_model(model_obj=mock_model)
             assert cost == 500000  # 500 sats * 1000 = msats
 
 
 async def test_get_max_cost_for_model_unknown() -> None:
-    mock_session = AsyncMock()
+    mock_model = Mock()
+    mock_model.sats_pricing = None
+    mock_model.id = "unknown-model"
 
-    # Mock the exec results to return no model override
-    async def async_mock_exec(query: Any) -> Any:
-        result = Mock()
-        result.first = Mock(return_value=None)
-        result.all = Mock(return_value=[])
-        return result
-
-    mock_session.exec = AsyncMock(side_effect=async_mock_exec)
-    mock_session.get = AsyncMock(return_value=None)
-
-    # Mock get_upstreams to return empty list
-    with patch("routstr.proxy.get_upstreams", return_value=[]):
-        with patch.object(settings, "fixed_cost_per_request", 100):
-            with patch.object(settings, "tolerance_percentage", 0):
-                cost = await get_max_cost_for_model(
-                    "unknown-model", session=mock_session, model_obj=None
-                )
-                assert cost == 100000
+    with patch.object(settings, "fixed_cost_per_request", 100):
+        with patch.object(settings, "tolerance_percentage", 0):
+            cost = get_max_cost_for_model(model_obj=mock_model)
+            assert cost == 100000
 
 
 async def test_get_max_cost_for_model_disabled() -> None:
-    mock_session = AsyncMock()
+    mock_model = Mock()
     with patch.object(settings, "fixed_pricing", True):
         with patch.object(settings, "fixed_cost_per_request", 200):
             with patch.object(settings, "tolerance_percentage", 0):
-                cost = await get_max_cost_for_model("any-model", session=mock_session)
+                cost = get_max_cost_for_model(model_obj=mock_model)
                 assert cost == 200000
 
 
 async def test_get_max_cost_for_model_tolerance() -> None:
-    from routstr.payment.models import Pricing
-
-    mock_session = AsyncMock()
+    from routstr.models.models import Pricing
 
     # Mock the model with sats_pricing
     mock_pricing = Pricing(
@@ -121,7 +105,5 @@ async def test_get_max_cost_for_model_tolerance() -> None:
 
     with patch.object(settings, "fixed_pricing", False):
         with patch.object(settings, "tolerance_percentage", 10):
-            cost = await get_max_cost_for_model(
-                "gpt-4", session=mock_session, model_obj=mock_model
-            )
+            cost = get_max_cost_for_model(model_obj=mock_model)
             assert cost == 450000  # 500 sats * 1000 * 0.9 = 450000

tests/unit/test_wallet.py

@@ -4,8 +4,13 @@ from unittest.mock import AsyncMock, Mock, patch
 
 import pytest
 
-from routstr.core.db import ApiKey
-from routstr.wallet import credit_balance, get_balance, recieve_token, send_token
+from routstr.core.db import TemporaryCredit
+from routstr.payment.wallet import (
+    credit_balance,
+    get_balance,
+    recieve_token,
+    send_token,
+)
 
 
 @pytest.mark.asyncio
@@ -15,7 +20,7 @@ async def test_get_balance() -> None:
     mock_wallet.load_mint = AsyncMock()
     mock_wallet.load_proofs = AsyncMock()
 
-    with patch("routstr.wallet.Wallet.with_db", return_value=mock_wallet):
+    with patch("routstr.payment.wallet.Wallet.with_db", return_value=mock_wallet):
         balance = await get_balance("sat")
         assert balance == 50000
 
@@ -43,7 +48,9 @@ async def test_recieve_token_valid() -> None:
     from routstr.core.settings import settings
 
     with patch.object(settings, "cashu_mints", ["http://mint:3338"]):
-        with patch("routstr.wallet.deserialize_token_from_string") as mock_deserialize:
+        with patch(
+            "routstr.payment.wallet.deserialize_token_from_string"
+        ) as mock_deserialize:
             mock_token = Mock()
             mock_token.keysets = ["keyset1"]
             mock_token.mint = "http://mint:3338"
@@ -54,7 +61,9 @@ async def test_recieve_token_valid() -> None:
 
             mock_wallet.load_mint = AsyncMock()
             mock_wallet.load_proofs = AsyncMock()
-            with patch("routstr.wallet.Wallet.with_db", return_value=mock_wallet):
+            with patch(
+                "routstr.payment.wallet.Wallet.with_db", return_value=mock_wallet
+            ):
                 amount, unit, mint = await recieve_token(token_str)
                 assert amount == 1000
                 assert unit == "sat"
@@ -65,8 +74,8 @@ async def test_recieve_token_valid() -> None:
 async def test_send_token() -> None:
     mock_wallet = Mock()
 
-    with patch("routstr.wallet.Wallet.with_db", return_value=mock_wallet):
-        with patch("routstr.wallet.send", return_value=(1000, "test_token")):
+    with patch("routstr.payment.wallet.Wallet.with_db", return_value=mock_wallet):
+        with patch("routstr.payment.wallet.send", return_value=(1000, "test_token")):
             token = await send_token(1000, "sat", "http://mint:3338")
             assert token == "test_token"
 
@@ -87,7 +96,7 @@ async def test_credit_balance() -> None:
     mock_session = AsyncMock()
 
     # Mock session.refresh to update the balance (simulates DB reload)
-    async def mock_refresh(key: ApiKey) -> None:
+    async def mock_refresh(key: TemporaryCredit) -> None:
         key.balance = 6000000
 
     mock_session.refresh.side_effect = mock_refresh
@@ -96,7 +105,7 @@ async def test_credit_balance() -> None:
 
     with patch.object(settings, "cashu_mints", ["http://mint:3338"]):
         with patch(
-            "routstr.wallet.recieve_token",
+            "routstr.payment.wallet.recieve_token",
             return_value=(1000, "sat", "http://mint:3338"),
         ):
             amount = await credit_balance(token_str, mock_key, mock_session)
@@ -112,7 +121,9 @@ async def test_credit_balance() -> None:
 async def test_recieve_token_untrusted_mint() -> None:
     mock_wallet = Mock()
 
-    with patch("routstr.wallet.deserialize_token_from_string") as mock_deserialize:
+    with patch(
+        "routstr.payment.wallet.deserialize_token_from_string"
+    ) as mock_deserialize:
         mock_token = Mock()
         mock_token.keysets = ["keyset1"]
         mock_token.mint = "http://untrusted:3338"
@@ -122,9 +133,9 @@ async def test_recieve_token_untrusted_mint() -> None:
 
         mock_wallet.load_mint = AsyncMock()
         mock_wallet.load_proofs = AsyncMock()
-        with patch("routstr.wallet.Wallet.with_db", return_value=mock_wallet):
+        with patch("routstr.payment.wallet.Wallet.with_db", return_value=mock_wallet):
             with patch(
-                "routstr.wallet.swap_to_primary_mint",
+                "routstr.payment.wallet.swap_to_primary_mint",
                 return_value=(900, "sat", "http://mint:3338"),
             ):
                 amount, unit, mint = await recieve_token("test_token")

ui/app/balances/page.tsx

@@ -29,7 +29,9 @@ export default function BalancesPage() {
         <div className='container max-w-6xl px-4 py-8 md:px-6 lg:px-8'>
           <div className='mb-8 flex flex-col gap-4 lg:flex-row lg:items-start lg:justify-between'>
             <div>
-              <h1 className='text-3xl font-bold tracking-tight'>Balances</h1>
+              <h1 className='text-3xl font-bold tracking-tight'>
+                Balances
+              </h1>
               <p className='text-muted-foreground mt-2'>
                 Monitor and manage wallet balances
               </p>

ui/app/login/page.tsx

@@ -78,8 +78,8 @@ export default function AdminLoginPage(): ReactElement {
   };
 
   return (
-    <div className='bg-background text-foreground flex min-h-screen items-center justify-center px-4 py-12'>
-      <Card className='border-border/60 bg-card/90 w-full max-w-md border shadow-2xl shadow-black/30 backdrop-blur'>
+    <div className='flex min-h-screen items-center justify-center bg-background px-4 py-12 text-foreground'>
+      <Card className='w-full max-w-md border border-border/60 bg-card/90 shadow-2xl shadow-black/30 backdrop-blur'>
         <CardHeader className='space-y-1'>
           <CardTitle className='text-center text-2xl font-bold'>
             Admin Login

ui/app/logs/log-details-dialog.tsx

@@ -97,7 +97,7 @@ export function LogDetailsDialog({
             <div>
               <h4 className='mb-2 text-sm font-medium'>Message</h4>
               <div className='bg-muted max-h-48 overflow-auto rounded-md p-3'>
-                <pre className='font-mono text-sm break-all whitespace-pre'>
+                <pre className='font-mono text-sm whitespace-pre break-all'>
                   {log.message}
                 </pre>
               </div>
@@ -116,12 +116,7 @@ export function LogDetailsDialog({
                         <Button
                           variant='outline'
                           size='sm'
-                          onClick={() =>
-                            copyToClipboard(
-                              String(log[field as keyof LogEntry] || ''),
-                              field
-                            )
-                          }
+                          onClick={() => copyToClipboard(String(log[field as keyof LogEntry] || ''), field)}
                           className='h-6 flex-shrink-0 px-2'
                         >
                           {copiedField === field ? (
@@ -180,9 +175,7 @@ export function LogDetailsDialog({
                 <Button
                   variant='outline'
                   size='sm'
-                  onClick={() =>
-                    copyToClipboard(JSON.stringify(log, null, 2), 'json')
-                  }
+                  onClick={() => copyToClipboard(JSON.stringify(log, null, 2), 'json')}
                   className='h-6 px-2'
                 >
                   {copiedField === 'json' ? (

ui/app/model/page.tsx

@@ -204,59 +204,47 @@ export default function ModelsPage() {
                                     </div>
                                   </div>
                                 </div>
-                                {totalModels === 0 && (
-                                  <Alert className='mb-4'>
+                                {totalModels === 0 ? (
+                                  <Alert>
                                     <AlertCircle className='h-4 w-4' />
                                     <AlertDescription>
                                       <div className='space-y-2'>
                                         <p className='font-medium'>
                                           No models found for this provider
                                         </p>
-                                        <div className='space-y-1 text-sm'>
-                                          <p className='font-medium'>
-                                            Common issues:
-                                          </p>
-                                          <ul className='ml-2 list-inside list-disc space-y-1'>
+                                        <div className='text-sm space-y-1'>
+                                          <p className='font-medium'>Common issues:</p>
+                                          <ul className='list-disc list-inside space-y-1 ml-2'>
                                             <li>
-                                              <strong>API credentials:</strong>{' '}
-                                              Check if the API key is correct
-                                              and has the right permissions
+                                              <strong>API credentials:</strong> Check if the API key is correct and has the right permissions
                                             </li>
                                             <li>
-                                              <strong>Base URL:</strong> Verify
-                                              the base URL is correct for your
-                                              provider
+                                              <strong>Base URL:</strong> Verify the base URL is correct for your provider
                                             </li>
                                             <li>
-                                              <strong>Network access:</strong>{' '}
-                                              Ensure the server can reach the
-                                              provider&apos;s API endpoint
+                                              <strong>Network access:</strong> Ensure the server can reach the provider&apos;s API endpoint
                                             </li>
                                             <li>
-                                              <strong>Provider status:</strong>{' '}
-                                              The upstream provider might be
-                                              temporarily unavailable
+                                              <strong>Provider status:</strong> The upstream provider might be temporarily unavailable
                                             </li>
                                           </ul>
                                           {groupData?.group_url && (
-                                            <p className='text-muted-foreground mt-2 text-xs'>
-                                              Current endpoint:{' '}
-                                              <code className='bg-muted rounded px-1'>
-                                                {groupData.group_url}
-                                              </code>
+                                            <p className='mt-2 text-xs text-muted-foreground'>
+                                              Current endpoint: <code className='bg-muted px-1 rounded'>{groupData.group_url}</code>
                                             </p>
                                           )}
                                         </div>
                                       </div>
                                     </AlertDescription>
                                   </Alert>
+                                ) : (
+                                  <ModelSelector
+                                    filterProvider={provider}
+                                    groupData={groupData}
+                                    showProviderActions={true}
+                                    showDeleteAllButton={false}
+                                  />
                                 )}
-                                <ModelSelector
-                                  filterProvider={provider}
-                                  groupData={groupData}
-                                  showProviderActions={true}
-                                  showDeleteAllButton={false}
-                                />
                               </div>
                             </TabsContent>
                           );

ui/app/page.tsx

@@ -53,7 +53,7 @@ export default function DashboardPage() {
       window.removeEventListener('storage', syncAuthState);
     };
   }, []);
-
+  
   const { data: btcUsdPrice } = useQuery({
     queryKey: ['btc-usd-price'],
     queryFn: fetchBtcUsdPrice,
@@ -131,13 +131,8 @@ export default function DashboardPage() {
         <SiteHeader />
         <div className='container max-w-7xl px-4 py-8 md:px-6 lg:px-8'>
           <div className='mb-8'>
-            <h1 className='mb-6 text-3xl font-bold tracking-tight'>
-              Dashboard
-            </h1>
-            <DashboardBalanceSummary
-              displayUnit={displayUnit}
-              usdPerSat={usdPerSat}
-            />
+            <h1 className='text-3xl font-bold tracking-tight mb-6'>Dashboard</h1>
+            <DashboardBalanceSummary displayUnit={displayUnit} usdPerSat={usdPerSat} />
           </div>
 
           <div className='mb-6 flex flex-col gap-4 lg:flex-row lg:items-end lg:justify-between'>
@@ -145,9 +140,8 @@ export default function DashboardPage() {
               <h2 className='text-xl font-semibold tracking-tight'>
                 Usage Analytics
               </h2>
-              <p className='text-muted-foreground mt-1 text-sm'>
-                Monitor requests, errors, and revenue over the last {timeRange}{' '}
-                hours
+              <p className='text-muted-foreground text-sm mt-1'>
+                Monitor requests, errors, and revenue over the last {timeRange} hours
               </p>
             </div>
             <div className='flex items-center gap-4'>
@@ -182,31 +176,27 @@ export default function DashboardPage() {
 
           <div className='space-y-6'>
             {summaryLoading ? (
-              <div className='py-8 text-center'>Loading summary...</div>
+              <div className='text-center py-8'>Loading summary...</div>
             ) : summaryData ? (
               <UsageSummaryCards summary={summaryData} />
             ) : null}
 
             <div className='grid gap-6 lg:grid-cols-2'>
               {metricsLoading ? (
-                <div className='col-span-2 py-8 text-center'>
+                <div className='text-center py-8 col-span-2'>
                   Loading metrics...
                 </div>
               ) : metricsData && metricsData.metrics.length > 0 ? (
                 <>
-                  <div className='col-span-full'>
+                  <div className="col-span-full">
                     <UsageMetricsChart
-                      data={
-                        metricsData.metrics.map((m) => ({
-                          ...m,
-                          revenue_sats: m.revenue_msats / 1000,
-                          refunds_sats: m.refunds_msats / 1000,
-                          net_revenue_sats:
-                            (m.revenue_msats - m.refunds_msats) / 1000,
-                        })) as Array<
-                          Record<string, unknown> & { timestamp: string }
-                        >
-                      }
+                      data={metricsData.metrics.map((m) => ({
+                        ...m,
+                        revenue_sats: m.revenue_msats / 1000,
+                        refunds_sats: m.refunds_msats / 1000,
+                        net_revenue_sats:
+                          (m.revenue_msats - m.refunds_msats) / 1000,
+                      })) as Array<Record<string, unknown> & { timestamp: string }>}
                       title='Revenue Over Time (sats)'
                       dataKeys={[
                         {
@@ -228,11 +218,7 @@ export default function DashboardPage() {
                     />
                   </div>
                   <UsageMetricsChart
-                    data={
-                      metricsData.metrics as Array<
-                        Record<string, unknown> & { timestamp: string }
-                      >
-                    }
+                    data={metricsData.metrics as Array<Record<string, unknown> & { timestamp: string }>}
                     title='Request Volume'
                     dataKeys={[
                       {
@@ -253,11 +239,7 @@ export default function DashboardPage() {
                     ]}
                   />
                   <UsageMetricsChart
-                    data={
-                      metricsData.metrics as Array<
-                        Record<string, unknown> & { timestamp: string }
-                      >
-                    }
+                    data={metricsData.metrics as Array<Record<string, unknown> & { timestamp: string }>}
                     title='Error Tracking'
                     dataKeys={[
                       {
@@ -278,11 +260,7 @@ export default function DashboardPage() {
                     ]}
                   />
                   <UsageMetricsChart
-                    data={
-                      metricsData.metrics as Array<
-                        Record<string, unknown> & { timestamp: string }
-                      >
-                    }
+                    data={metricsData.metrics as Array<Record<string, unknown> & { timestamp: string }>}
                     title='Payment Activity'
                     dataKeys={[
                       {
@@ -292,7 +270,7 @@ export default function DashboardPage() {
                       },
                     ]}
                   />
-                  <div className='space-y-6'>
+                  <div className="space-y-6">
                     {summaryData && summaryData.unique_models.length > 0 && (
                       <Card>
                         <CardHeader>
@@ -328,9 +306,7 @@ export default function DashboardPage() {
                                     key={type}
                                     className='flex items-center justify-between'
                                   >
-                                    <span className='text-sm font-medium'>
-                                      {type}
-                                    </span>
+                                    <span className='text-sm font-medium'>{type}</span>
                                     <span className='text-muted-foreground text-sm'>
                                       {count}
                                     </span>
@@ -359,18 +335,16 @@ export default function DashboardPage() {
             </div>
 
             {revenueByModelLoading ? (
-              <div className='py-8 text-center'>
-                Loading revenue by model...
-              </div>
+              <div className='text-center py-8'>Loading revenue by model...</div>
             ) : revenueByModelData && revenueByModelData.models.length > 0 ? (
-              <RevenueByModelTable
+              <RevenueByModelTable 
                 models={revenueByModelData.models}
                 totalRevenue={revenueByModelData.total_revenue_sats}
               />
             ) : null}
 
             {errorLoading ? (
-              <div className='py-8 text-center'>Loading errors...</div>
+              <div className='text-center py-8'>Loading errors...</div>
             ) : errorData ? (
               <ErrorDetailsTable errors={errorData.errors} />
             ) : null}

ui/app/providers/page.tsx

@@ -18,9 +18,7 @@ import {
   UpstreamProvider,
   CreateUpstreamProvider,
   UpdateUpstreamProvider,
-  AdminModel,
 } from '@/lib/api/services/admin';
-import { AddProviderModelDialog } from '@/components/AddProviderModelDialog';
 import { Skeleton } from '@/components/ui/skeleton';
 import {
   AlertCircle,
@@ -53,329 +51,9 @@ import {
 } from '@/components/ui/select';
 import { Switch } from '@/components/ui/switch';
 import { Tabs, TabsContent, TabsList, TabsTrigger } from '@/components/ui/tabs';
-import { useState, useEffect } from 'react';
+import { useState } from 'react';
 import { toast } from 'sonner';
 
-function ProviderBalance({
-  providerId,
-  platformUrl,
-}: {
-  providerId: number;
-  platformUrl?: string | null;
-}) {
-  const [isTopupDialogOpen, setIsTopupDialogOpen] = useState(false);
-  const [topupAmount, setTopupAmount] = useState('');
-  const [topupError, setTopupError] = useState('');
-  const [isHovered, setIsHovered] = useState(false);
-  const [invoiceData, setInvoiceData] = useState<{
-    payment_request: string;
-    invoice_id: string;
-  } | null>(null);
-  const [paymentStatus, setPaymentStatus] = useState<'pending' | 'paid' | null>(
-    null
-  );
-  const queryClient = useQueryClient();
-
-  const {
-    data: balanceData,
-    isLoading,
-    error,
-  } = useQuery({
-    queryKey: ['provider-balance', providerId],
-    queryFn: () => AdminService.getProviderBalance(providerId),
-    refetchInterval: 30000,
-    refetchOnWindowFocus: true,
-    retry: 1,
-  });
-
-  const { data: statusData } = useQuery({
-    queryKey: ['topup-status', providerId, invoiceData?.invoice_id],
-    queryFn: () =>
-      AdminService.checkTopupStatus(providerId, invoiceData!.invoice_id),
-    enabled: !!invoiceData && paymentStatus === 'pending',
-    refetchInterval: 2000,
-  });
-
-  useEffect(() => {
-    if (statusData?.paid === true) {
-      setPaymentStatus('paid');
-      queryClient.invalidateQueries({
-        queryKey: ['provider-balance', providerId],
-      });
-      toast.success('Payment received!', {
-        description: 'Your balance has been updated.',
-      });
-    }
-  }, [statusData, queryClient, providerId]);
-
-  const topupMutation = useMutation({
-    mutationFn: async (amount: number) => {
-      console.log('Calling top-up API with:', { providerId, amount });
-      try {
-        const result = await AdminService.initiateProviderTopup(
-          providerId,
-          amount
-        );
-        console.log('API returned:', result);
-        return result;
-      } catch (err) {
-        console.error('API call failed:', err);
-        throw err;
-      }
-    },
-    onSuccess: (data) => {
-      console.log('Top-up response:', data);
-      console.log('Type of data:', typeof data);
-      console.log('Keys in data:', Object.keys(data || {}));
-
-      if (data?.topup_data?.payment_request && data?.topup_data?.invoice_id) {
-        setInvoiceData({
-          payment_request: data.topup_data.payment_request as string,
-          invoice_id: data.topup_data.invoice_id as string,
-        });
-        setPaymentStatus('pending');
-      } else {
-        console.error('Missing invoice data:', data);
-        console.error('topup_data:', data?.topup_data);
-        toast.error('No invoice returned from provider');
-        setIsTopupDialogOpen(false);
-      }
-    },
-    onError: (error: Error) => {
-      console.error('Top-up mutation error:', error);
-      toast.error(`Failed to initiate top-up: ${error.message}`);
-    },
-  });
-
-  const handleTopup = () => {
-    // If no dialog open logic (which depends on API implementation),
-    // we check if we should redirect or open dialog based on available info
-    // But since this function is called inside the dialog, we might want to change
-    // how the "Top Up" button behaves instead.
-    const amount = parseFloat(topupAmount);
-
-    if (isNaN(amount)) {
-      setTopupError('Please enter a valid amount');
-      return;
-    }
-
-    if (amount < 1 || amount > 500) {
-      setTopupError('Amount must be between $1 and $500');
-      return;
-    }
-
-    topupMutation.mutate(amount);
-  };
-
-  const handleTopUpClick = () => {
-    // Check if the provider supports direct topup (currently only PPQ.AI effectively)
-    // We can infer this if it's NOT OpenRouter or OpenAI, or strictly checking provider capability
-    // For now, we'll try to initiate topup for anyone, but if we know it fails (or isn't implemented),
-    // we should redirect.
-    // However, the prompt asks to redirect if topup is not implemented.
-    // The backend throws 500/400 if not implemented.
-    // A better approach is to check if we have a platform URL and maybe redirect there
-    // if we know it's not supported.
-
-    // BUT, we don't know for sure if it's supported without checking metadata or trying.
-    // Let's rely on the "can_topup" metadata if available, but currently we only have "can_show_balance".
-
-    // Simple heuristic: If platformUrl exists and we suspect no direct topup, redirect?
-    // Actually, let's try to open the dialog, but if it's OpenRouter/OpenAI, maybe we just redirect?
-    // The user specifically mentioned "like in openrouter".
-
-    if (
-      platformUrl &&
-      (platformUrl.includes('openrouter.ai') ||
-        platformUrl.includes('openai.com'))
-    ) {
-      window.open(platformUrl, '_blank');
-      return;
-    }
-
-    setIsTopupDialogOpen(true);
-  };
-
-  const handleCloseDialog = () => {
-    setIsTopupDialogOpen(false);
-    setTopupAmount('');
-    setTopupError('');
-    setInvoiceData(null);
-    setPaymentStatus(null);
-  };
-
-  if (isLoading) {
-    return <Skeleton className='h-9 w-24' />;
-  }
-
-  if (error || !balanceData?.ok || !balanceData.balance_data) {
-    return null;
-  }
-
-  const balance = balanceData.balance_data;
-  let displayValue = 'N/A';
-
-  if (typeof balance === 'number') {
-    displayValue = `$${balance.toFixed(2)}`;
-  } else if (balance && typeof balance === 'object') {
-    // Legacy support for object response
-    const b = balance as Record<string, unknown>;
-    if (typeof b.balance === 'number') {
-      displayValue = `$${b.balance.toFixed(2)}`;
-    } else if (typeof b.balance === 'string') {
-      displayValue = b.balance;
-    } else if (b.amount !== undefined) {
-      displayValue = `$${Number(b.amount).toFixed(2)}`;
-    }
-  }
-
-  return (
-    <>
-      <Button
-        variant='outline'
-        size='sm'
-        onClick={handleTopUpClick}
-        onMouseEnter={() => setIsHovered(true)}
-        onMouseLeave={() => setIsHovered(false)}
-        className='w-full font-mono sm:w-auto'
-      >
-        {isHovered ? 'Top Up' : displayValue}
-      </Button>
-
-      <Dialog open={isTopupDialogOpen} onOpenChange={handleCloseDialog}>
-        <DialogContent className='sm:max-w-md'>
-          <DialogHeader>
-            <DialogTitle>
-              {paymentStatus === 'paid'
-                ? 'Payment Confirmed!'
-                : 'Top Up Balance'}
-            </DialogTitle>
-            <DialogDescription>
-              {paymentStatus === 'paid'
-                ? 'Your account balance has been updated.'
-                : invoiceData
-                  ? 'Scan the QR code or copy the Lightning invoice to pay.'
-                  : 'Enter the amount you want to add to your account balance.'}
-            </DialogDescription>
-          </DialogHeader>
-
-          {paymentStatus === 'paid' ? (
-            <div className='flex flex-col items-center gap-4 py-6'>
-              <div className='rounded-full bg-green-100 p-3 dark:bg-green-900'>
-                <svg
-                  className='h-12 w-12 text-green-600 dark:text-green-400'
-                  fill='none'
-                  strokeLinecap='round'
-                  strokeLinejoin='round'
-                  strokeWidth='2'
-                  viewBox='0 0 24 24'
-                  stroke='currentColor'
-                >
-                  <path d='M5 13l4 4L19 7'></path>
-                </svg>
-              </div>
-              <p className='text-center font-semibold'>Top-up successful!</p>
-            </div>
-          ) : invoiceData ? (
-            <div className='flex flex-col items-center gap-4 py-4'>
-              <div className='rounded-lg border-2 border-gray-200 p-2 dark:border-gray-800'>
-                {/* eslint-disable-next-line @next/next/no-img-element */}
-                <img
-                  src={`https://api.qrserver.com/v1/create-qr-code/?size=256x256&data=${encodeURIComponent(invoiceData.payment_request)}`}
-                  alt='Lightning Invoice QR Code'
-                  className='h-64 w-64'
-                />
-              </div>
-              <div className='w-full space-y-2'>
-                <Label htmlFor='invoice'>Lightning Invoice</Label>
-                <div className='flex gap-2'>
-                  <Input
-                    id='invoice'
-                    value={invoiceData.payment_request}
-                    readOnly
-                    className='font-mono text-xs'
-                  />
-                  <Button
-                    size='sm'
-                    variant='outline'
-                    onClick={() => {
-                      navigator.clipboard.writeText(
-                        invoiceData.payment_request
-                      );
-                      toast.success('Invoice copied to clipboard!');
-                    }}
-                  >
-                    Copy
-                  </Button>
-                </div>
-              </div>
-              {paymentStatus === 'pending' && (
-                <p className='text-muted-foreground text-center text-sm'>
-                  Waiting for payment...
-                </p>
-              )}
-            </div>
-          ) : (
-            <div className='grid gap-4 py-4'>
-              <div className='grid gap-2'>
-                <Label htmlFor='topup_amount'>Amount (USD)</Label>
-                <Input
-                  id='topup_amount'
-                  type='number'
-                  placeholder='Enter amount (1-500)'
-                  value={topupAmount}
-                  onChange={(e) => {
-                    setTopupAmount(e.target.value);
-                    setTopupError('');
-                  }}
-                  min='1'
-                  max='500'
-                  step='0.01'
-                />
-                {topupError && (
-                  <p className='text-sm text-red-600 dark:text-red-400'>
-                    {topupError}
-                  </p>
-                )}
-              </div>
-            </div>
-          )}
-
-          <DialogFooter>
-            {paymentStatus === 'paid' ? (
-              <Button onClick={handleCloseDialog} className='w-full'>
-                Done
-              </Button>
-            ) : invoiceData ? (
-              <Button
-                variant='outline'
-                onClick={handleCloseDialog}
-                className='w-full'
-              >
-                Cancel
-              </Button>
-            ) : (
-              <>
-                <Button variant='outline' onClick={handleCloseDialog}>
-                  Cancel
-                </Button>
-                <Button
-                  onClick={handleTopup}
-                  disabled={topupMutation.isPending || !topupAmount}
-                >
-                  {topupMutation.isPending
-                    ? 'Processing...'
-                    : 'Generate Invoice'}
-                </Button>
-              </>
-            )}
-          </DialogFooter>
-        </DialogContent>
-      </Dialog>
-    </>
-  );
-}
-
 export default function ProvidersPage() {
   const queryClient = useQueryClient();
   const [editingProvider, setEditingProvider] =
@@ -386,18 +64,6 @@ export default function ProvidersPage() {
     new Set()
   );
   const [viewingModels, setViewingModels] = useState<number | null>(null);
-  const [isCreatingAccount, setIsCreatingAccount] = useState(false);
-  const [modelDialogState, setModelDialogState] = useState<{
-    isOpen: boolean;
-    providerId: number | null;
-    mode: 'create' | 'edit' | 'override';
-    initialData?: AdminModel | null;
-  }>({
-    isOpen: false,
-    providerId: null,
-    mode: 'create',
-    initialData: null,
-  });
 
   const [formData, setFormData] = useState<CreateUpstreamProvider>({
     provider_type: 'openrouter',
@@ -405,13 +71,8 @@ export default function ProvidersPage() {
     api_key: '',
     api_version: null,
     enabled: true,
-    provider_fee: 1.06,
   });
 
-  const getProviderFeePlaceholder = (type: string) => {
-    return type === 'openrouter' ? 'Default: 1.06 (6%)' : 'Default: 1.01 (1%)';
-  };
-
   const { data: providerTypes = [] } = useQuery({
     queryKey: ['provider-types'],
     queryFn: () => AdminService.getProviderTypes(),
@@ -477,32 +138,6 @@ export default function ProvidersPage() {
     },
   });
 
-  const handleCreateAccount = async () => {
-    setIsCreatingAccount(true);
-    try {
-      const response = await AdminService.createProviderAccountByType(
-        formData.provider_type
-      );
-      if (response.ok && response.account_data.api_key) {
-        setFormData({
-          ...formData,
-          api_key: String(response.account_data.api_key),
-        });
-        toast.success(
-          'Account created successfully! API key has been filled in.'
-        );
-      } else {
-        toast.success('Account created, but no API key returned.');
-      }
-    } catch (error: unknown) {
-      const errorMessage =
-        error instanceof Error ? error.message : 'Unknown error';
-      toast.error(`Failed to create account: ${errorMessage}`);
-    } finally {
-      setIsCreatingAccount(false);
-    }
-  };
-
   const resetForm = () => {
     setFormData({
       provider_type: 'openrouter',
@@ -514,11 +149,7 @@ export default function ProvidersPage() {
   };
 
   const handleCreate = () => {
-    const data = { ...formData };
-    if (data.provider_fee === undefined || data.provider_fee === null) {
-      data.provider_fee = data.provider_type === 'openrouter' ? 1.06 : 1.01;
-    }
-    createMutation.mutate(data);
+    createMutation.mutate(formData);
   };
 
   const handleEdit = (provider: UpstreamProvider) => {
@@ -529,7 +160,6 @@ export default function ProvidersPage() {
       api_key: '',
       api_version: provider.api_version || null,
       enabled: provider.enabled,
-      provider_fee: provider.provider_fee,
     });
     setIsEditDialogOpen(true);
   };
@@ -541,7 +171,6 @@ export default function ProvidersPage() {
       base_url: formData.base_url,
       api_version: formData.api_version,
       enabled: formData.enabled,
-      provider_fee: formData.provider_fee,
     };
     if (formData.api_key) {
       updateData.api_key = formData.api_key;
@@ -570,16 +199,6 @@ export default function ProvidersPage() {
     return providerType?.platform_url || null;
   };
 
-  const canCreateAccount = (type: string) => {
-    const providerType = providerTypes.find((pt) => pt.id === type);
-    return providerType?.can_create_account || false;
-  };
-
-  const canShowBalance = (type: string) => {
-    const providerType = providerTypes.find((pt) => pt.id === type);
-    return providerType?.can_show_balance || false;
-  };
-
   const toggleProviderExpansion = (providerId: number) => {
     const newExpanded = new Set(expandedProviders);
     if (newExpanded.has(providerId)) {
@@ -595,33 +214,6 @@ export default function ProvidersPage() {
     }
   };
 
-  const handleAddModel = (providerId: number) => {
-    setModelDialogState({
-      isOpen: true,
-      providerId,
-      mode: 'create',
-      initialData: null,
-    });
-  };
-
-  const handleEditModel = (providerId: number, model: AdminModel) => {
-    setModelDialogState({
-      isOpen: true,
-      providerId,
-      mode: 'edit',
-      initialData: model,
-    });
-  };
-
-  const handleOverrideModel = (providerId: number, model: AdminModel) => {
-    setModelDialogState({
-      isOpen: true,
-      providerId,
-      mode: 'override',
-      initialData: model,
-    });
-  };
-
   return (
     <SidebarProvider>
       <AppSidebar variant='inset' />
@@ -661,12 +253,11 @@ export default function ProvidersPage() {
                       <Select
                         value={formData.provider_type}
                         onValueChange={(value) => {
-                          setFormData((prev) => ({
-                            ...prev,
+                          setFormData({
+                            ...formData,
                             provider_type: value,
                             base_url: getDefaultBaseUrl(value),
-                            provider_fee: value === 'openrouter' ? 1.06 : 1.01,
-                          }));
+                          });
                         }}
                       >
                         <SelectTrigger>
@@ -701,30 +292,15 @@ export default function ProvidersPage() {
                     <div className='grid gap-2'>
                       <div className='flex items-center justify-between'>
                         <Label htmlFor='api_key'>API Key</Label>
-                        {canCreateAccount(formData.provider_type) ? (
-                          <Button
-                            type='button'
-                            variant='outline'
-                            size='sm'
-                            onClick={handleCreateAccount}
-                            disabled={isCreatingAccount}
-                            className='h-6 text-xs'
+                        {getPlatformUrl(formData.provider_type) && (
+                          <a
+                            href={getPlatformUrl(formData.provider_type)!}
+                            target='_blank'
+                            rel='noopener noreferrer'
+                            className='text-xs text-blue-600 hover:text-blue-800 dark:text-blue-400 dark:hover:text-blue-300'
                           >
-                            {isCreatingAccount
-                              ? 'Creating...'
-                              : 'Create Account'}
-                          </Button>
-                        ) : (
-                          getPlatformUrl(formData.provider_type) && (
-                            <a
-                              href={getPlatformUrl(formData.provider_type)!}
-                              target='_blank'
-                              rel='noopener noreferrer'
-                              className='text-xs text-blue-600 hover:text-blue-800 dark:text-blue-400 dark:hover:text-blue-300'
-                            >
-                              Get Your API Key Here →
-                            </a>
-                          )
+                            Get Your API Key Here →
+                          </a>
                         )}
                       </div>
                       <Input
@@ -763,32 +339,6 @@ export default function ProvidersPage() {
                       />
                       <Label htmlFor='enabled'>Enabled</Label>
                     </div>
-                    <div className='grid gap-2'>
-                      <Label htmlFor='provider_fee'>
-                        Provider Fee (Multiplier)
-                      </Label>
-                      <Input
-                        id='provider_fee'
-                        type='number'
-                        step='0.001'
-                        min='1.0'
-                        value={formData.provider_fee || ''}
-                        onChange={(e) =>
-                          setFormData({
-                            ...formData,
-                            provider_fee: e.target.value
-                              ? parseFloat(e.target.value)
-                              : undefined,
-                          })
-                        }
-                        placeholder={getProviderFeePlaceholder(
-                          formData.provider_type
-                        )}
-                      />
-                      <p className='text-muted-foreground text-xs'>
-                        1.01 means +1% e.g. currency exchange, card fees, etc.
-                      </p>
-                    </div>
                   </div>
                   <DialogFooter>
                     <Button
@@ -860,16 +410,7 @@ export default function ProvidersPage() {
                             {provider.base_url}
                           </CardDescription>
                         </div>
-                        <div className='flex flex-wrap items-center gap-2'>
-                          {canShowBalance(provider.provider_type) &&
-                            provider.api_key && (
-                              <ProviderBalance
-                                providerId={provider.id}
-                                platformUrl={getPlatformUrl(
-                                  provider.provider_type
-                                )}
-                              />
-                            )}
+                        <div className='grid grid-cols-3 gap-2 sm:flex sm:flex-nowrap'>
                           <Button
                             variant='outline'
                             size='sm'
@@ -932,21 +473,9 @@ export default function ProvidersPage() {
                                 // No provided models - show custom models directly without tabs
                                 <div className='space-y-2'>
                                   {providerModels.db_models.length === 0 ? (
-                                    <div className='flex flex-col items-center justify-center gap-2 py-4'>
-                                      <div className='text-muted-foreground text-sm'>
-                                        No models configured. Add custom models
-                                        to use this provider.
-                                      </div>
-                                      <Button
-                                        variant='outline'
-                                        size='sm'
-                                        onClick={() =>
-                                          handleAddModel(provider.id)
-                                        }
-                                      >
-                                        <Plus className='mr-2 h-4 w-4' />
-                                        Add Custom Model
-                                      </Button>
+                                    <div className='text-muted-foreground py-4 text-center text-sm'>
+                                      No models configured. Add custom models to
+                                      use this provider.
                                     </div>
                                   ) : (
                                     <div className='space-y-2'>
@@ -977,24 +506,9 @@ export default function ProvidersPage() {
                                               {model.description || model.name}
                                             </div>
                                           </div>
-                                          <div className='flex items-center gap-2'>
-                                            <div className='text-muted-foreground text-xs whitespace-nowrap'>
-                                              {model.context_length?.toLocaleString()}{' '}
-                                              tokens
-                                            </div>
-                                            <Button
-                                              variant='ghost'
-                                              size='icon'
-                                              className='h-8 w-8'
-                                              onClick={() =>
-                                                handleEditModel(
-                                                  provider.id,
-                                                  model
-                                                )
-                                              }
-                                            >
-                                              <Pencil className='h-4 w-4' />
-                                            </Button>
+                                          <div className='text-muted-foreground text-xs whitespace-nowrap'>
+                                            {model.context_length?.toLocaleString()}{' '}
+                                            tokens
                                           </div>
                                         </div>
                                       ))}
@@ -1045,24 +559,12 @@ export default function ProvidersPage() {
                                     value='custom'
                                     className='mt-4 space-y-2'
                                   >
-                                    <div className='flex items-center justify-between'>
-                                      {providerModels.db_models.length > 0 && (
-                                        <div className='text-muted-foreground text-sm'>
-                                          Custom models override or extend the
-                                          provider&apos;s catalog.
-                                        </div>
-                                      )}
-                                      <Button
-                                        variant='outline'
-                                        size='sm'
-                                        onClick={() =>
-                                          handleAddModel(provider.id)
-                                        }
-                                      >
-                                        <Plus className='mr-2 h-4 w-4' />
-                                        Add
-                                      </Button>
-                                    </div>
+                                    {providerModels.db_models.length > 0 && (
+                                      <div className='text-muted-foreground mb-3 text-sm'>
+                                        Custom models override or extend the
+                                        provider&apos;s catalog.
+                                      </div>
+                                    )}
                                     {providerModels.db_models.length === 0 ? (
                                       <div className='text-muted-foreground py-4 text-center text-sm'>
                                         No custom models configured
@@ -1098,24 +600,9 @@ export default function ProvidersPage() {
                                                     model.name}
                                                 </div>
                                               </div>
-                                              <div className='flex items-center gap-2'>
-                                                <div className='text-muted-foreground text-xs whitespace-nowrap'>
-                                                  {model.context_length?.toLocaleString()}{' '}
-                                                  tokens
-                                                </div>
-                                                <Button
-                                                  variant='ghost'
-                                                  size='icon'
-                                                  className='h-8 w-8'
-                                                  onClick={() =>
-                                                    handleEditModel(
-                                                      provider.id,
-                                                      model
-                                                    )
-                                                  }
-                                                >
-                                                  <Pencil className='h-4 w-4' />
-                                                </Button>
+                                              <div className='text-muted-foreground text-xs whitespace-nowrap'>
+                                                {model.context_length?.toLocaleString()}{' '}
+                                                tokens
                                               </div>
                                             </div>
                                           )
@@ -1150,25 +637,9 @@ export default function ProvidersPage() {
                                                   model.name}
                                               </div>
                                             </div>
-                                            <div className='flex items-center gap-2'>
-                                              <div className='text-muted-foreground text-xs whitespace-nowrap'>
-                                                {model.context_length?.toLocaleString()}{' '}
-                                                tokens
-                                              </div>
-                                              <Button
-                                                variant='outline'
-                                                size='sm'
-                                                className='h-7 text-xs'
-                                                onClick={() =>
-                                                  handleOverrideModel(
-                                                    provider.id,
-                                                    model
-                                                  )
-                                                }
-                                              >
-                                                <Plus className='mr-1 h-3 w-3' />
-                                                Override
-                                              </Button>
+                                            <div className='text-muted-foreground text-xs whitespace-nowrap'>
+                                              {model.context_length?.toLocaleString()}{' '}
+                                              tokens
                                             </div>
                                           </div>
                                         )
@@ -1203,12 +674,11 @@ export default function ProvidersPage() {
                 <Select
                   value={formData.provider_type}
                   onValueChange={(value) => {
-                    setFormData((prev) => ({
-                      ...prev,
+                    setFormData({
+                      ...formData,
                       provider_type: value,
                       base_url: getDefaultBaseUrl(value),
-                      provider_fee: value === 'openrouter' ? 1.06 : 1.01,
-                    }));
+                    });
                   }}
                 >
                   <SelectTrigger>
@@ -1292,32 +762,6 @@ export default function ProvidersPage() {
                 />
                 <Label htmlFor='edit_enabled'>Enabled</Label>
               </div>
-              <div className='grid gap-2'>
-                <Label htmlFor='edit_provider_fee'>
-                  Provider Fee (Multiplier)
-                </Label>
-                <Input
-                  id='edit_provider_fee'
-                  type='number'
-                  step='0.001'
-                  min='1.0'
-                  value={formData.provider_fee || ''}
-                  onChange={(e) =>
-                    setFormData({
-                      ...formData,
-                      provider_fee: e.target.value
-                        ? parseFloat(e.target.value)
-                        : undefined,
-                    })
-                  }
-                  placeholder={getProviderFeePlaceholder(
-                    formData.provider_type
-                  )}
-                />
-                <p className='text-muted-foreground text-xs'>
-                  1.01 means +1% e.g. currency exchange, card fees, etc.
-                </p>
-              </div>
             </div>
             <DialogFooter>
               <Button
@@ -1335,23 +779,6 @@ export default function ProvidersPage() {
             </DialogFooter>
           </DialogContent>
         </Dialog>
-
-        {modelDialogState.providerId && (
-          <AddProviderModelDialog
-            providerId={modelDialogState.providerId}
-            isOpen={modelDialogState.isOpen}
-            onClose={() =>
-              setModelDialogState((prev) => ({ ...prev, isOpen: false }))
-            }
-            onSuccess={() => {
-              queryClient.invalidateQueries({
-                queryKey: ['provider-models', modelDialogState.providerId],
-              });
-            }}
-            initialData={modelDialogState.initialData}
-            mode={modelDialogState.mode}
-          />
-        )}
       </SidebarInset>
     </SidebarProvider>
   );

ui/components/AddProviderModelDialog.tsx

@@ -1,956 +0,0 @@
-'use client';
-
-import React, { useEffect, useMemo, useState } from 'react';
-import { useForm } from 'react-hook-form';
-import { z } from 'zod';
-import { zodResolver } from '@hookform/resolvers/zod';
-import { useQuery } from '@tanstack/react-query';
-import { Button } from '@/components/ui/button';
-import { Input } from '@/components/ui/input';
-import { Textarea } from '@/components/ui/textarea';
-import {
-  Popover,
-  PopoverContent,
-  PopoverTrigger,
-} from '@/components/ui/popover';
-import {
-  Command,
-  CommandEmpty,
-  CommandGroup,
-  CommandInput,
-  CommandItem,
-  CommandList,
-} from '@/components/ui/command';
-import {
-  Dialog,
-  DialogContent,
-  DialogDescription,
-  DialogFooter,
-  DialogHeader,
-  DialogTitle,
-} from '@/components/ui/dialog';
-import {
-  Form,
-  FormControl,
-  FormDescription,
-  FormField,
-  FormItem,
-  FormLabel,
-  FormMessage,
-} from '@/components/ui/form';
-import { Switch } from '@/components/ui/switch';
-import { Loader2, Plus } from 'lucide-react';
-import { toast } from 'sonner';
-import { AdminService, type AdminModel } from '@/lib/api/services/admin';
-
-const listFromString = (value: string): string[] =>
-  value
-    .split(',')
-    .map((item) => item.trim())
-    .filter((item) => item.length > 0);
-
-const listToString = (value: string[] | undefined | null): string =>
-  value && value.length > 0 ? value.join(', ') : '';
-
-const FormSchema = z.object({
-  id: z.string().min(1, 'Model ID is required'),
-  name: z.string().min(1, 'Name is required'),
-  description: z.string().default(''),
-  context_length: z.coerce.number().min(0).default(8192),
-  modality: z.string().min(1, 'Modality is required'),
-  input_modalities_raw: z.string().default(''),
-  output_modalities_raw: z.string().default(''),
-  tokenizer: z.string().default(''),
-  instruct_type: z.string().default(''),
-  canonical_slug: z.string().default(''),
-  alias_ids_raw: z.string().default(''),
-  upstream_provider_id: z.string().default(''),
-  input_cost: z.coerce.number().min(0).default(0),
-  output_cost: z.coerce.number().min(0).default(0),
-  request_cost: z.coerce.number().min(0).default(0),
-  image_cost: z.coerce.number().min(0).default(0),
-  web_search_cost: z.coerce.number().min(0).default(0),
-  internal_reasoning_cost: z.coerce.number().min(0).default(0),
-  max_prompt_cost: z.coerce.number().min(0).default(0),
-  max_completion_cost: z.coerce.number().min(0).default(0),
-  max_cost: z.coerce.number().min(0).default(0),
-  per_request_limits_raw: z.string().default(''),
-  top_provider_context_length: z.coerce.number().min(0).optional(),
-  top_provider_max_completion_tokens: z.coerce.number().min(0).optional(),
-  top_provider_is_moderated: z.boolean().default(false),
-  enabled: z.boolean().default(true),
-});
-
-type FormData = z.output<typeof FormSchema>;
-
-export interface AddProviderModelDialogProps {
-  providerId: number;
-  isOpen: boolean;
-  onClose: () => void;
-  onSuccess: () => void;
-  initialData?: AdminModel | null;
-  mode?: 'create' | 'edit' | 'override';
-}
-
-export function AddProviderModelDialog({
-  providerId,
-  isOpen,
-  onClose,
-  onSuccess,
-  initialData,
-  mode = 'create',
-}: AddProviderModelDialogProps) {
-  const [isSubmitting, setIsSubmitting] = useState(false);
-  const [isPresetOpen, setIsPresetOpen] = useState(false);
-  const [selectedPresetLabel, setSelectedPresetLabel] =
-    useState('Select a preset');
-
-  const form = useForm<FormData>({
-    resolver: zodResolver(FormSchema) as never,
-    defaultValues: {
-      id: '',
-      name: '',
-      description: '',
-      context_length: 8192,
-      modality: 'text',
-      input_modalities_raw: 'text',
-      output_modalities_raw: 'text',
-      tokenizer: '',
-      instruct_type: '',
-      canonical_slug: '',
-      alias_ids_raw: '',
-      upstream_provider_id: '',
-      input_cost: 0,
-      output_cost: 0,
-      request_cost: 0,
-      image_cost: 0,
-      web_search_cost: 0,
-      internal_reasoning_cost: 0,
-      max_prompt_cost: 0,
-      max_completion_cost: 0,
-      max_cost: 0,
-      per_request_limits_raw: '',
-      top_provider_context_length: undefined,
-      top_provider_max_completion_tokens: undefined,
-      top_provider_is_moderated: false,
-      enabled: true,
-    },
-  });
-
-  const isOverride = useMemo(() => mode === 'override', [mode]);
-  const isEdit = useMemo(() => mode === 'edit', [mode]);
-  const { data: presets = [], isLoading: isLoadingPresets } = useQuery({
-    queryKey: ['openrouter-presets'],
-    queryFn: () => AdminService.getOpenRouterPresets(),
-    staleTime: 10 * 60 * 1000,
-    refetchOnWindowFocus: false,
-  });
-
-  useEffect(() => {
-    if (initialData) {
-      const architecture = initialData.architecture as Record<string, unknown>;
-      const pricing = initialData.pricing as Record<string, number>;
-      const topProvider = initialData.top_provider as Record<
-        string,
-        unknown
-      > | null;
-
-      form.reset({
-        id: initialData.id,
-        name: initialData.name,
-        description: initialData.description,
-        context_length: initialData.context_length,
-        modality:
-          typeof architecture?.modality === 'string'
-            ? architecture.modality
-            : 'text',
-        input_modalities_raw: listToString(
-          (architecture?.input_modalities as string[]) || []
-        ),
-        output_modalities_raw: listToString(
-          (architecture?.output_modalities as string[]) || []
-        ),
-        tokenizer:
-          typeof architecture?.tokenizer === 'string'
-            ? architecture.tokenizer
-            : '',
-        instruct_type:
-          typeof architecture?.instruct_type === 'string'
-            ? architecture.instruct_type
-            : '',
-        canonical_slug: initialData.canonical_slug || '',
-        alias_ids_raw: listToString(initialData.alias_ids),
-        upstream_provider_id:
-          typeof initialData.upstream_provider_id === 'string'
-            ? initialData.upstream_provider_id
-            : initialData.upstream_provider_id?.toString() || '',
-        input_cost: pricing?.prompt ?? 0,
-        output_cost: pricing?.completion ?? 0,
-        request_cost: pricing?.request ?? 0,
-        image_cost: pricing?.image ?? 0,
-        web_search_cost: pricing?.web_search ?? 0,
-        internal_reasoning_cost: pricing?.internal_reasoning ?? 0,
-        max_prompt_cost: pricing?.max_prompt_cost ?? 0,
-        max_completion_cost: pricing?.max_completion_cost ?? 0,
-        max_cost: pricing?.max_cost ?? 0,
-        per_request_limits_raw: initialData.per_request_limits
-          ? JSON.stringify(initialData.per_request_limits, null, 2)
-          : '',
-        top_provider_context_length:
-          typeof topProvider?.context_length === 'number'
-            ? topProvider.context_length
-            : undefined,
-        top_provider_max_completion_tokens:
-          typeof topProvider?.max_completion_tokens === 'number'
-            ? topProvider.max_completion_tokens
-            : undefined,
-        top_provider_is_moderated:
-          typeof topProvider?.is_moderated === 'boolean'
-            ? topProvider.is_moderated
-            : false,
-        enabled: initialData.enabled,
-      });
-    } else {
-      form.reset({
-        id: '',
-        name: '',
-        description: '',
-        context_length: 8192,
-        modality: 'text',
-        input_modalities_raw: 'text',
-        output_modalities_raw: 'text',
-        tokenizer: '',
-        instruct_type: '',
-        canonical_slug: '',
-        alias_ids_raw: '',
-        upstream_provider_id: '',
-        input_cost: 0,
-        output_cost: 0,
-        request_cost: 0,
-        image_cost: 0,
-        web_search_cost: 0,
-        internal_reasoning_cost: 0,
-        max_prompt_cost: 0,
-        max_completion_cost: 0,
-        max_cost: 0,
-        per_request_limits_raw: '',
-        top_provider_context_length: undefined,
-        top_provider_max_completion_tokens: undefined,
-        top_provider_is_moderated: false,
-        enabled: true,
-      });
-    }
-  }, [initialData, form, isOpen]);
-
-  const applyModelToForm = (model: AdminModel) => {
-    setSelectedPresetLabel(`${model.id} — ${model.name}`);
-    const architecture = model.architecture as Record<string, unknown>;
-    const pricing = model.pricing as Record<string, number>;
-    const topProvider = model.top_provider as Record<string, unknown> | null;
-
-    form.setValue('id', model.id);
-    form.setValue('name', model.name);
-    form.setValue('description', model.description || '');
-    form.setValue('context_length', model.context_length);
-    form.setValue(
-      'modality',
-      typeof architecture?.modality === 'string'
-        ? architecture.modality
-        : 'text'
-    );
-    form.setValue(
-      'input_modalities_raw',
-      listToString((architecture?.input_modalities as string[]) || [])
-    );
-    form.setValue(
-      'output_modalities_raw',
-      listToString((architecture?.output_modalities as string[]) || [])
-    );
-    form.setValue(
-      'tokenizer',
-      typeof architecture?.tokenizer === 'string' ? architecture.tokenizer : ''
-    );
-    form.setValue(
-      'instruct_type',
-      typeof architecture?.instruct_type === 'string'
-        ? architecture.instruct_type
-        : ''
-    );
-    form.setValue('canonical_slug', model.canonical_slug || '');
-    form.setValue('alias_ids_raw', listToString(model.alias_ids));
-    form.setValue(
-      'upstream_provider_id',
-      typeof model.upstream_provider_id === 'string'
-        ? model.upstream_provider_id
-        : model.upstream_provider_id?.toString() || ''
-    );
-    form.setValue('input_cost', pricing?.prompt ?? 0);
-    form.setValue('output_cost', pricing?.completion ?? 0);
-    form.setValue('request_cost', pricing?.request ?? 0);
-    form.setValue('image_cost', pricing?.image ?? 0);
-    form.setValue('web_search_cost', pricing?.web_search ?? 0);
-    form.setValue('internal_reasoning_cost', pricing?.internal_reasoning ?? 0);
-    form.setValue('max_prompt_cost', pricing?.max_prompt_cost ?? 0);
-    form.setValue('max_completion_cost', pricing?.max_completion_cost ?? 0);
-    form.setValue('max_cost', pricing?.max_cost ?? 0);
-    form.setValue(
-      'per_request_limits_raw',
-      model.per_request_limits
-        ? JSON.stringify(model.per_request_limits, null, 2)
-        : ''
-    );
-    form.setValue(
-      'top_provider_context_length',
-      typeof topProvider?.context_length === 'number'
-        ? topProvider.context_length
-        : undefined
-    );
-    form.setValue(
-      'top_provider_max_completion_tokens',
-      typeof topProvider?.max_completion_tokens === 'number'
-        ? topProvider.max_completion_tokens
-        : undefined
-    );
-    form.setValue(
-      'top_provider_is_moderated',
-      typeof topProvider?.is_moderated === 'boolean'
-        ? topProvider.is_moderated
-        : false
-    );
-    form.setValue('enabled', model.enabled);
-  };
-
-  const onSubmit = async (data: FormData) => {
-    setIsSubmitting(true);
-    try {
-      let perRequestLimits: Record<string, unknown> | null = null;
-      if (
-        data.per_request_limits_raw &&
-        data.per_request_limits_raw.trim().length
-      ) {
-        try {
-          perRequestLimits = JSON.parse(data.per_request_limits_raw);
-        } catch {
-          toast.error('Per-request limits must be valid JSON');
-          setIsSubmitting(false);
-          return;
-        }
-      }
-
-      const adminModel: AdminModel = {
-        id: data.id,
-        name: data.name,
-        description: data.description || '',
-        created: Math.floor(Date.now() / 1000),
-        context_length: data.context_length,
-        architecture: {
-          modality: data.modality,
-          input_modalities: listFromString(
-            data.input_modalities_raw || data.modality
-          ),
-          output_modalities: listFromString(
-            data.output_modalities_raw || data.modality
-          ),
-          tokenizer: data.tokenizer || '',
-          instruct_type: data.instruct_type?.trim() || null,
-        },
-        pricing: {
-          prompt: data.input_cost,
-          completion: data.output_cost,
-          request: data.request_cost,
-          image: data.image_cost,
-          web_search: data.web_search_cost,
-          internal_reasoning: data.internal_reasoning_cost,
-          max_prompt_cost: data.max_prompt_cost,
-          max_completion_cost: data.max_completion_cost,
-          max_cost: data.max_cost,
-        },
-        per_request_limits: perRequestLimits,
-        top_provider:
-          data.top_provider_context_length ||
-          data.top_provider_max_completion_tokens ||
-          data.top_provider_is_moderated
-            ? {
-                context_length: data.top_provider_context_length ?? null,
-                max_completion_tokens:
-                  data.top_provider_max_completion_tokens ?? null,
-                is_moderated: data.top_provider_is_moderated,
-              }
-            : null,
-        upstream_provider_id: data.upstream_provider_id?.trim().length
-          ? data.upstream_provider_id.trim()
-          : providerId,
-        canonical_slug: data.canonical_slug?.trim() || null,
-        alias_ids: listFromString(data.alias_ids_raw || ''),
-        enabled: data.enabled,
-      };
-
-      if (isEdit) {
-        await AdminService.updateProviderModel(providerId, data.id, adminModel);
-        toast.success('Model updated successfully');
-      } else {
-        await AdminService.createProviderModel(providerId, adminModel);
-        toast.success(
-          isOverride ? 'Model override created' : 'Model created successfully'
-        );
-      }
-
-      onSuccess();
-      onClose();
-    } catch (error: unknown) {
-      const message =
-        error instanceof Error ? error.message : 'Unknown error saving model';
-      toast.error(`Failed to save model: ${message}`);
-    } finally {
-      setIsSubmitting(false);
-    }
-  };
-
-  const title = isEdit
-    ? 'Edit Model'
-    : isOverride
-      ? 'Override Model'
-      : 'Add Custom Model';
-  const description = isOverride
-    ? 'Create a custom override for this upstream model'
-    : 'Add a new model configuration for this provider';
-
-  return (
-    <Dialog open={isOpen} onOpenChange={(open) => !open && onClose()}>
-      <DialogContent className='max-h-[90vh] overflow-y-auto sm:max-w-[720px]'>
-        <DialogHeader>
-          <DialogTitle className='flex items-center gap-2'>
-            <Plus className='h-4 w-4' />
-            {title}
-          </DialogTitle>
-          <DialogDescription>{description}</DialogDescription>
-        </DialogHeader>
-        {!isEdit && !isOverride && (
-          <div className='bg-muted/30 rounded-md border p-3'>
-            <div className='mb-2 text-sm font-medium'>Presets</div>
-            <div className='grid gap-2 sm:grid-cols-3 sm:items-start'>
-              <Popover open={isPresetOpen} onOpenChange={setIsPresetOpen}>
-                <PopoverTrigger asChild>
-                  <Button
-                    variant='outline'
-                    role='combobox'
-                    aria-expanded={isPresetOpen}
-                    className='w-full justify-between overflow-hidden text-left text-sm'
-                  >
-                    <span className='truncate'>
-                      {isLoadingPresets
-                        ? 'Loading presets...'
-                        : selectedPresetLabel}
-                    </span>
-                  </Button>
-                </PopoverTrigger>
-                <PopoverContent
-                  className='w-80 max-w-sm p-0'
-                  align='start'
-                  sideOffset={4}
-                  collisionPadding={12}
-                  onOpenAutoFocus={(e) => e.preventDefault()}
-                >
-                  <Command shouldFilter={true}>
-                    <CommandInput placeholder='Search presets...' />
-                    <CommandList
-                      className='max-h-64 overflow-y-auto overscroll-contain'
-                      onWheel={(e) => e.stopPropagation()}
-                    >
-                      {isLoadingPresets ? (
-                        <CommandEmpty>Loading presets...</CommandEmpty>
-                      ) : presets.length === 0 ? (
-                        <CommandEmpty>No presets available.</CommandEmpty>
-                      ) : (
-                        <CommandGroup heading='Presets'>
-                          {presets.map((preset) => (
-                            <CommandItem
-                              key={preset.id}
-                              value={preset.id}
-                              onSelect={() => {
-                                applyModelToForm(preset);
-                                setIsPresetOpen(false);
-                              }}
-                            >
-                              <div className='flex flex-col text-sm'>
-                                <span className='font-medium'>{preset.id}</span>
-                                <span className='text-muted-foreground text-xs'>
-                                  {preset.name}
-                                </span>
-                              </div>
-                            </CommandItem>
-                          ))}
-                        </CommandGroup>
-                      )}
-                    </CommandList>
-                  </Command>
-                </PopoverContent>
-              </Popover>
-            </div>
-            <div className='text-muted-foreground mt-1 text-xs'>
-              Prefill fields from a preset model definition, then adjust as
-              needed.
-            </div>
-          </div>
-        )}
-        <Form {...form}>
-          <form onSubmit={form.handleSubmit(onSubmit)} className='space-y-4'>
-            <div className='grid grid-cols-1 gap-4 sm:grid-cols-2'>
-              <FormField
-                control={form.control}
-                name='id'
-                render={({ field }) => (
-                  <FormItem>
-                    <FormLabel>Model ID *</FormLabel>
-                    <FormControl>
-                      <Input
-                        placeholder='e.g., gpt-5.1'
-                        {...field}
-                        disabled={isOverride || isEdit}
-                      />
-                    </FormControl>
-                    <FormDescription>
-                      Unique identifier for the model
-                    </FormDescription>
-                    <FormMessage />
-                  </FormItem>
-                )}
-              />
-
-              <FormField
-                control={form.control}
-                name='name'
-                render={({ field }) => (
-                  <FormItem>
-                    <FormLabel>Display Name *</FormLabel>
-                    <FormControl>
-                      <Input placeholder='e.g., GPT-5.1' {...field} />
-                    </FormControl>
-                    <FormMessage />
-                  </FormItem>
-                )}
-              />
-            </div>
-
-            <FormField
-              control={form.control}
-              name='description'
-              render={({ field }) => (
-                <FormItem>
-                  <FormLabel>Description</FormLabel>
-                  <FormControl>
-                    <Textarea
-                      placeholder='Brief description...'
-                      {...field}
-                      rows={2}
-                    />
-                  </FormControl>
-                  <FormMessage />
-                </FormItem>
-              )}
-            />
-
-            <div className='grid grid-cols-1 gap-4 sm:grid-cols-2'>
-              <FormField
-                control={form.control}
-                name='modality'
-                render={({ field }) => (
-                  <FormItem>
-                    <FormLabel>Modality *</FormLabel>
-                    <FormControl>
-                      <Input
-                        placeholder='text, text+image->text, image->text, etc.'
-                        {...field}
-                      />
-                    </FormControl>
-                    <FormDescription>
-                      Composite modality label (e.g., text+image-&gt;text)
-                    </FormDescription>
-                    <FormMessage />
-                  </FormItem>
-                )}
-              />
-
-              <FormField
-                control={form.control}
-                name='context_length'
-                render={({ field }) => (
-                  <FormItem>
-                    <FormLabel>Context Length</FormLabel>
-                    <FormControl>
-                      <Input type='number' {...field} />
-                    </FormControl>
-                    <FormMessage />
-                  </FormItem>
-                )}
-              />
-            </div>
-
-            <div className='grid grid-cols-1 gap-4 sm:grid-cols-2'>
-              <FormField
-                control={form.control}
-                name='input_modalities_raw'
-                render={({ field }) => (
-                  <FormItem>
-                    <FormLabel>Input Modalities</FormLabel>
-                    <FormControl>
-                      <Input placeholder='text, image, file' {...field} />
-                    </FormControl>
-                    <FormDescription>Comma-separated list</FormDescription>
-                    <FormMessage />
-                  </FormItem>
-                )}
-              />
-
-              <FormField
-                control={form.control}
-                name='output_modalities_raw'
-                render={({ field }) => (
-                  <FormItem>
-                    <FormLabel>Output Modalities</FormLabel>
-                    <FormControl>
-                      <Input placeholder='text, image' {...field} />
-                    </FormControl>
-                    <FormDescription>Comma-separated list</FormDescription>
-                    <FormMessage />
-                  </FormItem>
-                )}
-              />
-            </div>
-
-            <div className='grid grid-cols-1 gap-4 sm:grid-cols-2'>
-              <FormField
-                control={form.control}
-                name='tokenizer'
-                render={({ field }) => (
-                  <FormItem>
-                    <FormLabel>Tokenizer</FormLabel>
-                    <FormControl>
-                      <Input placeholder='e.g., GPT, tiktoken' {...field} />
-                    </FormControl>
-                    <FormMessage />
-                  </FormItem>
-                )}
-              />
-
-              <FormField
-                control={form.control}
-                name='instruct_type'
-                render={({ field }) => (
-                  <FormItem>
-                    <FormLabel>Instruct Type</FormLabel>
-                    <FormControl>
-                      <Input placeholder='e.g., chat, completion' {...field} />
-                    </FormControl>
-                    <FormMessage />
-                  </FormItem>
-                )}
-              />
-            </div>
-
-            <div className='grid grid-cols-1 gap-4 sm:grid-cols-2'>
-              <FormField
-                control={form.control}
-                name='canonical_slug'
-                render={({ field }) => (
-                  <FormItem>
-                    <FormLabel>Canonical Slug</FormLabel>
-                    <FormControl>
-                      <Input
-                        placeholder='google/gemini-3-pro-preview-20251117'
-                        {...field}
-                      />
-                    </FormControl>
-                    <FormMessage />
-                  </FormItem>
-                )}
-              />
-
-              <FormField
-                control={form.control}
-                name='alias_ids_raw'
-                render={({ field }) => (
-                  <FormItem>
-                    <FormLabel>Alias IDs</FormLabel>
-                    <FormControl>
-                      <Input placeholder='alias-1, alias-2' {...field} />
-                    </FormControl>
-                    <FormDescription>Comma-separated list</FormDescription>
-                    <FormMessage />
-                  </FormItem>
-                )}
-              />
-            </div>
-
-            <div className='grid grid-cols-1 gap-4 sm:grid-cols-2'>
-              <FormField
-                control={form.control}
-                name='upstream_provider_id'
-                render={({ field }) => (
-                  <FormItem>
-                    <FormLabel>Upstream Provider ID</FormLabel>
-                    <FormControl>
-                      <Input placeholder='gemini' {...field} />
-                    </FormControl>
-                    <FormDescription>
-                      Defaults to current provider if left blank
-                    </FormDescription>
-                    <FormMessage />
-                  </FormItem>
-                )}
-              />
-
-              <FormField
-                control={form.control}
-                name='per_request_limits_raw'
-                render={({ field }) => (
-                  <FormItem>
-                    <FormLabel>Per-request Limits (JSON)</FormLabel>
-                    <FormControl>
-                      <Textarea
-                        placeholder='{"requests_per_min": 60}'
-                        {...field}
-                        rows={4}
-                      />
-                    </FormControl>
-                    <FormDescription>
-                      JSON object; leave empty for none
-                    </FormDescription>
-                    <FormMessage />
-                  </FormItem>
-                )}
-              />
-            </div>
-
-            <div className='bg-muted/20 space-y-4 rounded-md border p-4'>
-              <h4 className='text-sm font-medium'>
-                Pricing (USD per 1M tokens)
-              </h4>
-              <div className='grid grid-cols-1 gap-4 sm:grid-cols-2'>
-                <FormField
-                  control={form.control}
-                  name='input_cost'
-                  render={({ field }) => (
-                    <FormItem>
-                      <FormLabel>Input Cost</FormLabel>
-                      <FormControl>
-                        <Input type='number' step='0.01' {...field} />
-                      </FormControl>
-                      <FormMessage />
-                    </FormItem>
-                  )}
-                />
-                <FormField
-                  control={form.control}
-                  name='output_cost'
-                  render={({ field }) => (
-                    <FormItem>
-                      <FormLabel>Output Cost</FormLabel>
-                      <FormControl>
-                        <Input type='number' step='0.01' {...field} />
-                      </FormControl>
-                      <FormMessage />
-                    </FormItem>
-                  )}
-                />
-                <FormField
-                  control={form.control}
-                  name='request_cost'
-                  render={({ field }) => (
-                    <FormItem>
-                      <FormLabel>Per Request Cost</FormLabel>
-                      <FormControl>
-                        <Input type='number' step='0.0001' {...field} />
-                      </FormControl>
-                      <FormMessage />
-                    </FormItem>
-                  )}
-                />
-                <FormField
-                  control={form.control}
-                  name='image_cost'
-                  render={({ field }) => (
-                    <FormItem>
-                      <FormLabel>Image Cost (per image)</FormLabel>
-                      <FormControl>
-                        <Input type='number' step='0.0001' {...field} />
-                      </FormControl>
-                      <FormMessage />
-                    </FormItem>
-                  )}
-                />
-                <FormField
-                  control={form.control}
-                  name='web_search_cost'
-                  render={({ field }) => (
-                    <FormItem>
-                      <FormLabel>Web Search Cost</FormLabel>
-                      <FormControl>
-                        <Input type='number' step='0.0001' {...field} />
-                      </FormControl>
-                      <FormMessage />
-                    </FormItem>
-                  )}
-                />
-                <FormField
-                  control={form.control}
-                  name='internal_reasoning_cost'
-                  render={({ field }) => (
-                    <FormItem>
-                      <FormLabel>Internal Reasoning Cost</FormLabel>
-                      <FormControl>
-                        <Input type='number' step='0.0001' {...field} />
-                      </FormControl>
-                      <FormMessage />
-                    </FormItem>
-                  )}
-                />
-                <FormField
-                  control={form.control}
-                  name='max_prompt_cost'
-                  render={({ field }) => (
-                    <FormItem>
-                      <FormLabel>Max Prompt Cost</FormLabel>
-                      <FormControl>
-                        <Input type='number' step='0.0001' {...field} />
-                      </FormControl>
-                      <FormMessage />
-                    </FormItem>
-                  )}
-                />
-                <FormField
-                  control={form.control}
-                  name='max_completion_cost'
-                  render={({ field }) => (
-                    <FormItem>
-                      <FormLabel>Max Completion Cost</FormLabel>
-                      <FormControl>
-                        <Input type='number' step='0.0001' {...field} />
-                      </FormControl>
-                      <FormMessage />
-                    </FormItem>
-                  )}
-                />
-                <FormField
-                  control={form.control}
-                  name='max_cost'
-                  render={({ field }) => (
-                    <FormItem>
-                      <FormLabel>Max Total Cost</FormLabel>
-                      <FormControl>
-                        <Input type='number' step='0.0001' {...field} />
-                      </FormControl>
-                      <FormMessage />
-                    </FormItem>
-                  )}
-                />
-              </div>
-            </div>
-
-            <div className='bg-muted/20 space-y-4 rounded-md border p-4'>
-              <h4 className='text-sm font-medium'>Top Provider (optional)</h4>
-              <div className='grid grid-cols-1 gap-4 sm:grid-cols-3'>
-                <FormField
-                  control={form.control}
-                  name='top_provider_context_length'
-                  render={({ field }) => (
-                    <FormItem>
-                      <FormLabel>Context Length</FormLabel>
-                      <FormControl>
-                        <Input
-                          type='number'
-                          {...field}
-                          value={field.value ?? ''}
-                        />
-                      </FormControl>
-                      <FormMessage />
-                    </FormItem>
-                  )}
-                />
-                <FormField
-                  control={form.control}
-                  name='top_provider_max_completion_tokens'
-                  render={({ field }) => (
-                    <FormItem>
-                      <FormLabel>Max Completion Tokens</FormLabel>
-                      <FormControl>
-                        <Input
-                          type='number'
-                          {...field}
-                          value={field.value ?? ''}
-                        />
-                      </FormControl>
-                      <FormMessage />
-                    </FormItem>
-                  )}
-                />
-                <FormField
-                  control={form.control}
-                  name='top_provider_is_moderated'
-                  render={({ field }) => (
-                    <FormItem className='flex flex-row items-center justify-between rounded-lg border p-4'>
-                      <div className='space-y-0.5'>
-                        <FormLabel className='text-base'>
-                          Is Moderated
-                        </FormLabel>
-                        <FormDescription>
-                          Whether provider enforces moderation
-                        </FormDescription>
-                      </div>
-                      <FormControl>
-                        <Switch
-                          checked={field.value}
-                          onCheckedChange={field.onChange}
-                        />
-                      </FormControl>
-                    </FormItem>
-                  )}
-                />
-              </div>
-            </div>
-
-            <FormField
-              control={form.control}
-              name='enabled'
-              render={({ field }) => (
-                <FormItem className='flex flex-row items-center justify-between rounded-lg border p-4'>
-                  <div className='space-y-0.5'>
-                    <FormLabel className='text-base'>Enabled</FormLabel>
-                    <FormDescription>Enable this model for use</FormDescription>
-                  </div>
-                  <FormControl>
-                    <Switch
-                      checked={field.value}
-                      onCheckedChange={field.onChange}
-                    />
-                  </FormControl>
-                </FormItem>
-              )}
-            />
-
-            <DialogFooter>
-              <Button
-                type='button'
-                variant='outline'
-                onClick={onClose}
-                disabled={isSubmitting}
-              >
-                Cancel
-              </Button>
-              <Button type='submit' disabled={isSubmitting}>
-                {isSubmitting && (
-                  <Loader2 className='mr-2 h-4 w-4 animate-spin' />
-                )}
-                {isEdit
-                  ? 'Save Changes'
-                  : isOverride
-                    ? 'Create Override'
-                    : 'Create Model'}
-              </Button>
-            </DialogFooter>
-          </form>
-        </Form>
-      </DialogContent>
-    </Dialog>
-  );
-}

ui/components/ModelSelector.tsx

@@ -2,14 +2,15 @@
 
 import React, { useState, useMemo } from 'react';
 import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query';
-import { type Model, type GroupSettings } from '@/lib/api/schemas/models';
 import {
-  AdminService,
-  type AdminModelGroup,
-  type AdminModel,
-} from '@/lib/api/services/admin';
+  type Model,
+  type ManualModel,
+  type GroupSettings,
+} from '@/lib/api/schemas/models';
+import { AdminService, type AdminModelGroup } from '@/lib/api/services/admin';
 type ModelGroup = AdminModelGroup;
-import { AddProviderModelDialog } from '@/components/AddProviderModelDialog';
+import { AddModelForm } from '@/components/AddModelForm';
+import { EditModelForm } from '@/components/EditModelForm';
 import { EditGroupForm } from '@/components/EditGroupForm';
 import { CollectModelsDialog } from '@/components/CollectModelsDialog';
 import { formatCost } from '@/lib/services/costValidation';
@@ -79,23 +80,14 @@ export function ModelSelector({
 }: ModelSelectorProps) {
   const [selectedModelId, setSelectedModelId] = useState<string>('');
   const [, setHoveredModelId] = useState<string | null>(null);
+  const [isAddFormOpen, setIsAddFormOpen] = useState(false);
+  const [editingModel, setEditingModel] = useState<Model | null>(null);
   const [editingGroup, setEditingGroup] = useState<{
     provider: string;
     models: Model[];
     groupData: ModelGroup;
   } | null>(null);
   const [isCollectDialogOpen, setIsCollectDialogOpen] = useState(false);
-  const [modelDialogState, setModelDialogState] = useState<{
-    isOpen: boolean;
-    providerId: number | null;
-    mode: 'create' | 'edit' | 'override';
-    initialData?: AdminModel | null;
-  }>({
-    isOpen: false,
-    providerId: null,
-    mode: 'create',
-    initialData: null,
-  });
 
   // Bulk selection state
   const [selectedModels, setSelectedModels] = useState<Set<string>>(new Set());
@@ -500,104 +492,44 @@ export function ModelSelector({
     return hasIndividualApiKey || hasIndividualUrl;
   };
 
-  const handleAddModelClick = (providerId: number) => {
-    setModelDialogState({
-      isOpen: true,
-      providerId,
-      mode: 'create',
-      initialData: null,
-    });
-  };
+  // Handle manual model addition
+  const handleAddModel = async (newModel: ManualModel) => {
+    try {
+      // Find or create the provider group
+      let providerGroup = groups.find((g) => g.provider === newModel.provider);
 
-  const handleEditModelClick = (model: Model) => {
-    if (!model.provider_id) {
-      toast.error('Provider ID missing for model');
-      return;
+      if (!providerGroup) {
+        providerGroup = await AdminService.createModelGroup({
+          provider: newModel.provider,
+          group_api_key: undefined,
+        });
+      }
+
+      const modelData = {
+        name: newModel.name,
+        full_name: newModel.name,
+        input_cost: newModel.input_cost,
+        output_cost: newModel.output_cost,
+        provider: newModel.provider,
+        modelType: newModel.modelType,
+        description: newModel.description || undefined,
+        contextLength: newModel.contextLength,
+      };
+
+      await AdminService.createModel(modelData);
+      await refetchModels();
+      toast.success(`Model "${newModel.name}" added successfully!`);
+    } catch (error) {
+      console.error('Error adding model:', error);
+      toast.error('Failed to add model. Please try again.');
+      throw error; // Re-throw to let the form handle the error
     }
-    const providerId = parseInt(model.provider_id);
-
-    // Construct AdminModel from Model
-    const adminModel: AdminModel = {
-      id: model.id,
-      name: model.name,
-      description: model.description || '',
-      created: new Date(model.createdAt).getTime() / 1000,
-      context_length: model.contextLength || 0,
-      architecture: {
-        modality: model.modelType,
-        input_modalities: [model.modelType],
-        output_modalities: [model.modelType],
-        tokenizer: '',
-        instruct_type: null,
-      },
-      pricing: {
-        prompt: model.input_cost,
-        completion: model.output_cost,
-        request: model.min_cost_per_request,
-        image: 0,
-        web_search: 0,
-        internal_reasoning: 0,
-      },
-      per_request_limits: null,
-      top_provider: null,
-      upstream_provider_id: providerId,
-      enabled: model.isEnabled,
-    };
-
-    setModelDialogState({
-      isOpen: true,
-      providerId,
-      mode: 'edit',
-      initialData: adminModel,
-    });
-  };
-
-  const handleOverrideModelClick = (model: Model) => {
-    if (!model.provider_id) {
-      toast.error('Provider ID missing for model');
-      return;
-    }
-    const providerId = parseInt(model.provider_id);
-
-    // Construct AdminModel from Model
-    const adminModel: AdminModel = {
-      id: model.id,
-      name: model.name,
-      description: model.description || '',
-      created: new Date(model.createdAt).getTime() / 1000,
-      context_length: model.contextLength || 0,
-      architecture: {
-        modality: model.modelType,
-        input_modalities: [model.modelType],
-        output_modalities: [model.modelType],
-        tokenizer: '',
-        instruct_type: null,
-      },
-      pricing: {
-        prompt: model.input_cost,
-        completion: model.output_cost,
-        request: model.min_cost_per_request,
-        image: 0,
-        web_search: 0,
-        internal_reasoning: 0,
-      },
-      per_request_limits: null,
-      top_provider: null,
-      upstream_provider_id: providerId,
-      enabled: model.isEnabled,
-    };
-
-    setModelDialogState({
-      isOpen: true,
-      providerId,
-      mode: 'override',
-      initialData: adminModel,
-    });
   };
 
   // Handle model update
   const handleModelUpdate = async () => {
     await refetchModels();
+    setEditingModel(null);
   };
 
   // Handle group update
@@ -900,18 +832,11 @@ export function ModelSelector({
             Delete All Overrides Permanently
           </Button>
         )}
-        {/* Model Management Actions */}
-        {groupData && (
-          <Button
-            onClick={() => handleAddModelClick(parseInt(groupData.id))}
-            className='gap-2'
-            variant='outline'
-          >
-            <CheckSquare className='h-4 w-4' />
-            Add Custom Model
-          </Button>
-        )}
-        {/*
+        {/* Model Management Actions
+        <Button onClick={() => setIsAddFormOpen(true)} className='gap-2'>
+          <Plus className='h-4 w-4' />
+          Add Model
+        </Button>
         <Button
           onClick={() => setIsCollectDialogOpen(true)}
           variant='outline'
@@ -1155,28 +1080,15 @@ export function ModelSelector({
                               </Button>
                             </DropdownMenuTrigger>
                             <DropdownMenuContent align='end'>
-                              {model.api_key_type !== 'remote' && (
-                                <DropdownMenuItem
-                                  onClick={(e) => {
-                                    e.stopPropagation();
-                                    handleEditModelClick(model);
-                                  }}
-                                >
-                                  <Edit3 className='mr-2 h-4 w-4' />
-                                  Edit Model
-                                </DropdownMenuItem>
-                              )}
-                              {model.api_key_type === 'remote' && (
-                                <DropdownMenuItem
-                                  onClick={(e) => {
-                                    e.stopPropagation();
-                                    handleOverrideModelClick(model);
-                                  }}
-                                >
-                                  <Edit3 className='mr-2 h-4 w-4' />
-                                  Override
-                                </DropdownMenuItem>
-                              )}
+                              <DropdownMenuItem
+                                onClick={(e) => {
+                                  e.stopPropagation();
+                                  setEditingModel(model);
+                                }}
+                              >
+                                <Edit3 className='mr-2 h-4 w-4' />
+                                Edit Model
+                              </DropdownMenuItem>
                               <DropdownMenuSeparator />
                               {model.soft_deleted ? (
                                 <DropdownMenuItem
@@ -1303,16 +1215,23 @@ export function ModelSelector({
       })}
 
       {/* Forms and Dialogs */}
-      {modelDialogState.providerId && (
-        <AddProviderModelDialog
-          providerId={modelDialogState.providerId}
-          isOpen={modelDialogState.isOpen}
-          onClose={() =>
-            setModelDialogState((prev) => ({ ...prev, isOpen: false }))
+      <AddModelForm
+        isOpen={isAddFormOpen}
+        onModelAdd={handleAddModel}
+        onCancel={() => setIsAddFormOpen(false)}
+      />
+
+      {editingModel && (
+        <EditModelForm
+          model={editingModel}
+          providerId={
+            editingModel.provider_id
+              ? parseInt(editingModel.provider_id)
+              : undefined
           }
-          onSuccess={handleModelUpdate}
-          initialData={modelDialogState.initialData}
-          mode={modelDialogState.mode}
+          onModelUpdate={handleModelUpdate}
+          onCancel={() => setEditingModel(null)}
+          isOpen={!!editingModel}
         />
       )}
 

ui/components/currency-toggle.tsx

@@ -48,26 +48,20 @@ export function CurrencyToggle() {
   return (
     <DropdownMenu>
       <DropdownMenuTrigger asChild>
-        <Button
-          variant='ghost'
-          size='sm'
-          className='h-9 w-9 w-auto gap-2 px-0 px-3 font-normal'
-        >
-          <Coins className='h-4 w-4' />
-          <span className='hidden sm:inline-block'>
-            {getLabel(displayUnit)}
-          </span>
-          <span className='uppercase sm:hidden'>{displayUnit}</span>
+        <Button variant="ghost" size="sm" className="h-9 w-9 px-0 gap-2 w-auto px-3 font-normal">
+          <Coins className="h-4 w-4" />
+          <span className="hidden sm:inline-block">{getLabel(displayUnit)}</span>
+          <span className="sm:hidden uppercase">{displayUnit}</span>
         </Button>
       </DropdownMenuTrigger>
-      <DropdownMenuContent align='end'>
+      <DropdownMenuContent align="end">
         <DropdownMenuItem onClick={() => setDisplayUnit('msat')}>
           Millisatoshis (mSAT)
         </DropdownMenuItem>
         <DropdownMenuItem onClick={() => setDisplayUnit('sat')}>
           Satoshis (sat)
         </DropdownMenuItem>
-        <DropdownMenuItem
+        <DropdownMenuItem 
           onClick={() => setDisplayUnit('usd')}
           disabled={!usdPerSat}
         >
@@ -77,3 +71,4 @@ export function CurrencyToggle() {
     </DropdownMenu>
   );
 }
+

ui/components/dashboard-balance-summary.tsx

@@ -96,3 +96,4 @@ export function DashboardBalanceSummary({
     </div>
   );
 }
+

ui/components/error-details-table.tsx

@@ -24,7 +24,7 @@ export function ErrorDetailsTable({ errors }: ErrorDetailsTableProps) {
           <CardTitle>Recent Errors</CardTitle>
         </CardHeader>
         <CardContent>
-          <p className='text-muted-foreground py-8 text-center'>
+          <p className='text-muted-foreground text-center py-8'>
             No errors found in the selected time period
           </p>
         </CardContent>

ui/components/landing/cheat-sheet.tsx

@@ -116,9 +116,7 @@ export function CheatSheet(): JSX.Element {
   const [topupToken, setTopupToken] = useState('');
   const [apiKeyInput, setApiKeyInput] = useState('');
   const [walletInfo, setWalletInfo] = useState<WalletSnapshot | null>(null);
-  const [refundReceipt, setRefundReceipt] = useState<RefundReceipt | null>(
-    null
-  );
+  const [refundReceipt, setRefundReceipt] = useState<RefundReceipt | null>(null);
   const [isCreatingKey, setIsCreatingKey] = useState(false);
   const [isTopupLoading, setIsTopupLoading] = useState(false);
   const [isRefunding, setIsRefunding] = useState(false);
@@ -334,21 +332,22 @@ export function CheatSheet(): JSX.Element {
   const showCreateDetails =
     hasInteractedCreate || initialToken.trim().length > 0;
   const showManageDetails = hasInteractedManage || Boolean(walletInfo);
-  const showTopupDetails = hasInteractedTopup || topupToken.trim().length > 0;
+  const showTopupDetails =
+    hasInteractedTopup || topupToken.trim().length > 0;
   const refundToken = refundReceipt?.token ?? null;
   const canTopup = Boolean(activeApiKey);
 
   return (
-    <div className='from-background via-background to-muted min-h-screen bg-gradient-to-b'>
+    <div className='min-h-screen bg-gradient-to-b from-background via-background to-muted'>
       <main className='mx-auto flex w-full max-w-6xl flex-col gap-8 px-4 py-10 sm:px-6 lg:px-8'>
         <section className='relative space-y-3 text-center md:text-left'>
-          <div className='absolute top-0 right-0 hidden md:block'>
+          <div className='absolute right-0 top-0 hidden md:block'>
             <Button asChild size='sm' className='px-3 text-xs'>
               <a href='/login'>Admin</a>
             </Button>
           </div>
-          <div className='text-muted-foreground inline-flex items-center gap-2 rounded-full border px-3 py-1 text-[0.65rem] tracking-wider uppercase'>
-            <Bolt className='text-primary h-4 w-4' />
+          <div className='inline-flex items-center gap-2 rounded-full border px-3 py-1 text-[0.65rem] uppercase tracking-wider text-muted-foreground'>
+            <Bolt className='h-4 w-4 text-primary' />
             Routstr cheat sheet
           </div>
           <h1 className='text-3xl font-semibold tracking-tight sm:text-4xl'>
@@ -366,10 +365,10 @@ export function CheatSheet(): JSX.Element {
             <CardHeader className='flex flex-row items-start justify-between gap-4'>
               <div>
                 <CardTitle className='flex items-center gap-2 text-lg'>
-                  <ShieldCheck className='text-primary h-4 w-4' />
+                  <ShieldCheck className='h-4 w-4 text-primary' />
                   Node identity
                 </CardTitle>
-                <p className='text-muted-foreground text-xs tracking-wide uppercase'>
+                <p className='text-xs uppercase tracking-wide text-muted-foreground'>
                   /v1/info snapshot
                 </p>
               </div>
@@ -405,7 +404,7 @@ export function CheatSheet(): JSX.Element {
                   </div>
                   <dl className='grid gap-4 sm:grid-cols-2'>
                     <div>
-                      <dt className='text-muted-foreground text-xs tracking-wide uppercase'>
+                      <dt className='text-muted-foreground text-xs uppercase tracking-wide'>
                         Version
                       </dt>
                       <dd className='text-base font-medium'>
@@ -413,7 +412,7 @@ export function CheatSheet(): JSX.Element {
                       </dd>
                     </div>
                     <div>
-                      <dt className='text-muted-foreground text-xs tracking-wide uppercase'>
+                      <dt className='text-muted-foreground text-xs uppercase tracking-wide'>
                         HTTP
                       </dt>
                       <dd className='text-base font-medium break-all'>
@@ -422,7 +421,7 @@ export function CheatSheet(): JSX.Element {
                     </div>
                     {nodeInfo.onion_url && (
                       <div className='sm:col-span-2'>
-                        <dt className='text-muted-foreground text-xs tracking-wide uppercase'>
+                        <dt className='text-muted-foreground text-xs uppercase tracking-wide'>
                           Onion
                         </dt>
                         <dd className='text-base font-medium break-all'>
@@ -432,10 +431,10 @@ export function CheatSheet(): JSX.Element {
                     )}
                     {nodeInfo.npub && (
                       <div className='sm:col-span-2'>
-                        <dt className='text-muted-foreground text-xs tracking-wide uppercase'>
+                        <dt className='text-muted-foreground text-xs uppercase tracking-wide'>
                           npub
                         </dt>
-                        <dd className='flex items-center gap-2 font-mono text-sm break-all'>
+                        <dd className='flex items-center gap-2 break-all text-sm font-mono'>
                           {nodeInfo.npub}
                           <Button
                             variant='ghost'
@@ -450,7 +449,7 @@ export function CheatSheet(): JSX.Element {
                     )}
                   </dl>
                   <div className='space-y-2'>
-                    <p className='text-muted-foreground text-xs tracking-wide uppercase'>
+                    <p className='text-xs uppercase tracking-wide text-muted-foreground'>
                       Cashu mints
                     </p>
                     <div className='flex flex-wrap gap-2'>
@@ -479,10 +478,10 @@ export function CheatSheet(): JSX.Element {
           <Card>
             <CardHeader className='flex flex-row items-center justify-between'>
               <CardTitle className='flex items-center gap-2 text-lg'>
-                <Terminal className='text-primary h-4 w-4' />
+                <Terminal className='h-4 w-4 text-primary' />
                 Quick docs
               </CardTitle>
-              <span className='text-muted-foreground text-xs tracking-wide uppercase'>
+              <span className='text-xs uppercase tracking-wide text-muted-foreground'>
                 curl-ready
               </span>
             </CardHeader>
@@ -503,10 +502,8 @@ export function CheatSheet(): JSX.Element {
                   <Copy className='h-4 w-4' />
                 </Button>
               </div>
-              <div className='bg-muted rounded-lg p-4 font-mono text-sm leading-6'>
-                <pre className='break-all whitespace-pre-wrap'>
-                  {curlSnippet}
-                </pre>
+              <div className='rounded-lg bg-muted p-4 font-mono text-sm leading-6'>
+                <pre className='whitespace-pre-wrap break-all'>{curlSnippet}</pre>
               </div>
               <div className='flex gap-2'>
                 <Button
@@ -535,16 +532,16 @@ export function CheatSheet(): JSX.Element {
         <Card>
           <CardHeader className='space-y-1'>
             <CardTitle className='flex items-center gap-2 text-xl'>
-              <KeyRound className='text-primary h-5 w-5' />
+              <KeyRound className='h-5 w-5 text-primary' />
               API key workflow
             </CardTitle>
-            <p className='text-muted-foreground text-xs tracking-wide uppercase'>
+            <p className='text-xs uppercase tracking-wide text-muted-foreground'>
               Sections expand as soon as you interact
             </p>
           </CardHeader>
           <CardContent className='space-y-6'>
             <section className='space-y-2'>
-              <header className='text-muted-foreground flex items-center justify-between text-[0.7rem] tracking-wider uppercase'>
+              <header className='flex items-center justify-between text-[0.7rem] uppercase tracking-wider text-muted-foreground'>
                 <span>1 · Create key</span>
                 {showCreateDetails && (
                   <span className='text-primary'>Cashu token detected</span>
@@ -567,7 +564,7 @@ export function CheatSheet(): JSX.Element {
                   >
                     {isCreatingKey ? 'Creating…' : 'Create API key'}
                   </Button>
-                  <span className='text-muted-foreground text-xs'>
+                  <span className='text-xs text-muted-foreground'>
                     Redeems instantly and returns <code>sk-</code> key.
                   </span>
                 </div>
@@ -577,7 +574,7 @@ export function CheatSheet(): JSX.Element {
             <Separator />
 
             <section className='space-y-2'>
-              <header className='text-muted-foreground flex items-center justify-between text-[0.7rem] tracking-wider uppercase'>
+              <header className='flex items-center justify-between text-[0.7rem] uppercase tracking-wider text-muted-foreground'>
                 <span>2 · Manage key</span>
                 {walletInfo && (
                   <span className='text-primary'>
@@ -621,7 +618,7 @@ export function CheatSheet(): JSX.Element {
               {showManageDetails && (
                 <div className='grid gap-3 sm:grid-cols-2'>
                   <div className='rounded-lg border p-3'>
-                    <p className='text-muted-foreground text-[0.65rem] tracking-wide uppercase'>
+                    <p className='text-[0.65rem] uppercase tracking-wide text-muted-foreground'>
                       Spendable
                     </p>
                     <p className='text-xl font-semibold'>
@@ -636,7 +633,7 @@ export function CheatSheet(): JSX.Element {
                     )}
                   </div>
                   <div className='rounded-lg border p-3'>
-                    <p className='text-muted-foreground text-[0.65rem] tracking-wide uppercase'>
+                    <p className='text-[0.65rem] uppercase tracking-wide text-muted-foreground'>
                       Reserved
                     </p>
                     <p className='text-xl font-semibold'>
@@ -657,12 +654,10 @@ export function CheatSheet(): JSX.Element {
             <Separator />
 
             <section className='space-y-2'>
-              <header className='text-muted-foreground flex items-center justify-between text-[0.7rem] tracking-wider uppercase'>
+              <header className='flex items-center justify-between text-[0.7rem] uppercase tracking-wider text-muted-foreground'>
                 <span>3 · Top up</span>
                 {showTopupDetails && (
-                  <span
-                    className={canTopup ? 'text-primary' : 'text-destructive'}
-                  >
+                  <span className={canTopup ? 'text-primary' : 'text-destructive'}>
                     {canTopup ? 'Ready to redeem' : 'Paste API key first'}
                   </span>
                 )}
@@ -686,14 +681,14 @@ export function CheatSheet(): JSX.Element {
                   >
                     {isTopupLoading ? 'Topping up…' : 'Top up this key'}
                   </Button>
-                  <span className='text-muted-foreground text-xs'>
-                    {canTopup ? (
-                      <>
-                        Adds balance to the same <code>sk-</code> token.
-                      </>
-                    ) : (
-                      'Enter your sk- key above to unlock top ups.'
-                    )}
+                  <span className='text-xs text-muted-foreground'>
+                    {canTopup
+                      ? (
+                          <>
+                            Adds balance to the same <code>sk-</code> token.
+                          </>
+                        )
+                      : 'Enter your sk- key above to unlock top ups.'}
                   </span>
                 </div>
               )}
@@ -702,7 +697,7 @@ export function CheatSheet(): JSX.Element {
             <Separator />
 
             <section className='space-y-2'>
-              <header className='text-muted-foreground flex items-center justify-between text-[0.7rem] tracking-wider uppercase'>
+              <header className='flex items-center justify-between text-[0.7rem] uppercase tracking-wider text-muted-foreground'>
                 <span>4 · Refund</span>
                 {refundReceipt && <span className='text-primary'>Done</span>}
               </header>
@@ -715,13 +710,13 @@ export function CheatSheet(): JSX.Element {
                 >
                   {isRefunding ? 'Processing…' : 'Refund remaining balance'}
                 </Button>
-                <span className='text-muted-foreground text-xs'>
+                <span className='text-xs text-muted-foreground'>
                   Burns the key and returns a fresh Cashu token.
                 </span>
               </div>
               {refundToken && (
-                <div className='bg-muted/30 space-y-2 rounded-lg border p-4'>
-                  <div className='text-muted-foreground flex items-center justify-between text-[0.7rem] tracking-wider uppercase'>
+                <div className='space-y-2 rounded-lg border bg-muted/30 p-4'>
+                  <div className='flex items-center justify-between text-[0.7rem] uppercase tracking-wider text-muted-foreground'>
                     <span>Cashu refund token</span>
                     <Button
                       variant='outline'
@@ -748,3 +743,4 @@ export function CheatSheet(): JSX.Element {
     </div>
   );
 }
+

ui/components/revenue-by-model-table.tsx

@@ -41,11 +41,8 @@ export function RevenueByModelTable({
     <Card>
       <CardHeader>
         <CardTitle>Revenue by Model</CardTitle>
-        <p className='text-muted-foreground text-sm'>
-          Total Revenue:{' '}
-          <span className='text-foreground font-mono font-medium'>
-            {formatAmount(totalRevenue)}
-          </span>
+        <p className="text-sm text-muted-foreground">
+          Total Revenue: <span className="font-mono font-medium text-foreground">{formatAmount(totalRevenue)}</span>
         </p>
       </CardHeader>
       <CardContent>
@@ -53,62 +50,48 @@ export function RevenueByModelTable({
           <TableHeader>
             <TableRow>
               <TableHead>Model</TableHead>
-              <TableHead className='text-right'>Requests</TableHead>
-              <TableHead className='text-right'>Successful</TableHead>
-              <TableHead className='text-right'>Failed</TableHead>
-              <TableHead className='text-right'>Revenue</TableHead>
-              <TableHead className='w-[100px]'>Share</TableHead>
-              <TableHead className='text-right'>Refunds</TableHead>
-              <TableHead className='text-right'>Net Revenue</TableHead>
-              <TableHead className='text-right'>Avg/Request</TableHead>
+              <TableHead className="text-right">Requests</TableHead>
+              <TableHead className="text-right">Successful</TableHead>
+              <TableHead className="text-right">Failed</TableHead>
+              <TableHead className="text-right">Revenue</TableHead>
+              <TableHead className="w-[100px]">Share</TableHead>
+              <TableHead className="text-right">Refunds</TableHead>
+              <TableHead className="text-right">Net Revenue</TableHead>
+              <TableHead className="text-right">Avg/Request</TableHead>
             </TableRow>
           </TableHeader>
           <TableBody>
             {models.length === 0 ? (
               <TableRow>
-                <TableCell
-                  colSpan={9}
-                  className='text-muted-foreground text-center'
-                >
+                <TableCell colSpan={9} className="text-center text-muted-foreground">
                   No model data available
                 </TableCell>
               </TableRow>
             ) : (
               models.map((model) => {
-                const share =
-                  totalRevenue > 0
-                    ? (model.revenue_sats / totalRevenue) * 100
-                    : 0;
+                const share = totalRevenue > 0 ? (model.revenue_sats / totalRevenue) * 100 : 0;
                 return (
                   <TableRow key={model.model}>
-                    <TableCell className='font-medium'>{model.model}</TableCell>
-                    <TableCell className='text-right font-mono'>
-                      {model.requests}
-                    </TableCell>
-                    <TableCell className='text-right font-mono text-green-600'>
-                      {model.successful}
-                    </TableCell>
-                    <TableCell className='text-right font-mono text-red-600'>
-                      {model.failed}
-                    </TableCell>
-                    <TableCell className='text-right font-mono'>
+                    <TableCell className="font-medium">{model.model}</TableCell>
+                    <TableCell className="text-right font-mono">{model.requests}</TableCell>
+                    <TableCell className="text-right text-green-600 font-mono">{model.successful}</TableCell>
+                    <TableCell className="text-right text-red-600 font-mono">{model.failed}</TableCell>
+                    <TableCell className="text-right font-mono">
                       {formatAmount(model.revenue_sats)}
                     </TableCell>
                     <TableCell>
-                      <div className='flex items-center gap-2'>
-                        <Progress value={share} className='h-2' />
-                        <span className='text-muted-foreground w-8 text-right text-xs'>
-                          {share.toFixed(0)}%
-                        </span>
+                      <div className="flex items-center gap-2">
+                        <Progress value={share} className="h-2" />
+                        <span className="text-xs text-muted-foreground w-8 text-right">{share.toFixed(0)}%</span>
                       </div>
                     </TableCell>
-                    <TableCell className='text-right font-mono text-red-500'>
+                    <TableCell className="text-right text-red-500 font-mono">
                       {formatAmount(model.refunds_sats)}
                     </TableCell>
-                    <TableCell className='text-right font-mono font-semibold'>
+                    <TableCell className="text-right font-semibold font-mono">
                       {formatAmount(model.net_revenue_sats)}
                     </TableCell>
-                    <TableCell className='text-muted-foreground text-right font-mono'>
+                    <TableCell className="text-right text-muted-foreground font-mono">
                       {formatAmount(model.avg_revenue_per_request)}
                     </TableCell>
                   </TableRow>

ui/components/temporary-balances.tsx

@@ -9,7 +9,6 @@ import {
   Key,
   Clock,
   DollarSign,
-  Activity,
 } from 'lucide-react';
 import { AdminService, TemporaryBalance } from '@/lib/api/services/admin';
 import {
@@ -56,21 +55,17 @@ export function TemporaryBalances({
 
   const calculateTotals = (balances: TemporaryBalance[]) => {
     let totalBalance = 0;
-    let totalSpent = 0;
-    let totalRequests = 0;
 
     balances.forEach((balance) => {
       totalBalance += balance.balance || 0;
-      totalSpent += balance.total_spent || 0;
-      totalRequests += balance.total_requests || 0;
     });
 
-    return { totalBalance, totalSpent, totalRequests };
+    return { totalBalance };
   };
 
   const totals = data
     ? calculateTotals(data)
-    : { totalBalance: 0, totalSpent: 0, totalRequests: 0 };
+    : { totalBalance: 0 };
 
   return (
     <>
@@ -142,32 +137,6 @@ export function TemporaryBalances({
                     {formatBalance(totals.totalBalance)}
                   </div>
                 </div>
-                <div className='rounded-lg border border-green-200 bg-gradient-to-r from-green-50 to-emerald-50 p-4'>
-                  <div className='flex items-center justify-between'>
-                    <div className='flex items-center gap-2'>
-                      <Activity className='h-5 w-5 text-green-600' />
-                      <span className='text-sm font-medium text-green-800'>
-                        Total Spent
-                      </span>
-                    </div>
-                  </div>
-                  <div className='mt-2 text-2xl font-bold text-green-900'>
-                    {formatBalance(totals.totalSpent)}
-                  </div>
-                </div>
-                <div className='rounded-lg border border-purple-200 bg-gradient-to-r from-purple-50 to-pink-50 p-4'>
-                  <div className='flex items-center justify-between'>
-                    <div className='flex items-center gap-2'>
-                      <Key className='h-5 w-5 text-purple-600' />
-                      <span className='text-sm font-medium text-purple-800'>
-                        Total Requests
-                      </span>
-                    </div>
-                  </div>
-                  <div className='mt-2 text-2xl font-bold text-purple-900'>
-                    {totals.totalRequests.toLocaleString()}
-                  </div>
-                </div>
               </div>
 
               {/* Table */}
@@ -176,9 +145,9 @@ export function TemporaryBalances({
                 <div className='bg-muted hidden grid-cols-6 gap-2 p-3 text-sm font-semibold md:grid'>
                   <div>Hashed Key</div>
                   <div className='text-right'>Balance</div>
-                  <div className='text-right'>Total Spent</div>
-                  <div className='text-right'>Total Requests</div>
                   <div>Refund Address</div>
+                  <div>Refund Mint</div>
+                  <div>Currency</div>
                   <div className='text-right'>Expiry Time</div>
                 </div>
 
@@ -199,22 +168,22 @@ export function TemporaryBalances({
                         <div className='text-right font-mono'>
                           {formatBalance(balance.balance)}
                         </div>
-                        <div className='text-right font-mono'>
-                          {formatBalance(balance.total_spent)}
-                        </div>
-                        <div className='text-right font-mono'>
-                          {balance.total_requests.toLocaleString()}
-                        </div>
                         <div className='max-w-32 truncate font-mono text-xs break-all'>
                           {balance.refund_address || '-'}
                         </div>
+                        <div className='max-w-32 truncate font-mono text-xs break-all'>
+                          {balance.refund_mint_url || '-'}
+                        </div>
+                        <div className='truncate font-mono text-xs'>
+                          {balance.refund_currency || '-'}
+                        </div>
                         <div className='text-right font-mono text-xs'>
-                          {balance.key_expiry_time ? (
+                          {balance.refund_expiration_time ? (
                             <div className='flex items-center justify-end gap-1'>
                               <Clock className='h-3 w-3' />
                               <span>
                                 {new Date(
-                                  balance.key_expiry_time * 1000
+                                  balance.refund_expiration_time * 1000
                                 ).toLocaleDateString()}
                               </span>
                             </div>
@@ -246,34 +215,26 @@ export function TemporaryBalances({
                           </div>
                           <div className='space-y-1'>
                             <div className='text-muted-foreground text-xs font-medium'>
-                              Spent
+                              Currency
                             </div>
                             <div className='truncate font-mono text-sm'>
-                              {formatBalance(balance.total_spent)}
+                              {balance.refund_currency || '-'}
                             </div>
                           </div>
                         </div>
 
                         <div className='grid grid-cols-2 gap-3'>
-                          <div className='space-y-1'>
-                            <div className='text-muted-foreground text-xs font-medium'>
-                              Requests
-                            </div>
-                            <div className='truncate font-mono text-sm'>
-                              {balance.total_requests.toLocaleString()}
-                            </div>
-                          </div>
                           <div className='space-y-1'>
                             <div className='text-muted-foreground text-xs font-medium'>
                               Expires
                             </div>
                             <div className='font-mono text-xs'>
-                              {balance.key_expiry_time ? (
+                              {balance.refund_expiration_time ? (
                                 <div className='flex items-center gap-1'>
                                   <Clock className='h-3 w-3 flex-shrink-0' />
                                   <span className='truncate'>
                                     {new Date(
-                                      balance.key_expiry_time * 1000
+                                      balance.refund_expiration_time * 1000
                                     ).toLocaleDateString()}
                                   </span>
                                 </div>
@@ -294,6 +255,17 @@ export function TemporaryBalances({
                             </div>
                           </div>
                         )}
+                        
+                         {balance.refund_mint_url && (
+                          <div className='space-y-1 pt-2'>
+                            <div className='text-muted-foreground text-xs font-medium'>
+                              Refund Mint
+                            </div>
+                            <div className='font-mono text-xs break-all'>
+                              {balance.refund_mint_url}
+                            </div>
+                          </div>
+                        )}
                       </div>
                     </div>
                   ))

ui/components/usage-metrics-chart.tsx

@@ -90,11 +90,7 @@ export function UsageMetricsChart({
                 boxShadow: '0 4px 6px -1px rgb(0 0 0 / 0.1)',
               }}
               itemStyle={{ fontSize: '12px' }}
-              labelStyle={{
-                fontSize: '12px',
-                color: 'hsl(var(--muted-foreground))',
-                marginBottom: '8px',
-              }}
+              labelStyle={{ fontSize: '12px', color: 'hsl(var(--muted-foreground))', marginBottom: '8px' }}
             />
             <Legend wrapperStyle={{ fontSize: '12px', paddingTop: '16px' }} />
             {dataKeys.map((dataKey) => (

ui/components/usage-summary-cards.tsx

@@ -114,19 +114,15 @@ export function UsageSummaryCards({ summary }: UsageSummaryCardsProps) {
   return (
     <div className='grid gap-6 md:grid-cols-2 lg:grid-cols-4'>
       {cards.map((card) => (
-        <Card key={card.title} className='hover:bg-muted/50 transition-colors'>
+        <Card key={card.title} className="hover:bg-muted/50 transition-colors">
           <CardHeader className='flex flex-row items-center justify-between space-y-0 pb-2'>
-            <CardTitle className='text-muted-foreground text-sm font-medium'>
-              {card.title}
-            </CardTitle>
-            <div className={`bg-secondary rounded-full p-2`}>
-              <card.icon className={`h-4 w-4 ${card.color}`} />
+            <CardTitle className='text-sm font-medium text-muted-foreground'>{card.title}</CardTitle>
+            <div className={`p-2 rounded-full bg-secondary`}>
+                <card.icon className={`h-4 w-4 ${card.color}`} />
             </div>
           </CardHeader>
           <CardContent>
-            <div className='text-2xl font-bold tracking-tight'>
-              {card.value}
-            </div>
+            <div className='text-2xl font-bold tracking-tight'>{card.value}</div>
           </CardContent>
         </Card>
       ))}

ui/lib/api/client.ts

@@ -64,11 +64,6 @@ class ApiClient {
         data,
         config
       );
-      console.log(`POST response from ${endpoint}:`, {
-        status: response.status,
-        data: response.data,
-        headers: response.headers,
-      });
       return response.data;
     } catch (error) {
       this.handleAuthError(error);

ui/lib/api/services/admin.ts

@@ -7,9 +7,6 @@ export const ProviderTypeSchema = z.object({
   default_base_url: z.string(),
   fixed_base_url: z.boolean(),
   platform_url: z.string().nullable(),
-  can_create_account: z.boolean(),
-  can_topup: z.boolean(),
-  can_show_balance: z.boolean(),
 });
 
 export const UpstreamProviderSchema = z.object({
@@ -67,9 +64,7 @@ export const AdminModelSchema = z.object({
   pricing: AdminModelPricingSchema.or(z.record(z.any())),
   per_request_limits: z.record(z.any()).nullable().optional(),
   top_provider: z.record(z.any()).nullable().optional(),
-  upstream_provider_id: z.union([z.string(), z.number()]).nullable().optional(),
-  canonical_slug: z.string().nullable().optional(),
-  alias_ids: z.array(z.string()).nullable().optional(),
+  upstream_provider_id: z.number().nullable().optional(),
   enabled: z.boolean().default(true),
 });
 
@@ -816,9 +811,7 @@ export class AdminService {
     if (search) params.append('search', search);
     params.append('limit', limit.toString());
 
-    return await apiClient.get<LogResponse>(
-      `/admin/api/logs?${params.toString()}`
-    );
+    return await apiClient.get<LogResponse>(`/admin/api/logs?${params.toString()}`);
   }
 
   static async getLogDates(): Promise<{ dates: string[] }> {
@@ -863,66 +856,15 @@ export class AdminService {
       `/admin/api/usage/revenue-by-model?hours=${hours}&limit=${limit}`
     );
   }
-
-  static async createProviderAccountByType(providerType: string): Promise<{
-    ok: boolean;
-    account_data: Record<string, unknown>;
-    message: string;
-  }> {
-    return await apiClient.post<{
-      ok: boolean;
-      account_data: Record<string, unknown>;
-      message: string;
-    }>('/admin/api/upstream-providers/create-account', {
-      provider_type: providerType,
-    });
-  }
-
-  static async initiateProviderTopup(
-    providerId: number,
-    amount: number
-  ): Promise<{
-    ok: boolean;
-    topup_data: Record<string, unknown>;
-    message: string;
-  }> {
-    return await apiClient.post<{
-      ok: boolean;
-      topup_data: Record<string, unknown>;
-      message: string;
-    }>(`/admin/api/upstream-providers/${providerId}/topup`, {
-      amount: amount,
-    });
-  }
-
-  static async checkTopupStatus(
-    providerId: number,
-    invoiceId: string
-  ): Promise<{ ok: boolean; paid: boolean }> {
-    return await apiClient.get<{
-      ok: boolean;
-      paid: boolean;
-    }>(`/admin/api/upstream-providers/${providerId}/topup/${invoiceId}/status`);
-  }
-
-  static async getProviderBalance(providerId: number): Promise<{
-    ok: boolean;
-    balance_data: number | null | Record<string, unknown>;
-  }> {
-    return await apiClient.get<{
-      ok: boolean;
-      balance_data: number | null | Record<string, unknown>;
-    }>(`/admin/api/upstream-providers/${providerId}/balance`);
-  }
 }
 
 export const TemporaryBalanceSchema = z.object({
   hashed_key: z.string(),
   balance: z.number(),
-  total_spent: z.number(),
-  total_requests: z.number(),
   refund_address: z.string().nullable(),
-  key_expiry_time: z.number().nullable(),
+  refund_mint_url: z.string().nullable(),
+  refund_currency: z.string().nullable(),
+  refund_expiration_time: z.number().nullable(),
 });
 
 export type TemporaryBalance = z.infer<typeof TemporaryBalanceSchema>;

ui/lib/stores/currency.ts

@@ -18,3 +18,4 @@ export const useCurrencyStore = create<CurrencyState>()(
     }
   )
 );
+

uv.lock

@@ -1890,7 +1890,6 @@ dependencies = [
     { name = "marshmallow" },
     { name = "mdurl" },
     { name = "nostr" },
-    { name = "openai" },
     { name = "pillow" },
     { name = "python-json-logger" },
     { name = "secp256k1" },
@@ -1924,7 +1923,6 @@ requires-dist = [
     { name = "marshmallow", specifier = ">=3.13,<4.0" },
     { name = "mdurl", specifier = "==0.1.2" },
     { name = "nostr", specifier = ">=0.0.2" },
-    { name = "openai", specifier = ">=1.98.0" },
     { name = "pillow", specifier = ">=10" },
     { name = "python-json-logger", specifier = ">=2.0.0" },
     { name = "secp256k1", git = "https://github.com/saschanaz/secp256k1-py?branch=upgrade060" },