owlmem 0.9.1 → 0.9.2

package/owlmem.mjs

@@ -41,7 +41,7 @@ import { join, dirname, basename, relative, isAbsolute } from "node:path";
 import { fileURLToPath } from "node:url";
 import { spawn, spawnSync } from "node:child_process";
 
-export const VERSION = "0.9.1";
+export const VERSION = "0.9.2";
 const BASE = (process.env.OWLMEM_URL || "https://owlmem.com").replace(/\/$/, "");
 const DIR = process.env.OWLMEM_HOME || join(process.env.XDG_CONFIG_HOME || join(homedir(), ".config"), "owlmem");
 const enc = new TextEncoder(), dec = new TextDecoder();
@@ -1237,6 +1237,7 @@ export async function wakeLoop(o) {
       await sleep(backoff); continue;
     }
     backoff = 0;
+    if (Array.isArray(r.data.agents)) cfg.live = r.data.agents;       // which AIs the owner lets us read Claude quota for (read_quota)
     for (const t of r.data.tasks || []) {
       const plan = isReply(t) ? { kind: "scratch" } : o.plan ? o.plan(t) : planWorkdir(t);      // a reply is read in an empty folder
       log((isReply(t) ? "Reply #" + t.id + " (task #" + t.task_id + ")" : jobLabel(t)) + " for " + t.agent + ": starting " + t.client + (plan.dir ? " in " + plan.dir : "") + ".");
@@ -1420,7 +1421,9 @@ export function usageLine(client, line, f) {
   const rl = j?.payload?.type === "token_count" ? j.payload.rate_limits : null;
   if (rl && at >= (f.st?.limit_at || 0)) {   // Codex says how much of its plan's window is used and when it resets
     const w = [rl.primary, rl.secondary].filter((x) => x && typeof x.used_percent === "number").sort((a, b) => b.used_percent - a.used_percent)[0];
-    f.st = { ...f.st, limit_at: at, ok_at: at, used_percent: w?.used_percent, resets_at: w?.resets_at, window_minutes: w?.window_minutes,
+    const windows = [rl.primary, rl.secondary].filter((x) => x && typeof x.used_percent === "number")
+      .map((x) => ({ minutes: x.window_minutes, used_percent: x.used_percent, resets_at: x.resets_at }));
+    f.st = { ...f.st, limit_at: at, ok_at: at, used_percent: w?.used_percent, resets_at: w?.resets_at, window_minutes: w?.window_minutes, windows,
              ...(rl.rate_limit_reached_type ? { block: "limit", block_at: at } : {}) };
   }
   const info = j?.payload?.type === "token_count" ? j.payload.info : null, last = info?.last_token_usage, d = usageDay(j?.timestamp);
@@ -1479,10 +1482,39 @@ export function usageStatus(state, clients) {
     const o = out[f.client] || (out[f.client] = {});
     if ((f.st.ok_at || 0) > (o.ok_at || 0)) o.ok_at = f.st.ok_at;
     if ((f.st.block_at || 0) > (o.block_at || 0)) Object.assign(o, { block: f.st.block, block_at: f.st.block_at, resets_at: f.st.resets_at ?? o.resets_at });
-    if ((f.st.limit_at || 0) > (o.limit_at || 0)) Object.assign(o, { limit_at: f.st.limit_at, used_percent: f.st.used_percent, window_minutes: f.st.window_minutes, resets_at: f.st.resets_at });
+    if ((f.st.limit_at || 0) > (o.limit_at || 0)) Object.assign(o, { limit_at: f.st.limit_at, used_percent: f.st.used_percent, window_minutes: f.st.window_minutes, resets_at: f.st.resets_at, windows: f.st.windows });
   }
   return out;
 }
+// Claude Code's own quota (0.9.2, only when the AI's owner turned "Read Claude quota" on in the console): ask Anthropic's
+// usage endpoint with the sign-in Claude Code keeps on this computer, the same one its /usage screen uses. The sign-in is
+// only sent to api.anthropic.com and never leaves this computer otherwise; OwlMem gets the percentages and reset times.
+// An expired sign-in isn't refreshed here (Claude Code does that itself): no quota this round.
+export function claudeToken(h = home(), env = process.env, plat = process.platform, now = Date.now()) {
+  let raw = null;
+  try { raw = readFileSync(join(env.CLAUDE_CONFIG_DIR || join(h, ".claude"), ".credentials.json"), "utf8"); } catch {}
+  if (!raw && plat === "darwin") {
+    const r = spawnSync("security", ["find-generic-password", "-s", "Claude Code-credentials", "-w"], { encoding: "utf8", timeout: 10000 });
+    if (r.status === 0) raw = r.stdout;
+  }
+  let o; try { o = JSON.parse(raw || "null")?.claudeAiOauth; } catch { return null; }
+  if (!o?.accessToken || (o.expiresAt && o.expiresAt < now + 60000)) return null;
+  return o.accessToken;
+}
+export function claudeWindows(j) {
+  const sec = (v) => { const t = Date.parse(v); return isNaN(t) ? undefined : Math.floor(t / 1000); };
+  return [["five_hour", 300], ["seven_day", 10080]].filter(([k]) => typeof j?.[k]?.utilization === "number")
+    .map(([k, minutes]) => ({ minutes, used_percent: j[k].utilization, resets_at: sec(j[k].resets_at) }));
+}
+async function claudeQuota(fetchImpl = fetch) {
+  const tok = claudeToken();
+  if (!tok) return null;
+  const r = await fetchImpl("https://api.anthropic.com/api/oauth/usage", { headers: { Authorization: "Bearer " + tok, "anthropic-beta": "oauth-2025-04-20", "User-Agent": "owlmem-cli/" + VERSION },
+                                                                          signal: AbortSignal.timeout(15000) });
+  if (!r.ok) return null;
+  const w = claudeWindows(await r.json());
+  return w.length ? w : null;
+}
 async function usageTick(cfg, api, log, dir = DIR) {
   const file = join(wakeDir(dir), "usage.json");
   let state = {}; try { state = JSON.parse(readFileSync(file, "utf8")); } catch {}
@@ -1491,7 +1523,12 @@ async function usageTick(cfg, api, log, dir = DIR) {
   if (!clients.length) return;
   scanUsage(state, Object.fromEntries(Object.entries(usageRoots()).filter(([c]) => clients.includes(c))));
   try { writeFileSync(file + ".tmp", JSON.stringify(state)); renameSync(file + ".tmp", file); } catch (e) { log("usage: " + e.message); }
-  const r = await api("usage", { days: usageReport(state, clients), status: usageStatus(state, clients) });
+  const status = usageStatus(state, clients);
+  if (clients.includes("claude-code") && (cfg.live || []).some((a) => a.client === "claude-code" && a.read_quota)) {
+    const w = await claudeQuota().catch(() => null);
+    if (w) status["claude-code"] = { ...status["claude-code"], windows: w, limit_at: Math.floor(Date.now() / 1000) };
+  }
+  const r = await api("usage", { days: usageReport(state, clients), status });
   if (r.status !== 200 && r.status !== 404) log("usage: OwlMem answered " + (r.data?.detail || "HTTP " + r.status));
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "owlmem",
-  "version": "0.9.1",
+  "version": "0.9.2",
   "description": "OwlMem command line: connect your AI apps to OwlMem in one command (npx -y owlmem install), an end-to-end encrypted vault for your AI agents, and a local MCP server.",
   "type": "module",
   "bin": { "owlmem": "owlmem.mjs" },