polydev-ai 1.8.4 → 1.8.6

package/lib/cliManager.js

@@ -46,7 +46,7 @@ class CLIManager {
           chat: [],
           version: ['--version'],
           auth_status: ['--print', 'test auth'], // Use --print to test auth
-          test_prompt: ['--print']
+          test_prompt: ['--print', '--output-format', 'json'] // Use JSON output to get model info
         },
         install_instructions: 'Install via: npm install -g @anthropic-ai/claude-code',
         auth_instructions: 'Authenticate with Claude Code'
@@ -510,6 +510,27 @@ This is a known issue with @google/gemini-cli@0.3.4 and older Node.js versions.`
           );
 
           if (!result.error) {
+            // Special handling for Claude Code JSON output
+            if (providerId === 'claude_code') {
+              const jsonResult = this.parseClaudeCodeJsonResponse(result.stdout || '');
+              if (jsonResult) {
+                console.log(`[Polydev CLI] Claude Code detected model: ${jsonResult.model_used} (from JSON output)`);
+                return {
+                  success: true,
+                  content: jsonResult.content,
+                  tokens_used: jsonResult.tokens_used,
+                  latency_ms: Date.now() - startTime,
+                  provider: providerId,
+                  mode: 'args',
+                  model_used: jsonResult.model_used,
+                  cost_usd: jsonResult.cost_usd,
+                  model_usage: jsonResult.model_usage,
+                  timestamp: new Date()
+                };
+              }
+            }
+            
+            // Fallback: Standard text output handling
             const content = this.cleanCliResponse(result.stdout || '');
             
             // Detect actual model from CLI output
@@ -747,6 +768,66 @@ This is a known issue with @google/gemini-cli@0.3.4 and older Node.js versions.`
     return null; // Could not detect model from output
   }
 
+  // Parse Claude Code JSON response to extract model and content
+  parseClaudeCodeJsonResponse(stdout) {
+    if (!stdout || !stdout.trim()) return null;
+    
+    try {
+      // Try to parse as JSON
+      const json = JSON.parse(stdout.trim());
+      
+      // Check if it's a valid Claude Code response
+      if (json.type !== 'result' || !json.result) {
+        return null;
+      }
+      
+      // Extract content
+      const content = json.result;
+      
+      // Extract primary model from modelUsage
+      // 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 - that's the primary
+        primaryModel = modelNames[0];
+      } else if (modelNames.length > 1) {
+        // 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
+      let totalTokens = 0;
+      for (const usage of Object.values(modelUsage)) {
+        totalTokens += (usage.inputTokens || 0) + (usage.outputTokens || 0) + 
+                       (usage.cacheReadInputTokens || 0) + (usage.cacheCreationInputTokens || 0);
+      }
+      
+      return {
+        content,
+        model_used: primaryModel,
+        tokens_used: totalTokens || json.usage?.input_tokens + json.usage?.output_tokens || 0,
+        cost_usd: json.total_cost_usd || 0,
+        model_usage: modelUsage,
+        session_id: json.session_id,
+        duration_ms: json.duration_ms
+      };
+    } catch (e) {
+      // Not valid JSON, return null to fall back to text parsing
+      return null;
+    }
+  }
+
   async executeCodexExec(executable, commandArgs, prompt, timeoutMs, model = null) {
     if (!executable) {
       throw new Error('Missing Codex executable');

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "polydev-ai",
-  "version": "1.8.4",
+  "version": "1.8.6",
   "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.3",
+    "polydev-ai": "^1.8.5",
     "posthog-js": "^1.157.2",
     "prismjs": "^1.30.0",
     "react": "^18.3.1",