@aomi-labs/client 0.1.17 → 0.1.19

package/README.md

@@ -307,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.17",
+  version: "0.1.19",
   description: "Platform-agnostic TypeScript client for the Aomi backend API",
   type: "module",
   main: "./dist/index.cjs",
@@ -53,9 +53,10 @@ var package_default = {
     "clean:dist": "rm -rf dist"
   },
   dependencies: {
-    "@getpara/aa-alchemy": "2.18.0",
-    "@getpara/aa-pimlico": "2.18.0",
-    viem: "^2.40.3"
+    "@alchemy/wallet-apis": "5.0.0-beta.22",
+    "@getpara/aa-alchemy": "2.21.0",
+    "@getpara/aa-pimlico": "2.21.0",
+    viem: "^2.47.11"
   }
 };
 
@@ -68,17 +69,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);
 }
@@ -202,7 +203,7 @@ 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,
@@ -1157,6 +1158,39 @@ var AomiClient = class {
     }
     return postState(this.baseUrl, "/api/control/model", payload, sessionId, apiKey);
   }
+  // ===========================================================================
+  // Batch Simulation
+  // ===========================================================================
+  /**
+   * Simulate transactions as an atomic batch.
+   * Each tx sees state changes from previous txs (e.g., approve → swap).
+   * Sends full tx payloads — the backend does not look up by ID.
+   */
+  async simulateBatch(sessionId, transactions, options) {
+    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 payload = {
+      transactions,
+      from: options == null ? void 0 : options.from,
+      chain_id: options == null ? void 0 : options.chainId
+    };
+    const response = await fetch(url, {
+      method: "POST",
+      headers,
+      body: JSON.stringify(payload)
+    });
+    if (!response.ok) {
+      const body = await response.text().catch(() => "");
+      throw new Error(`HTTP ${response.status}: ${response.statusText}${body ? `
+${body}` : ""}`);
+    }
+    return await response.json();
+  }
 };
 
 // src/types.ts
@@ -2940,6 +2974,92 @@ async function sessionCommand(runtime) {
   );
 }
 
+// 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, _b, _c;
+  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.");
+  }
+  const pendingTxs = txIds.map((txId) => 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 transactions = pendingTxs.map((tx) => {
+    var _a4;
+    return {
+      to: tx.to,
+      value: tx.value,
+      data: tx.data,
+      label: (_a4 = tx.description) != null ? _a4 : tx.id
+    };
+  });
+  const response = await client.simulateBatch(
+    state.sessionId,
+    transactions,
+    {
+      from: (_a3 = state.publicKey) != null ? _a3 : void 0,
+      chainId: (_b = state.chainId) != null ? _b : void 0
+    }
+  );
+  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 ${(_c = failed == null ? void 0 : failed.step) != null ? _c : "?"}.${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";
@@ -3277,6 +3397,25 @@ async function createAlchemyAAState(options) {
   if (ownerParams.kind === "unsupported_adapter") {
     return getUnsupportedAdapterState(plan, ownerParams.adapter);
   }
+  if (owner.kind === "direct") {
+    try {
+      return await createAlchemyWalletApisState({
+        plan,
+        chain,
+        privateKey: owner.privateKey,
+        apiKey,
+        gasPolicyId,
+        mode: plan.mode
+      });
+    } catch (error) {
+      return {
+        plan,
+        AA: null,
+        isPending: false,
+        error: error instanceof Error ? error : new Error(String(error))
+      };
+    }
+  }
   try {
     const smartAccount = await createAlchemySmartAccount(__spreadProps(__spreadValues({}, ownerParams.ownerParams), {
       apiKey,
@@ -3308,6 +3447,46 @@ async function createAlchemyAAState(options) {
     };
   }
 }
+async function createAlchemyWalletApisState(params) {
+  const { createSmartWalletClient, alchemyWalletTransport } = await import("@alchemy/wallet-apis");
+  const signer = privateKeyToAccount(params.privateKey);
+  const walletClient = createSmartWalletClient(__spreadValues({
+    transport: alchemyWalletTransport({ apiKey: params.apiKey }),
+    chain: params.chain,
+    signer
+  }, params.gasPolicyId ? { paymaster: { policyId: params.gasPolicyId } } : {}));
+  let accountAddress = signer.address;
+  if (params.mode === "4337") {
+    const account = await walletClient.requestAccount();
+    accountAddress = account.address;
+  }
+  const sendCalls = async (calls) => {
+    var _a3, _b;
+    const result = await walletClient.sendCalls(__spreadProps(__spreadValues({}, params.mode === "4337" ? { account: accountAddress } : {}), {
+      calls
+    }));
+    const status = await walletClient.waitForCallsStatus({ id: result.id });
+    const transactionHash = (_b = (_a3 = status.receipts) == null ? void 0 : _a3[0]) == null ? void 0 : _b.transactionHash;
+    if (!transactionHash) {
+      throw new Error("Alchemy Wallets API did not return a transaction hash.");
+    }
+    return { transactionHash };
+  };
+  const AA = {
+    provider: "alchemy",
+    mode: params.mode,
+    AAAddress: accountAddress,
+    delegationAddress: params.mode === "7702" ? signer.address : void 0,
+    sendTransaction: async (call) => sendCalls([call]),
+    sendBatchTransaction: async (calls) => sendCalls(calls)
+  };
+  return {
+    plan: params.plan,
+    AA,
+    isPending: false,
+    error: null
+  };
+}
 async function createPimlicoAAState(options) {
   var _a3;
   const {
@@ -3491,7 +3670,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) {
@@ -3507,7 +3686,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;
@@ -3669,7 +3848,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(
@@ -3739,6 +3918,46 @@ 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) => {
+            var _a4;
+            return {
+              to: tx.to,
+              value: tx.value,
+              data: tx.data,
+              label: (_a4 = tx.description) != null ? _a4 : tx.id
+            };
+          }),
+          {
+            from: account.address,
+            chainId: primaryChainId
+          }
+        );
+        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,
@@ -3875,6 +4094,8 @@ Usage:
                         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
@@ -3914,7 +4135,7 @@ 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
@@ -3959,6 +4180,9 @@ async function main(runtime) {
     case "sign":
       await signCommand(runtime);
       break;
+    case "simulate":
+      await simulateCommand(runtime);
+      break;
     case "status":
       await statusCommand(runtime);
       break;
@@ -3996,7 +4220,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);
@@ -4009,7 +4233,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,39 @@ var AomiClient = class {
     }
     return postState(this.baseUrl, "/api/control/model", payload, sessionId, apiKey);
   }
+  // ===========================================================================
+  // Batch Simulation
+  // ===========================================================================
+  /**
+   * Simulate transactions as an atomic batch.
+   * Each tx sees state changes from previous txs (e.g., approve → swap).
+   * Sends full tx payloads — the backend does not look up by ID.
+   */
+  async simulateBatch(sessionId, transactions, options) {
+    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 payload = {
+      transactions,
+      from: options == null ? void 0 : options.from,
+      chain_id: options == null ? void 0 : options.chainId
+    };
+    const response = await fetch(url, {
+      method: "POST",
+      headers,
+      body: JSON.stringify(payload)
+    });
+    if (!response.ok) {
+      const body = await response.text().catch(() => "");
+      throw new Error(`HTTP ${response.status}: ${response.statusText}${body ? `
+${body}` : ""}`);
+    }
+    return await response.json();
+  }
 };
 
 // src/types.ts
@@ -1956,6 +1989,25 @@ async function createAlchemyAAState(options) {
   if (ownerParams.kind === "unsupported_adapter") {
     return getUnsupportedAdapterState(plan, ownerParams.adapter);
   }
+  if (owner.kind === "direct") {
+    try {
+      return await createAlchemyWalletApisState({
+        plan,
+        chain,
+        privateKey: owner.privateKey,
+        apiKey,
+        gasPolicyId,
+        mode: plan.mode
+      });
+    } catch (error) {
+      return {
+        plan,
+        AA: null,
+        isPending: false,
+        error: error instanceof Error ? error : new Error(String(error))
+      };
+    }
+  }
   try {
     const smartAccount = await (0, import_aa_alchemy.createAlchemySmartAccount)(__spreadProps(__spreadValues({}, ownerParams.ownerParams), {
       apiKey,
@@ -1987,6 +2039,46 @@ async function createAlchemyAAState(options) {
     };
   }
 }
+async function createAlchemyWalletApisState(params) {
+  const { createSmartWalletClient, alchemyWalletTransport } = await import("@alchemy/wallet-apis");
+  const signer = (0, import_accounts.privateKeyToAccount)(params.privateKey);
+  const walletClient = createSmartWalletClient(__spreadValues({
+    transport: alchemyWalletTransport({ apiKey: params.apiKey }),
+    chain: params.chain,
+    signer
+  }, params.gasPolicyId ? { paymaster: { policyId: params.gasPolicyId } } : {}));
+  let accountAddress = signer.address;
+  if (params.mode === "4337") {
+    const account = await walletClient.requestAccount();
+    accountAddress = account.address;
+  }
+  const sendCalls = async (calls) => {
+    var _a, _b;
+    const result = await walletClient.sendCalls(__spreadProps(__spreadValues({}, params.mode === "4337" ? { account: accountAddress } : {}), {
+      calls
+    }));
+    const status = await walletClient.waitForCallsStatus({ id: result.id });
+    const transactionHash = (_b = (_a = status.receipts) == null ? void 0 : _a[0]) == null ? void 0 : _b.transactionHash;
+    if (!transactionHash) {
+      throw new Error("Alchemy Wallets API did not return a transaction hash.");
+    }
+    return { transactionHash };
+  };
+  const AA = {
+    provider: "alchemy",
+    mode: params.mode,
+    AAAddress: accountAddress,
+    delegationAddress: params.mode === "7702" ? signer.address : void 0,
+    sendTransaction: async (call) => sendCalls([call]),
+    sendBatchTransaction: async (calls) => sendCalls(calls)
+  };
+  return {
+    plan: params.plan,
+    AA,
+    isPending: false,
+    error: null
+  };
+}
 async function createPimlicoAAState(options) {
   var _a;
   const {