@openai/agents-core 0.8.5 → 0.9.0

package/dist/sandbox/runtime/sessionSerialization.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sessionSerialization.js","sourceRoot":"","sources":["../../../src/sandbox/runtime/sessionSerialization.ts"],"names":[],"mappings":";;AASA,oEA+FC;AAtGD,oDAKwB;AAEjB,KAAK,UAAU,4BAA4B,CAAC,IAWlD;IACC,MAAM,EACJ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,sBAAsB,EACtB,qBAAqB,EACrB,oBAAoB,EACpB,wBAAwB,GACzB,GAAG,IAAI,CAAC;IACT,IAAI,CAAC,MAAM,EAAE,qBAAqB,IAAI,kBAAkB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QACpE,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,eAAe,GAAG,IAAA,mDAAoC,EAC1D,YAAY,EACZ,MAAM,CACP,CAAC;IACF,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC1B,gCAAgC,CAAC,eAAe,CAAC,CAAC;IACpD,CAAC;IACD,KAAK,MAAM,CAAC,eAAe,EAAE,OAAO,CAAC,IAAI,kBAAkB,CAAC,OAAO,EAAE,EAAE,CAAC;QACtE,MAAM,gBAAgB,GACpB,sBAAsB,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,eAAe,CAAC;QACjE,OAAO,eAAe,CAAC,eAAe,CAAC,CAAC;QACxC,MAAM,cAAc,GAAG,qBAAqB,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QAClE,IACE,cAAc;YACd,CAAC,oBAAoB;YACrB,MAAM,CAAC,2BAA2B;YAClC,CAAC,CAAC,MAAM,MAAM,CAAC,2BAA2B,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAC1D,CAAC;YACD,SAAS;QACX,CAAC;QACD,MAAM,gBAAgB,GACpB,cAAc;YACd,oBAAoB;YACpB,MAAM,CAAC,6BAA6B;YAClC,CAAC,CAAC,MAAM,MAAM,CAAC,6BAA6B,CAAC,OAAO,CAAC,KAAK,CAAC;YAC3D,CAAC,CAAC,IAAI,CAAC;QACX,gFAAgF;QAChF,+CAA+C;QAC/C,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,qBAAqB,CAAC,OAAO,CAAC,KAAK,EAAE;YACtE,oBAAoB,EAAE,cAAc,IAAI,CAAC,CAAC,oBAAoB;YAC9D,gBAAgB;YAChB,uBAAuB,EAAE,cAAc,IAAI,CAAC,oBAAoB;SACjE,CAAC,CAAC;QACH,eAAe,CAAC,eAAe,CAAC,GAAG;YACjC,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,eAAe;YACf,gBAAgB;YAChB,YAAY,EAAE,IAAA,qCAAsB,EAClC,MAAM,CAAC,SAAS,EAChB,OAAO,CAAC,KAAK,EACb,aAAa,CACd;YACD,GAAG,CAAC,cAAc,IAAI,oBAAoB;gBACxC,CAAC,CAAC;oBACE,qBAAqB,EAAE,IAAI;oBAC3B,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC;iBACzD;gBACH,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;IACJ,CAAC;IAED,MAAM,eAAe,GACnB,wBAAwB,IAAI,eAAe,CAAC,wBAAwB,CAAC;QACnE,CAAC,CAAC,wBAAwB;QAC1B,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;IACtC,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,YAAY,GAAG,eAAe,CAAC,eAAe,CAAC,CAAC;IACtD,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO;QACL,SAAS,EAAE,YAAY,CAAC,SAAS;QACjC,eAAe;QACf,gBAAgB,EAAE,YAAY,CAAC,gBAAgB;QAC/C,YAAY,EAAE,YAAY,CAAC,YAAY;QACvC,eAAe;KAChB,CAAC;AACJ,CAAC;AAED,SAAS,gCAAgC,CACvC,eAA8D;IAE9D,KAAK,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC;QAChE,IAAI,KAAK,CAAC,qBAAqB,EAAE,CAAC;YAChC,OAAO,eAAe,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/sandbox/runtime/sessionSerialization.mjs

@@ -0,0 +1,71 @@
+import { getPreviousSerializedSessionsByAgent, toSessionStateEnvelope, } from "./sessionState.mjs";
+export async function serializeSandboxRuntimeState(args) {
+    const { client, sandboxState, sessionsByAgentKey, sessionAgentNamesByKey, ownedSessionAgentKeys, includeOwnedSessions, preferredCurrentAgentKey, } = args;
+    if (!client?.serializeSessionState || sessionsByAgentKey.size === 0) {
+        return undefined;
+    }
+    const sessionsByAgent = getPreviousSerializedSessionsByAgent(sandboxState, client);
+    if (!includeOwnedSessions) {
+        dropPreservedOwnedSessionEntries(sessionsByAgent);
+    }
+    for (const [currentAgentKey, session] of sessionsByAgentKey.entries()) {
+        const currentAgentName = sessionAgentNamesByKey.get(currentAgentKey) ?? currentAgentKey;
+        delete sessionsByAgent[currentAgentKey];
+        const isOwnedSession = ownedSessionAgentKeys.has(currentAgentKey);
+        if (isOwnedSession &&
+            !includeOwnedSessions &&
+            client.canPersistOwnedSessionState &&
+            !(await client.canPersistOwnedSessionState(session.state))) {
+            continue;
+        }
+        const reuseLiveSession = isOwnedSession &&
+            includeOwnedSessions &&
+            client.canReusePreservedOwnedSession
+            ? await client.canReusePreservedOwnedSession(session.state)
+            : true;
+        // Start from previous entries so handoff agents that were not touched this turn
+        // keep their sandbox sessions in the RunState.
+        const providerState = await client.serializeSessionState(session.state, {
+            preserveOwnedSession: isOwnedSession && !!includeOwnedSessions,
+            reuseLiveSession,
+            willCloseAfterSerialize: isOwnedSession && !includeOwnedSessions,
+        });
+        sessionsByAgent[currentAgentKey] = {
+            backendId: client.backendId,
+            currentAgentKey,
+            currentAgentName,
+            sessionState: toSessionStateEnvelope(client.backendId, session.state, providerState),
+            ...(isOwnedSession && includeOwnedSessions
+                ? {
+                    preservedOwnedSession: true,
+                    ...(reuseLiveSession ? {} : { reuseLiveSession: false }),
+                }
+                : {}),
+        };
+    }
+    const currentAgentKey = preferredCurrentAgentKey && sessionsByAgent[preferredCurrentAgentKey]
+        ? preferredCurrentAgentKey
+        : Object.keys(sessionsByAgent)[0];
+    if (!currentAgentKey) {
+        return undefined;
+    }
+    const currentEntry = sessionsByAgent[currentAgentKey];
+    if (!currentEntry) {
+        return undefined;
+    }
+    return {
+        backendId: currentEntry.backendId,
+        currentAgentKey,
+        currentAgentName: currentEntry.currentAgentName,
+        sessionState: currentEntry.sessionState,
+        sessionsByAgent,
+    };
+}
+function dropPreservedOwnedSessionEntries(sessionsByAgent) {
+    for (const [agentKey, entry] of Object.entries(sessionsByAgent)) {
+        if (entry.preservedOwnedSession) {
+            delete sessionsByAgent[agentKey];
+        }
+    }
+}
+//# sourceMappingURL=sessionSerialization.mjs.map

package/dist/sandbox/runtime/sessionSerialization.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"sessionSerialization.mjs","sourceRoot":"","sources":["../../../src/sandbox/runtime/sessionSerialization.ts"],"names":[],"mappings":"OAEO,EACL,oCAAoC,EACpC,sBAAsB,GAGvB;AAED,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAAC,IAWlD;IACC,MAAM,EACJ,MAAM,EACN,YAAY,EACZ,kBAAkB,EAClB,sBAAsB,EACtB,qBAAqB,EACrB,oBAAoB,EACpB,wBAAwB,GACzB,GAAG,IAAI,CAAC;IACT,IAAI,CAAC,MAAM,EAAE,qBAAqB,IAAI,kBAAkB,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QACpE,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,eAAe,GAAG,oCAAoC,CAC1D,YAAY,EACZ,MAAM,CACP,CAAC;IACF,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC1B,gCAAgC,CAAC,eAAe,CAAC,CAAC;IACpD,CAAC;IACD,KAAK,MAAM,CAAC,eAAe,EAAE,OAAO,CAAC,IAAI,kBAAkB,CAAC,OAAO,EAAE,EAAE,CAAC;QACtE,MAAM,gBAAgB,GACpB,sBAAsB,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,eAAe,CAAC;QACjE,OAAO,eAAe,CAAC,eAAe,CAAC,CAAC;QACxC,MAAM,cAAc,GAAG,qBAAqB,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QAClE,IACE,cAAc;YACd,CAAC,oBAAoB;YACrB,MAAM,CAAC,2BAA2B;YAClC,CAAC,CAAC,MAAM,MAAM,CAAC,2BAA2B,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAC1D,CAAC;YACD,SAAS;QACX,CAAC;QACD,MAAM,gBAAgB,GACpB,cAAc;YACd,oBAAoB;YACpB,MAAM,CAAC,6BAA6B;YAClC,CAAC,CAAC,MAAM,MAAM,CAAC,6BAA6B,CAAC,OAAO,CAAC,KAAK,CAAC;YAC3D,CAAC,CAAC,IAAI,CAAC;QACX,gFAAgF;QAChF,+CAA+C;QAC/C,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,qBAAqB,CAAC,OAAO,CAAC,KAAK,EAAE;YACtE,oBAAoB,EAAE,cAAc,IAAI,CAAC,CAAC,oBAAoB;YAC9D,gBAAgB;YAChB,uBAAuB,EAAE,cAAc,IAAI,CAAC,oBAAoB;SACjE,CAAC,CAAC;QACH,eAAe,CAAC,eAAe,CAAC,GAAG;YACjC,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,eAAe;YACf,gBAAgB;YAChB,YAAY,EAAE,sBAAsB,CAClC,MAAM,CAAC,SAAS,EAChB,OAAO,CAAC,KAAK,EACb,aAAa,CACd;YACD,GAAG,CAAC,cAAc,IAAI,oBAAoB;gBACxC,CAAC,CAAC;oBACE,qBAAqB,EAAE,IAAI;oBAC3B,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC;iBACzD;gBACH,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;IACJ,CAAC;IAED,MAAM,eAAe,GACnB,wBAAwB,IAAI,eAAe,CAAC,wBAAwB,CAAC;QACnE,CAAC,CAAC,wBAAwB;QAC1B,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAAC;IACtC,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,YAAY,GAAG,eAAe,CAAC,eAAe,CAAC,CAAC;IACtD,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO;QACL,SAAS,EAAE,YAAY,CAAC,SAAS;QACjC,eAAe;QACf,gBAAgB,EAAE,YAAY,CAAC,gBAAgB;QAC/C,YAAY,EAAE,YAAY,CAAC,YAAY;QACvC,eAAe;KAChB,CAAC;AACJ,CAAC;AAED,SAAS,gCAAgC,CACvC,eAA8D;IAE9D,KAAK,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,EAAE,CAAC;QAChE,IAAI,KAAK,CAAC,qBAAqB,EAAE,CAAC;YAChC,OAAO,eAAe,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/sandbox/runtime/sessionState.d.ts

@@ -0,0 +1,26 @@
+import type { RunState } from '../../runState';
+import type { Agent, AgentOutputType } from '../../agent';
+import type { SandboxClient } from '../client';
+import { type SandboxSessionState, type SandboxSessionStateEnvelope } from '../session';
+export type SerializedSandboxSessionEntry = {
+    backendId: string;
+    currentAgentKey: string;
+    currentAgentName: string;
+    sessionState: SandboxSessionStateEnvelope;
+    preservedOwnedSession?: boolean;
+    reuseLiveSession?: boolean;
+};
+export type SerializedSandboxState = {
+    backendId: string;
+    currentAgentKey: string;
+    currentAgentName: string;
+    sessionState: SandboxSessionStateEnvelope;
+    sessionsByAgent: Record<string, SerializedSandboxSessionEntry>;
+};
+export declare function getSerializedSandboxState<TContext>(runState: RunState<TContext, Agent<TContext, AgentOutputType>> | undefined): SerializedSandboxState | undefined;
+export declare function toSessionStateEnvelope(backendId: string, state: SandboxSessionState, providerState: Record<string, unknown>): SandboxSessionStateEnvelope;
+export declare function assertSessionStateEnvelope(client: SandboxClient, envelope: SandboxSessionStateEnvelope): void;
+export declare function deserializeSandboxSessionStateEntry(client: SandboxClient, serializedEntry: SerializedSandboxSessionEntry | undefined): Promise<SandboxSessionState | undefined>;
+export declare function getSerializedSessionEntryForAgent(sandboxState: SerializedSandboxState | undefined, currentAgentKey: string): SerializedSandboxSessionEntry | undefined;
+export declare function getPreviousSerializedSessionsByAgent(sandboxState: SerializedSandboxState | undefined, client: SandboxClient): Record<string, SerializedSandboxSessionEntry>;
+export declare function hasPreservedOwnedSessions(sandboxState: SerializedSandboxState | undefined): boolean;

package/dist/sandbox/runtime/sessionState.js

@@ -0,0 +1,113 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getSerializedSandboxState = getSerializedSandboxState;
+exports.toSessionStateEnvelope = toSessionStateEnvelope;
+exports.assertSessionStateEnvelope = assertSessionStateEnvelope;
+exports.deserializeSandboxSessionStateEntry = deserializeSandboxSessionStateEntry;
+exports.getSerializedSessionEntryForAgent = getSerializedSessionEntryForAgent;
+exports.getPreviousSerializedSessionsByAgent = getPreviousSerializedSessionsByAgent;
+exports.hasPreservedOwnedSessions = hasPreservedOwnedSessions;
+const errors_1 = require("../../errors.js");
+const session_1 = require("../session.js");
+const manifestPersistence_1 = require("../sandboxes/shared/manifestPersistence.js");
+function getSerializedSandboxState(runState) {
+    return runState?._sandbox;
+}
+function toSessionStateEnvelope(backendId, state, providerState) {
+    const persistentManifest = (0, manifestPersistence_1.serializeManifest)(state.manifest);
+    return {
+        version: session_1.SANDBOX_SESSION_STATE_VERSION,
+        backendId,
+        manifest: (0, manifestPersistence_1.serializeManifestRecord)(persistentManifest),
+        ...(state.snapshot !== undefined ? { snapshot: state.snapshot } : {}),
+        ...(state.snapshotFingerprint !== undefined
+            ? { snapshotFingerprint: state.snapshotFingerprint }
+            : {}),
+        ...(state.snapshotFingerprintVersion !== undefined
+            ? { snapshotFingerprintVersion: state.snapshotFingerprintVersion }
+            : {}),
+        workspaceReady: state.workspaceReady ?? true,
+        ...(state.exposedPorts
+            ? { exposedPorts: structuredClone(state.exposedPorts) }
+            : {}),
+        // Keep provider-owned fields separate from the SDK envelope so version and backend
+        // checks can run before deserializing provider-specific state.
+        providerState,
+    };
+}
+function assertSessionStateEnvelope(client, envelope) {
+    if (envelope.version !== session_1.SANDBOX_SESSION_STATE_VERSION) {
+        throw new errors_1.UserError(`Sandbox session state version ${envelope.version} is not supported. Please use version ${session_1.SANDBOX_SESSION_STATE_VERSION}.`);
+    }
+    if (envelope.backendId !== client.backendId) {
+        throw new errors_1.UserError('RunState sandbox session backend does not match the configured sandbox client.');
+    }
+}
+async function deserializeSandboxSessionStateEntry(client, serializedEntry) {
+    if (!serializedEntry) {
+        return undefined;
+    }
+    if (serializedEntry.backendId !== client.backendId) {
+        throw new errors_1.UserError('RunState sandbox backend does not match the configured sandbox client.');
+    }
+    if (!client.deserializeSessionState) {
+        throw new errors_1.UserError('Sandbox client must implement deserializeSessionState() to resume RunState sandbox state.');
+    }
+    const envelope = serializedEntry.sessionState;
+    assertSessionStateEnvelope(client, envelope);
+    return await client.deserializeSessionState(providerStateWithSdkEnvelopeFields(envelope));
+}
+function providerStateWithSdkEnvelopeFields(envelope) {
+    return {
+        ...envelope.providerState,
+        manifest: envelope.manifest,
+        ...(envelope.snapshot !== undefined ? { snapshot: envelope.snapshot } : {}),
+        ...(envelope.snapshotFingerprint !== undefined
+            ? { snapshotFingerprint: envelope.snapshotFingerprint }
+            : {}),
+        ...(envelope.snapshotFingerprintVersion !== undefined
+            ? {
+                snapshotFingerprintVersion: envelope.snapshotFingerprintVersion,
+            }
+            : {}),
+        workspaceReady: envelope.workspaceReady,
+        ...(envelope.exposedPorts
+            ? { exposedPorts: structuredClone(envelope.exposedPorts) }
+            : {}),
+    };
+}
+function getSerializedSessionEntryForAgent(sandboxState, currentAgentKey) {
+    if (!sandboxState) {
+        return undefined;
+    }
+    return (sandboxState.sessionsByAgent?.[currentAgentKey] ??
+        (sandboxState.currentAgentKey === currentAgentKey
+            ? {
+                backendId: sandboxState.backendId,
+                currentAgentKey: sandboxState.currentAgentKey,
+                currentAgentName: sandboxState.currentAgentName,
+                sessionState: sandboxState.sessionState,
+            }
+            : undefined));
+}
+function getPreviousSerializedSessionsByAgent(sandboxState, client) {
+    if (!sandboxState || sandboxState.backendId !== client.backendId) {
+        return {};
+    }
+    const entries = {
+        ...(sandboxState.sessionsByAgent ?? {}),
+    };
+    if (sandboxState.currentAgentKey && !entries[sandboxState.currentAgentKey]) {
+        entries[sandboxState.currentAgentKey] = {
+            backendId: sandboxState.backendId,
+            currentAgentKey: sandboxState.currentAgentKey,
+            currentAgentName: sandboxState.currentAgentName,
+            sessionState: sandboxState.sessionState,
+        };
+    }
+    return Object.fromEntries(Object.entries(entries).filter(([, entry]) => entry.backendId === client.backendId));
+}
+function hasPreservedOwnedSessions(sandboxState) {
+    return Object.values(sandboxState?.sessionsByAgent ?? {}).some((entry) => entry.preservedOwnedSession);
+}
+//# sourceMappingURL=sessionState.js.map

package/dist/sandbox/runtime/sessionState.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sessionState.js","sourceRoot":"","sources":["../../../src/sandbox/runtime/sessionState.ts"],"names":[],"mappings":";;AA+BA,8DAIC;AAED,wDAyBC;AAED,gEAcC;AAED,kFAsBC;AAwBD,8EAmBC;AAED,oFAyBC;AAED,8DAMC;AApLD,4CAAyC;AAIzC,2CAIoB;AACpB,oFAGiD;AAmBjD,SAAgB,yBAAyB,CACvC,QAA0E;IAE1E,OAAO,QAAQ,EAAE,QAA8C,CAAC;AAClE,CAAC;AAED,SAAgB,sBAAsB,CACpC,SAAiB,EACjB,KAA0B,EAC1B,aAAsC;IAEtC,MAAM,kBAAkB,GAAG,IAAA,uCAAiB,EAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IAC7D,OAAO;QACL,OAAO,EAAE,uCAA6B;QACtC,SAAS;QACT,QAAQ,EAAE,IAAA,6CAAuB,EAAC,kBAAkB,CAAC;QACrD,GAAG,CAAC,KAAK,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACrE,GAAG,CAAC,KAAK,CAAC,mBAAmB,KAAK,SAAS;YACzC,CAAC,CAAC,EAAE,mBAAmB,EAAE,KAAK,CAAC,mBAAmB,EAAE;YACpD,CAAC,CAAC,EAAE,CAAC;QACP,GAAG,CAAC,KAAK,CAAC,0BAA0B,KAAK,SAAS;YAChD,CAAC,CAAC,EAAE,0BAA0B,EAAE,KAAK,CAAC,0BAA0B,EAAE;YAClE,CAAC,CAAC,EAAE,CAAC;QACP,cAAc,EAAE,KAAK,CAAC,cAAc,IAAI,IAAI;QAC5C,GAAG,CAAC,KAAK,CAAC,YAAY;YACpB,CAAC,CAAC,EAAE,YAAY,EAAE,eAAe,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE;YACvD,CAAC,CAAC,EAAE,CAAC;QACP,mFAAmF;QACnF,+DAA+D;QAC/D,aAAa;KACd,CAAC;AACJ,CAAC;AAED,SAAgB,0BAA0B,CACxC,MAAqB,EACrB,QAAqC;IAErC,IAAI,QAAQ,CAAC,OAAO,KAAK,uCAA6B,EAAE,CAAC;QACvD,MAAM,IAAI,kBAAS,CACjB,iCAAiC,QAAQ,CAAC,OAAO,yCAAyC,uCAA6B,GAAG,CAC3H,CAAC;IACJ,CAAC;IACD,IAAI,QAAQ,CAAC,SAAS,KAAK,MAAM,CAAC,SAAS,EAAE,CAAC;QAC5C,MAAM,IAAI,kBAAS,CACjB,gFAAgF,CACjF,CAAC;IACJ,CAAC;AACH,CAAC;AAEM,KAAK,UAAU,mCAAmC,CACvD,MAAqB,EACrB,eAA0D;IAE1D,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,eAAe,CAAC,SAAS,KAAK,MAAM,CAAC,SAAS,EAAE,CAAC;QACnD,MAAM,IAAI,kBAAS,CACjB,wEAAwE,CACzE,CAAC;IACJ,CAAC;IACD,IAAI,CAAC,MAAM,CAAC,uBAAuB,EAAE,CAAC;QACpC,MAAM,IAAI,kBAAS,CACjB,2FAA2F,CAC5F,CAAC;IACJ,CAAC;IACD,MAAM,QAAQ,GAAG,eAAe,CAAC,YAAY,CAAC;IAC9C,0BAA0B,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IAC7C,OAAO,MAAM,MAAM,CAAC,uBAAuB,CACzC,kCAAkC,CAAC,QAAQ,CAAC,CAC7C,CAAC;AACJ,CAAC;AAED,SAAS,kCAAkC,CACzC,QAAqC;IAErC,OAAO;QACL,GAAG,QAAQ,CAAC,aAAa;QACzB,QAAQ,EAAE,QAAQ,CAAC,QAAQ;QAC3B,GAAG,CAAC,QAAQ,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3E,GAAG,CAAC,QAAQ,CAAC,mBAAmB,KAAK,SAAS;YAC5C,CAAC,CAAC,EAAE,mBAAmB,EAAE,QAAQ,CAAC,mBAAmB,EAAE;YACvD,CAAC,CAAC,EAAE,CAAC;QACP,GAAG,CAAC,QAAQ,CAAC,0BAA0B,KAAK,SAAS;YACnD,CAAC,CAAC;gBACE,0BAA0B,EAAE,QAAQ,CAAC,0BAA0B;aAChE;YACH,CAAC,CAAC,EAAE,CAAC;QACP,cAAc,EAAE,QAAQ,CAAC,cAAc;QACvC,GAAG,CAAC,QAAQ,CAAC,YAAY;YACvB,CAAC,CAAC,EAAE,YAAY,EAAE,eAAe,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE;YAC1D,CAAC,CAAC,EAAE,CAAC;KACR,CAAC;AACJ,CAAC;AAED,SAAgB,iCAAiC,CAC/C,YAAgD,EAChD,eAAuB;IAEvB,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,CACL,YAAY,CAAC,eAAe,EAAE,CAAC,eAAe,CAAC;QAC/C,CAAC,YAAY,CAAC,eAAe,KAAK,eAAe;YAC/C,CAAC,CAAC;gBACE,SAAS,EAAE,YAAY,CAAC,SAAS;gBACjC,eAAe,EAAE,YAAY,CAAC,eAAe;gBAC7C,gBAAgB,EAAE,YAAY,CAAC,gBAAgB;gBAC/C,YAAY,EAAE,YAAY,CAAC,YAAY;aACxC;YACH,CAAC,CAAC,SAAS,CAAC,CACf,CAAC;AACJ,CAAC;AAED,SAAgB,oCAAoC,CAClD,YAAgD,EAChD,MAAqB;IAErB,IAAI,CAAC,YAAY,IAAI,YAAY,CAAC,SAAS,KAAK,MAAM,CAAC,SAAS,EAAE,CAAC;QACjE,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,OAAO,GAAkD;QAC7D,GAAG,CAAC,YAAY,CAAC,eAAe,IAAI,EAAE,CAAC;KACxC,CAAC;IACF,IAAI,YAAY,CAAC,eAAe,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,eAAe,CAAC,EAAE,CAAC;QAC3E,OAAO,CAAC,YAAY,CAAC,eAAe,CAAC,GAAG;YACtC,SAAS,EAAE,YAAY,CAAC,SAAS;YACjC,eAAe,EAAE,YAAY,CAAC,eAAe;YAC7C,gBAAgB,EAAE,YAAY,CAAC,gBAAgB;YAC/C,YAAY,EAAE,YAAY,CAAC,YAAY;SACxC,CAAC;IACJ,CAAC;IAED,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,CAC5B,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,KAAK,MAAM,CAAC,SAAS,CACpD,CACF,CAAC;AACJ,CAAC;AAED,SAAgB,yBAAyB,CACvC,YAAgD;IAEhD,OAAO,MAAM,CAAC,MAAM,CAAC,YAAY,EAAE,eAAe,IAAI,EAAE,CAAC,CAAC,IAAI,CAC5D,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,qBAAqB,CACvC,CAAC;AACJ,CAAC"}

package/dist/sandbox/runtime/sessionState.mjs

@@ -0,0 +1,104 @@
+import { UserError } from "../../errors.mjs";
+import { SANDBOX_SESSION_STATE_VERSION, } from "../session.mjs";
+import { serializeManifest, serializeManifestRecord, } from "../sandboxes/shared/manifestPersistence.mjs";
+export function getSerializedSandboxState(runState) {
+    return runState?._sandbox;
+}
+export function toSessionStateEnvelope(backendId, state, providerState) {
+    const persistentManifest = serializeManifest(state.manifest);
+    return {
+        version: SANDBOX_SESSION_STATE_VERSION,
+        backendId,
+        manifest: serializeManifestRecord(persistentManifest),
+        ...(state.snapshot !== undefined ? { snapshot: state.snapshot } : {}),
+        ...(state.snapshotFingerprint !== undefined
+            ? { snapshotFingerprint: state.snapshotFingerprint }
+            : {}),
+        ...(state.snapshotFingerprintVersion !== undefined
+            ? { snapshotFingerprintVersion: state.snapshotFingerprintVersion }
+            : {}),
+        workspaceReady: state.workspaceReady ?? true,
+        ...(state.exposedPorts
+            ? { exposedPorts: structuredClone(state.exposedPorts) }
+            : {}),
+        // Keep provider-owned fields separate from the SDK envelope so version and backend
+        // checks can run before deserializing provider-specific state.
+        providerState,
+    };
+}
+export function assertSessionStateEnvelope(client, envelope) {
+    if (envelope.version !== SANDBOX_SESSION_STATE_VERSION) {
+        throw new UserError(`Sandbox session state version ${envelope.version} is not supported. Please use version ${SANDBOX_SESSION_STATE_VERSION}.`);
+    }
+    if (envelope.backendId !== client.backendId) {
+        throw new UserError('RunState sandbox session backend does not match the configured sandbox client.');
+    }
+}
+export async function deserializeSandboxSessionStateEntry(client, serializedEntry) {
+    if (!serializedEntry) {
+        return undefined;
+    }
+    if (serializedEntry.backendId !== client.backendId) {
+        throw new UserError('RunState sandbox backend does not match the configured sandbox client.');
+    }
+    if (!client.deserializeSessionState) {
+        throw new UserError('Sandbox client must implement deserializeSessionState() to resume RunState sandbox state.');
+    }
+    const envelope = serializedEntry.sessionState;
+    assertSessionStateEnvelope(client, envelope);
+    return await client.deserializeSessionState(providerStateWithSdkEnvelopeFields(envelope));
+}
+function providerStateWithSdkEnvelopeFields(envelope) {
+    return {
+        ...envelope.providerState,
+        manifest: envelope.manifest,
+        ...(envelope.snapshot !== undefined ? { snapshot: envelope.snapshot } : {}),
+        ...(envelope.snapshotFingerprint !== undefined
+            ? { snapshotFingerprint: envelope.snapshotFingerprint }
+            : {}),
+        ...(envelope.snapshotFingerprintVersion !== undefined
+            ? {
+                snapshotFingerprintVersion: envelope.snapshotFingerprintVersion,
+            }
+            : {}),
+        workspaceReady: envelope.workspaceReady,
+        ...(envelope.exposedPorts
+            ? { exposedPorts: structuredClone(envelope.exposedPorts) }
+            : {}),
+    };
+}
+export function getSerializedSessionEntryForAgent(sandboxState, currentAgentKey) {
+    if (!sandboxState) {
+        return undefined;
+    }
+    return (sandboxState.sessionsByAgent?.[currentAgentKey] ??
+        (sandboxState.currentAgentKey === currentAgentKey
+            ? {
+                backendId: sandboxState.backendId,
+                currentAgentKey: sandboxState.currentAgentKey,
+                currentAgentName: sandboxState.currentAgentName,
+                sessionState: sandboxState.sessionState,
+            }
+            : undefined));
+}
+export function getPreviousSerializedSessionsByAgent(sandboxState, client) {
+    if (!sandboxState || sandboxState.backendId !== client.backendId) {
+        return {};
+    }
+    const entries = {
+        ...(sandboxState.sessionsByAgent ?? {}),
+    };
+    if (sandboxState.currentAgentKey && !entries[sandboxState.currentAgentKey]) {
+        entries[sandboxState.currentAgentKey] = {
+            backendId: sandboxState.backendId,
+            currentAgentKey: sandboxState.currentAgentKey,
+            currentAgentName: sandboxState.currentAgentName,
+            sessionState: sandboxState.sessionState,
+        };
+    }
+    return Object.fromEntries(Object.entries(entries).filter(([, entry]) => entry.backendId === client.backendId));
+}
+export function hasPreservedOwnedSessions(sandboxState) {
+    return Object.values(sandboxState?.sessionsByAgent ?? {}).some((entry) => entry.preservedOwnedSession);
+}
+//# sourceMappingURL=sessionState.mjs.map

package/dist/sandbox/runtime/sessionState.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"sessionState.mjs","sourceRoot":"","sources":["../../../src/sandbox/runtime/sessionState.ts"],"names":[],"mappings":"OAAO,EAAE,SAAS,EAAE;OAIb,EACL,6BAA6B,GAG9B;OACM,EACL,iBAAiB,EACjB,uBAAuB,GACxB;AAmBD,MAAM,UAAU,yBAAyB,CACvC,QAA0E;IAE1E,OAAO,QAAQ,EAAE,QAA8C,CAAC;AAClE,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,SAAiB,EACjB,KAA0B,EAC1B,aAAsC;IAEtC,MAAM,kBAAkB,GAAG,iBAAiB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IAC7D,OAAO;QACL,OAAO,EAAE,6BAA6B;QACtC,SAAS;QACT,QAAQ,EAAE,uBAAuB,CAAC,kBAAkB,CAAC;QACrD,GAAG,CAAC,KAAK,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACrE,GAAG,CAAC,KAAK,CAAC,mBAAmB,KAAK,SAAS;YACzC,CAAC,CAAC,EAAE,mBAAmB,EAAE,KAAK,CAAC,mBAAmB,EAAE;YACpD,CAAC,CAAC,EAAE,CAAC;QACP,GAAG,CAAC,KAAK,CAAC,0BAA0B,KAAK,SAAS;YAChD,CAAC,CAAC,EAAE,0BAA0B,EAAE,KAAK,CAAC,0BAA0B,EAAE;YAClE,CAAC,CAAC,EAAE,CAAC;QACP,cAAc,EAAE,KAAK,CAAC,cAAc,IAAI,IAAI;QAC5C,GAAG,CAAC,KAAK,CAAC,YAAY;YACpB,CAAC,CAAC,EAAE,YAAY,EAAE,eAAe,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE;YACvD,CAAC,CAAC,EAAE,CAAC;QACP,mFAAmF;QACnF,+DAA+D;QAC/D,aAAa;KACd,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,MAAqB,EACrB,QAAqC;IAErC,IAAI,QAAQ,CAAC,OAAO,KAAK,6BAA6B,EAAE,CAAC;QACvD,MAAM,IAAI,SAAS,CACjB,iCAAiC,QAAQ,CAAC,OAAO,yCAAyC,6BAA6B,GAAG,CAC3H,CAAC;IACJ,CAAC;IACD,IAAI,QAAQ,CAAC,SAAS,KAAK,MAAM,CAAC,SAAS,EAAE,CAAC;QAC5C,MAAM,IAAI,SAAS,CACjB,gFAAgF,CACjF,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,mCAAmC,CACvD,MAAqB,EACrB,eAA0D;IAE1D,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,IAAI,eAAe,CAAC,SAAS,KAAK,MAAM,CAAC,SAAS,EAAE,CAAC;QACnD,MAAM,IAAI,SAAS,CACjB,wEAAwE,CACzE,CAAC;IACJ,CAAC;IACD,IAAI,CAAC,MAAM,CAAC,uBAAuB,EAAE,CAAC;QACpC,MAAM,IAAI,SAAS,CACjB,2FAA2F,CAC5F,CAAC;IACJ,CAAC;IACD,MAAM,QAAQ,GAAG,eAAe,CAAC,YAAY,CAAC;IAC9C,0BAA0B,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IAC7C,OAAO,MAAM,MAAM,CAAC,uBAAuB,CACzC,kCAAkC,CAAC,QAAQ,CAAC,CAC7C,CAAC;AACJ,CAAC;AAED,SAAS,kCAAkC,CACzC,QAAqC;IAErC,OAAO;QACL,GAAG,QAAQ,CAAC,aAAa;QACzB,QAAQ,EAAE,QAAQ,CAAC,QAAQ;QAC3B,GAAG,CAAC,QAAQ,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3E,GAAG,CAAC,QAAQ,CAAC,mBAAmB,KAAK,SAAS;YAC5C,CAAC,CAAC,EAAE,mBAAmB,EAAE,QAAQ,CAAC,mBAAmB,EAAE;YACvD,CAAC,CAAC,EAAE,CAAC;QACP,GAAG,CAAC,QAAQ,CAAC,0BAA0B,KAAK,SAAS;YACnD,CAAC,CAAC;gBACE,0BAA0B,EAAE,QAAQ,CAAC,0BAA0B;aAChE;YACH,CAAC,CAAC,EAAE,CAAC;QACP,cAAc,EAAE,QAAQ,CAAC,cAAc;QACvC,GAAG,CAAC,QAAQ,CAAC,YAAY;YACvB,CAAC,CAAC,EAAE,YAAY,EAAE,eAAe,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE;YAC1D,CAAC,CAAC,EAAE,CAAC;KACR,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,iCAAiC,CAC/C,YAAgD,EAChD,eAAuB;IAEvB,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,CACL,YAAY,CAAC,eAAe,EAAE,CAAC,eAAe,CAAC;QAC/C,CAAC,YAAY,CAAC,eAAe,KAAK,eAAe;YAC/C,CAAC,CAAC;gBACE,SAAS,EAAE,YAAY,CAAC,SAAS;gBACjC,eAAe,EAAE,YAAY,CAAC,eAAe;gBAC7C,gBAAgB,EAAE,YAAY,CAAC,gBAAgB;gBAC/C,YAAY,EAAE,YAAY,CAAC,YAAY;aACxC;YACH,CAAC,CAAC,SAAS,CAAC,CACf,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,oCAAoC,CAClD,YAAgD,EAChD,MAAqB;IAErB,IAAI,CAAC,YAAY,IAAI,YAAY,CAAC,SAAS,KAAK,MAAM,CAAC,SAAS,EAAE,CAAC;QACjE,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,MAAM,OAAO,GAAkD;QAC7D,GAAG,CAAC,YAAY,CAAC,eAAe,IAAI,EAAE,CAAC;KACxC,CAAC;IACF,IAAI,YAAY,CAAC,eAAe,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,eAAe,CAAC,EAAE,CAAC;QAC3E,OAAO,CAAC,YAAY,CAAC,eAAe,CAAC,GAAG;YACtC,SAAS,EAAE,YAAY,CAAC,SAAS;YACjC,eAAe,EAAE,YAAY,CAAC,eAAe;YAC7C,gBAAgB,EAAE,YAAY,CAAC,gBAAgB;YAC/C,YAAY,EAAE,YAAY,CAAC,YAAY;SACxC,CAAC;IACJ,CAAC;IAED,OAAO,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,CAC5B,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,KAAK,MAAM,CAAC,SAAS,CACpD,CACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,YAAgD;IAEhD,OAAO,MAAM,CAAC,MAAM,CAAC,YAAY,EAAE,eAAe,IAAI,EAAE,CAAC,CAAC,IAAI,CAC5D,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,qBAAqB,CACvC,CAAC;AACJ,CAAC"}

package/dist/sandbox/runtime/spans.d.ts

@@ -0,0 +1 @@
+export declare function withSandboxSpan<T>(name: string, data: Record<string, unknown>, fn: () => Promise<T>): Promise<T>;

package/dist/sandbox/runtime/spans.js

@@ -0,0 +1,51 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.withSandboxSpan = withSandboxSpan;
+const tracing_1 = require("../../tracing/index.js");
+const events_1 = require("../events.js");
+async function withSandboxSpan(name, data, fn) {
+    const startedAt = Date.now();
+    await (0, events_1.emitSandboxEvent)({
+        type: 'sandbox_operation',
+        name,
+        phase: 'start',
+        timestamp: new Date(startedAt).toISOString(),
+        data: { ...data },
+    });
+    const runWithEvents = async () => {
+        try {
+            const result = await fn();
+            await (0, events_1.emitSandboxEvent)({
+                type: 'sandbox_operation',
+                name,
+                phase: 'end',
+                timestamp: new Date().toISOString(),
+                data: { ...data },
+                durationMs: Date.now() - startedAt,
+            });
+            return result;
+        }
+        catch (error) {
+            await (0, events_1.emitSandboxEvent)({
+                type: 'sandbox_operation',
+                name,
+                phase: 'error',
+                timestamp: new Date().toISOString(),
+                data: { ...data },
+                durationMs: Date.now() - startedAt,
+                error: (0, events_1.serializeSandboxEventError)(error),
+            });
+            throw error;
+        }
+    };
+    if (!(0, tracing_1.getCurrentTrace)()) {
+        return await runWithEvents();
+    }
+    return await (0, tracing_1.withCustomSpan)(async () => await runWithEvents(), {
+        data: {
+            name,
+            data,
+        },
+    });
+}
+//# sourceMappingURL=spans.js.map

package/dist/sandbox/runtime/spans.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"spans.js","sourceRoot":"","sources":["../../../src/sandbox/runtime/spans.ts"],"names":[],"mappings":";;AAGA,0CAkDC;AArDD,oDAAgE;AAChE,yCAAyE;AAElE,KAAK,UAAU,eAAe,CACnC,IAAY,EACZ,IAA6B,EAC7B,EAAoB;IAEpB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,IAAA,yBAAgB,EAAC;QACrB,IAAI,EAAE,mBAAmB;QACzB,IAAI;QACJ,KAAK,EAAE,OAAO;QACd,SAAS,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE;QAC5C,IAAI,EAAE,EAAE,GAAG,IAAI,EAAE;KAClB,CAAC,CAAC;IAEH,MAAM,aAAa,GAAG,KAAK,IAAgB,EAAE;QAC3C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,EAAE,EAAE,CAAC;YAC1B,MAAM,IAAA,yBAAgB,EAAC;gBACrB,IAAI,EAAE,mBAAmB;gBACzB,IAAI;gBACJ,KAAK,EAAE,KAAK;gBACZ,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,IAAI,EAAE,EAAE,GAAG,IAAI,EAAE;gBACjB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aACnC,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAA,yBAAgB,EAAC;gBACrB,IAAI,EAAE,mBAAmB;gBACzB,IAAI;gBACJ,KAAK,EAAE,OAAO;gBACd,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,IAAI,EAAE,EAAE,GAAG,IAAI,EAAE;gBACjB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBAClC,KAAK,EAAE,IAAA,mCAA0B,EAAC,KAAK,CAAC;aACzC,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC,CAAC;IAEF,IAAI,CAAC,IAAA,yBAAe,GAAE,EAAE,CAAC;QACvB,OAAO,MAAM,aAAa,EAAE,CAAC;IAC/B,CAAC;IAED,OAAO,MAAM,IAAA,wBAAc,EAAC,KAAK,IAAI,EAAE,CAAC,MAAM,aAAa,EAAE,EAAE;QAC7D,IAAI,EAAE;YACJ,IAAI;YACJ,IAAI;SACL;KACF,CAAC,CAAC;AACL,CAAC"}

package/dist/sandbox/runtime/spans.mjs

@@ -0,0 +1,48 @@
+import { getCurrentTrace, withCustomSpan } from "../../tracing/index.mjs";
+import { emitSandboxEvent, serializeSandboxEventError } from "../events.mjs";
+export async function withSandboxSpan(name, data, fn) {
+    const startedAt = Date.now();
+    await emitSandboxEvent({
+        type: 'sandbox_operation',
+        name,
+        phase: 'start',
+        timestamp: new Date(startedAt).toISOString(),
+        data: { ...data },
+    });
+    const runWithEvents = async () => {
+        try {
+            const result = await fn();
+            await emitSandboxEvent({
+                type: 'sandbox_operation',
+                name,
+                phase: 'end',
+                timestamp: new Date().toISOString(),
+                data: { ...data },
+                durationMs: Date.now() - startedAt,
+            });
+            return result;
+        }
+        catch (error) {
+            await emitSandboxEvent({
+                type: 'sandbox_operation',
+                name,
+                phase: 'error',
+                timestamp: new Date().toISOString(),
+                data: { ...data },
+                durationMs: Date.now() - startedAt,
+                error: serializeSandboxEventError(error),
+            });
+            throw error;
+        }
+    };
+    if (!getCurrentTrace()) {
+        return await runWithEvents();
+    }
+    return await withCustomSpan(async () => await runWithEvents(), {
+        data: {
+            name,
+            data,
+        },
+    });
+}
+//# sourceMappingURL=spans.mjs.map

package/dist/sandbox/runtime/spans.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"spans.mjs","sourceRoot":"","sources":["../../../src/sandbox/runtime/spans.ts"],"names":[],"mappings":"OAAO,EAAE,eAAe,EAAE,cAAc,EAAE;OACnC,EAAE,gBAAgB,EAAE,0BAA0B,EAAE;AAEvD,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,IAAY,EACZ,IAA6B,EAC7B,EAAoB;IAEpB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,gBAAgB,CAAC;QACrB,IAAI,EAAE,mBAAmB;QACzB,IAAI;QACJ,KAAK,EAAE,OAAO;QACd,SAAS,EAAE,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE;QAC5C,IAAI,EAAE,EAAE,GAAG,IAAI,EAAE;KAClB,CAAC,CAAC;IAEH,MAAM,aAAa,GAAG,KAAK,IAAgB,EAAE;QAC3C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,EAAE,EAAE,CAAC;YAC1B,MAAM,gBAAgB,CAAC;gBACrB,IAAI,EAAE,mBAAmB;gBACzB,IAAI;gBACJ,KAAK,EAAE,KAAK;gBACZ,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,IAAI,EAAE,EAAE,GAAG,IAAI,EAAE;gBACjB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aACnC,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,gBAAgB,CAAC;gBACrB,IAAI,EAAE,mBAAmB;gBACzB,IAAI;gBACJ,KAAK,EAAE,OAAO;gBACd,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,IAAI,EAAE,EAAE,GAAG,IAAI,EAAE;gBACjB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBAClC,KAAK,EAAE,0BAA0B,CAAC,KAAK,CAAC;aACzC,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC,CAAC;IAEF,IAAI,CAAC,eAAe,EAAE,EAAE,CAAC;QACvB,OAAO,MAAM,aAAa,EAAE,CAAC;IAC/B,CAAC;IAED,OAAO,MAAM,cAAc,CAAC,KAAK,IAAI,EAAE,CAAC,MAAM,aAAa,EAAE,EAAE;QAC7D,IAAI,EAAE;YACJ,IAAI;YACJ,IAAI;SACL;KACF,CAAC,CAAC;AACL,CAAC"}

package/dist/sandbox/runtime/toolRehydration.d.ts

@@ -0,0 +1,34 @@
+import type { Agent } from '../../agent';
+import type { ProcessedResponse } from '../../runner/types';
+import type { ApplyPatchTool, ComputerTool, FunctionTool, ShellTool, Tool } from '../../tool';
+import * as protocol from '../../types/protocol';
+export declare function processedResponseRequiresExecutionToolRehydration(processedResponse: ProcessedResponse<any> | undefined): boolean;
+export declare function getSerializedFunctionToolPlaceholder<TContext>(args: {
+    agent: Agent<any, any>;
+    baseAgentTools: Tool<any>[];
+    serializedTool: unknown;
+    toolCall: protocol.FunctionCallItem;
+    toolIdentity: string;
+    allowSerializedExecutionToolPlaceholder: boolean;
+}): FunctionTool<TContext> | undefined;
+export declare function getSerializedComputerToolPlaceholder(args: {
+    agent: Agent<any, any>;
+    baseAgentTools: Tool<any>[];
+    serializedTool: unknown;
+    toolName: string;
+    allowSerializedExecutionToolPlaceholder: boolean;
+}): ComputerTool<any, any> | undefined;
+export declare function getSerializedShellToolPlaceholder(args: {
+    agent: Agent<any, any>;
+    baseAgentTools: Tool<any>[];
+    serializedTool: unknown;
+    toolName: string;
+    allowSerializedExecutionToolPlaceholder: boolean;
+}): ShellTool | undefined;
+export declare function getSerializedApplyPatchToolPlaceholder(args: {
+    agent: Agent<any, any>;
+    baseAgentTools: Tool<any>[];
+    serializedTool: unknown;
+    toolName: string;
+    allowSerializedExecutionToolPlaceholder: boolean;
+}): ApplyPatchTool | undefined;