openpond 0.2.0 → 0.2.1

package/dist/chunks/chunk-26AMV5ZI.js

@@ -1,634 +0,0 @@
-import { createRequire as __openpondCreateRequire } from "node:module"; var require = __openpondCreateRequire(import.meta.url);
-import {
-  resolveApiBaseUrlOption
-} from "./chunk-SMNFI2FU.js";
-import {
-  optionString,
-  parseBooleanOption,
-  parseIntegerOption,
-  parseNumberOption
-} from "./chunk-MUMPWI4O.js";
-
-// src/cli/training.ts
-import { readFile } from "node:fs/promises";
-import { request as requestHttp } from "node:http";
-import { randomUUID } from "node:crypto";
-import os from "node:os";
-import path from "node:path";
-import { createInterface } from "node:readline/promises";
-var DEFAULT_LOCAL_TRAINING_API_URL = "http://127.0.0.1:17874";
-var TERMINAL_STATUSES = /* @__PURE__ */ new Set([
-  "cancelled",
-  "completed",
-  "failed",
-  "succeeded"
-]);
-async function runTrainingCommand(options, rest, dependencies = {}) {
-  const subcommand = rest[0];
-  const id = rest[1];
-  if (!subcommand || ![
-    "start",
-    "status",
-    "watch",
-    "logs",
-    "cancel",
-    "resume",
-    "artifacts",
-    "benchmark",
-    "create-project"
-  ].includes(
-    subcommand
-  ) || !id) {
-    throw new Error(
-      "usage: training <create-project|start|status|watch|logs|cancel|resume|artifacts|benchmark> <taskset-id|model-run-id|run-id|model-id>"
-    );
-  }
-  if (rest.length > 2) {
-    throw new Error(`training ${subcommand} accepts exactly one identifier`);
-  }
-  const baseUrl = resolveApiBaseUrlOption(options) ?? process.env.OPENPOND_LOCAL_API_URL?.replace(/\/$/, "") ?? DEFAULT_LOCAL_TRAINING_API_URL;
-  const client = new TrainingApiClient({
-    baseUrl,
-    request: dependencies.request ?? await createLocalAuthenticatedRequest(baseUrl)
-  });
-  const json = parseBooleanOption(options.json);
-  if (subcommand === "create-project") {
-    const project = await createManagedModelProject({
-      client,
-      options,
-      tasksetId: id
-    });
-    printResult(project, json);
-    return;
-  }
-  if (subcommand === "benchmark") {
-    await startHarnessRefinerBenchmark({
-      client,
-      modelId: id,
-      options,
-      json,
-      sleep: dependencies.sleep
-    });
-    return;
-  }
-  if (subcommand === "resume") {
-    const run = await client.modelRun(id, "resume", "POST");
-    printResult(run, json);
-    await watchTraining({
-      client,
-      runId: id,
-      json,
-      intervalMs: parseIntegerOption(options.intervalMs, "interval-ms") ?? 2e3,
-      sleep: dependencies.sleep
-    });
-    return;
-  }
-  if (subcommand === "start") {
-    await startTraining({
-      client,
-      dependencies,
-      json,
-      modelRunId: id,
-      options
-    });
-    return;
-  }
-  if (subcommand === "watch") {
-    await watchTraining({
-      client,
-      runId: id,
-      json,
-      intervalMs: parseIntegerOption(options.intervalMs, "interval-ms") ?? 2e3,
-      sleep: dependencies.sleep
-    });
-    return;
-  }
-  const endpoint = subcommand === "status" ? "status" : subcommand === "logs" ? "logs" : subcommand === "artifacts" ? "artifacts" : "cancel";
-  const result = await client.modelRun(
-    id,
-    endpoint,
-    subcommand === "cancel" ? "POST" : "GET"
-  );
-  printResult(result, json);
-}
-async function createLocalAuthenticatedRequest(baseUrl) {
-  const url = new URL(baseUrl);
-  if (url.protocol !== "http:" || !["127.0.0.1", "localhost", "::1"].includes(url.hostname)) {
-    return fetch;
-  }
-  const appHome = process.env.OPENPOND_HOME?.trim() || path.join(os.homedir(), ".openpond", "openpond-app");
-  const token = (await readFile(path.join(appHome, "token"), "utf8")).trim();
-  if (!token) {
-    throw new Error("OpenPond local capability token is empty.");
-  }
-  return (async (resource, init = {}) => {
-    const target = resource instanceof Request ? new URL(resource.url) : new URL(String(resource));
-    const headers = new Headers(init.headers);
-    headers.set("authorization", `Bearer ${token}`);
-    const nodeHeaders = {};
-    headers.forEach((value, name) => {
-      nodeHeaders[name] = value;
-    });
-    return new Promise((resolve, reject) => {
-      let settled = false;
-      const request = requestHttp(
-        target,
-        {
-          method: init.method ?? "GET",
-          headers: nodeHeaders
-        },
-        (response) => {
-          const chunks = [];
-          response.on("data", (chunk) => chunks.push(chunk));
-          response.once("error", rejectOnce);
-          response.once("end", () => {
-            if (settled) return;
-            settled = true;
-            const responseHeaders = new Headers();
-            for (const [name, value] of Object.entries(response.headers)) {
-              if (Array.isArray(value)) {
-                for (const item of value) responseHeaders.append(name, item);
-              } else if (value !== void 0) {
-                responseHeaders.set(name, value);
-              }
-            }
-            resolve(
-              new Response(new Uint8Array(Buffer.concat(chunks)), {
-                status: response.statusCode ?? 500,
-                statusText: response.statusMessage,
-                headers: responseHeaders
-              })
-            );
-          });
-        }
-      );
-      const rejectOnce = (error) => {
-        if (settled) return;
-        settled = true;
-        reject(error);
-      };
-      request.once("error", rejectOnce);
-      if (init.signal) {
-        const abort = () => request.destroy(
-          init.signal?.reason instanceof Error ? init.signal.reason : new Error("Training API request aborted.")
-        );
-        if (init.signal.aborted) abort();
-        else init.signal.addEventListener("abort", abort, { once: true });
-      }
-      if (init.body !== void 0 && init.body !== null) {
-        if (typeof init.body !== "string" && !(init.body instanceof Uint8Array)) {
-          request.destroy(
-            new Error("Local Training API supports string or byte request bodies.")
-          );
-          return;
-        }
-        request.write(init.body);
-      }
-      request.end();
-    });
-  });
-}
-var TrainingApiClient = class {
-  baseUrl;
-  request;
-  constructor(input) {
-    this.baseUrl = input.baseUrl.replace(/\/$/, "");
-    this.request = input.request ?? fetch;
-  }
-  modelRun(id, action, method, body) {
-    return this.json(
-      `/v1/training/model-runs/${encodeURIComponent(id)}/${action}`,
-      method,
-      body
-    );
-  }
-  prepareModelProject(id, body) {
-    return this.json(
-      `/v1/training/model-projects/${encodeURIComponent(id)}/training/prepare`,
-      "POST",
-      body
-    );
-  }
-  startModelProject(id, body) {
-    return this.json(
-      `/v1/training/model-projects/${encodeURIComponent(id)}/training/start`,
-      "POST",
-      body
-    );
-  }
-  trainingState() {
-    return this.json("/v1/training", "GET");
-  }
-  startHarnessRefinerBenchmark(modelId, body) {
-    return this.json(
-      `/v1/training/models/${encodeURIComponent(modelId)}/harness-refiner-benchmark`,
-      "POST",
-      body
-    );
-  }
-  importTasksetDraftPackage(body) {
-    return this.json("/v1/training/taskset-drafts/import", "POST", body);
-  }
-  publishTasksetDraft(draftId, modelId) {
-    return this.json(
-      `/v1/training/taskset-drafts/${encodeURIComponent(draftId)}/publish`,
-      "POST",
-      modelId ? { modelId } : {}
-    );
-  }
-  tasksetReadiness(tasksetId) {
-    return this.json("/v1/training/readiness", "POST", { tasksetId });
-  }
-  deleteTasksetDraft(draftId) {
-    return this.json(
-      `/v1/training/taskset-drafts/${encodeURIComponent(draftId)}`,
-      "DELETE"
-    );
-  }
-  saveModelProject(project) {
-    return this.json("/v1/training/models", "PUT", project);
-  }
-  async json(pathname, method, body) {
-    const response = await this.request(`${this.baseUrl}${pathname}`, {
-      method,
-      headers: body === void 0 ? { Accept: "application/json" } : {
-        Accept: "application/json",
-        "Content-Type": "application/json"
-      },
-      body: body === void 0 ? void 0 : JSON.stringify(body)
-    });
-    const text = await response.text();
-    let payload = null;
-    if (text) {
-      try {
-        payload = JSON.parse(text);
-      } catch {
-        payload = text;
-      }
-    }
-    if (!response.ok) {
-      const message = payload && typeof payload === "object" && "error" in payload && typeof payload.error === "string" ? payload.error : `${response.status} ${response.statusText}`.trim();
-      throw new Error(`Training API request failed: ${message}`);
-    }
-    return payload;
-  }
-};
-async function startHarnessRefinerBenchmark(input) {
-  const state = await input.client.trainingState();
-  const project = modelProject(state, input.modelId);
-  const providerId = optionString(input.options, "provider") || "openpond";
-  const modelId = optionString(input.options, "model") || "openpond-chat";
-  const effort = optionString(input.options, "reasoningEffort") || "high";
-  const maximumSpendUsd = parseNumberOption(input.options.maxSpend, "max-spend") ?? 10;
-  if (maximumSpendUsd <= 0) throw new Error("max-spend must be greater than zero");
-  if (parseBooleanOption(input.options.detach)) {
-    throw new Error("Harness Refiner benchmark runs must be watched to terminal state.");
-  }
-  const run = await input.client.startHarnessRefinerBenchmark(input.modelId, {
-    profileId: project.profileId,
-    model: { providerId, modelId },
-    reasoningEffort: effort === "none" ? "none" : effort,
-    maximumSpendUsd
-  });
-  printResult(run, input.json);
-  const runId = objectString(run, "id");
-  if (!runId) throw new Error("Benchmark start did not return a Model Run id.");
-  await watchTraining({
-    client: input.client,
-    runId,
-    json: input.json,
-    intervalMs: parseIntegerOption(input.options.intervalMs, "interval-ms") ?? 2e3,
-    sleep: input.sleep
-  });
-}
-function modelProject(value, modelId) {
-  if (!value || typeof value !== "object" || !("modelProjects" in value)) {
-    throw new Error("Training state did not include Models.");
-  }
-  const projects = Array.isArray(value.modelProjects) ? value.modelProjects : [];
-  const project = projects.find(
-    (candidate) => candidate && typeof candidate === "object" && "id" in candidate && candidate.id === modelId
-  );
-  const profileId = project && typeof project === "object" ? objectString(project, "profileId") : null;
-  if (!profileId) throw new Error(`Model ${modelId} was not found.`);
-  return { profileId };
-}
-function objectString(value, key) {
-  if (!value || typeof value !== "object" || !(key in value)) return null;
-  const candidate = value[key];
-  return typeof candidate === "string" && candidate ? candidate : null;
-}
-async function createManagedModelProject(input) {
-  const recipePath = optionString(input.options, "recipe");
-  if (!recipePath) {
-    throw new Error("training create-project requires --recipe <versioned-recipe.json>");
-  }
-  const recipe = await readJsonObject(recipePath, "Training recipe");
-  const state = await input.client.trainingState();
-  const taskset = trainingStateRecord(state, "tasksets", input.tasksetId, "Taskset");
-  const profileId = optionString(input.options, "profile") || objectString(taskset, "profileId");
-  if (!profileId) {
-    throw new Error(
-      `The Taskset has no Profile id (received ${typeof taskset.profileId}: ${String(taskset.profileId)}).`
-    );
-  }
-  const tasksetProfileId = objectString(taskset, "profileId");
-  if (tasksetProfileId !== profileId) {
-    throw new Error(
-      `Taskset ${input.tasksetId} belongs to Profile ${tasksetProfileId ?? "unknown"}, not ${profileId}.`
-    );
-  }
-  const projectId = optionString(input.options, "projectId") || `model_${randomUUID()}`;
-  const projects = trainingStateRecords(state, "modelProjects");
-  const existingProject = projects.find(
-    (project) => objectString(project, "id") === projectId
-  );
-  const rolloutPlacement = enumOption(
-    input.options,
-    "rolloutPlacement",
-    ["local", "remote"],
-    "local"
-  );
-  const gpuPlacementObjective = enumOption(
-    input.options,
-    "gpuPlacementObjective",
-    ["fast", "balanced", "economical"],
-    "balanced"
-  );
-  const runPreset = enumOption(
-    input.options,
-    "runPreset",
-    ["small", "standard", "custom", "small_experiment"],
-    "custom"
-  );
-  const maximumSpendUsd = parseNumberOption(input.options.maxSpend, "max-spend") ?? null;
-  if (maximumSpendUsd !== null && maximumSpendUsd < 0) {
-    throw new Error("max-spend must be non-negative");
-  }
-  const retentionDays = parseIntegerOption(
-    input.options.retentionDays,
-    "retention-days"
-  ) ?? null;
-  if (retentionDays !== null && retentionDays < 0) {
-    throw new Error("retention-days must be non-negative");
-  }
-  const timestamp = (/* @__PURE__ */ new Date()).toISOString();
-  const contentHash = objectString(taskset, "contentHash");
-  const revision = objectNumber(taskset, "revision");
-  if (!contentHash || !revision || revision < 1) {
-    throw new Error("The Taskset does not have an immutable published revision.");
-  }
-  const method = objectString(recipe, "method");
-  if (!method) throw new Error("Training recipe must declare its method.");
-  const baseModel = managedBasePreferenceFromRecipe(recipe);
-  if (existingProject) {
-    const existingSetup = objectRecord(existingProject, "trainingSetup");
-    const existingTasksetRef = objectRecord(existingSetup, "tasksetRef");
-    if (objectString(existingTasksetRef, "id") !== input.tasksetId || objectString(existingProject, "profileId") !== profileId) {
-      throw new Error(
-        `Model Project ${projectId} already exists for a different Profile or Taskset.`
-      );
-    }
-    if (objectRecord(existingSetup, "baseModel")) {
-      throw new Error(`Model Project ${projectId} already exists.`);
-    }
-    return input.client.saveModelProject({
-      ...existingProject,
-      revision: objectNumber(existingProject, "revision") ?? 1,
-      defaultBaseModel: baseModel,
-      trainingSetup: {
-        ...existingSetup,
-        baseModel,
-        recipe
-      },
-      updatedAt: timestamp
-    });
-  }
-  return input.client.saveModelProject({
-    schemaVersion: "openpond.modelProject.v2",
-    id: projectId,
-    profileId,
-    revision: 1,
-    name: optionString(input.options, "name") || objectString(taskset, "name") || "Managed Model",
-    objective: optionString(input.options, "objective") || objectString(taskset, "objective"),
-    defaultBaseModel: baseModel,
-    defaultDestinationId: optionString(input.options, "destination") || "openpond_managed",
-    trainingSetup: {
-      tasksetRef: { id: input.tasksetId, revision, contentHash },
-      tasksetRelease: null,
-      harnessRelease: null,
-      baseModel,
-      method,
-      destinationId: optionString(input.options, "destination") || "openpond_managed",
-      managedRolloutPlacement: rolloutPlacement,
-      managedGpuPlacementObjective: gpuPlacementObjective,
-      managedGpuRequirement: "any",
-      runPreset,
-      recipe,
-      preferredMaximumSpendUsd: maximumSpendUsd,
-      preferredRetentionDays: retentionDays
-    },
-    hosted: null,
-    tasksetSyncs: [],
-    createdAt: timestamp,
-    updatedAt: timestamp
-  });
-}
-function managedBasePreferenceFromRecipe(recipe) {
-  const recipeBase = objectRecord(recipe, "baseModel");
-  const modelId = objectString(recipeBase, "id");
-  const revision = objectString(recipeBase, "revision");
-  const tokenizerRevision = objectString(recipeBase, "tokenizerRevision");
-  const chatTemplateHash = objectString(recipeBase, "chatTemplateHash");
-  if (!modelId || !revision || !tokenizerRevision || !chatTemplateHash) {
-    throw new Error(
-      "Managed GRPO recipe must pin baseModel id, revision, tokenizerRevision, and chatTemplateHash."
-    );
-  }
-  return {
-    schemaVersion: "openpond.baseModelPreference.v1",
-    modelId,
-    revision,
-    tokenizerRevision,
-    chatTemplateHash,
-    modelAssetId: null,
-    source: "managed"
-  };
-}
-function trainingStateRecord(state, key, id, label) {
-  const match = trainingStateRecords(state, key).find(
-    (candidate) => objectString(candidate, "id") === id
-  );
-  if (!match) throw new Error(`${label} ${id} was not found.`);
-  return match;
-}
-function trainingStateRecords(state, key) {
-  if (!state || typeof state !== "object" || !(key in state)) return [];
-  const value = state[key];
-  return Array.isArray(value) ? value.filter(
-    (candidate) => Boolean(candidate) && typeof candidate === "object" && !Array.isArray(candidate)
-  ) : [];
-}
-function objectNumber(value, key) {
-  if (!value || typeof value !== "object" || !(key in value)) return null;
-  const candidate = value[key];
-  return typeof candidate === "number" && Number.isFinite(candidate) ? candidate : null;
-}
-function objectRecord(value, key) {
-  if (!value || typeof value !== "object" || !(key in value)) return null;
-  const candidate = value[key];
-  return candidate && typeof candidate === "object" && !Array.isArray(candidate) ? candidate : null;
-}
-function enumOption(options, key, values, fallback) {
-  const value = optionString(options, key) || fallback;
-  if (!values.includes(value)) {
-    throw new Error(`${key} must be one of ${values.join(", ")}`);
-  }
-  return value;
-}
-async function readJsonObject(file, label) {
-  let parsed;
-  try {
-    parsed = JSON.parse(await readFile(file, "utf8"));
-  } catch (error) {
-    throw new Error(
-      `Could not read ${label} ${file}: ${error instanceof Error ? error.message : String(error)}`
-    );
-  }
-  if (!parsed || typeof parsed !== "object" || Array.isArray(parsed)) {
-    throw new Error(`${label} must be a JSON object.`);
-  }
-  return parsed;
-}
-async function startTraining(input) {
-  const maximumSpendUsd = parseNumberOption(
-    input.options.maxSpend,
-    "max-spend"
-  ) ?? null;
-  if (maximumSpendUsd !== null && maximumSpendUsd < 0) {
-    throw new Error("max-spend must be non-negative");
-  }
-  const retentionDays = parseIntegerOption(input.options.retentionDays, "retention-days") ?? null;
-  if (retentionDays !== null && retentionDays <= 0) {
-    throw new Error("retention-days must be a positive integer");
-  }
-  const manifestPath = optionString(input.options, "manifest");
-  const manifest = manifestPath ? await readManifest(manifestPath) : void 0;
-  const comparisonSeriesEntryId = optionString(
-    input.options,
-    "comparisonSeriesEntry"
-  ) || null;
-  const controls = { maximumSpendUsd, retentionDays };
-  const preparation = await input.client.prepareModelProject(
-    input.modelRunId,
-    controls
-  );
-  if (!input.json) {
-    console.log("Preparation:");
-    console.log(JSON.stringify(preparation, null, 2));
-  }
-  const confirmed = parseBooleanOption(input.options.yes) || await confirmStart(input.dependencies.confirm);
-  if (!confirmed) {
-    throw new Error("Training start cancelled.");
-  }
-  const result = await input.client.startModelProject(
-    input.modelRunId,
-    {
-      ...controls,
-      exportApproved: parseBooleanOption(input.options.exportApproved),
-      manifest,
-      comparisonSeriesEntryId
-    }
-  );
-  printResult(result, input.json);
-  if (parseBooleanOption(input.options.detach)) return;
-  const runId = readRunId(result) ?? input.modelRunId;
-  await watchTraining({
-    client: input.client,
-    runId,
-    json: input.json,
-    intervalMs: 2e3,
-    sleep: input.dependencies.sleep
-  });
-}
-async function watchTraining(input) {
-  if (input.intervalMs < 250) {
-    throw new Error("interval-ms must be at least 250");
-  }
-  const sleep = input.sleep ?? ((milliseconds) => new Promise((resolve) => setTimeout(resolve, milliseconds)));
-  let previous = "";
-  for (; ; ) {
-    const status = await input.client.modelRun(
-      input.runId,
-      "status",
-      "GET"
-    );
-    const serialized = JSON.stringify(status);
-    if (input.json || serialized !== previous) {
-      printResult(status, input.json);
-      previous = serialized;
-    }
-    if (TERMINAL_STATUSES.has(readStatus(status))) return;
-    await sleep(input.intervalMs);
-  }
-}
-async function readManifest(path2) {
-  let parsed;
-  try {
-    parsed = JSON.parse(await readFile(path2, "utf8"));
-  } catch (error) {
-    throw new Error(
-      `Could not read Harness Run Manifest ${path2}: ${error instanceof Error ? error.message : String(error)}`
-    );
-  }
-  if (!parsed || typeof parsed !== "object" || Array.isArray(parsed)) {
-    throw new Error("Harness Run Manifest must be a JSON object.");
-  }
-  return parsed;
-}
-async function confirmStart(injected) {
-  const message = "Start this exact prepared training run? Downloads, provisioning, or spend may begin.";
-  if (injected) return injected(message);
-  if (!process.stdin.isTTY || !process.stdout.isTTY) {
-    throw new Error(
-      "training start requires --yes when standard input is not interactive"
-    );
-  }
-  const prompt = createInterface({
-    input: process.stdin,
-    output: process.stdout
-  });
-  try {
-    const answer = await prompt.question(`${message} [y/N] `);
-    return answer.trim().toLowerCase() === "y" || answer.trim().toLowerCase() === "yes";
-  } finally {
-    prompt.close();
-  }
-}
-function readRunId(value) {
-  if (!value || typeof value !== "object") return null;
-  const job = "job" in value ? value.job : null;
-  if (!job || typeof job !== "object" || !("id" in job)) return null;
-  return typeof job.id === "string" ? job.id : null;
-}
-function readStatus(value) {
-  if (!value || typeof value !== "object") return "";
-  if ("state" in value && typeof value.state === "string") {
-    return value.state;
-  }
-  return "status" in value && typeof value.status === "string" ? value.status : "";
-}
-function printResult(value, json) {
-  if (json) {
-    console.log(JSON.stringify(value));
-    return;
-  }
-  console.log(JSON.stringify(value, null, 2));
-}
-
-export {
-  DEFAULT_LOCAL_TRAINING_API_URL,
-  runTrainingCommand,
-  createLocalAuthenticatedRequest,
-  TrainingApiClient
-};