@yunazgr/pi-companion 0.3.2 → 0.3.3

package/README.md

@@ -2,7 +2,9 @@
 
 Lightweight, local-first remote control for Pi sessions.
 
-**New in 0.3.2:** responsive automation and run-history tables, rounded status chips, matching clay-icon headers, and a single-column automation workspace with full-width mobile controls. **Edit automation** scrolls to and focuses the inline editor. Pull to refresh Overview, Sessions, Automations, or Settings—or use the Refresh button; unsaved settings stay intact. Camera-policy guidance now explains HTTPS proxy configuration. Existing rich results, staged job editing, optional retries, and answers-only automation sessions are retained. See [automations](docs/automations.md).
+**New in 0.3.3:** Provider usage now shows measured session tokens and cost without pretending they are subscription quotas. Compatible extensions can still supply real weekly/5-hour limits. Mobile sessions, automations and usage use cards; the overview has compact activity filters; automation buttons share a consistent style; pull-to-refresh is gesture-only. No data or provider credentials are collected by the daemon outside the already shared session metadata.
+
+**New in 0.3.2:** responsive automation and run-history tables, rounded status chips, matching clay-icon headers, and a single-column automation workspace with full-width mobile controls. **Edit automation** scrolls to and focuses the inline editor. Pull to refresh Overview, Sessions, Automations, or Settings—with a gesture; unsaved settings stay intact. Camera-policy guidance now explains HTTPS proxy configuration. Existing rich results, staged job editing, optional retries, and answers-only automation sessions are retained. See [automations](docs/automations.md).
 
 Pi Companion is deliberately not another agent runtime. Pi owns execution and conversation state. A single Rust daemon owns session discovery, pairing, temporary file exchange, browser fan-out, and the embedded web UI.
 
@@ -192,7 +194,7 @@ Network loss and daemon restarts reconnect automatically with bounded backoff. H
 
 Shared session title, model, effort and working directory refresh on Pi events and once per second while sharing, including while idle. Session detail shows context-window usage and estimated session cost. Native Pi context and recorded usage costs take precedence; independent extension reports fill unavailable fields. Missing usage is shown as unavailable, not zero.
 
-**Settings → Usage** is available on the console and paired devices. It displays the latest snapshot per provider: weekly usage, 5-hour usage when reported, reset times, source and snapshot age. Quota snapshots are not added across sessions or accounts. These are extension-reported limits, not billing totals or locally inferred quotas; providers without a quota report remain unavailable.
+**Settings → Usage** is available on the console and paired devices. It displays the latest snapshot per provider: weekly usage, 5-hour usage when reported, reset times, source and snapshot age. Quota snapshots are not added across sessions or accounts. Those limits are extension-reported, not billing totals or locally inferred quotas. Companion additionally reports actual session token consumption and cost grouped by the model's provider from Pi's recorded assistant messages; this is session-only data, not an account-wide tally. Settings selects the newest consumption independently of quota freshness and displays their timestamps and session-ended indicators separately. When no quota adapter is installed, 5-hour and weekly quotas still show as unavailable rather than fabricated percentages.
 
 Extensions can publish a provider-neutral event without importing Companion:
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yunazgr/pi-companion",
-  "version": "0.3.2",
+  "version": "0.3.3",
   "description": "Local-first remote control for Pi sessions: live activity feed, steering, git diff, file drop and phone pairing from a single Rust daemon.",
   "keywords": [
     "pi-package",

package/src/protocol.ts

@@ -43,6 +43,11 @@ export type ProviderUsage = {
   updatedAt: string;
   weekly?: UsageWindow;
   fiveHour?: UsageWindow;
+  /** Observed consumption in this shared Pi session, not account quota utilization. */
+  sessionTokens?: number;
+  sessionCost?: number;
+  /** Native consumption freshness, independent of the account quota report. */
+  sessionUpdatedAt?: string;
 };
 export type SessionTelemetry = {
   context?: { tokens?: number; window?: number; percent?: number; source?: string };

package/src/telemetry.ts

@@ -51,6 +51,7 @@ export function normalizeTelemetry(value: unknown, source = "extension", now = D
 /** Independent sources may supply different fields; one broken/missing adapter cannot erase another. */
 export class TelemetryRelay {
   private sources = new Map<string, { telemetry: SessionTelemetry; contextAt?: number; costAt?: number }>();
+  private nativeUsage = new Map<string, { fingerprint: string; updatedAt: string }>();
 
   ingest(value: unknown, fallbackSource = "extension", now = Date.now()) {
     const raw = record(value);
@@ -140,6 +141,43 @@ export class TelemetryRelay {
       }
       if (known && Number.isFinite(total)) result.cost = { amount: total, currency: "USD", source: "native" };
     } catch { /* Preserve extension estimate when native usage is unavailable. */ }
+    // Native session entries are available without another plugin. They report actual
+    // consumption, not provider subscription limits, and must never fabricate quotas.
+    try {
+      const totals = new Map<string, { tokens: number; cost: number; hasTokens: boolean; hasCost: boolean }>();
+      for (const entry of ctx?.sessionManager?.getEntries?.() ?? []) {
+        const raw = record(entry);
+        if (raw?.type !== "message") continue;
+        const message = record(raw.message);
+        if (message?.role !== "assistant") continue;
+        const provider = text(message.provider);
+        const usage = record(message.usage);
+        if (!provider || !usage) continue;
+        const pieces = [usage.input, usage.output, usage.cacheRead, usage.cacheWrite].map(number);
+        const tokens = number(usage.totalTokens) ?? (pieces.some(value => value !== undefined)
+          ? pieces.reduce<number>((sum, value) => sum + (value ?? 0), 0) : undefined);
+        const cost = number(record(usage.cost)?.total);
+        if (tokens === undefined && cost === undefined) continue;
+        const total = totals.get(provider) ?? { tokens: 0, cost: 0, hasTokens: false, hasCost: false };
+        if (tokens !== undefined) { total.tokens += tokens; total.hasTokens = true; }
+        if (cost !== undefined) { total.cost += cost; total.hasCost = true; }
+        totals.set(provider, total);
+      }
+      for (const [provider, total] of totals) {
+        const fingerprint = JSON.stringify(total);
+        const cached = this.nativeUsage.get(provider);
+        const updatedAt = cached?.fingerprint === fingerprint ? cached.updatedAt : new Date(now).toISOString();
+        this.nativeUsage.set(provider, { fingerprint, updatedAt });
+        const old = providers.get(provider);
+        providers.set(provider, {
+          ...old, provider, source: old?.source ?? "Pi session", updatedAt: old?.updatedAt ?? updatedAt,
+          sessionUpdatedAt: updatedAt,
+          ...(total.hasTokens ? { sessionTokens: total.tokens } : {}),
+          ...(total.hasCost ? { sessionCost: total.cost } : {})
+        });
+      }
+      if (providers.size) result.providers = [...providers.values()];
+    } catch { /* Keep independently reported quota snapshots if native entries are unavailable. */ }
     return result;
   }
 }