@agentstep/agent-sdk 0.4.30 → 0.4.32

package/dist/{chunk-WHV4YUQB.js → chunk-USYY3L7G.js}

@@ -2,17 +2,17 @@ import {
   init_ids,
   newId
 } from "./chunk-F4WUVOLE.js";
+import {
+  init_clock,
+  nowMs
+} from "./chunk-HFDLUBWN.js";
 import {
   getDrizzle,
   init_drizzle
-} from "./chunk-SLAOS6JN.js";
+} from "./chunk-R5T4LJSK.js";
 import {
   schema_exports
-} from "./chunk-LHHBOQUR.js";
-import {
-  init_clock,
-  nowMs
-} from "./chunk-HFDLUBWN.js";
+} from "./chunk-ZMJ4EP4C.js";
 
 // src/db/api_keys.ts
 init_drizzle();

package/dist/{chunk-M36DF445.js → chunk-V2R7RWVY.js}

@@ -1,6 +1,6 @@
 import {
   getConfig
-} from "./chunk-EFDAVLKN.js";
+} from "./chunk-V7MTIMPB.js";
 
 // src/backends/gemini/auth.ts
 function buildGeminiAuthEnv() {

package/dist/{chunk-EFDAVLKN.js → chunk-V7MTIMPB.js}

@@ -1,14 +1,14 @@
+import {
+  init_clock,
+  nowMs
+} from "./chunk-HFDLUBWN.js";
 import {
   getDrizzle,
   init_drizzle
-} from "./chunk-SLAOS6JN.js";
+} from "./chunk-R5T4LJSK.js";
 import {
   schema_exports
-} from "./chunk-LHHBOQUR.js";
-import {
-  init_clock,
-  nowMs
-} from "./chunk-HFDLUBWN.js";
+} from "./chunk-ZMJ4EP4C.js";
 
 // src/config/index.ts
 init_drizzle();
@@ -53,7 +53,9 @@ function loadConfig() {
     ollamaUrl: process.env.OLLAMA_URL || readSetting("ollama_url") || "http://localhost:11434",
     anthropicPassthroughEnabled: parseBool(
       process.env.ANTHROPIC_PASSTHROUGH_ENABLED ?? readSetting("anthropic_passthrough_enabled")
-    )
+    ),
+    warmPoolSize: num(process.env.WARM_POOL_SIZE, 0),
+    warmPoolTtlMs: num(process.env.WARM_POOL_TTL_MS, 3e5)
   };
 }
 function parseBool(v) {

package/dist/{chunk-XUNTHJIE.js → chunk-WCDWL6ED.js}

@@ -459,6 +459,13 @@ function runMigrations(db) {
   if (envCols3.some((c) => c.name === "template_sprite")) {
     db.exec(`ALTER TABLE environments RENAME COLUMN template_sprite TO template_sandbox`);
   }
+  const sessColsQuota = db.prepare(`PRAGMA table_info(sessions)`).all();
+  if (!sessColsQuota.some((c) => c.name === "max_tokens")) {
+    db.exec(`ALTER TABLE sessions ADD COLUMN max_tokens INTEGER DEFAULT NULL`);
+  }
+  if (!sessColsQuota.some((c) => c.name === "max_wall_duration_ms")) {
+    db.exec(`ALTER TABLE sessions ADD COLUMN max_wall_duration_ms INTEGER DEFAULT NULL`);
+  }
 }
 var init_migrations = __esm({
   "src/db/migrations.ts"() {

package/dist/{chunk-QV5WI73T.js → chunk-WCNGMBIE.js}

@@ -12,15 +12,15 @@ import {
   listMemories,
   listMemoryStores,
   updateMemory
-} from "./chunk-7V277COF.js";
+} from "./chunk-SUGSHXND.js";
 import {
   jsonOk,
   routeWrap
-} from "./chunk-LOAFGJWC.js";
+} from "./chunk-3K4KIYEZ.js";
 import {
   getDb,
   init_client
-} from "./chunk-4C7XJI2V.js";
+} from "./chunk-6POQAFEC.js";
 import {
   badRequest,
   conflict,

package/dist/{chunk-FGQFNIRI.js → chunk-WQARLGBG.js}

@@ -1,6 +1,6 @@
 import {
   forwardToAnthropic
-} from "./chunk-L43UD7PK.js";
+} from "./chunk-JCW3ZRES.js";
 import {
   isAnthropicApiKey,
   isPassthroughAllowedPath
@@ -8,10 +8,10 @@ import {
 import {
   findByRawKey,
   hydratePermissions
-} from "./chunk-WHV4YUQB.js";
+} from "./chunk-USYY3L7G.js";
 import {
   getConfig
-} from "./chunk-EFDAVLKN.js";
+} from "./chunk-V7MTIMPB.js";
 import {
   unauthorized
 } from "./chunk-EZYKRG4W.js";

package/dist/{chunk-EN3ICYTL.js → chunk-WQHRP3JL.js}

@@ -1,21 +1,21 @@
-import {
-  resolveVaultSecrets
-} from "./chunk-IFXD4DH5.js";
 import {
   dockerProvider
-} from "./chunk-RMZRSYIJ.js";
+} from "./chunk-4XXQAVKE.js";
+import {
+  resolveVaultSecrets
+} from "./chunk-5IGBMS2U.js";
 import {
   appendEvent
-} from "./chunk-TRQX5AIZ.js";
+} from "./chunk-QHBXWPSO.js";
 import {
   getSession,
   getSessionRow,
   setSessionSandbox
-} from "./chunk-QWGJS6V6.js";
+} from "./chunk-2CPZFRJJ.js";
 import {
   deleteSprite,
   listSprites
-} from "./chunk-REGPK5NG.js";
+} from "./chunk-FXLUSECC.js";
 import {
   allSessionSandboxes,
   countInEnv,
@@ -24,20 +24,29 @@ import {
 } from "./chunk-EFOIR7R3.js";
 import {
   resolveContainerProvider
-} from "./chunk-C6JEPQ7T.js";
+} from "./chunk-WPK4ZPMG.js";
 import {
   getEnvironment,
-  getEnvironmentRow
-} from "./chunk-S5HVHCAK.js";
+  getEnvironmentRow,
+  listEnvironments
+} from "./chunk-X6IQ57SC.js";
+import {
+  addWarm,
+  claimWarm,
+  countInflight,
+  countWarm,
+  decrementInflight,
+  incrementInflight
+} from "./chunk-G7KUVNDY.js";
 import {
   getAgent
-} from "./chunk-TXUUUO5R.js";
+} from "./chunk-ZTH5JRZG.js";
 import {
   resolveBackend
-} from "./chunk-4B7PQTTS.js";
+} from "./chunk-EUIR2ADY.js";
 import {
   getConfig
-} from "./chunk-EFDAVLKN.js";
+} from "./chunk-V7MTIMPB.js";
 import {
   init_clock,
   nowMs
@@ -48,6 +57,7 @@ import {
 
 // src/containers/lifecycle.ts
 init_clock();
+import { ulid } from "ulid";
 var lcLog = (...args) => {
   if (process.env.DEBUG_LIFECYCLE === "1") console.log(...args);
 };
@@ -94,7 +104,8 @@ async function installSkills(sandboxName, provider, skills, engine) {
   }
 }
 function deriveSandboxName(sessionId) {
-  return `${SANDBOX_NAME_PREFIX}${sessionId.replace(/^sess_/, "").toLowerCase()}`;
+  const tail = sessionId.replace(/^[a-z]+_/, "").toLowerCase();
+  return `${SANDBOX_NAME_PREFIX}${tail}`;
 }
 async function acquireForFirstTurn(sessionId) {
   const row = getSessionRow(sessionId);
@@ -109,7 +120,8 @@ async function acquireForFirstTurn(sessionId) {
       `environment is not ready (state=${env.state})`
     );
   }
-  if (countInEnv(env.id) >= getConfig().maxSandboxesPerEnv) {
+  const config = getConfig();
+  if (countInEnv(env.id) + countWarm(env.id) >= config.maxSandboxesPerEnv) {
     throw new ApiError(503, "server_busy", "env sandbox pool exhausted");
   }
   const agent = getAgent(row.agent_id, row.agent_version);
@@ -128,6 +140,84 @@ async function acquireForFirstTurn(sessionId) {
     }
   }
   const sp = wrapProviderWithSecrets(provider, secrets);
+  const warm = claimWarm(env.id, backend.name, provider.name);
+  if (warm) {
+    lcLog(`[lifecycle] ${sessionId} claimed warm container: ${warm.sandboxName}`);
+    const name2 = warm.sandboxName;
+    try {
+      if (agent.engine === "claude") {
+        const customTools = agent.tools.filter(
+          (t) => t.type === "custom"
+        );
+        const allBridgeTools = [...customTools];
+        if (agent.threads_enabled) {
+          allBridgeTools.push({
+            type: "custom",
+            name: "spawn_agent",
+            description: "Spawn a sub-agent to handle a task. Returns the sub-agent's response.",
+            input_schema: {
+              type: "object",
+              properties: {
+                agent_id: { type: "string", description: "ID of the agent to spawn" },
+                prompt: { type: "string", description: "Task for the sub-agent" }
+              },
+              required: ["agent_id", "prompt"]
+            }
+          });
+        }
+        if (allBridgeTools.length > 0) {
+          const { installToolBridge } = await import("./backends/claude/index.js");
+          await installToolBridge(name2, allBridgeTools, sp);
+        }
+        if (agent.confirmation_mode) {
+          const { installPermissionHook } = await import("./backends/claude/index.js");
+          await installPermissionHook(name2, sp);
+        }
+      }
+      if (agent.skills && agent.skills.length > 0) {
+        lcLog(`[lifecycle] ${sessionId} installing ${agent.skills.length} skill(s) on warm container...`);
+        await installSkills(name2, sp, agent.skills, agent.engine);
+      }
+    } catch (err) {
+      await sp.delete(name2).catch(() => {
+      });
+      throw err;
+    }
+    const { listResources: listSessionResources2 } = await import("./db/session-resources.js");
+    const tableResources2 = listSessionResources2(sessionId);
+    if (tableResources2.length > 0) {
+      const mapped = tableResources2.map((r) => {
+        if (r.type === "file") return { type: "file", file_id: r.file_id, mount_path: r.mount_path };
+        if (r.type === "github_repository") {
+          const checkout = r.checkout;
+          return {
+            type: "github_repository",
+            repository_url: r.url,
+            mount_path: r.mount_path,
+            branch: checkout?.type === "branch" ? checkout.name : void 0,
+            commit: checkout?.type === "commit" ? checkout.name : void 0
+          };
+        }
+        return { type: "uri", uri: r.url, mount_path: r.mount_path };
+      });
+      await provisionResources(name2, mapped, sp);
+    } else {
+      const session = getSession(sessionId);
+      if (session?.resources && session.resources.length > 0) {
+        await provisionResources(name2, session.resources, sp);
+      }
+    }
+    register({
+      sandboxName: name2,
+      envId: env.id,
+      sessionId,
+      createdAt: nowMs(),
+      vaultSecrets: Object.keys(secrets).length > 0 ? secrets : void 0
+    });
+    setSessionSandbox(sessionId, name2);
+    void replenishWarmPool(env.id, backend.name, provider, secrets);
+    return name2;
+  }
   const name = deriveSandboxName(sessionId);
   lcLog(`[lifecycle] ${sessionId} creating container via ${sp.name}...`);
   try {
@@ -234,6 +324,80 @@ async function acquireForFirstTurn(sessionId) {
   setSessionSandbox(sessionId, name);
   return name;
 }
+function effectiveWarmPoolSize(envId) {
+  const cfg = getConfig();
+  const envObj = getEnvironment(envId);
+  return envObj?.config?.warm_pool_size ?? cfg.warmPoolSize;
+}
+async function replenishWarmPool(envId, engine, provider, secrets) {
+  if (!provider.supportsWarmPool) return;
+  const target = effectiveWarmPoolSize(envId);
+  if (target <= 0) return;
+  if (countWarm(envId) + countInflight(envId) >= target) return;
+  const cfg = getConfig();
+  if (countInEnv(envId) + countWarm(envId) + countInflight(envId) >= cfg.maxSandboxesPerEnv) return;
+  incrementInflight(envId);
+  try {
+    const sp = wrapProviderWithSecrets(provider, secrets);
+    const name = `ca-warm-${ulid().toLowerCase()}`;
+    lcLog(`[warm-pool] creating warm container for env=${envId} engine=${engine}: ${name}`);
+    await sp.create({ name });
+    const backend = resolveBackend(engine);
+    await backend.prepareOnSandbox(name, sp);
+    const cfg2 = getConfig();
+    addWarm({
+      sandboxName: name,
+      envId,
+      engine,
+      provider: provider.name,
+      createdAt: nowMs(),
+      expiresAt: nowMs() + cfg2.warmPoolTtlMs,
+      vaultSecrets: Object.keys(secrets).length > 0 ? secrets : void 0
+    });
+    lcLog(`[warm-pool] warm container ready: ${name} (env=${envId})`);
+  } catch (err) {
+    console.warn(`[warm-pool] replenish failed for env=${envId} engine=${engine}:`, err);
+  } finally {
+    decrementInflight(envId);
+  }
+}
+async function fillWarmPools() {
+  const cfg = getConfig();
+  if (cfg.warmPoolSize <= 0) return;
+  const envs = listEnvironments({ includeArchived: false, limit: 100 });
+  const readyEnvs = envs.filter((e) => e.state === "ready");
+  if (readyEnvs.length === 0) return;
+  const CONCURRENCY = 3;
+  let active = 0;
+  let idx = 0;
+  async function processNext() {
+    while (idx < readyEnvs.length) {
+      if (active >= CONCURRENCY) {
+        await new Promise((r) => setTimeout(r, 0));
+        continue;
+      }
+      const env = readyEnvs[idx++];
+      active++;
+      void fillOneEnv(env).finally(() => active--);
+    }
+  }
+  void processNext();
+}
+async function fillOneEnv(env) {
+  const target = effectiveWarmPoolSize(env.id);
+  if (target <= 0) return;
+  try {
+    const provider = await resolveContainerProvider(env.config?.provider);
+    if (!provider.supportsWarmPool) return;
+    const backend = resolveBackend("claude");
+    const needed = target - countWarm(env.id) - countInflight(env.id);
+    for (let i = 0; i < needed; i++) {
+      void replenishWarmPool(env.id, backend.name, provider, {});
+    }
+  } catch (err) {
+    console.warn(`[warm-pool] fillOneEnv failed for env=${env.id}:`, err);
+  }
+}
 async function provisionResources(sandboxName, resources, provider) {
   await provider.exec(sandboxName, ["mkdir", "-p", "/mnt/session/resources", "/mnt/session/uploads", "/mnt/session/outputs"]);
   const MAX_RESOURCE_BYTES = 50 * 1024 * 1024;
@@ -405,6 +569,8 @@ export {
   wrapProviderWithSecrets,
   installSkills,
   acquireForFirstTurn,
+  replenishWarmPool,
+  fillWarmPools,
   provisionResources,
   releaseSession,
   reconcileOrphanSandboxes,

package/dist/{chunk-S5HVHCAK.js → chunk-X6IQ57SC.js}

@@ -1,22 +1,22 @@
 import {
   DEFAULT_TENANT_ID
-} from "./chunk-BBYQMNGK.js";
+} from "./chunk-AR2TM7CR.js";
 import {
   init_ids,
   newId
 } from "./chunk-F4WUVOLE.js";
-import {
-  getDrizzle,
-  init_drizzle
-} from "./chunk-SLAOS6JN.js";
-import {
-  schema_exports
-} from "./chunk-LHHBOQUR.js";
 import {
   init_clock,
   nowMs,
   toIso
 } from "./chunk-HFDLUBWN.js";
+import {
+  getDrizzle,
+  init_drizzle
+} from "./chunk-R5T4LJSK.js";
+import {
+  schema_exports
+} from "./chunk-ZMJ4EP4C.js";
 
 // src/db/environments.ts
 init_drizzle();

package/dist/{chunk-WP5EWNGJ.js → chunk-X7VVDJ6T.js}

@@ -1,17 +1,17 @@
 import {
   listTraces
-} from "./chunk-SXTHM67U.js";
+} from "./chunk-6GP5IKXE.js";
 import {
   jsonOk,
   routeWrap
-} from "./chunk-LOAFGJWC.js";
+} from "./chunk-3K4KIYEZ.js";
 import {
   exportTrace
-} from "./chunk-F3BVYYRX.js";
+} from "./chunk-DYK2WTYW.js";
 import {
   listEventsByTrace,
   rowToManagedEvent
-} from "./chunk-5NBZZZKH.js";
+} from "./chunk-TH7WJLZC.js";
 import {
   badRequest,
   notFound

package/dist/{chunk-UF3NB27E.js → chunk-XMTKYR64.js}

@@ -1,32 +1,32 @@
 import {
   resolveRemoteSessionId
-} from "./chunk-YIOKUA23.js";
+} from "./chunk-L2RX552S.js";
 import {
   assertResourceTenant
 } from "./chunk-23UKWXJH.js";
 import {
   authenticateAndIntercept
-} from "./chunk-FGQFNIRI.js";
+} from "./chunk-WQARLGBG.js";
 import {
   forwardToAnthropic
-} from "./chunk-L43UD7PK.js";
+} from "./chunk-JCW3ZRES.js";
 import {
   ensureInitialized
-} from "./chunk-LR4TA74O.js";
+} from "./chunk-2VED3F37.js";
 import {
   getProxiedTenantId,
   isProxied
-} from "./chunk-NT2AI5IE.js";
+} from "./chunk-E7DD7F7J.js";
 import {
   subscribe
-} from "./chunk-TRQX5AIZ.js";
+} from "./chunk-QHBXWPSO.js";
 import {
   getSession
-} from "./chunk-QWGJS6V6.js";
+} from "./chunk-2CPZFRJJ.js";
 import {
   getDb,
   init_client
-} from "./chunk-4C7XJI2V.js";
+} from "./chunk-6POQAFEC.js";
 import {
   notFound,
   toResponse

package/dist/{chunk-7UJ2PIXJ.js → chunk-YD3T4PBA.js}

@@ -1,11 +1,11 @@
 import {
   jsonOk,
   routeWrap
-} from "./chunk-LOAFGJWC.js";
+} from "./chunk-3K4KIYEZ.js";
 import {
   readSetting,
   writeSetting
-} from "./chunk-EFDAVLKN.js";
+} from "./chunk-V7MTIMPB.js";
 import {
   badRequest
 } from "./chunk-EZYKRG4W.js";

package/dist/{chunk-EFCCO74P.js → chunk-YDZ7PCQO.js}

@@ -5,15 +5,15 @@ import {
 import {
   jsonOk,
   routeWrap
-} from "./chunk-LOAFGJWC.js";
+} from "./chunk-3K4KIYEZ.js";
 import {
   getSession,
   listSessions
-} from "./chunk-QWGJS6V6.js";
+} from "./chunk-2CPZFRJJ.js";
 import {
   getDb,
   init_client
-} from "./chunk-4C7XJI2V.js";
+} from "./chunk-6POQAFEC.js";
 import {
   notFound
 } from "./chunk-EZYKRG4W.js";

package/dist/{chunk-JN3G4Y4L.js → chunk-YNOONYPO.js}

@@ -1,16 +1,16 @@
 import {
   disableUpstreamKey,
   selectNextUpstreamKey
-} from "./chunk-YGTC5UP5.js";
+} from "./chunk-H7UKW666.js";
 import {
   listEntries
-} from "./chunk-7GNSB22P.js";
+} from "./chunk-ZVXIZ2JD.js";
 import {
   getSession
-} from "./chunk-QWGJS6V6.js";
+} from "./chunk-2CPZFRJJ.js";
 import {
   getConfig
-} from "./chunk-EFDAVLKN.js";
+} from "./chunk-V7MTIMPB.js";
 
 // src/providers/upstream-keys.ts
 var CONSECUTIVE_FAIL_THRESHOLD = 3;

package/dist/{chunk-H7Z4BWRI.js → chunk-YVHFSCAX.js}

@@ -4,11 +4,11 @@ import {
   getSources,
   getStats,
   searchSkills
-} from "./chunk-H3O7QBOQ.js";
+} from "./chunk-P3PHXVMA.js";
 import {
   jsonOk,
   routeWrap
-} from "./chunk-LOAFGJWC.js";
+} from "./chunk-3K4KIYEZ.js";
 
 // src/handlers/skills.ts
 async function handleGetSkillsCatalog(request) {

package/dist/{chunk-3KLJGU75.js → chunk-YYEDFQJY.js}

@@ -1,10 +1,10 @@
 import {
   syncAndCreateSession
-} from "./chunk-46XURNM7.js";
+} from "./chunk-R6V363NP.js";
 import {
   resolveRemoteSessionId,
   upsertSync
-} from "./chunk-YIOKUA23.js";
+} from "./chunk-L2RX552S.js";
 import {
   assertResourceTenant,
   tenantFilter
@@ -14,14 +14,14 @@ import {
 } from "./chunk-DC2UMEQH.js";
 import {
   kickoffEnvironmentSetup
-} from "./chunk-KA33M67F.js";
+} from "./chunk-C6AXM3M7.js";
 import {
   jsonOk,
   routeWrap
-} from "./chunk-LOAFGJWC.js";
+} from "./chunk-3K4KIYEZ.js";
 import {
   forwardToAnthropic
-} from "./chunk-L43UD7PK.js";
+} from "./chunk-JCW3ZRES.js";
 import {
   errors_exports,
   init_errors
@@ -31,18 +31,18 @@ import {
   isProxied,
   markProxied,
   unmarkProxied
-} from "./chunk-NT2AI5IE.js";
+} from "./chunk-E7DD7F7J.js";
 import {
   dropActor,
   getActor
 } from "./chunk-LAWTTG2E.js";
 import {
   releaseSession
-} from "./chunk-EN3ICYTL.js";
+} from "./chunk-WQHRP3JL.js";
 import {
   appendEvent,
   dropEmitter
-} from "./chunk-TRQX5AIZ.js";
+} from "./chunk-QHBXWPSO.js";
 import {
   archiveSession,
   createSession,
@@ -50,13 +50,13 @@ import {
   listSessions,
   updateSessionMutable,
   updateSessionStatus
-} from "./chunk-QWGJS6V6.js";
+} from "./chunk-2CPZFRJJ.js";
 import {
   getEnvironment
-} from "./chunk-S5HVHCAK.js";
+} from "./chunk-X6IQ57SC.js";
 import {
   getAgent
-} from "./chunk-TXUUUO5R.js";
+} from "./chunk-ZTH5JRZG.js";
 import {
   init_clock,
   nowMs
@@ -64,7 +64,7 @@ import {
 import {
   getDb,
   init_client
-} from "./chunk-4C7XJI2V.js";
+} from "./chunk-6POQAFEC.js";
 import {
   badRequest,
   notFound
@@ -137,6 +137,8 @@ var CreateSchema = z.object({
   title: z.string().nullish(),
   metadata: z.record(z.unknown()).optional(),
   max_budget_usd: z.number().positive().optional(),
+  max_tokens: z.number().int().positive().optional(),
+  max_wall_duration_ms: z.number().int().positive().optional(),
   resources: z.array(ResourceSchema).optional(),
   vault_ids: z.array(z.string()).optional()
 });
@@ -267,6 +269,8 @@ function handleCreateSession(request) {
           title: data.title ?? null,
           metadata: data.metadata ?? {},
           max_budget_usd: data.max_budget_usd ?? null,
+          max_tokens: data.max_tokens ?? null,
+          max_wall_duration_ms: data.max_wall_duration_ms ?? null,
           resources: data.resources?.length ? data.resources : null,
           vault_ids: data.vault_ids?.length ? data.vault_ids : null,
           api_key_id: auth.keyId,
@@ -326,6 +330,8 @@ function handleCreateSession(request) {
         title: data.title ?? null,
         metadata: data.metadata ?? {},
         max_budget_usd: data.max_budget_usd ?? null,
+        max_tokens: data.max_tokens ?? null,
+        max_wall_duration_ms: data.max_wall_duration_ms ?? null,
         resources: data.resources?.length ? data.resources : null,
         vault_ids: data.vault_ids?.length ? data.vault_ids : null,
         api_key_id: auth.keyId,

package/dist/{chunk-LHHBOQUR.js → chunk-ZMJ4EP4C.js}

@@ -126,6 +126,8 @@ var init_schema = __esm({
       parked_checkpoint_id: text("parked_checkpoint_id"),
       provider_name: text("provider_name").notNull().default("sprites"),
       max_budget_usd: real("max_budget_usd"),
+      max_tokens: integer("max_tokens"),
+      max_wall_duration_ms: integer("max_wall_duration_ms"),
       outcome_criteria_json: text("outcome_criteria_json"),
       resources_json: text("resources_json"),
       vault_ids_json: text("vault_ids_json"),