phewsh 0.5.4 → 0.6.1

package/commands/ai.js

@@ -1,6 +1,7 @@
 const fs = require('fs');
 const path = require('path');
 const os = require('os');
+const { trackSap } = require('../lib/supabase');
 
 const CONFIG_PATH = path.join(os.homedir(), '.phewsh', 'config.json');
 const INTENT_DIR = path.join(process.cwd(), '.intent');
@@ -35,14 +36,10 @@ function buildSystemPrompt(intentFiles) {
   return `You are a focused execution assistant. The user has structured intent artifacts that define what they are building. Use these as your primary context for every response — stay aligned with their vision, plan, and next actions.\n\n${sections}`;
 }
 
-async function streamResponse(apiKey, systemPrompt, userPrompt) {
+async function streamResponse(apiKey, systemPrompt, userPrompt, userId, accessToken) {
+  const model = 'claude-sonnet-4-6';
   const messages = [{ role: 'user', content: userPrompt }];
-  const body = {
-    model: 'claude-sonnet-4-6',
-    max_tokens: 1024,
-    messages,
-    stream: true,
-  };
+  const body = { model, max_tokens: 1024, messages, stream: true };
   if (systemPrompt) body.system = systemPrompt;
 
   const response = await fetch('https://api.anthropic.com/v1/messages', {
@@ -62,6 +59,9 @@ async function streamResponse(apiKey, systemPrompt, userPrompt) {
 
   process.stdout.write('\n');
 
+  let promptTokens = null;
+  let completionTokens = null;
+
   for await (const chunk of response.body) {
     const text = Buffer.from(chunk).toString('utf-8');
     const lines = text.split('\n').filter(l => l.startsWith('data: '));
@@ -73,11 +73,21 @@ async function streamResponse(apiKey, systemPrompt, userPrompt) {
         if (parsed.type === 'content_block_delta' && parsed.delta?.text) {
           process.stdout.write(parsed.delta.text);
         }
+        // Capture token counts from Anthropic stream
+        if (parsed.type === 'message_start' && parsed.message?.usage) {
+          promptTokens = parsed.message.usage.input_tokens;
+        }
+        if (parsed.type === 'message_delta' && parsed.usage) {
+          completionTokens = parsed.usage.output_tokens;
+        }
       } catch { /* skip malformed chunks */ }
     }
   }
 
   process.stdout.write('\n\n');
+
+  // SAP: fire-and-forget after stream completes
+  trackSap({ userId, source: 'cli', model, promptTokens, completionTokens, accessToken });
 }
 
 async function main() {
@@ -135,7 +145,7 @@ async function main() {
       console.log('\n  No .intent/ found — running without project context');
     }
 
-    await streamResponse(config.apiKey, systemPrompt, prompt);
+    await streamResponse(config.apiKey, systemPrompt, prompt, config.supabaseUserId, config.supabaseAccessToken);
     return;
   }
 

package/lib/supabase.js

@@ -77,4 +77,53 @@ async function upsert(table, data, accessToken) {
   return res.json();
 }
 
-module.exports = { sendOtp, verifyOtp, refreshSession, select, upsert, SUPABASE_URL, SUPABASE_ANON_KEY };
+// SAP: fire-and-forget per-inference tracking
+// Session ID — unique per CLI invocation, enables multi-prompt correlation
+const { randomUUID } = require('crypto');
+const CLI_SESSION_ID = randomUUID();
+
+// kWh estimates by model family (conservative)
+const MODEL_KWH = {
+  'claude-haiku':   0.00025,
+  'claude-sonnet':  0.00060,
+  'claude-opus':    0.00200,
+  'gemini-flash':   0.00020,
+  'deepseek':       0.00030,
+  'kimi':           0.00040,
+};
+
+function estimateKwh(model, totalTokens) {
+  let base = 0.0004;
+  if (model) {
+    const lower = model.toLowerCase();
+    for (const [key, val] of Object.entries(MODEL_KWH)) {
+      if (lower.includes(key)) { base = val; break; }
+    }
+  }
+  const scale = Math.max(0.5, Math.min(4, (totalTokens || 1000) / 1000));
+  return base * scale;
+}
+
+function trackSap({ userId, source = 'cli', model, promptTokens, completionTokens, accessToken } = {}) {
+  const totalTokens = (promptTokens || 0) + (completionTokens || 0);
+  const kwh = estimateKwh(model, totalTokens);
+  const co2_g = kwh * 500;   // US grid avg gCO2/kWh
+  const water_ml = kwh * 1800;
+
+  req('/rest/v1/rpc/track_sap_event', {
+    method: 'POST',
+    body: JSON.stringify({
+      p_user_id:           userId || null,
+      p_session_id:        CLI_SESSION_ID,
+      p_source:            source,
+      p_model:             model || null,
+      p_prompt_tokens:     promptTokens || null,
+      p_completion_tokens: completionTokens || null,
+      p_kwh:               kwh,
+      p_co2_g:             co2_g,
+      p_water_ml:          water_ml,
+    }),
+  }, accessToken).catch(() => { /* never surface SAP errors */ });
+}
+
+module.exports = { sendOtp, verifyOtp, refreshSession, select, upsert, trackSap, SUPABASE_URL, SUPABASE_ANON_KEY };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "phewsh",
-  "version": "0.5.4",
+  "version": "0.6.1",
   "description": "Turn intent into action. Structure your thinking, execute your next step.",
   "bin": {
     "phewsh": "bin/phewsh.js"