@tokenbuddy/tokenbuddy 1.0.6 → 1.0.7

package/dist/src/terminal-image.js

@@ -0,0 +1,135 @@
+import { spawn } from "child_process";
+import * as fs from "fs";
+import * as path from "path";
+function defaultRunCommand(command, args) {
+    return new Promise((resolve, reject) => {
+        const child = spawn(command, args, {
+            stdio: "ignore",
+        });
+        child.on("error", reject);
+        child.on("close", (code) => {
+            if (code === 0) {
+                resolve();
+                return;
+            }
+            reject(new Error(`${command} exited with ${code}`));
+        });
+    });
+}
+function terminalSupportsInlineImages(env, stdoutIsTTY) {
+    if (!stdoutIsTTY) {
+        return undefined;
+    }
+    const term = env.TERM || "";
+    if (env.KITTY_WINDOW_ID || term.includes("xterm-kitty")) {
+        return "kitty";
+    }
+    const termProgram = env.TERM_PROGRAM || "";
+    if (termProgram === "iTerm.app" || termProgram === "WezTerm" || env.WEZTERM_EXECUTABLE) {
+        return "iterm";
+    }
+    return undefined;
+}
+function basenameBase64(filePath) {
+    return Buffer.from(path.basename(filePath), "utf8").toString("base64");
+}
+function itermInlineImageSequence(filePath, image) {
+    const payload = image.toString("base64");
+    return `\u001B]1337;File=name=${basenameBase64(filePath)};inline=1;width=60%;preserveAspectRatio=1:${payload}\u0007\n`;
+}
+function kittyInlineImageSequence(filePath) {
+    const payload = Buffer.from(filePath, "utf8").toString("base64");
+    return `\u001B_Ga=T,t=f,f=100,c=60;${payload}\u001B\\\n`;
+}
+function openCommandForPlatform(platform, filePath) {
+    if (platform === "darwin") {
+        return {
+            command: "open",
+            args: [filePath],
+            label: `open ${filePath}`,
+        };
+    }
+    if (platform === "linux") {
+        return {
+            command: "xdg-open",
+            args: [filePath],
+            label: `xdg-open ${filePath}`,
+        };
+    }
+    if (platform === "win32") {
+        return {
+            command: "cmd",
+            args: ["/c", "start", "", filePath],
+            label: `start ${filePath}`,
+        };
+    }
+    return undefined;
+}
+export function detectTerminalImageDisplay(options = {}) {
+    return terminalSupportsInlineImages(options.env || process.env, options.stdoutIsTTY ?? Boolean(process.stdout.isTTY));
+}
+export async function displayTerminalImage(filePath, options = {}) {
+    const fileExists = options.fileExists || fs.existsSync;
+    const readFile = options.readFile || fs.readFileSync;
+    const write = options.write || ((chunk) => process.stdout.write(chunk));
+    const env = options.env || process.env;
+    const platform = options.platform || process.platform;
+    const stdoutIsTTY = options.stdoutIsTTY ?? Boolean(process.stdout.isTTY);
+    const openCommand = openCommandForPlatform(platform, filePath);
+    const fallbackCommand = openCommand?.label;
+    if (!fileExists(filePath)) {
+        return {
+            method: "manual",
+            displayed: false,
+            fallbackCommand,
+            message: `QR image file is missing: ${filePath}`,
+        };
+    }
+    const inlineProtocol = terminalSupportsInlineImages(env, stdoutIsTTY);
+    if (inlineProtocol === "kitty") {
+        write(kittyInlineImageSequence(filePath));
+        return {
+            method: "inline-kitty",
+            displayed: true,
+            fallbackCommand,
+            message: "Displayed the ClawTip wallet QR image inline using the Kitty graphics protocol.",
+        };
+    }
+    if (inlineProtocol === "iterm") {
+        write(itermInlineImageSequence(filePath, readFile(filePath)));
+        return {
+            method: "inline-iterm",
+            displayed: true,
+            fallbackCommand,
+            message: "Displayed the ClawTip wallet QR image inline using the iTerm2 image protocol.",
+        };
+    }
+    if (openCommand) {
+        try {
+            const runCommand = options.runCommand || defaultRunCommand;
+            await runCommand(openCommand.command, openCommand.args);
+            return {
+                method: "system-open",
+                displayed: true,
+                fallbackCommand,
+                message: `Opened the ClawTip wallet QR image with the system image viewer: ${openCommand.label}`,
+            };
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : String(error);
+            return {
+                method: "manual",
+                displayed: false,
+                fallbackCommand,
+                message: `Could not open QR image automatically: ${message}`,
+            };
+        }
+    }
+    return {
+        method: "manual",
+        displayed: false,
+        fallbackCommand,
+        message: "This terminal does not advertise inline image support and no system opener is available.",
+    };
+}
+//# sourceMappingURL=terminal-image.js.map

package/dist/src/terminal-image.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"terminal-image.js","sourceRoot":"","sources":["../../src/terminal-image.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAC;AACtC,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AA6B7B,SAAS,iBAAiB,CAAC,OAAe,EAAE,IAAc;IACxD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE;YACjC,KAAK,EAAE,QAAQ;SAChB,CAAC,CAAC;QACH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC1B,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;YACzB,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;gBACf,OAAO,EAAE,CAAC;gBACV,OAAO;YACT,CAAC;YACD,MAAM,CAAC,IAAI,KAAK,CAAC,GAAG,OAAO,gBAAgB,IAAI,EAAE,CAAC,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,4BAA4B,CACnC,GAAgB,EAChB,WAAoB;IAEpB,IAAI,CAAC,WAAW,EAAE,CAAC;QACjB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,IAAI,GAAG,GAAG,CAAC,IAAI,IAAI,EAAE,CAAC;IAC5B,IAAI,GAAG,CAAC,eAAe,IAAI,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;QACxD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,MAAM,WAAW,GAAG,GAAG,CAAC,YAAY,IAAI,EAAE,CAAC;IAC3C,IAAI,WAAW,KAAK,WAAW,IAAI,WAAW,KAAK,SAAS,IAAI,GAAG,CAAC,kBAAkB,EAAE,CAAC;QACvF,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,cAAc,CAAC,QAAgB;IACtC,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;AACzE,CAAC;AAED,SAAS,wBAAwB,CAAC,QAAgB,EAAE,KAAa;IAC/D,MAAM,OAAO,GAAG,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACzC,OAAO,yBAAyB,cAAc,CAAC,QAAQ,CAAC,6CAA6C,OAAO,UAAU,CAAC;AACzH,CAAC;AAED,SAAS,wBAAwB,CAAC,QAAgB;IAChD,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACjE,OAAO,8BAA8B,OAAO,YAAY,CAAC;AAC3D,CAAC;AAED,SAAS,sBAAsB,CAAC,QAAyB,EAAE,QAAgB;IACzE,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC1B,OAAO;YACL,OAAO,EAAE,MAAM;YACf,IAAI,EAAE,CAAC,QAAQ,CAAC;YAChB,KAAK,EAAE,QAAQ,QAAQ,EAAE;SAC1B,CAAC;IACJ,CAAC;IACD,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;QACzB,OAAO;YACL,OAAO,EAAE,UAAU;YACnB,IAAI,EAAE,CAAC,QAAQ,CAAC;YAChB,KAAK,EAAE,YAAY,QAAQ,EAAE;SAC9B,CAAC;IACJ,CAAC;IACD,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;QACzB,OAAO;YACL,OAAO,EAAE,KAAK;YACd,IAAI,EAAE,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,QAAQ,CAAC;YACnC,KAAK,EAAE,SAAS,QAAQ,EAAE;SAC3B,CAAC;IACJ,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,UAAuC,EAAE;IAEzC,OAAO,4BAA4B,CACjC,OAAO,CAAC,GAAG,IAAI,OAAO,CAAC,GAAG,EAC1B,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CACrD,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,QAAgB,EAChB,UAAuC,EAAE;IAEzC,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,EAAE,CAAC,UAAU,CAAC;IACvD,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC,YAAY,CAAC;IACrD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,CAAC,CAAC,KAAa,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;IAChF,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,OAAO,CAAC,GAAG,CAAC;IACvC,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,QAAQ,CAAC;IACtD,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACzE,MAAM,WAAW,GAAG,sBAAsB,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAC/D,MAAM,eAAe,GAAG,WAAW,EAAE,KAAK,CAAC;IAE3C,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC1B,OAAO;YACL,MAAM,EAAE,QAAQ;YAChB,SAAS,EAAE,KAAK;YAChB,eAAe;YACf,OAAO,EAAE,6BAA6B,QAAQ,EAAE;SACjD,CAAC;IACJ,CAAC;IAED,MAAM,cAAc,GAAG,4BAA4B,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;IACtE,IAAI,cAAc,KAAK,OAAO,EAAE,CAAC;QAC/B,KAAK,CAAC,wBAAwB,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC1C,OAAO;YACL,MAAM,EAAE,cAAc;YACtB,SAAS,EAAE,IAAI;YACf,eAAe;YACf,OAAO,EAAE,iFAAiF;SAC3F,CAAC;IACJ,CAAC;IAED,IAAI,cAAc,KAAK,OAAO,EAAE,CAAC;QAC/B,KAAK,CAAC,wBAAwB,CAAC,QAAQ,EAAE,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC;QAC9D,OAAO;YACL,MAAM,EAAE,cAAc;YACtB,SAAS,EAAE,IAAI;YACf,eAAe;YACf,OAAO,EAAE,+EAA+E;SACzF,CAAC;IACJ,CAAC;IAED,IAAI,WAAW,EAAE,CAAC;QAChB,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,iBAAiB,CAAC;YAC3D,MAAM,UAAU,CAAC,WAAW,CAAC,OAAO,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;YACxD,OAAO;gBACL,MAAM,EAAE,aAAa;gBACrB,SAAS,EAAE,IAAI;gBACf,eAAe;gBACf,OAAO,EAAE,oEAAoE,WAAW,CAAC,KAAK,EAAE;aACjG,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACvE,OAAO;gBACL,MAAM,EAAE,QAAQ;gBAChB,SAAS,EAAE,KAAK;gBAChB,eAAe;gBACf,OAAO,EAAE,0CAA0C,OAAO,EAAE;aAC7D,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO;QACL,MAAM,EAAE,QAAQ;QAChB,SAAS,EAAE,KAAK;QAChB,eAAe;QACf,OAAO,EAAE,0FAA0F;KACpG,CAAC;AACJ,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tokenbuddy/tokenbuddy",
-  "version": "1.0.6",
+  "version": "1.0.7",
   "description": "TokenBuddy Client CLI and Daemon",
   "main": "dist/src/index.js",
   "types": "dist/src/index.d.ts",

package/src/buyer-store.ts

@@ -11,6 +11,9 @@ const logger = createModuleLogger("tb-proxyd");
 export interface CachedToken {
   token: string;
   balanceMicros: number;
+  reservedMicros: number;
+  spentMicros: number;
+  balanceSource?: string;
 }
 
 export interface PaymentConfig {
@@ -66,6 +69,12 @@ export interface InferenceLedgerInput {
   promptTokens: number;
   completionTokens: number;
   billedMicros: number;
+  estimatedMicros?: number;
+  settledMicros?: number;
+  settledUsdMicros?: number;
+  priceVersion?: string;
+  balanceSnapshotMicros?: number;
+  balanceSource?: string;
   prompt?: string;
   response?: string;
 }
@@ -79,11 +88,28 @@ export interface SafeInferenceLedgerEntry {
   promptTokens: number;
   completionTokens: number;
   billedMicros: number;
+  estimatedMicros?: number;
+  settledMicros?: number;
+  settledUsdMicros?: number;
+  priceVersion?: string;
+  balanceSnapshotMicros?: number;
+  balanceSource?: string;
   promptHash?: string;
   responseHash?: string;
   createdAt: string;
 }
 
+export interface TokenBalanceSnapshotInput {
+  sellerKey: string;
+  token?: string;
+  tokenClass?: string;
+  balanceMicros: number;
+  reservedMicros?: number;
+  spentMicros?: number;
+  balanceSource: string;
+  expiresAt?: string;
+}
+
 export interface BuyerStoreSummary {
   journalMode: string;
   paymentsCount: number;
@@ -295,41 +321,83 @@ export class BuyerStore {
   }
 
   public getToken(sellerKey: string): CachedToken | undefined {
-    const stmt = this.db.prepare("SELECT token, balance_micros FROM token_cache WHERE seller_key = ?");
-    const row = stmt.get(sellerKey) as { token: string; balance_micros: number } | undefined;
+    const stmt = this.db.prepare(
+      "SELECT token, balance_micros, reserved_micros, spent_micros, balance_source FROM token_cache WHERE seller_key = ?"
+    );
+    const row = stmt.get(sellerKey) as {
+      token: string;
+      balance_micros: number;
+      reserved_micros: number;
+      spent_micros: number;
+      balance_source: string | null;
+    } | undefined;
     if (!row) {
       return undefined;
     }
     return {
       token: row.token,
-      balanceMicros: row.balance_micros
+      balanceMicros: row.balance_micros,
+      reservedMicros: row.reserved_micros,
+      spentMicros: row.spent_micros,
+      balanceSource: row.balance_source || undefined
     };
   }
 
   public saveToken(sellerKey: string, token: string, tokenClass: string, balanceMicros: number, expiresAt: string): void {
     const stmt = this.db.prepare(
       `INSERT OR REPLACE INTO token_cache (
-        seller_key, token, token_class, balance_micros, expires_at, updated_at
-      ) VALUES (?, ?, ?, ?, ?, ?)`
+        seller_key, token, token_class, balance_micros, reserved_micros,
+        spent_micros, balance_source, expires_at, updated_at
+      ) VALUES (?, ?, ?, ?, 0, 0, 'purchase_complete', ?, ?)`
     );
     stmt.run(sellerKey, token, tokenClass, balanceMicros, expiresAt, nowIso());
   }
 
-  public deductBalance(sellerKey: string, amountMicros: number): void {
-    const token = this.getToken(sellerKey);
+  public reconcileTokenBalance(input: TokenBalanceSnapshotInput): void {
+    const current = this.db.prepare(
+      "SELECT token, token_class, expires_at FROM token_cache WHERE seller_key = ?"
+    ).get(input.sellerKey) as {
+      token: string;
+      token_class: string;
+      expires_at: string;
+    } | undefined;
+    const token = input.token || current?.token;
+    const tokenClass = input.tokenClass || current?.token_class || "seller";
+    const expiresAt = input.expiresAt || current?.expires_at || new Date(Date.now() + 86400 * 1000).toISOString();
     if (!token) {
       return;
     }
-    const newBalance = Math.max(0, token.balanceMicros - amountMicros);
-    const stmt = this.db.prepare("UPDATE token_cache SET balance_micros = ?, updated_at = ? WHERE seller_key = ?");
-    stmt.run(newBalance, nowIso(), sellerKey);
-    logger.info("token.cache.debited", "token cache balance debited", {
-      sellerKey,
-      amountMicros,
-      balanceMicros: newBalance
+    this.db.prepare(
+      `INSERT OR REPLACE INTO token_cache (
+        seller_key, token, token_class, balance_micros, reserved_micros,
+        spent_micros, balance_source, expires_at, updated_at
+      ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`
+    ).run(
+      input.sellerKey,
+      token,
+      tokenClass,
+      input.balanceMicros,
+      input.reservedMicros ?? 0,
+      input.spentMicros ?? 0,
+      input.balanceSource,
+      expiresAt,
+      nowIso()
+    );
+    logger.info("token.cache.reconciled", "token cache reconciled from seller balance snapshot", {
+      sellerKey: input.sellerKey,
+      balanceMicros: input.balanceMicros,
+      reservedMicros: input.reservedMicros ?? 0,
+      spentMicros: input.spentMicros ?? 0,
+      balanceSource: input.balanceSource
     });
   }
 
+  public markTokenStale(sellerKey: string): void {
+    this.db.prepare(
+      "UPDATE token_cache SET balance_micros = 0, balance_source = 'stale', updated_at = ? WHERE seller_key = ?"
+    ).run(nowIso(), sellerKey);
+  }
+
   public listPayments(): PaymentConfig[] {
     const rows = this.db.prepare(
       `SELECT method, enabled, is_default, config_json, updated_at
@@ -500,8 +568,10 @@ export class BuyerStore {
     this.db.prepare(
       `INSERT INTO inference_ledger (
         request_id, seller_key, model_id, endpoint, status, prompt_tokens,
-        completion_tokens, billed_micros, prompt_hash, response_hash, created_at
-      ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`
+        completion_tokens, billed_micros, estimated_micros, settled_micros,
+        settled_usd_micros, price_version, balance_snapshot_micros, balance_source,
+        prompt_hash, response_hash, created_at
+      ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`
     ).run(
       input.requestId,
       input.sellerKey,
@@ -511,6 +581,12 @@ export class BuyerStore {
       input.promptTokens,
       input.completionTokens,
       input.billedMicros,
+      input.estimatedMicros ?? input.billedMicros,
+      input.settledMicros ?? null,
+      input.settledUsdMicros ?? null,
+      input.priceVersion || null,
+      input.balanceSnapshotMicros ?? null,
+      input.balanceSource || "unknown",
       safeHash(input.prompt) || null,
       safeHash(input.response) || null,
       nowIso()
@@ -520,7 +596,9 @@ export class BuyerStore {
   public listInferenceLedger(): SafeInferenceLedgerEntry[] {
     const rows = this.db.prepare(
       `SELECT request_id, seller_key, model_id, endpoint, status, prompt_tokens,
-              completion_tokens, billed_micros, prompt_hash, response_hash, created_at
+              completion_tokens, billed_micros, estimated_micros, settled_micros,
+              settled_usd_micros, price_version, balance_snapshot_micros, balance_source,
+              prompt_hash, response_hash, created_at
        FROM inference_ledger
        ORDER BY id ASC`
     ).all() as Array<{
@@ -532,6 +610,12 @@ export class BuyerStore {
       prompt_tokens: number;
       completion_tokens: number;
       billed_micros: number;
+      estimated_micros: number | null;
+      settled_micros: number | null;
+      settled_usd_micros: number | null;
+      price_version: string | null;
+      balance_snapshot_micros: number | null;
+      balance_source: string | null;
       prompt_hash: string | null;
       response_hash: string | null;
       created_at: string;
@@ -546,6 +630,12 @@ export class BuyerStore {
       promptTokens: row.prompt_tokens,
       completionTokens: row.completion_tokens,
       billedMicros: row.billed_micros,
+      estimatedMicros: row.estimated_micros ?? undefined,
+      settledMicros: row.settled_micros ?? undefined,
+      settledUsdMicros: row.settled_usd_micros ?? undefined,
+      priceVersion: row.price_version || undefined,
+      balanceSnapshotMicros: row.balance_snapshot_micros ?? undefined,
+      balanceSource: row.balance_source || undefined,
       promptHash: row.prompt_hash || undefined,
       responseHash: row.response_hash || undefined,
       createdAt: row.created_at
@@ -563,6 +653,9 @@ export class BuyerStore {
         token TEXT NOT NULL,
         token_class TEXT NOT NULL,
         balance_micros INTEGER NOT NULL,
+        reserved_micros INTEGER NOT NULL DEFAULT 0,
+        spent_micros INTEGER NOT NULL DEFAULT 0,
+        balance_source TEXT,
         expires_at TEXT NOT NULL,
         updated_at TEXT NOT NULL
       );
@@ -613,6 +706,12 @@ export class BuyerStore {
         prompt_tokens INTEGER NOT NULL,
         completion_tokens INTEGER NOT NULL,
         billed_micros INTEGER NOT NULL,
+        estimated_micros INTEGER,
+        settled_micros INTEGER,
+        settled_usd_micros INTEGER,
+        price_version TEXT,
+        balance_snapshot_micros INTEGER,
+        balance_source TEXT,
         prompt_hash TEXT,
         response_hash TEXT,
         created_at TEXT NOT NULL
@@ -639,6 +738,30 @@ export class BuyerStore {
         updated_at TEXT NOT NULL
       );
     `);
+    for (const [column, definition] of [
+      ["reserved_micros", "INTEGER NOT NULL DEFAULT 0"],
+      ["spent_micros", "INTEGER NOT NULL DEFAULT 0"],
+      ["balance_source", "TEXT"]
+    ]) {
+      this.ensureColumn("token_cache", column, definition);
+    }
+    for (const [column, definition] of [
+      ["estimated_micros", "INTEGER"],
+      ["settled_micros", "INTEGER"],
+      ["settled_usd_micros", "INTEGER"],
+      ["price_version", "TEXT"],
+      ["balance_snapshot_micros", "INTEGER"],
+      ["balance_source", "TEXT"]
+    ]) {
+      this.ensureColumn("inference_ledger", column, definition);
+    }
+  }
+
+  private ensureColumn(table: string, column: string, definition: string): void {
+    const rows = this.db.prepare(`PRAGMA table_info(${table})`).all() as Array<{ name: string }>;
+    if (!rows.some((row) => row.name === column)) {
+      this.db.exec(`ALTER TABLE ${table} ADD COLUMN ${column} ${definition}`);
+    }
   }
 
   private countRows(tableName: string): number {