trajectories-sh 1.23.1 → 1.24.1

package/dist/{auth-4UEZSE4C.js → auth-NNBHKNIF.js}

@@ -5,7 +5,7 @@ import {
   isLoggedIn,
   saveApiKey,
   saveSession
-} from "./chunk-Q2ZEN43L.js";
+} from "./chunk-ERO5HP43.js";
 
 // src/auth.ts
 import { createServer } from "http";

package/dist/{chunk-35B24E3O.js → chunk-72L6XPK6.js}

@@ -149,6 +149,7 @@ function unpackScreenshotChunks(jobDir) {
 
 export {
   CHUNK_DIR,
+  mimeFor,
   buildScreenshotChunks,
   unpackScreenshotChunks
 };

package/dist/{chunk-Q2ZEN43L.js → chunk-ERO5HP43.js}

@@ -29,7 +29,7 @@ function getAccessToken() {
   return loadConfig().access_token;
 }
 async function getAuthHeader() {
-  const envKey = process.env.TRAJECTORIES_API_KEY;
+  const envKey = process.env.TRAJECTORIES_API_KEY || process.env.TRAJECTORIES_TOKEN;
   if (envKey && envKey.trim()) return `Bearer ${envKey.trim()}`;
   const config = loadConfig();
   if (config.api_key) return `Bearer ${config.api_key}`;

package/dist/{chunk-2ELJH3IV.js → chunk-HHUWMQ5S.js}

@@ -1,7 +1,7 @@
 import {
   getApiUrl,
   getAuthHeader
-} from "./chunk-Q2ZEN43L.js";
+} from "./chunk-ERO5HP43.js";
 
 // src/quota.ts
 async function fetchOrgUsage() {

package/dist/chunk-LOJZPDGI.js

@@ -0,0 +1,242 @@
+import {
+  CHUNK_DIR,
+  mimeFor,
+  unpackScreenshotChunks
+} from "./chunk-72L6XPK6.js";
+import {
+  getApiUrl,
+  getAuthHeader
+} from "./chunk-ERO5HP43.js";
+
+// src/download.ts
+import { existsSync, mkdirSync, readdirSync, readFileSync, statSync, writeFileSync } from "fs";
+import { join, dirname } from "path";
+var UUID_RE = /[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}/i;
+var BATCH_SIZE = 100;
+var CONCURRENCY = 25;
+var DOWNLOAD_TIMEOUT_MS = 18e4;
+var EXIT = {
+  OK: 0,
+  ERROR: 1,
+  USAGE: 2,
+  AUTH_REQUIRED: 3,
+  NOT_FOUND: 4
+};
+var CliError = class extends Error {
+  constructor(message, exitCode = EXIT.ERROR) {
+    super(message);
+    this.exitCode = exitCode;
+  }
+  exitCode;
+};
+async function httpError(resp, what) {
+  let detail = "";
+  try {
+    const body = await resp.json();
+    detail = typeof body.detail === "string" ? body.detail : "";
+    if (resp.status === 401 || resp.status === 403) {
+      return new CliError(
+        `${what}: authentication required (${resp.status}). ${body.how_to_authenticate?.cli ?? "Set TRAJECTORIES_API_KEY in the environment."}`,
+        EXIT.AUTH_REQUIRED
+      );
+    }
+    if (resp.status === 404) {
+      return new CliError(
+        `${what}: not found (404)${detail ? `: ${detail}` : ""}. If it is private, set TRAJECTORIES_API_KEY for an org member's key.`,
+        EXIT.NOT_FOUND
+      );
+    }
+  } catch {
+  }
+  return new CliError(`${what}: HTTP ${resp.status}${detail ? ` ${detail}` : ""}`, EXIT.ERROR);
+}
+function parseJobId(input) {
+  const trimmed = input.trim();
+  if (UUID_RE.test(trimmed) && !trimmed.includes("/")) {
+    return trimmed.match(UUID_RE)[0];
+  }
+  const shortMatch = trimmed.match(/\/t\/([0-9a-f-]{36})/i);
+  if (shortMatch) return shortMatch[1];
+  const trajMatch = trimmed.match(/\/trajectories\/([0-9a-f-]{36})/i);
+  if (trajMatch) return trajMatch[1];
+  const trajViewerMatch = trimmed.match(/\/traj\/([0-9a-f-]{36})\/([^/]+)/i);
+  if (trajViewerMatch) {
+    return `org:${trajViewerMatch[1]}:${trajViewerMatch[2]}`;
+  }
+  const anyUuid = trimmed.match(UUID_RE);
+  if (anyUuid) return anyUuid[0];
+  throw new CliError(`Could not find a trajectory job ID in: ${input}`, EXIT.USAGE);
+}
+function parseProjectId(input) {
+  const trimmed = input.trim();
+  const m = trimmed.match(/\/projects\/([0-9a-f-]{36})/i) ?? (trimmed.includes("/") ? null : trimmed.match(UUID_RE));
+  if (!m) throw new CliError(`Could not find a project ID in: ${input}`, EXIT.USAGE);
+  return m[1] ?? m[0];
+}
+function isScreenshot(path) {
+  if (path.includes(`/${CHUNK_DIR}/`)) return true;
+  const parts = path.split("/");
+  return parts.length >= 3 && parts[1] === "agent" && mimeFor(parts[parts.length - 1]) != null;
+}
+async function planJob(input, options) {
+  const apiUrl = getApiUrl();
+  const auth = await getAuthHeader();
+  const headers = {};
+  if (auth) headers["Authorization"] = auth;
+  let jobId = parseJobId(input);
+  if (jobId.startsWith("org:")) {
+    const [, orgId, slug2] = jobId.split(":");
+    const configResp = await fetch(`${apiUrl}/api/viewer/${orgId}/${slug2}/api/jobs`, {
+      headers,
+      signal: AbortSignal.timeout(DOWNLOAD_TIMEOUT_MS)
+    });
+    if (!configResp.ok) throw await httpError(configResp, `Resolve job ${orgId}/${slug2}`);
+    const items = (await configResp.json()).items || [];
+    if (!items[0]?.id) throw new CliError(`Could not resolve job ID for ${orgId}/${slug2}`, EXIT.NOT_FOUND);
+    jobId = items[0].id;
+  }
+  const listResp = await fetch(`${apiUrl}/api/cli/download/${jobId}/files`, {
+    headers,
+    signal: AbortSignal.timeout(DOWNLOAD_TIMEOUT_MS)
+  });
+  if (!listResp.ok) throw await httpError(listResp, `List files for ${jobId}`);
+  const { slug, files: all } = await listResp.json();
+  const files = options.includeScreenshots ? all : all.filter((f) => !isScreenshot(f.path));
+  const trials = new Set(files.map((f) => f.path.split("/")[0]).filter((p) => p && !p.endsWith(".json") && !p.endsWith(".log")));
+  return {
+    jobId,
+    slug,
+    files,
+    bytes: files.reduce((s, f) => s + f.size, 0),
+    trials: trials.size,
+    trajectoryFiles: files.filter((f) => f.path.endsWith("/agent/trajectory.json")).length,
+    screenshotsSkipped: all.length - files.length
+  };
+}
+async function downloadPlannedJob(plan, outputDir, options) {
+  const apiUrl = getApiUrl();
+  const auth = await getAuthHeader();
+  const headers = {};
+  if (auth) headers["Authorization"] = auth;
+  const { jobId, files } = plan;
+  const log = options.quiet ? () => {
+  } : (m) => console.log(m);
+  const packedTrials = new Set(files.filter((f) => f.path.includes(`/${CHUNK_DIR}/`)).map((f) => f.path.split("/")[0]));
+  log(`  Job: ${plan.slug} (${jobId})`);
+  log(
+    `  ${files.length} files, ${(plan.bytes / 1024 / 1024).toFixed(1)} MB` + (packedTrials.size > 0 ? ` (screenshots of ${packedTrials.size} trial${packedTrials.size === 1 ? "" : "s"} packed)` : "") + (plan.screenshotsSkipped > 0 ? ` (${plan.screenshotsSkipped} screenshot files skipped; --include-screenshots to fetch)` : "")
+  );
+  mkdirSync(outputDir, { recursive: true });
+  const bar = options.quiet ? { tick: (_n = 0) => {
+  }, done: () => {
+  } } : new (await import("./progress-H7WNKHDR.js")).ProgressBar({ total: files.length, totalBytes: plan.bytes, label: "\u2193" });
+  let downloaded = 0;
+  let downloadedBytes = 0;
+  const failed = [];
+  for (let batchStart = 0; batchStart < files.length; batchStart += BATCH_SIZE) {
+    const batch = files.slice(batchStart, batchStart + BATCH_SIZE);
+    const urlResp = await fetch(`${apiUrl}/api/cli/download/${jobId}/signed-urls`, {
+      method: "POST",
+      headers: { ...headers, "Content-Type": "application/json" },
+      body: JSON.stringify({ paths: batch.map((f) => f.path) }),
+      signal: AbortSignal.timeout(DOWNLOAD_TIMEOUT_MS)
+    });
+    const urlMap = /* @__PURE__ */ new Map();
+    if (urlResp.ok) {
+      const { urls } = await urlResp.json();
+      for (const u of urls) if (u.signed_url) urlMap.set(u.path, u.signed_url);
+    }
+    let next = 0;
+    const worker = async () => {
+      while (next < batch.length) {
+        const file = batch[next++];
+        const signedUrl = urlMap.get(file.path);
+        const downloadUrl = signedUrl ?? `${apiUrl}/api/cli/download/${jobId}/file?path=${encodeURIComponent(file.path)}`;
+        const downloadHeaders = signedUrl ? {} : headers;
+        let ok = false;
+        for (let attempt = 0; attempt < 3 && !ok; attempt++) {
+          try {
+            const resp = await fetch(downloadUrl, { headers: downloadHeaders, signal: AbortSignal.timeout(DOWNLOAD_TIMEOUT_MS) });
+            if (!resp.ok) throw new Error(`HTTP ${resp.status}`);
+            const data = Buffer.from(await resp.arrayBuffer());
+            const outPath = join(outputDir, file.path);
+            mkdirSync(dirname(outPath), { recursive: true });
+            writeFileSync(outPath, data);
+            downloaded++;
+            downloadedBytes += data.length;
+            bar.tick(data.length);
+            ok = true;
+          } catch {
+            if (attempt < 2) await new Promise((r) => setTimeout(r, 1e3 * (attempt + 1)));
+          }
+        }
+        if (!ok) {
+          failed.push(file.path);
+          bar.tick(0);
+        }
+      }
+    };
+    await Promise.all(Array.from({ length: Math.min(CONCURRENCY, batch.length) }, worker));
+  }
+  bar.done();
+  let screenshotsUnpacked = 0;
+  if (packedTrials.size > 0) {
+    const unpacked = unpackScreenshotChunks(outputDir);
+    screenshotsUnpacked = unpacked.imagesWritten;
+    if (unpacked.imagesWritten > 0) {
+      log(`  \u2713 Unpacked ${unpacked.imagesWritten} screenshots across ${unpacked.trialsUnpacked} trial${unpacked.trialsUnpacked === 1 ? "" : "s"}`);
+    }
+  }
+  return { ...plan, outputDir, downloaded, downloadedBytes, failed, screenshotsUnpacked };
+}
+async function downloadJob(input, outputDir, options = { includeScreenshots: true, quiet: false }) {
+  const plan = await planJob(input, options);
+  return downloadPlannedJob(plan, outputDir, options);
+}
+function writeJsonl(rootDir, jobDirs, outFile) {
+  const lines = [];
+  for (const job of jobDirs) {
+    if (!existsSync(job.dir)) continue;
+    for (const trial of readdirSync(job.dir, { withFileTypes: true })) {
+      if (!trial.isDirectory()) continue;
+      const trajPath = join(job.dir, trial.name, "agent", "trajectory.json");
+      if (!existsSync(trajPath)) continue;
+      let result = null;
+      const resultPath = join(job.dir, trial.name, "result.json");
+      if (existsSync(resultPath)) {
+        try {
+          result = JSON.parse(readFileSync(resultPath, "utf-8"));
+        } catch {
+        }
+      }
+      const trajectory = JSON.parse(readFileSync(trajPath, "utf-8"));
+      lines.push(JSON.stringify({ job_id: job.jobId, job_slug: job.slug, trial_name: trial.name, result, trajectory }));
+    }
+  }
+  mkdirSync(dirname(outFile), { recursive: true });
+  writeFileSync(outFile, lines.map((l) => l + "\n").join(""));
+  return lines.length;
+}
+function dirBytes(dir) {
+  if (!existsSync(dir)) return 0;
+  let total = 0;
+  for (const e of readdirSync(dir, { withFileTypes: true })) {
+    const p = join(dir, e.name);
+    total += e.isDirectory() ? dirBytes(p) : statSync(p).size;
+  }
+  return total;
+}
+
+export {
+  DOWNLOAD_TIMEOUT_MS,
+  EXIT,
+  CliError,
+  httpError,
+  parseJobId,
+  parseProjectId,
+  planJob,
+  downloadPlannedJob,
+  downloadJob,
+  writeJsonl,
+  dirBytes
+};