@ycodium-ai/agent-antigravity 0.2.3236 → 0.2.3240

package/dist/index.js

@@ -26666,6 +26666,83 @@ function createAcpConnection(options2) {
   };
 }
 
+// ../../packages/ycodium-acp-executor/src/model-selection.ts
+function normalizeModelId(value) {
+  return value?.trim() || void 0;
+}
+function modelSelectionStateFromSetup(setup) {
+  const sessionModels = isRecord(setup.models) ? setup.models : void 0;
+  const modelConfigId = extractModelConfigId(setup);
+  const strategy = sessionModels === void 0 && modelConfigId !== void 0 ? "config-option" : "session/set_model";
+  const modelConfigOption = modelConfigId === void 0 ? void 0 : findSessionConfigOption(setup.configOptions, modelConfigId);
+  const configModel = modelConfigOption?.currentValue;
+  const currentModelId = strategy === "config-option" ? typeof configModel === "string" ? normalizeModelId(configModel) : void 0 : typeof sessionModels?.["currentModelId"] === "string" ? normalizeModelId(sessionModels["currentModelId"]) : void 0;
+  return {
+    strategy,
+    sessionModelsAdvertised: sessionModels !== void 0,
+    modelConfigId,
+    currentModelId,
+    advertisedModelValues: strategy === "config-option" ? modelConfigOption === void 0 ? [] : collectSessionConfigOptionValues(modelConfigOption) : collectSessionModelIds(sessionModels),
+    configOptions: setup.configOptions
+  };
+}
+
+// ../../packages/ycodium-acp-executor/src/mutex.ts
+var Mutex = class {
+  locked = false;
+  waiters = [];
+  get isLocked() {
+    return this.locked;
+  }
+  acquire() {
+    if (!this.locked) {
+      this.locked = true;
+      return Promise.resolve();
+    }
+    return new Promise((resolve6) => {
+      this.waiters.push(resolve6);
+    });
+  }
+  tryAcquire() {
+    if (this.locked) {
+      return false;
+    }
+    this.locked = true;
+    return true;
+  }
+  release() {
+    const next = this.waiters.shift();
+    if (next) {
+      next();
+      return;
+    }
+    this.locked = false;
+  }
+  async runExclusive(fn) {
+    await this.acquire();
+    try {
+      return await fn();
+    } finally {
+      this.release();
+    }
+  }
+};
+
+// ../../packages/ycodium-acp-executor/src/user-input-pending.ts
+var userInputAnswerOptionId = (question, answers) => {
+  const answer = answers[question.id];
+  const selected = typeof answer === "string" ? answer : Array.isArray(answer) && answer.length === 1 && typeof answer[0] === "string" ? answer[0] : void 0;
+  return selected !== void 0 && question.options.some((option) => option.id === selected) ? selected : void 0;
+};
+
+// ../../packages/ycodium-acp-executor/src/vendor-hooks.ts
+var defaultVendorHooks = {
+  completedStopReason: (response) => response.stopReason
+};
+
+// ../../packages/ycodium-acp-executor/src/executor/support.ts
+var DEFAULT_RESULT_DRAIN_TIMEOUT_MS = 2e3;
+
 // ../../packages/ycodium-acp-executor/src/executor/lifecycle.ts
 function createExecutorLifecycle({
   failIfDisposed,
@@ -26808,16 +26885,30 @@ function createExecutorLifecycle({
     }
   };
   let disposing;
-  const dispose = () => {
+  const dispose = (exitCode) => {
     if (disposing) return disposing;
     shared.disposing = true;
-    promptResults.stop();
-    settleAllApprovals("cancel");
-    settleAllUserInputs();
-    abortPendingSettlement("executor disposed");
-    notifyPromptSettled();
-    shared.abortLocalPrompt?.();
     disposing = (async () => {
+      if (exitCode !== void 0) {
+        let timer;
+        await Promise.race([
+          connection.closed.catch(() => void 0),
+          new Promise((resolve6) => {
+            timer = setTimeout(
+              resolve6,
+              options2.resultDrainTimeoutMs ?? DEFAULT_RESULT_DRAIN_TIMEOUT_MS
+            );
+          })
+        ]);
+        clearTimeout(timer);
+        await promptResults.wait(void 0, true);
+      }
+      promptResults.stop();
+      settleAllApprovals("cancel");
+      settleAllUserInputs();
+      abortPendingSettlement("executor disposed");
+      notifyPromptSettled();
+      shared.abortLocalPrompt?.();
       promptResults.finishUnreceived();
       await promptResults.wait();
       emit({
@@ -26825,7 +26916,10 @@ function createExecutorLifecycle({
         ...stamp(),
         provider: options2.provider,
         threadId: options2.threadId,
-        payload: { reason: "disposed" }
+        payload: exitCode === void 0 ? { reason: "disposed" } : {
+          reason: exitCode === null ? "process exited without an exit code" : `process exited with code ${exitCode}`,
+          exitKind: exitCode === 0 ? "graceful" : "error"
+        }
       });
       shared.disposed = true;
       shared.generation += 1;
@@ -26920,11 +27014,6 @@ var acpPromptFailureCategory = (error62, input2) => {
   return error62 instanceof AcpConnectionClosedError ? acpTransportIfEstablished(input2.sessionEstablished, input2.disposed) : void 0;
 };
 
-// ../../packages/ycodium-acp-executor/src/vendor-hooks.ts
-var defaultVendorHooks = {
-  completedStopReason: (response) => response.stopReason
-};
-
 // ../../packages/ycodium-acp-executor/src/executor/prompt.ts
 function createExecutorPrompt({
   isLive,
@@ -27030,8 +27119,12 @@ function createExecutorPrompt({
       }
     }
   };
-  const prompt = async (input2) => {
+  const failIfPromptUnavailable = () => {
     failIfDisposed();
+    if (shared.connectionClosed) throw new AcpExecutorDisposedError();
+  };
+  const prompt = async (input2) => {
+    failIfPromptUnavailable();
     const started = requireStarted();
     let record2;
     const settle2 = (rest) => settleTurn({
@@ -27043,7 +27136,7 @@ function createExecutorPrompt({
     let turnId;
     await threadMutex.acquire();
     try {
-      failIfDisposed();
+      failIfPromptUnavailable();
       turnId = selectPromptTurnId(
         input2,
         shared.promptsInFlight > 0 ? shared.activeTurnId : void 0,
@@ -27152,7 +27245,7 @@ function createExecutorPrompt({
           sessionEstablished: shared.startResult !== void 0,
           disposed: shared.disposed
         });
-        if (written && input2.onDispatched !== void 0 && !interruptedTurnIds.has(turnId)) {
+        if (!interruptedTurnIds.has(turnId)) {
           const errorMessage2 = error62 instanceof Error && error62.message.trim() || "ACP prompt failed";
           if (settle2({
             turnId,
@@ -27160,7 +27253,7 @@ function createExecutorPrompt({
             state: "failed",
             errorMessage: errorMessage2,
             ...failureCategory === void 0 ? {} : { failureCategory }
-          }))
+          }) && written && input2.onDispatched !== void 0)
             options2.log?.warn?.("turn failed after dispatch", {
               threadId: options2.threadId,
               failure: errorMessage2
@@ -27261,9 +27354,9 @@ function createPromptResults({
       }
     }
   };
-  const wait = async (turnId) => {
+  const wait = async (turnId, includeUnreceived = false) => {
     const claimed = [...prompts.values()].filter(
-      (record2) => !record2.finished && record2.response !== void 0 && (turnId === void 0 || record2.turnId === turnId)
+      (record2) => !record2.finished && (includeUnreceived || record2.response !== void 0) && (turnId === void 0 || record2.turnId === turnId)
     );
     if (claimed.length === 0) return;
     let timer;
@@ -27271,7 +27364,10 @@ function createPromptResults({
       const drained = await Promise.race([
         Promise.all(claimed.map((record2) => record2.done)).then(() => true),
         new Promise((resolve6) => {
-          timer = setTimeout(() => resolve6(false), options2.resultDrainTimeoutMs ?? 2e3);
+          timer = setTimeout(
+            () => resolve6(false),
+            options2.resultDrainTimeoutMs ?? DEFAULT_RESULT_DRAIN_TIMEOUT_MS
+          );
         })
       ]);
       if (drained) return;
@@ -27295,13 +27391,6 @@ function createPromptResults({
   return { add, receive, reject, stop, complete, hasInterrupted, forget, wait, finishUnreceived };
 }
 
-// ../../packages/ycodium-acp-executor/src/user-input-pending.ts
-var userInputAnswerOptionId = (question, answers) => {
-  const answer = answers[question.id];
-  const selected = typeof answer === "string" ? answer : Array.isArray(answer) && answer.length === 1 && typeof answer[0] === "string" ? answer[0] : void 0;
-  return selected !== void 0 && question.options.some((option) => option.id === selected) ? selected : void 0;
-};
-
 // ../../packages/ycodium-acp-executor/src/executor/requests.ts
 function createExecutorRequests({
   emit,
@@ -27865,27 +27954,6 @@ function acpInitializePayload(clientInfo, overrides) {
   };
 }
 
-// ../../packages/ycodium-acp-executor/src/model-selection.ts
-function normalizeModelId(value) {
-  return value?.trim() || void 0;
-}
-function modelSelectionStateFromSetup(setup) {
-  const sessionModels = isRecord(setup.models) ? setup.models : void 0;
-  const modelConfigId = extractModelConfigId(setup);
-  const strategy = sessionModels === void 0 && modelConfigId !== void 0 ? "config-option" : "session/set_model";
-  const modelConfigOption = modelConfigId === void 0 ? void 0 : findSessionConfigOption(setup.configOptions, modelConfigId);
-  const configModel = modelConfigOption?.currentValue;
-  const currentModelId = strategy === "config-option" ? typeof configModel === "string" ? normalizeModelId(configModel) : void 0 : typeof sessionModels?.["currentModelId"] === "string" ? normalizeModelId(sessionModels["currentModelId"]) : void 0;
-  return {
-    strategy,
-    sessionModelsAdvertised: sessionModels !== void 0,
-    modelConfigId,
-    currentModelId,
-    advertisedModelValues: strategy === "config-option" ? modelConfigOption === void 0 ? [] : collectSessionConfigOptionValues(modelConfigOption) : collectSessionModelIds(sessionModels),
-    configOptions: setup.configOptions
-  };
-}
-
 // ../../packages/ycodium-acp-executor/src/runtime-mode.ts
 function isRecord5(value) {
   return typeof value === "object" && value !== null && !Array.isArray(value);
@@ -28482,47 +28550,6 @@ function createExecutorUpdates({
   return { emitCompactionState, handleSessionUpdate };
 }
 
-// ../../packages/ycodium-acp-executor/src/mutex.ts
-var Mutex = class {
-  locked = false;
-  waiters = [];
-  get isLocked() {
-    return this.locked;
-  }
-  acquire() {
-    if (!this.locked) {
-      this.locked = true;
-      return Promise.resolve();
-    }
-    return new Promise((resolve6) => {
-      this.waiters.push(resolve6);
-    });
-  }
-  tryAcquire() {
-    if (this.locked) {
-      return false;
-    }
-    this.locked = true;
-    return true;
-  }
-  release() {
-    const next = this.waiters.shift();
-    if (next) {
-      next();
-      return;
-    }
-    this.locked = false;
-  }
-  async runExclusive(fn) {
-    await this.acquire();
-    try {
-      return await fn();
-    } finally {
-      this.release();
-    }
-  }
-};
-
 // ../../packages/ycodium-acp-executor/src/openCommands.ts
 function readOpenToolCalls(toolCalls, turnId) {
   const running = [];
@@ -28894,10 +28921,10 @@ function createAcpExecutor(options2) {
     setNativeMode,
     compact,
     openToolCalls: () => readOpenToolCalls(toolCalls, shared.activeTurnId),
-    dispose: () => {
+    dispose: (exitCode) => {
       disposeOnce ??= (async () => {
         try {
-          await disposeLifecycle();
+          await disposeLifecycle(exitCode);
         } finally {
           await protocolLogSink?.close();
         }
@@ -28908,7 +28935,8 @@ function createAcpExecutor(options2) {
 }
 
 // ../../packages/ycodium-acp-executor/src/session-registry.ts
-function createAcpSessionRegistry(threads, stopBooked) {
+function createAcpSessionRegistry(threads, stopBooked, log) {
+  const exits = /* @__PURE__ */ new WeakMap();
   const requireThread = (threadId) => {
     const booked = threads.get(threadId);
     if (!booked) throw new Error(`unknown thread ${threadId}`);
@@ -28917,7 +28945,8 @@ function createAcpSessionRegistry(threads, stopBooked) {
   };
   const stopSession = (threadId, booked = threads.get(threadId)) => {
     if (!booked) return Promise.resolve();
-    booked.closing ??= Promise.resolve().then(() => stopBooked(booked)).then(() => {
+    const close = { exitCode: exits.get(booked), hasExited: () => exits.has(booked) };
+    booked.closing ??= Promise.resolve().then(() => stopBooked(booked, close)).then(() => {
       if (threads.get(threadId) === booked) threads.delete(threadId);
     }).catch((cause) => {
       if (threads.get(threadId) === booked) delete booked.closing;
@@ -28928,7 +28957,16 @@ function createAcpSessionRegistry(threads, stopBooked) {
   const stopAll = () => Promise.all([...threads].map(([threadId, booked]) => stopSession(threadId, booked))).then(
     () => void 0
   );
-  return { requireThread, stopSession, stopAll };
+  const registerSession = (threadId, booked) => {
+    threads.set(threadId, booked);
+    void booked.handle.exited.then(async (exitCode) => {
+      exits.set(booked, exitCode);
+      await stopSession(threadId, booked);
+    }).catch((cause) => {
+      log?.warn("ACP process exit cleanup failed", { threadId, error: String(cause) });
+    });
+  };
+  return { requireThread, registerSession, stopSession, stopAll };
 }
 
 // ../../packages/ycodium-acp-executor/src/session-cleanup.ts
@@ -28941,8 +28979,11 @@ async function cleanupAcpSession(input2) {
     failures.push(new Error(`forwarding failed: ${String(cause)}`, { cause }));
   });
   for (const [step, cleanup] of [
-    ["executor.dispose", () => input2.executor.dispose()],
-    ["stopProcess", input2.stopProcess],
+    ["executor.dispose", () => input2.executor.dispose(input2.processExit?.exitCode)],
+    [
+      "stopProcess",
+      () => input2.processExit?.hasExited() ? Promise.resolve() : input2.stopProcess()
+    ],
     ["forwarding", () => forwarding]
   ]) {
     try {
@@ -30440,6 +30481,7 @@ function createAntigravitySession({
   dirs,
   events,
   threads,
+  registerSession,
   mutable,
   resolveInput,
   stopSession,
@@ -30643,7 +30685,7 @@ function createAntigravitySession({
           ...started.modelId === void 0 ? {} : { model: started.modelId },
           ...input2.bindingGeneration === void 0 ? {} : { bindingGeneration: input2.bindingGeneration }
         };
-        threads.set(input2.threadId, {
+        registerSession(input2.threadId, {
           session,
           turns: [],
           executor,
@@ -32551,12 +32593,13 @@ async function createAntigravityInstance(config2, host, deps) {
     processEnv
   };
   const resolveExecutable = () => deps.runtime.resolve(resolveInput);
-  const stopBooked = async (booked) => {
+  const stopBooked = async (booked, processExit) => {
     for (const stopped of booked.background.finish()) events.push(stopped);
     await cleanupAcpSession({
       executor: booked.executor,
       stopProcess: () => stopAgentProcess(booked.handle, exitWaitMs, host.log, clock),
-      forwarding: booked.forwarding
+      forwarding: booked.forwarding,
+      processExit
     });
     if (!await removeRuntimeTempDir(booked.runtimeTemp)) {
       host.log.warn("could not remove a runtime temp directory; it is swept on the next start", {
@@ -32564,7 +32607,11 @@ async function createAntigravityInstance(config2, host, deps) {
       });
     }
   };
-  const { requireThread, stopSession, stopAll } = createAcpSessionRegistry(threads, stopBooked);
+  const { requireThread, registerSession, stopSession, stopAll } = createAcpSessionRegistry(
+    threads,
+    stopBooked,
+    host.log
+  );
   const helpers = createAntigravityTextGeneration({
     host,
     profile: dirs.profile,
@@ -32694,6 +32741,7 @@ async function createAntigravityInstance(config2, host, deps) {
     dirs,
     events,
     threads,
+    registerSession,
     mutable,
     resolveInput,
     stopSession,
@@ -33879,7 +33927,7 @@ var executors = [
 ];
 var index_default = definePlugin({
   id: "agent-antigravity",
-  apiVersion: 25,
+  apiVersion: 26,
   executors,
   activate(ctx) {
     const runtime = createRuntimeManager({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ycodium-ai/agent-antigravity",
-  "version": "0.2.3236",
+  "version": "0.2.3240",
   "description": "Google Antigravity executor: runs the Antigravity agent over Agent Client Protocol stdio.",
   "keywords": [
     "ycodium",
@@ -31,8 +31,8 @@
     "typescript": "~6.0.3",
     "vite-plus": "0.2.2",
     "yauzl": "^3.4.0",
-    "@ycodium-ai/plugin-api": "0.2.3236",
-    "ycodium-acp-executor": "0.2.3236",
+    "@ycodium-ai/plugin-api": "0.2.3240",
+    "ycodium-acp-executor": "0.2.3240",
     "ycodium-executor-testing": "0.0.0"
   },
   "ycodium": {