aium

AI usage monitor: a Python CLI core that polls your AI providers and a GNOME Shell extension that displays the results. Think of it as a system monitor, but for AI providers.

Features

• Panel indicator — robot icon with monthly spend and balance (stacked two-line label).
• Popover — per-provider: balance, monthly spend, rate-limit quota windows, sparklines (spend or quota trend), plan, and a peak/off-peak dot for providers with dynamic pricing (DeepSeek).
• Clickable providers — open the provider's usage/dashboard page.
• Provider abstraction — add a provider with one file + one registry entry.
• Secrets — API keys in the system keyring; OAuth providers reuse the CLI's own credential files (Codex, Claude Code, Antigravity).
• History — SQLite time-series of balances, quota and usage.
• systemd timer — polls every 60 minutes, no resident daemon.

Screenshots

Installation

Quick install (CLI + timer + extension)

curl -fsSL https://raw.githubusercontent.com/jonykalavera/aium/main/install.sh | bash

Pin a specific release with AIUM_VERSION:

curl -fsSL https://raw.githubusercontent.com/jonykalavera/aium/main/install.sh | AIUM_VERSION=v0.1.1 bash

The script installs the aium CLI from PyPI (aium-cli), the systemd user timer (polls every 60 min) and the GNOME Shell extension. Requires curl, unzip, and one of uv / pipx / pip. Restart GNOME Shell (logout/login) to load the extension.

CLI only

pipx install aium-cli     # or: uv tool install aium-cli
aium init

The CLI reads its config from ~/.config/aium/ (YAML) and keeps history in ~/.local/share/aium/ (SQLite).

Extension only

Grab aium@jonykalavera.zip from the releases page, extract it to ~/.local/share/gnome-shell/extensions/aium@jonykalavera/, compile the schema (glib-compile-schemas schemas) and enable it with gnome-extensions enable aium@jonykalavera.

From source (development)

./local-install.sh         # installs the CLI (uv tool), the systemd timer and the extension from this repo

Quick start

aium providers add deepseek
aium keys set deepseek                  # prompts for the API key (keyring)
aium poll
aium status

Providers

Balance semantics. "Balance" means different things per provider:

• Prepaid credit (balance/credits) — real money on the account. OpenRouter reports it as credits because BYOK/free usage is billed elsewhere and does not decrement it.
• Budget (budget) — Anthropic's remaining monthly spending limit, not money you hold.
• None — the provider exposes no balance (quota/plan only).

The aggregated Prepaid balance total sums only prepaid-credit balances; budget and quota-only providers are shown per-row but excluded from the total.

OAuth providers reuse the CLI's own credential files (~/.codex/auth.json, ~/.claude/.credentials.json, ~/.gemini/oauth_creds.json) — no API key needed. Their usage endpoints are private/undocumented and may break.

Commands

aium providers add|list|show|update|remove
aium providers update deepseek --peak-window 00:30-16:30   # UTC high-tariff window
aium keys set|list|delete
aium poll                 # fetch every provider, persist history, refresh cache
aium status               # show the last cached status
aium history <id>         # balance history for a provider

peak_window marks the UTC peak (high-tariff) hours as 'HH:MM-HH:MM' (wrapping across midnight allowed); the provider row shows a peak/off-peak indicator. DeepSeek defaults to 00:30-16:30.

Architecture

src/aium/            # Python core (config, secrets, history, polling, providers)
extension/           # GNOME Shell outlet (GJS) — reads ~/.cache/aium/status.json
  lib/               #   pure logic (unit-tested), no gi:// imports
  tests/             #   GJS unit tests (gjs -m extension/tests/run.js)
systemd/             # aium-poll timer/service (user units)
tests/               # pytest (core)

The extension is a passive outlet: no provider CRUD, no secrets. It renders ~/.cache/aium/status.json, whose schema (models.py::StatusFile) is the contract between the core and the extension.

Development

uv sync --all-groups --locked   # install dev deps
make uv.check                   # ruff lint + ty typecheck + format check
make uv.test                    # pytest --cov aium --blockage (network blocked)
make ext-test                   # GJS unit tests for the extension logic

Tools: ruff (lint/format), ty (type checker), pytest (coverage + network blockage). Recipes use bare commands; prefix with uv. to run inside the uv environment. To iterate on the extension on Wayland (no hot reload), use ./scripts/dev-nested.sh (requires mutter-devkit).

License

MIT — see LICENSE.