@tuned-tensor/cli 0.20.0 → 0.21.0

package/dist/local-runtime.js

@@ -20,6 +20,10 @@ import {
   createLocalStore,
   datasetFormatSchema,
   datasetPrebuiltSchema,
+  decisionHyperparametersSchema,
+  decisionLabels,
+  decisionPlaceholderIssues,
+  decisionQuestionSchema,
   defaultBaseModelRevision,
   defaultLocalHome,
   evalExampleResultSchema,
@@ -40,6 +44,7 @@ import {
   hardenExistingLocalStore,
   initLocalRunnerConfigFile,
   initLocalSpecFile,
+  isDecisionSpecFile,
   isFoundationSpecFile,
   isTerminalRunState,
   jsonFieldMetricsSchema,
@@ -49,6 +54,7 @@ import {
   loadRunRequest,
   localAdapterSpecFileSchema,
   localBehaviorSpecFileSchema,
+  localDecisionSpecFileSchema,
   localFoundationSpecFileSchema,
   localPathsOverlap,
   localRunnerConfigSchema,
@@ -88,7 +94,7 @@ import {
   withHuggingFaceCacheEnvironment,
   withOfflineHuggingFaceCacheEnvironment,
   writePrivateJsonAtomic
-} from "./chunk-4Q54NHKR.js";
+} from "./chunk-PVLX7LJI.js";
 
 // src/local-runtime/index.ts
 import { readFileSync as readFileSync3, realpathSync } from "fs";
@@ -972,7 +978,7 @@ var COMMAND_DEFINITIONS = {
     description: "Create a local tunedtensor.json behavior spec.",
     options: [
       { name: "--name", value: "name", description: "Behavior spec name" },
-      { name: "--engine", value: "engine", description: "adapter (default) or foundation" },
+      { name: "--engine", value: "engine", description: "adapter (default), foundation or decision" },
       { name: "--model", value: "model", description: "Base model ID" },
       { name: "--output", value: "path", description: "Output spec path" },
       { name: "--profile", value: "profile", description: "Compatibility host profile (spark); creates only tunedtensor.json" },
@@ -1226,6 +1232,7 @@ async function configSelectionFromArgv(argv, adjacentTo, input) {
 async function configFromArgv(argv, adjacentTo) {
   return (await configSelectionFromArgv(argv, adjacentTo)).config;
 }
+var DECISION_SERVE_MESSAGE = "Decision models are not served through vLLM. Load the tuned model directory with `laya.load(<model_dir>)`; see docs/local-runtime/decision.md.";
 async function loadCliBehaviorSpec(inputPath, runId) {
   const input = await loadLocalRunInput(inputPath, {
     ...runId ? { runId } : {}
@@ -1235,6 +1242,11 @@ async function loadCliBehaviorSpec(inputPath, runId) {
       `This spec uses engine "foundation". Use \`tt pipeline run --spec ${input.path}\`; the legacy adapter runner cannot execute foundation specs.`
     );
   }
+  if (input.kind === "decision-spec") {
+    throw new Error(
+      `This spec uses engine "decision". Use \`tt pipeline run --spec ${input.path}\`; the legacy adapter runner cannot execute decision specs.`
+    );
+  }
   if (input.kind !== "spec") {
     throw new Error(`TT expects a tunedtensor.json behavior spec, not a full run request: ${input.path}`);
   }
@@ -1405,6 +1417,7 @@ async function modelSystemPrompt(args) {
     if (isFoundationSpecFile(local.data)) {
       throw new Error("Foundation specs have no Hugging Face base model. Use `tt serve foundation --checkpoint <model-dir> --tokenizer <tokenizer.json>`.");
     }
+    if (isDecisionSpecFile(local.data)) throw new Error(DECISION_SERVE_MESSAGE);
     spec = local.data;
   } else {
     const runRequestPath = join5(args.store.paths.runsDir, args.model.run_id, "request.json");
@@ -1556,6 +1569,7 @@ async function serveBaseModelFromCli(args) {
   if (spec && isFoundationSpecFile(spec)) {
     throw new Error("Foundation specs have no Hugging Face base model. Use `tt serve foundation --checkpoint <model-dir> --tokenizer <tokenizer.json>`.");
   }
+  if (spec && isDecisionSpecFile(spec)) throw new Error(DECISION_SERVE_MESSAGE);
   let baseModel = spec?.base_model;
   if (!baseModel) {
     const store = createLocalStore(args.config.storeRoot);
@@ -1689,6 +1703,7 @@ async function main(argv) {
       const input = await loadLocalRunInput(inputPath);
       if (input.kind === "foundation-spec") {
         foundationSpec = input.spec;
+      } else if (input.kind === "decision-spec") {
       } else if (input.kind === "spec") {
         request = input.request;
       } else {
@@ -1731,8 +1746,8 @@ async function main(argv) {
       throw new Error(`--profile must be spark, got: ${profile}`);
     }
     const engine = readOption(argv, "--engine") ?? "adapter";
-    if (engine !== "adapter" && engine !== "foundation") {
-      throw new Error(`--engine must be adapter or foundation, got: ${engine}`);
+    if (engine !== "adapter" && engine !== "foundation" && engine !== "decision") {
+      throw new Error(`--engine must be adapter, foundation or decision, got: ${engine}`);
     }
     if (engine === "foundation" && readOption(argv, "--model")) {
       throw new Error("Foundation specs do not take --model; they train a tokenizer and GPT from scratch.");
@@ -1740,9 +1755,9 @@ async function main(argv) {
     if (readOption(argv, "--config")) throw new Error("tt init uses one tunedtensor.json; put overrides in runtime/evaluation instead of --config.");
     const spec = await initLocalSpecFile({
       outputPath,
-      name: readOption(argv, "--name") ?? (engine === "foundation" ? "Foundation chat model" : "Local Tuned Tensor Spec"),
+      name: readOption(argv, "--name") ?? (engine === "foundation" ? "Foundation chat model" : engine === "decision" ? "Local decision model" : "Local Tuned Tensor Spec"),
       engine,
-      baseModel: engine === "foundation" ? void 0 : readOption(argv, "--model") ?? "Qwen/Qwen3.5-2B",
+      baseModel: engine === "foundation" ? void 0 : readOption(argv, "--model") ?? (engine === "decision" ? void 0 : "Qwen/Qwen3.5-2B"),
       force: hasFlag(argv, "--force")
     });
     const configPath = void 0;
@@ -1770,6 +1785,20 @@ async function main(argv) {
       });
       return;
     }
+    if (loaded.kind === "decision-spec") {
+      const validation = validateBehaviorSpec(loaded);
+      if (!validation.valid) throw new Error(`Invalid decision spec:
+${validation.errors.join("\n")}`);
+      printJson({
+        ok: true,
+        input_path: loaded.path,
+        engine: "decision",
+        base_model: loaded.spec.base_model,
+        warnings: validation.warnings,
+        pipeline: "tt pipeline run --spec"
+      });
+      return;
+    }
     if (loaded.kind !== "spec") {
       throw new Error(`TT expects a tunedtensor.json behavior spec, not a full run request: ${loaded.path}`);
     }
@@ -2013,6 +2042,10 @@ export {
   createLocalStore,
   datasetFormatSchema,
   datasetPrebuiltSchema,
+  decisionHyperparametersSchema,
+  decisionLabels,
+  decisionPlaceholderIssues,
+  decisionQuestionSchema,
   defaultLocalHome,
   evalExampleResultSchema,
   evalReportSchema,
@@ -2034,6 +2067,7 @@ export {
   hardenExistingLocalStore,
   initLocalRunnerConfigFile,
   initLocalSpecFile,
+  isDecisionSpecFile,
   isFoundationSpecFile,
   isTerminalRunState,
   jsonFieldMetricsSchema,
@@ -2043,6 +2077,7 @@ export {
   loadRunRequest,
   localAdapterSpecFileSchema,
   localBehaviorSpecFileSchema,
+  localDecisionSpecFileSchema,
   localFoundationSpecFileSchema,
   localPathsOverlap,
   localRunnerConfigSchema,