polydev-ai 1.8.5 → 1.8.7

package/lib/cliManager.js

@@ -785,21 +785,25 @@ This is a known issue with @google/gemini-cli@0.3.4 and older Node.js versions.`
       const content = json.result;
       
       // Extract primary model from modelUsage
-      // Priority: Opus > Sonnet > Haiku (prefer larger models as they do the main reasoning)
+      // The primary model is the one with highest cost - that's the user's configured main model
+      // (Haiku is used internally for quick tasks, but the expensive model is what the user chose)
       let primaryModel = 'cli_default';
       const modelUsage = json.modelUsage || {};
       const modelNames = Object.keys(modelUsage);
       
       if (modelNames.length === 1) {
-        // Only one model used
+        // Only one model used - that's the primary
         primaryModel = modelNames[0];
       } else if (modelNames.length > 1) {
-        // Multiple models - prefer Opus > Sonnet > Haiku
-        const opusModel = modelNames.find(m => m.includes('opus'));
-        const sonnetModel = modelNames.find(m => m.includes('sonnet'));
-        const haikuModel = modelNames.find(m => m.includes('haiku'));
-        
-        primaryModel = opusModel || sonnetModel || haikuModel || modelNames[0];
+        // Multiple models - the one with highest cost is the user's configured main model
+        let highestCost = -1;
+        for (const [modelName, usage] of Object.entries(modelUsage)) {
+          const cost = usage.costUSD || 0;
+          if (cost > highestCost) {
+            highestCost = cost;
+            primaryModel = modelName;
+          }
+        }
       }
       
       // Calculate total tokens

package/mcp/stdio-wrapper.js

@@ -826,7 +826,11 @@ class StdioMCPWrapper {
     if (successfulClis.length > 0) {
       // Show successful CLI responses
       for (const cliResult of successfulClis) {
-        formatted += `🟢 **Local CLI Response** (${cliResult.provider_id} - ${cliResult.mode || 'args'} mode)\n\n`;
+        // Display actual model name if detected, otherwise show provider ID
+        const modelDisplay = cliResult.model_used && cliResult.model_used !== 'cli_default' && cliResult.model_used !== 'cli_default_fallback'
+          ? cliResult.model_used 
+          : cliResult.provider_id;
+        formatted += `🟢 **Local CLI Response** (${modelDisplay})\n\n`;
         formatted += `${cliResult.content}\n\n`;
         formatted += `*Latency: ${cliResult.latency_ms || 0}ms | Tokens: ${cliResult.tokens_used || 0}*\n\n`;
         formatted += `---\n\n`;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "polydev-ai",
-  "version": "1.8.5",
+  "version": "1.8.7",
   "description": "Agentic workflow assistant with CLI integration - get diverse perspectives from multiple LLMs when stuck or need enhanced reasoning",
   "keywords": [
     "mcp",
@@ -67,7 +67,7 @@
     "lucide-react": "^0.542.0",
     "marked": "^16.2.1",
     "next": "^15.5.7",
-    "polydev-ai": "^1.8.4",
+    "polydev-ai": "^1.8.7",
     "posthog-js": "^1.157.2",
     "prismjs": "^1.30.0",
     "react": "^18.3.1",