@agentstep/agent-sdk 0.5.0 → 0.5.2

package/dist/{chunk-JFOHGHW5.js → chunk-4B5H2MFO.js}

@@ -1,21 +1,21 @@
+import {
+  resolveVaultSecrets
+} from "./chunk-O2X4XBU7.js";
 import {
   dockerProvider
 } from "./chunk-4XXQAVKE.js";
-import {
-  resolveVaultSecrets
-} from "./chunk-T3HMVHDG.js";
 import {
   appendEvent
-} from "./chunk-5MERXOLJ.js";
+} from "./chunk-NRX5WHHN.js";
 import {
   getSession,
   getSessionRow,
   setSessionSandbox
-} from "./chunk-KUWJJD6O.js";
+} from "./chunk-TSR33JMZ.js";
 import {
   deleteSprite,
   listSprites
-} from "./chunk-NKQVOAWN.js";
+} from "./chunk-VMLXV2H6.js";
 import {
   allSessionSandboxes,
   countInEnv,
@@ -24,12 +24,12 @@ import {
 } from "./chunk-EFOIR7R3.js";
 import {
   resolveContainerProvider
-} from "./chunk-UE6DNLSV.js";
+} from "./chunk-LT5PRO54.js";
 import {
   getEnvironment,
   getEnvironmentRow,
   listEnvironments
-} from "./chunk-H6OT5GUL.js";
+} from "./chunk-7APAOJIZ.js";
 import {
   addWarm,
   claimWarm,
@@ -40,13 +40,13 @@ import {
 } from "./chunk-G7KUVNDY.js";
 import {
   getAgent
-} from "./chunk-BCIFFAGW.js";
+} from "./chunk-M3752GFZ.js";
 import {
   resolveBackend
-} from "./chunk-GV6GUSCP.js";
+} from "./chunk-Q2JYAAJO.js";
 import {
   getConfig
-} from "./chunk-X6QIWZ33.js";
+} from "./chunk-UGH5JDXX.js";
 import {
   init_clock,
   nowMs
@@ -123,15 +123,22 @@ async function acquireForFirstTurn(sessionId) {
     );
   }
   const config = getConfig();
-  if (countInEnv(env.id) + countWarm(env.id) >= config.maxSandboxesPerEnv) {
-    throw new ApiError(503, "server_busy", "env sandbox pool exhausted");
+  const envObj = getEnvironment(row.environment_id);
+  const maxSandboxes = envObj?.config?.max_sandboxes ?? config.maxSandboxesPerEnv;
+  const active = countInEnv(env.id);
+  const warmCount = countWarm(env.id);
+  if (active + warmCount >= maxSandboxes) {
+    throw new ApiError(
+      503,
+      "server_busy",
+      `environment ${env.id} sandbox pool exhausted (${active} active + ${warmCount} warm, limit ${maxSandboxes})`
+    );
   }
   const agent = getAgent(row.agent_id, row.agent_version);
   if (!agent) {
     throw new ApiError(404, "not_found_error", "agent not found for session");
   }
   const backend = resolveBackend(agent.engine);
-  const envObj = getEnvironment(row.environment_id);
   const provider = await resolveContainerProvider(envObj?.config?.provider);
   const sessionForSecrets = getSession(sessionId);
   const secrets = sessionForSecrets?.vault_ids?.length ? resolveVaultSecrets(sessionForSecrets.vault_ids) : {};
@@ -209,6 +216,7 @@ async function acquireForFirstTurn(sessionId) {
         await provisionResources(name2, session.resources, sp);
       }
     }
+    await mountMemoryStores(name2, sessionId, sp);
     register({
       sandboxName: name2,
       envId: env.id,
@@ -316,6 +324,7 @@ async function acquireForFirstTurn(sessionId) {
       await provisionResources(name, session.resources, sp);
     }
   }
+  await mountMemoryStores(name, sessionId, sp);
   register({
     sandboxName: name,
     envId: env.id,
@@ -337,7 +346,9 @@ async function replenishWarmPool(envId, engine, provider, secrets) {
   if (target <= 0) return;
   if (countWarm(envId) + countInflight(envId) >= target) return;
   const cfg = getConfig();
-  if (countInEnv(envId) + countWarm(envId) + countInflight(envId) >= cfg.maxSandboxesPerEnv) return;
+  const envObj = getEnvironment(envId);
+  const maxSandboxes = envObj?.config?.max_sandboxes ?? cfg.maxSandboxesPerEnv;
+  if (countInEnv(envId) + countWarm(envId) + countInflight(envId) >= maxSandboxes) return;
   incrementInflight(envId);
   try {
     const sp = wrapProviderWithSecrets(provider, secrets);
@@ -391,7 +402,8 @@ async function fillOneEnv(env) {
   try {
     const provider = await resolveContainerProvider(env.config?.provider);
     if (!provider.supportsWarmPool) return;
-    const backend = resolveBackend("claude");
+    const engine = env.config?.default_engine ?? "claude";
+    const backend = resolveBackend(engine);
     const needed = target - countWarm(env.id) - countInflight(env.id);
     for (let i = 0; i < needed; i++) {
       void replenishWarmPool(env.id, backend.name, provider, {});
@@ -507,6 +519,36 @@ async function provisionResources(sandboxName, resources, provider) {
     }
   }
 }
+async function mountMemoryStores(sandboxName, sessionId, provider) {
+  const { listResources: listSessionResources } = await import("./db/session-resources.js");
+  const { getMemoryStore, listMemories } = await import("./db/memory.js");
+  const resources = listSessionResources(sessionId);
+  const memStoreResources = resources.filter((r) => r.type === "memory_store" && r.memory_store_id);
+  if (memStoreResources.length === 0) return;
+  await provider.exec(sandboxName, ["mkdir", "-p", "/mnt/memory"]);
+  for (const r of memStoreResources) {
+    const store = getMemoryStore(r.memory_store_id);
+    if (!store) {
+      lcLog(`[lifecycle] memory store not found: ${r.memory_store_id}`);
+      continue;
+    }
+    const storeName = store.name.replace(/[^a-zA-Z0-9_.-]/g, "_");
+    const storeDir = `/mnt/memory/${storeName}`;
+    await provider.exec(sandboxName, ["mkdir", "-p", storeDir]);
+    const memories = listMemories(store.id);
+    for (const mem of memories) {
+      const memPath = `${storeDir}/${mem.path}`;
+      const dir = memPath.substring(0, memPath.lastIndexOf("/"));
+      if (dir && dir !== storeDir) {
+        await provider.exec(sandboxName, ["mkdir", "-p", dir]);
+      }
+      await provider.exec(sandboxName, ["sh", "-c", 'cat > "$1"', "sh", memPath], {
+        stdin: mem.content
+      });
+    }
+    lcLog(`[lifecycle] ${sessionId} mounted ${memories.length} memories from store "${store.name}" at ${storeDir}`);
+  }
+}
 async function releaseSession(sessionId) {
   const entry = unregister(sessionId);
   const row = getSessionRow(sessionId);
@@ -574,6 +616,7 @@ export {
   replenishWarmPool,
   fillWarmPools,
   provisionResources,
+  mountMemoryStores,
   releaseSession,
   reconcileOrphanSandboxes,
   reconcileDockerOrphanSandboxes

package/dist/{chunk-OZFSKR2W.js → chunk-5525T7PL.js}

@@ -7,7 +7,7 @@ import {
   jsonOk,
   paginatedOk,
   routeWrap
-} from "./chunk-QTXAWC5J.js";
+} from "./chunk-6URMGCAC.js";
 import {
   archiveVault,
   createVault,
@@ -19,14 +19,14 @@ import {
   listVaults,
   setEntry,
   updateVault
-} from "./chunk-VP527YC5.js";
+} from "./chunk-BYIU4WAQ.js";
 import {
   getAgent
-} from "./chunk-BCIFFAGW.js";
+} from "./chunk-M3752GFZ.js";
 import {
   getDb,
   init_client
-} from "./chunk-W3JMIUHV.js";
+} from "./chunk-KQTTKYZY.js";
 import {
   badRequest,
   conflict,

package/dist/{chunk-AXBM7HAR.js → chunk-5EFY6UGV.js}

@@ -1,7 +1,7 @@
 import {
   jsonOk,
   routeWrap
-} from "./chunk-QTXAWC5J.js";
+} from "./chunk-6URMGCAC.js";
 
 // src/handlers/whoami.ts
 function handleWhoami(request) {

package/dist/chunk-5UTTNXNJ.js

@@ -0,0 +1,166 @@
+import {
+  archiveThread,
+  getThread,
+  listThreads
+} from "./chunk-7IYL2O7H.js";
+import {
+  assertResourceTenant
+} from "./chunk-23UKWXJH.js";
+import {
+  jsonOk,
+  paginatedOk,
+  routeWrap
+} from "./chunk-6URMGCAC.js";
+import {
+  subscribe
+} from "./chunk-NRX5WHHN.js";
+import {
+  getSession
+} from "./chunk-TSR33JMZ.js";
+import {
+  listEvents,
+  rowToManagedEvent
+} from "./chunk-7JAUY5SG.js";
+import {
+  getDb,
+  init_client
+} from "./chunk-KQTTKYZY.js";
+import {
+  badRequest,
+  notFound
+} from "./chunk-EZYKRG4W.js";
+
+// src/handlers/threads.ts
+init_client();
+function loadSession(auth, sessionId) {
+  const row = getDb().prepare(`SELECT tenant_id FROM sessions WHERE id = ?`).get(sessionId);
+  if (!row) throw notFound(`session ${sessionId} not found`);
+  assertResourceTenant(auth, row.tenant_id, `session ${sessionId} not found`);
+  const session = getSession(sessionId);
+  if (!session) throw notFound(`session ${sessionId} not found`);
+  return session;
+}
+function handleListThreads(request, sessionId) {
+  return routeWrap(request, async ({ auth }) => {
+    loadSession(auth, sessionId);
+    const url = new URL(request.url);
+    const limit = url.searchParams.get("limit");
+    const order = url.searchParams.get("order");
+    const requestedLimit = limit ? Number(limit) : 50;
+    const data = listThreads(sessionId, {
+      limit: requestedLimit,
+      order: order ?? void 0
+    });
+    return paginatedOk(data, requestedLimit);
+  });
+}
+function handleGetThread(request, sessionId, threadId) {
+  return routeWrap(request, async ({ auth }) => {
+    loadSession(auth, sessionId);
+    const thread = getThread(sessionId, threadId);
+    if (!thread) throw notFound(`thread ${threadId} not found`);
+    return jsonOk(thread);
+  });
+}
+function handleArchiveThread(request, sessionId, threadId) {
+  return routeWrap(request, async ({ auth }) => {
+    loadSession(auth, sessionId);
+    const existing = getThread(sessionId, threadId);
+    if (!existing) throw notFound(`thread ${threadId} not found`);
+    if (existing.status !== "idle") {
+      throw badRequest(`cannot archive thread in status "${existing.status}" \u2014 must be idle`);
+    }
+    const archived = archiveThread(sessionId, threadId);
+    if (!archived) throw notFound(`thread ${threadId} not found`);
+    return jsonOk(archived);
+  });
+}
+function handleListThreadEvents(request, sessionId, threadId) {
+  return routeWrap(request, async ({ auth }) => {
+    loadSession(auth, sessionId);
+    const thread = getThread(sessionId, threadId);
+    if (!thread) throw notFound(`thread ${threadId} not found`);
+    const url = new URL(request.url);
+    const limit = url.searchParams.get("limit");
+    const order = url.searchParams.get("order");
+    const afterSeq = url.searchParams.get("after_seq");
+    const requestedLimit = limit ? Number(limit) : 50;
+    const allEvents = listEvents(sessionId, {
+      limit: 500,
+      // fetch more than needed since we filter
+      order: order ?? "asc",
+      afterSeq: afterSeq ? Number(afterSeq) : void 0
+    });
+    const threadEvents = allEvents.filter((e) => {
+      const payload = JSON.parse(e.payload_json);
+      return payload.thread_id === threadId || e.type.startsWith("session.thread_");
+    }).slice(0, requestedLimit).map(rowToManagedEvent);
+    return paginatedOk(threadEvents, requestedLimit);
+  });
+}
+function handleStreamThreadEvents(request, sessionId, threadId) {
+  return routeWrap(request, async ({ auth }) => {
+    loadSession(auth, sessionId);
+    const thread = getThread(sessionId, threadId);
+    if (!thread) throw notFound(`thread ${threadId} not found`);
+    const url = new URL(request.url);
+    const lastEventId = request.headers.get("last-event-id");
+    const afterSeq = lastEventId ? Number(lastEventId) : Number(url.searchParams.get("after_seq") ?? "0");
+    const encoder = new TextEncoder();
+    let controller;
+    const stream = new ReadableStream({
+      start(c) {
+        controller = c;
+      }
+    });
+    const write = (payload) => {
+      try {
+        controller.enqueue(encoder.encode(payload));
+      } catch {
+      }
+    };
+    const sub = subscribe(
+      sessionId,
+      Number.isFinite(afterSeq) ? afterSeq : 0,
+      (evt) => {
+        if (evt.thread_id === threadId) {
+          write(`id: ${evt.seq}
+event: ${evt.type}
+data: ${JSON.stringify(evt)}
+
+`);
+        }
+      }
+    );
+    const keepalive = setInterval(() => {
+      write(`data: {"type":"ping"}
+
+`);
+    }, 15e3);
+    const abort = () => {
+      clearInterval(keepalive);
+      sub.unsubscribe();
+      try {
+        controller.close();
+      } catch {
+      }
+    };
+    request.signal.addEventListener("abort", abort);
+    return new Response(stream, {
+      headers: {
+        "Content-Type": "text/event-stream",
+        "Cache-Control": "no-cache",
+        Connection: "keep-alive",
+        "X-Accel-Buffering": "no"
+      }
+    });
+  });
+}
+
+export {
+  handleListThreads,
+  handleGetThread,
+  handleArchiveThread,
+  handleListThreadEvents,
+  handleStreamThreadEvents
+};

package/dist/{chunk-4RDILRIO.js → chunk-5YGYUGDQ.js}

@@ -1,13 +1,13 @@
 import {
   jsonOk,
   routeWrap
-} from "./chunk-QTXAWC5J.js";
+} from "./chunk-6URMGCAC.js";
 import {
   resolveContainerProvider
-} from "./chunk-UE6DNLSV.js";
+} from "./chunk-LT5PRO54.js";
 import {
   getConfig
-} from "./chunk-X6QIWZ33.js";
+} from "./chunk-UGH5JDXX.js";
 
 // src/handlers/providers.ts
 var LOCAL_PROVIDERS = ["docker", "apple-container", "podman"];

package/dist/{chunk-HWR2HYQJ.js → chunk-6DPJVMEC.js}

@@ -1,6 +1,6 @@
 import {
   getConfig
-} from "./chunk-X6QIWZ33.js";
+} from "./chunk-UGH5JDXX.js";
 
 // src/backends/pi/auth.ts
 function buildPiAuthEnv() {

package/dist/{chunk-PNZF7HIU.js → chunk-6KJS42ZL.js}

@@ -1,7 +1,3 @@
-import {
-  buildClaudeArgs,
-  buildClaudeAuthEnv
-} from "./chunk-GIMDS46L.js";
 import {
   PERMISSION_BRIDGE_DIR,
   PERMISSION_HOOK_SCRIPT_PATH,
@@ -19,13 +15,17 @@ import {
 import {
   createClaudeTranslator
 } from "./chunk-JFYE5BFI.js";
+import {
+  buildClaudeArgs,
+  buildClaudeAuthEnv
+} from "./chunk-ANS2TQYT.js";
 import {
   CLAUDE_WRAPPER_PATH,
   installClaudeWrapper
-} from "./chunk-IU457W7Q.js";
+} from "./chunk-QBJOYOVF.js";
 import {
   getConfig
-} from "./chunk-X6QIWZ33.js";
+} from "./chunk-UGH5JDXX.js";
 
 // src/backends/claude/index.ts
 function buildTurn(input) {
@@ -33,7 +33,8 @@ function buildTurn(input) {
   const argsBase = buildClaudeArgs({
     agent,
     claudeSessionId: backendSessionId,
-    confirmationMode: agent.confirmation_mode
+    confirmationMode: agent.confirmation_mode,
+    memoryStores: input.memoryStores
   });
   const env = buildClaudeAuthEnv();
   const customTools = agent.tools.filter((t) => t.type === "custom");

package/dist/{chunk-LQP6XGFU.js → chunk-6KQFP32G.js}

@@ -1,13 +1,13 @@
 import {
   resolveContainerProvider
-} from "./chunk-UE6DNLSV.js";
+} from "./chunk-LT5PRO54.js";
 import {
   getEnvironmentRow,
   updateEnvironmentState
-} from "./chunk-H6OT5GUL.js";
+} from "./chunk-7APAOJIZ.js";
 import {
   installClaudeWrapper
-} from "./chunk-IU457W7Q.js";
+} from "./chunk-QBJOYOVF.js";
 import {
   init_ids,
   newId

package/dist/{chunk-DMMNAQUM.js → chunk-6L4AGHAV.js}

@@ -10,10 +10,10 @@ import {
 import {
   getDrizzle,
   init_drizzle
-} from "./chunk-PZNAQBHQ.js";
+} from "./chunk-FILGQF2G.js";
 import {
   schema_exports
-} from "./chunk-ILHIHMO3.js";
+} from "./chunk-VC5DELR4.js";
 
 // src/db/files.ts
 init_drizzle();

package/dist/{chunk-WEC625LQ.js → chunk-6TJBZXD4.js}

@@ -1,6 +1,6 @@
 import {
   getConfig
-} from "./chunk-X6QIWZ33.js";
+} from "./chunk-UGH5JDXX.js";
 
 // src/backends/codex/auth.ts
 function buildCodexAuthEnv() {

package/dist/{chunk-QTXAWC5J.js → chunk-6URMGCAC.js}

@@ -4,13 +4,13 @@ import {
 } from "./chunk-D2XITRN6.js";
 import {
   authenticateAndIntercept
-} from "./chunk-GCQDNUS2.js";
+} from "./chunk-WFRHLTKI.js";
 import {
   checkAndBump
 } from "./chunk-HVUWXUUI.js";
 import {
   ensureInitialized
-} from "./chunk-PN3AWRMX.js";
+} from "./chunk-YDP3YA5D.js";
 import {
   captureException
 } from "./chunk-3MQ2FWXS.js";

package/dist/{chunk-DJZSPWG2.js → chunk-6WUYI3DJ.js}

@@ -1,17 +1,17 @@
 import {
   listTraces
-} from "./chunk-XSNJ7NT2.js";
+} from "./chunk-G4DTLYOD.js";
 import {
   jsonOk,
   routeWrap
-} from "./chunk-QTXAWC5J.js";
+} from "./chunk-6URMGCAC.js";
 import {
   exportTrace
-} from "./chunk-JP7Y3TKK.js";
+} from "./chunk-FSVDF57K.js";
 import {
   listEventsByTrace,
   rowToManagedEvent
-} from "./chunk-KGBKIJPF.js";
+} from "./chunk-7JAUY5SG.js";
 import {
   badRequest,
   notFound

package/dist/{chunk-H6OT5GUL.js → chunk-7APAOJIZ.js}

@@ -1,6 +1,6 @@
 import {
   DEFAULT_TENANT_ID
-} from "./chunk-I26QP3A3.js";
+} from "./chunk-UQZFFG7U.js";
 import {
   init_ids,
   newId
@@ -13,10 +13,10 @@ import {
 import {
   getDrizzle,
   init_drizzle
-} from "./chunk-PZNAQBHQ.js";
+} from "./chunk-FILGQF2G.js";
 import {
   schema_exports
-} from "./chunk-ILHIHMO3.js";
+} from "./chunk-VC5DELR4.js";
 
 // src/db/environments.ts
 init_drizzle();

package/dist/{chunk-HVLYE4S5.js → chunk-7EDLLXFQ.js}

@@ -14,7 +14,7 @@ import {
 import {
   getDb,
   init_client
-} from "./chunk-W3JMIUHV.js";
+} from "./chunk-KQTTKYZY.js";
 
 // src/db/credentials.ts
 init_client();