novaprime 1.3.1 → 1.3.3

package/package.json

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

package/src/agent.js

@@ -2,7 +2,7 @@
 const os = require('os');
 const ora = require('ora');
 const tools = require('./tools');
-const { c, aiLabel, error } = require('./ui');
+const { c, aiLabel, error, warn } = require('./ui');
 const { Renderer } = require('./render');
 
 const SYSTEM_PROMPT =
@@ -36,7 +36,7 @@ async function streamMessage(server, key, messages) {
     res = await fetch(server.replace(/\/$/, '') + '/v1/messages', {
       method: 'POST',
       headers: { 'content-type': 'application/json', 'x-novaprime-key': key },
-      body: JSON.stringify({ max_tokens: 8192, system: SYSTEM_PROMPT, tools: tools.definitions, messages, stream: true }),
+      body: JSON.stringify({ max_tokens: 16000, system: SYSTEM_PROMPT, tools: tools.definitions, messages, stream: true }),
     });
   } catch (err) { stopSpin(); return { error: 'Could not reach NovaPrime: ' + err.message }; }
 
@@ -56,7 +56,6 @@ async function streamMessage(server, key, messages) {
   while (true) {
     const { done, value } = await reader.read();
     if (done) break;
-    stopSpin();
     buffer += decoder.decode(value, { stream: true });
     let idx;
     while ((idx = buffer.indexOf('\n\n')) !== -1) {
@@ -66,17 +65,21 @@ async function streamMessage(server, key, messages) {
       let json; try { json = JSON.parse(dataLine.slice(5).trim()); } catch (_) { continue; }
 
       if (json.type === 'content_block_start') {
-        blocks[json.index] = json.content_block.type === 'tool_use'
-          ? { type: 'tool_use', id: json.content_block.id, name: json.content_block.name, _json: '' }
+        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' && spinning) spinner.text = c.muted('preparing ' + cb.name + '…');
       } 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; }
           b.text += json.delta.text;
           renderer.feed(json.delta.text);
         } else if (json.delta.type === 'input_json_delta') {
           b._json += json.delta.partial_json;
+          if (spinning) spinner.text = c.muted((b.name === 'run_command' ? 'preparing command' : 'writing ' + (b.name || 'file')) + '… ' + b._json.length + ' chars');
         }
       } else if (json.type === 'message_delta') {
         if (json.delta && json.delta.stop_reason) stopReason = json.delta.stop_reason;
@@ -93,7 +96,7 @@ async function streamMessage(server, key, messages) {
     }
     return { type: 'text', text: b.text };
   });
-  return { content, stopReason };
+  return { content, stopReason, printed: labelShown };
 }
 
 async function runTurn(server, key, messages) {
@@ -112,6 +115,12 @@ async function runTurn(server, key, messages) {
       messages.push({ role: 'user', content: toolResults });
       continue;
     }
+    // never finish silently
+    if (result.stopReason === 'max_tokens') {
+      warn('The response was too long and got cut off. Try a smaller or more specific request (e.g. one file at a time).');
+    } else if (!result.printed && !toolUses.length) {
+      warn('No response received. Please try again in a moment.');
+    }
     return;
   }
 }

package/src/tools.js

@@ -84,11 +84,13 @@ async function execute(name, input) {
         return items.join('\n') || '(empty)';
       }
       case 'write_file': {
-        console.log(c.warn('\n  ✎ write file: ') + c.bold(input.path) +
-          c.muted(`  (${(input.content || '').length} chars)`));
-        if (!(await confirm('Allow writing this file?'))) return 'DENIED: user did not allow writing the file.';
+        console.log('');
+        console.log(c.amber('  ╭─ permission ') + c.dim('─────────────────────────────'));
+        console.log(c.amber('  │ ') + c.bold('Write file') + '   ' + c.white(input.path) + c.muted(`  (${(input.content || '').length} chars)`));
+        if (!(await confirm(c.amber('  ╰─ allow?')))) { console.log(c.dim('  · skipped')); return 'DENIED: user did not allow writing the file.'; }
         fs.mkdirSync(path.dirname(path.resolve(input.path)), { recursive: true });
         fs.writeFileSync(input.path, input.content);
+        console.log(c.green('  ✓ wrote ') + c.white(input.path));
         return 'OK: wrote ' + input.path;
       }
       case 'edit_file': {
@@ -96,16 +98,21 @@ async function execute(name, input) {
         const count = before.split(input.old_string).length - 1;
         if (count === 0) return 'ERROR: old_string not found in ' + input.path;
         if (count > 1) return `ERROR: old_string matches ${count} times; make it more specific.`;
-        console.log(c.warn('\n  ✎ edit file: ') + c.bold(input.path));
-        console.log(c.danger('  - ' + input.old_string.split('\n').join('\n  - ')));
-        console.log(c.accent('  + ' + input.new_string.split('\n').join('\n  + ')));
-        if (!(await confirm('Allow this edit?'))) return 'DENIED: user did not allow the edit.';
+        console.log('');
+        console.log(c.amber('  ╭─ permission ') + c.dim('─────────────────────────────'));
+        console.log(c.amber('  │ ') + c.bold('Edit file') + '   ' + c.white(input.path));
+        console.log(c.red('  │ - ' + input.old_string.split('\n').join('\n  │ - ')));
+        console.log(c.green('  │ + ' + input.new_string.split('\n').join('\n  │ + ')));
+        if (!(await confirm(c.amber('  ╰─ allow?')))) { console.log(c.dim('  · skipped')); return 'DENIED: user did not allow the edit.'; }
         fs.writeFileSync(input.path, before.replace(input.old_string, input.new_string));
+        console.log(c.green('  ✓ edited ') + c.white(input.path));
         return 'OK: edited ' + input.path;
       }
       case 'run_command': {
-        console.log(c.warn('\n  $ ') + c.bold(input.command));
-        if (!(await confirm('Run this command?'))) return 'DENIED: user did not allow running the command.';
+        console.log('');
+        console.log(c.red('  ╭─ permission · run command ') + c.dim('───────────────────'));
+        console.log(c.red('  │ ') + c.bold(input.command));
+        if (!(await confirm(c.red('  ╰─ run this command?')))) { console.log(c.dim('  · skipped')); return 'DENIED: user did not allow running the command.'; }
         const r = spawnSync(input.command, { shell: true, encoding: 'utf8', timeout: 1000 * 120 });
         const out = (r.stdout || '') + (r.stderr || '');
         return clip(`exit_code=${r.status}\n${out}`.trim());