@aomi-labs/client 0.1.16 → 0.1.18

package/README.md

@@ -126,6 +126,7 @@ npx @aomi-labs/client chat "swap 1 ETH" --verbose        # stream tool calls + r
 npx @aomi-labs/client app list                           # list available apps
 npx @aomi-labs/client model list                         # list available models
 npx @aomi-labs/client model set claude-sonnet-4          # switch the current session model
+npx @aomi-labs/client session new                        # create a fresh active session
 npx @aomi-labs/client secret list                        # list configured secret handles
 npx @aomi-labs/client --secret ALCHEMY_API_KEY=...       # ingest a secret for the active session
 npx @aomi-labs/client log                                # show full conversation history
@@ -149,6 +150,31 @@ npx @aomi-labs/client chat "send 0 ETH to myself" \
 The address is persisted in the state file, so subsequent commands in the same
 session don't need it again.
 
+### Chain selection
+
+Use `--chain <id>` for the current command when the task is chain-specific:
+
+```bash
+$ npx @aomi-labs/client chat "swap 1 POL for USDC on Polygon" --chain 137
+```
+
+Use `AOMI_CHAIN_ID` when several consecutive commands should share the same
+chain context.
+
+### Fresh sessions
+
+Use `--new-session` when you want a command to start a fresh backend/local
+session instead of reusing the currently active one:
+
+```bash
+$ npx @aomi-labs/client chat "show my balances" --new-session
+$ npx @aomi-labs/client --secret ALCHEMY_API_KEY=... --new-session
+$ npx @aomi-labs/client session new
+```
+
+This is useful when starting a new operator flow or a new external chat thread
+and you do not want stale session state to bleed into the next run.
+
 ### Model selection
 
 The CLI can discover and switch backend models for the active session:
@@ -281,7 +307,7 @@ All config can be passed as flags (which take priority over env vars):
 
 | Flag                    | Env Variable      | Default                | Description                                  |
 | ----------------------- | ----------------- | ---------------------- | -------------------------------------------- |
-| `--backend-url`         | `AOMI_BASE_URL`   | `https://api.aomi.dev` | Backend URL                                  |
+| `--backend-url`         | `AOMI_BACKEND_URL`   | `https://api.aomi.dev` | Backend URL                                  |
 | `--api-key`             | `AOMI_API_KEY`    | —                      | API key for non-default apps                 |
 | `--app`                 | `AOMI_APP`        | `default`              | App                                          |
 | `--model`               | `AOMI_MODEL`      | —                      | Model rig to apply before chat               |

package/dist/cli.js

@@ -22,7 +22,7 @@ var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
 // package.json
 var package_default = {
   name: "@aomi-labs/client",
-  version: "0.1.16",
+  version: "0.1.18",
   description: "Platform-agnostic TypeScript client for the Aomi backend API",
   type: "module",
   main: "./dist/index.cjs",
@@ -68,17 +68,17 @@ var CliExit = class extends Error {
 };
 function fatal(message) {
   const RED = "\x1B[31m";
-  const DIM3 = "\x1B[2m";
-  const RESET3 = "\x1B[0m";
+  const DIM2 = "\x1B[2m";
+  const RESET2 = "\x1B[0m";
   const lines = message.split("\n");
   const [headline, ...details] = lines;
-  console.error(`${RED}\u274C ${headline}${RESET3}`);
+  console.error(`${RED}\u274C ${headline}${RESET2}`);
   for (const detail of details) {
     if (!detail.trim()) {
       console.error("");
       continue;
     }
-    console.error(`${DIM3}${detail}${RESET3}`);
+    console.error(`${DIM2}${detail}${RESET2}`);
   }
   throw new CliExit(1);
 }
@@ -139,6 +139,12 @@ function parseSecret(value, secrets) {
     secrets[value.slice(0, eqIdx)] = value.slice(eqIdx + 1);
   }
 }
+function normalizePrivateKey(value) {
+  if (value === void 0) return void 0;
+  const trimmed = value.trim();
+  if (!trimmed) return void 0;
+  return trimmed.startsWith("0x") ? trimmed : `0x${trimmed}`;
+}
 function parseArgs(argv) {
   const raw = argv.slice(2);
   const command = raw[0] && !raw[0].startsWith("-") ? raw[0] : void 0;
@@ -196,12 +202,15 @@ function getConfig(parsed) {
     fatal("`--aa-provider` and `--aa-mode` cannot be used with `--eoa`.");
   }
   return {
-    baseUrl: (_d = (_c = parsed.flags["backend-url"]) != null ? _c : process.env.AOMI_BASE_URL) != null ? _d : "https://api.aomi.dev",
+    baseUrl: (_d = (_c = parsed.flags["backend-url"]) != null ? _c : process.env.AOMI_BACKEND_URL) != null ? _d : "https://api.aomi.dev",
     apiKey: (_e = parsed.flags["api-key"]) != null ? _e : process.env.AOMI_API_KEY,
     app: (_g = (_f = parsed.flags["app"]) != null ? _f : process.env.AOMI_APP) != null ? _g : "default",
     model: (_h = parsed.flags["model"]) != null ? _h : process.env.AOMI_MODEL,
+    freshSession: parsed.flags["new-session"] === "true",
     publicKey: (_i = parsed.flags["public-key"]) != null ? _i : process.env.AOMI_PUBLIC_KEY,
-    privateKey: (_j = parsed.flags["private-key"]) != null ? _j : process.env.PRIVATE_KEY,
+    privateKey: normalizePrivateKey(
+      (_j = parsed.flags["private-key"]) != null ? _j : process.env.PRIVATE_KEY
+    ),
     chainRpcUrl: (_k = parsed.flags["rpc-url"]) != null ? _k : process.env.CHAIN_RPC_URL,
     chain: parseChainId((_l = parsed.flags["chain"]) != null ? _l : process.env.AOMI_CHAIN_ID),
     secrets: parsed.secrets,
@@ -1148,6 +1157,31 @@ var AomiClient = class {
     }
     return postState(this.baseUrl, "/api/control/model", payload, sessionId, apiKey);
   }
+  // ===========================================================================
+  // Batch Simulation
+  // ===========================================================================
+  /**
+   * Simulate pending transactions as an atomic batch.
+   * Each tx sees state changes from previous txs (e.g., approve → swap).
+   */
+  async simulateBatch(sessionId, txIds) {
+    const url = joinApiPath(this.baseUrl, "/api/simulate");
+    const headers = new Headers(
+      withSessionHeader(sessionId, { "Content-Type": "application/json" })
+    );
+    if (this.apiKey) {
+      headers.set(API_KEY_HEADER, this.apiKey);
+    }
+    const response = await fetch(url, {
+      method: "POST",
+      headers,
+      body: JSON.stringify({ tx_ids: txIds })
+    });
+    if (!response.ok) {
+      throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+    }
+    return await response.json();
+  }
 };
 
 // src/types.ts
@@ -1792,20 +1826,29 @@ function buildCliUserState(publicKey, chainId) {
 }
 
 // src/cli/context.ts
-function getOrCreateSession(runtime) {
+function buildSessionState(runtime) {
+  const { config } = runtime;
+  return {
+    sessionId: crypto.randomUUID(),
+    baseUrl: config.baseUrl,
+    app: config.app,
+    model: config.model,
+    apiKey: config.apiKey,
+    publicKey: config.publicKey,
+    chainId: config.chain,
+    clientId: crypto.randomUUID()
+  };
+}
+function createFreshSessionState(runtime) {
+  const state = buildSessionState(runtime);
+  writeState(state);
+  return state;
+}
+function getOrCreateSession(runtime, options = {}) {
   const { config } = runtime;
-  let state = readState();
+  let state = options.fresh || config.freshSession ? createFreshSessionState(runtime) : readState();
   if (!state) {
-    state = {
-      sessionId: crypto.randomUUID(),
-      baseUrl: config.baseUrl,
-      app: config.app,
-      apiKey: config.apiKey,
-      publicKey: config.publicKey,
-      chainId: config.chain,
-      clientId: crypto.randomUUID()
-    };
-    writeState(state);
+    state = createFreshSessionState(runtime);
   } else {
     let changed = false;
     if (config.baseUrl !== state.baseUrl) {
@@ -1987,7 +2030,9 @@ async function chatCommand(runtime) {
     fatal("Usage: aomi chat <message>");
   }
   const verbose = runtime.parsed.flags["verbose"] === "true" || runtime.parsed.flags["v"] === "true";
-  const { session, state } = getOrCreateSession(runtime);
+  const { session, state } = getOrCreateSession(runtime, {
+    fresh: runtime.config.freshSession
+  });
   try {
     await ingestSecretsIfPresent(runtime, state, session.client);
     await applyRequestedModelIfPresent(runtime, session, state);
@@ -2740,7 +2785,9 @@ async function ingestSecretsCommand(runtime) {
   if (secretEntries.length === 0) {
     fatal("Usage: aomi --secret NAME=value [NAME=value ...]");
   }
-  const { session, state } = getOrCreateSession(runtime);
+  const { session, state } = getOrCreateSession(runtime, {
+    fresh: runtime.config.freshSession
+  });
   try {
     const handles = await ingestSecretsIfPresent(
       runtime,
@@ -2873,6 +2920,12 @@ async function sessionsCommand(_runtime) {
 async function sessionCommand(runtime) {
   const subcommand = runtime.parsed.positional[0];
   const selector = runtime.parsed.positional[1];
+  if (subcommand === "new") {
+    const state = createFreshSessionState(runtime);
+    console.log(`Active session set to ${state.sessionId} (new).`);
+    printDataFileLocation();
+    return;
+  }
   if (subcommand === "resume") {
     if (!selector) {
       fatal("Usage: aomi session resume <session-id|session-N|N>");
@@ -2908,10 +2961,82 @@ async function sessionCommand(runtime) {
     return;
   }
   fatal(
-    "Usage: aomi session list\n       aomi session resume <session-id|session-N|N>\n       aomi session delete <session-id|session-N|N>"
+    "Usage: aomi session list\n       aomi session new\n       aomi session resume <session-id|session-N|N>\n       aomi session delete <session-id|session-N|N>"
   );
 }
 
+// src/cli/commands/simulate.ts
+function requirePendingTx(state, txId) {
+  var _a3;
+  const pendingTx = ((_a3 = state.pendingTxs) != null ? _a3 : []).find((tx) => tx.id === txId);
+  if (!pendingTx) {
+    fatal(
+      `No pending transaction with id "${txId}".
+Run \`aomi tx\` to see available IDs.`
+    );
+  }
+  return pendingTx;
+}
+async function simulateCommand(runtime) {
+  var _a3;
+  const state = readState();
+  if (!state) {
+    fatal("No active session. Run `aomi chat` first.");
+  }
+  const txIds = runtime.parsed.positional;
+  if (txIds.length === 0) {
+    fatal("Usage: aomi simulate <tx-id> [<tx-id> ...]\nRun `aomi tx` to see available IDs.");
+  }
+  for (const txId of txIds) {
+    requirePendingTx(state, txId);
+  }
+  console.log(
+    `${DIM}Simulating ${txIds.length} transaction(s) as atomic batch...${RESET}`
+  );
+  const client = new AomiClient({
+    baseUrl: state.baseUrl,
+    apiKey: state.apiKey
+  });
+  const response = await client.simulateBatch(state.sessionId, txIds);
+  const { result } = response;
+  const modeLabel = result.stateful ? "stateful (Anvil snapshot)" : "stateless (independent eth_call)";
+  console.log(`
+Batch simulation (${modeLabel}):`);
+  console.log(`From: ${result.from} | Network: ${result.network}
+`);
+  for (const step of result.steps) {
+    const icon = step.success ? `${GREEN}\u2713${RESET}` : `\x1B[31m\u2717${RESET}`;
+    const label = step.label || `Step ${step.step}`;
+    const gasInfo = step.gas_used ? ` | gas: ${step.gas_used.toLocaleString()}` : "";
+    console.log(`  ${icon} ${step.step}. ${label}`);
+    console.log(`    ${DIM}to: ${step.tx.to} | value: ${step.tx.value_eth} ETH${gasInfo}${RESET}`);
+    if (!step.success && step.revert_reason) {
+      console.log(`    \x1B[31mRevert: ${step.revert_reason}${RESET}`);
+    }
+  }
+  if (result.total_gas) {
+    console.log(`
+${DIM}Total gas: ${result.total_gas.toLocaleString()}${RESET}`);
+  }
+  if (result.fee) {
+    const feeEth = (Number(result.fee.amount_wei) / 1e18).toFixed(6);
+    console.log(
+      `Service fee: ${feeEth} ETH \u2192 ${result.fee.recipient}`
+    );
+  }
+  console.log();
+  if (result.batch_success) {
+    console.log(
+      `${GREEN}All steps passed.${RESET} Run \`aomi sign ${txIds.join(" ")}\` to execute.`
+    );
+  } else {
+    const failed = result.steps.find((s) => !s.success);
+    console.log(
+      `\x1B[31mBatch failed at step ${(_a3 = failed == null ? void 0 : failed.step) != null ? _a3 : "?"}.${RESET} Fix the issue and re-queue, or run \`aomi sign\` on the successful prefix.`
+    );
+  }
+}
+
 // src/cli/commands/wallet.ts
 import { createWalletClient, http } from "viem";
 import { privateKeyToAccount as privateKeyToAccount2 } from "viem/accounts";
@@ -3463,7 +3588,7 @@ function txCommand() {
   }
   printDataFileLocation();
 }
-function requirePendingTx(state, txId) {
+function requirePendingTx2(state, txId) {
   var _a3;
   const pendingTx = ((_a3 = state.pendingTxs) != null ? _a3 : []).find((tx) => tx.id === txId);
   if (!pendingTx) {
@@ -3479,7 +3604,7 @@ function requirePendingTxs(state, txIds) {
   if (uniqueIds.length !== txIds.length) {
     fatal("Duplicate transaction IDs are not allowed in a single `aomi sign` call.");
   }
-  return uniqueIds.map((txId) => requirePendingTx(state, txId));
+  return uniqueIds.map((txId) => requirePendingTx2(state, txId));
 }
 function rewriteSessionState(runtime, state) {
   let changed = false;
@@ -3641,7 +3766,7 @@ async function executeTransactionWithFallback(params) {
   }
 }
 async function signCommand(runtime) {
-  var _a3, _b;
+  var _a3, _b, _c, _d;
   const txIds = runtime.parsed.positional;
   if (txIds.length === 0) {
     fatal(
@@ -3711,6 +3836,34 @@ async function signCommand(runtime) {
       if (callList.length > 1 && rpcUrl && new Set(callList.map((call) => call.chainId)).size > 1) {
         fatal("A single `--rpc-url` override cannot be used for a mixed-chain multi-sign request.");
       }
+      try {
+        const simResponse = await session.client.simulateBatch(
+          state.sessionId,
+          pendingTxs.map((tx) => tx.id)
+        );
+        const { result: sim } = simResponse;
+        if (!sim.batch_success) {
+          const failed = sim.steps.find((s) => !s.success);
+          fatal(
+            `Simulation failed at step ${(_c = failed == null ? void 0 : failed.step) != null ? _c : "?"}: ${(_d = failed == null ? void 0 : failed.revert_reason) != null ? _d : "unknown"}`
+          );
+        }
+        if (sim.fee) {
+          const feeEth = (Number(sim.fee.amount_wei) / 1e18).toFixed(6);
+          console.log(
+            `Fee:     ${feeEth} ETH \u2192 ${sim.fee.recipient.slice(0, 10)}...`
+          );
+          callList.push({
+            to: sim.fee.recipient,
+            value: sim.fee.amount_wei,
+            chainId: primaryChainId
+          });
+        }
+      } catch (e) {
+        console.log(
+          `${DIM}Simulation unavailable, skipping fee injection.${RESET}`
+        );
+      }
       const decision = resolveCliExecutionDecision({
         config: runtime.config,
         chain,
@@ -3840,12 +3993,15 @@ Usage:
   aomi model set <rig>  Set the active model for the current session
   aomi chain list       List supported chains
   aomi session list     List local sessions with metadata
+  aomi session new      Start a fresh local/backend session and make it active
   aomi session resume <id>
                         Resume a local session (session-id or session-N)
   aomi session delete <id>
                         Delete a local session file (session-id or session-N)
   aomi log              Show full conversation history with tool results
   aomi tx               List pending and signed transactions
+  aomi simulate <tx-id> [<tx-id> ...]
+                        Batch-simulate pending txs atomically (approve \u2192 swap)
   aomi sign <tx-id> [<tx-id> ...] [--eoa | --aa] [--aa-provider <name>] [--aa-mode <mode>]
                         Sign and submit a pending transaction
   aomi secret list      List configured secrets for the active session
@@ -3860,6 +4016,8 @@ Options:
   --api-key <key>       API key for non-default apps
   --app <name>          App (default: "default")
   --model <rig>         Set the active model for this session
+  --new-session         Create a fresh active session for this command
+  --chain <id>          Active chain for chat/session context
   --public-key <addr>   Wallet address (so the agent knows your wallet)
   --private-key <key>   Hex private key for signing
   --rpc-url <url>       RPC URL for transaction submission
@@ -3883,10 +4041,11 @@ Default signing behavior:
                         fall back to EOA automatically if AA is unavailable
 
 Environment (overridden by flags):
-  AOMI_BASE_URL         Backend URL
+  AOMI_BACKEND_URL         Backend URL
   AOMI_API_KEY          API key
   AOMI_APP              App
   AOMI_MODEL            Model rig
+  AOMI_CHAIN_ID         Active chain for chat/session context
   AOMI_PUBLIC_KEY       Wallet address
   AOMI_AA_PROVIDER      AA provider: alchemy | pimlico
   AOMI_AA_MODE          AA mode: 4337 | 7702
@@ -3927,6 +4086,9 @@ async function main(runtime) {
     case "sign":
       await signCommand(runtime);
       break;
+    case "simulate":
+      await simulateCommand(runtime);
+      break;
     case "status":
       await statusCommand(runtime);
       break;
@@ -3964,7 +4126,7 @@ function isPnpmExecWrapper() {
 async function runCli(argv = process.argv) {
   const runtime = createRuntime(argv);
   const RED = "\x1B[31m";
-  const RESET3 = "\x1B[0m";
+  const RESET2 = "\x1B[0m";
   const strictExit = process.env.AOMI_CLI_STRICT_EXIT === "1";
   try {
     await main(runtime);
@@ -3977,7 +4139,7 @@ async function runCli(argv = process.argv) {
       return;
     }
     const message = err instanceof Error ? err.message : String(err);
-    console.error(`${RED}\u274C ${message}${RESET3}`);
+    console.error(`${RED}\u274C ${message}${RESET2}`);
     process.exit(1);
   }
 }

package/dist/index.cjs

@@ -615,6 +615,31 @@ var AomiClient = class {
     }
     return postState(this.baseUrl, "/api/control/model", payload, sessionId, apiKey);
   }
+  // ===========================================================================
+  // Batch Simulation
+  // ===========================================================================
+  /**
+   * Simulate pending transactions as an atomic batch.
+   * Each tx sees state changes from previous txs (e.g., approve → swap).
+   */
+  async simulateBatch(sessionId, txIds) {
+    const url = joinApiPath(this.baseUrl, "/api/simulate");
+    const headers = new Headers(
+      withSessionHeader(sessionId, { "Content-Type": "application/json" })
+    );
+    if (this.apiKey) {
+      headers.set(API_KEY_HEADER, this.apiKey);
+    }
+    const response = await fetch(url, {
+      method: "POST",
+      headers,
+      body: JSON.stringify({ tx_ids: txIds })
+    });
+    if (!response.ok) {
+      throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+    }
+    return await response.json();
+  }
 };
 
 // src/types.ts