pip install agent-budget

uv pip install agent-budget

# Create a budget
agent-budget create --name "API Costs" --limit 500 --period monthly --category api_calls

# Record a cost
agent-budget cost record <budget_id> --amount 2.50 --source "openai-gpt4" --description "Chat completion"

# Check budget status
agent-budget show <budget_id>

# List all budgets
agent-budget list

# Project spending
agent-budget analyze project <budget_id>

# Breakdown by source
agent-budget analyze by-source <budget_id>

# Start the MCP server
agent-budget-server

{
  "mcpServers": {
    "agent-budget": {
      "command": "python",
      "args": ["-m", "agent_budget.server"]
    }
  }
}

from agent_budget.engine import BudgetEngine
from agent_budget.models import BudgetPeriod, CostCategory

engine = BudgetEngine()

# Create a budget
budget = engine.create_budget(
    name="API Costs",
    limit=500.0,
    period=BudgetPeriod.MONTHLY,
    category=CostCategory.API_CALLS,
)

# Record costs
entry, alerts = engine.record_cost(
    budget_id=budget.id,
    amount=2.50,
    source="openai-gpt4",
    description="Chat completion",
)

# Check if budget is on track
projection = engine.project_spending(budget.id)
print(f"On track: {projection.on_track}")
print(f"Projected total: {projection.projected_total}")

# Get cost breakdown
by_source = engine.get_costs_by_source(budget.id)
by_category = engine.get_costs_by_category(budget.id)

agent_budget/
├── __init__.py      # Package init
├── models.py        # Pydantic models (Budget, CostEntry, etc.)
├── engine.py        # Business logic (BudgetEngine)
├── store.py         # JSON persistence layer
├── server.py        # MCP server
└── cli.py           # Click CLI