@owney/sdk 0.7.25-beta.15 → 0.7.25-beta.16

package/dist/index.js

@@ -370,20 +370,44 @@ function observeWithdrawalSubmission(sdk, chainId, token) {
         };
         const notify = (outcome) => {
           if (outcome === "rejected" && matching.length !== 1) return;
+          const evidence = matching.length === 1 ? { ...request2 } : {
+            attempted: request2.attempted,
+            method: request2.method,
+            path: request2.path,
+            ...request2.elapsedMs !== void 0 ? { elapsedMs: request2.elapsedMs } : {}
+          };
           for (const observer of matching) {
             try {
-              observer.notify(outcome, { ...request2 });
+              observer.notify(outcome, evidence);
             } catch {
             }
           }
         };
         notify("attempted");
+        let callArgs = args;
+        if (withdrawal && client.client?.defaults) {
+          const configIndex = method === "get" ? 1 : 2;
+          const config = args[configIndex];
+          const transforms = config?.transformResponse !== void 0 ? config.transformResponse : client.client.defaults.transformResponse;
+          const captureStatus = (data, _headers, status) => {
+            if (typeof status === "number" && Number.isInteger(status) && status >= 100 && status <= 599) request2.httpStatus = status;
+            return data;
+          };
+          callArgs = [...args];
+          callArgs[configIndex] = { ...config, transformResponse: [
+            ...Array.isArray(transforms) ? transforms : typeof transforms === "function" ? [transforms] : [],
+            captureStatus
+          ] };
+        }
         try {
-          return await original.apply(this, args);
+          return await original.apply(this, callArgs);
         } catch (error) {
           const message = error instanceof Error ? error.message : "";
-          request2.failure = message === "Unauthorized: Invalid API key" ? "unauthorized" : message === "Forbidden: Access denied" ? "forbidden" : message === "Rate limit exceeded. Please try again later." ? "rate-limited" : /^Idle balance \d+(?:\.\d+)? (USDC|WETH|ETH|EURC) is below the minimum withdrawal of \d+(?:\.\d+)? \1\.?$/.test(message) ? "insufficient-balance" : /^User doesn't have any active session key\.?$/.test(message) ? "session-key-missing" : message === "Network error: Unable to reach the server" ? "network" : "other";
-          if (["unauthorized", "forbidden", "rate-limited", "insufficient-balance", "session-key-missing"].includes(request2.failure)) notify("rejected");
+          const status = error?.response?.status;
+          if (request2.httpStatus === void 0 && typeof status === "number" && Number.isInteger(status) && status >= 100 && status <= 599)
+            request2.httpStatus = status;
+          request2.failure = message === "Unauthorized: Invalid API key" ? "unauthorized" : message === "Forbidden: Access denied" ? "forbidden" : message === "Rate limit exceeded. Please try again later." ? "rate-limited" : /^Idle balance \d+(?:\.\d+)? (USDC|WETH|ETH|EURC) is below the minimum withdrawal of \d+(?:\.\d+)? \1\.?$/.test(message) ? "insufficient-balance" : /^User doesn't have any active session key\.?$/.test(message) ? "session-key-missing" : method === "post" && request2.httpStatus === 400 && message === "Partial withdrawal already in progress" ? "withdrawal-in-progress" : message === "Network error: Unable to reach the server" ? "network" : "other";
+          if (["unauthorized", "forbidden", "rate-limited", "insufficient-balance", "session-key-missing", "withdrawal-in-progress"].includes(request2.failure) && (request2.httpStatus === void 0 || request2.httpStatus >= 400 && request2.httpStatus < 500)) notify("rejected");
           throw error;
         } finally {
           request2.elapsedMs = Math.max(0, Date.now() - startedAt);
@@ -1464,6 +1488,7 @@ var VIEM_CHAIN = { 8453: base, 42161: arbitrum, 1: mainnet };
 var SPLITTING_ASSETS = ["USDC", "WETH"];
 var POOL_POLICY_CHAIN_IDS = [8453, 42161, 1];
 var EARNINGS_REFRESH_COOLDOWN_MS = 60 * 60 * 1e3;
+var WITHDRAW_BALANCE_TIMEOUT_MS = 2e3;
 var ZyfaiAgent = class _ZyfaiAgent {
   id = "zyfai";
   supportedChainIds = [8453, 42161, 1];
@@ -2378,9 +2403,74 @@ var ZyfaiAgent = class _ZyfaiAgent {
     }
   }
   // --- IAgent: Fund operations ---
+  /** Only after a definitive endpoint rejection, combine a new provider
+   * snapshot with the Safe's exact idle balance. Never parse its rounded error
+   * amount as proof of zero, or use this to skip sending a new MAX request. */
+  async assertNotEmptyRejectedWithdrawal(state, chainId, token, owner) {
+    const internals = this.sdk;
+    const generation = this.earningsGeneration;
+    let expired = false;
+    const unavailable = () => new OwneyError(
+      "WITHDRAW_BALANCE_UNAVAILABLE",
+      "Could not verify Zyfai's balance after its withdrawal rejection.",
+      { chainId, token },
+      this.id
+    );
+    const assertScope = () => {
+      if (expired) throw unavailable();
+      const account = internals.walletClient?.account?.address;
+      if (generation !== this.earningsGeneration || state.chainId !== chainId || state.walletAddress.toLowerCase() !== owner.toLowerCase() || this.connectedAddress?.toLowerCase() !== owner.toLowerCase() || this.connectedChainId !== chainId || account && account.toLowerCase() !== owner.toLowerCase())
+        throw new OwneyError("NOT_CONNECTED", "Wallet changed while preparing Zyfai withdrawal.", void 0, this.id);
+    };
+    const read2 = async () => {
+      assertScope();
+      const [wallet, registered, portfolio] = await Promise.all([
+        this.sdk.getSmartWalletAddress(owner, chainId),
+        this.sdk.getSmartWalletByEOA(owner),
+        this.sdk.getPortfolio(owner).catch(() => void 0)
+      ]);
+      assertScope();
+      if (!isAddress(wallet.address) || !registered.smartWallet || registered.smartWallet.toLowerCase() !== wallet.address.toLowerCase())
+        throw new OwneyError(
+          "AGENT_AUTH_FAILED",
+          "Could not verify Zyfai's withdrawal wallet.",
+          { reason: "ZYFAI_SESSION_UNAVAILABLE", phase: "prepare", chainId, token },
+          this.id
+        );
+      if (hasEmptyWithdrawalSnapshot(portfolio, chainId, token)) {
+        const asset = SupportedAssets.find((value) => value.chainId === chainId && value.symbol === token);
+        const idle = asset ? await this.getPublicClient(chainId).readContract({
+          address: asset.address,
+          abi: erc20Abi2,
+          functionName: "balanceOf",
+          args: [wallet.address]
+        }).catch(() => void 0) : void 0;
+        assertScope();
+        if (idle === 0n) throw new OwneyError(
+          "WITHDRAW_INSUFFICIENT_BALANCE",
+          "No funds remain in Zyfai for this asset.",
+          { available: "0", chainId, token },
+          this.id
+        );
+      }
+    };
+    let timer;
+    try {
+      await Promise.race([read2(), new Promise((_, reject) => {
+        timer = setTimeout(() => {
+          expired = true;
+          reject(unavailable());
+        }, WITHDRAW_BALANCE_TIMEOUT_MS);
+      })]);
+    } finally {
+      expired = true;
+      if (timer) clearTimeout(timer);
+    }
+  }
   /** Called only after the execution endpoint explicitly rejected a request.
    * A timeout or an accepted request never authorizes repair and resubmission. */
   async recoverWithdrawalSession(state, chainId, token, owner) {
+    await this.assertNotEmptyRejectedWithdrawal(state, chainId, token, owner);
     const internals = this.sdk;
     const assertScope = () => {
       const account = internals.walletClient?.account?.address;
@@ -2388,35 +2478,6 @@ var ZyfaiAgent = class _ZyfaiAgent {
         throw new OwneyError("NOT_CONNECTED", "Wallet changed while preparing Zyfai withdrawal.", void 0, this.id);
     };
     assertScope();
-    const [wallet, registered, portfolio] = await Promise.all([
-      this.sdk.getSmartWalletAddress(owner, chainId),
-      this.sdk.getSmartWalletByEOA(owner),
-      this.sdk.getPortfolio(owner).catch(() => void 0)
-    ]);
-    assertScope();
-    if (!isAddress(wallet.address) || !registered.smartWallet || registered.smartWallet.toLowerCase() !== wallet.address.toLowerCase())
-      throw new OwneyError(
-        "AGENT_AUTH_FAILED",
-        "Could not verify Zyfai's withdrawal wallet.",
-        { reason: "ZYFAI_SESSION_UNAVAILABLE", phase: "prepare", chainId, token },
-        this.id
-      );
-    if (hasEmptyWithdrawalSnapshot(portfolio, chainId, token)) {
-      const asset = SupportedAssets.find((value) => value.chainId === chainId && value.symbol === token);
-      const idle = asset ? await this.getPublicClient(chainId).readContract({
-        address: asset.address,
-        abi: erc20Abi2,
-        functionName: "balanceOf",
-        args: [wallet.address]
-      }).catch(() => void 0) : void 0;
-      assertScope();
-      if (idle === 0n) throw new OwneyError(
-        "WITHDRAW_INSUFFICIENT_BALANCE",
-        "No funds remain in Zyfai for this asset.",
-        { available: "0", chainId, token },
-        this.id
-      );
-    }
     const walletState = await this.readWalletState(owner);
     assertScope();
     const unavailable = () => new OwneyError(
@@ -2468,6 +2529,13 @@ var ZyfaiAgent = class _ZyfaiAgent {
           raw2 = await this.sdk.withdrawFunds(owner, validChainId, amount, token);
           break;
         } catch (error) {
+          if (!amount && observation.rejected() && observation.request()?.failure === "insufficient-balance") {
+            try {
+              await this.assertNotEmptyRejectedWithdrawal(state, validChainId, token, owner);
+            } catch (proofError) {
+              if (proofError instanceof OwneyError && (proofError.code === "WITHDRAW_INSUFFICIENT_BALANCE" || proofError.code === "NOT_CONNECTED")) throw proofError;
+            }
+          }
           if (!observation.rejected() || observation.request()?.failure !== "session-key-missing") throw error;
           if (attempt === 1)
             throw new OwneyError(
@@ -2561,6 +2629,67 @@ var ZyfaiAgent = class _ZyfaiAgent {
   async getFundingWallet(state, chainId) {
     return (await this.sdk.getSmartWalletAddress(state.walletAddress, isValidChainId(chainId))).address;
   }
+  /** A positive Safe balance is a withdrawal planning floor when the portfolio
+   * index reports zero. Idle zero cannot prove that Zyfai's protocol positions
+   * are empty: its SDK does not expose a complete contract catalog for them. */
+  async getWithdrawableBalance(state, chainId, token) {
+    const validChainId = isValidChainId(chainId);
+    const asset = SupportedAssets.find((value) => value.chainId === validChainId && value.symbol === token);
+    if (!asset) throw new OwneyError("WITHDRAW_NO_PERMITTED_TOKENS", "Asset selected is not supported", void 0, this.id);
+    const owner = state.walletAddress;
+    const generation = this.earningsGeneration;
+    const internals = this.sdk;
+    let expired = false;
+    const unavailable = () => new OwneyError(
+      "WITHDRAW_BALANCE_UNAVAILABLE",
+      "Could not verify Zyfai's withdrawal balance. Its vault positions may still contain funds.",
+      { reason: "ZYFAI_VAULT_BALANCE_UNVERIFIED", chainId, token },
+      this.id
+    );
+    const assertScope = () => {
+      if (expired) throw unavailable();
+      const account = internals.walletClient?.account?.address;
+      if (generation !== this.earningsGeneration || this.connectedAddress?.toLowerCase() !== owner.toLowerCase() || this.connectedChainId !== validChainId || state.walletAddress.toLowerCase() !== owner.toLowerCase() || state.chainId !== validChainId || account && account.toLowerCase() !== owner.toLowerCase())
+        throw new OwneyError("NOT_CONNECTED", "Wallet changed while checking Zyfai's withdrawal balance.", void 0, this.id);
+    };
+    let timer;
+    const read2 = async () => {
+      assertScope();
+      const [wallet, registered] = await Promise.all([
+        this.sdk.getSmartWalletAddress(owner, validChainId),
+        this.sdk.getSmartWalletByEOA(owner)
+      ]);
+      assertScope();
+      if (!isAddress(wallet.address) || wallet.isDeployed !== true || wallet.isOwner !== true || !registered.smartWallet || registered.smartWallet.toLowerCase() !== wallet.address.toLowerCase()) throw unavailable();
+      const client = this.getPublicClient(validChainId);
+      const blockNumber = await client.getBlockNumber({ cacheTime: 0 });
+      assertScope();
+      const idle = await client.readContract({
+        address: asset.address,
+        abi: erc20Abi2,
+        functionName: "balanceOf",
+        args: [wallet.address],
+        blockNumber
+      });
+      assertScope();
+      if (typeof idle !== "bigint" || idle <= 0n) throw unavailable();
+      return idle;
+    };
+    try {
+      return await Promise.race([read2(), new Promise((_, reject) => {
+        timer = setTimeout(() => {
+          expired = true;
+          reject(unavailable());
+        }, WITHDRAW_BALANCE_TIMEOUT_MS);
+      })]);
+    } catch (error) {
+      if (error instanceof OwneyError) throw error;
+      throw unavailable();
+    } finally {
+      expired = true;
+      if (timer) clearTimeout(timer);
+    }
+  }
   async getBalances(state, chainId) {
     const requestedAt = Date.now();
     const { smartWallet, chainId: validChainId } = await this.resolveSmartWallet(state, chainId);
@@ -4573,6 +4702,8 @@ var YIELDSEEKER_SETTLEMENT_WINDOW_MS = 5 * 6e4;
 var YIELDSEEKER_ACTIVITY_CACHE_MS = 6e4;
 var YIELDSEEKER_MAX_WITHDRAW_SWEEPS = 3;
 var YIELDSEEKER_MAX_POSITION_VERIFICATION_MS = 2e3;
+var YIELDSEEKER_WITHDRAWAL_CANDIDATE_LIMIT = 64;
+var YIELDSEEKER_VAULTS_PER_RPC_BATCH = 10;
 function generateYieldseekerUsername() {
   const suffix = globalThis.crypto.randomUUID().replaceAll("-", "").slice(0, YIELDSEEKER_USERNAME_RANDOM_LENGTH).toLowerCase();
   return `${YIELDSEEKER_USERNAME_PREFIX}${suffix}`;
@@ -5096,17 +5227,38 @@ var YieldseekerAgent = class {
     let settlement = null;
     try {
       const maxSettlementBasis = amount === void 0 ? this.snapshotMovement(state, chainId, context, 0n) : null;
-      this.advancePortfolioVersion(this.contextKey(state, chainId, asset));
-      const portfolioRead = amount === void 0 ? this.loadActivityContext(state, chainId, context, {}).then((value) => ({ positions: value.positions, feeRateBps: 0 })) : this.loadPortfolioContext(state, chainId, context).then((value) => ({ positions: value.positions, feeRateBps: value.snapshot.feeRateBps }));
+      const contextKey = this.contextKey(state, chainId, asset);
+      const priorPositions = this.withdrawalVaultHints(state, chainId, context);
+      this.advancePortfolioVersion(contextKey);
+      const boundedVerification = async (read2, timeoutMs = YIELDSEEKER_MAX_POSITION_VERIFICATION_MS) => {
+        let timer;
+        try {
+          return await Promise.race([read2, new Promise((resolve) => {
+            timer = setTimeout(() => resolve(void 0), Math.max(0, timeoutMs));
+          })]);
+        } finally {
+          if (timer) clearTimeout(timer);
+        }
+      };
+      const feeRateRead = amount === void 0 ? Promise.resolve(0) : boundedVerification(
+        (() => {
+          try {
+            return this.portfolioSnapshot(state, chainId, context, contextKey).read.then((value) => value.agentSnapshot.feeRateBps);
+          } catch (error) {
+            return Promise.reject(error);
+          }
+        })().catch(() => void 0)
+      );
+      const positionsRead = this.readWithdrawalPositions(state, chainId, context);
       const metadata = YIELDSEEKER_ASSET_METADATA[asset];
       const agentWallet = getAddress4(context.wallet.walletAddress);
-      const feeTrackerRead = withPaidRpcDiagnostics(() => this.getReceiptClient().readContract({
-        address: agentWallet,
-        abi: YIELDSEEKER_AGENT_WALLET_ABI,
-        functionName: "feeTracker"
-      }), chainId, "eth_call", this.id);
-      const readOwnerFunds = async () => {
-        const feeTracker = await feeTrackerRead;
+      const readOwnerFunds = async (blockNumber) => {
+        const feeTracker = await withPaidRpcDiagnostics(() => this.getReceiptClient().readContract({
+          address: agentWallet,
+          abi: YIELDSEEKER_AGENT_WALLET_ABI,
+          functionName: "feeTracker",
+          blockNumber
+        }), chainId, "eth_call", this.id);
         this.assertSession(generation, chainId, asset);
         if (!isAddress3(feeTracker) || /^0x0{40}$/i.test(feeTracker))
           throw this.invalidResponse("wallet fee tracker");
@@ -5115,61 +5267,58 @@ var YieldseekerAgent = class {
             address: metadata.address,
             abi: erc20Abi3,
             functionName: "balanceOf",
-            args: [agentWallet]
+            args: [agentWallet],
+            blockNumber
           }), chainId, "eth_call", this.id),
           withPaidRpcDiagnostics(() => this.getReceiptClient().readContract({
             address: getAddress4(feeTracker),
             abi: YIELDSEEKER_FEE_TRACKER_ABI,
             functionName: "getFeesOwed",
-            args: [agentWallet]
+            args: [agentWallet],
+            blockNumber
           }), chainId, "eth_call", this.id)
         ]);
         this.assertSession(generation, chainId, asset);
         if (typeof idle2 !== "bigint" || idle2 < 0n || typeof fees !== "bigint" || fees < 0n)
           throw this.invalidResponse("wallet withdrawal balance");
-        return { idle: idle2, fees, available: idle2 > fees ? idle2 - fees : 0n };
+        return { blockNumber, idle: idle2, fees, available: idle2 > fees ? idle2 - fees : 0n };
       };
-      const [portfolio, initialFunds] = await Promise.all([portfolioRead, readOwnerFunds()]);
+      const [funds, feeRateBps] = await Promise.all([
+        positionsRead.then((positions2) => this.resolveWithdrawalLiquidity(
+          state,
+          chainId,
+          context,
+          positions2,
+          priorPositions,
+          amount === void 0 ? void 0 : BigInt(amount)
+        )),
+        feeRateRead
+      ]);
+      const initialFunds = funds;
       const idle = initialFunds.idle;
-      let positions = portfolio.positions;
+      const maxLiquidity = amount === void 0 ? funds : void 0;
+      let positions = [...funds.positions, ...funds.indexedPositions.filter((position2) => !funds.verifiedVaults.includes(position2.yieldOption.address.toLowerCase()))];
       let positionsStatus = "empty";
-      const validatePositions = (values) => {
-        const seen = /* @__PURE__ */ new Set();
-        for (const value of values) {
-          const address = value?.yieldOption?.address;
-          if (typeof address !== "string" || !isAddress3(address) || seen.has(address.toLowerCase()) || typeof value.assetsRaw !== "string" || !/^\d+$/.test(value.assetsRaw) || typeof value.withdrawableAssetsRaw !== "string" || !/^\d+$/.test(value.withdrawableAssetsRaw))
-            throw this.invalidResponse("yield positions");
-          seen.add(address.toLowerCase());
-        }
-        return values;
-      };
+      const validatePositions = (values) => this.validateWithdrawalPositions(values, chainId, asset);
+      validatePositions(positions);
       const zeroValuePositions = /* @__PURE__ */ new WeakSet();
       const hasPositions = (values) => values.some((position2) => !zeroValuePositions.has(position2) && (BigInt(position2.assetsRaw) > 0n || BigInt(position2.withdrawableAssetsRaw) > 0n));
       const knownPositions = /* @__PURE__ */ new Map();
       const rememberPositions = (values) => {
         for (const position2 of values) knownPositions.set(position2.yieldOption.address.toLowerCase(), position2);
       };
-      const boundedVerification = async (read2, timeoutMs = YIELDSEEKER_MAX_POSITION_VERIFICATION_MS) => {
-        let timer;
-        try {
-          return await Promise.race([read2, new Promise((resolve) => {
-            timer = setTimeout(() => resolve(void 0), Math.max(0, timeoutMs));
-          })]);
-        } finally {
-          if (timer) clearTimeout(timer);
-        }
-      };
       const verifyKnownPositions = async (blockNumber, values = [...knownPositions.values()], timeoutMs = YIELDSEEKER_MAX_POSITION_VERIFICATION_MS) => {
         const verifiedPositions = [];
         if (timeoutMs <= 0) return { verified: false, positions: verifiedPositions };
-        const reads = values.map(async (position2) => {
+        let active = true;
+        const checks = this.readVaultLiquidity(values.map((position2) => position2.yieldOption), agentWallet, blockNumber).catch(() => []);
+        const reads = values.map(async (position2, index) => {
           const address = getAddress4(position2.yieldOption.address);
           try {
-            const [underlying, shares, withdrawable] = await Promise.all([
-              this.getReceiptClient().readContract({ address, abi: erc4626Abi, functionName: "asset", blockNumber }),
-              this.getReceiptClient().readContract({ address, abi: erc20Abi3, functionName: "balanceOf", args: [agentWallet], blockNumber }),
-              this.getReceiptClient().readContract({ address, abi: erc4626Abi, functionName: "maxWithdraw", args: [agentWallet], blockNumber })
-            ]);
+            const group = (await checks).slice(index * 3, index * 3 + 3);
+            if (!active || group.length !== 3 || group.some((value) => value.status !== "success")) return null;
+            this.assertSession(generation, chainId, asset);
+            const [underlying, shares, withdrawable] = group.map((value) => value.result);
             if (typeof underlying !== "string" || !isAddress3(underlying) || underlying.toLowerCase() !== metadata.address.toLowerCase() || typeof shares !== "bigint" || shares < 0n || typeof withdrawable !== "bigint" || withdrawable < 0n) return null;
             const verified = { ...position2, assetsRaw: shares.toString(), withdrawableAssetsRaw: withdrawable.toString() };
             if (shares > 0n && withdrawable === 0n) {
@@ -5204,6 +5353,7 @@ var YieldseekerAgent = class {
           }
         });
         const result = await boundedVerification(Promise.all(reads), timeoutMs);
+        active = false;
         this.assertSession(generation, chainId, asset);
         return {
           verified: result !== void 0 && result.every((position2) => position2 !== null),
@@ -5222,7 +5372,7 @@ var YieldseekerAgent = class {
         return validatePositions(response.yieldPositions);
       };
       if (amount === void 0) {
-        validatePositions(positions);
+        rememberPositions(funds.indexedPositions);
         rememberPositions(positions);
         positionsStatus = hasPositions(positions) ? "remaining" : "empty";
       }
@@ -5238,8 +5388,8 @@ var YieldseekerAgent = class {
         const zeroPayoutCandidates = positions.filter((position2) => BigInt(position2.assetsRaw) > 0n || BigInt(position2.withdrawableAssetsRaw) > 0n);
         if (amount === void 0 && zeroPayoutCandidates.length) {
           const deadline = Date.now() + YIELDSEEKER_MAX_POSITION_VERIFICATION_MS;
-          const blockNumber = await boundedVerification(
-            Promise.resolve().then(() => this.getReceiptClient().getBlockNumber()).catch(() => void 0)
+          const blockNumber = initialFunds.blockNumber ?? await boundedVerification(
+            Promise.resolve().then(() => this.getReceiptClient().getBlockNumber({ cacheTime: 0 })).catch(() => void 0)
           );
           const proof = typeof blockNumber === "bigint" ? await verifyKnownPositions(blockNumber, zeroPayoutCandidates, deadline - Date.now()) : { verified: false, positions: [] };
           positions = proof.positions;
@@ -5363,8 +5513,9 @@ var YieldseekerAgent = class {
           remaining: remaining.toString()
         });
       }
-      const availableForOwner = confirmedMovement ? (await readOwnerFunds()).available : initialFunds.available;
-      const feeRate = portfolio.feeRateBps;
+      const ownerFunds = confirmedMovement ? await readOwnerFunds(latestUnwindBlock) : initialFunds;
+      const availableForOwner = ownerFunds.available;
+      const feeRate = feeRateBps ?? 0;
       const maxFee = confirmedMovement && Number.isInteger(feeRate) && feeRate >= 0 && feeRate <= 1e3 ? (grossUnwound * BigInt(feeRate) + 9999n) / 10000n : 0n;
       const shortfall = availableForOwner < requested ? requested - availableForOwner : 0n;
       if (availableForOwner === 0n || amount !== void 0 && shortfall > maxFee) {
@@ -5439,7 +5590,7 @@ var YieldseekerAgent = class {
       settlement = plannedMovement;
       const evidence = receipt ? inspectFundingReceipt(receipt, { ...base6, txHash }, metadata.address, "withdraw") : { ...base6, txHash, confirmedAmount: payout.toString(), transfer: "confirmed", delivery: "confirmed" };
       notifyFunding(onProgress, evidence);
-      let completion;
+      let completion = maxLiquidity && !maxLiquidity.complete ? "unverified" : void 0;
       if (amount === void 0) {
         const verificationDeadline = Date.now() + YIELDSEEKER_MAX_POSITION_VERIFICATION_MS;
         const knownAtPayout = new Set(knownPositions.keys());
@@ -5486,10 +5637,14 @@ var YieldseekerAgent = class {
           { asset, available: "0", txHash },
           this.id
         );
+      const confirmedAmount = evidence.confirmedAmount && /^\d+$/.test(evidence.confirmedAmount) ? BigInt(evidence.confirmedAmount) : void 0;
+      const feeIncrease = ownerFunds.fees - initialFunds.fees;
+      const feeAdjustment = amount !== void 0 && receipt && initialFunds.blockNumber !== void 0 && latestUnwindBlock !== void 0 && initialFunds.blockNumber <= latestUnwindBlock && receipt.blockNumber >= latestUnwindBlock && evidence.transfer === "confirmed" && evidence.delivery === "confirmed" && confirmedAmount === payout && feeIncrease > 0n && feeIncrease <= maxFee && requested - payout === feeIncrease ? { grossAmount: requested.toString(), feeAmount: feeIncrease.toString() } : void 0;
       return {
         txHash,
         type: amount === void 0 && !completion ? "full" : "partial",
         ...completion ? { completion } : {},
+        ...feeAdjustment ? { feeAdjustment } : {},
         amount: evidence.confirmedAmount ?? payout.toString(),
         evidence
       };
@@ -5505,6 +5660,236 @@ var YieldseekerAgent = class {
       }
     }
   }
+  /** A withdrawal-only liquidity floor; display snapshots never authorize it. */
+  async getWithdrawableBalance(state, chainId, asset, minimumAmount) {
+    this.assertChain(chainId);
+    this.assertAsset(asset);
+    const generation = this.readGeneration;
+    const context = await this.findAgent(state, chainId, asset);
+    this.assertSession(generation, chainId, asset);
+    if (!context) throw new OwneyError(
+      "WITHDRAW_BALANCE_UNAVAILABLE",
+      "The Yieldseeker withdrawal wallet could not be verified.",
+      { asset },
+      this.id
+    );
+    const hints = this.withdrawalVaultHints(state, chainId, context);
+    const positions = await this.readWithdrawalPositions(state, chainId, context);
+    const proof = await this.resolveWithdrawalLiquidity(state, chainId, context, positions, hints, minimumAmount);
+    this.assertSession(generation, chainId, asset);
+    if (proof.available === 0n && !proof.complete) throw new OwneyError(
+      "WITHDRAW_BALANCE_UNAVAILABLE",
+      "Yieldseeker withdrawal liquidity could not be verified yet.",
+      { asset },
+      this.id
+    );
+    return proof.available;
+  }
+  async readVaultLiquidity(options, wallet, blockNumber) {
+    const owner = getAddress4(wallet);
+    const batches = Array.from({ length: Math.ceil(options.length / YIELDSEEKER_VAULTS_PER_RPC_BATCH) }, (_, index) => options.slice(index * YIELDSEEKER_VAULTS_PER_RPC_BATCH, (index + 1) * YIELDSEEKER_VAULTS_PER_RPC_BATCH));
+    const results = await Promise.all(batches.map((batch) => this.getReceiptClient().multicall({
+      blockNumber,
+      allowFailure: true,
+      contracts: batch.flatMap((option) => {
+        const address = getAddress4(option.address);
+        return [
+          { address, abi: erc4626Abi, functionName: "asset" },
+          { address, abi: erc20Abi3, functionName: "balanceOf", args: [owner] },
+          { address, abi: erc4626Abi, functionName: "maxWithdraw", args: [owner] }
+        ];
+      })
+    })));
+    return results.flat();
+  }
+  validateWithdrawalPositions(values, chainId, asset) {
+    const underlying = YIELDSEEKER_ASSET_METADATA[asset].address.toLowerCase();
+    const seen = /* @__PURE__ */ new Set();
+    for (const value of values) {
+      const option = value?.yieldOption;
+      if (!option || typeof option.address !== "string" || !isAddress3(option.address) || seen.has(option.address.toLowerCase()) || option.chainId !== void 0 && option.chainId !== chainId || option.assetAddress !== void 0 && (typeof option.assetAddress !== "string" || option.assetAddress.toLowerCase() !== underlying) || typeof value.assetsRaw !== "string" || !/^\d+$/.test(value.assetsRaw) || typeof value.withdrawableAssetsRaw !== "string" || !/^\d+$/.test(value.withdrawableAssetsRaw))
+        throw this.invalidResponse("yield positions");
+      seen.add(option.address.toLowerCase());
+    }
+    return values;
+  }
+  withdrawalVaultHints(state, chainId, context) {
+    const key2 = this.contextKey(state, chainId, context.asset);
+    const positions = this.readCache.get(`positions:${key2}`)?.value?.yieldPositions ?? [];
+    const hints = new Map(this.validateWithdrawalPositions(positions, chainId, context.asset).map((position2) => [position2.yieldOption.address.toLowerCase(), position2]));
+    const historic = this.readCache.get(`historic:${key2}`)?.value?.position;
+    const receivedAssets = new Set((historic?.movements ?? []).filter((movement) => movement.chainId === chainId && movement.agentWalletId === context.wallet.agentWalletId && movement.toAddress?.toLowerCase() === context.wallet.walletAddress.toLowerCase()).filter((movement) => typeof movement.assetAddress === "string").map((movement) => movement.assetAddress.toLowerCase()));
+    for (const option of this.yieldOptions.get(context.asset)?.value ?? []) {
+      if (option.chainId !== chainId || typeof option.assetAddress !== "string" || option.assetAddress.toLowerCase() !== YIELDSEEKER_ASSET_METADATA[context.asset].address.toLowerCase() || !isAddress3(option.address) || !receivedAssets.has(option.address.toLowerCase()) || hints.has(option.address.toLowerCase())) continue;
+      hints.set(option.address.toLowerCase(), {
+        yieldOption: option,
+        assetsRaw: "0",
+        assetsBase: "0",
+        assetsUsd: 0,
+        withdrawableAssetsRaw: "0",
+        withdrawableAssetsBase: "0"
+      });
+    }
+    return [...hints.values()];
+  }
+  async readWithdrawalPositions(state, chainId, context) {
+    const generation = this.readGeneration;
+    let timer;
+    try {
+      const positions = await Promise.race([
+        this.loadActivityContext(state, chainId, context, {}).then((value) => value.positions).catch((error) => {
+          if (!(error instanceof OwneyError) || !["AGENT_API_ERROR", "AGENT_TIMEOUT"].includes(error.code)) throw error;
+          return void 0;
+        }),
+        new Promise((resolve) => {
+          timer = setTimeout(() => resolve(void 0), YIELDSEEKER_MAX_POSITION_VERIFICATION_MS);
+        })
+      ]);
+      this.assertSession(generation, chainId, context.asset);
+      return positions === void 0 ? void 0 : this.validateWithdrawalPositions(positions, chainId, context.asset);
+    } finally {
+      if (timer) clearTimeout(timer);
+    }
+  }
+  async resolveWithdrawalLiquidity(state, chainId, context, positions, previousPositions = [], requested) {
+    const generation = this.readGeneration;
+    const contextKey = this.contextKey(state, chainId, context.asset);
+    const version = this.portfolioVersion(contextKey);
+    const metadata = YIELDSEEKER_ASSET_METADATA[context.asset];
+    const unavailable = () => new OwneyError(
+      "WITHDRAW_BALANCE_UNAVAILABLE",
+      "Yieldseeker withdrawal liquidity could not be verified yet.",
+      { asset: context.asset, reason: "YIELDSEEKER_LIQUIDITY_UNVERIFIED" },
+      this.id
+    );
+    let active = true;
+    let knownProof;
+    const hasUnverifiedIndexedFunds = (proof) => proof.indexedPositions.some((position2) => !proof.verifiedVaults.includes(position2.yieldOption.address.toLowerCase()) && BigInt(position2.withdrawableAssetsRaw) > 0n);
+    const assertReadCurrent = () => {
+      this.assertSession(generation, chainId, context.asset);
+      if (!active || this.portfolioVersion(contextKey) !== version) throw unavailable();
+    };
+    const read2 = async () => {
+      const client = this.getReceiptClient();
+      const blockNumber = await client.getBlockNumber({ cacheTime: 0 });
+      const wallet = getAddress4(context.wallet.walletAddress);
+      assertReadCurrent();
+      if (typeof blockNumber !== "bigint") throw unavailable();
+      const indexedPositions = this.validateWithdrawalPositions(positions ?? [], chainId, context.asset);
+      const candidates = /* @__PURE__ */ new Map();
+      for (const position2 of this.validateWithdrawalPositions([...previousPositions], chainId, context.asset))
+        candidates.set(position2.yieldOption.address.toLowerCase(), position2.yieldOption);
+      for (const position2 of indexedPositions) candidates.set(position2.yieldOption.address.toLowerCase(), position2.yieldOption);
+      const options = [...candidates.values()];
+      if (options.length > YIELDSEEKER_WITHDRAWAL_CANDIDATE_LIMIT) throw unavailable();
+      const [checks, idle, feeTracker] = await Promise.all([
+        this.readVaultLiquidity(options, wallet, blockNumber),
+        client.readContract({
+          address: metadata.address,
+          abi: erc20Abi3,
+          functionName: "balanceOf",
+          args: [wallet],
+          blockNumber
+        }),
+        client.readContract({
+          address: wallet,
+          abi: YIELDSEEKER_AGENT_WALLET_ABI,
+          functionName: "feeTracker",
+          blockNumber
+        })
+      ]);
+      if (typeof idle !== "bigint" || idle < 0n || !isAddress3(feeTracker) || /^0x0{40}$/i.test(feeTracker)) throw unavailable();
+      assertReadCurrent();
+      const fees = await client.readContract({
+        address: getAddress4(feeTracker),
+        abi: YIELDSEEKER_FEE_TRACKER_ABI,
+        functionName: "getFeesOwed",
+        args: [wallet],
+        blockNumber
+      });
+      if (typeof fees !== "bigint" || fees < 0n) throw unavailable();
+      const verified = [];
+      const verifiedVaults = [];
+      let deployed = 0n;
+      const collect = (options2, results) => {
+        let valid2 = true;
+        options2.forEach((option, index) => {
+          const group = results.slice(index * 3, index * 3 + 3);
+          if (group.length !== 3 || group.some((value) => value.status !== "success")) {
+            valid2 = false;
+            return;
+          }
+          const [underlying, shares, withdrawable] = group.map((value) => value.result);
+          if (typeof underlying !== "string" || !isAddress3(underlying) || underlying.toLowerCase() !== metadata.address.toLowerCase() || typeof shares !== "bigint" || shares < 0n || typeof withdrawable !== "bigint" || withdrawable < 0n || shares === 0n && withdrawable > 0n) {
+            valid2 = false;
+            return;
+          }
+          verifiedVaults.push(option.address.toLowerCase());
+          deployed += withdrawable;
+          verified.push({
+            yieldOption: option,
+            assetsRaw: shares.toString(),
+            assetsBase: "0",
+            assetsUsd: 0,
+            withdrawableAssetsRaw: withdrawable.toString(),
+            withdrawableAssetsBase: withdrawable.toString()
+          });
+        });
+        return valid2;
+      };
+      const allKnownVerified = collect(options, checks);
+      const proof = (complete) => ({
+        blockNumber,
+        idle,
+        fees,
+        available: idle + deployed > fees ? idle + deployed - fees : 0n,
+        positions: [...verified],
+        verifiedVaults: [...verifiedVaults],
+        indexedPositions,
+        complete
+      });
+      knownProof = proof(false);
+      const indexed = new Map(indexedPositions.map((position2) => [position2.yieldOption.address.toLowerCase(), position2]));
+      const discrepancy = verified.some((position2) => {
+        const row = indexed.get(position2.yieldOption.address.toLowerCase());
+        return row ? BigInt(position2.withdrawableAssetsRaw) < BigInt(row.withdrawableAssetsRaw) : BigInt(position2.assetsRaw) > 0n;
+      });
+      const typedCoveredByIdle = requested !== void 0 && idle >= requested + fees;
+      const expand = positions === void 0 || indexedPositions.length === 0 && !typedCoveredByIdle || !allKnownVerified || discrepancy || knownProof.available === 0n || requested !== void 0 && requested > knownProof.available;
+      if (!expand) {
+        assertReadCurrent();
+        return proof(true);
+      }
+      const catalog = await this.loadYieldOptions(context.asset).catch(() => void 0);
+      assertReadCurrent();
+      const eligible = /* @__PURE__ */ new Map();
+      for (const option of catalog ?? []) {
+        if (option?.chainId === chainId && typeof option.assetAddress === "string" && option.assetAddress.toLowerCase() === metadata.address.toLowerCase() && isAddress3(option.address))
+          eligible.set(option.address.toLowerCase(), option);
+      }
+      const extra = [...eligible].filter(([address]) => !candidates.has(address)).map(([, option]) => option);
+      if (options.length + extra.length > YIELDSEEKER_WITHDRAWAL_CANDIDATE_LIMIT) throw unavailable();
+      const allExtraVerified = collect(extra, await this.readVaultLiquidity(extra, wallet, blockNumber));
+      assertReadCurrent();
+      const result = proof(eligible.size > 0 && allKnownVerified && allExtraVerified);
+      if (result.available === 0n && !result.complete && !hasUnverifiedIndexedFunds(result)) throw unavailable();
+      return result;
+    };
+    let timer;
+    try {
+      return await Promise.race([read2(), new Promise((_, reject) => {
+        timer = setTimeout(() => reject(unavailable()), YIELDSEEKER_MAX_POSITION_VERIFICATION_MS);
+      })]);
+    } catch (error) {
+      assertReadCurrent();
+      if (knownProof && (knownProof.available > 0n || hasUnverifiedIndexedFunds(knownProof)) && (!(error instanceof OwneyError) || error.code === "WITHDRAW_BALANCE_UNAVAILABLE")) return knownProof;
+      if (error instanceof OwneyError) throw error;
+      throw unavailable();
+    } finally {
+      active = false;
+      if (timer) clearTimeout(timer);
+    }
+  }
   async getBalances(state, chainId, asset) {
     this.assertChain(chainId);
     if (asset !== void 0) this.assertAsset(asset);
@@ -6702,6 +7087,7 @@ function aggregateApyByChainAndAsset(agentApys, agentBalances) {
 
 // src/client.ts
 import {
+  parseUnits as parseUnits2,
   createPublicClient as createPublicClient4,
   createWalletClient as createWalletClient4,
   custom as custom4
@@ -8674,11 +9060,10 @@ var OwneySDK = class {
       };
     }
     const requested = BigInt(amount);
-    const aggregated = await this.getBalances(void 0, { chainId, asset: token });
+    const { balances: aggregated, failures: balanceFailures } = await this.readAggregatedBalances({ chainId, asset: token });
     const unavailableAgents = eligibleAgents.filter(
       (agent) => !(agent.id in aggregated.agentBalances)
     );
-    const unavailableAgentIds = unavailableAgents.map((agent) => agent.id);
     const balances = projectAgentBalancesForAsset(
       eligibleAgents,
       aggregated.agentBalances,
@@ -8686,6 +9071,44 @@ var OwneySDK = class {
       asset,
       assetInfo.decimals
     );
+    const indexedAvailable = balances.reduce((sum, entry) => sum + entry.balance, 0n);
+    await Promise.all(balances.map(async (entry) => {
+      if (entry.balance !== 0n && indexedAvailable >= requested || !entry.agent.getWithdrawableBalance) return;
+      const originalFailure = balanceFailures.get(entry.agent.id);
+      if (originalFailure instanceof OwneyError && ![
+        "AGENT_API_ERROR",
+        "AGENT_TIMEOUT",
+        "AGENT_RATE_LIMITED"
+      ].includes(originalFailure.code)) return;
+      if (this.isUserRejectedWithdrawal(originalFailure)) throw originalFailure;
+      if (entry.balance !== 0n && entry.agent.id !== "yieldseeker") {
+        const total = aggregated.agentBalances[entry.agent.id]?.assetBalances?.find((value) => value.chainId === chainId && value.asset.toUpperCase() === token);
+        if (!total || parseUnits2(total.amount, assetInfo.decimals) <= entry.balance) return;
+      }
+      try {
+        const otherAvailable = indexedAvailable - entry.balance;
+        const minimumAmount = requested > otherAvailable ? requested - otherAvailable : void 0;
+        const verified = await entry.agent.getWithdrawableBalance(state, chainId, token, minimumAmount);
+        if (typeof verified !== "bigint" || verified < 0n)
+          throw new OwneyError("WITHDRAW_BALANCE_UNAVAILABLE", "Invalid withdrawal balance proof.", {}, entry.agent.id);
+        if (this.state !== state || this.state.walletAddress !== state.walletAddress || this.state.chainId !== chainId)
+          throw new OwneyError("NOT_CONNECTED", "Account changed while planning withdrawal.");
+        entry.balance = verified;
+        const unavailableIndex = unavailableAgents.indexOf(entry.agent);
+        if (unavailableIndex !== -1) unavailableAgents.splice(unavailableIndex, 1);
+        if (aggregated.agentErrors) delete aggregated.agentErrors[entry.agent.id];
+      } catch (error) {
+        entry.balance = 0n;
+        if (!unavailableAgents.includes(entry.agent)) unavailableAgents.push(entry.agent);
+        aggregated.agentErrors = {
+          ...aggregated.agentErrors,
+          [entry.agent.id]: error instanceof Error ? error.message : String(error)
+        };
+      }
+    }));
+    if (this.state !== state || this.state.chainId !== chainId)
+      throw new OwneyError("NOT_CONNECTED", "Account changed while planning withdrawal.");
+    const unavailableAgentIds = unavailableAgents.map((agent) => agent.id);
     const totalAvailable = balances.reduce((sum, x) => sum + x.balance, 0n);
     if (totalAvailable === 0n && unavailableAgents.length > 0) {
       throw new OwneyError(
@@ -8770,7 +9193,18 @@ var OwneySDK = class {
       );
     }
     const withdrawn = sumWithdrawnAmount(results);
-    if (withdrawn < requested) {
+    const verifiedFeeDeductions = orderedPlans.reduce((total, plan) => {
+      const result = results[plan.agent.id];
+      const adjustment = result?.feeAdjustment;
+      const evidence = result?.evidence;
+      const submitted = submissionEvidence.get(plan.agent.id);
+      if (plan.agent.id !== "yieldseeker" || !adjustment || !evidence || !submitted || typeof result.amount !== "string" || !/^\d+$/.test(result.amount) || typeof adjustment.grossAmount !== "string" || !/^\d+$/.test(adjustment.grossAmount) || typeof adjustment.feeAmount !== "string" || !/^\d+$/.test(adjustment.feeAmount) || evidence.agentId !== plan.agent.id || evidence.chainId !== chainId || evidence.token !== token || typeof evidence.walletAddress !== "string" || evidence.walletAddress.toLowerCase() !== state.walletAddress.toLowerCase() || typeof evidence.agentWallet !== "string" || !/^0x[\da-f]{40}$/i.test(evidence.agentWallet) || typeof result.txHash !== "string" || !/^0x[\da-f]{64}$/i.test(result.txHash) || typeof evidence.txHash !== "string" || typeof submitted.txHash !== "string" || evidence.txHash?.toLowerCase() !== result.txHash.toLowerCase() || submitted.txHash?.toLowerCase() !== result.txHash.toLowerCase() || submitted.agentId !== evidence.agentId || submitted.chainId !== chainId || submitted.token !== token || typeof submitted.walletAddress !== "string" || typeof submitted.agentWallet !== "string" || submitted.walletAddress.toLowerCase() !== state.walletAddress.toLowerCase() || submitted.agentWallet?.toLowerCase() !== evidence.agentWallet.toLowerCase() || evidence.transfer !== "confirmed" || evidence.delivery !== "confirmed" || submitted.transfer !== "confirmed" || submitted.delivery !== "confirmed" || evidence.confirmedAmount !== result.amount || submitted.confirmedAmount !== result.amount || evidence.intendedAmount !== result.amount || submitted.intendedAmount !== result.amount) return total;
+      const fee = BigInt(adjustment.feeAmount);
+      const net = BigInt(result.amount);
+      const gross = BigInt(adjustment.grossAmount);
+      return fee > 0n && net > 0n && gross === plan.planned && net + fee === gross ? total + fee : total;
+    }, 0n);
+    if (withdrawn + verifiedFeeDeductions < requested) {
       throw new OwneyError(
         "WITHDRAW_PARTIAL_FAILURE",
         `Withdrew "${withdrawn.toString()}" of requested "${amount}" of "${asset}". One or more agents failed; remaining balance could not cover the shortfall.`,
@@ -8794,23 +9228,42 @@ var OwneySDK = class {
    * @returns {AgentBalance} for a single agent, or {OwneyBalances} with totalBalance and per-agent breakdown
    */
   async getBalances(agentId, options) {
-    const { state, chainId, assertCurrent } = this.captureReadScope(options?.chainId);
-    const asset = options?.asset;
-    const readBalance = (agent) => this.readAgent(
-      agent,
-      "balances",
-      () => asset ? agent.getBalances(state, chainId, asset) : agent.getBalances(state, chainId),
-      asset ? { asset } : null
-    );
     if (agentId) {
+      const { state, chainId, assertCurrent } = this.captureReadScope(options?.chainId);
+      const asset = options?.asset;
       const agent = this.getAgent(agentId);
-      const result = await readBalance(agent);
+      const result = await this.readAgent(
+        agent,
+        "balances",
+        () => asset ? agent.getBalances(state, chainId, asset) : agent.getBalances(state, chainId),
+        asset ? { asset } : null
+      );
       assertCurrent();
       return {
         ...result,
         balanceComposition: agent.balanceComposition ?? "tokens-include-positions"
       };
     }
+    const { balances, failures } = await this.readAggregatedBalances(options);
+    if (Object.keys(balances.agentBalances).length === 0) {
+      throw new OwneyError(
+        "BALANCE_ALL_FAILED",
+        "Failed to fetch balances for all active agents.",
+        { agentErrors: balances.agentErrors ?? {}, failures: [...failures.values()] }
+      );
+    }
+    return balances;
+  }
+  /** Preserve failure objects for withdrawal recovery without changing public balance results. */
+  async readAggregatedBalances(options) {
+    const { state, chainId, assertCurrent } = this.captureReadScope(options?.chainId);
+    const asset = options?.asset;
+    const readBalance = (agent) => this.readAgent(
+      agent,
+      "balances",
+      () => asset ? agent.getBalances(state, chainId, asset) : agent.getBalances(state, chainId),
+      asset ? { asset } : null
+    );
     let totalBalance = 0;
     const results = {};
     const entries = [...this.getActiveAgents().entries()];
@@ -8827,39 +9280,33 @@ var OwneySDK = class {
       })
     );
     assertCurrent();
-    let successCount = 0;
     const agentErrors = {};
     const agentRetryAt = {};
-    const agentFailures = [];
+    const agentFailures = /* @__PURE__ */ new Map();
     for (let i = 0; i < balanceResults.length; i++) {
       const settledResult = balanceResults[i];
-      const [agentId2] = entries[i];
+      const [agentId] = entries[i];
       if (settledResult.status === "fulfilled") {
         const [id, b] = settledResult.value;
         results[id] = b;
         totalBalance += Number(b.totalBalance);
-        successCount += 1;
         continue;
       }
       const reason = settledResult.reason;
-      agentFailures.push(reason);
+      agentFailures.set(agentId, reason);
       const retryDelay = rateLimitDelay(reason);
-      if (retryDelay !== void 0) agentRetryAt[agentId2] = Date.now() + retryDelay;
-      agentErrors[agentId2] = reason instanceof Error ? reason.message : String(reason);
-      console.error(`[owney-sdk] Balance fetch failed for agent "${agentId2}":`, reason);
-    }
-    if (successCount === 0) {
-      throw new OwneyError(
-        "BALANCE_ALL_FAILED",
-        "Failed to fetch balances for all active agents.",
-        { agentErrors, failures: agentFailures }
-      );
+      if (retryDelay !== void 0) agentRetryAt[agentId] = Date.now() + retryDelay;
+      agentErrors[agentId] = reason instanceof Error ? reason.message : String(reason);
+      console.error(`[owney-sdk] Balance fetch failed for agent "${agentId}":`, reason);
     }
     return {
-      totalBalance: String(totalBalance),
-      totalBalanceAsset: "usdc",
-      agentBalances: results,
-      ...Object.keys(agentErrors).length > 0 ? { agentErrors, agentRetryAt } : {}
+      balances: {
+        totalBalance: String(totalBalance),
+        totalBalanceAsset: "usdc",
+        agentBalances: results,
+        ...Object.keys(agentErrors).length > 0 ? { agentErrors, agentRetryAt } : {}
+      },
+      failures: agentFailures
     };
   }
   /**