@runuai/host 0.8.3 → 0.8.5

package/images/standard/Dockerfile

@@ -219,15 +219,32 @@ RUN bash -lc '\
   asdf reshim nodejs; \
 '
 
-# Kimi Code CLI (Moonshot) — single-binary install to /home/node/.kimi-code/bin.
-# The adapter invokes it by absolute path (docker exec ignores login-shell PATH),
-# and task-up copies the operator subscription creds in. Non-fatal: a Kimi CDN
-# hiccup must not break the image for Claude/Codex (the kimi engine just won't
-# be runnable until rebuilt).
-RUN bash -lc '\
-  curl -fsSL https://code.kimi.com/kimi-code/install.sh | bash \
-    || echo "[warn] kimi-code install failed — kimi engine unavailable in this image"; \
-'
+# Optional agent CLIs — installed ONLY when the host has them configured
+# (INSTALL_* build args from standard-image.ts, gated on the operator's creds
+# and folded into the rebuild hash, so a new login rebuilds). Single-binary
+# installs to /home/node/.{kimi-code,grok}/bin; the adapters invoke them by
+# absolute path and task-up copies the subscription creds in. Non-fatal: a CDN
+# hiccup must never break the image for Claude/Codex.
+ARG INSTALL_KIMI=0
+RUN if [ "$INSTALL_KIMI" = "1" ]; then \
+      bash -lc 'curl -fsSL https://code.kimi.com/kimi-code/install.sh | bash' \
+        || echo "[warn] kimi-code install failed — kimi engine unavailable in this image"; \
+    fi
+
+ARG INSTALL_GROK=0
+RUN if [ "$INSTALL_GROK" = "1" ]; then \
+      bash -lc 'curl -fsSL https://x.ai/cli/install.sh | bash' \
+        || echo "[warn] grok install failed — grok engine unavailable in this image"; \
+    fi
+
+# Cursor Agent (Anysphere) — installs to /home/node/.local/bin/cursor-agent.
+# Auth is CURSOR_API_KEY (env-injected by the adapter), so nothing to copy at
+# task-up — just the binary.
+ARG INSTALL_CURSOR=0
+RUN if [ "$INSTALL_CURSOR" = "1" ]; then \
+      bash -lc 'curl https://cursor.com/install -fsS | bash' \
+        || echo "[warn] cursor install failed — cursor engine unavailable in this image"; \
+    fi
 
 ENV PATH=/home/node/.local/bin:$PATH
 

package/lib/agents/cursor.ts

@@ -0,0 +1,324 @@
+/**
+ * CursorSession — a real AgentSession backed by **Cursor Agent** (`cursor-agent`),
+ * run inside the task container:
+ *
+ *   docker exec -i -e CURSOR_API_KEY=… task-<id>-app-1 \
+ *     /home/node/.local/bin/cursor-agent -p "<prompt>" \
+ *     --output-format stream-json --stream-partial-output --force --trust \
+ *     -m <model> [--resume <sessionId>]
+ *
+ * One-shot per turn (like Kimi/Grok); sends serialized; continuity via
+ * `--resume <sessionId>` captured from the stream. Cursor's stream-json is
+ * essentially Claude Code's — the RICHEST of the extra engines (streaming text
+ * AND tool cards):
+ *
+ *   {"type":"system","subtype":"init","session_id":…}          → session id
+ *   {"type":"thinking","subtype":"delta"|"completed",…}        → skipped
+ *   {"type":"assistant","message":{content:[{text}]},"timestamp_ms":…}
+ *                                                              → message_delta
+ *   {"type":"tool_call","subtype":"started","call_id","tool_call":{…}}
+ *                                                              → tool_call card
+ *   {"type":"result","subtype":"success"|"error","result",session_id}
+ *                                                              → message_complete + end
+ *
+ * AUTH is a Cursor API key (`CURSOR_API_KEY`) — env-injected at exec, like the
+ * Claude token (no config-dir copy). Cloud never sees it (ADR-015). `-p` runs
+ * headless with `--force` (auto-run tools; the container is the sandbox) and
+ * `--trust` (skip the workspace-trust prompt). Cursor has no system-prompt
+ * flag, so the briefing folds into the first turn's prompt.
+ */
+import { spawn, type ChildProcess } from "node:child_process";
+
+import { newId } from "../ulid";
+import { register } from "./registry";
+import type {
+  AgentEvent,
+  AgentEventHandler,
+  AgentKind,
+  AgentSession,
+  RosterAgent,
+} from "./types";
+
+const CURSOR_BIN = "/home/node/.local/bin/cursor-agent";
+
+// A curated slice of Cursor's ~32 models (full ids from `cursor-agent
+// --list-models`). `auto` lets Cursor pick; the rest are the frontier. The UI
+// also accepts any typed id (unverified round-trip). UPDATE as Cursor's lineup
+// changes.
+const CURSOR_MODELS = [
+  "auto",
+  "composer-2.5",
+  "claude-opus-4-8-thinking-high",
+  "gpt-5.6-sol-high",
+  "cursor-grok-4.5-high",
+  "claude-fable-5-thinking-high",
+];
+const CURSOR_DEFAULT_MODEL = "auto";
+const CURSOR_EFFORTS: string[] = []; // effort is baked into the model id
+
+// ---------------------------------------------------------------------------
+// Pure protocol mapping.
+// ---------------------------------------------------------------------------
+
+function assistantText(msg: Record<string, unknown>): string {
+  const message = msg.message as { content?: unknown } | undefined;
+  const content = Array.isArray(message?.content) ? message!.content : [];
+  let out = "";
+  for (const raw of content) {
+    const block = raw as { type?: unknown; text?: unknown };
+    if (block.type === "text" && typeof block.text === "string") out += block.text;
+  }
+  return out;
+}
+
+/** "shellToolCall" → "shell", "readFileToolCall" → "readFile". */
+function prettyTool(name: string): string {
+  return name.replace(/ToolCall$/, "") || "tool";
+}
+
+export interface MappedCursorLine {
+  textDelta?: string;
+  toolCall?: { id: string; title: string; detail: string };
+  end?: boolean;
+  finalText?: string;
+  sessionId?: string;
+  errorText?: string;
+}
+
+export function mapCursorLine(line: string): MappedCursorLine {
+  const t = line.trim();
+  if (!t.startsWith("{")) return {};
+  let m: Record<string, unknown>;
+  try {
+    m = JSON.parse(t) as Record<string, unknown>;
+  } catch {
+    return {};
+  }
+
+  if (m.type === "system") {
+    return typeof m.session_id === "string" ? { sessionId: m.session_id } : {};
+  }
+  if (m.type === "assistant") {
+    // Streaming partials carry timestamp_ms; the final full message (finalized
+    // by `result`) does not — only stream the partials to avoid doubling.
+    if (typeof m.timestamp_ms === "number") {
+      const text = assistantText(m);
+      if (text) return { textDelta: text };
+    }
+    return {};
+  }
+  if (m.type === "tool_call" && m.subtype === "started") {
+    const tc = (m.tool_call ?? {}) as Record<string, unknown>;
+    const [name, spec] = Object.entries(tc)[0] ?? ["tool", {}];
+    const args = ((spec as Record<string, unknown>)?.args ?? {}) as Record<string, unknown>;
+    const pick = args.command ?? args.path ?? args.filePath ?? args.file_path ?? args.query;
+    return {
+      toolCall: {
+        id: typeof m.call_id === "string" ? m.call_id : newId(),
+        title: prettyTool(name),
+        detail: typeof pick === "string" ? pick : "",
+      },
+    };
+  }
+  if (m.type === "result") {
+    const errored = m.is_error === true;
+    return {
+      end: true,
+      sessionId: typeof m.session_id === "string" ? m.session_id : undefined,
+      finalText: typeof m.result === "string" ? m.result : undefined,
+      errorText: errored
+        ? typeof m.result === "string"
+          ? m.result
+          : "cursor turn failed"
+        : undefined,
+    };
+  }
+  // thinking / user / other → nothing.
+  return {};
+}
+
+// ---------------------------------------------------------------------------
+// Session
+// ---------------------------------------------------------------------------
+
+export type Spawner = (args: string[]) => ChildProcess;
+const defaultSpawn: Spawner = (args) =>
+  spawn("docker", args, { stdio: ["ignore", "pipe", "pipe"] });
+
+export class CursorSession implements AgentSession {
+  readonly agentId: string;
+  readonly kind: AgentKind = "cursor";
+
+  private readonly containerName: string;
+  private readonly model?: string;
+  private readonly systemPreamble: string;
+  private readonly agentEnv: Record<string, string>;
+  private readonly spawner: Spawner;
+  private readonly apiKey: string;
+
+  private readonly handlers = new Set<AgentEventHandler>();
+  private sessionId: string | null = null;
+  private current: ChildProcess | null = null;
+  private queue: Promise<void> = Promise.resolve();
+  private sentPreamble = false;
+  private closed = false;
+
+  constructor(args: {
+    taskId: string;
+    agent: RosterAgent;
+    containerName: string;
+    systemPreamble: string;
+    agentEnv?: Record<string, string>;
+    spawner?: Spawner;
+    apiKey?: string;
+  }) {
+    this.agentId = args.agent.id;
+    this.containerName = args.containerName;
+    this.model = args.agent.model;
+    this.systemPreamble = args.systemPreamble;
+    this.agentEnv = args.agentEnv ?? {};
+    this.spawner = args.spawner ?? defaultSpawn;
+    this.apiKey = args.apiKey ?? process.env.CURSOR_API_KEY ?? "";
+  }
+
+  onEvent(handler: AgentEventHandler): () => void {
+    this.handlers.add(handler);
+    return () => this.handlers.delete(handler);
+  }
+
+  private emit(event: AgentEvent): void {
+    if (this.closed && event.type !== "exit") return;
+    for (const h of this.handlers) h(event);
+  }
+
+  async send(text: string): Promise<void> {
+    if (this.closed) return;
+    this.queue = this.queue.then(() => this.runTurn(text));
+    return this.queue;
+  }
+
+  private buildArgs(prompt: string): string[] {
+    const args = ["exec", "-i", "-u", "node", "-e", `CURSOR_API_KEY=${this.apiKey}`];
+    for (const [k, v] of Object.entries(this.agentEnv)) {
+      args.push("-e", `${k}=${v}`);
+    }
+    args.push(
+      this.containerName,
+      CURSOR_BIN,
+      "-p",
+      prompt,
+      "--output-format",
+      "stream-json",
+      "--stream-partial-output",
+      "--force", // container is the sandbox — auto-run tools
+      "--trust", // skip the workspace-trust prompt in headless
+    );
+    if (this.model && this.model !== "auto") args.push("-m", this.model);
+    if (this.sessionId) args.push("--resume", this.sessionId);
+    return args;
+  }
+
+  private async runTurn(text: string): Promise<void> {
+    if (this.closed) return;
+    // No system-prompt flag — fold the briefing into the first turn; later
+    // turns carry it via the resumed session.
+    const prompt =
+      !this.sentPreamble && this.systemPreamble.trim().length > 0
+        ? `${this.systemPreamble}\n\n---\n\n${text}`
+        : text;
+    this.sentPreamble = true;
+
+    const child = this.spawner(this.buildArgs(prompt));
+    this.current = child;
+
+    let acc = "";
+    let sawText = false;
+    let buf = "";
+    const consume = (chunk: string): void => {
+      buf += chunk;
+      let nl: number;
+      while ((nl = buf.indexOf("\n")) >= 0) {
+        const line = buf.slice(0, nl);
+        buf = buf.slice(nl + 1);
+        const m = mapCursorLine(line);
+        if (m.sessionId) this.sessionId = m.sessionId;
+        if (typeof m.textDelta === "string") {
+          sawText = true;
+          acc += m.textDelta;
+          this.emit({ type: "message_delta", text: m.textDelta });
+        }
+        if (m.toolCall) {
+          this.emit({
+            type: "tool_call",
+            id: m.toolCall.id,
+            title: m.toolCall.title,
+            detail: m.toolCall.detail,
+          });
+        }
+        if (m.end) {
+          if (m.errorText) this.emit({ type: "error", message: m.errorText });
+          const finalText = m.finalText ?? acc;
+          if (sawText || finalText) {
+            this.emit({ type: "message_complete", text: finalText });
+          }
+        }
+      }
+    };
+    child.stdout?.on("data", (b: Buffer) => consume(b.toString("utf8")));
+    let stderr = "";
+    child.stderr?.on("data", (b: Buffer) => {
+      stderr += b.toString("utf8");
+    });
+
+    await new Promise<void>((resolve) => {
+      const finish = (code: number | null, spawnErr?: string): void => {
+        this.current = null;
+        if (buf.trim().length > 0) consume("\n");
+        if (this.closed) return resolve();
+        if (spawnErr) {
+          this.emit({ type: "error", message: `cursor spawn failed: ${spawnErr}` });
+        } else if (code !== 0) {
+          const tail = stderr.trim().slice(-500);
+          this.emit({
+            type: "error",
+            message: `cursor exited ${code ?? "null"}${tail ? `: ${tail}` : ""}`,
+          });
+        }
+        this.emit({ type: "turn_complete" });
+        resolve();
+      };
+      child.on("exit", (code) => finish(code));
+      child.on("error", (err) => finish(null, err.message));
+    });
+  }
+
+  async interrupt(): Promise<void> {
+    this.current?.kill("SIGKILL");
+  }
+
+  async resolvePermission(): Promise<void> {
+    /* --force auto-approves; no permission requests are emitted */
+  }
+
+  async close(): Promise<void> {
+    if (this.closed) return;
+    this.closed = true;
+    this.current?.kill("SIGKILL");
+    this.current = null;
+    this.emit({ type: "exit", code: 0 });
+    this.handlers.clear();
+  }
+}
+
+register({
+  kind: "cursor",
+  label: "Cursor",
+  supportedModels: () => [...CURSOR_MODELS],
+  defaultModel: CURSOR_DEFAULT_MODEL,
+  supportedEfforts: () => [...CURSOR_EFFORTS],
+  // Gated on a Cursor API key in the host env (loaded from .env.local); the
+  // adapter injects it into the container. No key → engine not advertised.
+  available: () => Boolean(process.env.CURSOR_API_KEY),
+  create: async ({ taskId, agent, containerName, systemPreamble, agentEnv }) =>
+    new CursorSession({ taskId, agent, containerName, systemPreamble, agentEnv }),
+});

package/lib/agents/factory.ts

@@ -18,6 +18,8 @@
 import "./claude";
 import "./codex";
 import "./kimi";
+import "./grok";
+import "./cursor";
 
 import { factoryFor } from "./registry";
 import type { AgentSession, AgentSessionFactory } from "./types";

package/lib/agents/grok.ts

@@ -0,0 +1,255 @@
+/**
+ * GrokSession — a real AgentSession backed by xAI's **Grok** CLI, run inside
+ * the task container:
+ *
+ *   docker exec -i task-<id>-app-1 ~/.grok/bin/grok -p "<prompt>" \
+ *     --output-format streaming-json --permission-mode bypassPermissions \
+ *     -m grok-4.5 --system-prompt-override "<briefing>" [-r <sessionId>]
+ *
+ * Like Kimi (and unlike Claude's persistent stdin / Codex's app-server), Grok
+ * headless mode is **one-shot per turn**: `-p` runs one prompt to completion,
+ * streams newline-delimited JSON, and exits. Each `send()` spawns a fresh
+ * `docker exec`; sends are **serialized** so a second message can't race a
+ * running turn. Continuity is `-r <sessionId>` — captured from the turn-end
+ * event (per-agent isolation, since each session tracks its own id).
+ *
+ * Stream-json vocabulary (`--output-format streaming-json`, verified):
+ *   {"type":"thought","data":"…"}                 → skipped (internal reasoning)
+ *   {"type":"text","data":"…"}                     → message_delta (streamed)
+ *   {"type":"end","stopReason","sessionId",…}      → message_complete + turn end
+ * Tool calls happen (num_turns>1) but aren't surfaced as events in headless
+ * mode, so there are no tool cards — the agent works, you get the streamed
+ * answer.
+ *
+ * AUTH is the operator's Grok subscription: `~/.grok/{auth.json, config.toml}`
+ * is docker-cp'd into the container at task-up (the Codex/Kimi config-dir
+ * pattern; the macOS `bin/` is NOT copied — the image supplies the Linux
+ * binary). auth.json is an OIDC token with a refresh_token, so the CLI renews
+ * it in-container. Cloud never sees it (ADR-015). `--system-prompt-override`
+ * carries the channel briefing (no fold-into-turn-1 needed).
+ */
+import { spawn, type ChildProcess } from "node:child_process";
+import { existsSync } from "node:fs";
+import { homedir } from "node:os";
+import { join } from "node:path";
+
+import { register } from "./registry";
+import type {
+  AgentEvent,
+  AgentEventHandler,
+  AgentKind,
+  AgentSession,
+  RosterAgent,
+} from "./types";
+
+const GROK_BIN = "/home/node/.grok/bin/grok";
+
+// `grok models` lists what the account can run. grok-4.5 is the current
+// default; UPDATE WHEN xAI CHANGES the lineup.
+const GROK_MODELS = ["grok-4.5"];
+const GROK_DEFAULT_MODEL = "grok-4.5";
+// Grok has no per-invocation reasoning-effort flag.
+const GROK_EFFORTS: string[] = [];
+
+function grokAuthPath(): string {
+  return join(process.env.UAI_OWNER_HOME?.trim() || homedir(), ".grok", "auth.json");
+}
+
+// ---------------------------------------------------------------------------
+// Pure protocol mapping — one streaming-json line → deltas / end.
+// ---------------------------------------------------------------------------
+
+export interface MappedGrokLine {
+  /** A streamed text chunk (assistant answer), if this line carried one. */
+  textDelta?: string;
+  /** True on the turn-end event. */
+  end?: boolean;
+  /** The session id from the end event, for `-r` on the next turn. */
+  sessionId?: string;
+}
+
+export function mapGrokLine(line: string): MappedGrokLine {
+  const trimmed = line.trim();
+  if (!trimmed.startsWith("{")) return {};
+  let msg: Record<string, unknown>;
+  try {
+    msg = JSON.parse(trimmed) as Record<string, unknown>;
+  } catch {
+    return {};
+  }
+  if (msg.type === "text" && typeof msg.data === "string") {
+    return { textDelta: msg.data };
+  }
+  if (msg.type === "end") {
+    return {
+      end: true,
+      sessionId: typeof msg.sessionId === "string" ? msg.sessionId : undefined,
+    };
+  }
+  // "thought" (reasoning) and any other types: no user-facing event.
+  return {};
+}
+
+// ---------------------------------------------------------------------------
+// Session
+// ---------------------------------------------------------------------------
+
+export type Spawner = (args: string[]) => ChildProcess;
+const defaultSpawn: Spawner = (args) =>
+  spawn("docker", args, { stdio: ["ignore", "pipe", "pipe"] });
+
+export class GrokSession implements AgentSession {
+  readonly agentId: string;
+  readonly kind: AgentKind = "grok";
+
+  private readonly containerName: string;
+  private readonly model?: string;
+  private readonly systemPreamble: string;
+  private readonly agentEnv: Record<string, string>;
+  private readonly spawner: Spawner;
+
+  private readonly handlers = new Set<AgentEventHandler>();
+  private sessionId: string | null = null;
+  private current: ChildProcess | null = null;
+  private queue: Promise<void> = Promise.resolve();
+  private closed = false;
+
+  constructor(args: {
+    taskId: string;
+    agent: RosterAgent;
+    containerName: string;
+    systemPreamble: string;
+    agentEnv?: Record<string, string>;
+    spawner?: Spawner;
+  }) {
+    this.agentId = args.agent.id;
+    this.containerName = args.containerName;
+    this.model = args.agent.model;
+    this.systemPreamble = args.systemPreamble;
+    this.agentEnv = args.agentEnv ?? {};
+    this.spawner = args.spawner ?? defaultSpawn;
+  }
+
+  onEvent(handler: AgentEventHandler): () => void {
+    this.handlers.add(handler);
+    return () => this.handlers.delete(handler);
+  }
+
+  private emit(event: AgentEvent): void {
+    if (this.closed && event.type !== "exit") return;
+    for (const h of this.handlers) h(event);
+  }
+
+  async send(text: string): Promise<void> {
+    if (this.closed) return;
+    this.queue = this.queue.then(() => this.runTurn(text));
+    return this.queue;
+  }
+
+  private buildArgs(prompt: string): string[] {
+    const args = ["exec", "-i", "-u", "node"];
+    for (const [k, v] of Object.entries(this.agentEnv)) {
+      args.push("-e", `${k}=${v}`);
+    }
+    args.push(
+      this.containerName,
+      GROK_BIN,
+      "-p",
+      prompt,
+      "--output-format",
+      "streaming-json",
+      // The container is the isolation boundary — auto-run tools.
+      "--permission-mode",
+      "bypassPermissions",
+    );
+    if (this.model) args.push("-m", this.model);
+    if (this.systemPreamble.trim().length > 0) {
+      args.push("--system-prompt-override", this.systemPreamble);
+    }
+    // Resume the same conversation across turns (captured from the last end).
+    if (this.sessionId) args.push("-r", this.sessionId);
+    return args;
+  }
+
+  private async runTurn(text: string): Promise<void> {
+    if (this.closed) return;
+    const child = this.spawner(this.buildArgs(text));
+    this.current = child;
+
+    let acc = "";
+    let sawText = false;
+    let buf = "";
+    const consume = (chunk: string): void => {
+      buf += chunk;
+      let nl: number;
+      while ((nl = buf.indexOf("\n")) >= 0) {
+        const line = buf.slice(0, nl);
+        buf = buf.slice(nl + 1);
+        const m = mapGrokLine(line);
+        if (m.sessionId) this.sessionId = m.sessionId;
+        if (typeof m.textDelta === "string") {
+          sawText = true;
+          acc += m.textDelta;
+          this.emit({ type: "message_delta", text: m.textDelta });
+        }
+      }
+    };
+    child.stdout?.on("data", (b: Buffer) => consume(b.toString("utf8")));
+    let stderr = "";
+    child.stderr?.on("data", (b: Buffer) => {
+      stderr += b.toString("utf8");
+    });
+
+    await new Promise<void>((resolve) => {
+      const finish = (code: number | null, spawnErr?: string): void => {
+        this.current = null;
+        if (buf.trim().length > 0) consume("\n"); // flush trailing line
+        if (this.closed) return resolve();
+        if (spawnErr) {
+          this.emit({ type: "error", message: `grok spawn failed: ${spawnErr}` });
+        } else if (code !== 0) {
+          const tail = stderr.trim().slice(-500);
+          this.emit({
+            type: "error",
+            message: `grok exited ${code ?? "null"}${tail ? `: ${tail}` : ""}`,
+          });
+        }
+        // Finalize the streamed message (no-op text if the turn produced none).
+        if (sawText) this.emit({ type: "message_complete", text: acc });
+        this.emit({ type: "turn_complete" });
+        resolve();
+      };
+      child.on("exit", (code) => finish(code));
+      child.on("error", (err) => finish(null, err.message));
+    });
+  }
+
+  async interrupt(): Promise<void> {
+    this.current?.kill("SIGKILL");
+  }
+
+  // bypassPermissions auto-approves; no permission requests are emitted.
+  async resolvePermission(): Promise<void> {
+    /* no-op */
+  }
+
+  async close(): Promise<void> {
+    if (this.closed) return;
+    this.closed = true;
+    this.current?.kill("SIGKILL");
+    this.current = null;
+    this.emit({ type: "exit", code: 0 });
+    this.handlers.clear();
+  }
+}
+
+register({
+  kind: "grok",
+  label: "Grok",
+  supportedModels: () => [...GROK_MODELS],
+  defaultModel: GROK_DEFAULT_MODEL,
+  supportedEfforts: () => [...GROK_EFFORTS],
+  available: () => existsSync(grokAuthPath()),
+  create: async ({ taskId, agent, containerName, systemPreamble, agentEnv }) =>
+    new GrokSession({ taskId, agent, containerName, systemPreamble, agentEnv }),
+});

package/lib/standard-image.ts

@@ -18,7 +18,9 @@
 
 import { spawn } from "node:child_process";
 import { createHash } from "node:crypto";
+import { existsSync } from "node:fs";
 import { readdir, readFile } from "node:fs/promises";
+import { homedir } from "node:os";
 import { dirname, join, resolve } from "node:path";
 import { fileURLToPath } from "node:url";
 
@@ -67,7 +69,29 @@ const CONTEXT_HASH_LABEL = "com.runuai.context-hash";
  * by relative path). Null when the context can't be read — the caller then
  * keeps whatever image exists.
  */
-async function hashBuildContext(): Promise<string | null> {
+/**
+ * Which OPTIONAL agent CLIs the operator has actually configured — so the
+ * image installs only those, not every engine on every host. Keyed on the
+ * same credential each adapter's `available()` checks (kimi/grok are copied
+ * into containers at task-up; claude/codex are always baked). Folded into the
+ * build hash below, so logging into a new engine triggers a rebuild.
+ */
+export function configuredOptionalEngines(): {
+  kimi: boolean;
+  grok: boolean;
+  cursor: boolean;
+} {
+  const home = process.env.UAI_OWNER_HOME?.trim() || homedir();
+  return {
+    kimi: existsSync(join(home, ".kimi-code", "credentials", "kimi-code.json")),
+    grok: existsSync(join(home, ".grok", "auth.json")),
+    // Cursor auths via CURSOR_API_KEY (env, loaded from .env.local), not a
+    // config file — install it when the key is present.
+    cursor: Boolean(process.env.CURSOR_API_KEY),
+  };
+}
+
+async function hashBuildContext(extra = ""): Promise<string | null> {
   try {
     const root = standardImageDir();
     const files: string[] = [];
@@ -87,6 +111,10 @@ async function hashBuildContext(): Promise<string | null> {
       hash.update(await readFile(join(root, rel)));
       hash.update("\0");
     }
+    // Build args (which optional engines are installed) are part of the image
+    // identity — a config change must invalidate the label so it rebuilds.
+    hash.update(extra);
+    hash.update("\0");
     return hash.digest("hex").slice(0, 32);
   } catch {
     return null;
@@ -304,7 +332,20 @@ export async function ensureStandardImage(): Promise<void> {
   //    landed). The build context is content-hashed into an image label;
   //    a mismatch triggers a rebuild (layer cache keeps it cheap).
   let imageReady = false;
-  const contextHash = await hashBuildContext();
+  // Install only the optional engines the operator has configured; the flags
+  // are build args AND part of the content hash (so a new login rebuilds).
+  const engines = configuredOptionalEngines();
+  const engineArgs = [
+    "--build-arg",
+    `INSTALL_KIMI=${engines.kimi ? 1 : 0}`,
+    "--build-arg",
+    `INSTALL_GROK=${engines.grok ? 1 : 0}`,
+    "--build-arg",
+    `INSTALL_CURSOR=${engines.cursor ? 1 : 0}`,
+  ];
+  const contextHash = await hashBuildContext(
+    `kimi=${engines.kimi ? 1 : 0};grok=${engines.grok ? 1 : 0};cursor=${engines.cursor ? 1 : 0}`,
+  );
   const inspect = await run("docker", [
     "image",
     "inspect",
@@ -338,6 +379,7 @@ export async function ensureStandardImage(): Promise<void> {
       "build",
       "-t",
       STANDARD_IMAGE_TAG,
+      ...engineArgs,
       ...(contextHash !== null
         ? ["--label", `${CONTEXT_HASH_LABEL}=${contextHash}`]
         : []),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@runuai/host",
-  "version": "0.8.3",
+  "version": "0.8.5",
   "description": "Uai host — runs ephemeral AI coding tasks in Docker on a machine you control.",
   "license": "MIT",
   "author": "Diogo Perillo <diogo.perillo@gmail.com>",

package/scripts/agent/task-up.sh

@@ -492,6 +492,21 @@ fi
 docker exec -u root "$app_container" \
   chown -R node:node /home/node/.kimi-code >/dev/null 2>&1 || true
 
+# Copy Grok subscription/config into a task-private /home/node/.grok. The Linux
+# `grok` binary is baked into the image; here we copy the arch-independent
+# auth + config (auth.json is an OIDC token the CLI refreshes in-container).
+docker exec -u root "$app_container" \
+  mkdir -p /home/node/.grok >/dev/null 2>&1 || true
+for grok_item in auth.json config.toml models_cache.json agent_id; do
+  if [ -e "$UAI_OWNER_HOME/.grok/$grok_item" ]; then
+    docker cp "$UAI_OWNER_HOME/.grok/$grok_item" \
+      "$app_container":/home/node/.grok/ >/dev/null 2>&1 \
+      || log "warning: docker cp of .grok/$grok_item failed; grok may need re-login"
+  fi
+done
+docker exec -u root "$app_container" \
+  chown -R node:node /home/node/.grok >/dev/null 2>&1 || true
+
 # Copy the same resolved SSH identity (task creator's per-user key when present,
 # else the operator identity — see above) into the container, so the agent signs
 # + pushes with the key whose .pub the user registered on GitHub. ADR-027 drops