샌드박스에서 Muse Code 실행하기

import { writeFile } from "node:fs/promises";
import { createSandboxClientFromEnv } from "@coreweave/cwsandbox/node";

const repository = process.argv[2];
if (!repository) throw new Error("Pass a repository URL.");
const metaApiKey = process.env.META_API_KEY;
if (!metaApiKey) throw new Error("Set META_API_KEY.");

const client = createSandboxClientFromEnv();
const sandbox = await client.create({
  containerImage: "node:22-bookworm",
  resources: { cpu: "2", memory: "4Gi" },
  maxLifetimeSeconds: 1800,
  waitUntilRunning: false,
  tags: ["muse-code"],
});
try {
  await sandbox.wait();
  const commands = [
    ["bash", "-o", "pipefail", "-ec", "curl -fsSL https://dev.meta.ai/install.sh | bash"],
    ["bash", "-ec", '"$HOME/.local/bin/muse" --version'],
    ["git", "clone", "--", repository, "/workspace/project"],
  ];
  for (const command of commands) {
    const result = await sandbox.commands.run(command, { timeoutMs: 600_000, check: true });
    process.stdout.write(result.stdout);
  }

  const agent = await sandbox.commands.start(
    [
      "bash", "-ec",
      'IFS= read -r META_API_KEY; export META_API_KEY; ' +
        'exec "$HOME/.local/bin/muse" exec --yolo "$1"',
      "bash",
      "Create sparkline.mjs, a dependency-free Node.js CLI that converts its " +
        "numeric arguments to a sparkline using ▁▂▃▄▅▆▇█. Scale from the minimum " +
        "to the maximum and round to the nearest bar index. " +
        "Running node sparkline.mjs 2 4 8 4 2 must print ▁▃█▃▁ followed by " +
        "a newline. Run that command to test it.",
    ],
    {
      cwd: "/workspace/project",
      stdin: true,
      timeoutMs: 600_000,
    },
  );
  // 출력이 없는 요청이 중단되지 않도록 Muse가 종료될 때까지 stdin을 열어 둡니다.
  await agent.stdin.write(metaApiKey + "\n");
  const result = await agent.wait();
  process.stdout.write(result.stdout);
  process.stderr.write(result.stderr);
  if (!result.ok) throw new Error(`Muse Code exited with code ${result.exitCode}.`);

  const chart = await sandbox.commands.run(
    ["node", "sparkline.mjs", "2", "4", "8", "4", "2"],
    { cwd: "/workspace/project", check: true },
  );
  if (chart.stdout !== "▁▃█▃▁\n") {
    throw new Error(`Unexpected sparkline: ${JSON.stringify(chart.stdout)}`);
  }
  process.stdout.write(chart.stdout);
  const source = await sandbox.commands.run(
    ["cat", "/workspace/project/sparkline.mjs"], { check: true },
  );
  await writeFile("sparkline.mjs", source.stdout, "utf8");
} finally {
  await sandbox.stop();
}