@moxt-ai/mobius 0.0.177 → 0.0.178

package/dist/index.js

@@ -20954,6 +20954,7 @@ var MAX_CONTENT_LENGTH = 1e6;
 var MAX_JSON_DEPTH = 16;
 var MAX_JSON_KEYS = 1e3;
 var SESSION_EVENT_V2_KIND = "session.event.v2";
+var SESSION_EVENT_V2_BATCH_MAX_SIZE = 1e3;
 var boundedString = external_exports.string().max(MAX_CONTENT_LENGTH);
 var shortString = external_exports.string().max(2048);
 var externalContentKeySchema = external_exports.string().regex(/^[a-f0-9]{64}$/u);
@@ -21325,6 +21326,7 @@ var sessionEventV2Schema = external_exports.discriminatedUnion("type", [
     type: external_exports.literal("lifecycle-metadata")
   }).readonly()
 ]);
+var sessionEventBatchV2Schema = external_exports.array(sessionEventV2Schema).min(1).max(SESSION_EVENT_V2_BATCH_MAX_SIZE).readonly();
 var sessionExportV2Schema = external_exports.object({
   binding: external_exports.object({
     bindingId: identifierStringSchema,
@@ -21367,7 +21369,7 @@ var mobiusSessionExportV2Schema = external_exports.object({
   formatVersion: external_exports.literal("2"),
   session: sessionExportV2Schema
 }).readonly();
-var sessionEventEnvelopeV2Schema = external_exports.object({ event: sessionEventV2Schema, kind: external_exports.literal(SESSION_EVENT_V2_KIND) }).readonly();
+var sessionEventEnvelopeV2Schema = external_exports.object({ events: sessionEventBatchV2Schema, kind: external_exports.literal(SESSION_EVENT_V2_KIND) }).readonly();
 var sessionEventLogPageV2Schema = external_exports.object({
   daemonGenerationId: identifierStringSchema,
   events: external_exports.array(sessionEventV2Schema).max(MAX_EVENT_COUNT).readonly(),
@@ -23339,14 +23341,13 @@ var AgentConnection = class {
   get inProcessAgent() {
     return this.#process.kind === "in-process" ? this.#process.endpoint : null;
   }
-  captureProcessTree = async () => this.#process.kind === "child" ? await captureProcessTree(this.#process.child) : [];
-  closeProcess = async (capturedProcesses = []) => {
+  closeProcess = async () => {
     try {
       if (this.#process.kind === "in-process") {
         await this.#process.endpoint.close();
         return;
       }
-      await stopProcessTree(this.#process.child, capturedProcesses);
+      await stopProcessTree(this.#process.child);
     } finally {
       this.connection.close();
     }
@@ -30973,16 +30974,11 @@ var AgentSession = class _AgentSession {
     }
   };
   cancel = async () => {
-    const capturedProcesses = await this.#connection.captureProcessTree();
-    try {
-      await this.#interactions.cancelPendingInteractions();
-      if (this.#turn.running) {
-        await this.#connection.connection.agent.notify(methods3.agent.session.cancel, {
-          sessionId: this.#updates.sessionId()
-        });
-      }
-    } finally {
-      await this.closeProcess(capturedProcesses);
+    await this.#interactions.cancelPendingInteractions();
+    if (this.#turn.running) {
+      await this.#connection.connection.agent.notify(methods3.agent.session.cancel, {
+        sessionId: this.#updates.sessionId()
+      });
     }
   };
   respondPermission = (options) => this.#interactions.respondPermission(options);
@@ -31182,7 +31178,7 @@ var AgentSession = class _AgentSession {
       await this.closeProcess();
     }
   };
-  closeProcess = (capturedProcesses = []) => this.#connection.closeProcess(capturedProcesses);
+  closeProcess = () => this.#connection.closeProcess();
   race = (operation, exitMessage, signal) => this.#connection.race(operation, exitMessage, signal);
   assertNativeSession = (sessionId) => {
     if (sessionId !== this.#updates.sessionId()) {
@@ -32323,13 +32319,7 @@ var AgentRuntimeManager = class {
     await this.#executions.cancel(sessionId, async () => {
       const runtime = this.#executionProcesses.get(sessionId);
       if (runtime !== void 0) {
-        try {
-          await runtime.cancel();
-        } finally {
-          if (this.#executionProcesses.detach(sessionId, runtime)) {
-            this.#processes.releaseClosed(runtime);
-          }
-        }
+        await runtime.cancel();
       }
     });
   };
@@ -35117,7 +35107,7 @@ var CodexAcpAgent = class {
   };
   #completeTurn = async (turn) => {
     const active = this.#turn;
-    if (active === null) {
+    if (active === null || active.turn.id !== turn.id) {
       return;
     }
     await this.#updateTail;
@@ -43007,24 +42997,33 @@ async function runRelayConnection({
 // src/relay-connection/session-event-publisher.ts
 import { WebSocket as WebSocket8 } from "undici";
 var MAX_BUFFERED_BYTES = 32 * 1024 * 1024;
+var SESSION_EVENT_V2_BATCH_WINDOW_MILLISECONDS = 200;
 var DaemonSessionEventPublisher = class {
+  #pendingV2Events = [];
+  #pendingV2Task = null;
   #socket = null;
+  #timer;
+  constructor(timer) {
+    this.#timer = timer;
+  }
   connect = (socket) => {
+    this.dropPendingV2Events();
     this.#socket = socket;
   };
   disconnect = (socket) => {
     if (this.#socket === socket) {
       this.#socket = null;
+      this.dropPendingV2Events();
     }
   };
   close = () => {
     this.#socket = null;
+    this.dropPendingV2Events();
   };
   publishOperationFailure = (sessionId, requestId, code, message) => {
     this.#send({ kind: "session.operation.failed", sessionId, requestId, code, message });
   };
   publishStatus = (event) => this.#send(event);
-  publishV2 = (event) => this.#send({ event, kind: SESSION_EVENT_V2_KIND });
   #send = (message) => {
     const socket = this.#socket;
     if (socket === null || socket.readyState !== WebSocket8.OPEN) {
@@ -43040,6 +43039,44 @@ var DaemonSessionEventPublisher = class {
     } catch {
     }
   };
+  publishV2 = (event) => {
+    const socket = this.#socket;
+    if (socket === null || socket.readyState !== WebSocket8.OPEN) {
+      return;
+    }
+    this.#pendingV2Events.push(event);
+    if (this.#pendingV2Events.length >= SESSION_EVENT_V2_BATCH_MAX_SIZE) {
+      this.flushV2Events(socket);
+      return;
+    }
+    this.#pendingV2Task ??= this.#timer.schedule(
+      () => this.flushV2Events(socket),
+      SESSION_EVENT_V2_BATCH_WINDOW_MILLISECONDS
+    );
+  };
+  dropPendingV2Events = () => {
+    this.#pendingV2Task?.cancel();
+    this.#pendingV2Task = null;
+    this.#pendingV2Events = [];
+  };
+  flushV2Events = (socket) => {
+    this.#pendingV2Task?.cancel();
+    this.#pendingV2Task = null;
+    const events = this.#pendingV2Events;
+    this.#pendingV2Events = [];
+    if (events.length === 0 || this.#socket !== socket || socket.readyState !== WebSocket8.OPEN) {
+      return;
+    }
+    const encoded = encodeDaemonToServiceMessage({ events, kind: SESSION_EVENT_V2_KIND });
+    if (socket.bufferedAmount + Buffer.byteLength(encoded, "utf8") > MAX_BUFFERED_BYTES) {
+      socket.close(1013, "Live event transport overloaded");
+      return;
+    }
+    try {
+      socket.send(encoded);
+    } catch {
+    }
+  };
 };
 
 // src/relay-connection/timer.ts
@@ -43283,7 +43320,7 @@ async function runDaemon(config2, audit, signal, timer = systemDaemonTimer, opti
   const sessionStore = new SqliteAcpSessionStore(sessionDatabase, eventsV2);
   await using externalContent = new LocalExternalContentStore(stateDirectory);
   const operations = new DaemonOperationState();
-  const publisher = new DaemonSessionEventPublisher();
+  const publisher = new DaemonSessionEventPublisher(timer);
   const eventLogV2 = new DaemonSessionEventLogV2(
     publisher,
     (sessionId, event) => agentRuntime.recordLifecycleEventV2(sessionId, event),