novaprime 1.3.4 → 1.3.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "novaprime",
-  "version": "1.3.4",
+  "version": "1.3.5",
   "description": "NovaPrime — an AI coding assistant in your terminal, powered by GLM.",
   "bin": {
     "novaprime": "bin/novaprime.js"

package/src/agent.js

@@ -51,7 +51,7 @@ async function streamMessage(server, key, messages) {
   }
 
   const blocks = [];
-  let stopReason = null, labelShown = false, buffer = '';
+  let stopReason = null, labelShown = false, buffer = '', usedModel = null;
   const renderer = new Renderer();
   const decoder = new TextDecoder();
   const reader = res.body.getReader();
@@ -67,22 +67,25 @@ async function streamMessage(server, key, messages) {
       if (!dataLine) continue;
       let json; try { json = JSON.parse(dataLine.slice(5).trim()); } catch (_) { continue; }
 
-      if (json.type === 'content_block_start') {
+      if (json.type === 'message_start') {
+        usedModel = (json.message && json.message.model) || usedModel;
+        if (spinning) spinner.text = c.muted('thinking · ' + (usedModel || ''));
+      } else if (json.type === 'content_block_start') {
         const cb = json.content_block || {};
         blocks[json.index] = cb.type === 'tool_use'
           ? { type: 'tool_use', id: cb.id, name: cb.name, _json: '' }
           : { type: 'text', text: '' };
-        if (cb.type === 'tool_use') ensureSpin(c.muted('preparing ' + cb.name + '…'));
+        if (cb.type === 'tool_use') ensureSpin(c.muted('writing with ' + (usedModel || 'model') + '…'));
       } else if (json.type === 'content_block_delta') {
         const b = blocks[json.index]; if (!b) continue;
         if (json.delta.type === 'text_delta') {
           stopSpin();
-          if (!labelShown) { aiLabel(); labelShown = true; }
+          if (!labelShown) { aiLabel(usedModel); labelShown = true; }
           b.text += json.delta.text;
           renderer.feed(json.delta.text);
         } else if (json.delta.type === 'input_json_delta') {
           b._json += json.delta.partial_json;
-          ensureSpin(c.muted((b.name === 'run_command' ? 'preparing command' : 'writing ' + (b.name || 'file')) + '… ' + b._json.length + ' chars'));
+          ensureSpin(c.muted('writing with ' + (usedModel || 'model') + '… ' + b._json.length + ' chars'));
         }
       } else if (json.type === 'message_delta') {
         if (json.delta && json.delta.stop_reason) stopReason = json.delta.stop_reason;

package/src/ui.js

@@ -98,7 +98,7 @@ function inputBoxOpen() {
 }
 function inputBoxClose() { process.stdout.write('\n'); }
 
-function aiLabel() { process.stdout.write('\n' + c.violet('● ') + c.violet.bold('Nova Prime') + '\n'); }
+function aiLabel(model) { process.stdout.write('\n' + c.violet('● ') + c.violet.bold('Nova Prime') + (model ? c.dim('   · ' + model) : '') + '\n'); }
 
 function info(msg) { console.log(c.muted(msg)); }
 function hint(msg) { console.log(c.dim(msg)); }