@agentclientprotocol/codex-acp 2.5.1-preview.10 → 2.5.1-preview.11

package/dist/index.js

@@ -32216,6 +32216,76 @@ function normalizeJsonObject(value) {
   );
 }
 
+// src/AcpV2SessionInfo.ts
+var AcpV2SessionInfo = class {
+  constructor(send, initialMetadata = null) {
+    this.send = send;
+    this.applyMetadata(JSON.parse(JSON.stringify({ _meta: initialMetadata }))._meta);
+  }
+  send;
+  metadata = /* @__PURE__ */ new Map();
+  tail = Promise.resolve();
+  disposed = false;
+  update(update, notificationMetadata) {
+    let copy;
+    try {
+      copy = JSON.parse(JSON.stringify({ update, _meta: notificationMetadata }));
+    } catch (error62) {
+      return Promise.reject(error62);
+    }
+    return this.enqueue(async () => {
+      this.applyMetadata(copy.update._meta);
+      await this.publish(copy.update, copy._meta);
+    });
+  }
+  patchMetadata(change) {
+    return this.enqueue(async () => {
+      const current = JSON.parse(JSON.stringify(Object.fromEntries(this.metadata)));
+      const patch = change(current);
+      if (patch === null) return;
+      const update = JSON.parse(JSON.stringify({ sessionUpdate: "session_info_update", _meta: patch }));
+      this.applyMetadata(update._meta);
+      await this.publish(update);
+    });
+  }
+  seedMetadata(metadata) {
+    const copy = JSON.parse(JSON.stringify({ _meta: metadata }))._meta;
+    return this.enqueue(async () => {
+      this.applyMetadata(copy);
+    });
+  }
+  snapshot(title) {
+    return this.enqueue(async () => {
+      await this.publish({ sessionUpdate: "session_info_update", title, _meta: null });
+      if (this.metadata.size > 0) {
+        await this.publish({ sessionUpdate: "session_info_update", _meta: Object.fromEntries(this.metadata) });
+      }
+    });
+  }
+  dispose() {
+    this.disposed = true;
+    this.metadata.clear();
+  }
+  applyMetadata(patch) {
+    if (patch === null) this.metadata.clear();
+    else if (isRecord2(patch)) for (const [key, value] of Object.entries(patch)) {
+      if (value === null) this.metadata.delete(key);
+      else this.metadata.set(key, value);
+    }
+  }
+  enqueue(operation) {
+    const next = this.tail.then(async () => {
+      if (!this.disposed) await operation();
+    });
+    this.tail = next.catch(() => {
+    });
+    return next;
+  }
+  publish(update, notificationMetadata) {
+    return this.disposed ? Promise.resolve() : this.send(update, notificationMetadata);
+  }
+};
+
 // src/JsonSnapshot.ts
 import { types } from "node:util";
 var JsonObject = class {
@@ -32773,6 +32843,10 @@ var AcpV2Connection = class {
         if (method === methods2.client.elicitation.complete) return this.sendElicitationCompletion(params);
         const sessionId = clientNotificationSessionId(method, params);
         if (sessionId !== null && this.sessionDelivery && !this.sessionDelivery.canPublishSession(sessionId)) return Promise.resolve();
+        if (sessionId !== null) {
+          const { update, _meta } = params;
+          if (SessionUpdate.isSessionInfoUpdate(update)) return this.sessionInfoFor(sessionId).update(update, _meta);
+        }
         return client.notify(method, params);
       },
       request: (method, params, options) => {
@@ -32793,6 +32867,7 @@ var AcpV2Connection = class {
   client;
   isTurnRunning = () => true;
   foreground = /* @__PURE__ */ new Map();
+  sessionInfo = /* @__PURE__ */ new Map();
   sessionDelivery = null;
   sendElicitationCompletion;
   setClientDelivery(delivery) {
@@ -32817,9 +32892,12 @@ var AcpV2Connection = class {
    * Sends a session update, rendered in the v2 wire shape, through the v2 `session/update`
    * binding. One update can render to several v2 updates (or none), sent in order.
    */
-  async updateSession(sessionId, update) {
+  async updateSession(sessionId, update, metadata) {
     for (const v2Update of toV2SessionUpdates(update)) {
-      await this.client.notify(methods2.client.session.update, { sessionId, update: v2Update });
+      await this.client.notify(
+        methods2.client.session.update,
+        { sessionId, update: v2Update, ...metadata !== void 0 ? { _meta: metadata } : {} }
+      );
     }
   }
   /** Sends a `state_update`, which only exists on v2 and so needs no rendering. */
@@ -32832,6 +32910,20 @@ var AcpV2Connection = class {
   snapshotForegroundState(sessionId) {
     return this.foregroundFor(sessionId).snapshot();
   }
+  resetSessionInfo(sessionId, metadata) {
+    this.sessionInfo.get(sessionId)?.dispose();
+    this.sessionInfo.delete(sessionId);
+    this.sessionInfoFor(sessionId, metadata);
+  }
+  refreshSessionInfo(sessionId, metadata) {
+    return metadata ? this.sessionInfoFor(sessionId).seedMetadata(metadata) : Promise.resolve();
+  }
+  snapshotSessionInfo(sessionId, title) {
+    return this.sessionInfoFor(sessionId).snapshot(title);
+  }
+  patchSessionMetadata(sessionId, change) {
+    return this.sessionInfoFor(sessionId).patchMetadata(change);
+  }
   startForegroundItem(sessionId, itemId) {
     return this.foregroundFor(sessionId).startItem(itemId);
   }
@@ -32844,6 +32936,28 @@ var AcpV2Connection = class {
   forgetSession(sessionId) {
     this.foreground.get(sessionId)?.dispose();
     this.foreground.delete(sessionId);
+    this.sessionInfo.get(sessionId)?.dispose();
+    this.sessionInfo.delete(sessionId);
+  }
+  sessionInfoFor(sessionId, metadata = null) {
+    if (this.sessionDelivery && !this.sessionDelivery.canPublishSession(sessionId)) {
+      const closed = new AcpV2SessionInfo(async () => {
+      });
+      closed.dispose();
+      return closed;
+    }
+    let info = this.sessionInfo.get(sessionId);
+    if (!info) {
+      info = new AcpV2SessionInfo((update, notificationMetadata) => {
+        if (this.sessionDelivery && !this.sessionDelivery.canPublishSession(sessionId)) return Promise.resolve();
+        return this.transportClient.notify(
+          methods2.client.session.update,
+          { sessionId, update, ...notificationMetadata !== void 0 ? { _meta: notificationMetadata } : {} }
+        );
+      }, metadata);
+      this.sessionInfo.set(sessionId, info);
+    }
+    return info;
   }
   foregroundFor(sessionId, running = false) {
     if (this.sessionDelivery && !this.sessionDelivery.canPublishSession(sessionId)) {
@@ -32934,7 +33048,7 @@ var AcpV2Connection = class {
         }
         if (method === methods.client.session.update) {
           const notification = params;
-          return this.updateSession(notification.sessionId, notification.update);
+          return this.updateSession(notification.sessionId, notification.update, notification._meta);
         }
         if (method === methods.client.elicitation.complete) {
           return this.completeElicitation(params);
@@ -33003,6 +33117,10 @@ var ACPSessionConnection = class _ACPSessionConnection {
   forSession(sessionId) {
     return new _ACPSessionConnection(this.connection, sessionId);
   }
+  /** Domain writers can rebuild an owned metadata value without changing v2's shallow patch semantics. */
+  patchSessionMetadata(change) {
+    return this.v2Connection?.patchSessionMetadata(this.sessionId, change) ?? Promise.resolve();
+  }
   /** Replay shares live report filtering, with v2 entity initialization after conversion. */
   replayUpdate(update, initialized) {
     return this.v2Connection ? this.v2Connection.replaySessionUpdate(this.sessionId, update, initialized) : this.update(update);
@@ -37765,7 +37883,17 @@ ${diff}`, "");
     }
   }
   async clearSessionFailure() {
+    const failure3 = this.sessionState.sessionFailure;
     delete this.sessionState.sessionFailure;
+    if (!failure3 || this.session.protocolVersion !== 2) return;
+    await this.session.patchSessionMetadata((metadata) => {
+      const jetbrains = recordOrNull(metadata[JETBRAINS_META_KEY]);
+      const air = recordOrNull(jetbrains?.[AIR_META_KEY]);
+      const published = recordOrNull(air?.[AIR_SESSION_FAILURE_KEY]);
+      if (!air || published?.["id"] !== failure3.id || published?.["revision"] !== failure3.revision) return null;
+      const { [AIR_SESSION_FAILURE_KEY]: _failure, ...currentAir } = air;
+      return { [JETBRAINS_META_KEY]: { ...jetbrains, [AIR_META_KEY]: currentAir } };
+    });
   }
   async completeSuccessfulTurn(turnId) {
     this.lastSessionNotice = void 0;
@@ -37773,7 +37901,7 @@ ${diff}`, "");
     const active = this.sessionState.sessionFailure;
     if (active?.id !== this.activeFailureIdByScope.get(turnId) || active?.severity !== "warning") return;
     this.activeFailureIdByScope.delete(turnId);
-    delete this.sessionState.sessionFailure;
+    await this.clearSessionFailure();
   }
   async handleFailedTurn(turn) {
     if (turn.status === "failed" && this.isAuthenticationRequiredError(turn.error?.codexErrorInfo ?? null)) {
@@ -38502,7 +38630,7 @@ ${event.details}` : event.summary;
     if (activeId === void 0 || this.failuresById.get(activeId)?.severity !== "warning") return;
     this.activeFailureIdByScope.delete(turnId);
     if (this.sessionState.sessionFailure?.id === activeId) {
-      delete this.sessionState.sessionFailure;
+      void this.clearSessionFailure().catch((error62) => logger.error("Failed to clear session failure metadata", error62));
     }
   }
   recordRetryWarning(params, attributeToTurn = true) {
@@ -41095,7 +41223,7 @@ var package_default = {
   publishConfig: {
     access: "public"
   },
-  version: "2.5.1-preview.10",
+  version: "2.5.1-preview.11",
   description: "",
   main: "dist/index.js",
   bin: {
@@ -50702,7 +50830,7 @@ You have been logged out. Please try again.`);
       this.sessionIndex.titleWriter(sessionId)
     );
     sessionState.promptSuggestions = this.createPromptSuggestions(sessionState);
-    this.installSessionState(sessionState);
+    this.installSessionState(sessionState, sessionMetadata.threadStatus);
     AcpV2Connection.of(this.connection)?.restoreForegroundState(sessionId, sessionMetadata.threadStatus?.type === "active");
     this.recovery?.markLoaded(sessionState, openGeneration);
     if (settleTrackerReady) {
@@ -50865,12 +50993,22 @@ You have been logged out. Please try again.`);
       new ACPSessionConnection(this.connection, sessionId)
     );
   }
-  installSessionState(sessionState) {
+  installSessionState(sessionState, threadStatus) {
     const previous = this.sessions.get(sessionState.sessionId);
     previous?.asyncTasks.clear();
     void previous?.promptSuggestions?.clear();
     this.sessions.set(sessionState.sessionId, sessionState);
     this.clientDelivery.openSession(sessionState.sessionId);
+    const v2 = AcpV2Connection.of(this.connection);
+    if (v2) {
+      const metadata = threadStatus ? { codex: { threadStatus } } : null;
+      if (previous) {
+        if (previous.sessionFailure) sessionState.sessionFailure = previous.sessionFailure;
+        void v2.refreshSessionInfo(sessionState.sessionId, metadata);
+      } else {
+        v2.resetSessionInfo(sessionState.sessionId, metadata);
+      }
+    }
   }
   /**
    * Installs a baseline session-scoped subscription, so a Codex-initiated turn starting before
@@ -51173,10 +51311,7 @@ You have been logged out. Please try again.`);
   async snapshotSessionStateV2(sessionId) {
     await this.codexAcpClient.waitForSessionNotifications(sessionId);
     const state = this.getSessionState(sessionId);
-    await new ACPSessionConnection(this.connection, sessionId).update({
-      sessionUpdate: "session_info_update",
-      title: state.sessionTitle
-    });
+    await AcpV2Connection.of(this.connection).snapshotSessionInfo(sessionId, state.sessionTitle);
     await AcpV2Connection.of(this.connection).snapshotForegroundState(sessionId);
   }
   async forkSession(params) {
@@ -52499,7 +52634,7 @@ You have been logged out. Please try again.`);
       this.sessionIndex.titleWriter(sessionId)
     );
     sessionState.promptSuggestions = this.createPromptSuggestions(sessionState);
-    this.installSessionState(sessionState);
+    this.installSessionState(sessionState, thread.status);
     AcpV2Connection.of(this.connection)?.restoreForegroundState(sessionId, thread.status.type === "active");
     this.recovery?.markLoaded(sessionState, openGeneration);
     if (knownAccount === "pending") {

package/package.json

@@ -3,7 +3,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "2.5.1-preview.10",
+  "version": "2.5.1-preview.11",
   "description": "",
   "main": "dist/index.js",
   "bin": {