@tuned-tensor/cli 0.19.0 → 0.21.0

package/dist/{chunk-4Q54NHKR.js → chunk-PVLX7LJI.js}

@@ -615,23 +615,52 @@ var TRAINING_MODELS = [
     gradientCheckpointing: true
   }
 ];
-function resolveTrainingModel(modelId) {
+var HUGGING_FACE_REPO_ID = /^[A-Za-z0-9][A-Za-z0-9._-]{0,95}\/[A-Za-z0-9][A-Za-z0-9._-]{0,95}$/;
+function uncertifiedTrainingModel(id) {
+  return {
+    id,
+    family: "uncertified",
+    modelLoader: "causal_lm",
+    defaultLearningRate: 1e-4,
+    defaultPerDeviceBatchSize: 1,
+    defaultGradientAccumulationSteps: 8,
+    defaultLoraRank: 16,
+    defaultLoraAlpha: 32,
+    defaultLoraDropout: 0.05,
+    defaultMaxSeqLength: 2048,
+    loraTargetModules: "all-linear",
+    gradientCheckpointing: true
+  };
+}
+function findCertifiedTrainingModel(modelId) {
   const normalized = modelId.trim().toLowerCase();
-  const model = TRAINING_MODELS.find(
-    (candidate) => candidate.id.toLowerCase() === normalized
+  return TRAINING_MODELS.find((candidate) => candidate.id.toLowerCase() === normalized);
+}
+function isCertifiedTrainingModel(modelId) {
+  return findCertifiedTrainingModel(modelId) !== void 0;
+}
+function resolveTrainingModel(modelId) {
+  const model = findCertifiedTrainingModel(modelId);
+  if (model) return model;
+  const id = modelId.trim();
+  if (HUGGING_FACE_REPO_ID.test(id)) return uncertifiedTrainingModel(id);
+  throw new Error(
+    `Unsupported base model "${modelId}". Use a certified model (${TRAINING_MODELS.map((item) => item.id).join(", ")}) or a Hugging Face repo id such as owner/name.`
+  );
+}
+function uncertifiedRevisionError(modelId) {
+  return new Error(
+    `${modelId} is not a TT-certified base model. To fine-tune it anyway, set hyperparameters.base_model_revision to its 40-character commit SHA (see https://huggingface.co/${modelId}/commits/main). Uncertified models load as text causal LMs with safetensors weights and no remote code.`
   );
-  if (!model) {
-    throw new Error(
-      `Unsupported base model "${modelId}". Supported models: ${TRAINING_MODELS.map((item) => item.id).join(", ")}`
-    );
-  }
-  return model;
 }
 function canonicalizeTrainingModel(modelId) {
   return resolveTrainingModel(modelId).id;
 }
 function resolveRequestedBaseModelRevision(modelId, requestedRevision) {
   const model = resolveTrainingModel(modelId);
+  if (!isCertifiedTrainingModel(model.id) && !requestedRevision) {
+    throw uncertifiedRevisionError(model.id);
+  }
   if (model.defaultRevision && requestedRevision && requestedRevision !== model.defaultRevision) {
     throw new Error(
       `${model.id} must use certified revision ${model.defaultRevision}; got ${requestedRevision}.`
@@ -640,7 +669,7 @@ function resolveRequestedBaseModelRevision(modelId, requestedRevision) {
   return requestedRevision ?? model.defaultRevision;
 }
 function defaultBaseModelRevision(modelId) {
-  return resolveRequestedBaseModelRevision(modelId);
+  return resolveTrainingModel(modelId).defaultRevision;
 }
 function resolveModelLoader(modelId) {
   return resolveTrainingModel(modelId).modelLoader;
@@ -703,6 +732,7 @@ function assertCertifiedBaseModelConfig(value, label = "base-model config.json",
     throw new Error(`${label} must contain a JSON object.`);
   }
   const config = value;
+  if (expectedModelId && !isCertifiedTrainingModel(expectedModelId)) return;
   const expectedFamily = expectedModelId ? resolveTrainingModel(expectedModelId).family : void 0;
   if (config.model_type === "qwen3_5") {
     if (expectedFamily && expectedFamily !== "qwen3_5") {
@@ -846,6 +876,7 @@ var TUNED_TENSOR_DIR = ".tuned-tensor";
 var DEFAULT_ARTIFACT_ROOT = `${TUNED_TENSOR_DIR}/artifacts`;
 var DEFAULT_PROJECT_STORE_ROOT = `${TUNED_TENSOR_DIR}/store`;
 var DEFAULT_FOUNDATION_RUNS_DIR = `${TUNED_TENSOR_DIR}/foundation-runs`;
+var DEFAULT_DECISION_RUNS_DIR = `${TUNED_TENSOR_DIR}/decision-runs`;
 function envHome(env = process.env) {
   const home = env.HOME?.trim();
   return home ? resolve3(home) : homedir();
@@ -902,6 +933,38 @@ function pythonEnvironmentPath(kind, hash, env = process.env) {
   return next;
 }
 
+// src/local-runtime/decision-models.ts
+var LAYA_REVISION = "55cf4c4ebb4ebe31b2550e8bdf3bd21b99753851";
+var DECISION_MODELS = [
+  {
+    id: "convaiinnovations/laya",
+    family: "laya",
+    revision: LAYA_REVISION,
+    parameters: "421M (ModernBERT-large encoder + typed decision head)",
+    license: "Apache-2.0",
+    defaultLearningRate: 2e-5,
+    defaultBatchSize: 8,
+    defaultEpochs: 3
+  }
+];
+var DEFAULT_DECISION_MODEL = DECISION_MODELS[0].id;
+function resolveDecisionModel(modelId) {
+  const normalized = modelId.trim().toLowerCase();
+  const model = DECISION_MODELS.find((candidate) => candidate.id.toLowerCase() === normalized);
+  if (!model) {
+    throw new Error(
+      `Unsupported decision model "${modelId}". Supported decision models: ${DECISION_MODELS.map((item) => item.id).join(", ")}`
+    );
+  }
+  return model;
+}
+function isDecisionModel(modelId) {
+  return DECISION_MODELS.some((candidate) => candidate.id.toLowerCase() === modelId.trim().toLowerCase());
+}
+function canonicalizeDecisionModel(modelId) {
+  return resolveDecisionModel(modelId).id;
+}
+
 // src/local-runtime/contracts.ts
 var datasetFormatSchema = z.literal("chat_jsonl");
 var baseModelRevisionSchema = z.string().regex(/^[0-9a-f]{40}$/i, "base_model_revision must be a 40-character Hugging Face commit SHA").transform((value) => value.toLowerCase());
@@ -974,6 +1037,15 @@ var fineTuneHyperparametersSchema = z.object({
   /** Immutable Hugging Face commit used by every stage of the run. */
   base_model_revision: baseModelRevisionSchema.optional()
 }).strict();
+function requirePinnedUncertifiedModel(spec, context) {
+  const baseModel = "spec_snapshot" in spec ? spec.spec_snapshot.base_model : spec.base_model;
+  if (isCertifiedTrainingModel(baseModel) || spec.hyperparameters?.base_model_revision) return;
+  context.addIssue({
+    code: z.ZodIssueCode.custom,
+    path: ["hyperparameters", "base_model_revision"],
+    message: uncertifiedRevisionError(baseModel).message
+  });
+}
 var fineTuneRunRequestSchema = z.object({
   run_id: z.string().uuid(),
   user_id: z.string().min(1),
@@ -983,6 +1055,7 @@ var fineTuneRunRequestSchema = z.object({
   hyperparameters: fineTuneHyperparametersSchema.default({ n_epochs: 1 }),
   dataset_prebuilt: datasetPrebuiltSchema.optional()
 }).strict().superRefine((request, context) => {
+  requirePinnedUncertifiedModel(request, context);
   if (request.spec_snapshot.examples.length === 0 && !request.dataset_prebuilt) {
     context.addIssue({
       code: z.ZodIssueCode.custom,
@@ -1205,9 +1278,54 @@ var localAdapterSpecFileSchema = specSnapshotSchema.extend({
   id: z.string().uuid().optional(),
   hyperparameters: fineTuneHyperparametersSchema.optional(),
   dataset_prebuilt: datasetPrebuiltSchema.optional()
+}).strict().superRefine(requirePinnedUncertifiedModel);
+var decisionQuestionSchema = z.discriminatedUnion("type", [
+  z.object({
+    type: z.literal("choice"),
+    /** Answer label -> description the model reads. Key order is option order. */
+    criteria: z.record(z.string().min(1), z.string()).refine(
+      (criteria) => Object.keys(criteria).length >= 2 && Object.keys(criteria).length <= 32,
+      "A choice decision needs 2 to 32 labels."
+    )
+  }).strict(),
+  z.object({
+    type: z.literal("noul"),
+    criteria: z.object({ true: z.string().optional(), false: z.string().optional() }).strict().optional()
+  }).strict(),
+  z.object({
+    type: z.literal("score"),
+    /** Level descriptions, index 0 first; example outputs are level indices. */
+    criteria: z.array(z.string().min(1)).min(2).max(10)
+  }).strict()
+]);
+var decisionHyperparametersSchema = z.object({
+  n_epochs: z.number().int().min(1).max(50).optional(),
+  learning_rate: z.number().positive().max(1).optional(),
+  batch_size: z.number().int().min(1).max(256).optional(),
+  /** Train only the decision head; faster on CPU, usually less accurate. */
+  freeze_encoder: z.boolean().optional(),
+  /** Shuffle option order per training row so the head learns meaning, not position. */
+  shuffle_options: z.boolean().optional(),
+  seed: z.number().int().min(0).max(2147483647).optional(),
+  device: z.enum(["auto", "cpu", "cuda", "mps"]).optional()
 }).strict();
+var localDecisionSpecFileSchema = behaviorSpecSchema.extend({
+  ...projectWorkflowFields,
+  engine: z.literal("decision"),
+  id: z.string().uuid().optional(),
+  // Refine first: a union member must report an issue, not throw, for other engines' specs.
+  base_model: z.string().refine(isDecisionModel, `Unsupported decision model. Supported decision models: ${DECISION_MODELS.map((model) => model.id).join(", ")}`).transform((value) => canonicalizeDecisionModel(value)),
+  decision: decisionQuestionSchema,
+  hyperparameters: decisionHyperparametersSchema.optional()
+});
+function decisionLabels(question) {
+  if (question.type === "choice") return Object.keys(question.criteria);
+  if (question.type === "noul") return ["false", "true"];
+  return question.criteria.map((_, index) => String(index));
+}
 var localBehaviorSpecFileSchema = z.union([
   localFoundationSpecFileSchema,
+  localDecisionSpecFileSchema,
   localAdapterSpecFileSchema
 ]).superRefine((spec, ctx) => {
   if (spec.pipeline && isFoundationPipeline(parsePipeline(spec.pipeline)) !== (spec.engine === "foundation")) {
@@ -1216,14 +1334,142 @@ var localBehaviorSpecFileSchema = z.union([
   if (spec.engine === "foundation" && (spec.evaluation || spec.runtime?.storeRoot || spec.runtime?.paths)) {
     ctx.addIssue({ code: "custom", path: ["runtime"], message: "Foundation uses foundation settings for evaluation and checkpoints; runtime.artifactRoot and runtime.gpu are supported." });
   }
+  if (spec.engine === "decision" && (spec.evaluation || spec.runtime?.gpu || spec.runtime?.storeRoot || spec.runtime?.paths)) {
+    ctx.addIssue({ code: "custom", path: ["runtime"], message: "Decision specs support runtime.artifactRoot only; the device is set in hyperparameters.device." });
+  }
 });
 function isFoundationSpecFile(spec) {
   return spec.engine === "foundation";
 }
+function isDecisionSpecFile(spec) {
+  return spec.engine === "decision";
+}
+
+// src/local-runtime/dataset.ts
+import { readFile as readFile2 } from "fs/promises";
+async function loadNormalizedChatJsonl(path) {
+  const text = await readFile2(path, "utf8");
+  const rows = [];
+  const jsonLines = [];
+  for (const [index, line] of text.split(/\r?\n/).entries()) {
+    if (!line.trim()) continue;
+    let value;
+    try {
+      value = JSON.parse(line);
+    } catch (error) {
+      throw new Error(`Invalid chat JSONL row ${index + 1}: malformed JSON`, { cause: error });
+    }
+    if (!value || typeof value !== "object" || Array.isArray(value)) {
+      throw new Error(`Invalid chat JSONL row ${index + 1}: expected an object`);
+    }
+    const messages = value.messages;
+    if (!Array.isArray(messages) || ![2, 3].includes(messages.length) || messages.some(
+      (message) => !message || typeof message !== "object" || Array.isArray(message) || !["system", "user", "assistant"].includes(String(message.role)) || typeof message.content !== "string"
+    )) {
+      throw new Error(
+        `Invalid chat JSONL row ${index + 1}: expected an optional system message, one user message, and one assistant answer`
+      );
+    }
+    const row = { messages };
+    const offset = row.messages.length === 3 ? 1 : 0;
+    if (offset === 1 && row.messages[0]?.role !== "system" || row.messages[offset]?.role !== "user" || !row.messages[offset]?.content.trim() || row.messages[offset + 1]?.role !== "assistant" || !row.messages[offset + 1]?.content.trim()) {
+      throw new Error(
+        `Invalid chat JSONL row ${index + 1}: expected an optional system message, one non-empty user message, and one non-empty assistant answer`
+      );
+    }
+    rows.push(row);
+    jsonLines.push(JSON.stringify(row));
+  }
+  if (rows.length === 0) throw new Error(`Chat JSONL contains no examples: ${path}`);
+  return {
+    rows,
+    jsonl: jsonLines.join("\n")
+  };
+}
+async function normalizeChatJsonlForRelocation(path) {
+  return (await loadNormalizedChatJsonl(path)).jsonl;
+}
+function buildSystemMessage(spec) {
+  const parts = [];
+  const guidelines = spec.guidelines.map((value) => value.trim()).filter(Boolean);
+  const constraints = spec.constraints.map((value) => value.trim()).filter(Boolean);
+  if (spec.system_prompt.trim()) parts.push(spec.system_prompt.trim());
+  if (guidelines.length > 0) {
+    parts.push(`Guidelines:
+${guidelines.map((guideline) => `- ${guideline}`).join("\n")}`);
+  }
+  if (constraints.length > 0) {
+    parts.push(`Constraints:
+${constraints.map((constraint) => `- ${constraint}`).join("\n")}`);
+  }
+  return parts.join("\n\n") || "Follow the demonstrated behavior.";
+}
+function exampleToChatRow(spec, example) {
+  const system = buildSystemMessage(spec);
+  return {
+    messages: [
+      { role: "system", content: system },
+      { role: "user", content: example.input },
+      { role: "assistant", content: example.output }
+    ]
+  };
+}
+function compileSpecToJsonl(spec) {
+  return spec.examples.map((example) => JSON.stringify(exampleToChatRow(spec, example))).join("\n");
+}
+function examplesFromSpec(spec) {
+  return spec.examples;
+}
+async function examplesFromChatJsonl(path) {
+  const normalized = await loadNormalizedChatJsonl(path);
+  const examples = [];
+  for (const [index, row] of normalized.rows.entries()) {
+    const user = row.messages.at(-2);
+    const assistant = row.messages.at(-1);
+    examples.push({
+      input: user.content,
+      output: assistant.content
+    });
+  }
+  return examples;
+}
+async function evaluationSuiteFromChatJsonl(path, configuredSystem) {
+  const normalized = await loadNormalizedChatJsonl(path);
+  const inputIdentities = /* @__PURE__ */ new Set();
+  for (const row of normalized.rows) {
+    const input = row.messages.at(-2).content;
+    const identity = input.trim().replace(/\s+/g, " ").toLowerCase();
+    if (inputIdentities.has(identity)) {
+      throw new Error(
+        "General regression data contains duplicate inputs; evaluation prompts must be unique."
+      );
+    }
+    inputIdentities.add(identity);
+  }
+  const systems = new Set(
+    normalized.rows.map((row) => row.messages.find((message) => message.role === "system")?.content).filter((value) => value !== void 0)
+  );
+  if (systems.size > 1) {
+    throw new Error("General regression rows must use one consistent system message.");
+  }
+  const embeddedSystem = [...systems][0];
+  if (configuredSystem !== void 0 && embeddedSystem !== void 0 && configuredSystem !== embeddedSystem) {
+    throw new Error(
+      "evaluation.generalRegression.systemPrompt must match the system message embedded in the dataset."
+    );
+  }
+  return {
+    examples: normalized.rows.map((row) => ({
+      input: row.messages.at(-2).content,
+      output: row.messages.at(-1).content
+    })),
+    system: configuredSystem ?? embeddedSystem ?? "You are a helpful assistant."
+  };
+}
 
 // src/local-runtime/local-project.ts
 import { createHash as createHash2, randomUUID as randomUUID2 } from "crypto";
-import { readFile as readFile2, stat as stat2, writeFile as writeFile2 } from "fs/promises";
+import { readFile as readFile3, stat as stat2, writeFile as writeFile2 } from "fs/promises";
 import { dirname as dirname2, isAbsolute as isAbsolute3, relative as relative2, resolve as resolve4, sep as sep2 } from "path";
 import { homedir as homedir2 } from "os";
 import { mkdir as mkdir2 } from "fs/promises";
@@ -1432,6 +1678,34 @@ async function initLocalSpecFile(args) {
     });
     await mkdir2(dirname2(args.outputPath), { recursive: true });
     await writeFile2(args.outputPath, `${JSON.stringify(spec2, null, 2)}
+`, "utf8");
+    return spec2;
+  }
+  if (args.engine === "decision") {
+    const spec2 = localDecisionSpecFileSchema.parse({
+      engine: "decision",
+      id: randomUUID2(),
+      name: args.name,
+      description: "Typed decision model: one labelled answer per input.",
+      system_prompt: "Describe the decision this local model should make.",
+      guidelines: [],
+      constraints: [],
+      base_model: args.baseModel ?? DEFAULT_DECISION_MODEL,
+      decision: {
+        type: "choice",
+        criteria: {
+          first_label: "Replace this with what the first label means.",
+          second_label: "Replace this with what the second label means."
+        }
+      },
+      examples: [
+        { input: "Replace this with a representative input.", output: "first_label" },
+        { input: "Replace this with a different input.", output: "second_label" }
+      ],
+      hyperparameters: { n_epochs: 3 }
+    });
+    await mkdir2(dirname2(args.outputPath), { recursive: true });
+    await writeFile2(args.outputPath, `${JSON.stringify(spec2, null, 2)}
 `, "utf8");
     return spec2;
   }
@@ -1491,7 +1765,7 @@ async function initLocalRunnerConfigFile(args) {
 }
 function placeholderIssues(systemPrompt, examples) {
   const issues = [];
-  if (/describe the behavior this local model should learn/i.test(systemPrompt)) {
+  if (/describe the (behavior|decision) this local model should (learn|make)/i.test(systemPrompt)) {
     issues.push("system_prompt still contains the generated placeholder");
   }
   examples.forEach((example, index) => {
@@ -1513,6 +1787,14 @@ function assertLocalRunInputReady(request) {
 function foundationPlaceholderIssues(spec) {
   return placeholderIssues(spec.system_prompt, spec.examples);
 }
+function decisionPlaceholderIssues(spec) {
+  const issues = placeholderIssues(spec.system_prompt, spec.examples);
+  const descriptions = spec.decision.type === "noul" ? Object.values(spec.decision.criteria ?? {}) : Object.values(spec.decision.criteria);
+  if (descriptions.some((value) => /replace this with/i.test(value ?? ""))) {
+    issues.push("decision.criteria still contains generated placeholder text");
+  }
+  return issues;
+}
 function assertFoundationSpecReady(spec) {
   const issues = foundationPlaceholderIssues(spec);
   if (issues.length > 0) {
@@ -1520,9 +1802,9 @@ function assertFoundationSpecReady(spec) {
   }
 }
 function runRequestFromLocalSpec(spec, options = {}) {
-  if (isFoundationSpecFile(spec)) {
+  if (isFoundationSpecFile(spec) || isDecisionSpecFile(spec)) {
     throw new Error(
-      "Foundation specs cannot be converted into a LoRA run request. Use `tt pipeline run --spec`."
+      `${isFoundationSpecFile(spec) ? "Foundation" : "Decision"} specs cannot be converted into a LoRA run request. Use \`tt pipeline run --spec\`.`
     );
   }
   const {
@@ -1570,6 +1852,9 @@ function parseLocalRunInput(rawInput, path, options = {}) {
   if (isFoundationSpecFile(spec)) {
     return { kind: "foundation-spec", path, spec };
   }
+  if (isDecisionSpecFile(spec)) {
+    return { kind: "decision-spec", path, spec };
+  }
   return {
     kind: "spec",
     path,
@@ -1579,7 +1864,7 @@ function parseLocalRunInput(rawInput, path, options = {}) {
 }
 async function loadLocalRunInput(path, options = {}) {
   return parseLocalRunInput(
-    JSON.parse(await readFile2(path, "utf8")),
+    JSON.parse(await readFile3(path, "utf8")),
     path,
     options
   );
@@ -1594,7 +1879,7 @@ function validateBehaviorSpec(input) {
     return { valid: false, errors: ["Use a behavior spec, not a run request."], warnings };
   }
   const { spec } = input;
-  const placeholders = input.kind === "foundation-spec" ? foundationPlaceholderIssues(input.spec) : generatedPlaceholderIssues(input.request);
+  const placeholders = input.kind === "foundation-spec" ? foundationPlaceholderIssues(input.spec) : input.kind === "decision-spec" ? decisionPlaceholderIssues(input.spec) : generatedPlaceholderIssues(input.request);
   if (placeholders.length) {
     errors.push(`Edit the generated behavior spec before training: ${placeholders.join("; ")}.`);
   }
@@ -1609,6 +1894,29 @@ function validateBehaviorSpec(input) {
       errors.push("Foundation RL requires numeric expected outputs in every example.");
     }
   }
+  if (input.kind === "decision-spec") {
+    const labels = decisionLabels(input.spec.decision);
+    const normalized = new Map(labels.map((label) => [label.trim().toLowerCase(), label]));
+    if (normalized.size !== labels.length) {
+      errors.push("decision labels must stay distinct ignoring case and surrounding spaces.");
+    }
+    const counts = /* @__PURE__ */ new Map();
+    for (const [index, example] of spec.examples.entries()) {
+      const label = normalized.get(example.output.trim().toLowerCase());
+      if (!label) {
+        errors.push(`examples[${index}].output must be one of the decision labels: ${labels.join(", ")}.`);
+      } else {
+        counts.set(label, (counts.get(label) ?? 0) + 1);
+      }
+    }
+    const missing = labels.filter((label) => !counts.has(label));
+    if (missing.length && spec.examples.length) {
+      warnings.push(`No examples use decision label(s) ${missing.join(", ")}; the tuned model learns those only from their descriptions.`);
+    }
+    if (buildSystemMessage(spec).length > 600) {
+      warnings.push("The compiled decision instruction exceeds 600 characters; the decision model reads about 190 tokens of instruction and options, so later text may be truncated.");
+    }
+  }
   for (const key of ["name", "system_prompt"]) {
     if (!spec[key].trim()) errors.push(`${key} must not be blank.`);
   }
@@ -1784,7 +2092,7 @@ ${validation.errors.join("\n")}`);
   const pipeline = parsePipeline2(recipe);
   if (isFoundationPipeline2(pipeline) !== foundation) {
     const recipeEngine = isFoundationPipeline2(pipeline) ? "a foundation" : "an adapter";
-    const specEngine = foundation ? "a foundation" : "an adapter";
+    const specEngine = foundation ? "a foundation" : input.kind === "decision-spec" ? "a decision" : "an adapter";
     throw new Error(`Pipeline is ${recipeEngine} recipe, but the behavior spec is ${specEngine} spec. Use a matching recipe or omit --file.`);
   }
   if (foundation) assertFoundationSettingsMatch(input.spec.foundation, pipeline);
@@ -1833,128 +2141,6 @@ function assertFoundationSettingsMatch(hp, plan) {
   }
 }
 
-// src/local-runtime/dataset.ts
-import { readFile as readFile3 } from "fs/promises";
-async function loadNormalizedChatJsonl(path) {
-  const text = await readFile3(path, "utf8");
-  const rows = [];
-  const jsonLines = [];
-  for (const [index, line] of text.split(/\r?\n/).entries()) {
-    if (!line.trim()) continue;
-    let value;
-    try {
-      value = JSON.parse(line);
-    } catch (error) {
-      throw new Error(`Invalid chat JSONL row ${index + 1}: malformed JSON`, { cause: error });
-    }
-    if (!value || typeof value !== "object" || Array.isArray(value)) {
-      throw new Error(`Invalid chat JSONL row ${index + 1}: expected an object`);
-    }
-    const messages = value.messages;
-    if (!Array.isArray(messages) || ![2, 3].includes(messages.length) || messages.some(
-      (message) => !message || typeof message !== "object" || Array.isArray(message) || !["system", "user", "assistant"].includes(String(message.role)) || typeof message.content !== "string"
-    )) {
-      throw new Error(
-        `Invalid chat JSONL row ${index + 1}: expected an optional system message, one user message, and one assistant answer`
-      );
-    }
-    const row = { messages };
-    const offset = row.messages.length === 3 ? 1 : 0;
-    if (offset === 1 && row.messages[0]?.role !== "system" || row.messages[offset]?.role !== "user" || !row.messages[offset]?.content.trim() || row.messages[offset + 1]?.role !== "assistant" || !row.messages[offset + 1]?.content.trim()) {
-      throw new Error(
-        `Invalid chat JSONL row ${index + 1}: expected an optional system message, one non-empty user message, and one non-empty assistant answer`
-      );
-    }
-    rows.push(row);
-    jsonLines.push(JSON.stringify(row));
-  }
-  if (rows.length === 0) throw new Error(`Chat JSONL contains no examples: ${path}`);
-  return {
-    rows,
-    jsonl: jsonLines.join("\n")
-  };
-}
-async function normalizeChatJsonlForRelocation(path) {
-  return (await loadNormalizedChatJsonl(path)).jsonl;
-}
-function buildSystemMessage(spec) {
-  const parts = [];
-  const guidelines = spec.guidelines.map((value) => value.trim()).filter(Boolean);
-  const constraints = spec.constraints.map((value) => value.trim()).filter(Boolean);
-  if (spec.system_prompt.trim()) parts.push(spec.system_prompt.trim());
-  if (guidelines.length > 0) {
-    parts.push(`Guidelines:
-${guidelines.map((guideline) => `- ${guideline}`).join("\n")}`);
-  }
-  if (constraints.length > 0) {
-    parts.push(`Constraints:
-${constraints.map((constraint) => `- ${constraint}`).join("\n")}`);
-  }
-  return parts.join("\n\n") || "Follow the demonstrated behavior.";
-}
-function exampleToChatRow(spec, example) {
-  const system = buildSystemMessage(spec);
-  return {
-    messages: [
-      { role: "system", content: system },
-      { role: "user", content: example.input },
-      { role: "assistant", content: example.output }
-    ]
-  };
-}
-function compileSpecToJsonl(spec) {
-  return spec.examples.map((example) => JSON.stringify(exampleToChatRow(spec, example))).join("\n");
-}
-function examplesFromSpec(spec) {
-  return spec.examples;
-}
-async function examplesFromChatJsonl(path) {
-  const normalized = await loadNormalizedChatJsonl(path);
-  const examples = [];
-  for (const [index, row] of normalized.rows.entries()) {
-    const user = row.messages.at(-2);
-    const assistant = row.messages.at(-1);
-    examples.push({
-      input: user.content,
-      output: assistant.content
-    });
-  }
-  return examples;
-}
-async function evaluationSuiteFromChatJsonl(path, configuredSystem) {
-  const normalized = await loadNormalizedChatJsonl(path);
-  const inputIdentities = /* @__PURE__ */ new Set();
-  for (const row of normalized.rows) {
-    const input = row.messages.at(-2).content;
-    const identity = input.trim().replace(/\s+/g, " ").toLowerCase();
-    if (inputIdentities.has(identity)) {
-      throw new Error(
-        "General regression data contains duplicate inputs; evaluation prompts must be unique."
-      );
-    }
-    inputIdentities.add(identity);
-  }
-  const systems = new Set(
-    normalized.rows.map((row) => row.messages.find((message) => message.role === "system")?.content).filter((value) => value !== void 0)
-  );
-  if (systems.size > 1) {
-    throw new Error("General regression rows must use one consistent system message.");
-  }
-  const embeddedSystem = [...systems][0];
-  if (configuredSystem !== void 0 && embeddedSystem !== void 0 && configuredSystem !== embeddedSystem) {
-    throw new Error(
-      "evaluation.generalRegression.systemPrompt must match the system message embedded in the dataset."
-    );
-  }
-  return {
-    examples: normalized.rows.map((row) => ({
-      input: row.messages.at(-2).content,
-      output: row.messages.at(-1).content
-    })),
-    system: configuredSystem ?? embeddedSystem ?? "You are a helpful assistant."
-  };
-}
-
 // src/local-runtime/run-reporter.ts
 function reportInBackground(action) {
   try {
@@ -2600,6 +2786,31 @@ var foundationRuntimeHash = (() => {
   return hash.digest("hex").slice(0, 20);
 })();
 var FOUNDATION_PYTHON_ENVIRONMENT = pythonEnvironmentPath("uv-foundation", foundationRuntimeHash);
+var decisionProject = join4(packageRoot, "training/decision");
+var decisionRuntimeHash = (() => {
+  const hash = createHash3("sha256");
+  for (const name of ["pyproject.toml", "uv.lock"]) {
+    hash.update(readFileSync(join4(decisionProject, name)));
+  }
+  return hash.digest("hex").slice(0, 20);
+})();
+var DECISION_PYTHON_ENVIRONMENT = pythonEnvironmentPath("uv-decision", decisionRuntimeHash);
+function buildDecisionPythonCommand(entrypoint, args = []) {
+  const commandArgs = [
+    "run",
+    "--frozen",
+    "--quiet",
+    "--project",
+    decisionProject,
+    "python",
+    join4(decisionProject, "src", entrypoint),
+    ...args
+  ];
+  return { command: "uv", commandArgs, displayCommand: ["uv", ...commandArgs] };
+}
+function withDecisionPythonEnvironment(env) {
+  return { ...env, UV_PROJECT_ENVIRONMENT: DECISION_PYTHON_ENVIRONMENT };
+}
 function buildBundledPythonCommand(entrypoint, args = []) {
   const target = entrypoint === "-c" ? entrypoint : join4(bundledProject, "src", entrypoint);
   const commandArgs = [
@@ -4640,7 +4851,7 @@ async function resolveBaseModelRevision(request, config) {
     request.hyperparameters.base_model_revision
   );
   if (config.paths.baseModel) {
-    return defaultBaseModelRevision(request.spec_snapshot.base_model);
+    return isCertifiedTrainingModel(request.spec_snapshot.base_model) ? defaultBaseModelRevision(request.spec_snapshot.base_model) : requested;
   }
   if (requested) return requested;
   if (!request.spec_snapshot.base_model.includes("/")) {
@@ -6026,6 +6237,12 @@ function formatCapabilitySummary(report) {
 }
 function warningsFromSnapshot(report, target) {
   const warnings = [];
+  if (target.engine === "decision") {
+    if (!report.cuda_available) {
+      warnings.push("No CUDA GPU detected: decision training runs on CPU or Apple MPS (roughly 15 s per 8-example step on a 4-core CPU).");
+    }
+    return warnings;
+  }
   if (target.engine === "foundation") {
     for (const kind of ["train", "finetune"]) {
       const verdict = report.foundation[kind];
@@ -6460,6 +6677,7 @@ async function assessHardware(options = {}) {
 }
 
 export {
+  fileUri,
   assertArtifactManifest,
   TRAINING_MODELS,
   resolveTrainingModel,
@@ -6467,9 +6685,12 @@ export {
   defaultBaseModelRevision,
   resolveModelLoader,
   assertUsableModelArtifact,
+  DECISION_MODELS,
+  resolveDecisionModel,
   DEFAULT_ARTIFACT_ROOT,
   DEFAULT_PROJECT_STORE_ROOT,
   DEFAULT_FOUNDATION_RUNS_DIR,
+  DEFAULT_DECISION_RUNS_DIR,
   expandUserPath,
   getConfigDir,
   getAgentConfigDir,
@@ -6495,8 +6716,20 @@ export {
   projectRuntimeSchema,
   localFoundationSpecFileSchema,
   localAdapterSpecFileSchema,
+  decisionQuestionSchema,
+  decisionHyperparametersSchema,
+  localDecisionSpecFileSchema,
+  decisionLabels,
   localBehaviorSpecFileSchema,
   isFoundationSpecFile,
+  isDecisionSpecFile,
+  normalizeChatJsonlForRelocation,
+  buildSystemMessage,
+  exampleToChatRow,
+  compileSpecToJsonl,
+  examplesFromSpec,
+  examplesFromChatJsonl,
+  evaluationSuiteFromChatJsonl,
   validateEvalCases,
   DEFAULT_LOCAL_SPEC_PATH,
   localPathsOverlap,
@@ -6506,6 +6739,7 @@ export {
   generatedPlaceholderIssues,
   assertLocalRunInputReady,
   foundationPlaceholderIssues,
+  decisionPlaceholderIssues,
   assertFoundationSpecReady,
   runRequestFromLocalSpec,
   parseLocalRunInput,
@@ -6522,17 +6756,12 @@ export {
   canonicalJson2 as canonicalJson,
   isFoundationPipeline3 as isFoundationPipeline,
   parsePipeline3 as parsePipeline,
-  normalizeChatJsonlForRelocation,
-  buildSystemMessage,
-  exampleToChatRow,
-  compileSpecToJsonl,
-  examplesFromSpec,
-  examplesFromChatJsonl,
-  evaluationSuiteFromChatJsonl,
   reportInBackground,
   sanitizeLogLine,
   forwardStreamLines,
   localRuntimePackageRoot,
+  buildDecisionPythonCommand,
+  withDecisionPythonEnvironment,
   buildBundledPythonCommand,
   withBundledPythonEnvironment,
   buildFoundationPythonCommand,
@@ -6550,6 +6779,9 @@ export {
   writePrivateJsonAtomic,
   createLocalStore,
   tokenF1,
+  deriveSampleSeed,
+  splitSpecExamples,
+  compareEvalReports,
   verifyModelPrefetchCacheReport,
   buildModelPrefetchPayload,
   verifyLocalBaseModel,
@@ -6567,10 +6799,11 @@ export {
   runLocalFineTune,
   configForSpec,
   resolveProjectConfig,
+  evaluateCapabilities,
   warningsFromSnapshot,
   readHardwareSnapshot,
   formatHostStatusLine,
   formatAgentHostBlock,
   assessHardware
 };
-//# sourceMappingURL=chunk-4Q54NHKR.js.map
+//# sourceMappingURL=chunk-PVLX7LJI.js.map