From 006422400d3724dde3bbe38462ce75bc809032d9 Mon Sep 17 00:00:00 2001
From: tom <tom@aisa.one>
Date: Fri, 11 Sep 2026 12:00:14 +0800
Subject: [PATCH] feat(account): show go-to-market balance

---
 src/commands/account.ts |  1 +
 src/types.ts            |  1 +
 tests/account.test.ts   | 10 ++++++----
 3 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/commands/account.ts b/src/commands/account.ts
index 5c6e9ee..685709e 100644
--- a/src/commands/account.ts
+++ b/src/commands/account.ts
@@ -35,6 +35,7 @@ export async function balanceAction(options: { json?: boolean } = {}): Promise<v
   const balance = res.data;
   console.log(`  Balance:             ${formatMicrosUSD(balance.account_balance_micros_usd)} ${balance.currency}`);
   console.log(`  Available with key:  ${formatMicrosUSD(balance.available_balance_micros_usd)} ${balance.currency}`);
+  console.log(`  GTM balance:         ${formatMicrosUSD(balance.go_to_market_balance_micros_usd)} ${balance.currency}`);
   if (balance.api_key.unlimited) {
     console.log("  API key limit:       Unlimited");
   } else {
diff --git a/src/types.ts b/src/types.ts
index 511d187..b01a44f 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -33,6 +33,7 @@ export interface BalanceResponse {
   currency: string;
   account_balance_micros_usd: number;
   available_balance_micros_usd: number;
+  go_to_market_balance_micros_usd: number;
   api_key: {
     unlimited: boolean;
     remaining_micros_usd: number;
diff --git a/tests/account.test.ts b/tests/account.test.ts
index a1c1008..f06dc79 100644
--- a/tests/account.test.ts
+++ b/tests/account.test.ts
@@ -5,6 +5,7 @@ const balance = {
   currency: "USD",
   account_balance_micros_usd: 2_500_000,
   available_balance_micros_usd: 1_000_000,
+  go_to_market_balance_micros_usd: 400_000,
   api_key: {
     unlimited: false,
     remaining_micros_usd: 1_000_000,
@@ -42,10 +43,11 @@ describe("balanceAction", () => {
     const [url, init] = fetchMock.mock.calls[0];
     expect(url).toBe("https://api.aisa.one/v1/credits/balance");
     expect((init?.headers as Record<string, string>).Authorization).toBe("Bearer test-balance-key");
-    expect(log.mock.calls.map(([line]) => line).join("\n")).toContain("Balance:");
-    expect(log.mock.calls.map(([line]) => line).join("\n")).toContain("$2.50 USD");
-    expect(log.mock.calls.map(([line]) => line).join("\n")).toContain("$1.00 USD");
-    expect(log.mock.calls.map(([line]) => line).join("\n")).not.toContain("test-balance-key");
+    const output = log.mock.calls.map(([line]) => line).join("\n");
+    expect(output).toContain("Balance:             $2.50 USD");
+    expect(output).toContain("Available with key:  $1.00 USD");
+    expect(output).toContain("GTM balance:         $0.40 USD");
+    expect(output).not.toContain("test-balance-key");
   });
 
   it("renders unlimited keys", async () => {