@hasna/economy 0.2.4 → 0.2.6

package/dist/mcp/index.js

@@ -15,6 +15,7 @@ var __export = (target, all) => {
     });
 };
 var __esm = (fn, res) => () => (fn && (res = fn(fn = 0)), res);
+var __require = import.meta.require;
 
 // src/lib/pricing.ts
 var exports_pricing = {};
@@ -88,6 +89,10 @@ var init_pricing = __esm(() => {
     "claude-3-opus": { inputPer1M: 15, outputPer1M: 75, cacheReadPer1M: 1.5, cacheWritePer1M: 18.75 },
     "claude-3-sonnet": { inputPer1M: 3, outputPer1M: 15, cacheReadPer1M: 0.3, cacheWritePer1M: 3.75 },
     "claude-3-haiku": { inputPer1M: 0.25, outputPer1M: 1.25, cacheReadPer1M: 0.03, cacheWritePer1M: 0.3 },
+    "gemini-2.0-flash": { inputPer1M: 0.075, outputPer1M: 0.3, cacheReadPer1M: 0, cacheWritePer1M: 0 },
+    "gemini-2.5-pro": { inputPer1M: 1.25, outputPer1M: 10, cacheReadPer1M: 0, cacheWritePer1M: 0 },
+    "gemini-1.5-pro": { inputPer1M: 1.25, outputPer1M: 5, cacheReadPer1M: 0, cacheWritePer1M: 0 },
+    "gemini-1.5-flash": { inputPer1M: 0.075, outputPer1M: 0.3, cacheReadPer1M: 0, cacheWritePer1M: 0 },
     "gpt-5.3-codex": { inputPer1M: 1.75, outputPer1M: 14, cacheReadPer1M: 0.44, cacheWritePer1M: 0 },
     "gpt-5.2-codex": { inputPer1M: 1.75, outputPer1M: 14, cacheReadPer1M: 0.44, cacheWritePer1M: 0 },
     "gpt-5-codex": { inputPer1M: 1.75, outputPer1M: 14, cacheReadPer1M: 0.44, cacheWritePer1M: 0 },
@@ -174,6 +179,16 @@ function initSchema(db) {
       updated_at TEXT NOT NULL
     );
 
+    CREATE TABLE IF NOT EXISTS goals (
+      id TEXT PRIMARY KEY,
+      period TEXT NOT NULL,
+      project_path TEXT,
+      agent TEXT,
+      limit_usd REAL NOT NULL,
+      created_at TEXT NOT NULL,
+      updated_at TEXT NOT NULL
+    );
+
     CREATE TABLE IF NOT EXISTS ingest_state (
       source TEXT NOT NULL,
       key TEXT NOT NULL,
@@ -206,6 +221,8 @@ function periodWhere(period) {
       return `timestamp >= DATE('now', '-7 days')`;
     case "month":
       return `timestamp >= DATE('now', '-30 days')`;
+    case "year":
+      return `timestamp >= DATE('now', '-365 days')`;
     case "all":
       return "1=1";
   }
@@ -218,6 +235,8 @@ function sessionPeriodWhere(period) {
       return `started_at >= DATE('now', '-7 days')`;
     case "month":
       return `started_at >= DATE('now', '-30 days')`;
+    case "year":
+      return `started_at >= DATE('now', '-365 days')`;
     case "all":
       return "1=1";
   }
@@ -267,6 +286,11 @@ function querySessions(db, filter = {}) {
     conditions.push("started_at >= ?");
     params.push(filter.since);
   }
+  if (filter.search) {
+    const q = `%${filter.search}%`;
+    conditions.push("(project_name LIKE ? OR agent LIKE ? OR id LIKE ?)");
+    params.push(q, q, `${filter.search}%`);
+  }
   const where = conditions.length ? `WHERE ${conditions.join(" AND ")}` : "";
   const limit = filter.limit ?? 50;
   const offset = filter.offset ?? 0;
@@ -319,16 +343,31 @@ function queryModelBreakdown(db) {
 }
 function queryProjectBreakdown(db) {
   return db.prepare(`
-    SELECT project_path, project_name,
-           COUNT(*) as sessions,
-           COALESCE(SUM(total_tokens), 0) as total_tokens,
-           COALESCE(SUM(request_count), 0) as requests,
-           COALESCE(SUM(total_cost_usd), 0) as cost_usd,
-           MAX(started_at) as last_active
-    FROM sessions
-    GROUP BY project_path ORDER BY cost_usd DESC
+    SELECT
+      s.project_path,
+      COALESCE(p.name, s.project_name) as project_name,
+      COUNT(DISTINCT s.id) as sessions,
+      COUNT(r.id) as requests,
+      COALESCE(SUM(r.cost_usd), COALESCE(SUM(s.total_cost_usd), 0)) as cost_usd,
+      COALESCE(SUM(r.input_tokens + r.output_tokens + r.cache_read_tokens + r.cache_create_tokens), 0) as total_tokens,
+      MAX(s.started_at) as last_active
+    FROM sessions s
+    LEFT JOIN projects p ON p.path = s.project_path OR p.name = s.project_name
+    LEFT JOIN requests r ON r.session_id = s.id
+    WHERE s.project_path != '' OR s.project_name != ''
+    GROUP BY s.project_path
+    ORDER BY cost_usd DESC
   `).all();
 }
+function queryDailyBreakdown(db, days = 30) {
+  return db.prepare(`
+    SELECT DATE(timestamp) as date, agent, COALESCE(SUM(cost_usd), 0) as cost_usd
+    FROM requests
+    WHERE timestamp >= DATE('now', ? || ' days')
+    GROUP BY DATE(timestamp), agent
+    ORDER BY date ASC
+  `).all(`-${days}`);
+}
 function listBudgets(db) {
   return db.prepare(`SELECT * FROM budgets ORDER BY created_at DESC`).all();
 }
@@ -358,6 +397,46 @@ function getBudgetStatuses(db) {
     };
   });
 }
+function upsertGoal(db, goal) {
+  db.prepare(`
+    INSERT OR REPLACE INTO goals
+      (id, period, project_path, agent, limit_usd, created_at, updated_at)
+    VALUES (?, ?, ?, ?, ?, ?, ?)
+  `).run(goal.id, goal.period, goal.project_path ?? null, goal.agent ?? null, goal.limit_usd, goal.created_at, goal.updated_at);
+}
+function deleteGoal(db, id) {
+  db.prepare(`DELETE FROM goals WHERE id = ?`).run(id);
+}
+function listGoals(db) {
+  return db.prepare(`SELECT * FROM goals ORDER BY created_at DESC`).all();
+}
+function getGoalStatuses(db) {
+  const goals = listGoals(db);
+  return goals.map((g) => {
+    const periodStart = g.period === "day" ? "DATE('now')" : g.period === "week" ? "DATE('now', '-7 days')" : g.period === "month" ? "DATE('now', '-30 days')" : "DATE('now', '-365 days')";
+    let spendQuery = `SELECT COALESCE(SUM(cost_usd), 0) as spend FROM requests WHERE timestamp >= ${periodStart}`;
+    const params = [];
+    if (g.project_path) {
+      spendQuery += ` AND session_id IN (SELECT id FROM sessions WHERE project_path = ?)`;
+      params.push(g.project_path);
+    }
+    if (g.agent) {
+      spendQuery += ` AND agent = ?`;
+      params.push(g.agent);
+    }
+    const row = db.prepare(spendQuery).get(...params);
+    const spend = row.spend;
+    const percent = g.limit_usd > 0 ? spend / g.limit_usd * 100 : 0;
+    return {
+      ...g,
+      current_spend_usd: spend,
+      percent_used: percent,
+      is_on_track: percent < 70,
+      is_at_risk: percent >= 70 && percent <= 100,
+      is_over: percent > 100
+    };
+  });
+}
 function getIngestState(db, source, key) {
   const row = db.prepare(`SELECT value FROM ingest_state WHERE source = ? AND key = ?`).get(source, key);
   return row?.value ?? null;
@@ -405,6 +484,9 @@ init_pricing();
 import { readdirSync, readFileSync, existsSync as existsSync2, statSync } from "fs";
 import { homedir as homedir2 } from "os";
 import { join as join2, basename } from "path";
+function autoDetectProject(cwd, projects) {
+  return projects.find((p) => cwd === p.path || cwd.startsWith(p.path + "/"));
+}
 var PROJECTS_DIR = join2(homedir2(), ".claude", "projects");
 function dirNameToPath(dirName) {
   return dirName.replace(/^-/, "/").replace(/-/g, "/").replace(/\/\//g, "/-");
@@ -433,11 +515,11 @@ async function ingestClaude(db, verbose = false, _telemetryDir) {
   let totalFiles = 0;
   let totalRequests = 0;
   const touchedSessions = new Set;
+  const registeredProjects = db.prepare(`SELECT path, name FROM projects ORDER BY LENGTH(path) DESC`).all();
   const projectDirs = readdirSync(PROJECTS_DIR, { withFileTypes: true }).filter((d) => d.isDirectory());
   for (const projectDirEntry of projectDirs) {
     const projectDirPath = join2(PROJECTS_DIR, projectDirEntry.name);
     const projectPath = dirNameToPath(projectDirEntry.name);
-    const projectName = basename(projectPath);
     const jsonlFiles = collectJsonlFiles(projectDirPath);
     for (const filePath of jsonlFiles) {
       const stateKey = filePath.replace(PROJECTS_DIR, "");
@@ -506,11 +588,13 @@ async function ingestClaude(db, verbose = false, _telemetryDir) {
         if (!touchedSessions.has(sessionId)) {
           const existing = db.prepare(`SELECT id FROM sessions WHERE id = ?`).get(sessionId);
           if (!existing) {
+            const effectiveCwd = sessionCwd || projectPath;
+            const detectedProject = autoDetectProject(effectiveCwd, registeredProjects);
             const session = {
               id: sessionId,
               agent: "claude",
-              project_path: sessionCwd || projectPath,
-              project_name: basename(sessionCwd || projectPath),
+              project_path: detectedProject ? detectedProject.path : effectiveCwd,
+              project_name: detectedProject ? detectedProject.name : "",
               started_at: timestamp,
               ended_at: null,
               total_cost_usd: 0,
@@ -541,24 +625,12 @@ import { join as join3, basename as basename2 } from "path";
 import { Database as Database2 } from "bun:sqlite";
 var CODEX_DB_PATH = join3(homedir3(), ".codex", "state_5.sqlite");
 var CODEX_CONFIG_PATH = join3(homedir3(), ".codex", "config.toml");
-function readCodexModel() {
-  if (!existsSync3(CODEX_CONFIG_PATH))
-    return "gpt-5.3-codex";
-  try {
-    const content = readFileSync2(CODEX_CONFIG_PATH, "utf-8");
-    const match = content.match(/^model\s*=\s*"([^"]+)"/m);
-    return match?.[1] ?? "gpt-5.3-codex";
-  } catch {
-    return "gpt-5.3-codex";
-  }
-}
 async function ingestCodex(db, verbose = false) {
   if (!existsSync3(CODEX_DB_PATH)) {
     if (verbose)
       console.log("Codex DB not found:", CODEX_DB_PATH);
     return { sessions: 0 };
   }
-  const model = readCodexModel();
   let codexDb = null;
   let ingested = 0;
   try {
@@ -596,6 +668,83 @@ async function ingestCodex(db, verbose = false) {
   return { sessions: ingested };
 }
 
+// src/ingest/gemini.ts
+init_database();
+import { readdirSync as readdirSync2, readFileSync as readFileSync3, existsSync as existsSync4, statSync as statSync2 } from "fs";
+import { homedir as homedir4 } from "os";
+import { join as join4 } from "path";
+var GEMINI_TMP_DIR = join4(homedir4(), ".gemini", "tmp");
+async function ingestGemini(db, verbose) {
+  if (!existsSync4(GEMINI_TMP_DIR)) {
+    if (verbose)
+      console.log("Gemini tmp dir not found:", GEMINI_TMP_DIR);
+    return { sessions: 0 };
+  }
+  let totalSessions = 0;
+  const touchedSessions = new Set;
+  let projectHashDirs = [];
+  try {
+    projectHashDirs = readdirSync2(GEMINI_TMP_DIR, { withFileTypes: true }).filter((d) => d.isDirectory() && /^[0-9a-f]{64}$/.test(d.name)).map((d) => join4(GEMINI_TMP_DIR, d.name));
+  } catch {
+    return { sessions: 0 };
+  }
+  for (const projectDir of projectHashDirs) {
+    const chatsDir = join4(projectDir, "chats");
+    if (!existsSync4(chatsDir))
+      continue;
+    let chatFiles = [];
+    try {
+      chatFiles = readdirSync2(chatsDir).filter((f) => f.endsWith(".json")).map((f) => join4(chatsDir, f));
+    } catch {
+      continue;
+    }
+    for (const filePath of chatFiles) {
+      const stateKey = filePath.replace(homedir4(), "~");
+      let fileMtime = "0";
+      try {
+        fileMtime = statSync2(filePath).mtimeMs.toString();
+      } catch {
+        continue;
+      }
+      const processed = getIngestState(db, "gemini", stateKey);
+      if (processed === fileMtime)
+        continue;
+      let chatData;
+      try {
+        chatData = JSON.parse(readFileSync3(filePath, "utf-8"));
+      } catch {
+        continue;
+      }
+      const sessionId = chatData.sessionId;
+      if (!sessionId)
+        continue;
+      const startTime = chatData.startTime ?? new Date().toISOString();
+      const existing = db.prepare(`SELECT id FROM sessions WHERE id = ?`).get(sessionId);
+      if (!existing) {
+        const session = {
+          id: sessionId,
+          agent: "gemini",
+          project_path: "",
+          project_name: "",
+          started_at: startTime,
+          ended_at: chatData.lastUpdated ?? null,
+          total_cost_usd: 0,
+          total_tokens: 0,
+          request_count: 0
+        };
+        upsertSession(db, session);
+        touchedSessions.add(sessionId);
+        totalSessions++;
+      }
+      setIngestState(db, "gemini", stateKey, fileMtime);
+    }
+  }
+  for (const sessionId of touchedSessions) {
+    rollupSession(db, sessionId);
+  }
+  return { sessions: totalSessions };
+}
+
 // src/mcp/index.ts
 init_pricing();
 var db = openDatabase();
@@ -612,24 +761,37 @@ function fmtSession(s) {
   return `${id} ${agent.padEnd(6)} ${cost.padEnd(10)} ${tok.padEnd(8)} ${proj}`;
 }
 var TOOLS = [
-  { name: "get_cost_summary", description: "Cost summary (total_usd, sessions, requests, tokens, human summary). period: today|week|month|all", inputSchema: { type: "object", properties: { period: { type: "string", enum: ["today", "week", "month", "all"] } } } },
+  { name: "get_cost_summary", description: "Cost summary (total_usd, sessions, requests, tokens, human summary). period: today|week|month|year|all", inputSchema: { type: "object", properties: { period: { type: "string", enum: ["today", "week", "month", "year", "all"] } } } },
   { name: "get_sessions", description: "List sessions. Returns compact table. Params: agent, project, limit(20)", inputSchema: { type: "object", properties: { agent: { type: "string" }, project: { type: "string" }, limit: { type: "number" } } } },
   { name: "get_top_sessions", description: "Top sessions by cost. Params: n(10), agent", inputSchema: { type: "object", properties: { n: { type: "number" }, agent: { type: "string" } } } },
   { name: "get_model_breakdown", description: "Cost per model. No params.", inputSchema: { type: "object", properties: {} } },
   { name: "get_project_breakdown", description: "Cost per project. No params.", inputSchema: { type: "object", properties: {} } },
   { name: "get_budget_status", description: "Budget limits vs spend, percent used, alert flags. No params.", inputSchema: { type: "object", properties: {} } },
-  { name: "sync", description: "Ingest new cost data. sources: all|claude|codex", inputSchema: { type: "object", properties: { sources: { type: "string", enum: ["all", "claude", "codex"] } } } },
+  { name: "get_daily", description: "Daily cost table by agent. Params: days(30)", inputSchema: { type: "object", properties: { days: { type: "number" } } } },
+  { name: "get_session_detail", description: "Per-request breakdown of a single session. Params: session_id (prefix ok)", inputSchema: { type: "object", properties: { session_id: { type: "string" } }, required: ["session_id"] } },
+  { name: "sync", description: "Ingest new cost data. sources: all|claude|codex|gemini", inputSchema: { type: "object", properties: { sources: { type: "string", enum: ["all", "claude", "codex", "gemini"] } } } },
   { name: "search_tools", description: "List tool names matching query. Use first to find relevant tools.", inputSchema: { type: "object", properties: { query: { type: "string" } } } },
-  { name: "describe_tools", description: "Get param hints for specific tools by name.", inputSchema: { type: "object", properties: { names: { type: "array", items: { type: "string" } } }, required: ["names"] } }
+  { name: "describe_tools", description: "Get param hints for specific tools by name.", inputSchema: { type: "object", properties: { names: { type: "array", items: { type: "string" } } }, required: ["names"] } },
+  { name: "get_goals", description: "All spending goals with current progress. No params.", inputSchema: { type: "object", properties: {} } },
+  { name: "set_goal", description: "Create/update a spending goal. period(day|week|month|year), limit_usd, project_path?, agent?", inputSchema: { type: "object", properties: { period: { type: "string" }, limit_usd: { type: "number" }, project_path: { type: "string" }, agent: { type: "string" } }, required: ["period", "limit_usd"] } },
+  { name: "remove_goal", description: "Delete a goal by id.", inputSchema: { type: "object", properties: { id: { type: "string" } }, required: ["id"] } },
+  { name: "register_agent", description: "Register agent session.", inputSchema: { type: "object", properties: { name: { type: "string" }, session_id: { type: "string" } }, required: ["name"] } },
+  { name: "heartbeat", description: "Update last_seen_at.", inputSchema: { type: "object", properties: { agent_id: { type: "string" } }, required: ["agent_id"] } },
+  { name: "set_focus", description: "Set active project context.", inputSchema: { type: "object", properties: { agent_id: { type: "string" }, project_id: { type: "string" } }, required: ["agent_id"] } }
 ];
 var TOOL_DESCRIPTIONS = {
-  get_cost_summary: "period(today|week|month|all) \u2192 {total_usd, sessions, requests, tokens, summary}",
+  get_cost_summary: "period(today|week|month|year|all) \u2192 {total_usd, sessions, requests, tokens, summary}",
   get_sessions: "agent(claude|codex), project(partial), limit(20) \u2192 compact session table",
   get_top_sessions: "n(10), agent(claude|codex) \u2192 top sessions by cost",
   get_model_breakdown: "no params \u2192 model, requests, tokens, cost",
   get_project_breakdown: "no params \u2192 project_name, sessions, cost",
   get_budget_status: "no params \u2192 budget limits, current spend, percent_used, is_over_alert",
-  sync: "sources(all|claude|codex) \u2192 {files, requests, sessions} ingested"
+  get_daily: "days(30) \u2192 daily cost table grouped by date and agent",
+  get_session_detail: "session_id(prefix ok) \u2192 per-request breakdown with model, tokens, cost",
+  sync: "sources(all|claude|codex|gemini) \u2192 {files, requests, sessions} ingested",
+  get_goals: "no params \u2192 period, scope, limit, spent, percent, status(ON TRACK/AT RISK/OVER)",
+  set_goal: "period(day|week|month|year), limit_usd, project_path?, agent? \u2192 creates/updates goal",
+  remove_goal: "id \u2192 deletes goal"
 };
 server.setRequestHandler(ListToolsRequestSchema, async () => ({ tools: TOOLS }));
 server.setRequestHandler(CallToolRequestSchema, async (req) => {
@@ -713,6 +875,48 @@ server.setRequestHandler(CallToolRequestSchema, async (req) => {
           lines.push(`${scope.padEnd(21)}${String(b["period"]).padEnd(9)}${fmtUsd(Number(b["current_spend_usd"])).padEnd(11)}${fmtUsd(Number(b["limit_usd"])).padEnd(11)}${pct}%`.padEnd(49) + `  ${status}`);
         }
         return { content: [{ type: "text", text: lines.join(`
+`) }] };
+      }
+      case "get_daily": {
+        const days = Number(a["days"] ?? 30);
+        const rows = queryDailyBreakdown(db, days);
+        const lines = ["date        claude     codex      gemini     total"];
+        const byDate = new Map;
+        for (const r of rows) {
+          const d = String(r["date"]);
+          const entry = byDate.get(d) ?? { claude: 0, codex: 0, gemini: 0 };
+          if (r["agent"] === "claude")
+            entry.claude += Number(r["cost_usd"]);
+          else if (r["agent"] === "codex")
+            entry.codex += Number(r["cost_usd"]);
+          else if (r["agent"] === "gemini")
+            entry.gemini += Number(r["cost_usd"]);
+          byDate.set(d, entry);
+        }
+        for (const [date, costs] of [...byDate.entries()].sort()) {
+          const total = costs.claude + costs.codex + costs.gemini;
+          lines.push(`${date}  ${fmtUsd(costs.claude).padEnd(11)}${fmtUsd(costs.codex).padEnd(11)}${fmtUsd(costs.gemini).padEnd(11)}${fmtUsd(total)}`);
+        }
+        return { content: [{ type: "text", text: lines.join(`
+`) }] };
+      }
+      case "get_session_detail": {
+        const sid = String(a["session_id"] ?? "");
+        const session = db.prepare(`SELECT * FROM sessions WHERE id = ? OR id LIKE ?`).get(sid, `${sid}%`);
+        if (!session)
+          return { content: [{ type: "text", text: `Session not found: ${sid}` }], isError: true };
+        const requests = db.prepare(`SELECT * FROM requests WHERE session_id = ? ORDER BY timestamp ASC LIMIT 50`).all(session["id"]);
+        const lines = [
+          `session: ${String(session["id"]).slice(0, 16)}`,
+          `agent: ${session["agent"]}  project: ${session["project_name"] || "\u2014"}`,
+          `cost: ${fmtUsd(Number(session["total_cost_usd"]))}  tokens: ${fmtTok(Number(session["total_tokens"]))}  requests: ${session["request_count"]}`,
+          "",
+          "time      model                  input    output   cost"
+        ];
+        for (const r of requests) {
+          lines.push(`${String(r["timestamp"]).slice(11, 19)}  ${String(r["model"]).slice(0, 22).padEnd(23)}${fmtTok(Number(r["input_tokens"])).padEnd(9)}${fmtTok(Number(r["output_tokens"])).padEnd(9)}${fmtUsd(Number(r["cost_usd"]))}`);
+        }
+        return { content: [{ type: "text", text: lines.join(`
 `) }] };
       }
       case "sync": {
@@ -726,9 +930,71 @@ server.setRequestHandler(CallToolRequestSchema, async (req) => {
           const r = await ingestCodex(db);
           parts.push(`codex: ${r["sessions"]} sessions`);
         }
+        if (sources === "all" || sources === "gemini") {
+          const r = await ingestGemini(db);
+          parts.push(`gemini: ${r["sessions"]} sessions`);
+        }
         return { content: [{ type: "text", text: parts.join(`
 `) || "done" }] };
       }
+      case "get_goals": {
+        const goals = getGoalStatuses(db);
+        if (goals.length === 0)
+          return { content: [{ type: "text", text: "No goals set." }] };
+        const lines = ["period   scope                limit      spent      used%  status"];
+        for (const g of goals) {
+          const scope = String(g["project_path"] ?? g["agent"] ?? "global").slice(0, 20);
+          const pct = Number(g["percent_used"]).toFixed(1);
+          const status = g["is_over"] ? "OVER" : g["is_at_risk"] ? "AT RISK" : "ON TRACK";
+          lines.push(`${String(g["period"]).padEnd(9)}${scope.padEnd(21)}${fmtUsd(Number(g["limit_usd"])).padEnd(11)}${fmtUsd(Number(g["current_spend_usd"])).padEnd(11)}${pct}%  ${status}`);
+        }
+        return { content: [{ type: "text", text: lines.join(`
+`) }] };
+      }
+      case "set_goal": {
+        const { randomUUID } = await import("crypto");
+        const now = new Date().toISOString();
+        upsertGoal(db, {
+          id: randomUUID(),
+          period: String(a["period"] ?? "month"),
+          project_path: a["project_path"] ?? null,
+          agent: a["agent"] ?? null,
+          limit_usd: Number(a["limit_usd"]),
+          created_at: now,
+          updated_at: now
+        });
+        return { content: [{ type: "text", text: `Goal set: ${a["period"]} $${a["limit_usd"]}` }] };
+      }
+      case "remove_goal": {
+        deleteGoal(db, String(a["id"] ?? ""));
+        return { content: [{ type: "text", text: "Goal removed." }] };
+      }
+      case "register_agent": {
+        const n = String(args["name"] ?? "");
+        const ex = [..._econAgents.values()].find((x) => x.name === n);
+        if (ex) {
+          ex.last_seen_at = new Date().toISOString();
+          return { content: [{ type: "text", text: JSON.stringify(ex) }] };
+        }
+        const id = Math.random().toString(36).slice(2, 10);
+        const ag = { id, name: n, last_seen_at: new Date().toISOString() };
+        _econAgents.set(id, ag);
+        return { content: [{ type: "text", text: JSON.stringify(ag) }] };
+      }
+      case "heartbeat": {
+        const ag = _econAgents.get(String(args["agent_id"] ?? ""));
+        if (!ag)
+          return { content: [{ type: "text", text: `Agent not found` }], isError: true };
+        ag.last_seen_at = new Date().toISOString();
+        return { content: [{ type: "text", text: `\u2665 ${ag.name}` }] };
+      }
+      case "set_focus": {
+        const ag = _econAgents.get(String(args["agent_id"] ?? ""));
+        if (!ag)
+          return { content: [{ type: "text", text: `Agent not found` }], isError: true };
+        ag["project_id"] = args["project_id"];
+        return { content: [{ type: "text", text: String(args["project_id"] ? `Focus: ${args["project_id"]}` : "Focus cleared") }] };
+      }
       default:
         return { content: [{ type: "text", text: `Unknown tool: ${name}` }], isError: true };
     }
@@ -736,31 +1002,6 @@ server.setRequestHandler(CallToolRequestSchema, async (req) => {
     return { content: [{ type: "text", text: `Error: ${e instanceof Error ? e.message : String(e)}` }], isError: true };
   }
 });
-var _agentReg = new Map;
-server.tool("register_agent", "Register this agent session. Returns agent_id for use in heartbeat/set_focus.", { name: z.string(), session_id: z.string().optional() }, async (a) => {
-  const existing = [..._agentReg.values()].find((x) => x.name === a.name);
-  if (existing) {
-    existing.last_seen_at = new Date().toISOString();
-    return { content: [{ type: "text", text: JSON.stringify(existing) }] };
-  }
-  const id = Math.random().toString(36).slice(2, 10);
-  const ag = { id, name: a.name, last_seen_at: new Date().toISOString() };
-  _agentReg.set(id, ag);
-  return { content: [{ type: "text", text: JSON.stringify(ag) }] };
-});
-server.tool("heartbeat", "Update last_seen_at to signal agent is active.", { agent_id: z.string() }, async (a) => {
-  const ag = _agentReg.get(a.agent_id);
-  if (!ag)
-    return { content: [{ type: "text", text: `Agent not found: ${a.agent_id}` }], isError: true };
-  ag.last_seen_at = new Date().toISOString();
-  return { content: [{ type: "text", text: `\u2665 ${ag.name} \u2014 active` }] };
-});
-server.tool("set_focus", "Set active project context for this agent session.", { agent_id: z.string(), project_id: z.string().optional() }, async (a) => {
-  const ag = _agentReg.get(a.agent_id);
-  if (!ag)
-    return { content: [{ type: "text", text: `Agent not found: ${a.agent_id}` }], isError: true };
-  ag.project_id = a.project_id;
-  return { content: [{ type: "text", text: a.project_id ? `Focus: ${a.project_id}` : "Focus cleared" }] };
-});
+var _econAgents = new Map;
 var transport = new StdioServerTransport;
 await server.connect(transport);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hasna/economy",
-  "version": "0.2.4",
+  "version": "0.2.6",
   "description": "AI coding cost tracker — CLI + MCP server + REST API + web dashboard for Claude Code, Codex, and Gemini",
   "type": "module",
   "main": "dist/index.js",