@t2000/engine 1.24.9 → 1.24.10

package/dist/index.js

@@ -5786,6 +5786,120 @@ Only offer to execute actions you have tools for. If you retrieved a quote, data
 - Cap: at most ONE proactive block per turn.
 - Skip proactive blocks when nothing notable changed since the last turn, when the user is mid-flow on something else, or when you'd just be restating the financial-context block. Quality over quantity \u2014 a block ignored is worse than no block.`;
 
+// src/post-write-poll.ts
+async function pollForIndexerCatchup(options) {
+  const { suiRpcUrl, address, ceilingMs, pollIntervalMs, signal } = options;
+  const start = Date.now();
+  if (!suiRpcUrl) {
+    await sleepWithFallback(ceilingMs, signal);
+    return {
+      outcome: "fallback_no_rpc",
+      attempts: 0,
+      resolvedAtMs: Date.now() - start
+    };
+  }
+  if (!address) {
+    await sleepWithFallback(ceilingMs, signal);
+    return {
+      outcome: "fallback_no_address",
+      attempts: 0,
+      resolvedAtMs: Date.now() - start
+    };
+  }
+  let baseline;
+  try {
+    const coins = await fetchWalletCoins(address, suiRpcUrl);
+    baseline = new Map(
+      coins.map((c) => [c.coinType, BigInt(c.totalBalance)])
+    );
+  } catch (err) {
+    console.warn(
+      "[post-write-poll] baseline fetch failed; falling back to fixed sleep:",
+      err
+    );
+    await sleepWithFallback(ceilingMs, signal);
+    return {
+      outcome: "fallback_no_baseline",
+      attempts: 0,
+      resolvedAtMs: Date.now() - start
+    };
+  }
+  const maxAttempts = Math.max(1, Math.floor(ceilingMs / pollIntervalMs));
+  for (let attempt = 1; attempt <= maxAttempts; attempt++) {
+    if (signal.aborted) {
+      return {
+        outcome: "aborted",
+        attempts: attempt - 1,
+        resolvedAtMs: Date.now() - start
+      };
+    }
+    await new Promise((resolve) => {
+      const t = setTimeout(resolve, pollIntervalMs);
+      signal.addEventListener(
+        "abort",
+        () => {
+          clearTimeout(t);
+          resolve();
+        },
+        { once: true }
+      );
+    });
+    if (signal.aborted) {
+      return {
+        outcome: "aborted",
+        attempts: attempt,
+        resolvedAtMs: Date.now() - start
+      };
+    }
+    let current;
+    try {
+      const coins = await fetchWalletCoins(address, suiRpcUrl);
+      current = new Map(
+        coins.map((c) => [c.coinType, BigInt(c.totalBalance)])
+      );
+    } catch (err) {
+      console.warn(
+        `[post-write-poll] poll attempt ${attempt} failed; continuing:`,
+        err
+      );
+      continue;
+    }
+    if (balancesDiffer(baseline, current)) {
+      return {
+        outcome: "detected_change",
+        attempts: attempt,
+        resolvedAtMs: Date.now() - start
+      };
+    }
+  }
+  return {
+    outcome: "ceiling",
+    attempts: maxAttempts,
+    resolvedAtMs: Date.now() - start
+  };
+}
+function balancesDiffer(a, b) {
+  if (a.size !== b.size) return true;
+  for (const [coinType, balance] of a) {
+    if (b.get(coinType) !== balance) return true;
+  }
+  return false;
+}
+async function sleepWithFallback(ms, signal) {
+  if (signal.aborted) return;
+  await new Promise((resolve) => {
+    const t = setTimeout(resolve, ms);
+    signal.addEventListener(
+      "abort",
+      () => {
+        clearTimeout(t);
+        resolve();
+      },
+      { once: true }
+    );
+  });
+}
+
 // src/proactive-marker.ts
 var VALID_TYPES = /* @__PURE__ */ new Set([
   "idle_balance",
@@ -7867,19 +7981,21 @@ var QueryEngine = class {
       { has_wallet: this.walletAddress ? "1" : "0" }
     );
     const sleepStart = Date.now();
-    if (!signal.aborted) {
-      await new Promise((resolve) => {
-        const t = setTimeout(resolve, 1500);
-        signal.addEventListener("abort", () => {
-          clearTimeout(t);
-          resolve();
-        }, { once: true });
-      });
-    }
+    const pollResult = await pollForIndexerCatchup({
+      suiRpcUrl: this.suiRpcUrl,
+      address: this.walletAddress,
+      ceilingMs: 1500,
+      pollIntervalMs: 250,
+      signal
+    });
     getTelemetrySink().histogram(
       "engine.pwr.sleep_ms",
       Date.now() - sleepStart,
-      { aborted: signal.aborted ? "1" : "0" }
+      {
+        aborted: signal.aborted ? "1" : "0",
+        outcome: pollResult.outcome,
+        attempts: String(pollResult.attempts)
+      }
     );
     if (signal.aborted) return;
     const refreshStart = Date.now();