@prismatic-io/lux 0.0.2-preview.22 → 0.0.2-preview.24

package/src/drivers/grok-build/index.ts

@@ -0,0 +1,329 @@
+import { realpath } from "node:fs/promises";
+import {
+  type AgentDriver,
+  type Answer,
+  type Artifact,
+  type DriverEventMap,
+  defineDriver,
+  discoverCliVersion,
+  type ReadyState,
+  type StartContext,
+  selectModelAndReasoningEffort,
+  TypedEmitter,
+} from "../../core/index.js";
+import { AcpEvents } from "../shared/acp-events.js";
+import { permissionAnswer, permissionRequestText, records } from "../shared/acp-interaction.js";
+import { AcpTransport, asRecord, type RpcId, type RpcRecord } from "../shared/acp-transport.js";
+import { walkArtifacts } from "../shared/artifacts.js";
+import {
+  buildGrokBuildArgs,
+  type GrokBuildDriverConfig,
+  GrokBuildDriverConfigSchema,
+} from "./config.js";
+
+class GrokBuildDriver implements AgentDriver {
+  private readonly emitter = new TypedEmitter<DriverEventMap>();
+  private readonly events: AcpEvents;
+  private transport: AcpTransport | undefined;
+  private sessionId: string | undefined;
+  private cwd: string | undefined;
+  private artifactsDir: string | undefined;
+  private ready = false;
+  private runPromise: Promise<void> | undefined;
+  private terminal = false;
+  private idleTimer: NodeJS.Timeout | undefined;
+  private interruptCount = 0;
+  private readonly interrupts = new Map<
+    string,
+    { id: RpcId; params: RpcRecord; timer: NodeJS.Timeout }
+  >();
+
+  private readonly config: GrokBuildDriverConfig;
+
+  constructor(config: GrokBuildDriverConfig) {
+    this.config = config;
+    this.events = new AcpEvents("grok-build", config.maxToolCalls, config.maxQueuedBytes);
+  }
+
+  on<K extends keyof DriverEventMap>(
+    event: K,
+    handler: (e: DriverEventMap[K]) => void | Promise<void>,
+  ): void {
+    this.emitter.on(event, handler);
+  }
+  off<K extends keyof DriverEventMap>(
+    event: K,
+    handler: (e: DriverEventMap[K]) => void | Promise<void>,
+  ): void {
+    this.emitter.off(event, handler);
+  }
+
+  async start(ctx: StartContext): Promise<ReadyState> {
+    if (this.transport || this.terminal)
+      throw new Error("Grok.build driver cannot be started twice or after close");
+    this.cwd = await realpath(this.config.cwd ?? ctx.artifactsDir);
+    this.artifactsDir = await realpath(ctx.artifactsDir);
+    const cliVersion = await discoverCliVersion(this.config.command);
+    const transport = new AcpTransport({
+      label: "Grok.build ACP",
+      command: this.config.command,
+      args: buildGrokBuildArgs(this.config),
+      cwd: this.cwd,
+      env: { ...process.env, ...this.config.env },
+      signal: ctx.abortSignal,
+      maxLineBytes: this.config.maxLineBytes,
+      maxQueuedBytes: this.config.maxQueuedBytes,
+      onMessage: (message) => this.message(message),
+      onFailure: (error) => this.fail(error),
+      onActivity: () => this.bumpIdleTimer(),
+    });
+    this.transport = transport;
+    try {
+      await transport.spawned();
+      const initialized = await this.request("initialize", {
+        protocolVersion: 1,
+        clientInfo: { name: "lux", version: "1" },
+        clientCapabilities: { fs: { readTextFile: false, writeTextFile: false }, terminal: false },
+      });
+      if (initialized.protocolVersion !== 1)
+        throw new Error("Grok.build ACP returned an unsupported protocol version");
+      if (asRecord(initialized._meta)?.grokShell !== true)
+        throw new Error("Expected Grok.build ACP; check the configured command points to grok");
+      const session = await this.request("session/new", {
+        cwd: this.cwd,
+        mcpServers: [],
+        _meta: { autoMode: this.config.permissionMode === "auto", yoloMode: false },
+      });
+      if (typeof session.sessionId !== "string")
+        throw new Error("Grok.build ACP omitted sessionId");
+      this.sessionId = session.sessionId;
+      let nativeConfig = session.configOptions;
+      for (const [configId, value] of [
+        ["model", this.config.model],
+        ["reasoning_effort", this.config.reasoningEffort],
+      ] as const) {
+        const updated = await this.request("session/set_config_option", {
+          sessionId: this.sessionId,
+          configId,
+          value,
+        });
+        nativeConfig = updated.configOptions;
+        if (records(nativeConfig).find((option) => option.id === configId)?.currentValue !== value)
+          throw new Error(`Grok.build ACP did not apply requested ${configId} '${value}'`);
+      }
+      this.ready = true;
+      this.bumpIdleTimer();
+      this.runPromise = this.run(ctx.prompt).catch((error: unknown) => this.fail(error));
+      return {
+        agentId: "grok-build",
+        model: this.config.model,
+        ...(typeof asRecord(initialized._meta)?.agentVersion === "string"
+          ? { agentVersion: String(asRecord(initialized._meta)?.agentVersion) }
+          : {}),
+        detail: {
+          sessionId: this.sessionId,
+          transport: "acp",
+          protocolVersion: 1,
+          ...(cliVersion ? { cliVersion } : {}),
+          permissionMode: this.config.permissionMode,
+          sandbox: this.config.sandbox,
+          trustProject: this.config.trustProject,
+          reasoningEffort: this.config.reasoningEffort,
+          modelResolution: "reported",
+          nativeConfig: nativeConfig ?? [],
+          usageAvailability: "unknown",
+          isolation: "none",
+        },
+      };
+    } catch (error) {
+      await this.close();
+      throw error;
+    }
+  }
+
+  private request(method: string, params: RpcRecord): Promise<RpcRecord> {
+    if (!this.transport) throw new Error("Grok.build ACP has not started");
+    return this.transport.request(method, params, this.config.startupTimeoutMs);
+  }
+
+  private async run(prompt: string): Promise<void> {
+    const result = await this.transport?.request("session/prompt", {
+      sessionId: this.sessionId,
+      prompt: [{ type: "text", text: prompt }],
+    });
+    await this.transport?.drained();
+    if (this.terminal) return;
+    if (result?.stopReason !== "end_turn")
+      throw new Error(`Grok.build stopped with ${String(result?.stopReason ?? "unknown reason")}`);
+    if (this.interrupts.size > 0)
+      throw new Error("Grok.build ended with pending permission requests");
+    this.terminal = true;
+    clearTimeout(this.idleTimer);
+    await this.emitter.emit("done", {
+      exitReason: "done",
+      summary: {
+        sessionId: this.sessionId,
+        stopReason: result.stopReason,
+        usageAvailability: "unknown",
+      },
+    });
+  }
+
+  private async message(message: RpcRecord): Promise<void> {
+    if (this.terminal) return;
+    const method = String(message.method);
+    const params = asRecord(message.params) ?? {};
+    const id = message.id;
+    if (typeof id === "number" || typeof id === "string") {
+      if (params.sessionId !== this.sessionId) {
+        this.transport?.write({ id, error: { code: -32602, message: "Unknown session" } });
+        return;
+      }
+      if (method !== "session/request_permission") {
+        this.transport?.write({
+          id,
+          error: { code: -32601, message: `Unsupported method: ${method}` },
+        });
+        return;
+      }
+      if (++this.interruptCount > this.config.maxInterrupts) {
+        this.transport?.write({ id, result: { outcome: { outcome: "cancelled" } } });
+        throw new Error("Grok.build exceeded maxInterrupts");
+      }
+      const interruptId = `grok-build-${id}`;
+      if (this.interrupts.has(interruptId))
+        throw new Error("Grok.build reused a pending request ID");
+      const timer = setTimeout(() => {
+        this.transport?.write({ id, result: { outcome: { outcome: "cancelled" } } });
+        this.interrupts.delete(interruptId);
+        this.fail(
+          new Error(
+            `Grok.build permission review timed out after ${this.config.permissionTimeoutMs}ms`,
+          ),
+        );
+      }, this.config.permissionTimeoutMs);
+      this.interrupts.set(interruptId, { id, params, timer });
+      clearTimeout(this.idleTimer);
+      void this.emitter
+        .emit("interrupt", {
+          kind: "approve",
+          id: interruptId,
+          request: `${permissionRequestText(params)}\n\nOperation details:\n${JSON.stringify(params.toolCall)}`,
+          context: params,
+        })
+        .catch((error: unknown) => this.fail(error));
+      return;
+    }
+    if (method === "session/update") {
+      if (params.sessionId !== this.sessionId) return;
+      const update = asRecord(params.update);
+      if (update)
+        for (const event of this.events.update(update)) await this.emitter.emit("progress", event);
+    }
+  }
+
+  async respond(answer: Answer): Promise<void> {
+    if (this.terminal) throw new Error("Grok.build is closed");
+    const pending = this.interrupts.get(answer.id);
+    if (!pending) throw new Error(`Grok.build has no pending interrupt '${answer.id}'`);
+    const result = permissionAnswer(pending.params, answer);
+    this.transport?.write({ id: pending.id, result });
+    if (asRecord(result.outcome)?.outcome === "cancelled") {
+      this.fail(new Error("Grok.build offered no matching one-operation permission option"));
+      return;
+    }
+    clearTimeout(pending.timer);
+    this.interrupts.delete(answer.id);
+    this.bumpIdleTimer();
+  }
+
+  private bumpIdleTimer(): void {
+    clearTimeout(this.idleTimer);
+    if (!this.ready || this.terminal || this.interrupts.size > 0) return;
+    this.idleTimer = setTimeout(
+      () =>
+        this.fail(
+          new Error(`Grok.build emitted no event for ${this.config.idleTimeoutMs}ms`),
+          "idle-timeout",
+        ),
+      this.config.idleTimeoutMs,
+    );
+  }
+
+  private fail(error: unknown, exitReason: "error" | "idle-timeout" = "error"): void {
+    if (this.terminal) return;
+    this.terminal = true;
+    clearTimeout(this.idleTimer);
+    for (const pending of this.interrupts.values()) clearTimeout(pending.timer);
+    if (this.ready)
+      void this.transport?.drained().then(() =>
+        this.emitter.emit("error", {
+          exitReason,
+          reason: error instanceof Error ? error.message : String(error),
+        }),
+      );
+    void this.transport?.close();
+  }
+
+  async quiesce(): Promise<void> {
+    clearTimeout(this.idleTimer);
+    const cancelling = !this.terminal && this.sessionId !== undefined;
+    this.terminal = true;
+    for (const pending of this.interrupts.values()) clearTimeout(pending.timer);
+    if (cancelling) {
+      for (const pending of this.interrupts.values())
+        this.transport?.write({ id: pending.id, result: { outcome: { outcome: "cancelled" } } });
+      this.transport?.write({ method: "session/cancel", params: { sessionId: this.sessionId } });
+      let timer: NodeJS.Timeout | undefined;
+      try {
+        await Promise.race([
+          this.runPromise,
+          new Promise<void>((resolve) => {
+            timer = setTimeout(resolve, 200);
+          }),
+        ]);
+      } finally {
+        clearTimeout(timer);
+      }
+    }
+    for (const pending of this.interrupts.values()) clearTimeout(pending.timer);
+    this.interrupts.clear();
+    await this.transport?.close();
+  }
+
+  async collect(): Promise<Artifact[]> {
+    await this.quiesce();
+    if (!this.cwd) return [];
+    const artifacts = await walkArtifacts(this.cwd);
+    return this.cwd === this.artifactsDir
+      ? artifacts
+      : artifacts.map((artifact) => ({ ...artifact, root: this.cwd as string }));
+  }
+
+  async close(): Promise<void> {
+    await this.quiesce();
+    this.emitter.removeAllListeners();
+  }
+}
+
+export const grokBuildDriver = defineDriver({
+  name: "grok-build",
+  configSchema: GrokBuildDriverConfigSchema,
+  runtime: (config) => ({
+    commands: [config.command],
+    capabilities: {
+      interactive: true,
+      "tool-calls": true,
+      "tool-results": true,
+      artifacts: true,
+      usage: false,
+      cost: false,
+      isolation: false,
+    },
+  }),
+  selectHarness: selectModelAndReasoningEffort,
+  create: (config) => new GrokBuildDriver(config),
+});
+
+export type { GrokBuildDriverConfig };
+export { buildGrokBuildArgs, GrokBuildDriverConfigSchema };

package/src/index.ts

@@ -364,6 +364,7 @@ export {
   codexCatalog,
   copilotCatalog,
   cursorCatalog,
+  grokBuildCatalog,
   harnessCatalogFor,
 } from "./core/harness-catalogs/index.js";
 export {
@@ -528,6 +529,12 @@ export {
   CursorDriverConfigSchema,
   cursorDriver,
 } from "./drivers/cursor/index.js";
+export {
+  buildGrokBuildArgs,
+  type GrokBuildDriverConfig,
+  GrokBuildDriverConfigSchema,
+  grokBuildDriver,
+} from "./drivers/grok-build/index.js";
 export {
   type McpProbeDriverConfig,
   McpProbeDriverConfigSchema,
@@ -539,13 +546,11 @@ export {
   SubprocessDriverConfigSchema,
   subprocessDriver,
 } from "./drivers/subprocess/index.js";
-
 export {
   createReflectiveOptimizer,
   type ReflectiveOptimizerConfig,
   ReflectiveOptimizerConfigSchema,
 } from "./optimization/reflective.js";
-
 // Supported programmatic orchestration.
 export {
   applyCandidate,

package/src/orchestrator/config.ts

@@ -29,6 +29,7 @@ import { claudeCodeDriver } from "../drivers/claude-code/index.js";
 import { codexDriver } from "../drivers/codex/index.js";
 import { copilotDriver } from "../drivers/copilot/index.js";
 import { cursorDriver } from "../drivers/cursor/index.js";
+import { grokBuildDriver } from "../drivers/grok-build/index.js";
 import { mcpProbeDriver } from "../drivers/mcp/index.js";
 import { subprocessDriver } from "../drivers/subprocess/index.js";
 
@@ -188,6 +189,7 @@ const builtinDrivers: DriverPlugin<never>[] = [
   codexDriver,
   cursorDriver,
   copilotDriver,
+  grokBuildDriver,
   antigravityDriver,
   subprocessDriver,
   mcpProbeDriver,