sapper-iq 1.0.21 → 1.0.23

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "sapper-iq",
-  "version": "1.0.21",
+  "version": "1.0.23",
   "description": "AI-powered development assistant that executes commands and builds projects",
   "main": "sapper.mjs",
   "bin": {

package/sapper.mjs

@@ -166,8 +166,10 @@ WORKFLOW:
     }];
   }
 
-  const ask = () => {
-    safeQuestion(chalk.blue.bold('\nIbrahim ➔ ')).then(async (input) => {
+  const ask = async () => {
+    try {
+      const input = await safeQuestion(chalk.blue.bold('\nIbrahim ➔ '));
+      
       if (input.toLowerCase() === 'exit') process.exit();
       
       // Handle reset command
@@ -180,17 +182,27 @@ WORKFLOW:
           role: 'system',
           content: messages[0].content // Keep system prompt
         }];
-        return ask();
+        return await ask();
       }
       
       messages.push({ role: 'user', content: input });
 
+      let toolRounds = 0; // Prevent infinite loops
+      const MAX_TOOL_ROUNDS = 5;
+      
       let active = true;
       while (active) {
         if (stepMode) await safeQuestion(chalk.gray('[STEP] Press Enter to let AI think...'));
         
         spinner.start('Thinking...');
-        const response = await ollama.chat({ model: selectedModel, messages, stream: true });
+        let response;
+        try {
+          response = await ollama.chat({ model: selectedModel, messages, stream: true });
+        } catch (ollamaError) {
+          spinner.stop();
+          console.error(chalk.red('\n❌ Ollama error:'), ollamaError.message);
+          return await ask();
+        }
         spinner.stop();
 
         let msg = '';
@@ -203,10 +215,21 @@ WORKFLOW:
         messages.push({ role: 'assistant', content: msg });
 
         // Fixed regex: .+? (non-greedy) stops correctly before [/TOOL]
-        // Old regex [^\]\n]+ was broken - it stopped at ] which is at END of [/TOOL]
         const toolMatches = [...msg.matchAll(/\[TOOL:(\w+)\](.+?)(?:\]([\s\S]*?))?\[\/TOOL\]/g)];
         
         if (toolMatches.length > 0) {
+          toolRounds++;
+          
+          // Prevent infinite tool loops
+          if (toolRounds >= MAX_TOOL_ROUNDS) {
+            console.log(chalk.yellow(`\n⚠️  Tool limit reached (${MAX_TOOL_ROUNDS} rounds). Stopping auto-execution.`));
+            messages.push({ 
+              role: 'user', 
+              content: 'STOP using tools now. You have enough information. Please provide your analysis based on what you have read.' 
+            });
+            continue; // Let AI respond without tools
+          }
+          
           for (const match of toolMatches) {
             const [_, type, path, content] = match;
             console.log(chalk.cyan(`\n[ACTION] ${type} -> ${path}`));
@@ -222,32 +245,31 @@ WORKFLOW:
           }
           fs.writeFileSync(CONTEXT_FILE, JSON.stringify(messages));
           
-          // Warn if reading many files at once
           if (toolMatches.length > 30) {
             console.log(chalk.yellow('\n⚠️  Reading 30+ files! This might take time.'));
           }
         } else {
           // No tools found - check if malformed command
           if (msg.includes('[TOOL:') && msg.includes('[/]')) {
-            console.log(chalk.red('\n❌ Malformed tool command detected! Expected format: [TOOL:TYPE]path[/TOOL]'));
+            console.log(chalk.red('\n❌ Malformed tool command detected!'));
             messages.push({ 
               role: 'user', 
               content: 'ERROR: Your tool command is malformed. Use [TOOL:TYPE]path]content[/TOOL] or [TOOL:TYPE]path[/TOOL]' 
             });
           } else {
-            // Normal response without tools - save context and wait for next input
+            // Normal response - save and wait for next input
             fs.writeFileSync(CONTEXT_FILE, JSON.stringify(messages));
             active = false;
           }
         }
       }
-      ask();
-    }).catch((error) => {
+    } catch (error) {
       console.error(chalk.red('\n❌ Error:'), error.message);
-      ask(); // Continue despite error
-    });
+    }
+    // ALWAYS call ask() again with await - keep the conversation going
+    await ask();
   };
-  ask();
+  await ask();
 }
 
 runSapper();