Wait out auto-repair chain before failing on invoke --wait

README.md

@@ -83,12 +83,27 @@ Lark ships [Agent Skills](https://docs.getlark.ai/agents) that teach Claude Code
 npx skills add getlark/skills
 ```
 
+Or, equivalently, via the `getlark` CLI:
+
+```bash
+getlark skills install
+```
+
 Then ask the agent to run the `setup` skill. See the [Agents docs](https://docs.getlark.ai/agents) for the full skill catalog and the opt-in `PostToolUse` hook that validates your branch after every `git commit` or `git push`.
 
 ## CI Pipeline Usage
 
 The `--wait` flag makes it easy to use in CI pipelines. The command will block until the workflow completes and exit with a non-zero code on failure.
 
+If your account has auto-repair enabled for deterministic workflows, a failed execution is not counted as a failure right away. Instead `--wait` follows the repair chain before deciding the exit code:
+
+- If the failure summarization classifies it as an **app issue**, the workflow **fails**.
+- If the repair job itself **fails**, the workflow **fails**.
+- If the repair succeeds and the automatic re-run **passes**, the workflow **passes** (the test self-healed) and is reported as auto-repaired.
+- If the repair succeeds but the re-run **fails**, the workflow **fails**.
+
+Workflows in `ai_driven` mode, and accounts without auto-repair enabled, keep the previous fail-fast behavior. The `--timeout` covers the whole wait, including any repair chain.
+
 ### GitHub Actions Example
 
 Set the `GETLARK_API_KEY` environment variable in GitHub Actions secrets.
@@ -230,8 +245,8 @@ getlark workflows invoke --group-name "Checkout Flow" --wait
 | `--all`                    | Invoke all workflows                                                             |
 | `--group-id <groupId>`     | Invoke all workflows in a group (by group ID)                                    |
 | `--group-name <groupName>` | Invoke all workflows in a group (by group name)                                  |
-| `--wait`                   | Wait for the execution to finish (successfully or unsuccessfully) before exiting |
-| `--timeout <seconds>`      | Maximum time to wait in seconds (default: 600, requires `--wait`)                |
+| `--wait`                   | Wait for the execution to finish before exiting. When auto-repair is enabled, also waits out the repair chain for failed deterministic workflows (see [CI Pipeline Usage](#ci-pipeline-usage)) |
+| `--timeout <seconds>`      | Maximum time to wait in seconds (default: 600, requires `--wait`); covers the repair chain too |
 | `--verbose`                | Print verbose output (includes logs)                                             |
 
 One of `--workflow-ids`, `--all`, `--group-id`, or `--group-name` is required.

src/api/client.ts

@@ -14,11 +14,13 @@ import type {
   ListWorkflowGroupsResponse,
   ListWorkflowRepairsResponse,
   ListWorkflowsResponse,
+  SettingsResource,
   WorkflowExecutionResource,
   WorkflowGenerationResource,
   WorkflowGroupResource,
   WorkflowRepairResource,
   WorkflowResource,
+  WorkflowSummarizationResource,
 } from "./types.js";
 
 export class TimeoutError extends Error {
@@ -37,6 +39,38 @@ export interface PollOptions {
   ) => void | Promise<void>;
 }
 
+/** The stage of the post-failure repair chain currently being awaited. */
+export type RepairChainStage = "summarization" | "repair" | "re-execution";
+
+export interface RepairChainPollOptions {
+  timeoutMs: number;
+  pollIntervalMs: number;
+  onPoll?: (stage: RepairChainStage, elapsedMs: number) => void | Promise<void>;
+}
+
+/**
+ * The verdict of waiting out the auto-repair chain that follows a failed
+ * execution. `repaired` means the test self-healed (repair succeeded and the
+ * re-run passed); every `failure` reason counts as a genuine failure.
+ */
+export type RepairChainOutcome =
+  | {
+      result: "success";
+      reason: "repaired";
+      executionId: string;
+      summary: string | null;
+    }
+  | {
+      result: "failure";
+      reason:
+        | "app_issue"
+        | "summarization_failed"
+        | "repair_failed"
+        | "reexecution_failed";
+      executionId: string;
+      summary: string | null;
+    };
+
 export class GetLarkClient {
   private baseUrl: string;
   private apiKey: string;
@@ -312,6 +346,24 @@ export class GetLarkClient {
     return this.request<ListWorkflowEventsResponse>("GET", path);
   }
 
+  // ── Summarizations ─────────────────────────────────────────
+
+  async getWorkflowSummarization(
+    workflowId: string,
+    summarizationId: string,
+  ): Promise<WorkflowSummarizationResource> {
+    return this.request<WorkflowSummarizationResource>(
+      "GET",
+      `/workflows/${workflowId}/summarizations/${summarizationId}`,
+    );
+  }
+
+  // ── Settings ───────────────────────────────────────────────
+
+  async getSettings(): Promise<SettingsResource> {
+    return this.request<SettingsResource>("GET", "/settings");
+  }
+
   // ── Secret Contexts ────────────────────────────────────────
 
   async listSecretContexts(): Promise<ListSecretContextsResponse> {
@@ -475,6 +527,10 @@ export class GetLarkClient {
 
   // ── Polling ────────────────────────────────────────────────
 
+  private sleep(ms: number): Promise<void> {
+    return new Promise<void>((resolve) => setTimeout(resolve, ms));
+  }
+
   async pollWorkflowExecution(
     workflowId: string,
     executionId: string,
@@ -483,9 +539,6 @@ export class GetLarkClient {
     const { timeoutMs, pollIntervalMs, onPoll } = options;
     const startTime = Date.now();
 
-    const sleep = (ms: number) =>
-      new Promise<void>((resolve) => setTimeout(resolve, ms));
-
     const terminalStatuses = new Set(["success", "failure", "cancelled"]);
 
     while (true) {
@@ -507,7 +560,130 @@ export class GetLarkClient {
         );
       }
 
-      await sleep(pollIntervalMs);
+      await this.sleep(pollIntervalMs);
+    }
+  }
+
+  /**
+   * After an execution fails, an account with auto-repair enabled will run a
+   * summarization, an optional repair, and a follow-up re-execution. This
+   * follows that chain via the workflow event timeline and returns a verdict:
+   *
+   *   summarization "app_issue"        → failure (genuine app defect)
+   *   summarization not successful     → failure (inconclusive)
+   *   repair not successful            → failure
+   *   re-execution failure             → failure
+   *   re-execution success             → success (test self-healed)
+   */
+  async pollWorkflowRepairChain(
+    workflowId: string,
+    failedExecution: WorkflowExecutionResource,
+    options: RepairChainPollOptions,
+  ): Promise<RepairChainOutcome> {
+    const { timeoutMs, pollIntervalMs, onPoll } = options;
+    const startTime = Date.now();
+
+    const terminalStatuses = new Set(["success", "failure", "cancelled"]);
+    const failedAt = new Date(failedExecution.created_at).getTime();
+    const at = (ts: string | null) => (ts ? new Date(ts).getTime() : 0);
+
+    let stage: RepairChainStage = "summarization";
+
+    while (true) {
+      // Events are returned newest-first; reorder the events that belong to
+      // this failure (created after the failed execution) oldest-first so we
+      // can walk the chain in the order it happened.
+      const { workflow_events } = await this.listWorkflowEvents(workflowId, {
+        limit: 50,
+      });
+      const chain = workflow_events
+        .filter((e) => at(e.created_at) > failedAt)
+        .sort((a, b) => at(a.created_at) - at(b.created_at));
+
+      const elapsedMs = Date.now() - startTime;
+      await onPoll?.(stage, elapsedMs);
+
+      const summ = chain.find((e) => e.event_type === "summarization");
+      if (summ && terminalStatuses.has(summ.status)) {
+        if (summ.status !== "success") {
+          return {
+            result: "failure",
+            reason: "summarization_failed",
+            executionId: failedExecution.id,
+            summary: null,
+          };
+        }
+        const detail = await this.getWorkflowSummarization(workflowId, summ.id);
+        // Make sure this summarization is the one for our failed execution and
+        // not a newer one that raced ahead; if not, keep waiting for ours.
+        if (detail.workflow_execution_id !== failedExecution.id) {
+          if (elapsedMs >= timeoutMs) {
+            throw new TimeoutError(
+              `Timed out after ${Math.round(timeoutMs / 1000)}s waiting for repair of execution ${failedExecution.id} (stage: ${stage})`,
+            );
+          }
+          await this.sleep(pollIntervalMs);
+          continue;
+        }
+        if (detail.category === "app_issue") {
+          return {
+            result: "failure",
+            reason: "app_issue",
+            executionId: failedExecution.id,
+            summary: detail.summary,
+          };
+        }
+        // A test-side issue: an auto-repair should follow.
+        stage = "repair";
+
+        const repair = chain.find(
+          (e) =>
+            e.event_type === "repair" && at(e.created_at) >= at(summ.created_at),
+        );
+        if (repair && terminalStatuses.has(repair.status)) {
+          if (repair.status !== "success") {
+            return {
+              result: "failure",
+              reason: "repair_failed",
+              executionId: failedExecution.id,
+              summary: null,
+            };
+          }
+          // The repair succeeded; the backend re-runs the test.
+          stage = "re-execution";
+
+          const reExecution = chain.find(
+            (e) =>
+              e.event_type === "execution" &&
+              repair.stopped_at !== null &&
+              at(e.created_at) >= at(repair.stopped_at),
+          );
+          if (reExecution && terminalStatuses.has(reExecution.status)) {
+            if (reExecution.status === "success") {
+              return {
+                result: "success",
+                reason: "repaired",
+                executionId: reExecution.id,
+                summary: detail.summary,
+              };
+            }
+            return {
+              result: "failure",
+              reason: "reexecution_failed",
+              executionId: reExecution.id,
+              summary: detail.summary,
+            };
+          }
+        }
+      }
+
+      if (elapsedMs >= timeoutMs) {
+        throw new TimeoutError(
+          `Timed out after ${Math.round(timeoutMs / 1000)}s waiting for repair of execution ${failedExecution.id} (stage: ${stage})`,
+        );
+      }
+
+      await this.sleep(pollIntervalMs);
     }
   }
 }

src/api/types.ts

@@ -33,6 +33,10 @@ export interface WorkflowResource {
   last_repair_started_at: string | null;
   last_repair_stopped_at: string | null;
   last_repair_result_type: "success" | "failure" | "cancelled" | null;
+  last_summarization_id: string | null;
+  last_summarization_started_at: string | null;
+  last_summarization_stopped_at: string | null;
+  last_summarization_result_type: "success" | "failure" | "cancelled" | null;
   schedule: string | null;
   group_id: string | null;
   next_execution_at: string | null;
@@ -63,7 +67,8 @@ export interface WorkflowArtifactResource {
     | "video"
     | "javascript"
     | "python"
-    | "shellscript";
+    | "shellscript"
+    | "other";
   filename: string;
   presigned_url: string;
   presigned_url_expires_at: string;
@@ -130,6 +135,24 @@ export interface WorkflowRepairResource {
   updated_at: string;
 }
 
+export interface WorkflowSummarizationResource {
+  id: string;
+  workflow_id: string;
+  workflow_execution_id: string;
+  status: "pending" | "running" | "success" | "failure" | "cancelled";
+  // "app_issue" means the failure is a genuine defect in the app under test;
+  // any other category (e.g. "test_issue") indicates a test-side issue that
+  // an auto-repair may be able to fix.
+  category: "test_issue" | "app_issue";
+  started_at: string | null;
+  stopped_at: string | null;
+  summary: string | null;
+  secret_contexts: string[] | null;
+  artifacts: WorkflowArtifactResource[];
+  created_at: string;
+  updated_at: string;
+}
+
 export interface ListedWorkflowRepairResource {
   id: string;
   workflow_id: string;
@@ -149,7 +172,7 @@ export interface ListWorkflowRepairsResponse {
 export interface ListedWorkflowEventResource {
   id: string;
   workflow_id: string;
-  event_type: "generation" | "execution" | "repair";
+  event_type: "generation" | "execution" | "repair" | "summarization";
   status: "pending" | "running" | "success" | "failure" | "cancelled";
   started_at: string | null;
   stopped_at: string | null;
@@ -174,6 +197,13 @@ export interface ListWorkflowGroupsResponse {
   has_more: boolean;
 }
 
+export interface SettingsResource {
+  tasks_enabled: boolean;
+  deterministic_workflow_enabled: boolean;
+  auto_repair_deterministic_workflows_enabled: boolean;
+  qa_report_enabled: boolean;
+}
+
 export type JobType = "workflow_import";
 
 export type JobStatus =