@midscene/web 0.2.2 → 0.2.3-beta-20240815082813.0

package/dist/lib/index.js

@@ -22,6 +22,9 @@ var __spreadValues = (a, b) => {
   return a;
 };
 var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
+var __commonJS = (cb, mod) => function __require() {
+  return mod || (0, cb[__getOwnPropNames(cb)[0]])((mod = { exports: {} }).exports, mod), mod.exports;
+};
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -44,6 +47,301 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
 ));
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
+// ../../node_modules/.pnpm/dayjs@1.11.11/node_modules/dayjs/dayjs.min.js
+var require_dayjs_min = __commonJS({
+  "../../node_modules/.pnpm/dayjs@1.11.11/node_modules/dayjs/dayjs.min.js"(exports, module2) {
+    "use strict";
+    !function(t, e) {
+      "object" == typeof exports && "undefined" != typeof module2 ? module2.exports = e() : "function" == typeof define && define.amd ? define(e) : (t = "undefined" != typeof globalThis ? globalThis : t || self).dayjs = e();
+    }(exports, function() {
+      "use strict";
+      var t = 1e3, e = 6e4, n = 36e5, r = "millisecond", i = "second", s = "minute", u = "hour", a = "day", o = "week", c = "month", f = "quarter", h = "year", d = "date", l = "Invalid Date", $ = /^(\d{4})[-/]?(\d{1,2})?[-/]?(\d{0,2})[Tt\s]*(\d{1,2})?:?(\d{1,2})?:?(\d{1,2})?[.:]?(\d+)?$/, y = /\[([^\]]+)]|Y{1,4}|M{1,4}|D{1,2}|d{1,4}|H{1,2}|h{1,2}|a|A|m{1,2}|s{1,2}|Z{1,2}|SSS/g, M = { name: "en", weekdays: "Sunday_Monday_Tuesday_Wednesday_Thursday_Friday_Saturday".split("_"), months: "January_February_March_April_May_June_July_August_September_October_November_December".split("_"), ordinal: function(t2) {
+        var e2 = ["th", "st", "nd", "rd"], n2 = t2 % 100;
+        return "[" + t2 + (e2[(n2 - 20) % 10] || e2[n2] || e2[0]) + "]";
+      } }, m = function(t2, e2, n2) {
+        var r2 = String(t2);
+        return !r2 || r2.length >= e2 ? t2 : "" + Array(e2 + 1 - r2.length).join(n2) + t2;
+      }, v = { s: m, z: function(t2) {
+        var e2 = -t2.utcOffset(), n2 = Math.abs(e2), r2 = Math.floor(n2 / 60), i2 = n2 % 60;
+        return (e2 <= 0 ? "+" : "-") + m(r2, 2, "0") + ":" + m(i2, 2, "0");
+      }, m: function t2(e2, n2) {
+        if (e2.date() < n2.date())
+          return -t2(n2, e2);
+        var r2 = 12 * (n2.year() - e2.year()) + (n2.month() - e2.month()), i2 = e2.clone().add(r2, c), s2 = n2 - i2 < 0, u2 = e2.clone().add(r2 + (s2 ? -1 : 1), c);
+        return +(-(r2 + (n2 - i2) / (s2 ? i2 - u2 : u2 - i2)) || 0);
+      }, a: function(t2) {
+        return t2 < 0 ? Math.ceil(t2) || 0 : Math.floor(t2);
+      }, p: function(t2) {
+        return { M: c, y: h, w: o, d: a, D: d, h: u, m: s, s: i, ms: r, Q: f }[t2] || String(t2 || "").toLowerCase().replace(/s$/, "");
+      }, u: function(t2) {
+        return void 0 === t2;
+      } }, g = "en", D = {};
+      D[g] = M;
+      var p = "$isDayjsObject", S = function(t2) {
+        return t2 instanceof _ || !(!t2 || !t2[p]);
+      }, w = function t2(e2, n2, r2) {
+        var i2;
+        if (!e2)
+          return g;
+        if ("string" == typeof e2) {
+          var s2 = e2.toLowerCase();
+          D[s2] && (i2 = s2), n2 && (D[s2] = n2, i2 = s2);
+          var u2 = e2.split("-");
+          if (!i2 && u2.length > 1)
+            return t2(u2[0]);
+        } else {
+          var a2 = e2.name;
+          D[a2] = e2, i2 = a2;
+        }
+        return !r2 && i2 && (g = i2), i2 || !r2 && g;
+      }, O = function(t2, e2) {
+        if (S(t2))
+          return t2.clone();
+        var n2 = "object" == typeof e2 ? e2 : {};
+        return n2.date = t2, n2.args = arguments, new _(n2);
+      }, b = v;
+      b.l = w, b.i = S, b.w = function(t2, e2) {
+        return O(t2, { locale: e2.$L, utc: e2.$u, x: e2.$x, $offset: e2.$offset });
+      };
+      var _ = function() {
+        function M2(t2) {
+          this.$L = w(t2.locale, null, true), this.parse(t2), this.$x = this.$x || t2.x || {}, this[p] = true;
+        }
+        var m2 = M2.prototype;
+        return m2.parse = function(t2) {
+          this.$d = function(t3) {
+            var e2 = t3.date, n2 = t3.utc;
+            if (null === e2)
+              return /* @__PURE__ */ new Date(NaN);
+            if (b.u(e2))
+              return /* @__PURE__ */ new Date();
+            if (e2 instanceof Date)
+              return new Date(e2);
+            if ("string" == typeof e2 && !/Z$/i.test(e2)) {
+              var r2 = e2.match($);
+              if (r2) {
+                var i2 = r2[2] - 1 || 0, s2 = (r2[7] || "0").substring(0, 3);
+                return n2 ? new Date(Date.UTC(r2[1], i2, r2[3] || 1, r2[4] || 0, r2[5] || 0, r2[6] || 0, s2)) : new Date(r2[1], i2, r2[3] || 1, r2[4] || 0, r2[5] || 0, r2[6] || 0, s2);
+              }
+            }
+            return new Date(e2);
+          }(t2), this.init();
+        }, m2.init = function() {
+          var t2 = this.$d;
+          this.$y = t2.getFullYear(), this.$M = t2.getMonth(), this.$D = t2.getDate(), this.$W = t2.getDay(), this.$H = t2.getHours(), this.$m = t2.getMinutes(), this.$s = t2.getSeconds(), this.$ms = t2.getMilliseconds();
+        }, m2.$utils = function() {
+          return b;
+        }, m2.isValid = function() {
+          return !(this.$d.toString() === l);
+        }, m2.isSame = function(t2, e2) {
+          var n2 = O(t2);
+          return this.startOf(e2) <= n2 && n2 <= this.endOf(e2);
+        }, m2.isAfter = function(t2, e2) {
+          return O(t2) < this.startOf(e2);
+        }, m2.isBefore = function(t2, e2) {
+          return this.endOf(e2) < O(t2);
+        }, m2.$g = function(t2, e2, n2) {
+          return b.u(t2) ? this[e2] : this.set(n2, t2);
+        }, m2.unix = function() {
+          return Math.floor(this.valueOf() / 1e3);
+        }, m2.valueOf = function() {
+          return this.$d.getTime();
+        }, m2.startOf = function(t2, e2) {
+          var n2 = this, r2 = !!b.u(e2) || e2, f2 = b.p(t2), l2 = function(t3, e3) {
+            var i2 = b.w(n2.$u ? Date.UTC(n2.$y, e3, t3) : new Date(n2.$y, e3, t3), n2);
+            return r2 ? i2 : i2.endOf(a);
+          }, $2 = function(t3, e3) {
+            return b.w(n2.toDate()[t3].apply(n2.toDate("s"), (r2 ? [0, 0, 0, 0] : [23, 59, 59, 999]).slice(e3)), n2);
+          }, y2 = this.$W, M3 = this.$M, m3 = this.$D, v2 = "set" + (this.$u ? "UTC" : "");
+          switch (f2) {
+            case h:
+              return r2 ? l2(1, 0) : l2(31, 11);
+            case c:
+              return r2 ? l2(1, M3) : l2(0, M3 + 1);
+            case o:
+              var g2 = this.$locale().weekStart || 0, D2 = (y2 < g2 ? y2 + 7 : y2) - g2;
+              return l2(r2 ? m3 - D2 : m3 + (6 - D2), M3);
+            case a:
+            case d:
+              return $2(v2 + "Hours", 0);
+            case u:
+              return $2(v2 + "Minutes", 1);
+            case s:
+              return $2(v2 + "Seconds", 2);
+            case i:
+              return $2(v2 + "Milliseconds", 3);
+            default:
+              return this.clone();
+          }
+        }, m2.endOf = function(t2) {
+          return this.startOf(t2, false);
+        }, m2.$set = function(t2, e2) {
+          var n2, o2 = b.p(t2), f2 = "set" + (this.$u ? "UTC" : ""), l2 = (n2 = {}, n2[a] = f2 + "Date", n2[d] = f2 + "Date", n2[c] = f2 + "Month", n2[h] = f2 + "FullYear", n2[u] = f2 + "Hours", n2[s] = f2 + "Minutes", n2[i] = f2 + "Seconds", n2[r] = f2 + "Milliseconds", n2)[o2], $2 = o2 === a ? this.$D + (e2 - this.$W) : e2;
+          if (o2 === c || o2 === h) {
+            var y2 = this.clone().set(d, 1);
+            y2.$d[l2]($2), y2.init(), this.$d = y2.set(d, Math.min(this.$D, y2.daysInMonth())).$d;
+          } else
+            l2 && this.$d[l2]($2);
+          return this.init(), this;
+        }, m2.set = function(t2, e2) {
+          return this.clone().$set(t2, e2);
+        }, m2.get = function(t2) {
+          return this[b.p(t2)]();
+        }, m2.add = function(r2, f2) {
+          var d2, l2 = this;
+          r2 = Number(r2);
+          var $2 = b.p(f2), y2 = function(t2) {
+            var e2 = O(l2);
+            return b.w(e2.date(e2.date() + Math.round(t2 * r2)), l2);
+          };
+          if ($2 === c)
+            return this.set(c, this.$M + r2);
+          if ($2 === h)
+            return this.set(h, this.$y + r2);
+          if ($2 === a)
+            return y2(1);
+          if ($2 === o)
+            return y2(7);
+          var M3 = (d2 = {}, d2[s] = e, d2[u] = n, d2[i] = t, d2)[$2] || 1, m3 = this.$d.getTime() + r2 * M3;
+          return b.w(m3, this);
+        }, m2.subtract = function(t2, e2) {
+          return this.add(-1 * t2, e2);
+        }, m2.format = function(t2) {
+          var e2 = this, n2 = this.$locale();
+          if (!this.isValid())
+            return n2.invalidDate || l;
+          var r2 = t2 || "YYYY-MM-DDTHH:mm:ssZ", i2 = b.z(this), s2 = this.$H, u2 = this.$m, a2 = this.$M, o2 = n2.weekdays, c2 = n2.months, f2 = n2.meridiem, h2 = function(t3, n3, i3, s3) {
+            return t3 && (t3[n3] || t3(e2, r2)) || i3[n3].slice(0, s3);
+          }, d2 = function(t3) {
+            return b.s(s2 % 12 || 12, t3, "0");
+          }, $2 = f2 || function(t3, e3, n3) {
+            var r3 = t3 < 12 ? "AM" : "PM";
+            return n3 ? r3.toLowerCase() : r3;
+          };
+          return r2.replace(y, function(t3, r3) {
+            return r3 || function(t4) {
+              switch (t4) {
+                case "YY":
+                  return String(e2.$y).slice(-2);
+                case "YYYY":
+                  return b.s(e2.$y, 4, "0");
+                case "M":
+                  return a2 + 1;
+                case "MM":
+                  return b.s(a2 + 1, 2, "0");
+                case "MMM":
+                  return h2(n2.monthsShort, a2, c2, 3);
+                case "MMMM":
+                  return h2(c2, a2);
+                case "D":
+                  return e2.$D;
+                case "DD":
+                  return b.s(e2.$D, 2, "0");
+                case "d":
+                  return String(e2.$W);
+                case "dd":
+                  return h2(n2.weekdaysMin, e2.$W, o2, 2);
+                case "ddd":
+                  return h2(n2.weekdaysShort, e2.$W, o2, 3);
+                case "dddd":
+                  return o2[e2.$W];
+                case "H":
+                  return String(s2);
+                case "HH":
+                  return b.s(s2, 2, "0");
+                case "h":
+                  return d2(1);
+                case "hh":
+                  return d2(2);
+                case "a":
+                  return $2(s2, u2, true);
+                case "A":
+                  return $2(s2, u2, false);
+                case "m":
+                  return String(u2);
+                case "mm":
+                  return b.s(u2, 2, "0");
+                case "s":
+                  return String(e2.$s);
+                case "ss":
+                  return b.s(e2.$s, 2, "0");
+                case "SSS":
+                  return b.s(e2.$ms, 3, "0");
+                case "Z":
+                  return i2;
+              }
+              return null;
+            }(t3) || i2.replace(":", "");
+          });
+        }, m2.utcOffset = function() {
+          return 15 * -Math.round(this.$d.getTimezoneOffset() / 15);
+        }, m2.diff = function(r2, d2, l2) {
+          var $2, y2 = this, M3 = b.p(d2), m3 = O(r2), v2 = (m3.utcOffset() - this.utcOffset()) * e, g2 = this - m3, D2 = function() {
+            return b.m(y2, m3);
+          };
+          switch (M3) {
+            case h:
+              $2 = D2() / 12;
+              break;
+            case c:
+              $2 = D2();
+              break;
+            case f:
+              $2 = D2() / 3;
+              break;
+            case o:
+              $2 = (g2 - v2) / 6048e5;
+              break;
+            case a:
+              $2 = (g2 - v2) / 864e5;
+              break;
+            case u:
+              $2 = g2 / n;
+              break;
+            case s:
+              $2 = g2 / e;
+              break;
+            case i:
+              $2 = g2 / t;
+              break;
+            default:
+              $2 = g2;
+          }
+          return l2 ? $2 : b.a($2);
+        }, m2.daysInMonth = function() {
+          return this.endOf(c).$D;
+        }, m2.$locale = function() {
+          return D[this.$L];
+        }, m2.locale = function(t2, e2) {
+          if (!t2)
+            return this.$L;
+          var n2 = this.clone(), r2 = w(t2, e2, true);
+          return r2 && (n2.$L = r2), n2;
+        }, m2.clone = function() {
+          return b.w(this.$d, this);
+        }, m2.toDate = function() {
+          return new Date(this.valueOf());
+        }, m2.toJSON = function() {
+          return this.isValid() ? this.toISOString() : null;
+        }, m2.toISOString = function() {
+          return this.$d.toISOString();
+        }, m2.toString = function() {
+          return this.$d.toUTCString();
+        }, M2;
+      }(), k = _.prototype;
+      return O.prototype = k, [["$ms", r], ["$s", i], ["$m", s], ["$H", u], ["$W", a], ["$M", c], ["$y", h], ["$D", d]].forEach(function(t2) {
+        k[t2[1]] = function(e2) {
+          return this.$g(e2, t2[0], t2[1]);
+        };
+      }), O.extend = function(t2, e2) {
+        return t2.$i || (t2(e2, _, O), t2.$i = true), O;
+      }, O.locale = w, O.isDayjs = S, O.unix = function(t2) {
+        return O(1e3 * t2);
+      }, O.en = D[g], O.Ls = D, O.p = {}, O;
+    });
+  }
+});
+
 // src/index.ts
 var src_exports = {};
 __export(src_exports, {
@@ -125,6 +423,7 @@ var import_node_fs = __toESM(require("fs"));
 var import_node_path = __toESM(require("path"));
 var import_image = require("@midscene/core/image");
 var import_utils = require("@midscene/core/utils");
+var import_dayjs = __toESM(require_dayjs_min());
 
 // src/web-element.ts
 var WebElementInfo = class {
@@ -213,6 +512,13 @@ function findNearestPackageJson(dir) {
   }
   return findNearestPackageJson(parentDir);
 }
+function reportFileName(tag = "web") {
+  const dateTimeInFileName = (0, import_dayjs.default)().format("YYYY-MM-DD_HH-mm-ss-SSS");
+  return `${tag}-${dateTimeInFileName}`;
+}
+function printReportMsg(filepath) {
+  console.log("Midscene - report file updated:", filepath);
+}
 
 // src/common/tasks.ts
 var PageTaskExecutor = class {
@@ -574,31 +880,49 @@ var PageTaskExecutor = class {
 var PageAgent = class {
   constructor(page, opts) {
     this.page = page;
-    this.dumps = [
+    this.opts = Object.assign(
       {
-        groupName: (opts == null ? void 0 : opts.taskFile) || "unnamed",
-        executions: []
-      }
-    ];
-    this.testId = (opts == null ? void 0 : opts.testId) || String(process.pid);
+        generateReport: true,
+        groupName: "Midscene Report",
+        groupDescription: ""
+      },
+      opts || {}
+    );
+    this.dump = {
+      groupName: this.opts.groupName,
+      groupDescription: this.opts.groupDescription,
+      executions: []
+    };
     this.taskExecutor = new PageTaskExecutor(this.page, {
       cache: (opts == null ? void 0 : opts.cache) || { aiTasks: [] }
     });
+    this.reportFileName = reportFileName((opts == null ? void 0 : opts.testId) || "web");
   }
-  appendDump(execution) {
-    const currentDump = this.dumps[0];
+  appendExecutionDump(execution) {
+    const currentDump = this.dump;
     currentDump.executions.push(execution);
   }
+  dumpDataString() {
+    this.dump.groupName = this.opts.groupName;
+    this.dump.groupDescription = this.opts.groupDescription;
+    return (0, import_utils4.stringifyDumpData)(this.dump);
+  }
   writeOutActionDumps() {
-    this.dumpFile = (0, import_utils4.writeDumpFile)({
-      fileName: `run-${this.testId}`,
+    const generateReport = this.opts.generateReport;
+    this.reportFile = (0, import_utils4.writeLogFile)({
+      fileName: this.reportFileName,
       fileExt: import_utils4.groupedActionDumpFileExt,
-      fileContent: (0, import_utils4.stringifyDumpData)(this.dumps)
+      fileContent: this.dumpDataString(),
+      type: "dump",
+      generateReport
     });
+    if (generateReport) {
+      printReportMsg(this.reportFile);
+    }
   }
   async aiAction(taskPrompt) {
     const { executor } = await this.taskExecutor.action(taskPrompt);
-    this.appendDump(executor.dump());
+    this.appendExecutionDump(executor.dump());
     this.writeOutActionDumps();
     if (executor.isInErrorState()) {
       const errorTask = executor.latestErrorTask();
@@ -608,7 +932,7 @@ ${errorTask == null ? void 0 : errorTask.errorStack}`);
   }
   async aiQuery(demand) {
     const { output, executor } = await this.taskExecutor.query(demand);
-    this.appendDump(executor.dump());
+    this.appendExecutionDump(executor.dump());
     this.writeOutActionDumps();
     if (executor.isInErrorState()) {
       const errorTask = executor.latestErrorTask();
@@ -619,7 +943,7 @@ ${errorTask == null ? void 0 : errorTask.errorStack}`);
   }
   async aiAssert(assertion, msg) {
     const { output, executor } = await this.taskExecutor.assert(assertion);
-    this.appendDump(executor.dump());
+    this.appendExecutionDump(executor.dump());
     this.writeOutActionDumps();
     if (!(output == null ? void 0 : output.pass)) {
       const errMsg = msg || `Assertion failed: ${assertion}`;
@@ -647,13 +971,13 @@ ${reasonMsg}`);
 // src/playwright/cache.ts
 var import_node_fs2 = __toESM(require("fs"));
 var import_node_path2 = __toESM(require("path"));
-var import_utils6 = require("@midscene/core/utils");
+var import_utils7 = require("@midscene/core/utils");
 function writeTestCache(taskFile, taskTitle, taskCacheJson) {
   const packageJson = getPkgInfo();
-  (0, import_utils6.writeDumpFile)({
+  (0, import_utils7.writeLogFile)({
     fileName: `${taskFile}(${taskTitle})`,
     fileExt: "json",
-    fileContent: (0, import_utils6.stringifyDumpData)(
+    fileContent: (0, import_utils7.stringifyDumpData)(
       __spreadValues({
         pkgName: packageJson.name,
         pkgVersion: packageJson.version,
@@ -667,7 +991,7 @@ function writeTestCache(taskFile, taskTitle, taskCacheJson) {
 }
 function readTestCache(taskFile, taskTitle) {
   const cacheFile = (0, import_node_path2.join)(
-    (0, import_utils6.getDumpDirPath)("cache"),
+    (0, import_utils7.getLogDirByType)("cache"),
     `${taskFile}(${taskTitle}).json`
   );
   const pkgInfo = getPkgInfo();
@@ -721,32 +1045,47 @@ var groupAndCaseForTest = (testInfo) => {
   return { taskFile, taskTitle };
 };
 var midsceneAgentKeyId = "_midsceneAgentId";
+var midsceneDumpAnnotationId = "MIDSCENE_DUMP_ANNOTATION";
 var PlaywrightAiFixture = () => {
   const pageAgentMap = {};
-  const agentForPage = (page, opts) => {
+  const agentForPage = (page, testInfo) => {
     let idForPage = page[midsceneAgentKeyId];
     if (!idForPage) {
       idForPage = (0, import_node_crypto.randomUUID)();
       page[midsceneAgentKeyId] = idForPage;
-      const testCase = readTestCache(opts.taskFile, opts.taskTitle) || {
+      const { testId } = testInfo;
+      const { taskFile, taskTitle } = groupAndCaseForTest(testInfo);
+      const testCase = readTestCache(taskFile, taskTitle) || {
         aiTasks: []
       };
       pageAgentMap[idForPage] = new PageAgent(page, {
-        testId: `${opts.testId}-${idForPage}`,
-        taskFile: opts.taskFile,
-        cache: testCase
+        testId: `playwright-${testId}-${idForPage}`,
+        groupName: taskTitle,
+        groupDescription: taskFile,
+        cache: testCase,
+        generateReport: false
+        // we will generate it in the reporter
       });
     }
     return pageAgentMap[idForPage];
   };
+  const updateDumpAnnotation = (test, dump) => {
+    const currentAnnotation = test.annotations.find((item) => {
+      return item.type === midsceneDumpAnnotationId;
+    });
+    if (currentAnnotation) {
+      currentAnnotation.description = dump;
+    } else {
+      test.annotations.push({
+        type: midsceneDumpAnnotationId,
+        description: dump
+      });
+    }
+  };
   return {
     ai: async ({ page }, use, testInfo) => {
       const { taskFile, taskTitle } = groupAndCaseForTest(testInfo);
-      const agent = agentForPage(page, {
-        testId: testInfo.testId,
-        taskFile,
-        taskTitle
-      });
+      const agent = agentForPage(page, testInfo);
       await use(
         async (taskPrompt, opts) => {
           await page.waitForLoadState("networkidle");
@@ -757,79 +1096,33 @@ var PlaywrightAiFixture = () => {
       );
       const taskCacheJson = agent.taskExecutor.taskCache.generateTaskCache();
       writeTestCache(taskFile, taskTitle, taskCacheJson);
-      if (agent.dumpFile) {
-        testInfo.annotations.push({
-          type: "MIDSCENE_AI_ACTION",
-          description: JSON.stringify({
-            testId: testInfo.testId,
-            dumpPath: agent.dumpFile
-          })
-        });
-      }
+      updateDumpAnnotation(testInfo, agent.dumpDataString());
     },
     aiAction: async ({ page }, use, testInfo) => {
       const { taskFile, taskTitle } = groupAndCaseForTest(testInfo);
-      const agent = agentForPage(page, {
-        testId: testInfo.testId,
-        taskFile,
-        taskTitle
-      });
+      const agent = agentForPage(page, testInfo);
       await use(async (taskPrompt) => {
         await page.waitForLoadState("networkidle");
         await agent.aiAction(taskPrompt);
       });
-      if (agent.dumpFile) {
-        testInfo.annotations.push({
-          type: "MIDSCENE_AI_ACTION",
-          description: JSON.stringify({
-            testId: testInfo.testId,
-            dumpPath: agent.dumpFile
-          })
-        });
-      }
+      updateDumpAnnotation(testInfo, agent.dumpDataString());
     },
     aiQuery: async ({ page }, use, testInfo) => {
-      const { taskFile, taskTitle } = groupAndCaseForTest(testInfo);
-      const agent = agentForPage(page, {
-        testId: testInfo.testId,
-        taskFile,
-        taskTitle
-      });
+      const agent = agentForPage(page, testInfo);
       await use(async (demand) => {
         await page.waitForLoadState("networkidle");
         const result = await agent.aiQuery(demand);
         return result;
       });
-      if (agent.dumpFile) {
-        testInfo.annotations.push({
-          type: "MIDSCENE_AI_ACTION",
-          description: JSON.stringify({
-            testId: testInfo.testId,
-            dumpPath: agent.dumpFile
-          })
-        });
-      }
+      updateDumpAnnotation(testInfo, agent.dumpDataString());
     },
     aiAssert: async ({ page }, use, testInfo) => {
-      const { taskFile, taskTitle } = groupAndCaseForTest(testInfo);
-      const agent = agentForPage(page, {
-        testId: testInfo.testId,
-        taskFile,
-        taskTitle
-      });
+      const agent = agentForPage(page, testInfo);
       await use(async (assertion, errorMsg) => {
         await page.waitForLoadState("networkidle");
         await agent.aiAssert(assertion, errorMsg);
       });
-      if (agent.dumpFile) {
-        testInfo.annotations.push({
-          type: "MIDSCENE_AI_ACTION",
-          description: JSON.stringify({
-            testId: testInfo.testId,
-            dumpPath: agent.dumpFile
-          })
-        });
-      }
+      updateDumpAnnotation(testInfo, agent.dumpDataString());
     }
   };
 };