@tuned-tensor/cli 0.11.0 → 0.12.0-beta.0

package/dist/index.js

@@ -1,10 +1,11 @@
 #!/usr/bin/env node
 import {
+  TRAINING_MODELS,
   assertLocalRunInputReady,
   loadLocalRunInput,
   loadLocalRunnerConfig,
   runLocalPipeline
-} from "./chunk-DAP2Z7EZ.js";
+} from "./chunk-HDGBDHOB.js";
 
 // src/cli.ts
 import { spawn as spawn4 } from "child_process";
@@ -24,76 +25,6 @@ import { join } from "path";
 import { Type } from "typebox";
 import { Value } from "typebox/value";
 
-// src/base-models.ts
-var SUPPORTED_BASE_MODELS = [
-  "google/gemma-4-E2B-it",
-  "google/gemma-4-E4B-it",
-  "Qwen/Qwen3.5-2B",
-  "Qwen/Qwen3-VL-2B-Instruct",
-  "Qwen/Qwen3.5-4B",
-  "meta-llama/Llama-3.2-3B-Instruct",
-  "microsoft/Phi-4-mini-instruct",
-  "ibm-granite/granite-3.3-2b-instruct",
-  "bigcode/starcoder2-3b"
-];
-var BASE_MODEL_ALIASES = /* @__PURE__ */ new Map();
-for (const model of SUPPORTED_BASE_MODELS) {
-  BASE_MODEL_ALIASES.set(model.toLowerCase(), model);
-}
-for (const [alias, model] of [
-  ["google/gemma-4-E2B", "google/gemma-4-E2B-it"],
-  ["google/gemma-4-e2b", "google/gemma-4-E2B-it"],
-  ["google/gemma-4-e2b-it", "google/gemma-4-E2B-it"],
-  ["google/gemma-4-E4B", "google/gemma-4-E4B-it"],
-  ["google/gemma-4-e4b", "google/gemma-4-E4B-it"],
-  ["google/gemma-4-e4b-it", "google/gemma-4-E4B-it"],
-  ["qwen/qwen3.5-2b", "Qwen/Qwen3.5-2B"],
-  ["Qwen/Qwen3.5-2B-Base", "Qwen/Qwen3.5-2B"],
-  ["qwen/qwen3.5-2b-base", "Qwen/Qwen3.5-2B"],
-  ["qwen/qwen3-vl-2b-instruct", "Qwen/Qwen3-VL-2B-Instruct"],
-  ["Qwen/Qwen3-VL-2B", "Qwen/Qwen3-VL-2B-Instruct"],
-  ["qwen/qwen3-vl-2b", "Qwen/Qwen3-VL-2B-Instruct"],
-  ["qwen/qwen3.5-4b", "Qwen/Qwen3.5-4B"],
-  ["Qwen/Qwen3.5-4B-Base", "Qwen/Qwen3.5-4B"],
-  ["qwen/qwen3.5-4b-base", "Qwen/Qwen3.5-4B"],
-  ["meta-llama/llama-3.2-3b-instruct", "meta-llama/Llama-3.2-3B-Instruct"],
-  ["meta-llama/Llama-3.2-3B", "meta-llama/Llama-3.2-3B-Instruct"],
-  ["meta-llama/llama-3.2-3b", "meta-llama/Llama-3.2-3B-Instruct"],
-  ["microsoft/phi-4-mini-instruct", "microsoft/Phi-4-mini-instruct"],
-  ["phi-4-mini-instruct", "microsoft/Phi-4-mini-instruct"],
-  ["ibm-granite/granite-3.3-2b-instruct", "ibm-granite/granite-3.3-2b-instruct"],
-  ["granite-3.3-2b-instruct", "ibm-granite/granite-3.3-2b-instruct"],
-  ["bigcode/starcoder2-3b", "bigcode/starcoder2-3b"],
-  ["starcoder2-3b", "bigcode/starcoder2-3b"]
-]) {
-  BASE_MODEL_ALIASES.set(alias.toLowerCase(), model);
-}
-var UnsupportedBaseModelError = class extends Error {
-  constructor(model) {
-    const supported = SUPPORTED_BASE_MODELS.join(", ");
-    super(
-      `Unsupported base_model "${String(model)}". Supported base models: ${supported}`
-    );
-    this.name = "UnsupportedBaseModelError";
-  }
-};
-function canonicalizeBaseModel(model) {
-  if (typeof model !== "string" || !model.trim()) {
-    throw new UnsupportedBaseModelError(model);
-  }
-  const canonical = BASE_MODEL_ALIASES.get(model.trim().toLowerCase());
-  if (!canonical) {
-    throw new UnsupportedBaseModelError(model);
-  }
-  return canonical;
-}
-function canonicalizeSpecBaseModel(body) {
-  if ("base_model" in body && body.base_model !== void 0) {
-    return { ...body, base_model: canonicalizeBaseModel(body.base_model) };
-  }
-  return body;
-}
-
 // src/eval/rules.ts
 function validateSpec(spec) {
   const checks = [];
@@ -258,15 +189,15 @@ var FolderName = Type.String({
   pattern: "^[A-Za-z0-9][A-Za-z0-9._-]*$",
   description: "One new folder name directly beneath the current workspace"
 });
-var BaseModel = Type.Unsafe({
+var LocalBaseModel = Type.String({
   type: "string",
-  enum: [...SUPPORTED_BASE_MODELS]
+  enum: TRAINING_MODELS.map((model) => model.id)
 });
 var NonEmptyText = Type.String({ minLength: 1, maxLength: 1e3 });
 var LocalProjectSpecSchema = Type.Object({
   name: Type.String({ minLength: 1, maxLength: 255 }),
   description: Type.Optional(Type.String({ maxLength: 5e3 })),
-  base_model: BaseModel,
+  base_model: LocalBaseModel,
   system_prompt: Type.String({ minLength: 1, maxLength: 1e4 }),
   guidelines: Type.Array(NonEmptyText, { minItems: 1, maxItems: 50 }),
   constraints: Type.Optional(Type.Array(NonEmptyText, { maxItems: 50 })),
@@ -721,6 +652,83 @@ function ensureOpenRouterAppHeaders(modelsPath) {
 import { randomUUID } from "crypto";
 import { Type as Type2 } from "typebox";
 
+// src/base-models.ts
+var SUPPORTED_BASE_MODELS = [
+  "google/gemma-4-E2B-it",
+  "google/gemma-4-E4B-it",
+  "Qwen/Qwen3.5-2B",
+  "Qwen/Qwen3-VL-2B-Instruct",
+  "Qwen/Qwen3.5-4B",
+  "nvidia/NVIDIA-Nemotron-3.5-Lightning-30B-A3B-BF16",
+  "meta-models/Muse-Glimmer-30B",
+  "meta-llama/Llama-3.2-3B-Instruct",
+  "microsoft/Phi-4-mini-instruct",
+  "ibm-granite/granite-3.3-2b-instruct",
+  "bigcode/starcoder2-3b"
+];
+var BASE_MODEL_ALIASES = /* @__PURE__ */ new Map();
+for (const model of SUPPORTED_BASE_MODELS) {
+  BASE_MODEL_ALIASES.set(model.toLowerCase(), model);
+}
+for (const [alias, model] of [
+  ["google/gemma-4-E2B", "google/gemma-4-E2B-it"],
+  ["google/gemma-4-e2b", "google/gemma-4-E2B-it"],
+  ["google/gemma-4-e2b-it", "google/gemma-4-E2B-it"],
+  ["google/gemma-4-E4B", "google/gemma-4-E4B-it"],
+  ["google/gemma-4-e4b", "google/gemma-4-E4B-it"],
+  ["google/gemma-4-e4b-it", "google/gemma-4-E4B-it"],
+  ["qwen/qwen3.5-2b", "Qwen/Qwen3.5-2B"],
+  ["Qwen/Qwen3.5-2B-Base", "Qwen/Qwen3.5-2B"],
+  ["qwen/qwen3.5-2b-base", "Qwen/Qwen3.5-2B"],
+  ["qwen/qwen3-vl-2b-instruct", "Qwen/Qwen3-VL-2B-Instruct"],
+  ["Qwen/Qwen3-VL-2B", "Qwen/Qwen3-VL-2B-Instruct"],
+  ["qwen/qwen3-vl-2b", "Qwen/Qwen3-VL-2B-Instruct"],
+  ["qwen/qwen3.5-4b", "Qwen/Qwen3.5-4B"],
+  ["Qwen/Qwen3.5-4B-Base", "Qwen/Qwen3.5-4B"],
+  ["qwen/qwen3.5-4b-base", "Qwen/Qwen3.5-4B"],
+  [
+    "nvidia/nvidia-nemotron-3.5-lightning-30b-a3b-bf16",
+    "nvidia/NVIDIA-Nemotron-3.5-Lightning-30B-A3B-BF16"
+  ],
+  ["meta-models/muse-glimmer-30b", "meta-models/Muse-Glimmer-30B"],
+  ["meta-llama/llama-3.2-3b-instruct", "meta-llama/Llama-3.2-3B-Instruct"],
+  ["meta-llama/Llama-3.2-3B", "meta-llama/Llama-3.2-3B-Instruct"],
+  ["meta-llama/llama-3.2-3b", "meta-llama/Llama-3.2-3B-Instruct"],
+  ["microsoft/phi-4-mini-instruct", "microsoft/Phi-4-mini-instruct"],
+  ["phi-4-mini-instruct", "microsoft/Phi-4-mini-instruct"],
+  ["ibm-granite/granite-3.3-2b-instruct", "ibm-granite/granite-3.3-2b-instruct"],
+  ["granite-3.3-2b-instruct", "ibm-granite/granite-3.3-2b-instruct"],
+  ["bigcode/starcoder2-3b", "bigcode/starcoder2-3b"],
+  ["starcoder2-3b", "bigcode/starcoder2-3b"]
+]) {
+  BASE_MODEL_ALIASES.set(alias.toLowerCase(), model);
+}
+var UnsupportedBaseModelError = class extends Error {
+  constructor(model) {
+    const supported = SUPPORTED_BASE_MODELS.join(", ");
+    super(
+      `Unsupported base_model "${String(model)}". Supported base models: ${supported}`
+    );
+    this.name = "UnsupportedBaseModelError";
+  }
+};
+function canonicalizeBaseModel(model) {
+  if (typeof model !== "string" || !model.trim()) {
+    throw new UnsupportedBaseModelError(model);
+  }
+  const canonical = BASE_MODEL_ALIASES.get(model.trim().toLowerCase());
+  if (!canonical) {
+    throw new UnsupportedBaseModelError(model);
+  }
+  return canonical;
+}
+function canonicalizeSpecBaseModel(body) {
+  if ("base_model" in body && body.base_model !== void 0) {
+    return { ...body, base_model: canonicalizeBaseModel(body.base_model) };
+  }
+  return body;
+}
+
 // src/pipeline.ts
 import {
   canonicalPipeline as canonicalPortablePipeline,
@@ -837,7 +845,7 @@ var RunBody = Type2.Object({
   }, { additionalProperties: false })),
   split_ratios: Type2.Optional(SplitRatios)
 }, { additionalProperties: false });
-var BaseModel2 = Type2.Union(
+var BaseModel = Type2.Union(
   SUPPORTED_BASE_MODELS.map((model) => Type2.Literal(model))
 );
 var PipelineDocument = Type2.Object({
@@ -864,7 +872,7 @@ var CreateSpecBody = Type2.Object({
   guidelines: Type2.Optional(Guidelines),
   examples: Examples,
   constraints: Type2.Optional(Guidelines),
-  base_model: Type2.Optional(BaseModel2)
+  base_model: Type2.Optional(BaseModel)
 }, { additionalProperties: false });
 var UpdateSpecBody = Type2.Object({
   name: Type2.Optional(Type2.String({ minLength: 1, maxLength: 255 })),
@@ -873,7 +881,7 @@ var UpdateSpecBody = Type2.Object({
   guidelines: Type2.Optional(Guidelines),
   examples: Type2.Optional(Examples),
   constraints: Type2.Optional(Guidelines),
-  base_model: Type2.Optional(BaseModel2)
+  base_model: Type2.Optional(BaseModel)
 }, { additionalProperties: false });
 function record2(value) {
   return value !== null && typeof value === "object" && !Array.isArray(value) ? value : void 0;
@@ -1762,8 +1770,10 @@ var SLASH_COMMANDS = [
   { path: "/help", description: "Show commands; add a word to filter." },
   { path: "/status", description: "Show lightweight workflow status." },
   { path: "/context", description: "Show the current project and backend context." },
-  { path: "/mode cloud", description: "Use the managed cloud workflow." },
-  { path: "/mode local", description: "Use the local GPU workflow." },
+  { path: "/cloud", description: "Switch to the managed cloud workflow." },
+  { path: "/local", description: "Switch to the local GPU workflow." },
+  { path: "/mode cloud", description: "Alias for /cloud." },
+  { path: "/mode local", description: "Alias for /local." },
   { path: "/model", description: "Show the active model; /model <id> activates a local model." },
   { path: "/cd", description: "Change the shell's working directory." },
   { path: "/clear", description: "Clear the terminal." },
@@ -1874,6 +1884,14 @@ function cloudConfigPath(env, homeDirectory) {
   const configHome = env.XDG_CONFIG_HOME ? resolve(env.XDG_CONFIG_HOME) : join5(homeDirectory, ".config");
   return join5(configHome, "tuned-tensor", "config.json");
 }
+function agentSelectionFrom(env, configAgent) {
+  const stored = configAgent && typeof configAgent === "object" && !Array.isArray(configAgent) ? configAgent : {};
+  const provider = env.TUNED_TENSOR_AGENT_PROVIDER?.trim() || stringField(stored, "provider");
+  const model = env.TUNED_TENSOR_AGENT_MODEL?.trim() || stringField(stored, "model");
+  const thinking = env.TUNED_TENSOR_AGENT_THINKING?.trim() || stringField(stored, "thinking");
+  if (!provider && !model && !thinking) return void 0;
+  return { provider, model, thinking };
+}
 function redactApiKey(key) {
   if (!key) return void 0;
   if (key.length <= 4) return "\u2026";
@@ -1971,6 +1989,7 @@ async function discoverShellContext(options = {}) {
   const storedApiKey = stringField(cloudConfigJson.value, "api_key");
   const apiKey = environmentApiKey || storedApiKey;
   const baseUrl = env.TUNED_TENSOR_URL?.trim() || stringField(cloudConfigJson.value, "base_url") || "https://tunedtensor.com";
+  const agent = agentSelectionFrom(env, cloudConfigJson.value?.agent);
   const environmentTarget = targetFromEnvironment(env);
   if (env.TT_TARGET && !environmentTarget) {
     warnings.push(`Ignoring invalid TT_TARGET=${JSON.stringify(env.TT_TARGET)}; use cloud or local.`);
@@ -2001,12 +2020,19 @@ async function discoverShellContext(options = {}) {
       activeModelId,
       latestRun
     },
+    agent,
     warnings
   };
 }
 function valueOrDash(value) {
   return value === void 0 || value === "" ? "\u2014" : String(value);
 }
+function agentLabel(context) {
+  const agent = context.agent;
+  if (!agent?.provider && !agent?.model) return "not configured";
+  const id = [agent.provider, agent.model].filter(Boolean).join("/");
+  return agent.thinking ? `${id} (thinking ${agent.thinking})` : id;
+}
 function targetSourceLabel(source) {
   switch (source) {
     case "environment":
@@ -2028,6 +2054,7 @@ function formatShellContext(context, selectedTarget, targetSource) {
     `Spec ID        ${valueOrDash(context.spec?.id)}`,
     `Base model     ${valueOrDash(context.spec?.baseModel)}`,
     `Examples       ${valueOrDash(context.spec?.exampleCount)}`,
+    `Agent model    ${agentLabel(context)}`,
     `Cloud endpoint ${context.cloud.baseUrl}`,
     `Cloud auth     ${context.cloud.authenticated ? `yes (${context.cloud.keyPrefix})` : "no"}`,
     `Local config   ${context.local.configPath ?? "not found"}`,
@@ -2040,7 +2067,8 @@ function formatShellContext(context, selectedTarget, targetSource) {
 function formatShellStatus(context, selectedTarget) {
   const lines = [
     `Workflow       ${selectedTarget}`,
-    `Spec           ${context.spec?.name ?? (context.spec ? "unnamed" : "not found")}`
+    `Spec           ${context.spec?.name ?? (context.spec ? "unnamed" : "not found")}`,
+    `Agent model    ${agentLabel(context)}`
   ];
   if (selectedTarget === "cloud") {
     lines.push(
@@ -2219,6 +2247,8 @@ var SLASH_NAMES = /* @__PURE__ */ new Set([
   "status",
   "context",
   "mode",
+  "cloud",
+  "local",
   "model",
   "clear",
   "cd",
@@ -2382,14 +2412,22 @@ function helpText(mode, query, palette = false) {
 function activeModelLabel(snapshot) {
   return snapshot.mode === "local" ? snapshot.context.local.activeModelId ?? "base" : snapshot.context.spec?.baseModel ?? "\u2014";
 }
+function agentModelLabel(context) {
+  const agent = context.agent;
+  if (!agent?.provider && !agent?.model) return "not configured";
+  return [agent.provider, agent.model].filter(Boolean).join("/") || "not configured";
+}
 function renderShellBanner(snapshot) {
   const spec = snapshot.context.spec?.name ?? snapshot.context.spec?.path ?? "no spec";
   const heading = snapshot.version ? `${accent.bold("tt")} ${chalk.dim(`v${snapshot.version}`)}` : accent.bold("tt");
   const lines = [
     heading,
     `${accent(snapshot.mode)}${chalk.dim(
-      ` \xB7 ${snapshot.context.projectName} \xB7 ${spec} \xB7 model ${activeModelLabel(snapshot)}`
+      ` \xB7 ${snapshot.context.projectName} \xB7 ${spec}`
     )}`,
+    chalk.dim(
+      `agent ${agentModelLabel(snapshot.context)} \xB7 workflow model ${activeModelLabel(snapshot)}`
+    ),
     chalk.dim("ctrl+c stop/clear \xB7 ctrl+d exit \xB7 /help commands \xB7 tab complete"),
     "",
     chalk.dim("Ask TT anything. Known commands run directly.")
@@ -2535,6 +2573,14 @@ var TunedTensorShellSession = class _TunedTensorShellSession {
 `);
         return "continue";
       }
+      case "cloud":
+      case "local":
+        assertNoArgs(command.name, command.args);
+        this.mode = command.name;
+        this.modeSource = "session";
+        this.io.write(`${successMark()} Workflow switched to ${this.mode}.
+`);
+        return "continue";
       case "model": {
         if (command.args.length === 0) {
           await this.refreshContext();
@@ -8080,16 +8126,22 @@ function registerAgentCommands(parent, options) {
     const runtime = await options.getRuntime();
     const resolved = resolveAgentModelDefinition(runtime, selection);
     const authenticated = runtime.hasConfiguredAuth(selection.provider);
+    const providerName = runtime.getProviders().find(
+      (provider) => provider.id === selection.provider
+    )?.name;
     const value = {
       execution: "local",
       provider: selection.provider,
+      provider_name: providerName,
       model: selection.model,
+      model_name: resolved.model.name,
       thinking: selection.thinking,
       authenticated,
       supports_thinking: resolved.model.reasoning !== false
     };
+    const modelLabel = resolved.model.name && resolved.model.name !== selection.model ? `${resolved.model.name} (${selection.provider}/${selection.model})` : `${selection.provider}/${selection.model}`;
     options.output.write(isJsonMode() ? `${JSON.stringify(value, null, 2)}
-` : `Local agent: ${value.provider}/${value.model} (thinking: ${value.thinking}, auth: ${authenticated ? "configured" : "required"})
+` : `Local agent: ${modelLabel} (thinking: ${value.thinking}, auth: ${authenticated ? "configured" : "required"})
 `);
   });
   agent.command("models").description("List Pi provider models available to the local agent").option("--provider <provider>", "Filter by Pi provider ID").option("--all", "Include models whose provider auth is not configured").action(async (commandOptions) => {
@@ -8677,7 +8729,7 @@ async function runCli(version, runtime = {}) {
 }
 
 // src/index.ts
-runCli("0.11.0").catch((err) => {
+runCli("0.12.0-beta.0").catch((err) => {
   reportError(err);
   process.exit(1);
 });