Billing

// Credits are dollar-denominated: 1 credit = 1 cent
const usage = await theazo.billing.getCreditBalance()

console.log(usage.credits)   // 4823 — credits remaining
console.log(usage.currency)  // 'usd'

import { Theazo } from 'theazo'

const theazo = new Theazo({ apiKey: 'th_live_...' })

// Platform-wide budget — the monthly spend ceiling for the whole platform
await theazo.billing.setBudget({
  monthly: { amount: 50000, currency: 'usd' },  // $500/mo across the platform
})

// Per-user budget — the monthly ceiling applied to each end-user
await theazo.billing.setUserBudget({
  monthly: { amount: 2000, currency: 'usd' },    // $20/mo per user
})

// Session-level limit — caps a single session
const session = await theazo.sessions.forUser('user_alice', {
  limits: {
    maxCost: { amount: 500, currency: 'usd', period: 'day' },  // $5/day
    maxAgents: 3,
    maxComputeMinutes: 60,
  },
})

// Run-level limit — caps one agent.run() call
const agent = await session.agents.create({ compute: 'python' })
const result = await agent.run('Summarize this codebase', {
  maxCost: { amount: 50, currency: 'usd' },  // $0.50 max per run
})

// Example budget.threshold payload
{
  "event": "budget.threshold",
  "level": "user",                           // 'platform' | 'user' | 'session' | 'run'
  "userId": "user_alice",
  "threshold": 80,                           // percent consumed
  "consumed": { "amount": 1600, "currency": "usd" },
  "limit":    { "amount": 2000, "currency": "usd" },
  "period": "month"
}

// Example budget.exhausted payload
{
  "event": "budget.exhausted",
  "level": "user",
  "userId": "user_alice",
  "consumed": { "amount": 2001, "currency": "usd" },
  "limit":    { "amount": 2000, "currency": "usd" },
  "agentsPaused": ["agt_x1", "agt_x2"]
}

// Raise the budget and resume the paused session
await theazo.billing.setUserBudget({
  monthly: { amount: 5000, currency: 'usd' },
})

const session = await theazo.sessions.forUser('user_alice')
await session.resume()

// Set the environment when the session is created — agents inherit it
const session = await theazo.sessions.create({
  userId: 'user_alice',
  environment: 'development',   // 'development' | 'staging' | 'production'
})

// Every agent in this session runs (and bills) under that environment
await session.agents.create({ compute: 'python' })

const usage = await theazo.usage.forUser('user_alice', { period: 'month' })

// With BYOI compute, the provider (E2B/Fly/…) bills you for compute directly;
// usage.compute is that portion, usage.total is everything (compute + models + storage).
console.log(usage.compute)   // { minutes: 202, cost: { amount: 870, currency: 'usd' } }
console.log(usage.models)    // { calls: 87, tokens: { input, output }, cost: {...} }
console.log(usage.total)     // { amount: 1010, currency: 'usd' } — full picture

// agent.run.completed payload
{
  "event": "agent.run.completed",
  "agentId": "agt_abc123",
  "runId":   "run_xyz789",
  "userId":  "user_alice",
  "status":  "completed",
  "cost": {
    "compute":       { "amount": 15,  "currency": "usd" },   // sandbox runtime
    "model":         { "amount": 42,  "currency": "usd" },   // model tokens
    "storage":       { "amount": 2,   "currency": "usd" },   // snapshot size
    "orchestration": { "amount": 8,   "currency": "usd" },   // control plane
    "total":         { "amount": 67,  "currency": "usd" },
    "theazoBilled":  { "amount": 67,  "currency": "usd" },
    "providerDirect":{ "amount": 0,   "currency": "usd" }    // 0 = managed compute
  },
  "duration": 43,      // seconds
  "environment": "production"
}

// Get current balance
const balance = await theazo.billing.getCreditBalance()

console.log(balance.credits)   // 4823
console.log(balance.currency)  // 'usd'

// Query usage for one user (period: 'today' | 'week' | 'month')
const usage = await theazo.usage.forUser('user_alice', { period: 'month' })

console.log(usage.total)      // { amount: 1010, currency: 'usd' }
console.log(usage.compute)    // { minutes: 202, cost: {...} }
console.log(usage.models)     // { calls: 87, tokens: {...}, cost: {...} }
console.log(usage.breakdown)  // [{ date, total }, ...] — per-day

// Session-level usage
const session = await theazo.sessions.forUser('user_alice')
const sessionUsage = await session.usage()

console.log(sessionUsage.cost)                // { amount: 234, currency: 'usd' }
console.log(sessionUsage.computeMinutes)      // 47
console.log(sessionUsage.activeAgents)        // 2
console.log(sessionUsage.totalAgentsCreated)  // 12

// Stripe-compatible — attach directly to Stripe invoice line items
const stripe = await theazo.usage.export({ format: 'stripe', period: 'month' })

// CSV — spreadsheet-friendly rows
const csv = await theazo.usage.export({ format: 'csv', period: 'month' })

// JSON — full usage records
const json = await theazo.usage.export({ format: 'json', period: 'month' })