wolverine-ai 3.9.0 → 3.9.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wolverine-ai",
-  "version": "3.9.0",
+  "version": "3.9.1",
   "description": "Self-healing Node.js server framework powered by AI. Catches crashes, diagnoses errors, generates fixes, verifies, and restarts — automatically.",
   "main": "src/index.js",
   "bin": {

package/src/agent/agent-engine.js

@@ -446,6 +446,10 @@ class AgentEngine {
         return { success: false, summary: "Token budget exhausted", filesModified: this.filesModified, turnCount: this.turnCount, totalTokens: this.totalTokens };
       }
 
+      if (!response.choices || !response.choices[0]) {
+        console.log(chalk.red(`  ⚠️  AI returned no choices: ${JSON.stringify(response).slice(0, 200)}`));
+        return { success: false, summary: "AI returned empty response", filesModified: this.filesModified, turnCount: this.turnCount, totalTokens: this.totalTokens };
+      }
       const choice = response.choices[0];
       const assistantMessage = choice.message || choice;
       this.messages.push(assistantMessage);

package/src/core/ai-client.js

@@ -615,13 +615,27 @@ ${backupSourceCode ? `## Last Known Working Version\n\`\`\`javascript\n${backupS
 Include both if needed, or just one.`;
 
   const result = await aiCall({ model, systemPrompt, userPrompt, maxTokens: 2048, category: "coding" });
-  const content = result.content;
-  const cleaned = content.replace(/^```(?:json)?\n?/, "").replace(/\n?```$/, "");
+  const content = (result.content || "").trim();
+
+  // Strip thinking tags (Gemma), markdown fences, and any prefix text
+  let cleaned = content
+    .replace(/<\|channel>.*?<channel\|>/gs, "")
+    .replace(/<\|think\|>[\s\S]*?<\|\/think\|>/g, "")
+    .replace(/^```(?:json)?\s*/gm, "")
+    .replace(/```\s*$/gm, "")
+    .trim();
+
+  // Extract JSON object from response
+  const jsonMatch = cleaned.match(/\{[\s\S]*"(?:explanation|changes|commands)"[\s\S]*\}/);
+  if (jsonMatch) cleaned = jsonMatch[0];
 
   try {
     return JSON.parse(cleaned);
-  } catch (parseErr) {
-    throw new Error(`Failed to parse AI response as JSON.\nRaw response:\n${content}`);
+  } catch {
+    // Last resort: try to find any JSON object
+    const anyJson = cleaned.match(/\{[\s\S]*\}/);
+    if (anyJson) try { return JSON.parse(anyJson[0]); } catch {}
+    throw new Error(`AI response was not valid JSON`);
   }
 }