pi-agent-budget

Pi extension: AI coding cost tracker with widget, budget limits, and CNY pricing for Chinese models.

Packages

Package details

extension

Install pi-agent-budget from npm and Pi will load the resources declared by the package manifest.

npm repo home report

$ pi install npm:pi-agent-budget

Package: pi-agent-budget
Version: 0.1.6
Published: Jun 9, 2026
Downloads: not available
Author: rex_young
License: MIT
Types: extension
Size: 193 KB
Dependencies: 1 dependency · 2 peers

Pi manifest JSON

{
  "extensions": [
    "./src/index.ts"
  ],
  "image": "https://raw.githubusercontent.com/RexYoung000/pi-agent-budget/main/preview.png"
}

Security note

Pi packages can execute code and influence agent behavior. Review the source before installing third-party packages.

README

pi-agent-budget

Pi extension: real-time AI coding cost tracking with budget limits and native CNY pricing for Chinese models.

Track token usage and costs as you chat with AI. A live widget sits below the input box showing your spending, with soft/hard budget limits to prevent surprise bills.

Features

👁️ Real-time cost tracking — every assistant message is recorded: tokens × pricing = cost
📊 Status bar widget — live display below the editor: ¥0.16 / ¥10.00 ██░░░░ (model + cost + budget + progress bar)
🇨🇳 CNY pricing for Chinese models — deepseek / qwen / glm / kimi / doubao priced in RMB with auto FX conversion (1 USD ≈ 7.2 CNY)
🛡️ Soft/hard budget limits — 80% triggers a warning card, 100% blocks further input
📈 /budget dashboard — overview + budget settings + pricing table + config, all in a TUI panel
📋 /budget report — cost breakdowns by model / tool / session

Install

pi install npm:pi-agent-budget

Depends on better-sqlite3 (native C++ addon). Requires build toolchain on first install: Xcode CLT on macOS (xcode-select --install), build-essential on Linux.

Commands

/budget                     # Dashboard (overview / budget / pricing / settings)
/budget set <amount>        # Set session budget      e.g. /budget set 2
/budget set project <amount> # Set project daily budget
/budget report [range]      # Cost report             e.g. /budget report week
/budget by <dim> [range]    # Slice by model / tool / session
/budget widget              # Widget appearance settings
/budget export              # Export cost data

Configuration

Edit via /budget config or ~/.pi/budget.json:

{
  "session": { "usd": 2.00 },
  "pricingAnchor": "followModel",
  "widget": {
    "placement": "belowEditor",
    "compact": false,
    "showModel": true,
    "showProgressBar": true,
    "showContextWindow": true
  }
}

Setting	Description	Default
`session.usd`	Session budget in USD	Unlimited
`pricingAnchor`	Currency anchor: `followModel` / `cny` / `usd`	`followModel`
`widget.placement`	Widget position: `belowEditor` / `aboveEditor`	`belowEditor`
`widget.compact`	Minimal mode (cost only)	`false`
`softLimitRatio`	Soft limit threshold	`0.8`
`hardLimitRatio`	Hard limit threshold	`1.0`

Currency handling

followModel (default): Chinese models (deepseek/qwen/glm etc.) display in ¥, Western models display in $
cny: Everything in RMB, budget presets use RMB amounts
usd: Everything in USD

Chinese models use RMB price tables internally; all costs are stored in USD for consistent cross-model aggregation, with display-time conversion.

Data storage

File	Purpose
`~/.pi/budget.db`	SQLite (costs / budgets / alerts)
`~/.pi/budget.json`	User configuration

Development

npm run check          # TypeScript type check
pi -e ./src/index.ts   # Local dev testing

License

MIT