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

package/dist/index.js

@@ -583,7 +583,7 @@ function resolveAgentModel(runtime, selection) {
   const resolved = resolveAgentModelDefinition(runtime, selection);
   if (!runtime.hasConfiguredAuth(selection.provider)) {
     throw new Error(
-      `Authenticate Pi provider "${selection.provider}" first (for example with Pi's /login), then retry. Provider secrets are never accepted by tt flags.`
+      `Authenticate provider "${selection.provider}" first, then retry. Provider secrets are never accepted by tt flags.`
     );
   }
   return resolved;
@@ -594,7 +594,7 @@ function resolveAgentModelDefinition(runtime, selection) {
   );
   if (!provider) {
     throw new Error(
-      `Unknown Pi provider "${selection.provider}". Run \`tt agent models --all\` to list providers and models.`
+      `Unknown provider "${selection.provider}". Run \`tt agent models --all\` to list providers and models.`
     );
   }
   const model = runtime.getModel(selection.provider, selection.model);
@@ -1127,7 +1127,7 @@ You have no general filesystem tools. The only workspace-scoped local spec capab
 }
 function defaultAgent(options) {
   if (!options.streamSimple) {
-    throw new Error("The configured Pi model runtime cannot stream this model.");
+    throw new Error("The configured model runtime cannot stream this model.");
   }
   let calls = 0;
   return new Agent({
@@ -1570,6 +1570,10 @@ function readConfig() {
     return {};
   }
 }
+var configRevision = 0;
+function getConfigRevision() {
+  return configRevision;
+}
 function writeConfig(config) {
   const dir = getConfigDir();
   if (!existsSync2(dir)) mkdirSync2(dir, { recursive: true, mode: 448 });
@@ -1579,6 +1583,7 @@ function writeConfig(config) {
     mode: 384
   });
   chmodSync(path, 384);
+  configRevision += 1;
 }
 function updateConfig(partial) {
   const current = readConfig();
@@ -1770,11 +1775,7 @@ 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: "/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: "/model", description: "Show or change the TT agent model." },
   { path: "/cd", description: "Change the shell's working directory." },
   { path: "/clear", description: "Clear the terminal." },
   { path: "/exit", description: "Exit the TT shell." }
@@ -1994,8 +1995,8 @@ async function discoverShellContext(options = {}) {
   if (env.TT_TARGET && !environmentTarget) {
     warnings.push(`Ignoring invalid TT_TARGET=${JSON.stringify(env.TT_TARGET)}; use cloud or local.`);
   }
-  const inferredTarget = environmentTarget ?? (hasAdjacentLocalConfig ? "local" : "cloud");
-  const targetSource = environmentTarget ? "environment" : hasAdjacentLocalConfig ? "adjacent-config" : "default-cloud";
+  const inferredTarget = environmentTarget ?? "local";
+  const targetSource = environmentTarget ? "environment" : "default-local";
   const [activeModelId, latestRun] = await Promise.all([
     readActiveModelId(storeRoot),
     readLatestRun(storeRoot)
@@ -2037,15 +2038,13 @@ function targetSourceLabel(source) {
   switch (source) {
     case "environment":
       return "TT_TARGET";
-    case "adjacent-config":
-      return "adjacent local-runner.json";
     default:
       return "default";
   }
 }
 function formatShellContext(context, selectedTarget, targetSource) {
   const specLabel = context.spec ? context.spec.parseError ? `${context.spec.path} (invalid JSON)` : context.spec.path : "not found";
-  const sourceLabel = targetSource === "session" ? "session" : targetSource === "command-option" ? "--target" : targetSourceLabel(targetSource);
+  const sourceLabel = targetSource === "command-option" ? "--target" : targetSourceLabel(targetSource);
   const lines = [
     `Target         ${selectedTarget} (${sourceLabel})`,
     `Project        ${context.projectName}`,
@@ -2087,6 +2086,103 @@ function formatShellStatus(context, selectedTarget) {
   return lines;
 }
 
+// src/agent-control.ts
+function modelKey(model) {
+  return `${model.provider}/${model.id}`;
+}
+function modelRelevance(model, query) {
+  const q = query.trim().toLowerCase();
+  if (!q) return void 0;
+  const fullId = modelKey(model).toLowerCase();
+  const bareId = model.id.toLowerCase();
+  const name = model.name.toLowerCase();
+  if (fullId === q || bareId === q) return 0;
+  if (fullId.startsWith(q)) return 1;
+  if (bareId.startsWith(q)) return 2;
+  if (fullId.includes(q)) return 3;
+  if (name.includes(q)) return 4;
+  return void 0;
+}
+function listAgentModels(runtime, options = {}) {
+  let models = runtime.getModels(options.provider).filter(
+    (model) => options.includeUnauthenticated || runtime.hasConfiguredAuth(model.provider)
+  ).map((model) => ({
+    provider: model.provider,
+    id: model.id,
+    name: model.name ?? model.id,
+    authenticated: runtime.hasConfiguredAuth(model.provider),
+    thinking: model.reasoning !== false
+  }));
+  const query = options.query?.trim().toLowerCase();
+  if (query) {
+    const scored = [];
+    for (const model of models) {
+      const score = modelRelevance(model, query);
+      if (score !== void 0) scored.push({ model, score });
+    }
+    scored.sort(
+      (left, right) => left.score - right.score || modelKey(left.model).localeCompare(modelKey(right.model))
+    );
+    models = scored.map((entry) => entry.model);
+  } else {
+    models.sort((left, right) => modelKey(left).localeCompare(modelKey(right)));
+  }
+  if (options.limit !== void 0 && options.limit > 0) {
+    models = models.slice(0, options.limit);
+  }
+  return models;
+}
+function describeAgentModel(runtime, env) {
+  const selection = getAgentSelection(env);
+  if (!selection) return void 0;
+  const resolved = resolveAgentModelDefinition(runtime, selection);
+  const providerName = runtime.getProviders().find((provider) => provider.id === selection.provider)?.name;
+  return {
+    provider: selection.provider,
+    providerName,
+    model: selection.model,
+    modelName: resolved.model.name,
+    thinking: selection.thinking,
+    authenticated: runtime.hasConfiguredAuth(selection.provider),
+    supportsThinking: resolved.model.reasoning !== false
+  };
+}
+function setAgentModel(runtime, env, provider, model, options = {}) {
+  const current = getAgentSelection(env);
+  const requestedThinking = options.thinking ?? current?.thinking ?? "medium";
+  if (!AGENT_THINKING_LEVELS.includes(requestedThinking)) {
+    throw new Error(
+      `Agent thinking must be one of: ${AGENT_THINKING_LEVELS.join(", ")}`
+    );
+  }
+  const adjust = options.adjustThinking ?? true;
+  if (!adjust) {
+    const selection2 = {
+      provider,
+      model,
+      thinking: requestedThinking
+    };
+    resolveAgentModel(runtime, selection2);
+    updateConfig({ agent: selection2 });
+    return { selection: selection2, adjustedThinking: false };
+  }
+  const definition = resolveAgentModelDefinition(runtime, {
+    provider,
+    model,
+    thinking: "off"
+  });
+  const supportsThinking = definition.model.reasoning !== false;
+  const effectiveThinking = supportsThinking ? requestedThinking : "off";
+  const selection = {
+    provider,
+    model,
+    thinking: effectiveThinking
+  };
+  resolveAgentModel(runtime, selection);
+  updateConfig({ agent: selection });
+  return { selection, adjustedThinking: effectiveThinking !== requestedThinking };
+}
+
 // src/shell.ts
 var AGENT_SLASH_COMMANDS = /* @__PURE__ */ new Set([
   "new",
@@ -2217,7 +2313,7 @@ function routeShellCommand(input, activeMode) {
   const override = args2[0];
   if (override === "cloud" || override === "local") {
     if (args2.length === 1) {
-      throw new ShellParseError(`Add a ${override} command, or use \`/mode ${override}\` to switch workflows.`);
+      throw new ShellParseError(`Add a ${override} command, e.g. \`${override} runs list\`.`);
     }
     return { target: override, args: args2.slice(1) };
   }
@@ -2246,9 +2342,6 @@ var SLASH_NAMES = /* @__PURE__ */ new Set([
   "help",
   "status",
   "context",
-  "mode",
-  "cloud",
-  "local",
   "model",
   "clear",
   "cd",
@@ -2410,7 +2503,7 @@ function helpText(mode, query, palette = false) {
 `;
 }
 function activeModelLabel(snapshot) {
-  return snapshot.mode === "local" ? snapshot.context.local.activeModelId ?? "base" : snapshot.context.spec?.baseModel ?? "\u2014";
+  return snapshot.context.local.activeModelId ?? "base";
 }
 function agentModelLabel(context) {
   const agent = context.agent;
@@ -2465,20 +2558,19 @@ function resetShellPromptStyle() {
   return "";
 }
 var TunedTensorShellSession = class _TunedTensorShellSession {
-  constructor(runner, io, agent, env, contextProvider, version, initialContext) {
+  constructor(runner, io, agent, env, contextProvider, version, agentModelRuntime, initialContext) {
     this.runner = runner;
     this.io = io;
     this.agent = agent;
     this.env = env;
     this.contextProvider = contextProvider;
     this.version = version;
-    this.mode = initialContext.inferredTarget;
-    this.modeSource = initialContext.targetSource;
+    this.agentModelRuntime = agentModelRuntime;
     this.cwd = initialContext.cwd;
     this.context = initialContext;
   }
-  mode;
-  modeSource;
+  mode = "local";
+  modeSource = "default-local";
   cwd;
   context;
   static async create(options) {
@@ -2493,6 +2585,7 @@ var TunedTensorShellSession = class _TunedTensorShellSession {
       env,
       contextProvider,
       options.version,
+      options.agentModelRuntime,
       initialContext
     );
   }
@@ -2517,26 +2610,67 @@ var TunedTensorShellSession = class _TunedTensorShellSession {
   async refreshContext() {
     this.context = await this.contextProvider({ cwd: this.cwd, env: this.env });
     this.cwd = this.context.cwd;
-    if (this.modeSource !== "session" && this.modeSource !== "environment") {
-      this.mode = this.context.inferredTarget;
-      this.modeSource = this.context.targetSource;
-    }
   }
   writeLines(lines) {
     this.io.write(`${lines.join("\n")}
 `);
   }
-  modelLines() {
-    if (this.mode === "local") {
-      return styleDetailLines([
-        detailLine("Active model", this.context.local.activeModelId ?? "base"),
-        detailLine("Change", "/model <id> to activate a verified local model")
-      ]);
+  async requireAgentModelRuntime() {
+    if (!this.agentModelRuntime) {
+      throw new ShellParseError("The local agent model runtime is unavailable.");
     }
-    return styleDetailLines([
-      detailLine("Base model", this.context.spec?.baseModel ?? "\u2014"),
-      detailLine("Change", "edit base_model in tunedtensor.json, then run push")
+    return await this.agentModelRuntime();
+  }
+  renderAgentModel(runtime, query) {
+    let summary;
+    try {
+      summary = describeAgentModel(runtime, this.env);
+    } catch {
+      summary = void 0;
+    }
+    const label = summary ? `${summary.provider}/${summary.model}${summary.modelName && summary.modelName !== summary.model ? ` (${summary.modelName})` : ""} \u2014 thinking ${summary.thinking}, auth ${summary.authenticated ? "configured" : "required"}` : "not configured";
+    const lines = styleDetailLines([
+      detailLine("Agent model", label),
+      detailLine("Change", "/model <provider>/<model>"),
+      detailLine("Search", "/model <query>")
     ]);
+    const all = listAgentModels(runtime);
+    if (all.length === 0) {
+      lines.push("", chalk.dim("  no authenticated models \u2014 run `tt agent models --all` for the full catalog"));
+      return lines;
+    }
+    if (query) {
+      const matches = listAgentModels(runtime, { query });
+      lines.push("", chalk.bold(`Models matching ${JSON.stringify(query)}`));
+      if (matches.length === 0) {
+        lines.push(chalk.dim("  no matches"));
+      } else {
+        const shown2 = matches.slice(0, 10);
+        for (const model of shown2) {
+          lines.push(this.formatAgentModelChoice(model, summary));
+        }
+        if (matches.length > shown2.length) {
+          lines.push(chalk.dim(`  \u2026 ${matches.length - shown2.length} more \u2014 refine the query`));
+        }
+      }
+      return lines;
+    }
+    const preferred = summary ? all.filter((model) => model.provider === summary.provider) : [];
+    const rest = summary ? all.filter((model) => model.provider !== summary.provider) : all;
+    const shown = [...preferred, ...rest].slice(0, 8);
+    lines.push("", chalk.bold(summary ? "Suggestions" : "Available models"));
+    for (const model of shown) {
+      lines.push(this.formatAgentModelChoice(model, summary));
+    }
+    if (all.length > shown.length) {
+      lines.push(chalk.dim(`  \u2026 ${all.length - shown.length} more \u2014 /model <query> to search`));
+    }
+    return lines;
+  }
+  formatAgentModelChoice(model, summary) {
+    const active = summary?.provider === model.provider && summary?.model === model.id;
+    const displayName = model.name !== model.id ? `  ${model.name}` : "";
+    return `  ${accent(`${model.provider}/${model.id}`)}${displayName}${active ? chalk.dim(" (active)") : ""}`;
   }
   async handleSlash(command) {
     switch (command.name) {
@@ -2558,49 +2692,37 @@ var TunedTensorShellSession = class _TunedTensorShellSession {
           formatShellContext(this.context, this.mode, this.modeSource)
         ));
         return "continue";
-      case "mode": {
-        if (command.args.length === 0) {
-          this.io.write(`Current workflow: ${this.mode}
-`);
-          return "continue";
-        }
-        if (command.args.length !== 1 || !["cloud", "local"].includes(command.args[0])) {
-          throw new ShellParseError("Usage: /mode cloud|local");
-        }
-        this.mode = command.args[0];
-        this.modeSource = "session";
-        this.io.write(`${successMark()} Workflow switched to ${this.mode}.
-`);
-        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": {
+        const runtime = await this.requireAgentModelRuntime();
         if (command.args.length === 0) {
           await this.refreshContext();
-          this.writeLines(this.modelLines());
+          this.writeLines(this.renderAgentModel(runtime));
           return "continue";
         }
         if (command.args.length !== 1) {
-          throw new ShellParseError("Usage: /model [model-id]");
+          throw new ShellParseError("Usage: /model [<query> | <provider>/<model>]");
         }
-        if (this.mode !== "local") {
+        const target = command.args[0];
+        const slash = target.indexOf("/");
+        if (slash <= 0) {
+          await this.refreshContext();
+          this.writeLines(this.renderAgentModel(runtime, target));
+          return "continue";
+        }
+        if (slash === target.length - 1) {
           throw new ShellParseError(
-            "Activating models is a local workflow action. Use /mode local first; cloud specs change base_model in tunedtensor.json."
+            "Usage: /model <provider>/<model>, e.g. /model anthropic/claude-sonnet-4-5"
           );
         }
-        await this.runner({
-          target: "local",
-          args: ["models", "activate", command.args[0]],
-          cwd: this.cwd
-        });
+        const provider = target.slice(0, slash);
+        const model = target.slice(slash + 1);
+        const result2 = setAgentModel(runtime, this.env, provider, model);
         await this.refreshContext();
+        const note = result2.adjustedThinking ? " \u2014 thinking set to off for this model" : "";
+        this.io.write(
+          `${successMark()} Agent model: ${result2.selection.provider}/${result2.selection.model} (thinking ${result2.selection.thinking})${note}.
+`
+        );
         return "continue";
       }
       case "clear":
@@ -2740,7 +2862,8 @@ async function startInteractiveShell(options) {
     agent: options.agent,
     cwd: options.cwd,
     env: options.env,
-    version: options.version
+    version: options.version,
+    agentModelRuntime: options.agentModelRuntime
   });
   const completer = createCommandCompleter(() => session.snapshot().mode);
   readline = createInterface({
@@ -8110,61 +8233,47 @@ function registerPushCommand(parent) {
 }
 
 // src/commands/agent.ts
-function selectionOrHint(env) {
-  const selection = getAgentSelection(env);
-  if (!selection) {
-    throw new Error(
-      "The local agent is not configured. Run `tt agent configure --provider <provider> --model <model>`."
-    );
-  }
-  return selection;
-}
 function registerAgentCommands(parent, options) {
-  const agent = parent.command("agent").description("Configure the laptop-local Pi agent");
+  const agent = parent.command("agent").description("Configure the laptop-local TT agent");
   agent.command("status").description("Show the selected local provider, model, thinking, and auth state").action(async () => {
-    const selection = selectionOrHint(options.env);
     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 summary = describeAgentModel(runtime, options.env);
+    if (!summary) {
+      throw new Error(
+        "The local agent is not configured. Run `tt agent configure --provider <provider> --model <model>`."
+      );
+    }
     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
+      provider: summary.provider,
+      provider_name: summary.providerName,
+      model: summary.model,
+      model_name: summary.modelName,
+      thinking: summary.thinking,
+      authenticated: summary.authenticated,
+      supports_thinking: summary.supportsThinking
     };
-    const modelLabel = resolved.model.name && resolved.model.name !== selection.model ? `${resolved.model.name} (${selection.provider}/${selection.model})` : `${selection.provider}/${selection.model}`;
+    const modelLabel = summary.modelName && summary.modelName !== summary.model ? `${summary.modelName} (${summary.provider}/${summary.model})` : `${summary.provider}/${summary.model}`;
     options.output.write(isJsonMode() ? `${JSON.stringify(value, null, 2)}
-` : `Local agent: ${modelLabel} (thinking: ${value.thinking}, auth: ${authenticated ? "configured" : "required"})
+` : `Local agent: ${modelLabel} (thinking: ${value.thinking}, auth: ${value.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) => {
+  agent.command("models").description("List provider models available to the local agent").option("--provider <provider>", "Filter by provider ID").option("--all", "Include models whose provider auth is not configured").action(async (commandOptions) => {
     const runtime = await options.getRuntime();
     if (commandOptions.provider && !runtime.getProviders().some((provider) => provider.id === commandOptions.provider)) {
-      throw new Error(`Unknown Pi provider "${commandOptions.provider}".`);
-    }
-    const models = runtime.getModels(commandOptions.provider).filter(
-      (model) => commandOptions.all || runtime.hasConfiguredAuth(model.provider)
-    ).map((model) => ({
-      provider: model.provider,
-      id: model.id,
-      name: model.name ?? model.id,
-      authenticated: runtime.hasConfiguredAuth(model.provider),
-      thinking: model.reasoning !== false
-    }));
+      throw new Error(`Unknown provider "${commandOptions.provider}".`);
+    }
+    const models = listAgentModels(runtime, {
+      provider: commandOptions.provider,
+      includeUnauthenticated: commandOptions.all
+    });
     if (isJsonMode()) {
       options.output.write(`${JSON.stringify({ data: models }, null, 2)}
 `);
       return;
     }
     if (models.length === 0) {
-      options.output.write("No matching authenticated Pi models. Use --all to inspect the full catalog.\n");
+      options.output.write("No matching authenticated models. Use --all to inspect the full catalog.\n");
       return;
     }
     for (const model of models) {
@@ -8172,20 +8281,17 @@ function registerAgentCommands(parent, options) {
 `);
     }
   });
-  agent.command("configure").description("Select the local Pi provider, model, and thinking level").requiredOption("--provider <provider>", "Pi provider ID").requiredOption("--model <model>", "Pi model ID").option("--thinking <level>", "off, minimal, low, medium, high, xhigh, or max", "medium").action(async (commandOptions) => {
-    if (!AGENT_THINKING_LEVELS.includes(commandOptions.thinking)) {
-      throw new Error(`--thinking must be one of: ${AGENT_THINKING_LEVELS.join(", ")}`);
-    }
-    const selection = {
-      provider: commandOptions.provider,
-      model: commandOptions.model,
-      thinking: commandOptions.thinking
-    };
-    resolveAgentModel(await options.getRuntime(), selection);
-    updateConfig({ agent: selection });
-    const value = { execution: "local", ...selection };
+  agent.command("configure").description("Select the local provider, model, and thinking level").requiredOption("--provider <provider>", "Provider ID").requiredOption("--model <model>", "Model ID").option("--thinking <level>", "off, minimal, low, medium, high, xhigh, or max", "medium").action(async (commandOptions) => {
+    const result2 = setAgentModel(
+      await options.getRuntime(),
+      options.env,
+      commandOptions.provider,
+      commandOptions.model,
+      { thinking: commandOptions.thinking, adjustThinking: false }
+    );
+    const value = { execution: "local", ...result2.selection };
     options.output.write(isJsonMode() ? `${JSON.stringify(value, null, 2)}
-` : `Configured local agent: ${selection.provider}/${selection.model} (thinking: ${selection.thinking}).
+` : `Configured local agent: ${result2.selection.provider}/${result2.selection.model} (thinking: ${result2.selection.thinking}).
 `);
   });
 }
@@ -8355,14 +8461,14 @@ function formatCliUpdateNotice(update) {
 }
 
 // src/cli.ts
-async function createDefaultAgentClient(runtime, env, cloud, workspaceRoot) {
+async function createDefaultAgentClient(runtime, env, cloud, workspaceRoot, getModelRuntime) {
   const selection = getAgentSelection(env);
   if (!selection) {
     throw new Error(
       "The laptop-local agent is not configured. Run `tt agent models --all`, then `tt agent configure --provider <provider> --model <model>`."
     );
   }
-  const modelRuntime = runtime.modelRuntime ?? await createPiModelRuntime();
+  const modelRuntime = await getModelRuntime();
   const clientOpts = { apiKey: cloud.apiKey, baseUrl: cloud.baseUrl };
   const toolApi = runtime.agentToolApi ?? {
     get: async (path, query) => await get(path, query, clientOpts),
@@ -8395,11 +8501,23 @@ async function createDefaultAgentClient(runtime, env, cloud, workspaceRoot) {
     createAgent: runtime.createPiAgent
   });
 }
-function createLazyDefaultAgentClient(runtime, env, cloud, workspaceRoot) {
+function createModelRuntimeGetter(runtime) {
+  const injected = runtime.modelRuntime;
+  let promise;
+  return async () => {
+    if (injected) return injected;
+    promise ??= createPiModelRuntime();
+    return await promise;
+  };
+}
+function createLazyDefaultAgentClient(runtime, env, cloud, workspaceRoot, getModelRuntime) {
   let pending;
+  let builtRevision = -1;
   const client = () => {
-    if (!pending) {
-      const attempt = createDefaultAgentClient(runtime, env, cloud, workspaceRoot);
+    const revision = getConfigRevision();
+    if (!pending || builtRevision !== revision) {
+      builtRevision = revision;
+      const attempt = createDefaultAgentClient(runtime, env, cloud, workspaceRoot, getModelRuntime);
       pending = attempt;
       attempt.catch(() => {
         if (pending === attempt) pending = void 0;
@@ -8499,13 +8617,7 @@ function createProgram(version, runtime = {}) {
   const invokeLocal = runtime.runLocalCommand ?? executeLocalCommand;
   const launchShell = runtime.startShell ?? startInteractiveShell;
   const cwd = runtime.cwd ?? process.cwd();
-  const injectedModelRuntime = runtime.modelRuntime;
-  let modelRuntimePromise;
-  const getModelRuntime = async () => {
-    if (injectedModelRuntime) return injectedModelRuntime;
-    modelRuntimePromise ??= createPiModelRuntime();
-    return await modelRuntimePromise;
-  };
+  const getModelRuntime = createModelRuntimeGetter(runtime);
   const shellRunner = async (request2) => {
     const root = program.opts();
     const args2 = request2.target === "local" ? ["local", ...request2.args] : request2.args;
@@ -8524,7 +8636,7 @@ function createProgram(version, runtime = {}) {
       client: runtime.agentClient ?? createLazyDefaultAgentClient(runtime, shellEnvironment, {
         apiKey: root.apiKey ?? shellEnvironment.TUNED_TENSOR_API_KEY,
         baseUrl: root.baseUrl ?? shellEnvironment.TUNED_TENSOR_URL
-      }, cwd),
+      }, cwd, getModelRuntime),
       output,
       error
     });
@@ -8536,7 +8648,8 @@ function createProgram(version, runtime = {}) {
       error,
       cwd,
       env: shellEnvironment,
-      version
+      version,
+      agentModelRuntime: getModelRuntime
     });
   };
   program.name("tt").description("Tuned Tensor \u2014 converse, train, and inspect from one terminal").version(version).option("-k, --api-key <key>", "API key (overrides stored key)").option(
@@ -8640,7 +8753,7 @@ function createProgram(version, runtime = {}) {
     });
     applyExitCode(result2);
   });
-  program.command("shell").description("Open the interactive cloud/local terminal").action(openShell);
+  program.command("shell").description("Open the conversational terminal (local by default)").action(openShell);
   program.addHelpText(
     "after",
     `
@@ -8697,11 +8810,12 @@ async function runCli(version, runtime = {}) {
 `);
     }
     const invokeSelf = runtime.runSelfCommand ?? runSelfCommand;
+    const getModelRuntime = createModelRuntimeGetter(runtime);
     const agent = createShellAgent({
       client: runtime.agentClient ?? createLazyDefaultAgentClient(runtime, env, {
         apiKey: env.TUNED_TENSOR_API_KEY,
         baseUrl: env.TUNED_TENSOR_URL
-      }, runtime.cwd ?? process.cwd()),
+      }, runtime.cwd ?? process.cwd(), getModelRuntime),
       output,
       error
     });
@@ -8720,7 +8834,8 @@ async function runCli(version, runtime = {}) {
       error,
       cwd: runtime.cwd ?? process.cwd(),
       env,
-      version
+      version,
+      agentModelRuntime: getModelRuntime
     });
     return;
   }
@@ -8729,7 +8844,7 @@ async function runCli(version, runtime = {}) {
 }
 
 // src/index.ts
-runCli("0.12.0-beta.0").catch((err) => {
+runCli("0.12.0-beta.1").catch((err) => {
   reportError(err);
   process.exit(1);
 });