@cursor/july 0.1.107 → 0.1.109

package/dist/internal/session-engine.js

@@ -18,7 +18,6 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
 import { AsyncLocalStorage } from "node:async_hooks";
 import { mkdir, rm, writeFile } from "node:fs/promises";
 import { dirname, join } from "node:path";
-import { mergeABInstructions, resolveABMaxPlaygroundSessions, resolveABSplit, } from "../ab.js";
 import { SLACK_CHANNEL_ID } from "../channels/slack/constants.js";
 import { materializeSlackNudgeCarrier } from "../channels/slack/nudge-carrier.js";
 import { SlackAskDedupeStore, SlackNudgeStore, } from "../channels/slack/nudge-store.js";
@@ -28,19 +27,17 @@ import { cursorHostedFiles, isCursorHostedFilesAvailable, isCursorUserStoreFiles
 import { isFileSystemToolAlias, isMutatingFileSystemTool, localFileSystem, resolveFileSystemTools, resolveHostHarnessToolOptions, tempFs, unionFs, writeWorkspaceFilesToRoot, } from "../filesystem.js";
 import { storageKeysForArchitecture } from "../storage.js";
 import { DEFAULT_MODEL, DIRECT_TOOL_CALL_CHANNEL_ID, isCursorAccountTransport, isHostOauthConnectionTransport, isSymbolicConnectionTransport, } from "../types.js";
-import { ABCollector } from "./ab-collector.js";
-import { foldABStream } from "./ab-fold.js";
-import { buildABSnapshot } from "./ab-snapshot.js";
 import { resolveAdvertisedTools, } from "./advertise-tools.js";
 import { ApprovalGate } from "./approval-gate.js";
 import { ArtifactsStore } from "./artifacts-store.js";
 import { assertTimeArgs, normalizeAsOf, schemaTimeWindowKeys, sessionClock, } from "./as-of.js";
 import { SessionAuthError, samePrincipal } from "./auth.js";
+import { mergeChannelStateMutation, mergeChannelStateSkipNull, } from "./channel-state.js";
 import { mergeCloudOptions, resolveSessionRuntime } from "./cloud-merge.js";
 import { applyCloudUsageToEvents, cloudSdkAgentIdFromEvents, isUnpricedCompletedTurn, mapSdkTokenUsage, } from "./cloud-turn-cost.js";
 import { applyAgentContinuationDefaults, channelContinuationIdentity, httpContinuationIdentity, } from "./continuation-channel.js";
 import { continuationSource, resolveV2Key, V2EngineError, } from "./continuation-identity.js";
-import { describeError } from "./describe-error.js";
+import { describeError, describeErrorCode } from "./describe-error.js";
 import { isDevMode } from "./dev-mode.js";
 import { EventLogStore } from "./event-log.js";
 import { TurnEventMapper } from "./event-mapper.js";
@@ -52,6 +49,7 @@ import { createHostKvApi } from "./host-kv.js";
 import { createHostOtelApi } from "./host-otel.js";
 import { createHostPlatforms, createHostSlackClient, } from "./host-platforms.js";
 import { HostedAdmissionLocalSendError, isDuringHostedAdmission, } from "./hosted-admission-context.js";
+import { hostedSessionResumeKey } from "./hosted-catch-protocol.js";
 import { formatContinuationToken, newContinuationKey, newSessionId, newToolCallId, parseContinuationToken, } from "./ids.js";
 import { validateToolInput } from "./json-schema.js";
 import { isMcpNeedsAuthError, McpHost } from "./mcp-host.js";
@@ -75,6 +73,7 @@ import { resolveAgentV2Hosting } from "./v2-hosting.js";
 import { buildAgentsMdContent, buildCloudMemorySection, buildCloudSessionWorkspaceSection, buildCloudWorkspaceSection, ensureOwnGitRoot, isNestedInGitRepo, materializeWorkspace, SERVER_TOOLS_MCP_SERVER_NAME, } from "./workspace.js";
 import { syncSessionWorkspaceFiles, syncWorkspaceToStore, WorkspaceFileRefusal, wipeSessionWorkspace, } from "./workspace-sync.js";
 export const HTTP_CHANNEL_ID = "http";
+export { mergeChannelStateSkipNull } from "./channel-state.js";
 export { SCHEDULE_CHANNEL_PREFIX } from "./prompt-context.js";
 export class SessionBusyError extends Error {
     constructor(sessionId) {
@@ -284,9 +283,6 @@ export class SessionEngine {
             config: options.project.artifacts,
             sink: managedV2Files === null || managedV2Files === void 0 ? void 0 : managedV2Files(),
         });
-        this.abCollector = new ABCollector(options.project.abs, this.logger, (sessionId) => __awaiter(this, void 0, void 0, function* () { return (yield this.logs.get(sessionId)).snapshot(0); }), 
-        // Samples flow to the defineStorage `abs` table (no-op without one).
-        { persistSample: (sample) => { var _a; return (_a = this.storage) === null || _a === void 0 ? void 0 : _a.abSample(sample); } });
         // Peer and Cursor-account transports are symbolic until the serve host
         // resolves them; the host-side MCP registry starts with the concrete
         // connections and gains the rest via setResolvedConnections.
@@ -570,8 +566,8 @@ export class SessionEngine {
      */
     send(channelId_1, message_1) {
         return __awaiter(this, arguments, void 0, function* (channelId, message, options = {}) {
-            var _a;
-            var _b, _c, _d, _e, _f, _g;
+            var _a, _b;
+            var _c, _d, _e, _f, _g, _h;
             if (isDuringHostedAdmission()) {
                 // Only sends that bypassed the admission capture stub (typically a
                 // startup-captured `onStart` send) reach the engine inside the
@@ -592,7 +588,7 @@ export class SessionEngine {
                 sendOptions = Object.assign(Object.assign({}, options), { continuationToken: cKey });
                 skipContinuationLookup = cKey.startsWith("evt:");
             }
-            const auth = (_b = sendOptions.auth) !== null && _b !== void 0 ? _b : null;
+            const auth = (_c = sendOptions.auth) !== null && _c !== void 0 ? _c : null;
             let existing = skipContinuationLookup || sendOptions.continuationToken === undefined
                 ? undefined
                 : yield this.sessions.findByContinuation(channelId, sendOptions.continuationToken);
@@ -603,6 +599,16 @@ export class SessionEngine {
                 // fell outside the startup restore window, or restore is lazy-only).
                 existing = yield this.restoreSessionByContinuation(channelId, sendOptions.continuationToken);
             }
+            if (existing === undefined &&
+                sendOptions.hostedSessionId !== undefined &&
+                ((_a = sendOptions.mailboxContinuation) === null || _a === void 0 ? void 0 : _a.key) ===
+                    hostedSessionResumeKey(sendOptions.hostedSessionId)) {
+                // Explicit control-plane resume of an existing session: the CP named
+                // the session by its own id and stamped the matching resume key. The
+                // session keeps its original continuation key (a PR lane must stay
+                // addressable by its webhooks); this is a follow-up turn by id.
+                existing = yield this.getOrRestoreSession(sendOptions.hostedSessionId);
+            }
             if (existing !== undefined) {
                 if (sendOptions.hostedSessionId !== undefined &&
                     existing.sessionId !== sendOptions.hostedSessionId) {
@@ -638,7 +644,7 @@ export class SessionEngine {
                     record = yield this.updateRecord(record.sessionId, (current) => (Object.assign(Object.assign({}, current), { channelState: mergeChannelStateSkipNull(current.channelState, sendOptions.refreshState) })));
                 }
                 const since = (yield this.logs.get(record.sessionId)).count;
-                const coalesced = yield this.startTurn(record.sessionId, message, Object.assign({ admission: (_c = sendOptions.admission) !== null && _c !== void 0 ? _c : "preempt" }, (sendOptions.coalesceSourceTs === undefined
+                const coalesced = yield this.startTurn(record.sessionId, message, Object.assign({ admission: (_d = sendOptions.admission) !== null && _d !== void 0 ? _d : "preempt" }, (sendOptions.coalesceSourceTs === undefined
                     ? {}
                     : { sourceTs: sendOptions.coalesceSourceTs })));
                 return this.channelSession(record.sessionId, record, false, coalesced, since);
@@ -654,14 +660,14 @@ export class SessionEngine {
             const purpose = sendOptions.purpose === "eval" ? "eval" : "live";
             const frozenAsOf = normalizeAsOf(sendOptions.asOf);
             assertTimeArgs(this.project.agent.tools, frozenAsOf);
-            let record = {
-                sessionId: (_d = sendOptions.hostedSessionId) !== null && _d !== void 0 ? _d : newSessionId(),
+            const record = {
+                sessionId: (_e = sendOptions.hostedSessionId) !== null && _e !== void 0 ? _e : newSessionId(),
                 channelId,
-                mode: (_e = sendOptions.mode) !== null && _e !== void 0 ? _e : "chat",
+                mode: (_f = sendOptions.mode) !== null && _f !== void 0 ? _f : "chat",
                 purpose,
                 dryRun: sendOptions.dryRun === true ? true : undefined,
                 asOf: frozenAsOf,
-                continuationKey: (_f = sendOptions.continuationToken) !== null && _f !== void 0 ? _f : newContinuationKey(),
+                continuationKey: (_g = sendOptions.continuationToken) !== null && _g !== void 0 ? _g : newContinuationKey(),
                 auth,
                 title: sendOptions.title,
                 channelState,
@@ -675,7 +681,7 @@ export class SessionEngine {
                 updatedAt: now,
             };
             yield this.sessions.save(record);
-            (_a = this.storage) === null || _a === void 0 ? void 0 : _a.sessionRecord(record);
+            (_b = this.storage) === null || _b === void 0 ? void 0 : _b.sessionRecord(record);
             yield this.appendEvent(record.sessionId, {
                 type: "session.started",
                 data: {
@@ -683,9 +689,8 @@ export class SessionEngine {
                     dryRun: record.dryRun === true ? true : undefined,
                 },
             });
-            record = yield this.enrollABs(record);
             const since = (yield this.logs.get(record.sessionId)).count;
-            yield this.startTurn(record.sessionId, message, Object.assign({ admission: (_g = sendOptions.admission) !== null && _g !== void 0 ? _g : "preempt" }, (sendOptions.coalesceSourceTs === undefined
+            yield this.startTurn(record.sessionId, message, Object.assign({ admission: (_h = sendOptions.admission) !== null && _h !== void 0 ? _h : "preempt" }, (sendOptions.coalesceSourceTs === undefined
                 ? {}
                 : { sourceTs: sendOptions.coalesceSourceTs })));
             return this.channelSession(record.sessionId, record, true, false, since);
@@ -741,85 +746,6 @@ export class SessionEngine {
             return existing !== undefined;
         });
     }
-    /**
-     * Sticky A/B enrollment at session creation — before the first turn, so
-     * variants are in the durable stream (and on {@link SessionRecord.abs})
-     * ahead of model/instructions work. Failures persist as `variant: null`
-     * (sticky skip). Eval harness sessions (`purpose: "eval"`) are never
-     * enrolled.
-     */
-    enrollABs(record) {
-        return __awaiter(this, void 0, void 0, function* () {
-            if (!this.abCollector.enabled || record.purpose === "eval") {
-                return record;
-            }
-            const sessionInfo = this.sessionInfo(record);
-            const abs = {};
-            for (const experiment of this.project.abs) {
-                const def = experiment.definition;
-                const variantKeys = Object.keys(def.variants);
-                let variant = null;
-                try {
-                    variant = yield resolveABSplit(def.split, {
-                        agent: { name: this.project.name },
-                        channel: { id: record.channelId },
-                        session: sessionInfo,
-                        experiment: experiment.name,
-                        variants: variantKeys,
-                    });
-                    if (variant !== null && !(variant in def.variants)) {
-                        throw new Error(`split resolved to ${JSON.stringify(variant)}; expected one of ${variantKeys.join(", ")} or null to skip`);
-                    }
-                }
-                catch (error) {
-                    this.logger(`[agent-sdk] ab "${experiment.name}" enrollment failed (sticky skip): ${describeError(error)}`);
-                    variant = null;
-                }
-                abs[experiment.name] = variant;
-                yield this.appendEvent(record.sessionId, {
-                    type: "ab.assigned",
-                    data: { experiment: experiment.name, variant },
-                });
-            }
-            return this.updateRecord(record.sessionId, (current) => (Object.assign(Object.assign({}, current), { abs })));
-        });
-    }
-    /**
-     * Ensure {@link SessionRecord.abs} is populated (hydrate from the durable
-     * `ab.assigned` stream when missing — e.g. older sessions after upgrade).
-     */
-    ensureAbs(record) {
-        return __awaiter(this, void 0, void 0, function* () {
-            if (record.abs !== undefined || record.purpose === "eval") {
-                return record;
-            }
-            if (!this.abCollector.enabled) {
-                return record;
-            }
-            const events = yield this.listEvents(record.sessionId, 0);
-            const folded = foldABStream(events);
-            if (folded.arms.size === 0) {
-                return record;
-            }
-            const abs = Object.fromEntries(folded.arms);
-            return this.updateRecord(record.sessionId, (current) => (Object.assign(Object.assign({}, current), { abs })));
-        });
-    }
-    /**
-     * Agent config for this session: base project agent plus any per-variant
-     * `instructions` overlays from enrolled A/B arms.
-     */
-    agentForSession(record) {
-        const base = this.project.agent;
-        const instructions = mergeABInstructions(base.instructions, record.abs, this.project.abs.map((experiment) => ({
-            name: experiment.name,
-            definition: experiment.definition,
-        })));
-        if (instructions === base.instructions) {
-            return base;
-        }
-        return Object.assign(Object.assign({}, base), { instructions });
-    }
     /** Look up a session as a channel-scoped handle. */
     getChannelSession(channelId, sessionId) {
         return __awaiter(this, void 0, void 0, function* () {
@@ -1094,9 +1020,7 @@ export class SessionEngine {
             this.pendingTurns.add(sessionId);
             let lifecycleInstalled = false;
             try {
-                let record = yield this.updateRecord(sessionId, (current) => (Object.assign(Object.assign({}, current), { turnCount: current.turnCount + 1 })));
-                // Arms must be on the record before tools/instructions see SessionInfo.
-                record = yield this.ensureAbs(record);
+                const record = yield this.updateRecord(sessionId, (current) => (Object.assign(Object.assign({}, current), { turnCount: current.turnCount + 1 })));
                 const turnId = `turn_${record.turnCount}`;
                 const isFirstTurn = record.turnCount === 1;
                 yield this.appendEvent(sessionId, {
@@ -1118,11 +1042,17 @@ export class SessionEngine {
                     poll.unref();
                     try {
                         outcome = yield this.executeTurn(record, turnId, message, isFirstTurn, abort.signal);
+                        if (outcome.status === "cancelled") {
+                            outcome.cancellationReason = abort.signal.aborted
+                                ? "signal"
+                                : "runner";
+                        }
                     }
                     catch (error) {
                         outcome = {
                             status: "error",
                             errorMessage: describeError(error),
+                            errorCode: describeErrorCode(error),
                         };
                     }
                     finally {
@@ -1562,10 +1492,15 @@ export class SessionEngine {
     }
     executeTurn(record, turnId, message, isFirstTurn, signal) {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b, _c, _d, _e, _f, _g, _h;
+            var _a, _b, _c, _d, _e, _f, _g, _h, _j;
             const sessionId = record.sessionId;
             const workspaceDir = this.scaffoldDir(record);
             const harnessCwd = this.harnessCwd(record);
+            const turnChannelState = structuredClone((_a = record.channelState) !== null && _a !== void 0 ? _a : null);
+            const eventContext = {
+                channelState: structuredClone(turnChannelState),
+                session: this.sessionInfo(record),
+            };
             // Hold the open log so tool events append synchronously. Fire-and-forget
             // `appendEvent` races parallel tool completions and delays the live stream
             // until the next await tick — often after *all* parallel tools finish.
@@ -1573,7 +1508,7 @@ export class SessionEngine {
             const emit = (payload) => {
                 try {
                     const event = log.append(Object.assign(Object.assign({}, payload), { turnId }));
-                    this.queueDispatch(sessionId, event);
+                    this.queueDispatch(sessionId, event, eventContext);
                 }
                 catch (error) {
                     this.logger(`[agent-sdk] failed to record event for ${sessionId}: ${describeError(error)}`);
@@ -1606,12 +1541,10 @@ export class SessionEngine {
                 emit({ type: "turn.started", data: {} });
                 try {
                     // Scaffold agent files into the session workspace even when the harness
-                    // cwd is overridden (e.g. a PR worktree). A/B variant instruction
-                    // overlays are applied via agentForSession.
-                    const agent = this.agentForSession(record);
+                    // cwd is overridden (e.g. a PR worktree).
                     if (!this.hasRootOverride()) {
                         yield materializeWorkspace({
-                            agent,
+                            agent: this.project.agent,
                             workspaceDir,
                             seed: isFirstTurn,
                             stateRoot: this.stateRoot,
@@ -1633,14 +1566,14 @@ export class SessionEngine {
                     const pendingFiles = record.pendingWorkspaceFiles;
                     const pendingImages = record.pendingImages;
                     const runtime = resolveSessionRuntime({
-                        agentRuntime: agent.runtime,
+                        agentRuntime: this.project.agent.runtime,
                         cloudOverride: record.cloudOverride,
                     });
                     const hasPendingFiles = pendingFiles !== undefined && Object.keys(pendingFiles).length > 0;
                     if (hasPendingFiles && runtime !== "cloud") {
                         const root = yield this.resolveFileSystem({
                             sessionId,
-                            continuationKey: (_a = record.continuationKey) !== null && _a !== void 0 ? _a : undefined,
+                            continuationKey: (_b = record.continuationKey) !== null && _b !== void 0 ? _b : undefined,
                             cache: true,
                             cwd: harnessCwd,
                         });
@@ -1664,7 +1597,7 @@ export class SessionEngine {
                     const instructionsInvisibleToHarness = isFirstTurn &&
                         (this.hasBorrowedWorkspace(record) ||
                             (yield isNestedInGitRepo(harnessCwd)));
-                    const request = yield this.buildTurnRequest(record, agent, turnId, message, isFirstTurn, instructionsInvisibleToHarness, emit, (update) => __awaiter(this, void 0, void 0, function* () {
+                    const request = yield this.buildTurnRequest(record, turnId, message, isFirstTurn, instructionsInvisibleToHarness, emit, eventContext, (update) => __awaiter(this, void 0, void 0, function* () {
                         const turn = this.turnRoots.get(sessionId);
                         if (turn !== undefined) {
                             update = yield applyWebScratchUpdate({
@@ -1700,7 +1633,11 @@ export class SessionEngine {
                 catch (error) {
                     this.approvals.cancelSession(sessionId, "deny");
                     this.auths.cancelSession(sessionId, "deny");
-                    outcome = { status: "error", errorMessage: describeError(error) };
+                    outcome = {
+                        status: "error",
+                        errorMessage: describeError(error),
+                        errorCode: describeErrorCode(error),
+                    };
                 }
                 finally {
                     permit.release();
@@ -1718,16 +1655,17 @@ export class SessionEngine {
             if (mapper.title !== undefined) {
                 const title = mapper.title;
                 yield this.updateRecord(sessionId, (current) => (Object.assign(Object.assign({}, current), { title })));
+                eventContext.session = Object.assign(Object.assign({}, eventContext.session), { title });
             }
-            const finalResult = (_b = outcome.result) !== null && _b !== void 0 ? _b : mapper.finalText;
-            const finalUsage = (_c = outcome.usage) !== null && _c !== void 0 ? _c : mapper.usage;
+            const finalResult = (_c = outcome.result) !== null && _c !== void 0 ? _c : mapper.finalText;
+            const finalUsage = (_d = outcome.usage) !== null && _d !== void 0 ? _d : mapper.usage;
             const finalCost = finalUsage === undefined
                 ? undefined
                 : yield this.estimateTurnCost({
-                    model: (_d = this.project.agent.model) !== null && _d !== void 0 ? _d : DEFAULT_MODEL,
+                    model: (_e = this.project.agent.model) !== null && _e !== void 0 ? _e : DEFAULT_MODEL,
                     usage: finalUsage,
                     agentId: record.sdkAgentId,
-                    conversationId: (_e = record.sdkAgentId) !== null && _e !== void 0 ? _e : sessionId,
+                    conversationId: (_f = record.sdkAgentId) !== null && _f !== void 0 ? _f : sessionId,
                 });
             if (finalCost !== undefined) {
                 outcome = Object.assign(Object.assign({}, outcome), { cost: finalCost });
@@ -1739,7 +1677,8 @@ export class SessionEngine {
                     text: finalResult,
                     dryRun: record.dryRun === true,
                     ctx: {
-                        session: this.sessionInfo(record),
+                        session: eventContext.session,
+                        channelState: turnChannelState,
                         host: this.hostForSession(sessionId),
                         artifacts: this.artifactsApi({ sessionId, turnId }),
                         stateRoot: this.stateRoot,
@@ -1765,7 +1704,7 @@ export class SessionEngine {
                         usage: finalUsage,
                         cost: finalCost,
                     },
-                });
+                }, eventContext);
             }
             else {
                 const interrupted = signal.aborted && outcome.status === "cancelled";
@@ -1773,24 +1712,25 @@ export class SessionEngine {
                     type: "turn.failed",
                     turnId,
                     data: {
+                        status: outcome.status,
                         message: interrupted
                             ? "turn interrupted"
-                            : ((_f = outcome.errorMessage) !== null && _f !== void 0 ? _f : (outcome.status === "cancelled"
+                            : ((_g = outcome.errorMessage) !== null && _g !== void 0 ? _g : (outcome.status === "cancelled"
                                 ? "turn cancelled"
                                 : "turn failed")),
                     },
-                });
+                }, eventContext);
             }
             const current = yield this.sessions.get(sessionId);
-            const mode = (_g = current === null || current === void 0 ? void 0 : current.mode) !== null && _g !== void 0 ? _g : record.mode;
+            const mode = (_h = current === null || current === void 0 ? void 0 : current.mode) !== null && _h !== void 0 ? _h : record.mode;
             if (mode === "task") {
                 yield this.appendEvent(sessionId, outcome.status === "finished"
                     ? { type: "session.completed", turnId, data: {} }
                     : {
                         type: "session.failed",
                         turnId,
-                        data: { message: (_h = outcome.errorMessage) !== null && _h !== void 0 ? _h : "task failed" },
-                    });
+                        data: { message: (_j = outcome.errorMessage) !== null && _j !== void 0 ? _j : "task failed" },
+                    }, eventContext);
                 yield this.wipeCloudSessionWorkspace(sessionId);
             }
             else {
@@ -1798,7 +1738,7 @@ export class SessionEngine {
                     type: "session.waiting",
                     turnId,
                     data: {},
-                });
+                }, eventContext);
             }
             return Object.assign(Object.assign({}, outcome), { result: finalResult, usage: finalUsage });
         });
@@ -1952,10 +1892,11 @@ export class SessionEngine {
         }
         return undefined;
     }
-    buildTurnRequest(record, agent, turnId, prompt, isFirstTurn, instructionsInvisibleToHarness, emit, onUpdate, signal, images) {
+    buildTurnRequest(record, turnId, prompt, isFirstTurn, instructionsInvisibleToHarness, emit, eventContext, onUpdate, signal, images) {
         return __awaiter(this, void 0, void 0, function* () {
             var _a, _b, _c;
             var _d;
+            const agent = this.project.agent;
             // Per-session cloud attach (channel/tool handoff) forces cloud runtime
             // even when the agent defaults to local for chat server tools.
             const runtime = resolveSessionRuntime({
@@ -2124,9 +2065,7 @@ export class SessionEngine {
             }
             return Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign({ sessionId: record.sessionId }, (record.sdkAgentId === undefined
                 ? {}
-                : { sdkAgentId: record.sdkAgentId })), { workspaceDir: this.harnessCwd(record), sandbox: authoredRoot
-                    ? false
-                    : ((_c = this.project.agent.local) === null || _c === void 0 ? void 0 : _c.sandbox) === true }), (runtime === "local" && harnessTools !== undefined
+                : { sdkAgentId: record.sdkAgentId })), { workspaceDir: this.harnessCwd(record), sandbox: authoredRoot ? false : ((_c = agent.local) === null || _c === void 0 ? void 0 : _c.sandbox) === true }), (runtime === "local" && harnessTools !== undefined
                 ? Object.assign(Object.assign({}, (harnessTools.tools === undefined
                     ? {}
                     : { tools: harnessTools.tools })), (harnessTools.disallowedTools === undefined
@@ -2142,7 +2081,7 @@ export class SessionEngine {
                     : undefined }), (runtime === "cloud" && !isCursorHostedFilesAvailable()
                 ? { agentServeAgent: this.project.name }
                 : {})), { customTools: runtime === "local"
-                    ? yield this.buildCustomTools(record, agent, turnId, emit, synthesized)
+                    ? yield this.buildCustomTools(record, turnId, emit, synthesized)
                     : {}, mcpServers: this.buildMcpServers(runtime, record.sessionId, yield this.evaluateConnectionAuth(record)) }), (runtime === "local" && this.hasAttachedAuthedConnections()
                 ? { perSendMcpServers: true }
                 : {})), { subagents: buildRunnerSubagents(agent), onUpdate,
@@ -2157,13 +2096,14 @@ export class SessionEngine {
                     }
                     yield this.updateRecord(record.sessionId, (current) => (Object.assign(Object.assign({}, current), { sdkAgentId })));
                     record.sdkAgentId = sdkAgentId;
+                    eventContext.session = Object.assign(Object.assign({}, eventContext.session), { sdkAgentId });
                     yield this.appendEvent(record.sessionId, {
                         type: "agent.bound",
                         turnId,
                         data: Object.assign({ sdkAgentId }, (sdkAgentId.startsWith("bc-")
                             ? { url: `https://cursor.com/agents/${sdkAgentId}` }
                             : {})),
-                    });
+                    }, eventContext);
                 }) });
         });
     }
@@ -2365,8 +2305,9 @@ export class SessionEngine {
             return evaluated;
         });
     }
-    buildCustomTools(record_1, agent_1, turnId_1, emit_1) {
-        return __awaiter(this, arguments, void 0, function* (record, agent, turnId, emit, extraTools = []) {
+    buildCustomTools(record_1, turnId_1, emit_1) {
+        return __awaiter(this, arguments, void 0, function* (record, turnId, emit, extraTools = []) {
+            const agent = this.project.agent;
             const tools = {};
             const register = (tool) => {
                 tools[tool.name] = Object.assign(Object.assign(Object.assign({ description: tool.description, inputSchema: tool.inputSchema }, (tool.outputSchema === undefined
@@ -3389,15 +3330,15 @@ export class SessionEngine {
     // ==========================================================================
     // Events: append, dispatch to channel handlers + hooks
     // ==========================================================================
-    appendEvent(sessionId, payload) {
+    appendEvent(sessionId, payload, context) {
         return __awaiter(this, void 0, void 0, function* () {
             const log = yield this.logs.get(sessionId);
             const event = log.append(payload);
-            this.queueDispatch(sessionId, event);
+            this.queueDispatch(sessionId, event, context);
             return event;
         });
     }
-    queueDispatch(sessionId, event) {
+    queueDispatch(sessionId, event, context) {
         var _a;
         var _b;
         // Single funnel for both append paths (appendEvent + the held-log emit),
@@ -3405,7 +3346,7 @@ export class SessionEngine {
         (_a = this.storage) === null || _a === void 0 ? void 0 : _a.event(event);
         const chain = (_b = this.dispatchChains.get(sessionId)) !== null && _b !== void 0 ? _b : Promise.resolve();
         const next = chain
-            .then(() => this.dispatchEvent(sessionId, event))
+            .then(() => this.dispatchEvent(sessionId, event, context))
             .catch((error) => {
             this.logger(`[agent-sdk] event dispatch failed for ${sessionId}: ${describeError(error)}`);
         })
@@ -3418,17 +3359,25 @@ export class SessionEngine {
         });
         this.dispatchChains.set(sessionId, next);
     }
-    dispatchEvent(sessionId, event) {
+    dispatchEvent(sessionId, event, context) {
         return __awaiter(this, void 0, void 0, function* () {
             var _a;
-            var _b;
+            var _b, _c;
             const record = yield this.sessions.get(sessionId);
             if (record === undefined) {
                 return;
             }
+            const channelState = context === undefined
+                ? ((_b = record.channelState) !== null && _b !== void 0 ? _b : null)
+                : context.channelState;
+            const channelStateBefore = context === undefined ? undefined : structuredClone(channelState);
             const sessionCtx = {
-                session: this.sessionInfo(record),
+                session: (_c = context === null || context === void 0 ? void 0 : context.session) !== null && _c !== void 0 ? _c : this.sessionInfo(record),
                 host: this.hostForSession(record.sessionId),
+                artifacts: this.artifactsApi({
+                    sessionId: record.sessionId,
+                    turnId: event.turnId,
+                }),
             };
             const channelEvents = this.channelEventHandlersFor(record.channelId);
             if (channelEvents !== undefined) {
@@ -3440,7 +3389,7 @@ export class SessionEngine {
                 ];
                 const channelCtx = {
                     channelId: record.channelId,
-                    state: ((_b = record.channelState) !== null && _b !== void 0 ? _b : null),
+                    state: channelState,
                     continuationToken: record.continuationKey,
                     setContinuationToken: (token) => this.setContinuationToken(sessionId, token),
                 };
@@ -3455,8 +3404,19 @@ export class SessionEngine {
                         this.logger(`[agent-sdk] channel "${record.channelId}" handler for ${event.type} threw: ${describeError(error)}`);
                     }
                 }
-                // Handlers mutate channel state in place; persist after each batch.
-                yield this.updateRecord(sessionId, (current) => (Object.assign({}, current)));
+                if (context === undefined || channelStateBefore === undefined) {
+                    // Handlers mutate live channel state in place; save the current record.
+                    yield this.updateRecord(sessionId, (current) => (Object.assign({}, current)));
+                }
+                else {
+                    const updatedState = channelCtx.state;
+                    context.channelState = updatedState;
+                    yield this.updateRecord(sessionId, (current) => (Object.assign(Object.assign({}, current), { channelState: mergeChannelStateMutation({
+                            current: current.channelState,
+                            before: channelStateBefore,
+                            after: updatedState,
+                        }) })));
+                }
             }
             for (const hook of this.project.hooks) {
                 const handlers = [
@@ -3471,19 +3431,13 @@ export class SessionEngine {
                         yield handler(event, Object.assign(Object.assign({}, sessionCtx), { agent: { name: this.project.name }, channel: {
                                 id: record.channelId,
                                 continuationToken: record.continuationKey,
-                            }, stateRoot: this.stateRoot, artifacts: this.artifactsApi({
-                                sessionId: record.sessionId,
-                                turnId: event.turnId,
-                            }) }));
+                            }, stateRoot: this.stateRoot }));
                     }
                     catch (error) {
                         this.logger(`[agent-sdk] hook "${hook.name}" handler for ${event.type} threw: ${describeError(error)}`);
                     }
                 }
             }
-            if (this.abCollector.enabled) {
-                yield this.abCollector.onEvent(event, sessionCtx.session, this.project.name);
-            }
             (_a = this.otel) === null || _a === void 0 ? void 0 : _a.onEvent(event, sessionCtx.session);
         });
     }
@@ -3542,6 +3496,10 @@ export class SessionEngine {
                     yield active;
                     continue;
                 }
+                const dispatch = this.dispatchChains.get(sessionId);
+                if (dispatch !== undefined) {
+                    yield dispatch.catch(() => { });
+                }
                 return (_a = this.lastOutcomes.get(sessionId)) !== null && _a !== void 0 ? _a : { status: "finished" };
             }
         });
@@ -3756,57 +3714,6 @@ export class SessionEngine {
             };
         });
     }
-    /**
-     * Fold durable session streams into per-session A/B assignments and
-     * aggregate arm metrics (`GET /v1/abs`). Collector-independent: always
-     * reads `events.ndjson`.
-     */
-    abSnapshot(caller, options) {
-        return __awaiter(this, void 0, void 0, function* () {
-            var _a, _b;
-            var _c;
-            const sessions = yield this.listSessions(caller, options);
-            const abConfig = this.project.abConfig;
-            const maxPlaygroundSessions = resolveABMaxPlaygroundSessions(abConfig === null || abConfig === void 0 ? void 0 : abConfig.maxPlaygroundSessions);
-            const foldedSessions = sessions.slice(0, maxPlaygroundSessions);
-            // Backfill for a replacement host with nothing local to fold (e.g.
-            // lazy-only restore before any session resumed): serve the persisted
-            // aggregate instead of an empty surface. Gated on the host having no
-            // sessions AT ALL — the persisted snapshot spans every principal, so a
-            // caller who merely owns no sessions must not see it. Loaded once per
-            // process (playground polls this route ~every 4s); skip the persist
-            // side effect below — re-saving a restored snapshot would echo stale
-            // data back into the store.
-            if (foldedSessions.length === 0 &&
-                this.storage !== undefined &&
-                (yield this.sessions.list()).length === 0) {
-                (_c = this.restoredAbSnapshot) !== null && _c !== void 0 ? _c : (this.restoredAbSnapshot = this.storage.getLatestAbSnapshot());
-                const restoredSnapshot = yield this.restoredAbSnapshot;
-                if (restoredSnapshot !== undefined) {
-                    return restoredSnapshot;
-                }
-            }
-            const abTable = (_a = this.project.storage) === null || _a === void 0 ? void 0 : _a.abs;
-            const snapshot = yield buildABSnapshot({
-                experiments: this.project.abs,
-                agentName: this.project.name,
-                sessions: foldedSessions,
-                loadRecord: (sessionId) => this.sessions.get(sessionId),
-                loadEvents: (sessionId) => this.listEvents(sessionId, 0),
-                sessionInfo: (record) => this.sessionInfo(record),
-                logger: this.logger,
-                config: {
-                    maxPlaygroundSessions,
-                    durableSamples: abTable !== undefined,
-                    durableSnapshots: (abTable === null || abTable === void 0 ? void 0 : abTable.putSnapshot) !== undefined,
-                },
-            });
-            // Side effect on the read path: playground polls GET /v1/abs ~every 4s;
-            // the coordinator throttles the actual table write.
-            (_b = this.storage) === null || _b === void 0 ? void 0 : _b.abSnapshot(snapshot);
-            return snapshot;
-        });
-    }
     eventStream(sessionId_1) {
         return __awaiter(this, arguments, void 0, function* (sessionId, startIndex = 0) {
             // Playground Trace replays this stream; backfill before the snapshot
@@ -4017,7 +3924,7 @@ export class SessionEngine {
             this.coalesceDrains.has(sessionId));
     }
     sessionInfo(record) {
-        return Object.assign(Object.assign(Object.assign({ id: record.sessionId, channelId: record.channelId, mode: record.mode, purpose: record.purpose === "eval" ? "eval" : "live", dryRun: record.dryRun === true ? true : undefined, asOf: record.asOf, auth: record.auth, continuationKey: record.continuationKey }, (record.abs === undefined ? {} : { abs: record.abs })), (record.title === undefined ? {} : { title: record.title })), (record.sdkAgentId === undefined
+        return Object.assign(Object.assign({ id: record.sessionId, channelId: record.channelId, mode: record.mode, purpose: record.purpose === "eval" ? "eval" : "live", dryRun: record.dryRun === true ? true : undefined, asOf: record.asOf, auth: record.auth, continuationKey: record.continuationKey }, (record.title === undefined ? {} : { title: record.title })), (record.sdkAgentId === undefined
             ? {}
             : { sdkAgentId: record.sdkAgentId }));
     }
@@ -4079,9 +3986,9 @@ export class SessionEngine {
      * serve start, before channels accept traffic: any persisted session
      * missing from the local `--state-root` is materialized there — events
      * as `events.ndjson`, record as `session.json` — after which every
-     * existing read path (continuation lookup, event replay/streams, A/B
-     * fold hydration) works unchanged. Local state always wins over the
-     * sink, so restarts on the same host are no-ops.
+     * existing read path (continuation lookup, event replay/streams) works
+     * unchanged. Local state always wins over the sink, so restarts on the
+     * same host are no-ops.
      *
      * Failures are logged and skipped; a broken store (including rows that
      * violate the SessionRecord shape) never blocks serve start.
@@ -4345,30 +4252,6 @@ function mergePendingWorkspaceFiles(current, incoming) {
     }
     return Object.assign(Object.assign({}, current), incoming);
 }
-/**
- * Shallow-merge a wake patch into durable channel state. Null / undefined
- * patch values leave the existing key alone so sticky fields survive.
- */
-export function mergeChannelStateSkipNull(current, patch) {
-    if (patch === undefined) {
-        return current;
-    }
-    if (typeof patch !== "object" ||
-        patch === null ||
-        Array.isArray(patch) ||
-        typeof current !== "object" ||
-        current === null ||
-        Array.isArray(current)) {
-        return structuredClone(patch);
-    }
-    const merged = Object.assign({}, current);
-    for (const [key, value] of Object.entries(patch)) {
-        if (value !== null && value !== undefined) {
-            merged[key] = value;
-        }
-    }
-    return merged;
-}
 function eventsNdjson(events) {
     if (events.length === 0) {
         return undefined;