@costlens/mcp-server 0.4.3 → 0.6.0

package/dist/cli.js

@@ -39,31 +39,41 @@ function resolveApiKey() {
     return void 0;
   }
 }
+async function apiFetch(path, opts = {}) {
+  const headers = { "Authorization": `Bearer ${API_KEY}`, ...opts.headers };
+  if (opts.body) headers["Content-Type"] = "application/json";
+  return fetch(`${API_BASE}${path}`, { ...opts, headers, signal: AbortSignal.timeout(5e3) });
+}
 async function loadPricing() {
   try {
-    const headers = {};
-    if (API_KEY) headers["Authorization"] = `Bearer ${API_KEY}`;
-    const res = await fetch(`${API_BASE}/v1/pricing`, { headers, signal: AbortSignal.timeout(5e3) });
+    const res = await apiFetch("/v1/pricing");
     if (res.ok) {
       const data = await res.json();
-      for (const m of data.models) {
-        import_classifier.PRICING[m.model] = { input: m.input, output: m.output };
-      }
+      for (const m of data.models) import_classifier.PRICING[m.model] = { input: m.input, output: m.output };
     }
   } catch {
   }
 }
+async function loadKeySettings() {
+  if (!API_KEY) return;
+  try {
+    const res = await apiFetch("/v1/plan");
+    if (!res.ok) return;
+    const data = await res.json();
+    trackingMode = data.keySettings?.trackingMode || "manual";
+    productivityEnabled = data.features?.includes("productivity_basic") ?? false;
+  } catch {
+  }
+}
 function startHeartbeat(sessionId) {
   stopHeartbeat();
   activeSessionId = sessionId;
   heartbeatInterval = setInterval(async () => {
     if (!activeSessionId || !API_KEY) return;
     try {
-      const res = await fetch(`${API_BASE}/v1/productivity/heartbeat`, {
+      const res = await apiFetch("/v1/productivity/heartbeat", {
         method: "POST",
-        headers: { "Authorization": `Bearer ${API_KEY}`, "Content-Type": "application/json" },
-        body: JSON.stringify({ session_id: activeSessionId }),
-        signal: AbortSignal.timeout(5e3)
+        body: JSON.stringify({ session_id: activeSessionId })
       });
       if (res.status === 410) {
         activeSessionId = null;
@@ -79,58 +89,65 @@ function stopHeartbeat() {
     heartbeatInterval = null;
   }
 }
-async function syncEvent(result, params) {
+async function ensureAutoSession() {
+  if (trackingMode !== "auto" || !API_KEY || !productivityEnabled || activeSessionId) return;
   try {
-    await fetch(`${API_BASE}/v1/classifier/events`, {
+    const res = await apiFetch("/v1/productivity/sessions", {
       method: "POST",
-      headers: { "Authorization": `Bearer ${API_KEY}`, "Content-Type": "application/json" },
-      body: JSON.stringify({
-        complexity: result.level,
-        confidence: result.confidence,
-        suggestedModel: result.suggestedModel,
-        currentModel: params.currentModel,
-        provider: params.provider,
-        staticScore: result.signals.staticScore
-      }),
-      signal: AbortSignal.timeout(5e3)
+      body: JSON.stringify({ task_description: "Auto-tracked MCP session" })
     });
+    if (res.ok) {
+      const data = await res.json();
+      if (data.data?.id) startHeartbeat(data.data.id);
+    }
   } catch {
   }
 }
-async function checkFeature(feature) {
-  if (!API_KEY) return false;
+async function autoLogEvent(eventType, metadata) {
+  if (trackingMode !== "auto" || !activeSessionId || !API_KEY) return;
   try {
-    const res = await fetch(`${API_BASE}/v1/plan`, {
-      headers: { "Authorization": `Bearer ${API_KEY}` },
-      signal: AbortSignal.timeout(5e3)
+    await apiFetch("/v1/productivity/events", {
+      method: "POST",
+      body: JSON.stringify({ session_id: activeSessionId, event_type: eventType, metadata })
     });
-    if (!res.ok) return false;
-    const plan = await res.json();
-    return plan.features?.includes(feature) ?? false;
   } catch {
-    return false;
+  }
+}
+async function onToolCall(toolName) {
+  toolCallCount++;
+  if (trackingMode === "auto") {
+    await ensureAutoSession();
   }
 }
 async function ensureProductivity() {
   if (!API_KEY) return "Not connected to CostLens. Run: npx @costlens/mcp-server setup";
   if (productivityEnabled === null) {
-    productivityEnabled = await checkFeature("productivity_basic");
+    await loadKeySettings();
   }
   if (!productivityEnabled) return "Productivity tracking requires Business plan or Productivity add-on.";
   return null;
 }
+async function checkFeature(feature) {
+  if (!API_KEY) return false;
+  try {
+    const res = await apiFetch("/v1/plan");
+    if (!res.ok) return false;
+    const plan = await res.json();
+    return plan.features?.includes(feature) ?? false;
+  } catch {
+    return false;
+  }
+}
 async function main() {
-  await loadPricing();
+  await Promise.all([loadPricing(), loadKeySettings()]);
   const transport = new import_stdio.StdioServerTransport();
   await server.connect(transport);
   const cleanup = async () => {
     if (activeSessionId && API_KEY) {
       try {
-        await fetch(`${API_BASE}/v1/productivity/sessions/${activeSessionId}`, {
+        await apiFetch(`/v1/productivity/sessions/${activeSessionId}`, {
           method: "PATCH",
-          headers: { "Authorization": `Bearer ${API_KEY}`, "Content-Type": "application/json" },
-          body: JSON.stringify({ outcome: "completed" }),
-          signal: AbortSignal.timeout(3e3)
+          body: JSON.stringify({ outcome: "completed", metadata: { toolCalls: toolCallCount } })
         });
       } catch {
       }
@@ -141,7 +158,7 @@ async function main() {
   process.on("SIGTERM", cleanup);
   process.on("SIGINT", cleanup);
 }
-var import_mcp, import_stdio, import_zod, import_classifier, import_fs, import_path, import_os, API_KEY, API_BASE, server, activeSessionId, heartbeatInterval, productivityEnabled;
+var import_mcp, import_stdio, import_zod, import_classifier, import_fs, import_path, import_os, API_KEY, API_BASE, activeSessionId, heartbeatInterval, trackingMode, productivityEnabled, toolCallCount, server;
 var init_index = __esm({
   "src/index.ts"() {
     "use strict";
@@ -154,42 +171,32 @@ var init_index = __esm({
     import_os = require("os");
     API_KEY = resolveApiKey();
     API_BASE = process.env.COSTLENS_API_URL || "https://api.costlens.dev";
-    server = new import_mcp.McpServer({
-      name: "@lens360/mcp-server",
-      version: "0.3.1"
-    });
     activeSessionId = null;
     heartbeatInterval = null;
+    trackingMode = "manual";
+    productivityEnabled = null;
+    toolCallCount = 0;
+    server = new import_mcp.McpServer({ name: "@lens360/mcp-server", version: "0.4.0" });
     server.tool(
       "get_spend_summary",
       "Get current spend summary \u2014 daily, weekly, monthly, and session totals",
       {},
       async () => {
-        if (!API_KEY) {
-          return { content: [{ type: "text", text: "Not connected to CostLens. Run: npx @costlens/mcp-server setup" }] };
-        }
+        await onToolCall("get_spend_summary");
+        if (!API_KEY) return { content: [{ type: "text", text: "Not connected to CostLens. Run: npx @costlens/mcp-server setup" }] };
         try {
-          const res = await fetch(`${API_BASE}/v1/spend`, {
-            headers: { "Authorization": `Bearer ${API_KEY}` },
-            signal: AbortSignal.timeout(5e3)
-          });
-          if (!res.ok) {
-            return { content: [{ type: "text", text: `Error: ${res.status} \u2014 ${await res.text()}` }] };
-          }
+          const res = await apiFetch("/v1/spend");
+          if (!res.ok) return { content: [{ type: "text", text: `Error: ${res.status} \u2014 ${await res.text()}` }] };
           const data = await res.json();
           const lines = [
-            `\u{1F4B0} Spend Summary (${data.key})`,
+            `Spend Summary (${data.key})`,
             ``,
             `Today:   $${data.daily.cost.toFixed(4)} (${data.daily.requests} requests)`,
             `Week:    $${data.weekly.cost.toFixed(4)} (${data.weekly.requests} requests)`,
             `Month:   $${data.monthly.cost.toFixed(4)} (${data.monthly.requests} requests)`
           ];
-          if (data.session) {
-            lines.push(`Session: $${data.session.cost.toFixed(4)} (${data.session.requests} requests) [${data.session.correlationId}]`);
-          }
-          if (data.dailyBudget) {
-            lines.push(``, `Budget:  $${data.budgetRemaining.toFixed(2)} remaining of $${data.dailyBudget.toFixed(2)}/day`);
-          }
+          if (data.session) lines.push(`Session: $${data.session.cost.toFixed(4)} (${data.session.requests} requests) [${data.session.correlationId}]`);
+          if (data.dailyBudget) lines.push(``, `Budget:  $${data.budgetRemaining.toFixed(2)} remaining of $${data.dailyBudget.toFixed(2)}/day`);
           return { content: [{ type: "text", text: lines.join("\n") }] };
         } catch (err) {
           return { content: [{ type: "text", text: `Error fetching spend: ${err.message}` }] };
@@ -208,6 +215,7 @@ var init_index = __esm({
         isFirstMessage: import_zod.z.boolean().optional()
       },
       async (params) => {
+        await onToolCall("costlens_suggest_model");
         const result = (0, import_classifier.classify)({
           prompt: params.prompt,
           currentModel: params.currentModel,
@@ -217,15 +225,27 @@ var init_index = __esm({
           isFirstMessage: params.isFirstMessage
         });
         if (API_KEY && result.suggestedModel) {
-          syncEvent(result, params).catch(() => {
+          apiFetch("/v1/classifier/events", {
+            method: "POST",
+            body: JSON.stringify({
+              complexity: result.level,
+              confidence: result.confidence,
+              suggestedModel: result.suggestedModel,
+              currentModel: params.currentModel,
+              provider: params.provider,
+              staticScore: result.signals.staticScore
+            })
+          }).catch(() => {
+          });
+          autoLogEvent("prompt", {
+            complexity: result.level,
+            model: result.suggestedModel,
+            provider: params.provider
           });
         }
-        return {
-          content: [{ type: "text", text: JSON.stringify(result, null, 2) }]
-        };
+        return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }] };
       }
     );
-    productivityEnabled = null;
     server.tool(
       "start_session",
       "Start a productivity tracking session",
@@ -235,14 +255,13 @@ var init_index = __esm({
         repo: import_zod.z.string().optional()
       },
       async (params) => {
+        await onToolCall("start_session");
         const err = await ensureProductivity();
         if (err) return { content: [{ type: "text", text: err }] };
         try {
-          const res = await fetch(`${API_BASE}/v1/productivity/sessions`, {
+          const res = await apiFetch("/v1/productivity/sessions", {
             method: "POST",
-            headers: { "Authorization": `Bearer ${API_KEY}`, "Content-Type": "application/json" },
-            body: JSON.stringify(params),
-            signal: AbortSignal.timeout(5e3)
+            body: JSON.stringify(params)
           });
           if (!res.ok) return { content: [{ type: "text", text: `Error: ${res.status} \u2014 ${await res.text()}` }] };
           const data = await res.json();
@@ -262,14 +281,13 @@ var init_index = __esm({
         artifacts: import_zod.z.array(import_zod.z.string()).optional()
       },
       async (params) => {
+        await onToolCall("end_session");
         const err = await ensureProductivity();
         if (err) return { content: [{ type: "text", text: err }] };
         try {
-          const res = await fetch(`${API_BASE}/v1/productivity/sessions/${params.session_id}`, {
+          const res = await apiFetch(`/v1/productivity/sessions/${params.session_id}`, {
             method: "PATCH",
-            headers: { "Authorization": `Bearer ${API_KEY}`, "Content-Type": "application/json" },
-            body: JSON.stringify({ outcome: params.outcome, artifacts: params.artifacts }),
-            signal: AbortSignal.timeout(5e3)
+            body: JSON.stringify({ outcome: params.outcome, artifacts: params.artifacts })
           });
           if (!res.ok) return { content: [{ type: "text", text: `Error: ${res.status} \u2014 ${await res.text()}` }] };
           const data = await res.json();
@@ -290,18 +308,14 @@ var init_index = __esm({
         metadata: import_zod.z.record(import_zod.z.any()).optional()
       },
       async (params) => {
+        await onToolCall("log_event");
         if (!API_KEY) return { content: [{ type: "text", text: "Not connected to CostLens. Run: npx @costlens/mcp-server setup" }] };
-        if (productivityEnabled === null) {
-          productivityEnabled = await checkFeature("productivity_basic");
-        }
         const hasFull = await checkFeature("productivity_full");
         if (!hasFull) return { content: [{ type: "text", text: "Event logging requires Productivity Insights add-on." }] };
         try {
-          const res = await fetch(`${API_BASE}/v1/productivity/events`, {
+          const res = await apiFetch("/v1/productivity/events", {
             method: "POST",
-            headers: { "Authorization": `Bearer ${API_KEY}`, "Content-Type": "application/json" },
-            body: JSON.stringify(params),
-            signal: AbortSignal.timeout(5e3)
+            body: JSON.stringify(params)
           });
           if (!res.ok) return { content: [{ type: "text", text: `Error: ${res.status} \u2014 ${await res.text()}` }] };
           return { content: [{ type: "text", text: "Event logged." }] };
@@ -313,22 +327,18 @@ var init_index = __esm({
     server.tool(
       "get_productivity_summary",
       "Get productivity summary for a time period",
-      {
-        period: import_zod.z.enum(["today", "week", "month"]).optional()
-      },
+      { period: import_zod.z.enum(["today", "week", "month"]).optional() },
       async (params) => {
+        await onToolCall("get_productivity_summary");
         const err = await ensureProductivity();
         if (err) return { content: [{ type: "text", text: err }] };
         try {
           const period = params.period || "week";
-          const res = await fetch(`${API_BASE}/v1/productivity/summary?period=${period}`, {
-            headers: { "Authorization": `Bearer ${API_KEY}` },
-            signal: AbortSignal.timeout(5e3)
-          });
+          const res = await apiFetch(`/v1/productivity/summary?period=${period}`);
           if (!res.ok) return { content: [{ type: "text", text: `Error: ${res.status} \u2014 ${await res.text()}` }] };
           const data = await res.json();
           const lines = [
-            `\u{1F4CA} Productivity Summary (${period})`,
+            `Productivity Summary (${period})`,
             ``,
             `Sessions:    ${data.sessions}`,
             `Time:        ${data.total_time_minutes} min`,
@@ -349,25 +359,15 @@ var init_index = __esm({
     server.tool(
       "get_github_metrics",
       "Get GitHub PR metrics \u2014 PRs merged, review rounds, first-pass rate, cost per PR",
-      {
-        period: import_zod.z.enum(["today", "week", "month"]).optional()
-      },
+      { period: import_zod.z.enum(["today", "week", "month"]).optional() },
       async (params) => {
-        if (!API_KEY) {
-          return { content: [{ type: "text", text: "Not connected to CostLens. Run: npx @costlens/mcp-server setup" }] };
-        }
+        await onToolCall("get_github_metrics");
+        if (!API_KEY) return { content: [{ type: "text", text: "Not connected to CostLens. Run: npx @costlens/mcp-server setup" }] };
         try {
           const period = params.period || "week";
-          const res = await fetch(`${API_BASE}/v1/productivity/github?period=${period}`, {
-            headers: { "Authorization": `Bearer ${API_KEY}` },
-            signal: AbortSignal.timeout(5e3)
-          });
-          if (res.status === 404) {
-            return { content: [{ type: "text", text: "GitHub not connected. Connect in Settings \u2192 GitHub at costlens.dev" }] };
-          }
-          if (!res.ok) {
-            return { content: [{ type: "text", text: `Error: ${res.status} \u2014 ${await res.text()}` }] };
-          }
+          const res = await apiFetch(`/v1/productivity/github?period=${period}`);
+          if (res.status === 404) return { content: [{ type: "text", text: "GitHub not connected. Connect in Settings \u2192 GitHub at costlens.dev" }] };
+          if (!res.ok) return { content: [{ type: "text", text: `Error: ${res.status} \u2014 ${await res.text()}` }] };
           const { data } = await res.json();
           const lines = [
             `GitHub Metrics (${period}):`,
@@ -398,6 +398,11 @@ var CONFIG_FILE = (0, import_path2.join)(CONFIG_DIR, "config.json");
 var API_BASE2 = process.env.COSTLENS_API_URL || "https://api.costlens.dev";
 var APP_URL = process.env.COSTLENS_APP_URL || "https://costlens.dev";
 function readConfig() {
+  try {
+    const projectConfig = JSON.parse((0, import_fs2.readFileSync)((0, import_path2.join)(process.cwd(), ".costlens.json"), "utf-8"));
+    if (projectConfig.key) return { apiKey: projectConfig.key };
+  } catch {
+  }
   try {
     return JSON.parse((0, import_fs2.readFileSync)(CONFIG_FILE, "utf-8"));
   } catch {
@@ -408,8 +413,103 @@ function writeConfig(config) {
   if (!(0, import_fs2.existsSync)(CONFIG_DIR)) (0, import_fs2.mkdirSync)(CONFIG_DIR, { recursive: true });
   (0, import_fs2.writeFileSync)(CONFIG_FILE, JSON.stringify(config, null, 2));
 }
+var HOOK_MARKER = "# costlens-hook";
+var POST_COMMIT_HOOK = `#!/bin/sh
+${HOOK_MARKER}
+# CostLens \u2014 tracks commits during AI sessions (metadata only, no code content)
+# Sends: hash, message, files changed, insertions, deletions, branch, timestamp
+# Remove this file to uninstall: .git/hooks/post-commit
+
+COSTLENS_KEY=$(cat ~/.costlens/config.json 2>/dev/null | grep -o '"apiKey"[[:space:]]*:[[:space:]]*"[^"]*"' | cut -d'"' -f4)
+if [ -z "$COSTLENS_KEY" ]; then exit 0; fi
+
+HASH=$(git rev-parse HEAD)
+MSG=$(git log -1 --format=%s | head -c 200)
+BRANCH=$(git rev-parse --abbrev-ref HEAD)
+FILES=$(git diff --numstat HEAD~1 HEAD 2>/dev/null | wc -l | tr -d ' ')
+STATS=$(git diff --shortstat HEAD~1 HEAD 2>/dev/null)
+INSERTIONS=$(echo "$STATS" | grep -o '[0-9]* insertion' | grep -o '[0-9]*' || echo "0")
+DELETIONS=$(echo "$STATS" | grep -o '[0-9]* deletion' | grep -o '[0-9]*' || echo "0")
+
+curl -s -X POST "\${COSTLENS_API_URL:-https://api.costlens.dev}/v1/hooks/commit" \\
+  -H "Authorization: Bearer $COSTLENS_KEY" \\
+  -H "Content-Type: application/json" \\
+  -d "{\\"hash\\":\\"$HASH\\",\\"message\\":\\"$MSG\\",\\"branch\\":\\"$BRANCH\\",\\"filesChanged\\":$FILES,\\"insertions\\":\${INSERTIONS:-0},\\"deletions\\":\${DELETIONS:-0}}" \\
+  --max-time 3 > /dev/null 2>&1 &
+
+exit 0
+`;
+function findGitRepos() {
+  const cwd = process.cwd();
+  const repos = [];
+  if ((0, import_fs2.existsSync)((0, import_path2.join)(cwd, ".git"))) repos.push(cwd);
+  const searchDirs = [
+    (0, import_path2.join)((0, import_os2.homedir)(), "Desktop"),
+    (0, import_path2.join)((0, import_os2.homedir)(), "Projects"),
+    (0, import_path2.join)((0, import_os2.homedir)(), "projects"),
+    (0, import_path2.join)((0, import_os2.homedir)(), "dev"),
+    (0, import_path2.join)((0, import_os2.homedir)(), "code"),
+    (0, import_path2.join)((0, import_os2.homedir)(), "workspace"),
+    (0, import_path2.join)((0, import_os2.homedir)(), "Workspace"),
+    (0, import_path2.join)((0, import_os2.homedir)(), "Desktop", "Workspace")
+  ];
+  for (const dir of searchDirs) {
+    if (!(0, import_fs2.existsSync)(dir)) continue;
+    try {
+      const entries = require("fs").readdirSync(dir, { withFileTypes: true });
+      for (const entry of entries) {
+        if (entry.isDirectory() && (0, import_fs2.existsSync)((0, import_path2.join)(dir, entry.name, ".git"))) {
+          repos.push((0, import_path2.join)(dir, entry.name));
+        }
+      }
+    } catch {
+    }
+  }
+  return [...new Set(repos)];
+}
+function installGitHook(repoPath) {
+  const hooksDir = (0, import_path2.join)(repoPath, ".git", "hooks");
+  const hookPath = (0, import_path2.join)(hooksDir, "post-commit");
+  if ((0, import_fs2.existsSync)(hookPath)) {
+    try {
+      const content = (0, import_fs2.readFileSync)(hookPath, "utf-8");
+      if (content.includes(HOOK_MARKER)) return "exists";
+      return "skipped";
+    } catch {
+      return "skipped";
+    }
+  }
+  try {
+    if (!(0, import_fs2.existsSync)(hooksDir)) (0, import_fs2.mkdirSync)(hooksDir, { recursive: true });
+    (0, import_fs2.writeFileSync)(hookPath, POST_COMMIT_HOOK);
+    (0, import_fs2.chmodSync)(hookPath, "755");
+    return "installed";
+  } catch {
+    return "skipped";
+  }
+}
+function installHooks() {
+  const repos = findGitRepos();
+  if (repos.length === 0) {
+    console.log("  \xB7 No git repos detected\n");
+    return;
+  }
+  console.log(`
+  Git hooks (tracks commits during AI sessions):
+`);
+  for (const repo of repos.slice(0, 10)) {
+    const name = repo.split("/").pop();
+    const result = installGitHook(repo);
+    if (result === "installed") console.log(`  + ${name} \u2014 hook installed`);
+    else if (result === "exists") console.log(`  \xB7 ${name} \u2014 already installed`);
+    else console.log(`  \xB7 ${name} \u2014 skipped (existing hook)`);
+  }
+  console.log(`
+  Hooks send: commit hash, message, branch, files changed.`);
+  console.log(`  No code content is ever sent. Remove .git/hooks/post-commit to uninstall.`);
+}
 async function login() {
-  console.log("\u{1F511} CostLens \u2014 Authenticating...\n");
+  console.log("  CostLens \u2014 Authenticating...\n");
   const sessionId = import_crypto.default.randomBytes(16).toString("hex");
   const authUrl = `${APP_URL}/cli-auth?session=${sessionId}`;
   console.log(`  Opening browser: ${authUrl}
@@ -437,27 +537,24 @@ async function login() {
         const data = await res.json();
         if (data.key) {
           writeConfig({ apiKey: data.key });
-          console.log("\n\u2713 Authenticated successfully");
+          console.log("\n  Authenticated successfully");
           console.log(`  Key saved to ${CONFIG_FILE}
 `);
-          if (process.argv[2] === "setup") {
-            init();
-          } else {
-            process.exit(0);
-          }
+          if (process.argv[2] === "setup") init();
+          else process.exit(0);
           return;
         }
       }
     } catch {
     }
   }
-  console.log("\n\u2717 Timed out. Try again.");
+  console.log("\n  Timed out. Try again.");
   process.exit(1);
 }
 function init() {
   const config = readConfig();
   if (!config.apiKey) {
-    console.log("\u2717 Not authenticated. Run: npx @costlens/mcp-server login\n");
+    console.log("  Not authenticated. Run: npx @costlens/mcp-server login\n");
     process.exit(1);
   }
   const mcpConfig = {
@@ -465,20 +562,10 @@ function init() {
       costlens: {
         command: "npx",
         args: ["-y", "@costlens/mcp-server"],
-        env: {
-          COSTLENS_MCP_KEY: config.apiKey
-        }
+        env: { COSTLENS_MCP_KEY: config.apiKey }
       }
     }
   };
-  console.log("\u{1F4CB} MCP Configuration:\n");
-  console.log(JSON.stringify(mcpConfig, null, 2));
-  console.log("\n  Add this to your agent config file:\n");
-  console.log("  Kiro:       ~/.kiro/settings/mcp.json");
-  console.log("  Cursor:     ~/.cursor/mcp.json");
-  console.log("  Claude Code: ~/.claude/mcp_servers.json");
-  console.log("  VS Code:    .vscode/mcp.json");
-  console.log("");
   const kiroPath = (0, import_path2.join)((0, import_os2.homedir)(), ".kiro", "settings", "mcp.json");
   const cursorPath = (0, import_path2.join)((0, import_os2.homedir)(), ".cursor", "mcp.json");
   const claudePath = (0, import_path2.join)((0, import_os2.homedir)(), ".claude", "mcp_servers.json");
@@ -487,6 +574,8 @@ function init() {
     { name: "Cursor", path: cursorPath },
     { name: "Claude Code", path: claudePath }
   ];
+  console.log("  Agent configuration:\n");
+  let configured = false;
   for (const { name, path } of paths) {
     if ((0, import_fs2.existsSync)(path)) {
       try {
@@ -494,50 +583,63 @@ function init() {
         if (!existing.mcpServers?.costlens) {
           existing.mcpServers = { ...existing.mcpServers, ...mcpConfig.mcpServers };
           (0, import_fs2.writeFileSync)(path, JSON.stringify(existing, null, 2));
-          console.log(`  \u2713 Auto-configured ${name} (${path})`);
+          console.log(`  + ${name} \u2014 configured`);
+          configured = true;
         } else {
-          console.log(`  \xB7 ${name} already configured`);
+          console.log(`  \xB7 ${name} \u2014 already configured`);
+          configured = true;
         }
       } catch {
-        console.log(`  \xB7 Could not auto-configure ${name}`);
+        console.log(`  \xB7 ${name} \u2014 could not configure`);
       }
     }
   }
+  if (!configured) {
+    console.log("  No agents detected. Add manually:\n");
+    console.log(JSON.stringify(mcpConfig, null, 2));
+    console.log("\n  Config paths:");
+    console.log("    Kiro:        ~/.kiro/settings/mcp.json");
+    console.log("    Cursor:      ~/.cursor/mcp.json");
+    console.log("    Claude Code: ~/.claude/mcp_servers.json");
+    console.log("    VS Code:     .vscode/mcp.json");
+  }
+  installHooks();
   console.log("\n\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500");
-  console.log("\n  \u2713 Setup complete. What's next:\n");
+  console.log("\n  Setup complete.\n");
   console.log("  1. Restart your coding agent");
-  console.log("  2. Start coding \u2014 sessions are tracked automatically");
-  console.log("  3. View your dashboard: https://costlens.dev/dashboard");
-  console.log("  4. Docs: https://costlens.dev/docs/mcp\n");
+  console.log("  2. Start coding \u2014 sessions tracked automatically");
+  console.log("  3. Dashboard: https://costlens.dev/dashboard\n");
   console.log("  Commands:");
   console.log("    npx @costlens/mcp-server status   \u2014 check connection");
   console.log("    npx @costlens/mcp-server login    \u2014 re-authenticate");
+  console.log("    npx @costlens/mcp-server hooks    \u2014 reinstall git hooks");
   console.log("");
+  process.exit(0);
 }
 async function status() {
   const config = readConfig();
   if (!config.apiKey) {
-    console.log("\u2717 Not authenticated. Run: npx @costlens/mcp-server login\n");
+    console.log("  Not authenticated. Run: npx @costlens/mcp-server login\n");
     process.exit(1);
   }
-  console.log("\u{1F4CA} CostLens Status\n");
+  console.log("  CostLens Status\n");
   try {
     const res = await fetch(`${API_BASE2}/v1/spend`, {
       headers: { Authorization: `Bearer ${config.apiKey}` },
       signal: AbortSignal.timeout(5e3)
     });
     if (!res.ok) {
-      console.log("  \u2717 Connection failed (invalid key or server error)");
+      console.log("  Connection failed (invalid key or server error)");
       process.exit(1);
     }
     const data = await res.json();
-    console.log(`  \u2713 Connected`);
+    console.log(`  Connected`);
     console.log(`  Key: ${config.apiKey.slice(0, 8)}...`);
     console.log(`  Today: $${(data.today || 0).toFixed(4)}`);
     console.log(`  This week: $${(data.week || 0).toFixed(4)}`);
     console.log(`  This month: $${(data.month || 0).toFixed(4)}`);
   } catch (e) {
-    console.log(`  \u2717 Could not connect: ${e.message}`);
+    console.log(`  Could not connect: ${e.message}`);
     process.exit(1);
   }
 }
@@ -546,11 +648,15 @@ if (command === "login") login();
 else if (command === "init") init();
 else if (command === "status") status();
 else if (command === "setup") setup();
+else if (command === "hooks") {
+  installHooks();
+  process.exit(0);
+} else if (command === "doctor") doctor();
 else {
   init_index();
 }
 async function setup() {
-  console.log("\u{1F527} CostLens \u2014 One-step setup\n");
+  console.log("  CostLens \u2014 One-step setup\n");
   const existing = readConfig();
   if (existing.apiKey) {
     console.log("  Already authenticated. Running init...\n");
@@ -559,3 +665,139 @@ async function setup() {
   }
   await login();
 }
+async function doctor() {
+  console.log("\n  CostLens Doctor\n");
+  const config = readConfig();
+  const issues = [];
+  let keyValid = false;
+  let planData = null;
+  let spendData = null;
+  if (!config.apiKey) {
+    console.log("  Key:        not configured");
+    issues.push("No API key. Run: npx @costlens/mcp-server setup");
+  } else {
+    try {
+      const res = await fetch(`${API_BASE2}/v1/plan`, {
+        headers: { Authorization: `Bearer ${config.apiKey}` },
+        signal: AbortSignal.timeout(5e3)
+      });
+      if (res.ok) {
+        planData = await res.json();
+        keyValid = true;
+        console.log(`  Key:        valid (${config.apiKey.slice(0, 10)}...)`);
+      } else {
+        console.log(`  Key:        invalid (${res.status})`);
+        issues.push("API key is invalid or expired. Run: npx @costlens/mcp-server login");
+      }
+    } catch (e) {
+      console.log(`  Key:        error (${e.message})`);
+      issues.push("Cannot reach CostLens API. Check your network.");
+    }
+  }
+  if (planData) {
+    console.log(`  Plan:       ${planData.plan}`);
+    console.log(`  Tracking:   ${planData.keySettings?.trackingMode || "manual"}`);
+  }
+  if (keyValid) {
+    try {
+      const res = await fetch(`${API_BASE2}/v1/spend`, {
+        headers: { Authorization: `Bearer ${config.apiKey}` },
+        signal: AbortSignal.timeout(5e3)
+      });
+      if (res.ok) {
+        spendData = await res.json();
+        const daily = spendData.daily?.cost ?? 0;
+        const budget = spendData.dailyBudget;
+        if (budget) {
+          console.log(`  Budget:     $${daily.toFixed(2)} / $${budget.toFixed(2)} today`);
+        } else {
+          console.log(`  Spend:      $${daily.toFixed(4)} today`);
+        }
+      }
+    } catch {
+    }
+  }
+  const kiroPath = (0, import_path2.join)((0, import_os2.homedir)(), ".kiro", "settings", "mcp.json");
+  const cursorPath = (0, import_path2.join)((0, import_os2.homedir)(), ".cursor", "mcp.json");
+  const claudePath = (0, import_path2.join)((0, import_os2.homedir)(), ".claude", "mcp_servers.json");
+  const agents = [
+    { name: "Kiro", path: kiroPath },
+    { name: "Cursor", path: cursorPath },
+    { name: "Claude Code", path: claudePath }
+  ];
+  let agentConfigured = false;
+  for (const { name, path } of agents) {
+    if ((0, import_fs2.existsSync)(path)) {
+      try {
+        const content = JSON.parse((0, import_fs2.readFileSync)(path, "utf-8"));
+        if (content.mcpServers?.costlens) {
+          console.log(`  MCP:        ${name} (configured)`);
+          agentConfigured = true;
+          break;
+        }
+      } catch {
+      }
+    }
+  }
+  if (!agentConfigured) {
+    console.log("  MCP:        not configured");
+    issues.push("No agent configured. Run: npx @costlens/mcp-server setup");
+  }
+  const cwd = process.cwd();
+  const hookPath = (0, import_path2.join)(cwd, ".git", "hooks", "post-commit");
+  if ((0, import_fs2.existsSync)(hookPath)) {
+    try {
+      const content = (0, import_fs2.readFileSync)(hookPath, "utf-8");
+      if (content.includes(HOOK_MARKER)) {
+        console.log("  Git hook:   installed (post-commit)");
+      } else {
+        console.log("  Git hook:   exists (not CostLens)");
+      }
+    } catch {
+      console.log("  Git hook:   error reading");
+    }
+  } else if ((0, import_fs2.existsSync)((0, import_path2.join)(cwd, ".git"))) {
+    console.log("  Git hook:   not installed");
+    issues.push("Git hook not installed. Run: npx @costlens/mcp-server hooks");
+  } else {
+    console.log("  Git hook:   no git repo in cwd");
+  }
+  const pkgPath = (0, import_path2.join)(cwd, "package.json");
+  if ((0, import_fs2.existsSync)(pkgPath)) {
+    try {
+      const pkg = JSON.parse((0, import_fs2.readFileSync)(pkgPath, "utf-8"));
+      const sdkVersion = pkg.dependencies?.costlens || pkg.devDependencies?.costlens;
+      if (sdkVersion) {
+        console.log(`  SDK:        installed (${sdkVersion})`);
+      } else {
+        console.log("  SDK:        not installed");
+      }
+    } catch {
+    }
+  }
+  if (keyValid) {
+    try {
+      const res = await fetch(`${API_BASE2}/v1/productivity/github?period=week`, {
+        headers: { Authorization: `Bearer ${config.apiKey}` },
+        signal: AbortSignal.timeout(5e3)
+      });
+      if (res.ok) {
+        console.log("  GitHub:     connected");
+      } else if (res.status === 404) {
+        console.log("  GitHub:     not connected");
+      }
+    } catch {
+    }
+  }
+  console.log("");
+  if (issues.length === 0) {
+    console.log("  All good.\n");
+  } else {
+    console.log("  Issues:\n");
+    for (const issue of issues) {
+      console.log(`    - ${issue}`);
+    }
+    console.log("");
+  }
+  process.exit(issues.length > 0 ? 1 : 0);
+}

package/dist/index.js

@@ -21,37 +21,46 @@ function resolveApiKey() {
 }
 var API_KEY = resolveApiKey();
 var API_BASE = process.env.COSTLENS_API_URL || "https://api.costlens.dev";
+var activeSessionId = null;
+var heartbeatInterval = null;
+var trackingMode = "manual";
+var productivityEnabled = null;
+var toolCallCount = 0;
+async function apiFetch(path, opts = {}) {
+  const headers = { "Authorization": `Bearer ${API_KEY}`, ...opts.headers };
+  if (opts.body) headers["Content-Type"] = "application/json";
+  return fetch(`${API_BASE}${path}`, { ...opts, headers, signal: AbortSignal.timeout(5e3) });
+}
 async function loadPricing() {
   try {
-    const headers = {};
-    if (API_KEY) headers["Authorization"] = `Bearer ${API_KEY}`;
-    const res = await fetch(`${API_BASE}/v1/pricing`, { headers, signal: AbortSignal.timeout(5e3) });
+    const res = await apiFetch("/v1/pricing");
     if (res.ok) {
       const data = await res.json();
-      for (const m of data.models) {
-        import_classifier.PRICING[m.model] = { input: m.input, output: m.output };
-      }
+      for (const m of data.models) import_classifier.PRICING[m.model] = { input: m.input, output: m.output };
     }
   } catch {
   }
 }
-var server = new import_mcp.McpServer({
-  name: "@lens360/mcp-server",
-  version: "0.3.1"
-});
-var activeSessionId = null;
-var heartbeatInterval = null;
+async function loadKeySettings() {
+  if (!API_KEY) return;
+  try {
+    const res = await apiFetch("/v1/plan");
+    if (!res.ok) return;
+    const data = await res.json();
+    trackingMode = data.keySettings?.trackingMode || "manual";
+    productivityEnabled = data.features?.includes("productivity_basic") ?? false;
+  } catch {
+  }
+}
 function startHeartbeat(sessionId) {
   stopHeartbeat();
   activeSessionId = sessionId;
   heartbeatInterval = setInterval(async () => {
     if (!activeSessionId || !API_KEY) return;
     try {
-      const res = await fetch(`${API_BASE}/v1/productivity/heartbeat`, {
+      const res = await apiFetch("/v1/productivity/heartbeat", {
         method: "POST",
-        headers: { "Authorization": `Bearer ${API_KEY}`, "Content-Type": "application/json" },
-        body: JSON.stringify({ session_id: activeSessionId }),
-        signal: AbortSignal.timeout(5e3)
+        body: JSON.stringify({ session_id: activeSessionId })
       });
       if (res.status === 410) {
         activeSessionId = null;
@@ -67,36 +76,57 @@ function stopHeartbeat() {
     heartbeatInterval = null;
   }
 }
+async function ensureAutoSession() {
+  if (trackingMode !== "auto" || !API_KEY || !productivityEnabled || activeSessionId) return;
+  try {
+    const res = await apiFetch("/v1/productivity/sessions", {
+      method: "POST",
+      body: JSON.stringify({ task_description: "Auto-tracked MCP session" })
+    });
+    if (res.ok) {
+      const data = await res.json();
+      if (data.data?.id) startHeartbeat(data.data.id);
+    }
+  } catch {
+  }
+}
+async function autoLogEvent(eventType, metadata) {
+  if (trackingMode !== "auto" || !activeSessionId || !API_KEY) return;
+  try {
+    await apiFetch("/v1/productivity/events", {
+      method: "POST",
+      body: JSON.stringify({ session_id: activeSessionId, event_type: eventType, metadata })
+    });
+  } catch {
+  }
+}
+async function onToolCall(toolName) {
+  toolCallCount++;
+  if (trackingMode === "auto") {
+    await ensureAutoSession();
+  }
+}
+var server = new import_mcp.McpServer({ name: "@lens360/mcp-server", version: "0.4.0" });
 server.tool(
   "get_spend_summary",
   "Get current spend summary \u2014 daily, weekly, monthly, and session totals",
   {},
   async () => {
-    if (!API_KEY) {
-      return { content: [{ type: "text", text: "Not connected to CostLens. Run: npx @costlens/mcp-server setup" }] };
-    }
+    await onToolCall("get_spend_summary");
+    if (!API_KEY) return { content: [{ type: "text", text: "Not connected to CostLens. Run: npx @costlens/mcp-server setup" }] };
     try {
-      const res = await fetch(`${API_BASE}/v1/spend`, {
-        headers: { "Authorization": `Bearer ${API_KEY}` },
-        signal: AbortSignal.timeout(5e3)
-      });
-      if (!res.ok) {
-        return { content: [{ type: "text", text: `Error: ${res.status} \u2014 ${await res.text()}` }] };
-      }
+      const res = await apiFetch("/v1/spend");
+      if (!res.ok) return { content: [{ type: "text", text: `Error: ${res.status} \u2014 ${await res.text()}` }] };
       const data = await res.json();
       const lines = [
-        `\u{1F4B0} Spend Summary (${data.key})`,
+        `Spend Summary (${data.key})`,
         ``,
         `Today:   $${data.daily.cost.toFixed(4)} (${data.daily.requests} requests)`,
         `Week:    $${data.weekly.cost.toFixed(4)} (${data.weekly.requests} requests)`,
         `Month:   $${data.monthly.cost.toFixed(4)} (${data.monthly.requests} requests)`
       ];
-      if (data.session) {
-        lines.push(`Session: $${data.session.cost.toFixed(4)} (${data.session.requests} requests) [${data.session.correlationId}]`);
-      }
-      if (data.dailyBudget) {
-        lines.push(``, `Budget:  $${data.budgetRemaining.toFixed(2)} remaining of $${data.dailyBudget.toFixed(2)}/day`);
-      }
+      if (data.session) lines.push(`Session: $${data.session.cost.toFixed(4)} (${data.session.requests} requests) [${data.session.correlationId}]`);
+      if (data.dailyBudget) lines.push(``, `Budget:  $${data.budgetRemaining.toFixed(2)} remaining of $${data.dailyBudget.toFixed(2)}/day`);
       return { content: [{ type: "text", text: lines.join("\n") }] };
     } catch (err) {
       return { content: [{ type: "text", text: `Error fetching spend: ${err.message}` }] };
@@ -115,6 +145,7 @@ server.tool(
     isFirstMessage: import_zod.z.boolean().optional()
   },
   async (params) => {
+    await onToolCall("costlens_suggest_model");
     const result = (0, import_classifier.classify)({
       prompt: params.prompt,
       currentModel: params.currentModel,
@@ -124,39 +155,39 @@ server.tool(
       isFirstMessage: params.isFirstMessage
     });
     if (API_KEY && result.suggestedModel) {
-      syncEvent(result, params).catch(() => {
+      apiFetch("/v1/classifier/events", {
+        method: "POST",
+        body: JSON.stringify({
+          complexity: result.level,
+          confidence: result.confidence,
+          suggestedModel: result.suggestedModel,
+          currentModel: params.currentModel,
+          provider: params.provider,
+          staticScore: result.signals.staticScore
+        })
+      }).catch(() => {
+      });
+      autoLogEvent("prompt", {
+        complexity: result.level,
+        model: result.suggestedModel,
+        provider: params.provider
       });
     }
-    return {
-      content: [{ type: "text", text: JSON.stringify(result, null, 2) }]
-    };
+    return { content: [{ type: "text", text: JSON.stringify(result, null, 2) }] };
   }
 );
-async function syncEvent(result, params) {
-  try {
-    await fetch(`${API_BASE}/v1/classifier/events`, {
-      method: "POST",
-      headers: { "Authorization": `Bearer ${API_KEY}`, "Content-Type": "application/json" },
-      body: JSON.stringify({
-        complexity: result.level,
-        confidence: result.confidence,
-        suggestedModel: result.suggestedModel,
-        currentModel: params.currentModel,
-        provider: params.provider,
-        staticScore: result.signals.staticScore
-      }),
-      signal: AbortSignal.timeout(5e3)
-    });
-  } catch {
+async function ensureProductivity() {
+  if (!API_KEY) return "Not connected to CostLens. Run: npx @costlens/mcp-server setup";
+  if (productivityEnabled === null) {
+    await loadKeySettings();
   }
+  if (!productivityEnabled) return "Productivity tracking requires Business plan or Productivity add-on.";
+  return null;
 }
 async function checkFeature(feature) {
   if (!API_KEY) return false;
   try {
-    const res = await fetch(`${API_BASE}/v1/plan`, {
-      headers: { "Authorization": `Bearer ${API_KEY}` },
-      signal: AbortSignal.timeout(5e3)
-    });
+    const res = await apiFetch("/v1/plan");
     if (!res.ok) return false;
     const plan = await res.json();
     return plan.features?.includes(feature) ?? false;
@@ -164,15 +195,6 @@ async function checkFeature(feature) {
     return false;
   }
 }
-var productivityEnabled = null;
-async function ensureProductivity() {
-  if (!API_KEY) return "Not connected to CostLens. Run: npx @costlens/mcp-server setup";
-  if (productivityEnabled === null) {
-    productivityEnabled = await checkFeature("productivity_basic");
-  }
-  if (!productivityEnabled) return "Productivity tracking requires Business plan or Productivity add-on.";
-  return null;
-}
 server.tool(
   "start_session",
   "Start a productivity tracking session",
@@ -182,14 +204,13 @@ server.tool(
     repo: import_zod.z.string().optional()
   },
   async (params) => {
+    await onToolCall("start_session");
     const err = await ensureProductivity();
     if (err) return { content: [{ type: "text", text: err }] };
     try {
-      const res = await fetch(`${API_BASE}/v1/productivity/sessions`, {
+      const res = await apiFetch("/v1/productivity/sessions", {
         method: "POST",
-        headers: { "Authorization": `Bearer ${API_KEY}`, "Content-Type": "application/json" },
-        body: JSON.stringify(params),
-        signal: AbortSignal.timeout(5e3)
+        body: JSON.stringify(params)
       });
       if (!res.ok) return { content: [{ type: "text", text: `Error: ${res.status} \u2014 ${await res.text()}` }] };
       const data = await res.json();
@@ -209,14 +230,13 @@ server.tool(
     artifacts: import_zod.z.array(import_zod.z.string()).optional()
   },
   async (params) => {
+    await onToolCall("end_session");
     const err = await ensureProductivity();
     if (err) return { content: [{ type: "text", text: err }] };
     try {
-      const res = await fetch(`${API_BASE}/v1/productivity/sessions/${params.session_id}`, {
+      const res = await apiFetch(`/v1/productivity/sessions/${params.session_id}`, {
         method: "PATCH",
-        headers: { "Authorization": `Bearer ${API_KEY}`, "Content-Type": "application/json" },
-        body: JSON.stringify({ outcome: params.outcome, artifacts: params.artifacts }),
-        signal: AbortSignal.timeout(5e3)
+        body: JSON.stringify({ outcome: params.outcome, artifacts: params.artifacts })
       });
       if (!res.ok) return { content: [{ type: "text", text: `Error: ${res.status} \u2014 ${await res.text()}` }] };
       const data = await res.json();
@@ -237,18 +257,14 @@ server.tool(
     metadata: import_zod.z.record(import_zod.z.any()).optional()
   },
   async (params) => {
+    await onToolCall("log_event");
     if (!API_KEY) return { content: [{ type: "text", text: "Not connected to CostLens. Run: npx @costlens/mcp-server setup" }] };
-    if (productivityEnabled === null) {
-      productivityEnabled = await checkFeature("productivity_basic");
-    }
     const hasFull = await checkFeature("productivity_full");
     if (!hasFull) return { content: [{ type: "text", text: "Event logging requires Productivity Insights add-on." }] };
     try {
-      const res = await fetch(`${API_BASE}/v1/productivity/events`, {
+      const res = await apiFetch("/v1/productivity/events", {
         method: "POST",
-        headers: { "Authorization": `Bearer ${API_KEY}`, "Content-Type": "application/json" },
-        body: JSON.stringify(params),
-        signal: AbortSignal.timeout(5e3)
+        body: JSON.stringify(params)
       });
       if (!res.ok) return { content: [{ type: "text", text: `Error: ${res.status} \u2014 ${await res.text()}` }] };
       return { content: [{ type: "text", text: "Event logged." }] };
@@ -260,22 +276,18 @@ server.tool(
 server.tool(
   "get_productivity_summary",
   "Get productivity summary for a time period",
-  {
-    period: import_zod.z.enum(["today", "week", "month"]).optional()
-  },
+  { period: import_zod.z.enum(["today", "week", "month"]).optional() },
   async (params) => {
+    await onToolCall("get_productivity_summary");
     const err = await ensureProductivity();
     if (err) return { content: [{ type: "text", text: err }] };
     try {
       const period = params.period || "week";
-      const res = await fetch(`${API_BASE}/v1/productivity/summary?period=${period}`, {
-        headers: { "Authorization": `Bearer ${API_KEY}` },
-        signal: AbortSignal.timeout(5e3)
-      });
+      const res = await apiFetch(`/v1/productivity/summary?period=${period}`);
       if (!res.ok) return { content: [{ type: "text", text: `Error: ${res.status} \u2014 ${await res.text()}` }] };
       const data = await res.json();
       const lines = [
-        `\u{1F4CA} Productivity Summary (${period})`,
+        `Productivity Summary (${period})`,
         ``,
         `Sessions:    ${data.sessions}`,
         `Time:        ${data.total_time_minutes} min`,
@@ -296,25 +308,15 @@ server.tool(
 server.tool(
   "get_github_metrics",
   "Get GitHub PR metrics \u2014 PRs merged, review rounds, first-pass rate, cost per PR",
-  {
-    period: import_zod.z.enum(["today", "week", "month"]).optional()
-  },
+  { period: import_zod.z.enum(["today", "week", "month"]).optional() },
   async (params) => {
-    if (!API_KEY) {
-      return { content: [{ type: "text", text: "Not connected to CostLens. Run: npx @costlens/mcp-server setup" }] };
-    }
+    await onToolCall("get_github_metrics");
+    if (!API_KEY) return { content: [{ type: "text", text: "Not connected to CostLens. Run: npx @costlens/mcp-server setup" }] };
     try {
       const period = params.period || "week";
-      const res = await fetch(`${API_BASE}/v1/productivity/github?period=${period}`, {
-        headers: { "Authorization": `Bearer ${API_KEY}` },
-        signal: AbortSignal.timeout(5e3)
-      });
-      if (res.status === 404) {
-        return { content: [{ type: "text", text: "GitHub not connected. Connect in Settings \u2192 GitHub at costlens.dev" }] };
-      }
-      if (!res.ok) {
-        return { content: [{ type: "text", text: `Error: ${res.status} \u2014 ${await res.text()}` }] };
-      }
+      const res = await apiFetch(`/v1/productivity/github?period=${period}`);
+      if (res.status === 404) return { content: [{ type: "text", text: "GitHub not connected. Connect in Settings \u2192 GitHub at costlens.dev" }] };
+      if (!res.ok) return { content: [{ type: "text", text: `Error: ${res.status} \u2014 ${await res.text()}` }] };
       const { data } = await res.json();
       const lines = [
         `GitHub Metrics (${period}):`,
@@ -331,17 +333,15 @@ server.tool(
   }
 );
 async function main() {
-  await loadPricing();
+  await Promise.all([loadPricing(), loadKeySettings()]);
   const transport = new import_stdio.StdioServerTransport();
   await server.connect(transport);
   const cleanup = async () => {
     if (activeSessionId && API_KEY) {
       try {
-        await fetch(`${API_BASE}/v1/productivity/sessions/${activeSessionId}`, {
+        await apiFetch(`/v1/productivity/sessions/${activeSessionId}`, {
           method: "PATCH",
-          headers: { "Authorization": `Bearer ${API_KEY}`, "Content-Type": "application/json" },
-          body: JSON.stringify({ outcome: "completed" }),
-          signal: AbortSignal.timeout(3e3)
+          body: JSON.stringify({ outcome: "completed", metadata: { toolCalls: toolCallCount } })
         });
       } catch {
       }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@costlens/mcp-server",
-  "version": "0.4.3",
+  "version": "0.6.0",
   "description": "MCP server for AI cost optimization with prompt complexity classification",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",