pi-kimi-keepalive 0.3.2 → 0.3.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-kimi-keepalive",
-  "version": "0.3.2",
+  "version": "0.3.5",
   "description": "Prompt-cache keepalive for Kimi (kimi-coding) sessions in the Pi coding agent. Replays the last real provider request while idle so the automatic prefix cache never expires.",
   "type": "module",
   "license": "MIT",

package/src/index.ts

@@ -51,7 +51,22 @@ import {
 
 const STATE_DIR = join(homedir(), ".pi", "cache-keepalive");
 const STATE_FILE = join(STATE_DIR, "state.json");
-const AUTH_FILE = join(process.env.PI_AGENT_DIR ?? join(homedir(), ".pi", "agent"), "auth.json");
+
+/** Package version, read from the package.json that ships this extension. */
+function ownVersion(): string {
+  try {
+    const pkg = JSON.parse(
+      readFileSync(new URL("../../package.json", import.meta.url), "utf8"),
+    ) as { version?: unknown };
+    return typeof pkg.version === "string" ? pkg.version : "unknown";
+  } catch {
+    return "unknown";
+  }
+}
+/** Live OAuth state; resolved lazily so tests can redirect it via PI_AGENT_DIR. */
+function authFile(): string {
+  return join(process.env.PI_AGENT_DIR ?? join(homedir(), ".pi", "agent"), "auth.json");
+}
 
 interface PersistedConfig {
   enabled: boolean;
@@ -473,22 +488,26 @@ export default function (pi: ExtensionAPI) {
    */
   function probeHeaders(): Record<string, string> | null {
     if (!capture) return null;
-    const headers = buildProbeHeaders(capture.headers);
-    if (!headers.authorization) {
-      try {
-        const raw = JSON.parse(readFileSync(AUTH_FILE, "utf8")) as Record<string, unknown>;
-        const entry = (raw["kimi-coding"] ?? null) as { access?: unknown } | null;
-        if (entry && typeof entry.access === "string" && entry.access.length > 0) {
-          headers.authorization = `Bearer ${entry.access}`;
-        } else {
-          debug(`no kimi-coding access token in ${AUTH_FILE}`);
-        }
-      } catch (error) {
-        debug(`auth.json unavailable: ${error instanceof Error ? error.message : String(error)}`);
+    // Prefer the live token from auth.json: pi refreshes it there on real
+    // requests, while captured request headers freeze the token from capture
+    // time — hours later that Bearer can already be expired (HTTP 401).
+    // Fall back to the captured headers when auth.json is unavailable.
+    try {
+      const raw = JSON.parse(readFileSync(authFile(), "utf8")) as Record<string, unknown>;
+      const entry = (raw["kimi-coding"] ?? null) as { access?: unknown } | null;
+      if (entry && typeof entry.access === "string" && entry.access.length > 0) {
+        const headers = buildProbeHeaders(capture.headers);
+        headers.authorization = `Bearer ${entry.access}`;
+        return headers;
       }
+      debug(`no kimi-coding access token in ${authFile()}`);
+    } catch (error) {
+      debug(
+        `auth.json unavailable (${error instanceof Error ? error.message : String(error)}) — falling back to captured headers`,
+      );
     }
-    if (!headers.authorization) return null;
-    return headers;
+    const headers = buildProbeHeaders(capture.headers);
+    return headers.authorization ? headers : null;
   }
 
   async function runProbe(): Promise<boolean> {
@@ -710,7 +729,7 @@ export default function (pi: ExtensionAPI) {
       const savings = hasPricing(capture?.cost) ? formatUsd(stats.savedUsd) : "n/a (no price data)";
       ctx.ui.setStatus("cache-keepalive", `♥ ${state}`);
       ctx.ui.setWidget("cache-keepalive", [
-        "pi-kimi-keepalive",
+        `pi-kimi-keepalive v${ownVersion()}`,
         `  state:   ${state}`,
         `  probes:  ${stats.probes} sent · ${stats.hits} hits · ${stats.misses} misses · ${stats.errors} errors`,
         `  est.:    saved ${savings} · probe spend ${formatUsd(stats.spendUsd)} · cap ${config.spendCapUsd === null ? "none" : formatUsd(config.spendCapUsd)}`,
@@ -733,7 +752,7 @@ export default function (pi: ExtensionAPI) {
       ? `${capture.provider} (${capture.api ?? "unknown api"}) @ ${capture.baseUrl}`
       : "none yet — probes start after your first real turn";
     return [
-      "pi-kimi-keepalive",
+      `pi-kimi-keepalive v${ownVersion()}`,
       `  state:     ${config.enabled ? "on" : "off"}${pausedReason ? ` (paused: ${pausedReason})` : ""}`,
       `  capture:   ${route}`,
       `  mode: ${config.mode}${config.mode === "smart" ? ` · cadence ${formatDuration(config.intervalMs)}${lastProbeInputTokens > SMART_MAX_CONTEXT_TOKENS ? ` · context ${lastProbeInputTokens.toLocaleString()} > cap, frozen at floor` : ""}` : " (fixed via interval=)"}`,
@@ -952,8 +971,23 @@ export default function (pi: ExtensionAPI) {
     const raw = payload as Record<string, unknown>;
     if (!Array.isArray(raw.messages) || raw.messages.length === 0) return;
     if (raw.system !== undefined && !Array.isArray(raw.system) && typeof raw.system !== "string") return;
+    let clonedPayload: Record<string, unknown>;
+    try {
+      clonedPayload = structuredClone(raw);
+    } catch {
+      // DataCloneError: the payload carries non-cloneable values (BigInt,
+      // symbols, functions). Fall back to a JSON round-trip; probing never
+      // mutates the captured payload, so a by-reference capture is the last
+      // resort rather than failing the real request.
+      try {
+        clonedPayload = JSON.parse(JSON.stringify(raw)) as Record<string, unknown>;
+      } catch {
+        debug("payload not cloneable — capturing by reference");
+        clonedPayload = raw;
+      }
+    }
     capture = {
-      payload: structuredClone(raw),
+      payload: clonedPayload,
       headers: capturedHeaders,
       provider: model.provider,
       api: (model as { api?: string }).api,