@cursor/july 0.1.27 → 0.1.29

package/dist/internal/reminder-runner.js

@@ -13,10 +13,8 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
     });
 };
 import { randomBytes } from "node:crypto";
-import { join } from "node:path";
 import { parseDuration } from "./duration.js";
 import { defaultBusyPolicy, MAX_TIMER_MS, nextCronFireMs, ReminderValidationError, resolveSchedule, } from "./reminder-schedule.js";
-import { ReminderStore, } from "./reminder-store.js";
 import { APP_AUTH } from "./schedule-runner.js";
 import { SessionBusyError } from "./session-engine.js";
 export { ReminderValidationError } from "./reminder-schedule.js";
@@ -71,35 +69,72 @@ function buildFirePrompt(record) {
     return lines.join("\n");
 }
 export class ReminderRunner {
-    constructor(engine, logger = () => { }) {
+    constructor(engine, logger = () => { }, platformTimers) {
         this.engine = engine;
         this.logger = logger;
+        this.platformTimers = platformTimers;
+        this.records = new Map();
         this.handlers = new Map();
         this.timers = new Map();
         this.firing = new Set();
         /** Single-flight idle gates keyed by `channelId:continuationToken`. */
         this.idleGates = new Map();
         this.started = false;
-        this.store = new ReminderStore(join(engine.stateRoot, "reminders"));
     }
-    get storeRoot() {
-        return join(this.engine.stateRoot, "reminders");
+    /**
+     * Load reminder documents from the control plane into memory. Does not
+     * arm in-process timers — the CP owns clocks.
+     */
+    hydrate() {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (this.platformTimers === undefined) {
+                return;
+            }
+            let remote = [];
+            try {
+                remote = yield this.platformTimers.listReminders();
+            }
+            catch (error) {
+                this.logger(`[agentkit] reminders: control-plane list failed: ${error instanceof Error ? error.message : String(error)}`);
+                return;
+            }
+            this.records.clear();
+            for (const record of remote) {
+                this.records.set(record.id, record);
+            }
+            let disarmedOrphan = 0;
+            for (const record of [...this.records.values()]) {
+                if (record.status !== "armed") {
+                    continue;
+                }
+                if ((record.hasRunHandler === true || record.hasUntilFn === true) &&
+                    this.handlers.get(record.id) === undefined) {
+                    yield this.disarm(record, "handler_lost_on_restart");
+                    disarmedOrphan += 1;
+                }
+            }
+            if (remote.length > 0 || disarmedOrphan > 0) {
+                this.logger(`[agentkit] reminders: hydrated ${remote.length} reminder(s) from control plane` +
+                    (disarmedOrphan > 0
+                        ? `; disarmed ${disarmedOrphan} without live handlers`
+                        : ""));
+            }
+        });
     }
-    /** Start deadline timers for armed reminders (call after restore). */
+    /** Start deadline timers for armed reminders (tests / no control plane). */
     start() {
         return __awaiter(this, void 0, void 0, function* () {
             if (this.started)
                 return;
             this.started = true;
-            // Materialize sink-backed reminders onto local disk before arming
-            // (local wins when both sides have the same id).
-            yield this.hydrateFromStorage();
-            const armed = yield this.store.listArmed();
+            yield this.hydrate();
+            if (this.platformTimers !== undefined) {
+                return;
+            }
+            const armed = [...this.records.values()].filter((r) => r.status === "armed");
             let restored = 0;
             let disarmedOrphan = 0;
             for (const record of armed) {
-                // Run / function-until handlers are in-memory only. After restart they
-                // are gone — disarm rather than falling back to a generic prompt fire.
                 if ((record.hasRunHandler === true || record.hasUntilFn === true) &&
                     this.handlers.get(record.id) === undefined) {
                     yield this.disarm(record, "handler_lost_on_restart");
@@ -117,45 +152,49 @@ export class ReminderRunner {
             }
         });
     }
-    /**
-     * Pull reminder records from `defineStorage` into `<stateRoot>/reminders`
-     * when local disk is empty for that id. Failures are logged and skipped.
-     */
-    hydrateFromStorage() {
+    persist(record) {
         return __awaiter(this, void 0, void 0, function* () {
-            const storage = this.engine.storage;
-            if (storage === undefined || !storage.canBulkRestore) {
-                return;
+            if (this.platformTimers !== undefined) {
+                yield this.platformTimers.putReminder(record);
             }
-            let hydrated = 0;
-            try {
-                const remote = yield storage.listReminders();
-                for (const record of remote) {
-                    const local = yield this.store.get(record.id);
-                    if (local !== undefined) {
-                        continue;
+            this.records.set(record.id, record);
+        });
+    }
+    getRecord(id) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (this.platformTimers !== undefined) {
+                try {
+                    const remote = yield this.platformTimers.getReminder(id);
+                    if (remote === undefined) {
+                        this.records.delete(id);
+                        return undefined;
                     }
-                    // Local write only — re-mirroring would just echo the sink.
-                    yield this.store.save(record);
-                    hydrated += 1;
+                    this.records.set(id, remote);
+                    return remote;
+                }
+                catch (error) {
+                    this.logger(`[agentkit] reminders: control-plane get failed for ${id}: ${error instanceof Error ? error.message : String(error)}`);
+                    return this.records.get(id);
                 }
             }
-            catch (error) {
-                const message = error instanceof Error ? error.message : String(error);
-                this.logger(`[agentkit] reminders: storage hydrate failed: ${message}`);
-                return;
-            }
-            if (hydrated > 0) {
-                this.logger(`[agentkit] reminders: hydrated ${hydrated} reminder(s) from storage`);
-            }
+            return this.records.get(id);
         });
     }
-    /** Persist locally and mirror into the authored storage sink when present. */
-    persist(record) {
+    allRecords() {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a;
-            yield this.store.save(record);
-            (_a = this.engine.storage) === null || _a === void 0 ? void 0 : _a.reminder(record);
+            if (this.platformTimers !== undefined) {
+                try {
+                    const remote = yield this.platformTimers.listReminders();
+                    this.records.clear();
+                    for (const record of remote) {
+                        this.records.set(record.id, record);
+                    }
+                }
+                catch (error) {
+                    this.logger(`[agentkit] reminders: control-plane list failed: ${error instanceof Error ? error.message : String(error)}`);
+                }
+            }
+            return [...this.records.values()];
         });
     }
     stop() {
@@ -217,7 +256,7 @@ export class ReminderRunner {
                 run: input.run,
                 untilFn,
             });
-            if (this.started) {
+            if (this.started && this.platformTimers === undefined) {
                 this.armTimer(record);
             }
             return toInfo(record);
@@ -225,7 +264,7 @@ export class ReminderRunner {
     }
     list(filter) {
         return __awaiter(this, void 0, void 0, function* () {
-            const all = yield this.store.list();
+            const all = yield this.allRecords();
             return all
                 .filter((r) => {
                 if ((filter === null || filter === void 0 ? void 0 : filter.status) !== undefined && r.status !== filter.status) {
@@ -251,7 +290,7 @@ export class ReminderRunner {
     }
     cancel(id_1) {
         return __awaiter(this, arguments, void 0, function* (id, reason = "explicit_cancel") {
-            const record = yield this.store.get(id);
+            const record = yield this.getRecord(id);
             if (record === undefined) {
                 return undefined;
             }
@@ -268,7 +307,7 @@ export class ReminderRunner {
     /** Cancel all armed reminders matching a continuation token (e.g. PR closed). */
     cancelForContinuation(continuationToken, reason) {
         return __awaiter(this, void 0, void 0, function* () {
-            const armed = yield this.store.listArmed();
+            const armed = (yield this.allRecords()).filter((r) => r.status === "armed");
             let n = 0;
             for (const record of armed) {
                 if (record.continuationToken === continuationToken) {
@@ -287,7 +326,7 @@ export class ReminderRunner {
      */
     rekeyContinuation(channelId, oldToken, newToken) {
         return __awaiter(this, void 0, void 0, function* () {
-            const armed = yield this.store.listArmed();
+            const armed = (yield this.allRecords()).filter((r) => r.status === "armed");
             let n = 0;
             for (const record of armed) {
                 if (record.channelId !== channelId ||
@@ -300,14 +339,22 @@ export class ReminderRunner {
             return n;
         });
     }
-    /** Fire one reminder immediately (dev / hillclimb). */
-    dispatch(reminderId) {
-        return __awaiter(this, void 0, void 0, function* () {
-            const record = yield this.store.get(reminderId);
+    /** Fire one reminder immediately (dev / hillclimb / control plane). */
+    dispatch(reminderId_1) {
+        return __awaiter(this, arguments, void 0, function* (reminderId, opts = {}) {
+            const record = yield this.getRecord(reminderId);
             if (record === undefined) {
-                const ids = (yield this.store.list()).map((r) => r.id);
+                const ids = (yield this.allRecords()).map((r) => r.id);
                 throw new UnknownReminderError(reminderId, ids);
             }
+            if (opts.expectedFireCount !== undefined &&
+                record.fireCount !== opts.expectedFireCount) {
+                return {
+                    reminderId,
+                    outcome: "noop",
+                    reason: "stale_fire_count",
+                };
+            }
             const result = yield this.fire(record.id, { force: true });
             return Object.assign({ reminderId }, result);
         });
@@ -327,7 +374,7 @@ export class ReminderRunner {
                 // Keep recurring reminders alive after unexpected throws.
                 // One-shots stay armed so a failed wake can be retried / dispatched.
                 try {
-                    const current = yield this.store.get(record.id);
+                    const current = yield this.getRecord(record.id);
                     if (current === undefined || current.status !== "armed")
                         return;
                     const oneShot = current.scheduleKind === "delay" || current.scheduleKind === "at";
@@ -362,7 +409,7 @@ export class ReminderRunner {
             }
             this.firing.add(id);
             try {
-                const record = yield this.store.get(id);
+                const record = yield this.getRecord(id);
                 if (record === undefined || record.status !== "armed") {
                     this.clearTimer(id);
                     return { outcome: "noop", reason: "not_armed" };
@@ -399,7 +446,7 @@ export class ReminderRunner {
                         return { outcome: "fired", reason: "skipped_busy" };
                     }
                     yield this.whenSessionIdle(record.channelId, record.continuationToken);
-                    const still = yield this.store.get(id);
+                    const still = yield this.getRecord(id);
                     if (still === undefined || still.status !== "armed") {
                         return { outcome: "noop", reason: "disarmed_while_waiting" };
                     }
@@ -417,7 +464,7 @@ export class ReminderRunner {
                     return { outcome: "noop", reason: "missing_handler" };
                 }
                 // Cancel / disarm during the handler must win over reschedule.
-                const latest = yield this.store.get(id);
+                const latest = yield this.getRecord(id);
                 if (result.action === "stop") {
                     if (latest !== undefined && latest.status === "armed") {
                         yield this.disarm(latest, "run_stop");
@@ -590,7 +637,7 @@ export class ReminderRunner {
     }
     afterFire(record, _outcome, reschedule) {
         return __awaiter(this, void 0, void 0, function* () {
-            const current = yield this.store.get(record.id);
+            const current = yield this.getRecord(record.id);
             if (current === undefined || current.status !== "armed") {
                 return;
             }
@@ -619,7 +666,9 @@ export class ReminderRunner {
                         ? String(reschedule.delay)
                         : String(reschedule.at), updatedAt: new Date().toISOString() });
                 yield this.persist(updated);
-                this.armTimer(updated);
+                if (this.platformTimers === undefined) {
+                    this.armTimer(updated);
+                }
                 return;
             }
             if (oneShot) {
@@ -642,7 +691,9 @@ export class ReminderRunner {
             }
             const updated = Object.assign(Object.assign({}, current), { fireCount: current.fireCount + 1, nextFireAtMs, updatedAt: new Date().toISOString() });
             yield this.persist(updated);
-            this.armTimer(updated);
+            if (this.platformTimers === undefined) {
+                this.armTimer(updated);
+            }
         });
     }
 }

package/dist/internal/reminder-store.d.ts

@@ -1,10 +1,11 @@
 /**
- * Durable reminder instance store under `<stateRoot>/reminders/`.
+ * Reminder document store used by the local control plane
+ * (`<stateRoot>/control-plane/reminders/`).
  *
  * Handlers (`run` / function `until`) are not serializable — they live only
- * in the in-memory {@link ReminderRunner} registry. Disk holds schedule,
- * prompt, target, and status so one-shots / recurring ticks survive restart
- * for prompt-form reminders.
+ * in the in-memory {@link ReminderRunner} registry. The control plane holds
+ * schedule, prompt, target, and status so prompt-form reminders survive
+ * restart.
  */
 import type { ReminderBusyPolicy, ReminderScheduleKind, ReminderStatus } from "../types.js";
 export declare const REMINDER_SNAPSHOT_VERSION: 1;

package/dist/internal/reminder-store.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"reminder-store.d.ts","sourceRoot":"","sources":["../../src/internal/reminder-store.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAWH,OAAO,KAAK,EACV,kBAAkB,EAClB,oBAAoB,EACpB,cAAc,EACf,MAAM,aAAa,CAAC;AAErB,eAAO,MAAM,yBAAyB,EAAG,CAAU,CAAC;AAEpD,YAAY,EACV,kBAAkB,EAClB,oBAAoB,EACpB,cAAc,GACf,MAAM,aAAa,CAAC;AAErB,yDAAyD;AACzD,MAAM,WAAW,cAAc;IAC7B,OAAO,EAAE,OAAO,yBAAyB,CAAC;IAC1C,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,mEAAmE;IACnE,SAAS,EAAE,MAAM,CAAC;IAClB,yEAAyE;IACzE,iBAAiB,EAAE,MAAM,CAAC;IAC1B,0DAA0D;IAC1D,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACjC,YAAY,EAAE,oBAAoB,CAAC;IACnC,4EAA4E;IAC5E,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,6DAA6D;IAC7D,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB;;;;OAIG;IACH,aAAa,EAAE,OAAO,CAAC;IACvB,yEAAyE;IACzE,UAAU,EAAE,OAAO,CAAC;IACpB,+EAA+E;IAC/E,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,kBAAkB,CAAC;IAC/B,MAAM,EAAE,cAAc,CAAC;IACvB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,yEAAyE;AACzE,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,cAAc,CAkBxE;AAED,qBAAa,aAAa;IACZ,OAAO,CAAC,QAAQ,CAAC,OAAO;IAApC,YAA6B,OAAO,EAAE,MAAM,EAAI;IAEhD,OAAO,CAAC,QAAQ;IAIV,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC,CAEhC;IAEK,IAAI,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC,CAiBtC;IAEK,SAAS,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC,CAG3C;IAEK,GAAG,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,GAAG,SAAS,CAAC,CAsBzD;IAEK,IAAI,CAAC,MAAM,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,CAYhD;IAEK,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAEtC;CACF"}
+{"version":3,"file":"reminder-store.d.ts","sourceRoot":"","sources":["../../src/internal/reminder-store.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAWH,OAAO,KAAK,EACV,kBAAkB,EAClB,oBAAoB,EACpB,cAAc,EACf,MAAM,aAAa,CAAC;AAErB,eAAO,MAAM,yBAAyB,EAAG,CAAU,CAAC;AAEpD,YAAY,EACV,kBAAkB,EAClB,oBAAoB,EACpB,cAAc,GACf,MAAM,aAAa,CAAC;AAErB,yDAAyD;AACzD,MAAM,WAAW,cAAc;IAC7B,OAAO,EAAE,OAAO,yBAAyB,CAAC;IAC1C,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,mEAAmE;IACnE,SAAS,EAAE,MAAM,CAAC;IAClB,yEAAyE;IACzE,iBAAiB,EAAE,MAAM,CAAC;IAC1B,0DAA0D;IAC1D,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACjC,YAAY,EAAE,oBAAoB,CAAC;IACnC,4EAA4E;IAC5E,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,6DAA6D;IAC7D,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB;;;;OAIG;IACH,aAAa,EAAE,OAAO,CAAC;IACvB,yEAAyE;IACzE,UAAU,EAAE,OAAO,CAAC;IACpB,+EAA+E;IAC/E,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,kBAAkB,CAAC;IAC/B,MAAM,EAAE,cAAc,CAAC;IACvB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,yEAAyE;AACzE,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,cAAc,CAkBxE;AAED,qBAAa,aAAa;IACZ,OAAO,CAAC,QAAQ,CAAC,OAAO;IAApC,YAA6B,OAAO,EAAE,MAAM,EAAI;IAEhD,OAAO,CAAC,QAAQ;IAIV,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC,CAEhC;IAEK,IAAI,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC,CAiBtC;IAEK,SAAS,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC,CAG3C;IAEK,GAAG,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,cAAc,GAAG,SAAS,CAAC,CAsBzD;IAEK,IAAI,CAAC,MAAM,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,CAYhD;IAEK,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAEtC;CACF"}

package/dist/internal/reminder-store.js

@@ -1,10 +1,11 @@
 /**
- * Durable reminder instance store under `<stateRoot>/reminders/`.
+ * Reminder document store used by the local control plane
+ * (`<stateRoot>/control-plane/reminders/`).
  *
  * Handlers (`run` / function `until`) are not serializable — they live only
- * in the in-memory {@link ReminderRunner} registry. Disk holds schedule,
- * prompt, target, and status so one-shots / recurring ticks survive restart
- * for prompt-form reminders.
+ * in the in-memory {@link ReminderRunner} registry. The control plane holds
+ * schedule, prompt, target, and status so prompt-form reminders survive
+ * restart.
  */
 var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
     function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }

package/dist/internal/sdk-runner.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"sdk-runner.d.ts","sourceRoot":"","sources":["../../src/internal/sdk-runner.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAGH,OAAO,EAOL,KAAK,eAAe,EACpB,KAAK,cAAc,EACnB,KAAK,SAAS,EAEf,MAAM,aAAa,CAAC;AACrB,OAAO,EAEL,KAAK,YAAY,EAEjB,KAAK,iBAAiB,EACtB,KAAK,WAAW,EAEjB,MAAM,aAAa,CAAC;AAErB,2EAA2E;AAC3E,MAAM,WAAW,WAAW;IAC1B,OAAO,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IAC1D;;;;;OAKG;IACH,OAAO,CAAC,CAAC,OAAO,EAAE,oBAAoB,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;IACjE,OAAO,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CAC3B;AAED;;;;GAIG;AACH,MAAM,MAAM,oBAAoB,GAAG,IAAI,CACrC,iBAAiB,EACf,cAAc,GACd,gBAAgB,GAChB,SAAS,GACT,YAAY,GACZ,WAAW,GACX,OAAO,CACV,CAAC;AAEF,6EAA6E;AAC7E,MAAM,MAAM,cAAc,GACtB;IAAE,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAA;CAAE,GACzB;IAAE,QAAQ,CAAC,MAAM,EAAE,KAAK,CAAC;IAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;CAAE,CAAC;AAcxD,MAAM,WAAW,sBAAsB;IACrC,6EAA6E;IAC7E,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB;;;;OAIG;IACH,KAAK,CAAC,EAAE,eAAe,CAAC;IACxB,6DAA6D;IAC7D,MAAM,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;CACjC;AAwBD;;;;;GAKG;AACH,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,SAAS,GAAG,OAAO,CAKvE;AAED,qBAAa,eAAgB,YAAW,WAAW;IAKrC,OAAO,CAAC,QAAQ,CAAC,OAAO;IAJpC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAwC;IAC/D,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA+C;IACtE,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAkC;IAElE,YAA6B,OAAO,GAAE,sBAA2B,EAAI;IAE/D,OAAO,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,WAAW,CAAC,CAkC9D;YAEa,cAAc;IAyE5B;;;;OAIG;IACG,OAAO,CAAC,OAAO,EAAE,oBAAoB,GAAG,OAAO,CAAC,cAAc,CAAC,CAepE;IAEK,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAY7B;IAED,OAAO,CAAC,QAAQ;IA0BhB;;;;OAIG;IACH,OAAO,CAAC,UAAU;IASlB,OAAO,CAAC,GAAG;IAIX,OAAO,CAAC,aAAa;IAUrB,OAAO,CAAC,kBAAkB;IAc1B;;;OAGG;YACW,iBAAiB;YAkBjB,SAAS;YAOT,cAAc;YA6Bd,cAAc;IA6C5B,OAAO,CAAC,QAAQ;CAUjB;AAED,4EAA4E;AAC5E,wBAAgB,0BAA0B,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAE9D;AAED;;;;;GAKG;AACH,wBAAgB,yBAAyB,CACvC,KAAK,EAAE,OAAO,EACd,OAAO,EAAE,MAAM,GACd,OAAO,CAMT;AAED,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,YAAY,GAAG,cAAc,CAQpE;AAmED,wBAAgB,eAAe,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,GAAG,MAAM,CAExE"}
+{"version":3,"file":"sdk-runner.d.ts","sourceRoot":"","sources":["../../src/internal/sdk-runner.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAGH,OAAO,EAOL,KAAK,eAAe,EACpB,KAAK,cAAc,EACnB,KAAK,SAAS,EAEf,MAAM,aAAa,CAAC;AACrB,OAAO,EAEL,KAAK,YAAY,EAEjB,KAAK,iBAAiB,EACtB,KAAK,WAAW,EAEjB,MAAM,aAAa,CAAC;AAGrB,2EAA2E;AAC3E,MAAM,WAAW,WAAW;IAC1B,OAAO,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IAC1D;;;;;OAKG;IACH,OAAO,CAAC,CAAC,OAAO,EAAE,oBAAoB,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;IACjE,OAAO,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CAC3B;AAED;;;;GAIG;AACH,MAAM,MAAM,oBAAoB,GAAG,IAAI,CACrC,iBAAiB,EACf,cAAc,GACd,gBAAgB,GAChB,SAAS,GACT,YAAY,GACZ,WAAW,GACX,OAAO,CACV,CAAC;AAEF,6EAA6E;AAC7E,MAAM,MAAM,cAAc,GACtB;IAAE,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAA;CAAE,GACzB;IAAE,QAAQ,CAAC,MAAM,EAAE,KAAK,CAAC;IAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;CAAE,CAAC;AAcxD,MAAM,WAAW,sBAAsB;IACrC,6EAA6E;IAC7E,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB;;;;OAIG;IACH,KAAK,CAAC,EAAE,eAAe,CAAC;IACxB,6DAA6D;IAC7D,MAAM,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,CAAC;CACjC;AAwBD;;;;;GAKG;AACH,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,SAAS,GAAG,OAAO,CAKvE;AAED,qBAAa,eAAgB,YAAW,WAAW;IAKrC,OAAO,CAAC,QAAQ,CAAC,OAAO;IAJpC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAwC;IAC/D,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA+C;IACtE,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAkC;IAElE,YAA6B,OAAO,GAAE,sBAA2B,EAAI;IAE/D,OAAO,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,WAAW,CAAC,CAkC9D;YAEa,cAAc;IAmF5B;;;;OAIG;IACG,OAAO,CAAC,OAAO,EAAE,oBAAoB,GAAG,OAAO,CAAC,cAAc,CAAC,CAepE;IAEK,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAY7B;IAED,OAAO,CAAC,QAAQ;IA0BhB;;;;OAIG;IACH,OAAO,CAAC,UAAU;IASlB,OAAO,CAAC,GAAG;IAIX,OAAO,CAAC,aAAa;IAUrB,OAAO,CAAC,kBAAkB;IAc1B;;;OAGG;YACW,iBAAiB;YAkBjB,SAAS;YAOT,cAAc;YA6Bd,cAAc;IA6C5B,OAAO,CAAC,QAAQ;CAUjB;AAED,4EAA4E;AAC5E,wBAAgB,0BAA0B,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAE9D;AAED;;;;;GAKG;AACH,wBAAgB,yBAAyB,CACvC,KAAK,EAAE,OAAO,EACd,OAAO,EAAE,MAAM,GACd,OAAO,CAMT;AAED,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,YAAY,GAAG,cAAc,CAQpE;AAwGD,wBAAgB,eAAe,CAAC,MAAM,EAAE,IAAI,CAAC,SAAS,EAAE,OAAO,CAAC,GAAG,MAAM,CAExE"}

package/dist/internal/sdk-runner.js

@@ -17,6 +17,7 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
 import { mkdir } from "node:fs/promises";
 import { Agent, createAgentPlatform, JsonlLocalAgentStore, } from "@cursor/sdk";
 import { DEFAULT_MODEL, } from "../types.js";
+import { mapSdkTokenUsage, usageTokenSum } from "./cloud-turn-cost.js";
 /**
  * How long one SDK agent handle (and the Connect transport inside it) may be
  * reused before the runner retires it.
@@ -127,7 +128,7 @@ export class CursorSdkRunner {
                 }
                 try {
                     const result = yield run.wait();
-                    const outcome = toTurnOutcome(result);
+                    let outcome = toTurnOutcome(result);
                     if (outcome.status === "error") {
                         // An error result usually means the handle's transport is poisoned
                         // — most often a cached access token that aged past its 60-minute
@@ -137,6 +138,15 @@ export class CursorSdkRunner {
                         // transport instead of failing every turn until a restart.
                         this.evictAgent(request.sessionId, agentPromise);
                     }
+                    else if (request.runtime === "cloud" &&
+                        outcome.status === "finished" &&
+                        usageTokenSum(outcome.usage) <= 0 &&
+                        result.id !== "") {
+                        // Cloud `wait()` drops usage after stream reconnect / getRun replay,
+                        // or returns an all-zero stub. `getUsage` still has the run's tokens
+                        // (eventually); without them traces/ABs never record an estimated cost.
+                        outcome = yield fillMissingCloudUsage(agent, result.id, outcome);
+                    }
                     return outcome;
                 }
                 finally {
@@ -389,6 +399,39 @@ function toTurnOutcome(result) {
         ? { errorMessage: runErrorMessage(result) }
         : {})), (usage === undefined ? {} : { usage }));
 }
+/** Billing can lag briefly after `wait()`; retry before giving up on cost. */
+const CLOUD_USAGE_RETRY_DELAYS_MS = [0, 400, 1200];
+/**
+ * Cloud `wait()` often omits usage after a stream drop. Pull the run's
+ * tokens from the usage API so the engine can still estimate cost.
+ * Failures are ignored: a missing estimate is better than failing the turn.
+ */
+function fillMissingCloudUsage(agent, runId, outcome) {
+    return __awaiter(this, void 0, void 0, function* () {
+        for (const delayMs of CLOUD_USAGE_RETRY_DELAYS_MS) {
+            if (delayMs > 0) {
+                yield sleep(delayMs);
+            }
+            try {
+                const billed = yield agent.getUsage({ runId });
+                const mapped = mapSdkTokenUsage(billed.usage);
+                if (usageTokenSum(mapped) <= 0) {
+                    continue;
+                }
+                return Object.assign(Object.assign({}, outcome), { usage: mapped });
+            }
+            catch (_a) {
+                // Transient usage-API failures should still burn the retry budget.
+            }
+        }
+        return outcome;
+    });
+}
+function sleep(ms) {
+    return new Promise((resolve) => {
+        setTimeout(resolve, ms);
+    });
+}
 export function runErrorMessage(result) {
     var _a, _b;
     return ((_b = (_a = result.error) === null || _a === void 0 ? void 0 : _a.message) === null || _b === void 0 ? void 0 : _b.trim()) || "The Cursor agent run failed";

package/dist/internal/server.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/internal/server.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AA4BH,OAAO,KAAK,EACV,YAAY,EACZ,gBAAgB,EAOhB,YAAY,EACb,MAAM,aAAa,CAAC;AAqFrB,OAAO,EAAE,KAAK,WAAW,EAAmB,MAAM,iBAAiB,CAAC;AAiBpE,8EAA8E;AAC9E,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,YAAY,CAAC;CACvB;AAYD,6EAA6E;AAC7E,MAAM,WAAW,0BAA0B;IACzC,oEAAoE;IACpE,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAED;;;;GAIG;AACH,wBAAsB,WAAW,CAC/B,MAAM,EAAE,UAAU,EAAE,EACpB,OAAO,GAAE,YAAY,GAAG,0BAA+B,GACtD,OAAO,CAAC,gBAAgB,CAAC,CA4pB3B"}
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/internal/server.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AA4BH,OAAO,KAAK,EACV,YAAY,EACZ,gBAAgB,EAOhB,YAAY,EACb,MAAM,aAAa,CAAC;AA8FrB,OAAO,EAAE,KAAK,WAAW,EAAmB,MAAM,iBAAiB,CAAC;AAiBpE,8EAA8E;AAC9E,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,YAAY,CAAC;CACvB;AAaD,6EAA6E;AAC7E,MAAM,WAAW,0BAA0B;IACzC,oEAAoE;IACpE,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAED;;;;GAIG;AACH,wBAAsB,WAAW,CAC/B,MAAM,EAAE,UAAU,EAAE,EACpB,OAAO,GAAE,YAAY,GAAG,0BAA+B,GACtD,OAAO,CAAC,gBAAgB,CAAC,CAsrB3B"}