@openai/agents-core 0.8.4 → 0.9.0

package/dist/sandbox/runtime/agentPreparation.js

@@ -0,0 +1,178 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getDefaultSandboxInstructions = void 0;
+exports.cloneSandboxCapabilities = cloneSandboxCapabilities;
+exports.prepareSandboxAgent = prepareSandboxAgent;
+const defaultModel_1 = require("../../defaultModel.js");
+const errors_1 = require("../../errors.js");
+const modelSettings_1 = require("../../runner/modelSettings.js");
+const manifest_1 = require("../manifest.js");
+const prompts_1 = require("./prompts.js");
+const runAsManifest_1 = require("./runAsManifest.js");
+var prompts_2 = require("./prompts.js");
+Object.defineProperty(exports, "getDefaultSandboxInstructions", { enumerable: true, get: function () { return prompts_2.getDefaultSandboxInstructions; } });
+function cloneSandboxCapabilities(capabilities) {
+    return capabilities.map((capability) => capability.clone());
+}
+function prepareSandboxAgent({ agent, session, capabilities, runConfigModel, processManifest = true, }) {
+    const { model: resolvedModel, modelInstance: resolvedModelInstance } = resolveSandboxModel(agent, runConfigModel);
+    // Capabilities are cloned before binding because tools, instructions, and model
+    // sampling parameters can depend on the live session for this run.
+    const boundCapabilities = cloneSandboxCapabilities(capabilities ?? agent.capabilities).map((capability) => capability
+        .bind(session)
+        .bindRunAs(agent.runAs)
+        .bindModel(resolvedModel, resolvedModelInstance));
+    const boundCapabilityTypes = new Set(boundCapabilities.map((capability) => capability.type));
+    const runtimeManifest = processManifest
+        ? boundCapabilities.reduce((manifest, capability) => capability.processManifest(manifest), resolveManifest(agent, session))
+        : resolveManifest(agent, session);
+    for (const capability of boundCapabilities) {
+        const missingTypes = [...capability.requiredCapabilityTypes()].filter((requiredType) => !boundCapabilityTypes.has(requiredType));
+        if (missingTypes.length > 0) {
+            throw new errors_1.UserError(`${capability.type} requires missing capabilities: ${missingTypes.sort().join(', ')}`);
+        }
+    }
+    const providerData = boundCapabilities.reduce((samplingParams, capability) => 
+    // Provider data often nests model-specific settings; merge recursively so a
+    // capability can add its section without replacing user-supplied siblings.
+    mergeSandboxProviderData(samplingParams, capability.samplingParams({
+        model: resolvedModel,
+        ...(resolvedModelInstance
+            ? { modelInstance: resolvedModelInstance }
+            : {}),
+    })), {});
+    const tools = [
+        ...agent.tools,
+        ...boundCapabilities.flatMap((capability) => capability.tools()),
+    ];
+    const prepared = agent.clone({
+        capabilities: boundCapabilities,
+        tools,
+        modelSettings: {
+            ...agent.modelSettings,
+            providerData: mergeSandboxProviderData(agent.modelSettings.providerData, providerData),
+        },
+        instructions: async (runContext, preparedAgent) => {
+            const capabilityInstructions = await Promise.all(boundCapabilities.map((capability) => capability.instructions(runtimeManifest)));
+            const segments = [];
+            const baseInstructions = await renderBaseInstructions(agent, runContext, preparedAgent);
+            if (baseInstructions) {
+                segments.push(baseInstructions);
+            }
+            const agentInstructions = await renderAgentInstructions(agent.instructions, runContext, preparedAgent);
+            if (agentInstructions) {
+                segments.push((0, prompts_1.renderInstructionSection)('Agent instructions', agentInstructions));
+            }
+            const capabilityFragments = capabilityInstructions.filter((fragment) => Boolean(fragment));
+            if (capabilityFragments.length > 0) {
+                segments.push((0, prompts_1.renderInstructionSection)('Sandbox capability instructions', capabilityFragments.join('\n\n')));
+            }
+            const remoteMountPolicy = (0, prompts_1.renderRemoteMountPolicyInstructions)(runtimeManifest);
+            if (remoteMountPolicy) {
+                segments.push((0, prompts_1.renderInstructionSection)('Sandbox remote mount policy', remoteMountPolicy));
+            }
+            segments.push((0, prompts_1.renderFilesystemInstructions)(runtimeManifest));
+            return segments.join('\n\n');
+        },
+    });
+    if (tools.length === 0 && !agent.hasExplicitToolConfig()) {
+        // agent.clone() sees an explicit tools array, but an empty post-capability tool set
+        // should preserve the original "no explicit tools" semantics for tool choice.
+        prepared._toolsExplicitlyConfigured = false;
+    }
+    prepared.runtimeManifest = runtimeManifest;
+    return prepared;
+}
+function resolveManifest(agent, session) {
+    const manifestCandidate = session.state.manifest ?? agent.defaultManifest ?? new manifest_1.Manifest();
+    return (0, runAsManifest_1.manifestWithRunAsUser)((0, manifest_1.cloneManifest)(manifestCandidate), agent.runAs);
+}
+function resolveSandboxModel(agent, runConfigModel) {
+    let runConfigModelForSelection;
+    let runConfigModelInstance;
+    if (isResolvedSandboxRuntimeModel(runConfigModel)) {
+        runConfigModelForSelection = runConfigModel.model;
+        runConfigModelInstance = runConfigModel.modelInstance;
+    }
+    else {
+        runConfigModelForSelection = runConfigModel;
+        runConfigModelInstance =
+            runConfigModel && typeof runConfigModel !== 'string'
+                ? runConfigModel
+                : undefined;
+    }
+    const selectedModel = (0, modelSettings_1.selectModel)(agent.model, runConfigModelForSelection);
+    if (typeof selectedModel === 'string' && selectedModel.trim().length > 0) {
+        return {
+            model: selectedModel,
+            ...(selectedModel === runConfigModelForSelection && runConfigModelInstance
+                ? { modelInstance: runConfigModelInstance }
+                : {}),
+        };
+    }
+    if (selectedModel && typeof selectedModel !== 'string') {
+        return {
+            model: getSandboxRuntimeModelName(selectedModel) ?? (0, defaultModel_1.getDefaultModel)(),
+            modelInstance: selectedModel,
+        };
+    }
+    return {
+        model: (0, defaultModel_1.getDefaultModel)(),
+        ...(runConfigModelInstance
+            ? { modelInstance: runConfigModelInstance }
+            : {}),
+    };
+}
+function isResolvedSandboxRuntimeModel(model) {
+    return (typeof model === 'object' &&
+        model !== null &&
+        'model' in model &&
+        typeof model.model === 'string' &&
+        'modelInstance' in model);
+}
+function getSandboxRuntimeModelName(model) {
+    const candidate = model?.model ??
+        model?.name;
+    return typeof candidate === 'string' && candidate.trim().length > 0
+        ? candidate
+        : undefined;
+}
+async function renderAgentInstructions(instructions, runContext, agent) {
+    if (typeof instructions === 'function') {
+        return await instructions(runContext, agent);
+    }
+    return instructions;
+}
+async function renderBaseInstructions(agent, runContext, preparedAgent) {
+    const baseInstructions = agent.baseInstructions ?? (0, prompts_1.getDefaultSandboxInstructions)();
+    if (typeof baseInstructions === 'function') {
+        return await baseInstructions(runContext, preparedAgent);
+    }
+    return baseInstructions;
+}
+function mergeSandboxProviderData(base, update) {
+    if (!base) {
+        return mergePlainRecords({}, update);
+    }
+    return mergePlainRecords(base, update);
+}
+function mergePlainRecords(base, update) {
+    const merged = { ...base };
+    for (const [key, updateValue] of Object.entries(update)) {
+        const baseValue = merged[key];
+        if (isPlainRecord(baseValue) && isPlainRecord(updateValue)) {
+            merged[key] = mergePlainRecords(baseValue, updateValue);
+            continue;
+        }
+        merged[key] = updateValue;
+    }
+    return merged;
+}
+function isPlainRecord(value) {
+    if (typeof value !== 'object' || value === null || Array.isArray(value)) {
+        return false;
+    }
+    const prototype = Object.getPrototypeOf(value);
+    return prototype === Object.prototype || prototype === null;
+}
+//# sourceMappingURL=agentPreparation.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"agentPreparation.js","sourceRoot":"","sources":["../../../src/sandbox/runtime/agentPreparation.ts"],"names":[],"mappings":";;;AAmCA,4DAIC;AAED,kDA6IC;AArLD,wDAAqD;AACrD,4CAAyC;AAGzC,iEAAyD;AAGzD,6CAAsD;AAEtD,0CAKmB;AACnB,sDAAwD;AAExD,wCAA0D;AAAjD,wHAAA,6BAA6B,OAAA;AAiBtC,SAAgB,wBAAwB,CACtC,YAA0B;IAE1B,OAAO,YAAY,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC;AAC9D,CAAC;AAED,SAAgB,mBAAmB,CAA4C,EAC7E,KAAK,EACL,OAAO,EACP,YAAY,EACZ,cAAc,EACd,eAAe,GAAG,IAAI,GACqB;IAI3C,MAAM,EAAE,KAAK,EAAE,aAAa,EAAE,aAAa,EAAE,qBAAqB,EAAE,GAClE,mBAAmB,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;IAC7C,gFAAgF;IAChF,mEAAmE;IACnE,MAAM,iBAAiB,GAAG,wBAAwB,CAChD,YAAY,IAAI,KAAK,CAAC,YAAY,CACnC,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CACnB,UAAU;SACP,IAAI,CAAC,OAAO,CAAC;SACb,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC;SACtB,SAAS,CAAC,aAAa,EAAE,qBAAqB,CAAC,CACnD,CAAC;IACF,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAClC,iBAAiB,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,CACvD,CAAC;IACF,MAAM,eAAe,GAAG,eAAe;QACrC,CAAC,CAAC,iBAAiB,CAAC,MAAM,CACtB,CAAC,QAAQ,EAAE,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,eAAe,CAAC,QAAQ,CAAC,EAC9D,eAAe,CAAC,KAAK,EAAE,OAAO,CAAC,CAChC;QACH,CAAC,CAAC,eAAe,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAEpC,KAAK,MAAM,UAAU,IAAI,iBAAiB,EAAE,CAAC;QAC3C,MAAM,YAAY,GAAG,CAAC,GAAG,UAAU,CAAC,uBAAuB,EAAE,CAAC,CAAC,MAAM,CACnE,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC,oBAAoB,CAAC,GAAG,CAAC,YAAY,CAAC,CAC1D,CAAC;QACF,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,kBAAS,CACjB,GAAG,UAAU,CAAC,IAAI,mCAAmC,YAAY,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACtF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,MAAM,YAAY,GAAG,iBAAiB,CAAC,MAAM,CAC3C,CAAC,cAAc,EAAE,UAAU,EAAE,EAAE;IAC7B,4EAA4E;IAC5E,2EAA2E;IAC3E,wBAAwB,CACtB,cAAc,EACd,UAAU,CAAC,cAAc,CAAC;QACxB,KAAK,EAAE,aAAa;QACpB,GAAG,CAAC,qBAAqB;YACvB,CAAC,CAAC,EAAE,aAAa,EAAE,qBAAqB,EAAE;YAC1C,CAAC,CAAC,EAAE,CAAC;KACR,CAAC,CACH,EACH,EAAE,CACH,CAAC;IACF,MAAM,KAAK,GAAG;QACZ,GAAG,KAAK,CAAC,KAAK;QACd,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;KACjE,CAAC;IAEF,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC;QAC3B,YAAY,EAAE,iBAAiB;QAC/B,KAAK;QACL,aAAa,EAAE;YACb,GAAG,KAAK,CAAC,aAAa;YACtB,YAAY,EAAE,wBAAwB,CACpC,KAAK,CAAC,aAAa,CAAC,YAAmD,EACvE,YAAY,CACb;SACF;QACD,YAAY,EAAE,KAAK,EAAE,UAAU,EAAE,aAAa,EAAE,EAAE;YAChD,MAAM,sBAAsB,GAAG,MAAM,OAAO,CAAC,GAAG,CAC9C,iBAAiB,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CACnC,UAAU,CAAC,YAAY,CAAC,eAAe,CAAC,CACzC,CACF,CAAC;YACF,MAAM,QAAQ,GAAa,EAAE,CAAC;YAC9B,MAAM,gBAAgB,GAAG,MAAM,sBAAsB,CACnD,KAAK,EACL,UAAU,EACV,aAAa,CACd,CAAC;YACF,IAAI,gBAAgB,EAAE,CAAC;gBACrB,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAClC,CAAC;YAED,MAAM,iBAAiB,GAAG,MAAM,uBAAuB,CACrD,KAAK,CAAC,YAAY,EAClB,UAAU,EACV,aAAa,CACd,CAAC;YACF,IAAI,iBAAiB,EAAE,CAAC;gBACtB,QAAQ,CAAC,IAAI,CACX,IAAA,kCAAwB,EAAC,oBAAoB,EAAE,iBAAiB,CAAC,CAClE,CAAC;YACJ,CAAC;YAED,MAAM,mBAAmB,GAAG,sBAAsB,CAAC,MAAM,CACvD,CAAC,QAAQ,EAAsB,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,CACpD,CAAC;YACF,IAAI,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACnC,QAAQ,CAAC,IAAI,CACX,IAAA,kCAAwB,EACtB,iCAAiC,EACjC,mBAAmB,CAAC,IAAI,CAAC,MAAM,CAAC,CACjC,CACF,CAAC;YACJ,CAAC;YAED,MAAM,iBAAiB,GACrB,IAAA,6CAAmC,EAAC,eAAe,CAAC,CAAC;YACvD,IAAI,iBAAiB,EAAE,CAAC;gBACtB,QAAQ,CAAC,IAAI,CACX,IAAA,kCAAwB,EACtB,6BAA6B,EAC7B,iBAAiB,CAClB,CACF,CAAC;YACJ,CAAC;YAED,QAAQ,CAAC,IAAI,CAAC,IAAA,sCAA4B,EAAC,eAAe,CAAC,CAAC,CAAC;YAE7D,OAAO,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC/B,CAAC;KACF,CAAC,CAAC;IAEH,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,qBAAqB,EAAE,EAAE,CAAC;QACzD,oFAAoF;QACpF,8EAA8E;QAE5E,QAGD,CAAC,0BAA0B,GAAG,KAAK,CAAC;IACvC,CAAC;IAED,QAAQ,CAAC,eAAe,GAAG,eAAe,CAAC;IAC3C,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,eAAe,CACtB,KAAsC,EACtC,OAAgD;IAEhD,MAAM,iBAAiB,GACrB,OAAO,CAAC,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,eAAe,IAAI,IAAI,mBAAQ,EAAE,CAAC;IAEpE,OAAO,IAAA,qCAAqB,EAAC,IAAA,wBAAa,EAAC,iBAAiB,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,mBAAmB,CAC1B,KAAsC,EACtC,cAAoC;IAEpC,IAAI,0BAAsD,CAAC;IAC3D,IAAI,sBAAyC,CAAC;IAC9C,IAAI,6BAA6B,CAAC,cAAc,CAAC,EAAE,CAAC;QAClD,0BAA0B,GAAG,cAAc,CAAC,KAAK,CAAC;QAClD,sBAAsB,GAAG,cAAc,CAAC,aAAa,CAAC;IACxD,CAAC;SAAM,CAAC;QACN,0BAA0B,GAAG,cAAc,CAAC;QAC5C,sBAAsB;YACpB,cAAc,IAAI,OAAO,cAAc,KAAK,QAAQ;gBAClD,CAAC,CAAC,cAAc;gBAChB,CAAC,CAAC,SAAS,CAAC;IAClB,CAAC;IACD,MAAM,aAAa,GAAG,IAAA,2BAAW,EAAC,KAAK,CAAC,KAAK,EAAE,0BAA0B,CAAC,CAAC;IAC3E,IAAI,OAAO,aAAa,KAAK,QAAQ,IAAI,aAAa,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzE,OAAO;YACL,KAAK,EAAE,aAAa;YACpB,GAAG,CAAC,aAAa,KAAK,0BAA0B,IAAI,sBAAsB;gBACxE,CAAC,CAAC,EAAE,aAAa,EAAE,sBAAsB,EAAE;gBAC3C,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;IACJ,CAAC;IACD,IAAI,aAAa,IAAI,OAAO,aAAa,KAAK,QAAQ,EAAE,CAAC;QACvD,OAAO;YACL,KAAK,EAAE,0BAA0B,CAAC,aAAa,CAAC,IAAI,IAAA,8BAAe,GAAE;YACrE,aAAa,EAAE,aAAa;SAC7B,CAAC;IACJ,CAAC;IAED,OAAO;QACL,KAAK,EAAE,IAAA,8BAAe,GAAE;QACxB,GAAG,CAAC,sBAAsB;YACxB,CAAC,CAAC,EAAE,aAAa,EAAE,sBAAsB,EAAE;YAC3C,CAAC,CAAC,EAAE,CAAC;KACR,CAAC;AACJ,CAAC;AAED,SAAS,6BAA6B,CACpC,KAAsC;IAEtC,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,KAAK,IAAI;QACd,OAAO,IAAI,KAAK;QAChB,OAAQ,KAA6B,CAAC,KAAK,KAAK,QAAQ;QACxD,eAAe,IAAI,KAAK,CACzB,CAAC;AACJ,CAAC;AAED,SAAS,0BAA0B,CAAC,KAAY;IAC9C,MAAM,SAAS,GACZ,KAAuD,EAAE,KAAK;QAC9D,KAAuC,EAAE,IAAI,CAAC;IACjD,OAAO,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC;QACjE,CAAC,CAAC,SAAS;QACX,CAAC,CAAC,SAAS,CAAC;AAChB,CAAC;AAED,KAAK,UAAU,uBAAuB,CAIpC,YAA6D,EAC7D,UAAgC,EAChC,KAAyD;IAEzD,IAAI,OAAO,YAAY,KAAK,UAAU,EAAE,CAAC;QACvC,OAAO,MAAM,YAAY,CACvB,UAAU,EACV,KAAwC,CACzC,CAAC;IACJ,CAAC;IAED,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,KAAK,UAAU,sBAAsB,CAInC,KAAsC,EACtC,UAAgC,EAChC,aAAiE;IAEjE,MAAM,gBAAgB,GACpB,KAAK,CAAC,gBAAgB,IAAI,IAAA,uCAA6B,GAAE,CAAC;IAC5D,IAAI,OAAO,gBAAgB,KAAK,UAAU,EAAE,CAAC;QAC3C,OAAO,MAAM,gBAAgB,CAC3B,UAAU,EACV,aAAgD,CACjD,CAAC;IACJ,CAAC;IAED,OAAO,gBAAgB,CAAC;AAC1B,CAAC;AAED,SAAS,wBAAwB,CAC/B,IAAyC,EACzC,MAA+B;IAE/B,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,iBAAiB,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;IACvC,CAAC;IAED,OAAO,iBAAiB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;AACzC,CAAC;AAED,SAAS,iBAAiB,CACxB,IAA6B,EAC7B,MAA+B;IAE/B,MAAM,MAAM,GAA4B,EAAE,GAAG,IAAI,EAAE,CAAC;IAEpD,KAAK,MAAM,CAAC,GAAG,EAAE,WAAW,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QACxD,MAAM,SAAS,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QAC9B,IAAI,aAAa,CAAC,SAAS,CAAC,IAAI,aAAa,CAAC,WAAW,CAAC,EAAE,CAAC;YAC3D,MAAM,CAAC,GAAG,CAAC,GAAG,iBAAiB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;YACxD,SAAS;QACX,CAAC;QAED,MAAM,CAAC,GAAG,CAAC,GAAG,WAAW,CAAC;IAC5B,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACxE,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,SAAS,GAAG,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;IAC/C,OAAO,SAAS,KAAK,MAAM,CAAC,SAAS,IAAI,SAAS,KAAK,IAAI,CAAC;AAC9D,CAAC"}

package/dist/sandbox/runtime/agentPreparation.mjs

@@ -0,0 +1,172 @@
+import { getDefaultModel } from "../../defaultModel.mjs";
+import { UserError } from "../../errors.mjs";
+import { selectModel } from "../../runner/modelSettings.mjs";
+import { cloneManifest, Manifest } from "../manifest.mjs";
+import { getDefaultSandboxInstructions, renderFilesystemInstructions, renderInstructionSection, renderRemoteMountPolicyInstructions, } from "./prompts.mjs";
+import { manifestWithRunAsUser } from "./runAsManifest.mjs";
+export { getDefaultSandboxInstructions } from "./prompts.mjs";
+export function cloneSandboxCapabilities(capabilities) {
+    return capabilities.map((capability) => capability.clone());
+}
+export function prepareSandboxAgent({ agent, session, capabilities, runConfigModel, processManifest = true, }) {
+    const { model: resolvedModel, modelInstance: resolvedModelInstance } = resolveSandboxModel(agent, runConfigModel);
+    // Capabilities are cloned before binding because tools, instructions, and model
+    // sampling parameters can depend on the live session for this run.
+    const boundCapabilities = cloneSandboxCapabilities(capabilities ?? agent.capabilities).map((capability) => capability
+        .bind(session)
+        .bindRunAs(agent.runAs)
+        .bindModel(resolvedModel, resolvedModelInstance));
+    const boundCapabilityTypes = new Set(boundCapabilities.map((capability) => capability.type));
+    const runtimeManifest = processManifest
+        ? boundCapabilities.reduce((manifest, capability) => capability.processManifest(manifest), resolveManifest(agent, session))
+        : resolveManifest(agent, session);
+    for (const capability of boundCapabilities) {
+        const missingTypes = [...capability.requiredCapabilityTypes()].filter((requiredType) => !boundCapabilityTypes.has(requiredType));
+        if (missingTypes.length > 0) {
+            throw new UserError(`${capability.type} requires missing capabilities: ${missingTypes.sort().join(', ')}`);
+        }
+    }
+    const providerData = boundCapabilities.reduce((samplingParams, capability) => 
+    // Provider data often nests model-specific settings; merge recursively so a
+    // capability can add its section without replacing user-supplied siblings.
+    mergeSandboxProviderData(samplingParams, capability.samplingParams({
+        model: resolvedModel,
+        ...(resolvedModelInstance
+            ? { modelInstance: resolvedModelInstance }
+            : {}),
+    })), {});
+    const tools = [
+        ...agent.tools,
+        ...boundCapabilities.flatMap((capability) => capability.tools()),
+    ];
+    const prepared = agent.clone({
+        capabilities: boundCapabilities,
+        tools,
+        modelSettings: {
+            ...agent.modelSettings,
+            providerData: mergeSandboxProviderData(agent.modelSettings.providerData, providerData),
+        },
+        instructions: async (runContext, preparedAgent) => {
+            const capabilityInstructions = await Promise.all(boundCapabilities.map((capability) => capability.instructions(runtimeManifest)));
+            const segments = [];
+            const baseInstructions = await renderBaseInstructions(agent, runContext, preparedAgent);
+            if (baseInstructions) {
+                segments.push(baseInstructions);
+            }
+            const agentInstructions = await renderAgentInstructions(agent.instructions, runContext, preparedAgent);
+            if (agentInstructions) {
+                segments.push(renderInstructionSection('Agent instructions', agentInstructions));
+            }
+            const capabilityFragments = capabilityInstructions.filter((fragment) => Boolean(fragment));
+            if (capabilityFragments.length > 0) {
+                segments.push(renderInstructionSection('Sandbox capability instructions', capabilityFragments.join('\n\n')));
+            }
+            const remoteMountPolicy = renderRemoteMountPolicyInstructions(runtimeManifest);
+            if (remoteMountPolicy) {
+                segments.push(renderInstructionSection('Sandbox remote mount policy', remoteMountPolicy));
+            }
+            segments.push(renderFilesystemInstructions(runtimeManifest));
+            return segments.join('\n\n');
+        },
+    });
+    if (tools.length === 0 && !agent.hasExplicitToolConfig()) {
+        // agent.clone() sees an explicit tools array, but an empty post-capability tool set
+        // should preserve the original "no explicit tools" semantics for tool choice.
+        prepared._toolsExplicitlyConfigured = false;
+    }
+    prepared.runtimeManifest = runtimeManifest;
+    return prepared;
+}
+function resolveManifest(agent, session) {
+    const manifestCandidate = session.state.manifest ?? agent.defaultManifest ?? new Manifest();
+    return manifestWithRunAsUser(cloneManifest(manifestCandidate), agent.runAs);
+}
+function resolveSandboxModel(agent, runConfigModel) {
+    let runConfigModelForSelection;
+    let runConfigModelInstance;
+    if (isResolvedSandboxRuntimeModel(runConfigModel)) {
+        runConfigModelForSelection = runConfigModel.model;
+        runConfigModelInstance = runConfigModel.modelInstance;
+    }
+    else {
+        runConfigModelForSelection = runConfigModel;
+        runConfigModelInstance =
+            runConfigModel && typeof runConfigModel !== 'string'
+                ? runConfigModel
+                : undefined;
+    }
+    const selectedModel = selectModel(agent.model, runConfigModelForSelection);
+    if (typeof selectedModel === 'string' && selectedModel.trim().length > 0) {
+        return {
+            model: selectedModel,
+            ...(selectedModel === runConfigModelForSelection && runConfigModelInstance
+                ? { modelInstance: runConfigModelInstance }
+                : {}),
+        };
+    }
+    if (selectedModel && typeof selectedModel !== 'string') {
+        return {
+            model: getSandboxRuntimeModelName(selectedModel) ?? getDefaultModel(),
+            modelInstance: selectedModel,
+        };
+    }
+    return {
+        model: getDefaultModel(),
+        ...(runConfigModelInstance
+            ? { modelInstance: runConfigModelInstance }
+            : {}),
+    };
+}
+function isResolvedSandboxRuntimeModel(model) {
+    return (typeof model === 'object' &&
+        model !== null &&
+        'model' in model &&
+        typeof model.model === 'string' &&
+        'modelInstance' in model);
+}
+function getSandboxRuntimeModelName(model) {
+    const candidate = model?.model ??
+        model?.name;
+    return typeof candidate === 'string' && candidate.trim().length > 0
+        ? candidate
+        : undefined;
+}
+async function renderAgentInstructions(instructions, runContext, agent) {
+    if (typeof instructions === 'function') {
+        return await instructions(runContext, agent);
+    }
+    return instructions;
+}
+async function renderBaseInstructions(agent, runContext, preparedAgent) {
+    const baseInstructions = agent.baseInstructions ?? getDefaultSandboxInstructions();
+    if (typeof baseInstructions === 'function') {
+        return await baseInstructions(runContext, preparedAgent);
+    }
+    return baseInstructions;
+}
+function mergeSandboxProviderData(base, update) {
+    if (!base) {
+        return mergePlainRecords({}, update);
+    }
+    return mergePlainRecords(base, update);
+}
+function mergePlainRecords(base, update) {
+    const merged = { ...base };
+    for (const [key, updateValue] of Object.entries(update)) {
+        const baseValue = merged[key];
+        if (isPlainRecord(baseValue) && isPlainRecord(updateValue)) {
+            merged[key] = mergePlainRecords(baseValue, updateValue);
+            continue;
+        }
+        merged[key] = updateValue;
+    }
+    return merged;
+}
+function isPlainRecord(value) {
+    if (typeof value !== 'object' || value === null || Array.isArray(value)) {
+        return false;
+    }
+    const prototype = Object.getPrototypeOf(value);
+    return prototype === Object.prototype || prototype === null;
+}
+//# sourceMappingURL=agentPreparation.mjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"agentPreparation.mjs","sourceRoot":"","sources":["../../../src/sandbox/runtime/agentPreparation.ts"],"names":[],"mappings":"OACO,EAAE,eAAe,EAAE;OACnB,EAAE,SAAS,EAAE;OAGb,EAAE,WAAW,EAAE;OAGf,EAAE,aAAa,EAAE,QAAQ,EAAE;OAE3B,EACL,6BAA6B,EAC7B,4BAA4B,EAC5B,wBAAwB,EACxB,mCAAmC,GACpC;OACM,EAAE,qBAAqB,EAAE;OAEzB,EAAE,6BAA6B,EAAE;AAiBxC,MAAM,UAAU,wBAAwB,CACtC,YAA0B;IAE1B,OAAO,YAAY,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC,CAAC;AAC9D,CAAC;AAED,MAAM,UAAU,mBAAmB,CAA4C,EAC7E,KAAK,EACL,OAAO,EACP,YAAY,EACZ,cAAc,EACd,eAAe,GAAG,IAAI,GACqB;IAI3C,MAAM,EAAE,KAAK,EAAE,aAAa,EAAE,aAAa,EAAE,qBAAqB,EAAE,GAClE,mBAAmB,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;IAC7C,gFAAgF;IAChF,mEAAmE;IACnE,MAAM,iBAAiB,GAAG,wBAAwB,CAChD,YAAY,IAAI,KAAK,CAAC,YAAY,CACnC,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CACnB,UAAU;SACP,IAAI,CAAC,OAAO,CAAC;SACb,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC;SACtB,SAAS,CAAC,aAAa,EAAE,qBAAqB,CAAC,CACnD,CAAC;IACF,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAClC,iBAAiB,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,CACvD,CAAC;IACF,MAAM,eAAe,GAAG,eAAe;QACrC,CAAC,CAAC,iBAAiB,CAAC,MAAM,CACtB,CAAC,QAAQ,EAAE,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,eAAe,CAAC,QAAQ,CAAC,EAC9D,eAAe,CAAC,KAAK,EAAE,OAAO,CAAC,CAChC;QACH,CAAC,CAAC,eAAe,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAEpC,KAAK,MAAM,UAAU,IAAI,iBAAiB,EAAE,CAAC;QAC3C,MAAM,YAAY,GAAG,CAAC,GAAG,UAAU,CAAC,uBAAuB,EAAE,CAAC,CAAC,MAAM,CACnE,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC,oBAAoB,CAAC,GAAG,CAAC,YAAY,CAAC,CAC1D,CAAC;QACF,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,SAAS,CACjB,GAAG,UAAU,CAAC,IAAI,mCAAmC,YAAY,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACtF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,MAAM,YAAY,GAAG,iBAAiB,CAAC,MAAM,CAC3C,CAAC,cAAc,EAAE,UAAU,EAAE,EAAE;IAC7B,4EAA4E;IAC5E,2EAA2E;IAC3E,wBAAwB,CACtB,cAAc,EACd,UAAU,CAAC,cAAc,CAAC;QACxB,KAAK,EAAE,aAAa;QACpB,GAAG,CAAC,qBAAqB;YACvB,CAAC,CAAC,EAAE,aAAa,EAAE,qBAAqB,EAAE;YAC1C,CAAC,CAAC,EAAE,CAAC;KACR,CAAC,CACH,EACH,EAAE,CACH,CAAC;IACF,MAAM,KAAK,GAAG;QACZ,GAAG,KAAK,CAAC,KAAK;QACd,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;KACjE,CAAC;IAEF,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC;QAC3B,YAAY,EAAE,iBAAiB;QAC/B,KAAK;QACL,aAAa,EAAE;YACb,GAAG,KAAK,CAAC,aAAa;YACtB,YAAY,EAAE,wBAAwB,CACpC,KAAK,CAAC,aAAa,CAAC,YAAmD,EACvE,YAAY,CACb;SACF;QACD,YAAY,EAAE,KAAK,EAAE,UAAU,EAAE,aAAa,EAAE,EAAE;YAChD,MAAM,sBAAsB,GAAG,MAAM,OAAO,CAAC,GAAG,CAC9C,iBAAiB,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CACnC,UAAU,CAAC,YAAY,CAAC,eAAe,CAAC,CACzC,CACF,CAAC;YACF,MAAM,QAAQ,GAAa,EAAE,CAAC;YAC9B,MAAM,gBAAgB,GAAG,MAAM,sBAAsB,CACnD,KAAK,EACL,UAAU,EACV,aAAa,CACd,CAAC;YACF,IAAI,gBAAgB,EAAE,CAAC;gBACrB,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAClC,CAAC;YAED,MAAM,iBAAiB,GAAG,MAAM,uBAAuB,CACrD,KAAK,CAAC,YAAY,EAClB,UAAU,EACV,aAAa,CACd,CAAC;YACF,IAAI,iBAAiB,EAAE,CAAC;gBACtB,QAAQ,CAAC,IAAI,CACX,wBAAwB,CAAC,oBAAoB,EAAE,iBAAiB,CAAC,CAClE,CAAC;YACJ,CAAC;YAED,MAAM,mBAAmB,GAAG,sBAAsB,CAAC,MAAM,CACvD,CAAC,QAAQ,EAAsB,EAAE,CAAC,OAAO,CAAC,QAAQ,CAAC,CACpD,CAAC;YACF,IAAI,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACnC,QAAQ,CAAC,IAAI,CACX,wBAAwB,CACtB,iCAAiC,EACjC,mBAAmB,CAAC,IAAI,CAAC,MAAM,CAAC,CACjC,CACF,CAAC;YACJ,CAAC;YAED,MAAM,iBAAiB,GACrB,mCAAmC,CAAC,eAAe,CAAC,CAAC;YACvD,IAAI,iBAAiB,EAAE,CAAC;gBACtB,QAAQ,CAAC,IAAI,CACX,wBAAwB,CACtB,6BAA6B,EAC7B,iBAAiB,CAClB,CACF,CAAC;YACJ,CAAC;YAED,QAAQ,CAAC,IAAI,CAAC,4BAA4B,CAAC,eAAe,CAAC,CAAC,CAAC;YAE7D,OAAO,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC/B,CAAC;KACF,CAAC,CAAC;IAEH,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,qBAAqB,EAAE,EAAE,CAAC;QACzD,oFAAoF;QACpF,8EAA8E;QAE5E,QAGD,CAAC,0BAA0B,GAAG,KAAK,CAAC;IACvC,CAAC;IAED,QAAQ,CAAC,eAAe,GAAG,eAAe,CAAC;IAC3C,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,eAAe,CACtB,KAAsC,EACtC,OAAgD;IAEhD,MAAM,iBAAiB,GACrB,OAAO,CAAC,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,eAAe,IAAI,IAAI,QAAQ,EAAE,CAAC;IAEpE,OAAO,qBAAqB,CAAC,aAAa,CAAC,iBAAiB,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,mBAAmB,CAC1B,KAAsC,EACtC,cAAoC;IAEpC,IAAI,0BAAsD,CAAC;IAC3D,IAAI,sBAAyC,CAAC;IAC9C,IAAI,6BAA6B,CAAC,cAAc,CAAC,EAAE,CAAC;QAClD,0BAA0B,GAAG,cAAc,CAAC,KAAK,CAAC;QAClD,sBAAsB,GAAG,cAAc,CAAC,aAAa,CAAC;IACxD,CAAC;SAAM,CAAC;QACN,0BAA0B,GAAG,cAAc,CAAC;QAC5C,sBAAsB;YACpB,cAAc,IAAI,OAAO,cAAc,KAAK,QAAQ;gBAClD,CAAC,CAAC,cAAc;gBAChB,CAAC,CAAC,SAAS,CAAC;IAClB,CAAC;IACD,MAAM,aAAa,GAAG,WAAW,CAAC,KAAK,CAAC,KAAK,EAAE,0BAA0B,CAAC,CAAC;IAC3E,IAAI,OAAO,aAAa,KAAK,QAAQ,IAAI,aAAa,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzE,OAAO;YACL,KAAK,EAAE,aAAa;YACpB,GAAG,CAAC,aAAa,KAAK,0BAA0B,IAAI,sBAAsB;gBACxE,CAAC,CAAC,EAAE,aAAa,EAAE,sBAAsB,EAAE;gBAC3C,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;IACJ,CAAC;IACD,IAAI,aAAa,IAAI,OAAO,aAAa,KAAK,QAAQ,EAAE,CAAC;QACvD,OAAO;YACL,KAAK,EAAE,0BAA0B,CAAC,aAAa,CAAC,IAAI,eAAe,EAAE;YACrE,aAAa,EAAE,aAAa;SAC7B,CAAC;IACJ,CAAC;IAED,OAAO;QACL,KAAK,EAAE,eAAe,EAAE;QACxB,GAAG,CAAC,sBAAsB;YACxB,CAAC,CAAC,EAAE,aAAa,EAAE,sBAAsB,EAAE;YAC3C,CAAC,CAAC,EAAE,CAAC;KACR,CAAC;AACJ,CAAC;AAED,SAAS,6BAA6B,CACpC,KAAsC;IAEtC,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,KAAK,IAAI;QACd,OAAO,IAAI,KAAK;QAChB,OAAQ,KAA6B,CAAC,KAAK,KAAK,QAAQ;QACxD,eAAe,IAAI,KAAK,CACzB,CAAC;AACJ,CAAC;AAED,SAAS,0BAA0B,CAAC,KAAY;IAC9C,MAAM,SAAS,GACZ,KAAuD,EAAE,KAAK;QAC9D,KAAuC,EAAE,IAAI,CAAC;IACjD,OAAO,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC;QACjE,CAAC,CAAC,SAAS;QACX,CAAC,CAAC,SAAS,CAAC;AAChB,CAAC;AAED,KAAK,UAAU,uBAAuB,CAIpC,YAA6D,EAC7D,UAAgC,EAChC,KAAyD;IAEzD,IAAI,OAAO,YAAY,KAAK,UAAU,EAAE,CAAC;QACvC,OAAO,MAAM,YAAY,CACvB,UAAU,EACV,KAAwC,CACzC,CAAC;IACJ,CAAC;IAED,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,KAAK,UAAU,sBAAsB,CAInC,KAAsC,EACtC,UAAgC,EAChC,aAAiE;IAEjE,MAAM,gBAAgB,GACpB,KAAK,CAAC,gBAAgB,IAAI,6BAA6B,EAAE,CAAC;IAC5D,IAAI,OAAO,gBAAgB,KAAK,UAAU,EAAE,CAAC;QAC3C,OAAO,MAAM,gBAAgB,CAC3B,UAAU,EACV,aAAgD,CACjD,CAAC;IACJ,CAAC;IAED,OAAO,gBAAgB,CAAC;AAC1B,CAAC;AAED,SAAS,wBAAwB,CAC/B,IAAyC,EACzC,MAA+B;IAE/B,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,iBAAiB,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;IACvC,CAAC;IAED,OAAO,iBAAiB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;AACzC,CAAC;AAED,SAAS,iBAAiB,CACxB,IAA6B,EAC7B,MAA+B;IAE/B,MAAM,MAAM,GAA4B,EAAE,GAAG,IAAI,EAAE,CAAC;IAEpD,KAAK,MAAM,CAAC,GAAG,EAAE,WAAW,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QACxD,MAAM,SAAS,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QAC9B,IAAI,aAAa,CAAC,SAAS,CAAC,IAAI,aAAa,CAAC,WAAW,CAAC,EAAE,CAAC;YAC3D,MAAM,CAAC,GAAG,CAAC,GAAG,iBAAiB,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;YACxD,SAAS;QACX,CAAC;QAED,MAAM,CAAC,GAAG,CAAC,GAAG,WAAW,CAAC;IAC5B,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACxE,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,SAAS,GAAG,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;IAC/C,OAAO,SAAS,KAAK,MAAM,CAAC,SAAS,IAAI,SAAS,KAAK,IAAI,CAAC;AAC9D,CAAC"}

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

@@ -0,0 +1,5 @@
+export * from './agentPreparation';
+export * from './manager';
+export * from './runAsManifest';
+export * from './spans';
+export * from './toolRehydration';

package/dist/sandbox/runtime/index.js

@@ -0,0 +1,22 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./agentPreparation.js"), exports);
+__exportStar(require("./manager.js"), exports);
+__exportStar(require("./runAsManifest.js"), exports);
+__exportStar(require("./spans.js"), exports);
+__exportStar(require("./toolRehydration.js"), exports);
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/sandbox/runtime/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,wDAAmC;AACnC,+CAA0B;AAC1B,qDAAgC;AAChC,6CAAwB;AACxB,uDAAkC"}

package/dist/sandbox/runtime/index.mjs

@@ -0,0 +1,6 @@
+export * from "./agentPreparation.mjs";
+export * from "./manager.mjs";
+export * from "./runAsManifest.mjs";
+export * from "./spans.mjs";
+export * from "./toolRehydration.mjs";
+//# sourceMappingURL=index.mjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.mjs","sourceRoot":"","sources":["../../../src/sandbox/runtime/index.ts"],"names":[],"mappings":""}

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

@@ -0,0 +1,25 @@
+import type { Agent, AgentOutputType } from '../../agent';
+import type { RunState } from '../../runState';
+import type { SandboxClient } from '../client';
+import type { SandboxSessionLike, SandboxSessionState } from '../session';
+import type { SerializedSandboxSessionEntry, SerializedSandboxState } from './sessionState';
+export type LivePreservedOwnedSessionEntry = {
+    agentKey: string;
+    backendId: string;
+    currentAgentName: string;
+    session: SandboxSessionLike<SandboxSessionState>;
+};
+export declare function rememberLivePreservedOwnedSessions<TContext>(args: {
+    state: RunState<TContext, Agent<TContext, AgentOutputType>>;
+    serializedState: SerializedSandboxState;
+    sessionsByAgentKey: ReadonlyMap<string, SandboxSessionLike<SandboxSessionState>>;
+}): void;
+export declare function forgetLivePreservedOwnedSessions<TContext>(state: RunState<TContext, Agent<TContext, AgentOutputType>>): void;
+export declare function livePreservedOwnedSessionEntries<TContext>(state: RunState<TContext, Agent<TContext, AgentOutputType>>): LivePreservedOwnedSessionEntry[];
+export declare function preservedOwnedSessionAgentKeysWithoutLiveReuse(serializedState: SerializedSandboxState): Set<string>;
+export declare function livePreservedOwnedSession<TContext>(args: {
+    runState: RunState<TContext, Agent<TContext, AgentOutputType>> | undefined;
+    client: SandboxClient;
+    agentKey: string;
+    serializedEntry: SerializedSandboxSessionEntry | undefined;
+}): LivePreservedOwnedSessionEntry | undefined;

package/dist/sandbox/runtime/livePreservedSessions.js

@@ -0,0 +1,58 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.rememberLivePreservedOwnedSessions = rememberLivePreservedOwnedSessions;
+exports.forgetLivePreservedOwnedSessions = forgetLivePreservedOwnedSessions;
+exports.livePreservedOwnedSessionEntries = livePreservedOwnedSessionEntries;
+exports.preservedOwnedSessionAgentKeysWithoutLiveReuse = preservedOwnedSessionAgentKeysWithoutLiveReuse;
+exports.livePreservedOwnedSession = livePreservedOwnedSession;
+const livePreservedOwnedSessionsByRunState = new WeakMap();
+function rememberLivePreservedOwnedSessions(args) {
+    const liveSessions = new Map();
+    for (const [agentKey, entry] of Object.entries(args.serializedState.sessionsByAgent ?? {})) {
+        if (!entry.preservedOwnedSession) {
+            continue;
+        }
+        if (entry.reuseLiveSession === false) {
+            continue;
+        }
+        const session = args.sessionsByAgentKey.get(agentKey);
+        if (!session) {
+            continue;
+        }
+        liveSessions.set(agentKey, {
+            agentKey,
+            backendId: entry.backendId,
+            currentAgentName: entry.currentAgentName,
+            session,
+        });
+    }
+    if (liveSessions.size > 0) {
+        livePreservedOwnedSessionsByRunState.set(args.state, liveSessions);
+    }
+    else {
+        livePreservedOwnedSessionsByRunState.delete(args.state);
+    }
+}
+function forgetLivePreservedOwnedSessions(state) {
+    livePreservedOwnedSessionsByRunState.delete(state);
+}
+function livePreservedOwnedSessionEntries(state) {
+    return [...(livePreservedOwnedSessionsByRunState.get(state)?.values() ?? [])];
+}
+function preservedOwnedSessionAgentKeysWithoutLiveReuse(serializedState) {
+    return new Set(Object.entries(serializedState.sessionsByAgent ?? {})
+        .filter(([, entry]) => entry.preservedOwnedSession && entry.reuseLiveSession === false)
+        .map(([agentKey]) => agentKey));
+}
+function livePreservedOwnedSession(args) {
+    if (!args.serializedEntry?.preservedOwnedSession || !args.runState) {
+        return undefined;
+    }
+    const liveSessions = livePreservedOwnedSessionsByRunState.get(args.runState);
+    const liveEntry = liveSessions?.get(args.agentKey);
+    if (!liveEntry || liveEntry.backendId !== args.client.backendId) {
+        return undefined;
+    }
+    return liveEntry;
+}
+//# sourceMappingURL=livePreservedSessions.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"livePreservedSessions.js","sourceRoot":"","sources":["../../../src/sandbox/runtime/livePreservedSessions.ts"],"names":[],"mappings":";;AAqBA,gFAmCC;AAED,4EAIC;AAED,4EAIC;AAED,wGAWC;AAED,8DAeC;AAlFD,MAAM,oCAAoC,GAAG,IAAI,OAAO,EAGrD,CAAC;AAEJ,SAAgB,kCAAkC,CAAW,IAO5D;IACC,MAAM,YAAY,GAAG,IAAI,GAAG,EAA0C,CAAC;IACvE,KAAK,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAC5C,IAAI,CAAC,eAAe,CAAC,eAAe,IAAI,EAAE,CAC3C,EAAE,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,qBAAqB,EAAE,CAAC;YACjC,SAAS;QACX,CAAC;QACD,IAAI,KAAK,CAAC,gBAAgB,KAAK,KAAK,EAAE,CAAC;YACrC,SAAS;QACX,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACtD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,SAAS;QACX,CAAC;QACD,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE;YACzB,QAAQ;YACR,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;YACxC,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAED,IAAI,YAAY,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;QAC1B,oCAAoC,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;IACrE,CAAC;SAAM,CAAC;QACN,oCAAoC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1D,CAAC;AACH,CAAC;AAED,SAAgB,gCAAgC,CAC9C,KAA2D;IAE3D,oCAAoC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AACrD,CAAC;AAED,SAAgB,gCAAgC,CAC9C,KAA2D;IAE3D,OAAO,CAAC,GAAG,CAAC,oCAAoC,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AAChF,CAAC;AAED,SAAgB,8CAA8C,CAC5D,eAAuC;IAEvC,OAAO,IAAI,GAAG,CACZ,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,eAAe,IAAI,EAAE,CAAC;SAClD,MAAM,CACL,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CACZ,KAAK,CAAC,qBAAqB,IAAI,KAAK,CAAC,gBAAgB,KAAK,KAAK,CAClE;SACA,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,CACjC,CAAC;AACJ,CAAC;AAED,SAAgB,yBAAyB,CAAW,IAKnD;IACC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,qBAAqB,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACnE,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,YAAY,GAAG,oCAAoC,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC7E,MAAM,SAAS,GAAG,YAAY,EAAE,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACnD,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,SAAS,KAAK,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;QAChE,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC"}

package/dist/sandbox/runtime/livePreservedSessions.mjs

@@ -0,0 +1,51 @@
+const livePreservedOwnedSessionsByRunState = new WeakMap();
+export function rememberLivePreservedOwnedSessions(args) {
+    const liveSessions = new Map();
+    for (const [agentKey, entry] of Object.entries(args.serializedState.sessionsByAgent ?? {})) {
+        if (!entry.preservedOwnedSession) {
+            continue;
+        }
+        if (entry.reuseLiveSession === false) {
+            continue;
+        }
+        const session = args.sessionsByAgentKey.get(agentKey);
+        if (!session) {
+            continue;
+        }
+        liveSessions.set(agentKey, {
+            agentKey,
+            backendId: entry.backendId,
+            currentAgentName: entry.currentAgentName,
+            session,
+        });
+    }
+    if (liveSessions.size > 0) {
+        livePreservedOwnedSessionsByRunState.set(args.state, liveSessions);
+    }
+    else {
+        livePreservedOwnedSessionsByRunState.delete(args.state);
+    }
+}
+export function forgetLivePreservedOwnedSessions(state) {
+    livePreservedOwnedSessionsByRunState.delete(state);
+}
+export function livePreservedOwnedSessionEntries(state) {
+    return [...(livePreservedOwnedSessionsByRunState.get(state)?.values() ?? [])];
+}
+export function preservedOwnedSessionAgentKeysWithoutLiveReuse(serializedState) {
+    return new Set(Object.entries(serializedState.sessionsByAgent ?? {})
+        .filter(([, entry]) => entry.preservedOwnedSession && entry.reuseLiveSession === false)
+        .map(([agentKey]) => agentKey));
+}
+export function livePreservedOwnedSession(args) {
+    if (!args.serializedEntry?.preservedOwnedSession || !args.runState) {
+        return undefined;
+    }
+    const liveSessions = livePreservedOwnedSessionsByRunState.get(args.runState);
+    const liveEntry = liveSessions?.get(args.agentKey);
+    if (!liveEntry || liveEntry.backendId !== args.client.backendId) {
+        return undefined;
+    }
+    return liveEntry;
+}
+//# sourceMappingURL=livePreservedSessions.mjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"livePreservedSessions.mjs","sourceRoot":"","sources":["../../../src/sandbox/runtime/livePreservedSessions.ts"],"names":[],"mappings":"AAgBA,MAAM,oCAAoC,GAAG,IAAI,OAAO,EAGrD,CAAC;AAEJ,MAAM,UAAU,kCAAkC,CAAW,IAO5D;IACC,MAAM,YAAY,GAAG,IAAI,GAAG,EAA0C,CAAC;IACvE,KAAK,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAC5C,IAAI,CAAC,eAAe,CAAC,eAAe,IAAI,EAAE,CAC3C,EAAE,CAAC;QACF,IAAI,CAAC,KAAK,CAAC,qBAAqB,EAAE,CAAC;YACjC,SAAS;QACX,CAAC;QACD,IAAI,KAAK,CAAC,gBAAgB,KAAK,KAAK,EAAE,CAAC;YACrC,SAAS;QACX,CAAC;QACD,MAAM,OAAO,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACtD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,SAAS;QACX,CAAC;QACD,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE;YACzB,QAAQ;YACR,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;YACxC,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAED,IAAI,YAAY,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;QAC1B,oCAAoC,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;IACrE,CAAC;SAAM,CAAC;QACN,oCAAoC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1D,CAAC;AACH,CAAC;AAED,MAAM,UAAU,gCAAgC,CAC9C,KAA2D;IAE3D,oCAAoC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AACrD,CAAC;AAED,MAAM,UAAU,gCAAgC,CAC9C,KAA2D;IAE3D,OAAO,CAAC,GAAG,CAAC,oCAAoC,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AAChF,CAAC;AAED,MAAM,UAAU,8CAA8C,CAC5D,eAAuC;IAEvC,OAAO,IAAI,GAAG,CACZ,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,eAAe,IAAI,EAAE,CAAC;SAClD,MAAM,CACL,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CACZ,KAAK,CAAC,qBAAqB,IAAI,KAAK,CAAC,gBAAgB,KAAK,KAAK,CAClE;SACA,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,CACjC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAW,IAKnD;IACC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,qBAAqB,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACnE,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,YAAY,GAAG,oCAAoC,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC7E,MAAM,SAAS,GAAG,YAAY,EAAE,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACnD,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,SAAS,KAAK,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;QAChE,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC"}

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

@@ -0,0 +1,68 @@
+import type { Agent, AgentOutputType } from '../../agent';
+import type { RunState } from '../../runState';
+import type { AgentInputItem } from '../../types';
+import type { SandboxRunConfig } from '../client';
+import { type SandboxMemoryAgentRunner } from '../memory/generation';
+import { type SandboxRuntimeModel } from './agentPreparation';
+type SandboxPreparedAgent<TContext> = {
+    executionAgent: Agent<TContext, AgentOutputType>;
+    turnInput: AgentInputItem[];
+};
+export declare class SandboxRuntimeManager<TContext> {
+    private readonly sandboxConfig?;
+    private readonly runState?;
+    private readonly agentKeys;
+    private readonly acquiredAgents;
+    private readonly sessionsByAgent;
+    private readonly sessionsByAgentKey;
+    private readonly sessionAgentNamesByKey;
+    private readonly preparedAgents;
+    private readonly preparedSessions;
+    private readonly preparedManifestSignatures;
+    private readonly ownedSessionAgentKeys;
+    private readonly sessionStartPromises;
+    private activeMemory?;
+    private currentAgentId?;
+    constructor(args: {
+        startingAgent: Agent<TContext, AgentOutputType>;
+        sandboxConfig?: SandboxRunConfig;
+        runState?: RunState<TContext, Agent<TContext, AgentOutputType>>;
+    });
+    prepareAgent(args: {
+        currentAgent: Agent<TContext, AgentOutputType>;
+        turnInput: AgentInputItem[];
+        runConfigModel?: SandboxRuntimeModel;
+    }): Promise<SandboxPreparedAgent<TContext>>;
+    cleanup(state: RunState<TContext, Agent<TContext, AgentOutputType>>, options?: {
+        preserveOwnedSessions?: boolean;
+    }): Promise<void>;
+    private planCleanup;
+    private planCleanupForActiveSessions;
+    private planCleanupForSerializedStateOnly;
+    private executeCleanupPlan;
+    private closeLivePreservedOwnedSessions;
+    private runPreStopHooksBeforeRelease;
+    enqueueMemoryGeneration(state: RunState<TContext, Agent<TContext, AgentOutputType>>, args: {
+        exception?: unknown;
+        groupId?: string;
+        inputOverride?: string | AgentInputItem[];
+        sdkSessionId?: () => Promise<string | undefined>;
+        runAgent: SandboxMemoryAgentRunner;
+    }): Promise<void>;
+    adoptPreservedOwnedSessions(): Promise<boolean>;
+    private acquireAgent;
+    private ensureSession;
+    private registerSessionForAgent;
+    private ensureSessionStarted;
+    private getPreparedAgent;
+    private resumeSessionForAgent;
+    private serializeState;
+    private requireClient;
+    private resolveSnapshotSpec;
+    private agentKey;
+    private allocateRuntimeAgentKey;
+    private resolveConfiguredManifest;
+    private closeOwnedSessions;
+    private releaseAgents;
+}
+export {};