@aomi-labs/client 0.1.18 → 0.1.19

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.18",
+  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"
   }
 };
 
@@ -1161,10 +1162,11 @@ var AomiClient = class {
   // Batch Simulation
   // ===========================================================================
   /**
-   * Simulate pending transactions as an atomic batch.
+   * 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, txIds) {
+  async simulateBatch(sessionId, transactions, options) {
     const url = joinApiPath(this.baseUrl, "/api/simulate");
     const headers = new Headers(
       withSessionHeader(sessionId, { "Content-Type": "application/json" })
@@ -1172,13 +1174,20 @@ var AomiClient = class {
     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({ tx_ids: txIds })
+      body: JSON.stringify(payload)
     });
     if (!response.ok) {
-      throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+      const body = await response.text().catch(() => "");
+      throw new Error(`HTTP ${response.status}: ${response.statusText}${body ? `
+${body}` : ""}`);
     }
     return await response.json();
   }
@@ -2978,7 +2987,7 @@ Run \`aomi tx\` to see available IDs.`
   return pendingTx;
 }
 async function simulateCommand(runtime) {
-  var _a3;
+  var _a3, _b, _c;
   const state = readState();
   if (!state) {
     fatal("No active session. Run `aomi chat` first.");
@@ -2987,9 +2996,7 @@ async function simulateCommand(runtime) {
   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);
-  }
+  const pendingTxs = txIds.map((txId) => requirePendingTx(state, txId));
   console.log(
     `${DIM}Simulating ${txIds.length} transaction(s) as atomic batch...${RESET}`
   );
@@ -2997,7 +3004,23 @@ async function simulateCommand(runtime) {
     baseUrl: state.baseUrl,
     apiKey: state.apiKey
   });
-  const response = await client.simulateBatch(state.sessionId, txIds);
+  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(`
@@ -3032,7 +3055,7 @@ ${DIM}Total gas: ${result.total_gas.toLocaleString()}${RESET}`);
   } 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.`
+      `\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.`
     );
   }
 }
@@ -3374,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,
@@ -3405,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 {
@@ -3839,7 +3921,19 @@ async function signCommand(runtime) {
       try {
         const simResponse = await session.client.simulateBatch(
           state.sessionId,
-          pendingTxs.map((tx) => tx.id)
+          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) {

package/dist/index.cjs

@@ -619,10 +619,11 @@ var AomiClient = class {
   // Batch Simulation
   // ===========================================================================
   /**
-   * Simulate pending transactions as an atomic batch.
+   * 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, txIds) {
+  async simulateBatch(sessionId, transactions, options) {
     const url = joinApiPath(this.baseUrl, "/api/simulate");
     const headers = new Headers(
       withSessionHeader(sessionId, { "Content-Type": "application/json" })
@@ -630,13 +631,20 @@ var AomiClient = class {
     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({ tx_ids: txIds })
+      body: JSON.stringify(payload)
     });
     if (!response.ok) {
-      throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+      const body = await response.text().catch(() => "");
+      throw new Error(`HTTP ${response.status}: ${response.statusText}${body ? `
+${body}` : ""}`);
     }
     return await response.json();
   }
@@ -1981,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,
@@ -2012,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 {