@runuai/host 0.9.116 → 0.9.117

package/lib/orchestrator.ts

@@ -135,6 +135,7 @@ import type {
   HostEvent,
 } from "../src/protocol";
 import {
+  type TaskStartStage,
   HOST_ACCOUNT_SCOPE,
   MAX_AGENT_ID_CHARS,
   MAX_SECRETARY_DISPATCH_ID_CHARS,
@@ -455,6 +456,11 @@ export class Orchestrator {
     this.emitHost({ kind: "system.note", taskId, text });
   }
 
+  /** A task-start stage boundary (the launch screen on the task page). */
+  emitTaskStage(taskId: string, stage: TaskStartStage): void {
+    this.emitHost({ kind: "task.stage", taskId, stage, at: Date.now() });
+  }
+
   private emitHost(event: HostEvent): void {
     for (const fn of this.hostSubscribers) {
       try {
@@ -1889,6 +1895,7 @@ export class Orchestrator {
     // which the client caps at 10s (Codex) / 30s (Claude Code). Idempotent
     // and ~0.5s once the build is present; a failure arms its own 60s retry
     // rather than depending on another delivery/ensure to arrive.
+    this.emitTaskStage(channel.taskId, "tools");
     await this.ensureBrowserSetup(
       channel,
       initialRoster.some((a) => a.kind === "codex"),
@@ -1923,6 +1930,7 @@ export class Orchestrator {
       return false;
     }
 
+    this.emitTaskStage(channel.taskId, "agents");
     for (const agent of initialRoster) {
       if (!this.isActiveChannel(channel)) return false;
       // A prior partial start or an independently guarded recovery may already
@@ -1972,6 +1980,7 @@ export class Orchestrator {
     // so it both duplicated the opening turn AND re-injected it on a host
     // restart, making agents redo the first turn mid-task. The persona/mission
     // an agent needs live in the always-on system preamble, so nothing is lost.
+    this.emitTaskStage(channel.taskId, "ready");
     return true;
   }
 

package/lib/task-environment/index.ts

@@ -128,7 +128,7 @@ registry.register(appleProvider);
 // host) until machine-backing becomes a per-task property.
 // ---------------------------------------------------------------------------
 
-function machineTasksEnabled(): boolean {
+export function machineTasksEnabled(): boolean {
   return process.env.UAI_MACHINE_TASKS === "1";
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@runuai/host",
-  "version": "0.9.116",
+  "version": "0.9.117",
   "description": "Uai host — runs ephemeral AI tasks in containers on a machine you control.",
   "license": "MIT",
   "author": "Uai Tech <team@runuai.com>",

package/src/index.ts

@@ -77,6 +77,7 @@ import {
 } from "../lib/container-runtime";
 import { vnodeAdmissionProblem } from "../lib/vnode-pressure";
 import {
+  machineTasksEnabled,
   provisionTaskEnvironment,
   reconstructPersistedTaskEnvironment,
 } from "../lib/task-environment";
@@ -818,6 +819,13 @@ export const hostCommands: HostCommands = {
             "GitHub is connected on this host, but Uai could not prepare its credential for Git. Retry the task; if it persists, reconnect GitHub on this host.",
           );
         }
+        // Launch screen: the first stage boundary. A machine-backed task
+        // spends the next minute launching an EC2 instance; a container
+        // task goes straight to preparing its workspace.
+        orchestrator.emitTaskStage(
+          input.task.id,
+          machineTasksEnabled() ? "machine" : "workspace",
+        );
         try {
           const provisioned = gitCredential
             ? await gitCredential.run(() =>

package/src/protocol.ts

@@ -1831,6 +1831,15 @@ export type HostToCloud =
   | EngineConnectEventFrame
   | HostConfigAckFrame;
 
+/** Task start stages, in order (docs/agent-notes.md "task start screen"). The
+ *  host emits them as `task.stage` events while a task comes up; the page
+ *  shows a launch screen instead of the chat until `ready`. */
+export const TASK_START_STAGES = ["machine", "workspace", "tools", "agents", "ready"] as const;
+export type TaskStartStage = (typeof TASK_START_STAGES)[number];
+export function isTaskStartStage(value: unknown): value is TaskStartStage {
+  return (TASK_START_STAGES as readonly unknown[]).includes(value);
+}
+
 export type HostEvent =
   | {
       kind: "task.status_changed";
@@ -1838,6 +1847,7 @@ export type HostEvent =
       from: string;
       to: string;
     }
+  | { kind: "task.stage"; taskId: string; stage: TaskStartStage; at: number }
   | {
       kind: "task.log";
       taskId: string;