Normalize replay waitForLoadState targets

Co-authored-by: Shri Sukhani <shrisukhani@users.noreply.github.com>

Author: cursoragent

src/agent/shared/replay-special-actions.test.ts

@@ -163,6 +163,21 @@ describe("executeReplaySpecialAction", () => {
     expect(result?.output).toBe("Waited for load state: networkidle");
   });
 
+  it("defaults waitForLoadState to domcontentloaded for unsupported values", async () => {
+    const page = createPage();
+
+    const result = await executeReplaySpecialAction({
+      taskId: "task-loadstate-invalid",
+      actionType: "waitForLoadState",
+      arguments: ["interactive"],
+      page: page as unknown as Page,
+    });
+
+    expect(page.waitForLoadState).toHaveBeenCalledWith("domcontentloaded", undefined);
+    expect(result?.status).toBe("completed");
+    expect(result?.output).toBe("Waited for load state: domcontentloaded");
+  });
+
   it("fails extract replay when extracted object cannot be serialized", async () => {
     const circular: Record<string, unknown> = {};
     circular.self = circular;

src/agent/shared/replay-special-actions.ts

@@ -68,6 +68,14 @@ function normalizeWaitMs(value: unknown): number {
   return parsed >= 0 ? parsed : 1000;
 }
 
+function normalizeWaitUntil(value: unknown): "domcontentloaded" | "load" | "networkidle" {
+  const parsed = asNonEmptyTrimmedString(value);
+  if (parsed === "load" || parsed === "networkidle") {
+    return parsed;
+  }
+  return "domcontentloaded";
+}
+
 export async function executeReplaySpecialAction(
   params: ReplaySpecialActionInput
 ): Promise<TaskOutput | null> {
@@ -215,12 +223,12 @@ export async function executeReplaySpecialAction(
   }
 
   if (actionType === "waitForLoadState") {
-    const waitUntil = asNonEmptyTrimmedString(actionArgs?.[0]) ?? "domcontentloaded";
+    const waitUntil = normalizeWaitUntil(actionArgs?.[0]);
     const timeoutMs = asFiniteNumber(actionArgs?.[1]);
     const options =
       timeoutMs !== undefined ? { timeout: timeoutMs } : undefined;
     await page.waitForLoadState(
-      waitUntil as "domcontentloaded" | "load" | "networkidle",
+      waitUntil,
       options
     );
     markDomSnapshotDirty(page);