@tuned-tensor/cli 0.7.0 → 0.8.0

package/dist/index.js

@@ -948,6 +948,32 @@ function errorMessage(error) {
 var successMark = () => chalk.green("\u2713");
 var errorMark = () => chalk.red("\u2717");
 var accent = chalk.hex("#8B5CF6");
+var ANSI_BOLD = "\x1B[1m";
+var ANSI_BOLD_OFF = "\x1B[22m";
+var ANSI_CLEAR_TO_END = "\x1B[K";
+var ANSI_RESET = "\x1B[0m";
+function inputColorCodes() {
+  if (chalk.level === 0) return null;
+  if (chalk.level === 1) {
+    return {
+      background: "\x1B[100m",
+      accent: "\x1B[95m",
+      text: "\x1B[97m"
+    };
+  }
+  if (chalk.level === 2) {
+    return {
+      background: "\x1B[48;5;238m",
+      accent: "\x1B[38;5;183m",
+      text: "\x1B[38;5;255m"
+    };
+  }
+  return {
+    background: "\x1B[48;2;50;52;67m",
+    accent: "\x1B[38;2;196;181;253m",
+    text: "\x1B[38;2;245;243;255m"
+  };
+}
 var DETAIL_LABEL_WIDTH = 15;
 function styleDetailLines(lines) {
   return lines.map((line) => {
@@ -1012,30 +1038,38 @@ function helpText(mode, query, palette = false) {
 function activeModelLabel(snapshot) {
   return snapshot.mode === "local" ? snapshot.context.local.activeModelId ?? "base" : snapshot.context.spec?.baseModel ?? "\u2014";
 }
-function logoRows() {
-  const muted = chalk.dim("\u2588\u2588");
-  return [
-    `${chalk.hex("#A78BFA")("\u2588\u2588")} ${muted} ${muted}`,
-    `${muted} ${accent("\u2588\u2588")} ${muted}`,
-    `${muted} ${muted} ${chalk.hex("#7C3AED")("\u2588\u2588")}`
-  ];
-}
 function renderShellBanner(snapshot) {
   const spec = snapshot.context.spec?.name ?? snapshot.context.spec?.path ?? "no spec";
-  const heading = snapshot.version ? `${accent.bold("Tuned Tensor")} ${chalk.dim(`v${snapshot.version}`)}` : accent.bold("Tuned Tensor");
-  const textRows = [
+  const heading = snapshot.version ? `${accent.bold("tt")} ${chalk.dim(`v${snapshot.version}`)}` : accent.bold("tt");
+  const lines = [
     heading,
-    `${chalk.bold(snapshot.mode)} \xB7 ${snapshot.context.projectName} \xB7 ${spec} \xB7 model ${activeModelLabel(snapshot)}`,
-    chalk.dim("Ask TT anything \xB7 known commands run directly \xB7 /help for more")
+    chalk.dim(
+      `${snapshot.mode} \xB7 ${snapshot.context.projectName} \xB7 ${spec} \xB7 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.")
   ];
-  const logo = logoRows();
-  const lines = textRows.map((row, index) => `${logo[index]}  ${row}`);
   return `${lines.join("\n")}
 
 `;
 }
-function renderShellPrompt(snapshot) {
-  return `${accent("tt")} ${chalk.bold(snapshot.mode)} ${chalk.dim(snapshot.context.projectName)} \u203A `;
+function renderShellPrompt() {
+  const colors = inputColorCodes();
+  if (!colors) return "\u203A ";
+  return [
+    colors.background,
+    colors.accent,
+    ANSI_BOLD,
+    "\u203A",
+    ANSI_BOLD_OFF,
+    colors.text,
+    " ",
+    ANSI_CLEAR_TO_END
+  ].join("");
+}
+function resetShellPromptStyle() {
+  return chalk.level === 0 ? "" : ANSI_RESET;
 }
 var TunedTensorShellSession = class _TunedTensorShellSession {
   constructor(runner, io, agent, env, contextProvider, version, initialContext) {
@@ -1082,7 +1116,7 @@ var TunedTensorShellSession = class _TunedTensorShellSession {
     };
   }
   prompt() {
-    return renderShellPrompt(this.snapshot());
+    return renderShellPrompt();
   }
   banner() {
     return renderShellBanner(this.snapshot());
@@ -1312,10 +1346,12 @@ async function startInteractiveShell(options) {
   });
   readline.on("SIGINT", () => {
     if (session.interruptAgent()) {
-      output.write("\n");
+      output.write(`${resetShellPromptStyle()}
+`);
       return;
     }
-    output.write("\n");
+    output.write(`${resetShellPromptStyle()}
+`);
     readline?.setPrompt(session.prompt());
     readline?.write(null, { ctrl: true, name: "u" });
   });
@@ -1324,15 +1360,18 @@ async function startInteractiveShell(options) {
   readline.prompt();
   try {
     for await (const line of readline) {
+      output.write(resetShellPromptStyle());
       const action = await session.handleLine(line);
       if (action === "exit") {
         readline.close();
         break;
       }
+      output.write("\n");
       readline.setPrompt(session.prompt());
       readline.prompt();
     }
   } finally {
+    output.write(resetShellPromptStyle());
     readline.close();
   }
 }
@@ -1422,6 +1461,7 @@ var TunedTensorAgentSession = class {
   pendingActions = /* @__PURE__ */ new Map();
   activeRequest = null;
   responseStarted = false;
+  reasoningActive = false;
   lineOpen = false;
   get busy() {
     return this.activeRequest !== null;
@@ -1440,6 +1480,7 @@ var TunedTensorAgentSession = class {
   endOpenLine() {
     if (this.lineOpen) this.options.io.write("\n");
     this.lineOpen = false;
+    this.reasoningActive = false;
   }
   renderAction(action) {
     this.endOpenLine();
@@ -1472,15 +1513,32 @@ ${chalk2.yellow.bold("Approval required")}  ${chalk2.dim(action.id.slice(0, 8))}
     if (event.type === "text_delta") {
       const delta = stringValue2(payload.delta) ?? stringValue2(payload.text) ?? stringValue2(payload.content);
       if (!delta) return;
+      const followsReasoning = this.reasoningActive;
+      if (followsReasoning) this.endOpenLine();
       if (!this.responseStarted) {
-        this.options.io.write(`${accent2.bold("TT")}  `);
+        this.options.io.write(`
+${accent2.bold("tt")}  `);
         this.responseStarted = true;
         this.lineOpen = true;
+      } else if (followsReasoning) {
+        this.options.io.write("\n");
       }
       this.options.io.write(delta);
       this.lineOpen = !delta.endsWith("\n");
       return;
     }
+    if (event.type === "reasoning_delta") {
+      const delta = stringValue2(payload.delta) ?? stringValue2(payload.text) ?? stringValue2(payload.content);
+      if (!delta) return;
+      if (!this.reasoningActive) {
+        this.endOpenLine();
+        this.options.io.write("\n");
+        this.reasoningActive = true;
+      }
+      this.options.io.write(chalk2.dim.italic(delta));
+      this.lineOpen = !delta.endsWith("\n");
+      return;
+    }
     if (event.type === "tool_call") {
       this.endOpenLine();
       const name = stringValue2(payload.name) ?? stringValue2(isRecord(payload.tool_call) ? payload.tool_call.name : null) ?? "tool";
@@ -1538,6 +1596,7 @@ ${chalk2.yellow.bold("Approval required")}  ${chalk2.dim(action.id.slice(0, 8))}
     const controller = new AbortController();
     this.activeRequest = controller;
     this.responseStarted = false;
+    this.reasoningActive = false;
     this.lineOpen = false;
     try {
       const result = await operation(controller.signal);
@@ -1556,6 +1615,7 @@ ${chalk2.yellow.bold("Approval required")}  ${chalk2.dim(action.id.slice(0, 8))}
     } finally {
       this.activeRequest = null;
       this.responseStarted = false;
+      this.reasoningActive = false;
       this.lineOpen = false;
     }
   }
@@ -6989,7 +7049,7 @@ async function runCli(version, runtime = {}) {
 }
 
 // src/index.ts
-runCli("0.7.0").catch((err) => {
+runCli("0.8.0").catch((err) => {
   reportError(err);
   process.exit(1);
 });