@opengeni/worker-bundle 0.28.4-canary.0 → 0.28.4-canary.2

package/dist/activities/agent-turn/codex-selection-diagnostics.d.ts

@@ -0,0 +1 @@
+export { codexSelectionDiagnostics } from "@opengeni/db";

package/dist/activities/codex-fleet-shadow.d.ts

@@ -4,7 +4,7 @@ export declare const CODEX_FLEET_SHADOW_MAX_PAYLOAD_BYTES: number;
 export type CodexFleetActualDecisionV1 = {
     outcome: "selected" | "waiting" | "none";
     candidateKey: string | null;
-    reason: "lease_reused" | "pin" | "rotation" | "active" | "all_capped" | "allocator_disabled" | "none";
+    reason: "lease_reused" | "affinity_reused" | "assigned" | "switched" | "pin" | "rotation" | "active" | "all_capped" | "allocator_disabled" | "none";
 };
 export type CodexFleetShadowComparisonV1 = "match" | "different_candidate" | "different_outcome" | "not_comparable_truncated";
 export type CodexFleetShadowPayloadV1 = {

package/dist/activities/sandbox-lease.d.ts

@@ -60,7 +60,7 @@ export type DrainCaptureDisposition = "capture_required" | "archive_published";
  *  Test seams may return the legacy boolean shorthand. Production returns the
  *  typed outcome so the cold commit can distinguish a clean stop from definitive
  *  provider disappearance before capture. */
-export type TerminateBoxFn = (settings: ActivityServices["settings"], lease: NonNullable<Awaited<ReturnType<typeof readLease>>>, observability: ActivityServices["observability"], persistArchive: PersistArchiveFn, providerCaptureRequestId?: string, captureDisposition?: DrainCaptureDisposition, capturePolicy?: ProviderWorkspaceCapturePolicy | null, diskBackedArchives?: boolean) => Promise<boolean | ProviderTerminationOutcome>;
+export type TerminateBoxFn = (settings: ActivityServices["settings"], lease: NonNullable<Awaited<ReturnType<typeof readLease>>>, observability: ActivityServices["observability"], persistArchive: PersistArchiveFn, providerCaptureRequestId?: string, captureDisposition?: DrainCaptureDisposition, capturePolicy?: ProviderWorkspaceCapturePolicy | null, diskBackedArchives?: boolean, releaseFailedCapture?: () => Promise<void>) => Promise<boolean | ProviderTerminationOutcome>;
 export type SweepModalOrphansFn = (settings: ActivityServices["settings"], db: ActivityServices["db"], observability: ActivityServices["observability"]) => Promise<number>;
 export type InspectOpenSandboxKubernetesInventoryFn = (input: {
     namespace: string;
@@ -114,6 +114,14 @@ export declare class SandboxProviderCaptureTimeoutError extends Error {
     readonly name = "SandboxProviderCaptureTimeoutError";
     constructor(sandboxGroupId: string, backend: string, timeoutMs: number, leaseEpoch: number, instanceId: string);
 }
+export declare function awaitProviderCaptureWithLatePublication<T>(input: {
+    capture: Promise<T>;
+    timeoutMs: number;
+    timeoutError: SandboxProviderCaptureTimeoutError;
+    publishLate: (value: T) => Promise<void>;
+    observeLateFailure: (error: unknown) => void;
+    observeLatePublicationFailure: (error: unknown) => void;
+}): Promise<T>;
 export type SandboxLeaseSweepPlan = {
     drainable: ReapDrainable[];
     timeoutClass: SandboxDrainActivityInput["timeoutClass"];
@@ -199,4 +207,4 @@ export declare function sandboxDrainReason(lease: Pick<LeaseSnapshot, "rotationR
  * draining until its adapter can prove termination. A truly instance-less or
  * `none`-backed row is a no-op.
  */
-export declare function terminateProviderBox(settings: ActivityServices["settings"], lease: NonNullable<Awaited<ReturnType<typeof readLease>>>, observability: ActivityServices["observability"], persistArchive: PersistArchiveFn, createClientForBackend?: CreateSandboxClientForBackendFn, terminateModalById?: typeof terminateModalSandboxById, providerCaptureRequestId?: string, captureDisposition?: DrainCaptureDisposition, claimedCapturePolicy?: ProviderWorkspaceCapturePolicy | null, diskBackedArchives?: boolean): Promise<ProviderTerminationOutcome>;
+export declare function terminateProviderBox(settings: ActivityServices["settings"], lease: NonNullable<Awaited<ReturnType<typeof readLease>>>, observability: ActivityServices["observability"], persistArchive: PersistArchiveFn, createClientForBackend?: CreateSandboxClientForBackendFn, terminateModalById?: typeof terminateModalSandboxById, providerCaptureRequestId?: string, captureDisposition?: DrainCaptureDisposition, claimedCapturePolicy?: ProviderWorkspaceCapturePolicy | null, diskBackedArchives?: boolean, releaseFailedCapture?: () => Promise<void>): Promise<ProviderTerminationOutcome>;

package/dist/{activities-control-J5QNRWVH.js → activities-control-MKGEOBXT.js}

@@ -31,7 +31,7 @@ import {
   scheduledUserMessagePayload,
   withFirstPartyTools,
   workflowIdForSession
-} from "./chunk-ZNIKZQMN.js";
+} from "./chunk-OA5CSY5N.js";
 import {
   inspectOpenSandboxKubernetesInventory,
   recordCreditBalanceGauges,
@@ -5199,6 +5199,7 @@ import {
   appendSessionEventToSandboxGroup,
   bindRetainedProcessProviderIdentity,
   claimWorkspaceArchiveCapture,
+  releaseWorkspaceArchiveCapture,
   claimSandboxCheckpointArtifactsForGc,
   claimTerminalRetainedProcesses,
   countActiveRetainedProcessesByOwnerState,
@@ -5312,7 +5313,7 @@ async function awaitProviderCaptureWithLatePublication(input) {
     ]);
   } catch (error) {
     if (error !== timeoutError) throw error;
-    void input.capture.then(input.publishLate).catch(input.observeLateFailure);
+    void input.capture.then(input.publishLate, input.observeLateFailure).catch(input.observeLatePublicationFailure);
     throw error;
   } finally {
     if (timeout) clearTimeout(timeout);
@@ -5362,7 +5363,7 @@ function sandboxDrainActivityAttempt() {
   }
 }
 function createSandboxLeaseActivities(services, options = {}) {
-  const terminateBox = options.terminateBox ?? (async (settings, lease, observability, persistArchive, providerCaptureRequestId, captureDisposition, capturePolicy, diskBackedArchives) => await terminateProviderBox(
+  const terminateBox = options.terminateBox ?? (async (settings, lease, observability, persistArchive, providerCaptureRequestId, captureDisposition, capturePolicy, diskBackedArchives, releaseFailedCapture) => await terminateProviderBox(
     settings,
     lease,
     observability,
@@ -5372,7 +5373,8 @@ function createSandboxLeaseActivities(services, options = {}) {
     providerCaptureRequestId,
     captureDisposition,
     capturePolicy,
-    diskBackedArchives
+    diskBackedArchives,
+    releaseFailedCapture
   ));
   const sweepModalOrphans = options.sweepModalOrphans ?? sweepModalOrphansForConfiguredBackend;
   const inspectOpenSandboxInventory = options.inspectOpenSandboxKubernetesInventory ?? inspectOpenSandboxKubernetesInventory;
@@ -7131,7 +7133,18 @@ async function terminateDrainableBox(db, settings, row, observability, terminate
     captureClaim?.providerRequestId ?? attempt.operationId,
     captureDisposition,
     capturePolicy,
-    Boolean(objectStorage)
+    Boolean(objectStorage),
+    async () => {
+      if (!captureClaim || !lease.instanceId) return;
+      await releaseWorkspaceArchiveCapture(db, {
+        accountId,
+        workspaceId: row.workspaceId,
+        sandboxGroupId: row.sandboxGroupId,
+        captureId: captureClaim.id,
+        expectedEpoch: row.leaseEpoch,
+        expectedInstanceId: lease.instanceId
+      });
+    }
   );
   const terminated = typeof termination === "boolean" ? termination : termination.terminated;
   if (!terminated) {
@@ -7187,10 +7200,20 @@ function providerDrainLogIdentity(lease, backend) {
     instanceId: lease.instanceId
   };
 }
-async function terminateProviderBox(settings, lease, observability, persistArchive, createClientForBackend = createSandboxClientForBackend, terminateModalById = terminateModalSandboxById, providerCaptureRequestId, captureDisposition = "capture_required", claimedCapturePolicy, diskBackedArchives = false) {
+async function terminateProviderBox(settings, lease, observability, persistArchive, createClientForBackend = createSandboxClientForBackend, terminateModalById = terminateModalSandboxById, providerCaptureRequestId, captureDisposition = "capture_required", claimedCapturePolicy, diskBackedArchives = false, releaseFailedCapture) {
   const durableBackendId = lease.resumeBackendId ?? lease.backend;
   const backend = sandboxBackendForSdkBackendId(durableBackendId) ?? durableBackendId;
   const logIdentity = providerDrainLogIdentity(lease, backend);
+  const releaseAfterCaptureFailure = async () => {
+    try {
+      await releaseFailedCapture?.();
+    } catch (error) {
+      observability.warn("sandbox reaper: failed capture claim release failed", {
+        ...logIdentity,
+        error: error instanceof Error ? error.message : String(error)
+      });
+    }
+  };
   if (!backend || backend === "none") {
     return { terminated: true, providerMissingBeforeCapture: false };
   }
@@ -7351,7 +7374,14 @@ async function terminateProviderBox(settings, lease, observability, persistArchi
             await disposeWorkspaceArchive(archive);
           }
         },
+        observeLatePublicationFailure: (error) => {
+          observability.warn("sandbox reaper: late capture publication cleanup failed", {
+            ...logIdentity,
+            error: error instanceof Error ? error.message : String(error)
+          });
+        },
         observeLateFailure: (error) => {
+          void releaseAfterCaptureFailure();
           observability.warn("sandbox reaper: timed-out provider capture later failed", {
             ...logIdentity,
             error: error instanceof Error ? error.message : String(error)
@@ -7373,6 +7403,9 @@ async function terminateProviderBox(settings, lease, observability, persistArchi
         error: error instanceof Error ? error.message : String(error)
       }
     );
+    if (!(error instanceof SandboxProviderCaptureTimeoutError)) {
+      await releaseAfterCaptureFailure();
+    }
     throw error;
   }
   if (captureDisposition === "archive_published") {
@@ -10336,4 +10369,4 @@ export {
   createControlActivities,
   createControlActivitiesFromServices
 };
-//# sourceMappingURL=activities-control-J5QNRWVH.js.map
+//# sourceMappingURL=activities-control-MKGEOBXT.js.map