@mastra/platform-workspace 1.3.0-alpha.0 → 1.3.0-alpha.2

package/dist/index.js

@@ -835,6 +835,7 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 	_client;
 	_environmentId;
 	_sandboxId;
+	_seedCheckpointName;
 	_idleTimeoutMinutes;
 	_networkIsolation;
 	_env;
@@ -914,6 +915,12 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 	* failed/timed out, but now coalesced via `_leaseInFlight`).
 	*/
 	_transportReadyPromise = null;
+	/**
+	* The sidecar address of the most recent `start()`, kept so a timed-out
+	* probe can be restarted by a later exec ({@link _awaitTransportReady})
+	* instead of pinning the sandbox to the lease path for its lifetime.
+	*/
+	_probeTarget = null;
 	constructor(options = {}) {
 		super({
 			...options,
@@ -926,6 +933,7 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 		this._environmentId = options.environmentId ?? process.env.MASTRA_ENVIRONMENT_ID ?? "";
 		if (!this._environmentId && !options.sandboxId) throw new Error("environmentId is required");
 		this._sandboxId = options.sandboxId;
+		this._seedCheckpointName = options.seedCheckpointName;
 		this._idleTimeoutMinutes = options.idleTimeoutMinutes;
 		this._networkIsolation = options.networkIsolation;
 		this._env = options.env ?? {};
@@ -962,6 +970,7 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 			fetch: this._client.fetch,
 			environmentId: this._environmentId,
 			...options.sandboxId !== void 0 && { sandboxId: options.sandboxId },
+			...(options.seedCheckpointName ?? this._seedCheckpointName) !== void 0 && { seedCheckpointName: options.seedCheckpointName ?? this._seedCheckpointName },
 			idleTimeoutMinutes: options.idleTimeoutMinutes ?? this._idleTimeoutMinutes,
 			...this._networkIsolation !== void 0 && { networkIsolation: this._networkIsolation },
 			env: options.env ?? this._env,
@@ -1008,6 +1017,7 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 		if (!this._environmentId) throw new Error("environmentId is required");
 		const body = JSON.stringify({
 			id: this.id,
+			seedCheckpointName: this._seedCheckpointName,
 			environmentId: this._environmentId,
 			idleTimeoutMinutes: this._idleTimeoutMinutes,
 			networkIsolation: this._networkIsolation,
@@ -1066,14 +1076,24 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 	* HTTP round-trip.
 	*
 	* `null`/absent `instanceUrl` (proxy discovery failed, or an older proxy
-	* that predates the field) leaves the registry untouched — executes fall
-	* through to the lease path with no branch here.
+	* that predates the field) evicts any stale registry address and leaves
+	* executes on the lease path.
 	*/
 	_populateAddressFromResponse(json) {
 		if (!this._addressRegistry) return;
-		if (!json.instanceUrl) return;
+		if (!json.instanceUrl) {
+			this._probeGeneration++;
+			this._probeTarget = null;
+			this._transportReadyPromise = null;
+			this._addressRegistry.delete(json.id);
+			return;
+		}
 		this._addressRegistry.delete(json.id);
 		const generation = ++this._probeGeneration;
+		this._probeTarget = {
+			sandboxId: json.id,
+			instanceUrl: json.instanceUrl
+		};
 		this._transportReadyPromise = this._probeSidecarThenRegister(json.id, json.instanceUrl, generation);
 	}
 	/**
@@ -1083,8 +1103,9 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 	* fall back to the lease path until the probe succeeds.
 	*
 	* If the sidecar never comes up within {@link SIDECAR_PROBE_TIMEOUT_MS},
-	* the registry stays unpopulated and all execs go via lease for this
-	* sandbox's lifetime (or until a future `start()` re-runs the probe).
+	* the registry stays unpopulated, `_transportReadyPromise` is cleared,
+	* and a later {@link _awaitTransportReady} call restarts the probe
+	* instead of pinning this sandbox to the lease path.
 	*
 	* @param generation - The probe generation captured at call time. If this
 	*   no longer matches `_probeGeneration` when the probe succeeds, the probe
@@ -1118,6 +1139,7 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 			} catch {}
 			await new Promise((r) => setTimeout(r, SIDECAR_PROBE_INTERVAL_MS));
 		}
+		if (generation === this._probeGeneration && this._transportReadyPromise) this._transportReadyPromise = null;
 		this.logger.warn("platform-workspace probe timed out", {
 			sandboxId,
 			sessionId: this._client.sessionId,
@@ -1138,7 +1160,12 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 	*/
 	async _awaitTransportReady() {
 		if (this._sandboxId && this._addressRegistry?.get(this._sandboxId)) return;
-		if (!this._transportReadyPromise) return;
+		if (!this._transportReadyPromise) {
+			const target = this._probeTarget;
+			if (!target || this._sandboxId !== target.sandboxId) return;
+			const generation = ++this._probeGeneration;
+			this._transportReadyPromise = this._probeSidecarThenRegister(target.sandboxId, target.instanceUrl, generation);
+		}
 		await Promise.race([this._transportReadyPromise, new Promise((r) => setTimeout(r, TRANSPORT_READY_WAIT_MS))]);
 	}
 	/**
@@ -1205,6 +1232,8 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 		if (!this._sandboxId) return;
 		const destroyedSandboxId = this._sandboxId;
 		this._probeGeneration++;
+		this._probeTarget = null;
+		this._transportReadyPromise = null;
 		await this._client.request(`/sandbox/${encodeURIComponent(destroyedSandboxId)}`, { method: "DELETE" });
 		this._sandboxId = void 0;
 		this._createdAt = null;
@@ -1215,6 +1244,8 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 	async snapshot() {
 		await this.captureCheckpoint();
 	}
+	/** Snapshots persist real checkpoints that can seed future sandboxes. */
+	supportsCheckpoints = true;
 	/**
 	* Capture the sandbox's checkpoint on demand, outside any refresh timer the
 	* workspace-proxy owns internally.
@@ -1334,6 +1365,9 @@ var PlatformSandbox = class PlatformSandbox extends MastraSandbox {
 	* the cached `'running'` state (see `MastraSandbox._start`).
 	*/
 	_clearDestroyedState(destroyedSandboxId) {
+		this._probeGeneration++;
+		this._probeTarget = null;
+		this._transportReadyPromise = null;
 		this._sandboxId = void 0;
 		this._createdAt = null;
 		this._lease = null;