@minpeter/pss-runtime 0.1.0-next.2 → 0.1.0-next.3

package/dist/agent-resume.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent-resume.js","names":[],"sources":["../src/agent-resume.ts"],"sourcesContent":["import { ownsAgentNamespace } from \"./agent-namespace\";\nimport { StoredAgentRun } from \"./execution/run\";\nimport type {\n  ExecutionHost,\n  NotificationRecord,\n  RunRecord,\n} from \"./execution/types\";\nimport { type AgentRun, BufferedAgentRun } from \"./session/run\";\nimport { readDurableBackgroundChildRunState } from \"./subagent-background-child-run-state\";\nimport { buildDurableResumeGroups } from \"./subagent-background-resume-group\";\nimport { runBackgroundJob } from \"./subagent-background-runner\";\nimport type { RuntimeInputSink, Subagent, SubagentJob } from \"./subagent-types\";\n\nconst defaultResumeLeaseMs = 300_000;\n\ninterface ResumeAgentRunInput {\n  readonly host: ExecutionHost;\n  readonly ownerNamespace: string;\n  resumeNotification(notification: NotificationRecord): Promise<AgentRun>;\n  readonly runId: string;\n  readonly subagents: readonly Subagent[];\n}\n\nexport async function resumeAgentRun({\n  host,\n  ownerNamespace,\n  resumeNotification,\n  runId,\n  subagents,\n}: ResumeAgentRunInput): Promise<AgentRun | null> {\n  const run = await host.store.runs.get(runId);\n  if (!run) {\n    return null;\n  }\n  if (!canAccessRun(run, ownerNamespace)) {\n    return null;\n  }\n\n  if (run.kind === \"background-subagent\") {\n    return await resumeBackgroundSubagentRun({ host, run, subagents });\n  }\n\n  if (run.kind === \"notification\" && run.dedupeKey) {\n    const idempotencyKey = run.dedupeKey;\n    const claimed = await claimRun(host, run);\n    if (!claimed) {\n      return null;\n    }\n\n    const notification = await claimNotificationForRun({\n      host,\n      idempotencyKey,\n      ownerNamespace,\n    });\n    if (!notification) {\n      return null;\n    }\n\n    try {\n      const notificationRun = await resumeNotification(notification);\n      await completeNotificationRun(host, claimed.runId);\n      return notificationRun;\n    } catch (error) {\n      await host.store.notifications.releaseByIdempotencyKey(idempotencyKey);\n      throw error;\n    }\n  }\n\n  return null;\n}\n\nasync function claimNotificationForRun({\n  host,\n  idempotencyKey,\n  ownerNamespace,\n}: {\n  readonly host: ExecutionHost;\n  readonly idempotencyKey: string;\n  readonly ownerNamespace: string;\n}): Promise<NotificationRecord | null> {\n  const current =\n    await host.store.notifications.getByIdempotencyKey(idempotencyKey);\n  if (!ownsAgentNamespace(current?.ownerNamespace, ownerNamespace)) {\n    return null;\n  }\n\n  const claim =\n    await host.store.notifications.claimByIdempotencyKey(idempotencyKey);\n  if (claim.ok) {\n    if (ownsAgentNamespace(claim.record.ownerNamespace, ownerNamespace)) {\n      return claim.record;\n    }\n    await host.store.notifications.releaseByIdempotencyKey(idempotencyKey);\n    return null;\n  }\n\n  if (\n    claim.reason === \"already-claimed\" &&\n    ownsAgentNamespace(claim.record?.ownerNamespace, ownerNamespace)\n  ) {\n    return claim.record ?? null;\n  }\n\n  return null;\n}\n\nfunction canAccessRun(run: RunRecord, ownerNamespace: string): boolean {\n  if (run.ownerNamespace) {\n    return ownsAgentNamespace(run.ownerNamespace, ownerNamespace);\n  }\n\n  return (\n    run.sessionKey.startsWith(`parent:${ownerNamespace}:`) ||\n    run.parentRunId?.startsWith(`${ownerNamespace}:session:`) === true\n  );\n}\n\nexport async function completeNotificationRun(\n  host: ExecutionHost,\n  runId: string\n): Promise<void> {\n  const run = await host.store.runs.get(runId);\n  if (run?.kind !== \"notification\" || run.status === \"completed\") {\n    return;\n  }\n\n  await host.store.runs.update({ ...run, status: \"completed\" });\n}\n\nasync function resumeBackgroundSubagentRun({\n  host,\n  run,\n  subagents,\n}: {\n  readonly host: ExecutionHost;\n  readonly run: RunRecord;\n  readonly subagents: readonly Subagent[];\n}): Promise<AgentRun | null> {\n  const claimed = await claimRun(host, run);\n  if (!claimed) {\n    return null;\n  }\n\n  const checkpoint = await host.store.checkpoints.latest(run.runId);\n  const state = readDurableBackgroundChildRunState(checkpoint);\n  if (!state) {\n    throw new AgentResumeError(run.runId, \"missing background run state\");\n  }\n\n  const subagent = subagents.find(\n    (candidate) => candidate.name === state.subagent\n  );\n  if (!subagent) {\n    throw new AgentResumeError(\n      run.runId,\n      `missing subagent ${JSON.stringify(state.subagent)}`\n    );\n  }\n\n  const childSession = subagent.session(claimed.sessionKey);\n  const job: SubagentJob = {\n    abort: () => childSession.interrupt(),\n    childRunId: claimed.runId,\n    childRunLeaseId: claimed.lease?.leaseId,\n    cleanup: () => Promise.resolve(),\n    dedupeKey: claimed.dedupeKey,\n    delegateToolCallId: state.delegateToolCallId,\n    description: state.description,\n    executionHost: host,\n    groupId: state.groupId,\n    id: claimed.publicTaskId ?? claimed.runId,\n    ownerNamespace: claimed.ownerNamespace,\n    parentRunId: claimed.parentRunId,\n    parentSessionKey: state.parentSessionKey,\n    promise: Promise.resolve(),\n    sessionKey: claimed.sessionKey,\n    settled: false,\n    status: \"running\",\n    subagent: state.subagent,\n  };\n  const jobs = new Map([[job.id, job]]);\n  const groups = await buildDurableResumeGroups({\n    currentJob: job,\n    host,\n    run: claimed,\n    state,\n    jobs,\n  });\n  const parentSession = durableParentSession(host, run.runId);\n\n  job.promise = runBackgroundJob({\n    childSession,\n    groups,\n    jobs,\n    job,\n    parentSession,\n    prompt: state.prompt,\n  }).finally(() => {\n    job.settled = true;\n  });\n  await job.promise;\n\n  return new StoredAgentRun({\n    eventStore: host.store.events,\n    runId: run.runId,\n  });\n}\n\nasync function claimRun(\n  host: ExecutionHost,\n  run: RunRecord\n): Promise<RunRecord | null> {\n  const claim = await host.store.runs.claim(run.runId, {\n    attempt: (run.lease?.attempt ?? 0) + 1,\n    leaseId: crypto.randomUUID(),\n    leaseMs: defaultResumeLeaseMs,\n    nowMs: Date.now(),\n  });\n  return claim.ok ? claim.record : null;\n}\n\nfunction durableParentSession(\n  host: ExecutionHost,\n  runId: string\n): RuntimeInputSink {\n  return {\n    emitObserverEvent: (event) => host.store.events.append(runId, event).then(),\n    enqueueRuntimeInput: () => undefined,\n    notify: () => Promise.resolve(emptyRun()),\n  };\n}\n\nfunction emptyRun(): AgentRun {\n  const run = new BufferedAgentRun();\n  run.close();\n  return run;\n}\n\nclass AgentResumeError extends Error {\n  constructor(runId: string, reason: string) {\n    super(`Cannot resume agent run ${runId}: ${reason}`);\n    this.name = \"AgentResumeError\";\n  }\n}\n"],"mappings":";;;;;;;AAaA,MAAM,uBAAuB;AAU7B,eAAsB,eAAe,EACnC,MACA,gBACA,oBACA,OACA,aACgD;CAChD,MAAM,MAAM,MAAM,KAAK,MAAM,KAAK,IAAI,KAAK;CAC3C,IAAI,CAAC,KACH,OAAO;CAET,IAAI,CAAC,aAAa,KAAK,cAAc,GACnC,OAAO;CAGT,IAAI,IAAI,SAAS,uBACf,OAAO,MAAM,4BAA4B;EAAE;EAAM;EAAK;CAAU,CAAC;CAGnE,IAAI,IAAI,SAAS,kBAAkB,IAAI,WAAW;EAChD,MAAM,iBAAiB,IAAI;EAC3B,MAAM,UAAU,MAAM,SAAS,MAAM,GAAG;EACxC,IAAI,CAAC,SACH,OAAO;EAGT,MAAM,eAAe,MAAM,wBAAwB;GACjD;GACA;GACA;EACF,CAAC;EACD,IAAI,CAAC,cACH,OAAO;EAGT,IAAI;GACF,MAAM,kBAAkB,MAAM,mBAAmB,YAAY;GAC7D,MAAM,wBAAwB,MAAM,QAAQ,KAAK;GACjD,OAAO;EACT,SAAS,OAAO;GACd,MAAM,KAAK,MAAM,cAAc,wBAAwB,cAAc;GACrE,MAAM;EACR;CACF;CAEA,OAAO;AACT;AAEA,eAAe,wBAAwB,EACrC,MACA,gBACA,kBAKqC;CAGrC,IAAI,CAAC,oBAAmB,MADhB,KAAK,MAAM,cAAc,oBAAoB,cAAc,IAClC,gBAAgB,cAAc,GAC7D,OAAO;CAGT,MAAM,QACJ,MAAM,KAAK,MAAM,cAAc,sBAAsB,cAAc;CACrE,IAAI,MAAM,IAAI;EACZ,IAAI,mBAAmB,MAAM,OAAO,gBAAgB,cAAc,GAChE,OAAO,MAAM;EAEf,MAAM,KAAK,MAAM,cAAc,wBAAwB,cAAc;EACrE,OAAO;CACT;CAEA,IACE,MAAM,WAAW,qBACjB,mBAAmB,MAAM,QAAQ,gBAAgB,cAAc,GAE/D,OAAO,MAAM,UAAU;CAGzB,OAAO;AACT;AAEA,SAAS,aAAa,KAAgB,gBAAiC;CACrE,IAAI,IAAI,gBACN,OAAO,mBAAmB,IAAI,gBAAgB,cAAc;CAG9D,OACE,IAAI,WAAW,WAAW,UAAU,eAAe,EAAE,KACrD,IAAI,aAAa,WAAW,GAAG,eAAe,UAAU,MAAM;AAElE;AAEA,eAAsB,wBACpB,MACA,OACe;CACf,MAAM,MAAM,MAAM,KAAK,MAAM,KAAK,IAAI,KAAK;CAC3C,IAAI,KAAK,SAAS,kBAAkB,IAAI,WAAW,aACjD;CAGF,MAAM,KAAK,MAAM,KAAK,OAAO;EAAE,GAAG;EAAK,QAAQ;CAAY,CAAC;AAC9D;AAEA,eAAe,4BAA4B,EACzC,MACA,KACA,aAK2B;CAC3B,MAAM,UAAU,MAAM,SAAS,MAAM,GAAG;CACxC,IAAI,CAAC,SACH,OAAO;CAIT,MAAM,QAAQ,mCAAmC,MADxB,KAAK,MAAM,YAAY,OAAO,IAAI,KAAK,CACL;CAC3D,IAAI,CAAC,OACH,MAAM,IAAI,iBAAiB,IAAI,OAAO,8BAA8B;CAGtE,MAAM,WAAW,UAAU,MACxB,cAAc,UAAU,SAAS,MAAM,QAC1C;CACA,IAAI,CAAC,UACH,MAAM,IAAI,iBACR,IAAI,OACJ,oBAAoB,KAAK,UAAU,MAAM,QAAQ,GACnD;CAGF,MAAM,eAAe,SAAS,QAAQ,QAAQ,UAAU;CACxD,MAAM,MAAmB;EACvB,aAAa,aAAa,UAAU;EACpC,YAAY,QAAQ;EACpB,iBAAiB,QAAQ,OAAO;EAChC,eAAe,QAAQ,QAAQ;EAC/B,WAAW,QAAQ;EACnB,oBAAoB,MAAM;EAC1B,aAAa,MAAM;EACnB,eAAe;EACf,SAAS,MAAM;EACf,IAAI,QAAQ,gBAAgB,QAAQ;EACpC,gBAAgB,QAAQ;EACxB,aAAa,QAAQ;EACrB,kBAAkB,MAAM;EACxB,SAAS,QAAQ,QAAQ;EACzB,YAAY,QAAQ;EACpB,SAAS;EACT,QAAQ;EACR,UAAU,MAAM;CAClB;CACA,MAAM,OAAO,IAAI,IAAI,CAAC,CAAC,IAAI,IAAI,GAAG,CAAC,CAAC;CAUpC,IAAI,UAAU,iBAAiB;EAC7B;EACA,QAAA,MAXmB,yBAAyB;GAC5C,YAAY;GACZ;GACA,KAAK;GACL;GACA;EACF,CAAC;EAMC;EACA;EACA,eAPoB,qBAAqB,MAAM,IAAI,KAOvC;EACZ,QAAQ,MAAM;CAChB,CAAC,EAAE,cAAc;EACf,IAAI,UAAU;CAChB,CAAC;CACD,MAAM,IAAI;CAEV,OAAO,IAAI,eAAe;EACxB,YAAY,KAAK,MAAM;EACvB,OAAO,IAAI;CACb,CAAC;AACH;AAEA,eAAe,SACb,MACA,KAC2B;CAC3B,MAAM,QAAQ,MAAM,KAAK,MAAM,KAAK,MAAM,IAAI,OAAO;EACnD,UAAU,IAAI,OAAO,WAAW,KAAK;EACrC,SAAS,OAAO,WAAW;EAC3B,SAAS;EACT,OAAO,KAAK,IAAI;CAClB,CAAC;CACD,OAAO,MAAM,KAAK,MAAM,SAAS;AACnC;AAEA,SAAS,qBACP,MACA,OACkB;CAClB,OAAO;EACL,oBAAoB,UAAU,KAAK,MAAM,OAAO,OAAO,OAAO,KAAK,EAAE,KAAK;EAC1E,2BAA2B,KAAA;EAC3B,cAAc,QAAQ,QAAQ,SAAS,CAAC;CAC1C;AACF;AAEA,SAAS,WAAqB;CAC5B,MAAM,MAAM,IAAI,iBAAiB;CACjC,IAAI,MAAM;CACV,OAAO;AACT;AAEA,IAAM,mBAAN,cAA+B,MAAM;CACnC,YAAY,OAAe,QAAgB;EACzC,MAAM,2BAA2B,MAAM,IAAI,QAAQ;EACnD,KAAK,OAAO;CACd;AACF"}
+{"version":3,"file":"agent-resume.js","names":[],"sources":["../src/agent-resume.ts"],"sourcesContent":["import { ownsAgentNamespace } from \"./agent-namespace\";\nimport type {\n  ExecutionHost,\n  NotificationRecord,\n  RunRecord,\n} from \"./execution/types\";\nimport type { AgentRun } from \"./session/run\";\n\ninterface ResumeAgentRunInput {\n  readonly host: ExecutionHost;\n  readonly ownerNamespace: string;\n  resumeNotification(notification: NotificationRecord): Promise<AgentRun>;\n  readonly runId: string;\n}\n\nexport async function resumeAgentRun({\n  host,\n  ownerNamespace,\n  resumeNotification,\n  runId,\n}: ResumeAgentRunInput): Promise<AgentRun | null> {\n  const run = await host.store.runs.get(runId);\n  if (!run) {\n    return null;\n  }\n  if (!canAccessRun(run, ownerNamespace)) {\n    return null;\n  }\n\n  if (run.kind === \"notification\" && run.dedupeKey) {\n    const idempotencyKey = run.dedupeKey;\n    const claimed = await claimRun(host, run);\n    if (!claimed) {\n      return null;\n    }\n\n    const notification = await claimNotificationForRun({\n      host,\n      idempotencyKey,\n      ownerNamespace,\n    });\n    if (!notification) {\n      return null;\n    }\n\n    try {\n      const notificationRun = await resumeNotification(notification);\n      await completeNotificationRun(host, claimed.runId);\n      return notificationRun;\n    } catch (error) {\n      await host.store.notifications.releaseByIdempotencyKey(idempotencyKey);\n      throw error;\n    }\n  }\n\n  return null;\n}\n\nasync function claimNotificationForRun({\n  host,\n  idempotencyKey,\n  ownerNamespace,\n}: {\n  readonly host: ExecutionHost;\n  readonly idempotencyKey: string;\n  readonly ownerNamespace: string;\n}): Promise<NotificationRecord | null> {\n  const current =\n    await host.store.notifications.getByIdempotencyKey(idempotencyKey);\n  if (!ownsAgentNamespace(current?.ownerNamespace, ownerNamespace)) {\n    return null;\n  }\n\n  const claim =\n    await host.store.notifications.claimByIdempotencyKey(idempotencyKey);\n  if (claim.ok) {\n    if (ownsAgentNamespace(claim.record.ownerNamespace, ownerNamespace)) {\n      return claim.record;\n    }\n    await host.store.notifications.releaseByIdempotencyKey(idempotencyKey);\n    return null;\n  }\n\n  if (\n    claim.reason === \"already-claimed\" &&\n    ownsAgentNamespace(claim.record?.ownerNamespace, ownerNamespace)\n  ) {\n    return claim.record ?? null;\n  }\n\n  return null;\n}\n\nfunction canAccessRun(run: RunRecord, ownerNamespace: string): boolean {\n  if (run.ownerNamespace) {\n    return ownsAgentNamespace(run.ownerNamespace, ownerNamespace);\n  }\n\n  return (\n    run.sessionKey.startsWith(`parent:${ownerNamespace}:`) ||\n    run.parentRunId?.startsWith(`${ownerNamespace}:session:`) === true\n  );\n}\n\nexport async function completeNotificationRun(\n  host: ExecutionHost,\n  runId: string\n): Promise<void> {\n  const run = await host.store.runs.get(runId);\n  if (run?.kind !== \"notification\" || run.status === \"completed\") {\n    return;\n  }\n\n  await host.store.runs.update({ ...run, status: \"completed\" });\n}\n\nasync function claimRun(\n  host: ExecutionHost,\n  run: RunRecord\n): Promise<RunRecord | null> {\n  const claim = await host.store.runs.claim(run.runId, {\n    attempt: (run.lease?.attempt ?? 0) + 1,\n    leaseId: crypto.randomUUID(),\n    leaseMs: 300_000,\n    nowMs: Date.now(),\n  });\n  return claim.ok ? claim.record : null;\n}\n"],"mappings":";;AAeA,eAAsB,eAAe,EACnC,MACA,gBACA,oBACA,SACgD;CAChD,MAAM,MAAM,MAAM,KAAK,MAAM,KAAK,IAAI,KAAK;CAC3C,IAAI,CAAC,KACH,OAAO;CAET,IAAI,CAAC,aAAa,KAAK,cAAc,GACnC,OAAO;CAGT,IAAI,IAAI,SAAS,kBAAkB,IAAI,WAAW;EAChD,MAAM,iBAAiB,IAAI;EAC3B,MAAM,UAAU,MAAM,SAAS,MAAM,GAAG;EACxC,IAAI,CAAC,SACH,OAAO;EAGT,MAAM,eAAe,MAAM,wBAAwB;GACjD;GACA;GACA;EACF,CAAC;EACD,IAAI,CAAC,cACH,OAAO;EAGT,IAAI;GACF,MAAM,kBAAkB,MAAM,mBAAmB,YAAY;GAC7D,MAAM,wBAAwB,MAAM,QAAQ,KAAK;GACjD,OAAO;EACT,SAAS,OAAO;GACd,MAAM,KAAK,MAAM,cAAc,wBAAwB,cAAc;GACrE,MAAM;EACR;CACF;CAEA,OAAO;AACT;AAEA,eAAe,wBAAwB,EACrC,MACA,gBACA,kBAKqC;CAGrC,IAAI,CAAC,oBAAmB,MADhB,KAAK,MAAM,cAAc,oBAAoB,cAAc,IAClC,gBAAgB,cAAc,GAC7D,OAAO;CAGT,MAAM,QACJ,MAAM,KAAK,MAAM,cAAc,sBAAsB,cAAc;CACrE,IAAI,MAAM,IAAI;EACZ,IAAI,mBAAmB,MAAM,OAAO,gBAAgB,cAAc,GAChE,OAAO,MAAM;EAEf,MAAM,KAAK,MAAM,cAAc,wBAAwB,cAAc;EACrE,OAAO;CACT;CAEA,IACE,MAAM,WAAW,qBACjB,mBAAmB,MAAM,QAAQ,gBAAgB,cAAc,GAE/D,OAAO,MAAM,UAAU;CAGzB,OAAO;AACT;AAEA,SAAS,aAAa,KAAgB,gBAAiC;CACrE,IAAI,IAAI,gBACN,OAAO,mBAAmB,IAAI,gBAAgB,cAAc;CAG9D,OACE,IAAI,WAAW,WAAW,UAAU,eAAe,EAAE,KACrD,IAAI,aAAa,WAAW,GAAG,eAAe,UAAU,MAAM;AAElE;AAEA,eAAsB,wBACpB,MACA,OACe;CACf,MAAM,MAAM,MAAM,KAAK,MAAM,KAAK,IAAI,KAAK;CAC3C,IAAI,KAAK,SAAS,kBAAkB,IAAI,WAAW,aACjD;CAGF,MAAM,KAAK,MAAM,KAAK,OAAO;EAAE,GAAG;EAAK,QAAQ;CAAY,CAAC;AAC9D;AAEA,eAAe,SACb,MACA,KAC2B;CAC3B,MAAM,QAAQ,MAAM,KAAK,MAAM,KAAK,MAAM,IAAI,OAAO;EACnD,UAAU,IAAI,OAAO,WAAW,KAAK;EACrC,SAAS,OAAO,WAAW;EAC3B,SAAS;EACT,OAAO,KAAK,IAAI;CAClB,CAAC;CACD,OAAO,MAAM,KAAK,MAAM,SAAS;AACnC"}

package/dist/agent-session-entry.d.ts

@@ -3,8 +3,8 @@ import { AgentRun } from "./session/run.js";
 //#region src/agent-session-entry.d.ts
 interface SessionHandle {
   delete(): Promise<void>;
+  dispose(): Promise<void>;
   interrupt(): void;
-  kill(): Promise<void>;
   send(input: AgentInput): Promise<AgentRun>;
   steer(input: AgentInput): Promise<AgentRun>;
 }

package/dist/agent.d.ts

@@ -1,15 +1,15 @@
 import { AgentInput } from "./session/input.js";
 import { AgentHost } from "./execution/types.js";
-import { AgentOptions } from "./agent-options.js";
+import { AgentConstructionOptions, AgentOptions } from "./agent-options.js";
 import { AgentRun } from "./session/run.js";
 import { SessionHandle } from "./agent-session-entry.js";
 
 //#region src/agent.d.ts
 declare class Agent {
   #private;
-  readonly description?: string;
-  readonly name?: string;
-  constructor(options: AgentOptions);
+  readonly host: AgentHost;
+  readonly namespace?: string;
+  constructor(options: AgentConstructionOptions);
   send(input: AgentInput): Promise<AgentRun>;
   resume(runId: string): Promise<AgentRun | null>;
   session(key: string): SessionHandle;

package/dist/agent.js

@@ -1,46 +1,33 @@
 import { executionHost } from "./execution/host.js";
-import { cancelDurableChildRuns } from "./agent-child-runs.js";
-import { supportsBackgroundSubagents } from "./agent-host-capabilities.js";
 import { sessionStoreForHost } from "./agent-host-session-store.js";
-import { parentSessionNamespace, stableAgentNamespace } from "./agent-namespace.js";
-import { assertAgentOptions, hasRuntimeModel } from "./agent-options.js";
+import { stableAgentNamespace } from "./agent-namespace.js";
+import { assertAgentOptions, hasLanguageModel, hasRuntimeModel } from "./agent-options.js";
 import { resumeAgentRun } from "./agent-resume.js";
-import { assertSubagents } from "./agent-validation.js";
-import { ChildSessionCleanups } from "./child-session-cleanups.js";
 import { createInMemoryExecutionHost } from "./execution/memory.js";
 import { createLlm } from "./llm.js";
 import { AgentSession } from "./session/session.js";
-import { createSubagentTools } from "./subagents.js";
 //#region src/agent.ts
-var Agent = class Agent {
+var Agent = class {
 	#baseTools;
 	#llm;
 	#modelOptions;
-	#childSessionCleanups = new ChildSessionCleanups();
-	#sessionGenerations = /* @__PURE__ */ new Map();
 	#sessions = /* @__PURE__ */ new Map();
 	#sessionNamespace;
 	#store;
 	#host;
 	#plugins;
-	#subagents;
-	description;
-	name;
+	host;
+	namespace;
 	constructor(options) {
 		assertAgentOptions(options);
-		this.description = options.description;
-		this.name = options.name;
-		this.#sessionNamespace = stableAgentNamespace({
-			name: options.name,
-			namespace: options.namespace
-		});
+		this.namespace = options.namespace;
+		this.#sessionNamespace = stableAgentNamespace({ namespace: options.namespace });
 		this.#host = options.host ?? createInMemoryExecutionHost();
+		this.host = this.#host;
 		this.#store = sessionStoreForHost(this.#host);
 		this.#plugins = options.plugins ?? [];
-		assertSubagents(options, Agent, hasRuntimeModel(options));
-		this.#subagents = hasRuntimeModel(options) ? [] : options.subagents ?? [];
 		if (hasRuntimeModel(options)) this.#llm = options.model;
-		else {
+		else if (hasLanguageModel(options)) {
 			this.#baseTools = options.tools;
 			this.#modelOptions = {
 				instructions: options.instructions,
@@ -59,8 +46,7 @@ var Agent = class Agent {
 			host,
 			ownerNamespace: this.#sessionNamespace,
 			resumeNotification: (notification) => this.#resumeNotification(notification),
-			runId,
-			subagents: this.#subagents
+			runId
 		});
 	}
 	session(key) {
@@ -70,16 +56,7 @@ var Agent = class Agent {
 		const existing = this.#sessions.get(key);
 		if (existing) return existing;
 		let session;
-		const getSession = () => {
-			if (!session) throw new Error("Agent session is not initialized.");
-			return session;
-		};
-		const parentAgentNamespace = parentSessionNamespace({
-			generation: this.#sessionGenerations.get(key) ?? 0,
-			sessionKey: key,
-			sessionNamespace: this.#sessionNamespace
-		});
-		session = new AgentSession(this.#llm ?? createLlm(this.#createLlmOptionsForSession(key, parentAgentNamespace, (input, placement) => getSession().enqueueRuntimeInput(input, placement), (event) => getSession().emitObserverEvent(event), (input, options) => getSession().notify(input, options), () => getSession().currentTurnId(), () => parentAgentNamespace)), {
+		session = new AgentSession(this.#llm ?? createLlm(this.#createLlmOptionsForSession()), {
 			key,
 			store: this.#store
 		}, this.#plugins, { executionHost: executionHost(this.#host) });
@@ -88,18 +65,15 @@ var Agent = class Agent {
 			publicHandle: {
 				delete: async () => {
 					session.kill();
-					await this.#cancelDurableChildRunsBeforeLocalCleanup(key, parentAgentNamespace);
 					this.#evictSessionHandle(key);
 					await session.delete();
-					await this.#childSessionCleanups.delete(key);
 				},
-				interrupt: () => session.interrupt(),
-				kill: async () => {
+				dispose: () => {
 					session.kill();
-					await this.#cancelDurableChildRunsBeforeLocalCleanup(key, parentAgentNamespace);
 					this.#evictSessionHandle(key);
-					await this.#childSessionCleanups.delete(key);
+					return Promise.resolve();
 				},
+				interrupt: () => session.interrupt(),
 				send: (input) => session.send(input),
 				steer: (input) => session.steer(input)
 			}
@@ -107,48 +81,20 @@ var Agent = class Agent {
 		this.#sessions.set(key, entry);
 		return entry;
 	}
-	async #cancelDurableChildRunsBeforeLocalCleanup(key, parentAgentNamespace) {
-		try {
-			await cancelDurableChildRuns(this.#host, parentAgentNamespace);
-		} catch (error) {
-			this.#evictSessionHandle(key);
-			throw error;
-		}
-	}
 	#evictSessionHandle(key) {
 		this.#sessions.delete(key);
-		this.#sessionGenerations.set(key, (this.#sessionGenerations.get(key) ?? 0) + 1);
 	}
 	#resumeNotification(notification) {
 		return this.#sessionEntry(notification.sessionKey).notify(notification.input, { observerEvents: notification.observerEvents });
 	}
-	#createLlmOptionsForSession(key, parentAgentNamespace, enqueueRuntimeInput, emitObserverEvent, notify, currentBackgroundGroupId, currentRunId) {
+	#createLlmOptionsForSession() {
 		const modelOptions = this.#modelOptions;
 		if (!modelOptions) throw new Error("Agent: missing model options.");
-		const hostExecution = executionHost(this.#host);
-		const tools = this.#subagents.length === 0 ? this.#baseTools : {
-			...this.#baseTools,
-			...createSubagentTools({
-				backgroundSubagents: supportsBackgroundSubagents(this.#host, hostExecution),
-				executionHost: hostExecution,
-				parentAgentNamespace,
-				parentSession: {
-					currentBackgroundGroupId,
-					currentRunId,
-					emitObserverEvent,
-					enqueueRuntimeInput,
-					notify
-				},
-				parentSessionKey: key,
-				registerChildSession: (sessionKey, cleanup) => this.#childSessionCleanups.register(sessionKey, cleanup),
-				subagents: this.#subagents
-			})
-		};
 		return {
 			instructions: modelOptions.instructions,
 			model: modelOptions.model,
 			toolChoice: modelOptions.toolChoice,
-			tools
+			tools: this.#baseTools
 		};
 	}
 };

package/dist/agent.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent.js","names":["#baseTools","#llm","#modelOptions","#childSessionCleanups","#sessionGenerations","#sessions","#sessionNamespace","#store","#host","#plugins","#subagents","#resumeNotification","#sessionEntry","#createLlmOptionsForSession","#cancelDurableChildRunsBeforeLocalCleanup","#evictSessionHandle"],"sources":["../src/agent.ts"],"sourcesContent":["import type { ToolSet } from \"ai\";\nimport { cancelDurableChildRuns } from \"./agent-child-runs\";\nimport { supportsBackgroundSubagents } from \"./agent-host-capabilities\";\nimport { sessionStoreForHost } from \"./agent-host-session-store\";\nimport {\n  parentSessionNamespace,\n  stableAgentNamespace,\n} from \"./agent-namespace\";\nimport {\n  type AgentModelOptions,\n  type AgentOptions,\n  assertAgentOptions,\n  hasRuntimeModel,\n} from \"./agent-options\";\nimport { resumeAgentRun } from \"./agent-resume\";\nimport type { AgentSessionEntry, SessionHandle } from \"./agent-session-entry\";\nimport { assertSubagents } from \"./agent-validation\";\nimport { ChildSessionCleanups } from \"./child-session-cleanups\";\nimport { executionHost } from \"./execution/host\";\nimport { createInMemoryExecutionHost } from \"./execution/memory\";\nimport type { AgentHost, NotificationRecord } from \"./execution/types\";\nimport { createLlm, type RuntimeLlm } from \"./llm\";\nimport type { AgentPlugin } from \"./plugins\";\nimport type { UserInput } from \"./session/events\";\nimport type { AgentRun } from \"./session/run\";\nimport {\n  type AgentInput,\n  AgentSession,\n  type NotifyOptions,\n} from \"./session/session\";\nimport type { SessionStore } from \"./session/store/types\";\nimport { createSubagentTools } from \"./subagents\";\n\nexport type { AgentOptions } from \"./agent-options\";\nexport type { SessionHandle } from \"./agent-session-entry\";\nexport type { AgentHost } from \"./execution/types\";\n\nexport class Agent {\n  readonly #baseTools?: ToolSet;\n  readonly #llm?: RuntimeLlm;\n  readonly #modelOptions?: AgentModelOptions;\n  readonly #childSessionCleanups = new ChildSessionCleanups();\n  readonly #sessionGenerations = new Map<string, number>();\n  readonly #sessions = new Map<string, AgentSessionEntry>();\n  readonly #sessionNamespace: string;\n  readonly #store: SessionStore;\n  readonly #host: AgentHost;\n  readonly #plugins: readonly AgentPlugin[];\n  readonly #subagents: readonly Agent[];\n  readonly description?: string;\n  readonly name?: string;\n\n  constructor(options: AgentOptions) {\n    assertAgentOptions(options);\n\n    this.description = options.description;\n    this.name = options.name;\n    this.#sessionNamespace = stableAgentNamespace({\n      name: options.name,\n      namespace: options.namespace,\n    });\n    this.#host = options.host ?? createInMemoryExecutionHost();\n    this.#store = sessionStoreForHost(this.#host);\n    this.#plugins = options.plugins ?? [];\n    assertSubagents(options, Agent, hasRuntimeModel(options));\n    this.#subagents = hasRuntimeModel(options) ? [] : (options.subagents ?? []);\n    if (hasRuntimeModel(options)) {\n      this.#llm = options.model;\n    } else {\n      this.#baseTools = options.tools;\n      this.#modelOptions = {\n        instructions: options.instructions,\n        model: options.model,\n        toolChoice: options.toolChoice,\n      };\n    }\n  }\n\n  send(input: AgentInput): Promise<AgentRun> {\n    return this.session(\"default\").send(input);\n  }\n\n  async resume(runId: string): Promise<AgentRun | null> {\n    const host = executionHost(this.#host);\n    if (!host) {\n      throw new Error(\"Agent host does not support durable run resume.\");\n    }\n\n    return await resumeAgentRun({\n      host,\n      ownerNamespace: this.#sessionNamespace,\n      resumeNotification: (notification) =>\n        this.#resumeNotification(notification),\n      runId,\n      subagents: this.#subagents,\n    });\n  }\n\n  session(key: string): SessionHandle {\n    return this.#sessionEntry(key).publicHandle;\n  }\n\n  #sessionEntry(key: string): AgentSessionEntry {\n    const existing = this.#sessions.get(key);\n    if (existing) {\n      return existing;\n    }\n\n    let session: AgentSession | undefined;\n    const getSession = () => {\n      if (!session) {\n        throw new Error(\"Agent session is not initialized.\");\n      }\n      return session;\n    };\n    const parentAgentNamespace = parentSessionNamespace({\n      generation: this.#sessionGenerations.get(key) ?? 0,\n      sessionKey: key,\n      sessionNamespace: this.#sessionNamespace,\n    });\n    const llm =\n      this.#llm ??\n      createLlm(\n        this.#createLlmOptionsForSession(\n          key,\n          parentAgentNamespace,\n          (input: UserInput, placement?: \"turn-start\") =>\n            getSession().enqueueRuntimeInput(input, placement),\n          (event) => getSession().emitObserverEvent(event),\n          (input: UserInput, options?: NotifyOptions) =>\n            getSession().notify(input, options),\n          () => getSession().currentTurnId(),\n          () => parentAgentNamespace\n        )\n      );\n    session = new AgentSession(\n      llm,\n      { key, store: this.#store },\n      this.#plugins,\n      {\n        executionHost: executionHost(this.#host),\n      }\n    );\n    const publicHandle: SessionHandle = {\n      delete: async () => {\n        session.kill();\n        await this.#cancelDurableChildRunsBeforeLocalCleanup(\n          key,\n          parentAgentNamespace\n        );\n        this.#evictSessionHandle(key);\n        await session.delete();\n        await this.#childSessionCleanups.delete(key);\n      },\n      interrupt: () => session.interrupt(),\n      kill: async () => {\n        session.kill();\n        await this.#cancelDurableChildRunsBeforeLocalCleanup(\n          key,\n          parentAgentNamespace\n        );\n        this.#evictSessionHandle(key);\n        await this.#childSessionCleanups.delete(key);\n      },\n      send: (input) => session.send(input),\n      steer: (input) => session.steer(input),\n    };\n    const entry: AgentSessionEntry = {\n      notify: (input, options) => session.notify(input, options),\n      publicHandle,\n    };\n    this.#sessions.set(key, entry);\n    return entry;\n  }\n\n  async #cancelDurableChildRunsBeforeLocalCleanup(\n    key: string,\n    parentAgentNamespace: string\n  ): Promise<void> {\n    try {\n      await cancelDurableChildRuns(this.#host, parentAgentNamespace);\n    } catch (error) {\n      this.#evictSessionHandle(key);\n      throw error;\n    }\n  }\n\n  #evictSessionHandle(key: string): void {\n    this.#sessions.delete(key);\n    this.#sessionGenerations.set(\n      key,\n      (this.#sessionGenerations.get(key) ?? 0) + 1\n    );\n  }\n\n  #resumeNotification(notification: NotificationRecord): Promise<AgentRun> {\n    return this.#sessionEntry(notification.sessionKey).notify(\n      notification.input,\n      { observerEvents: notification.observerEvents }\n    );\n  }\n\n  #createLlmOptionsForSession(\n    key: string,\n    parentAgentNamespace: string,\n    enqueueRuntimeInput: AgentSession[\"enqueueRuntimeInput\"],\n    emitObserverEvent: AgentSession[\"emitObserverEvent\"],\n    notify: (input: UserInput, options?: NotifyOptions) => Promise<AgentRun>,\n    currentBackgroundGroupId: () => string | undefined,\n    currentRunId: () => string | undefined\n  ): Parameters<typeof createLlm>[0] {\n    const modelOptions = this.#modelOptions;\n    if (!modelOptions) {\n      throw new Error(\"Agent: missing model options.\");\n    }\n    const hostExecution = executionHost(this.#host);\n    const tools =\n      this.#subagents.length === 0\n        ? this.#baseTools\n        : {\n            ...this.#baseTools,\n            ...createSubagentTools({\n              backgroundSubagents: supportsBackgroundSubagents(\n                this.#host,\n                hostExecution\n              ),\n              executionHost: hostExecution,\n              parentAgentNamespace,\n              parentSession: {\n                currentBackgroundGroupId,\n                currentRunId,\n                emitObserverEvent,\n                enqueueRuntimeInput,\n                notify,\n              },\n              parentSessionKey: key,\n              registerChildSession: (sessionKey, cleanup) =>\n                this.#childSessionCleanups.register(sessionKey, cleanup),\n              subagents: this.#subagents,\n            }),\n          };\n\n    return {\n      instructions: modelOptions.instructions,\n      model: modelOptions.model,\n      toolChoice: modelOptions.toolChoice,\n      tools,\n    };\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;AAqCA,IAAa,QAAb,MAAa,MAAM;CACjB;CACA;CACA;CACA,wBAAiC,IAAI,qBAAqB;CAC1D,sCAA+B,IAAI,IAAoB;CACvD,4BAAqB,IAAI,IAA+B;CACxD;CACA;CACA;CACA;CACA;CACA;CACA;CAEA,YAAY,SAAuB;EACjC,mBAAmB,OAAO;EAE1B,KAAK,cAAc,QAAQ;EAC3B,KAAK,OAAO,QAAQ;EACpB,KAAKM,oBAAoB,qBAAqB;GAC5C,MAAM,QAAQ;GACd,WAAW,QAAQ;EACrB,CAAC;EACD,KAAKE,QAAQ,QAAQ,QAAQ,4BAA4B;EACzD,KAAKD,SAAS,oBAAoB,KAAKC,KAAK;EAC5C,KAAKC,WAAW,QAAQ,WAAW,CAAC;EACpC,gBAAgB,SAAS,OAAO,gBAAgB,OAAO,CAAC;EACxD,KAAKC,aAAa,gBAAgB,OAAO,IAAI,CAAC,IAAK,QAAQ,aAAa,CAAC;EACzE,IAAI,gBAAgB,OAAO,GACzB,KAAKT,OAAO,QAAQ;OACf;GACL,KAAKD,aAAa,QAAQ;GAC1B,KAAKE,gBAAgB;IACnB,cAAc,QAAQ;IACtB,OAAO,QAAQ;IACf,YAAY,QAAQ;GACtB;EACF;CACF;CAEA,KAAK,OAAsC;EACzC,OAAO,KAAK,QAAQ,SAAS,EAAE,KAAK,KAAK;CAC3C;CAEA,MAAM,OAAO,OAAyC;EACpD,MAAM,OAAO,cAAc,KAAKM,KAAK;EACrC,IAAI,CAAC,MACH,MAAM,IAAI,MAAM,iDAAiD;EAGnE,OAAO,MAAM,eAAe;GAC1B;GACA,gBAAgB,KAAKF;GACrB,qBAAqB,iBACnB,KAAKK,oBAAoB,YAAY;GACvC;GACA,WAAW,KAAKD;EAClB,CAAC;CACH;CAEA,QAAQ,KAA4B;EAClC,OAAO,KAAKE,cAAc,GAAG,EAAE;CACjC;CAEA,cAAc,KAAgC;EAC5C,MAAM,WAAW,KAAKP,UAAU,IAAI,GAAG;EACvC,IAAI,UACF,OAAO;EAGT,IAAI;EACJ,MAAM,mBAAmB;GACvB,IAAI,CAAC,SACH,MAAM,IAAI,MAAM,mCAAmC;GAErD,OAAO;EACT;EACA,MAAM,uBAAuB,uBAAuB;GAClD,YAAY,KAAKD,oBAAoB,IAAI,GAAG,KAAK;GACjD,YAAY;GACZ,kBAAkB,KAAKE;EACzB,CAAC;EAgBD,UAAU,IAAI,aAdZ,KAAKL,QACL,UACE,KAAKY,4BACH,KACA,uBACC,OAAkB,cACjB,WAAW,EAAE,oBAAoB,OAAO,SAAS,IAClD,UAAU,WAAW,EAAE,kBAAkB,KAAK,IAC9C,OAAkB,YACjB,WAAW,EAAE,OAAO,OAAO,OAAO,SAC9B,WAAW,EAAE,cAAc,SAC3B,oBACR,CACF,GAGA;GAAE;GAAK,OAAO,KAAKN;EAAO,GAC1B,KAAKE,UACL,EACE,eAAe,cAAc,KAAKD,KAAK,EACzC,CACF;EAyBA,MAAM,QAA2B;GAC/B,SAAS,OAAO,YAAY,QAAQ,OAAO,OAAO,OAAO;GACzD,cAAA;IAzBA,QAAQ,YAAY;KAClB,QAAQ,KAAK;KACb,MAAM,KAAKM,0CACT,KACA,oBACF;KACA,KAAKC,oBAAoB,GAAG;KAC5B,MAAM,QAAQ,OAAO;KACrB,MAAM,KAAKZ,sBAAsB,OAAO,GAAG;IAC7C;IACA,iBAAiB,QAAQ,UAAU;IACnC,MAAM,YAAY;KAChB,QAAQ,KAAK;KACb,MAAM,KAAKW,0CACT,KACA,oBACF;KACA,KAAKC,oBAAoB,GAAG;KAC5B,MAAM,KAAKZ,sBAAsB,OAAO,GAAG;IAC7C;IACA,OAAO,UAAU,QAAQ,KAAK,KAAK;IACnC,QAAQ,UAAU,QAAQ,MAAM,KAAK;GAI1B;EACb;EACA,KAAKE,UAAU,IAAI,KAAK,KAAK;EAC7B,OAAO;CACT;CAEA,MAAMS,0CACJ,KACA,sBACe;EACf,IAAI;GACF,MAAM,uBAAuB,KAAKN,OAAO,oBAAoB;EAC/D,SAAS,OAAO;GACd,KAAKO,oBAAoB,GAAG;GAC5B,MAAM;EACR;CACF;CAEA,oBAAoB,KAAmB;EACrC,KAAKV,UAAU,OAAO,GAAG;EACzB,KAAKD,oBAAoB,IACvB,MACC,KAAKA,oBAAoB,IAAI,GAAG,KAAK,KAAK,CAC7C;CACF;CAEA,oBAAoB,cAAqD;EACvE,OAAO,KAAKQ,cAAc,aAAa,UAAU,EAAE,OACjD,aAAa,OACb,EAAE,gBAAgB,aAAa,eAAe,CAChD;CACF;CAEA,4BACE,KACA,sBACA,qBACA,mBACA,QACA,0BACA,cACiC;EACjC,MAAM,eAAe,KAAKV;EAC1B,IAAI,CAAC,cACH,MAAM,IAAI,MAAM,+BAA+B;EAEjD,MAAM,gBAAgB,cAAc,KAAKM,KAAK;EAC9C,MAAM,QACJ,KAAKE,WAAW,WAAW,IACvB,KAAKV,aACL;GACE,GAAG,KAAKA;GACR,GAAG,oBAAoB;IACrB,qBAAqB,4BACnB,KAAKQ,OACL,aACF;IACA,eAAe;IACf;IACA,eAAe;KACb;KACA;KACA;KACA;KACA;IACF;IACA,kBAAkB;IAClB,uBAAuB,YAAY,YACjC,KAAKL,sBAAsB,SAAS,YAAY,OAAO;IACzD,WAAW,KAAKO;GAClB,CAAC;EACH;EAEN,OAAO;GACL,cAAc,aAAa;GAC3B,OAAO,aAAa;GACpB,YAAY,aAAa;GACzB;EACF;CACF;AACF"}
+{"version":3,"file":"agent.js","names":["#baseTools","#llm","#modelOptions","#sessions","#sessionNamespace","#store","#host","#plugins","#resumeNotification","#sessionEntry","#createLlmOptionsForSession","#evictSessionHandle"],"sources":["../src/agent.ts"],"sourcesContent":["import type { ToolSet } from \"ai\";\nimport { sessionStoreForHost } from \"./agent-host-session-store\";\nimport { stableAgentNamespace } from \"./agent-namespace\";\nimport {\n  type AgentConstructionOptions,\n  type AgentModelOptions,\n  assertAgentOptions,\n  hasLanguageModel,\n  hasRuntimeModel,\n} from \"./agent-options\";\nimport { resumeAgentRun } from \"./agent-resume\";\nimport type { AgentSessionEntry, SessionHandle } from \"./agent-session-entry\";\nimport { executionHost } from \"./execution/host\";\nimport { createInMemoryExecutionHost } from \"./execution/memory\";\nimport type { AgentHost, NotificationRecord } from \"./execution/types\";\nimport { createLlm, type RuntimeLlm } from \"./llm\";\nimport type { AgentPlugin } from \"./plugins\";\nimport type { AgentRun } from \"./session/run\";\nimport { type AgentInput, AgentSession } from \"./session/session\";\nimport type { SessionStore } from \"./session/store/types\";\n\nexport type { AgentOptions } from \"./agent-options\";\nexport type { SessionHandle } from \"./agent-session-entry\";\nexport type { AgentHost } from \"./execution/types\";\n\nexport class Agent {\n  readonly #baseTools?: ToolSet;\n  readonly #llm?: RuntimeLlm;\n  readonly #modelOptions?: AgentModelOptions;\n  readonly #sessions = new Map<string, AgentSessionEntry>();\n  readonly #sessionNamespace: string;\n  readonly #store: SessionStore;\n  readonly #host: AgentHost;\n  readonly #plugins: readonly AgentPlugin[];\n  readonly host: AgentHost;\n  readonly namespace?: string;\n  constructor(options: AgentConstructionOptions) {\n    assertAgentOptions(options);\n\n    this.namespace = options.namespace;\n    this.#sessionNamespace = stableAgentNamespace({\n      namespace: options.namespace,\n    });\n    this.#host = options.host ?? createInMemoryExecutionHost();\n    this.host = this.#host;\n    this.#store = sessionStoreForHost(this.#host);\n    this.#plugins = options.plugins ?? [];\n    if (hasRuntimeModel(options)) {\n      this.#llm = options.model;\n    } else if (hasLanguageModel(options)) {\n      this.#baseTools = options.tools;\n      this.#modelOptions = {\n        instructions: options.instructions,\n        model: options.model,\n        toolChoice: options.toolChoice,\n      };\n    }\n  }\n\n  send(input: AgentInput): Promise<AgentRun> {\n    return this.session(\"default\").send(input);\n  }\n\n  async resume(runId: string): Promise<AgentRun | null> {\n    const host = executionHost(this.#host);\n    if (!host) {\n      throw new Error(\"Agent host does not support durable run resume.\");\n    }\n\n    return await resumeAgentRun({\n      host,\n      ownerNamespace: this.#sessionNamespace,\n      resumeNotification: (notification) =>\n        this.#resumeNotification(notification),\n      runId,\n    });\n  }\n\n  session(key: string): SessionHandle {\n    return this.#sessionEntry(key).publicHandle;\n  }\n\n  #sessionEntry(key: string): AgentSessionEntry {\n    const existing = this.#sessions.get(key);\n    if (existing) {\n      return existing;\n    }\n\n    let session: AgentSession | undefined;\n    const llm = this.#llm ?? createLlm(this.#createLlmOptionsForSession());\n    session = new AgentSession(\n      llm,\n      { key, store: this.#store },\n      this.#plugins,\n      {\n        executionHost: executionHost(this.#host),\n      }\n    );\n    const publicHandle: SessionHandle = {\n      delete: async () => {\n        session.kill();\n        this.#evictSessionHandle(key);\n        await session.delete();\n      },\n      dispose: () => {\n        session.kill();\n        this.#evictSessionHandle(key);\n        return Promise.resolve();\n      },\n      interrupt: () => session.interrupt(),\n      send: (input) => session.send(input),\n      steer: (input) => session.steer(input),\n    };\n    const entry: AgentSessionEntry = {\n      notify: (input, options) => session.notify(input, options),\n      publicHandle,\n    };\n    this.#sessions.set(key, entry);\n    return entry;\n  }\n\n  #evictSessionHandle(key: string): void {\n    this.#sessions.delete(key);\n  }\n\n  #resumeNotification(notification: NotificationRecord): Promise<AgentRun> {\n    return this.#sessionEntry(notification.sessionKey).notify(\n      notification.input,\n      { observerEvents: notification.observerEvents }\n    );\n  }\n\n  #createLlmOptionsForSession(): Parameters<typeof createLlm>[0] {\n    const modelOptions = this.#modelOptions;\n    if (!modelOptions) {\n      throw new Error(\"Agent: missing model options.\");\n    }\n\n    return {\n      instructions: modelOptions.instructions,\n      model: modelOptions.model,\n      toolChoice: modelOptions.toolChoice,\n      tools: this.#baseTools,\n    };\n  }\n}\n"],"mappings":";;;;;;;;;AAyBA,IAAa,QAAb,MAAmB;CACjB;CACA;CACA;CACA,4BAAqB,IAAI,IAA+B;CACxD;CACA;CACA;CACA;CACA;CACA;CACA,YAAY,SAAmC;EAC7C,mBAAmB,OAAO;EAE1B,KAAK,YAAY,QAAQ;EACzB,KAAKI,oBAAoB,qBAAqB,EAC5C,WAAW,QAAQ,UACrB,CAAC;EACD,KAAKE,QAAQ,QAAQ,QAAQ,4BAA4B;EACzD,KAAK,OAAO,KAAKA;EACjB,KAAKD,SAAS,oBAAoB,KAAKC,KAAK;EAC5C,KAAKC,WAAW,QAAQ,WAAW,CAAC;EACpC,IAAI,gBAAgB,OAAO,GACzB,KAAKN,OAAO,QAAQ;OACf,IAAI,iBAAiB,OAAO,GAAG;GACpC,KAAKD,aAAa,QAAQ;GAC1B,KAAKE,gBAAgB;IACnB,cAAc,QAAQ;IACtB,OAAO,QAAQ;IACf,YAAY,QAAQ;GACtB;EACF;CACF;CAEA,KAAK,OAAsC;EACzC,OAAO,KAAK,QAAQ,SAAS,EAAE,KAAK,KAAK;CAC3C;CAEA,MAAM,OAAO,OAAyC;EACpD,MAAM,OAAO,cAAc,KAAKI,KAAK;EACrC,IAAI,CAAC,MACH,MAAM,IAAI,MAAM,iDAAiD;EAGnE,OAAO,MAAM,eAAe;GAC1B;GACA,gBAAgB,KAAKF;GACrB,qBAAqB,iBACnB,KAAKI,oBAAoB,YAAY;GACvC;EACF,CAAC;CACH;CAEA,QAAQ,KAA4B;EAClC,OAAO,KAAKC,cAAc,GAAG,EAAE;CACjC;CAEA,cAAc,KAAgC;EAC5C,MAAM,WAAW,KAAKN,UAAU,IAAI,GAAG;EACvC,IAAI,UACF,OAAO;EAGT,IAAI;EAEJ,UAAU,IAAI,aADF,KAAKF,QAAQ,UAAU,KAAKS,4BAA4B,CAAC,GAGnE;GAAE;GAAK,OAAO,KAAKL;EAAO,GAC1B,KAAKE,UACL,EACE,eAAe,cAAc,KAAKD,KAAK,EACzC,CACF;EAgBA,MAAM,QAA2B;GAC/B,SAAS,OAAO,YAAY,QAAQ,OAAO,OAAO,OAAO;GACzD,cAAA;IAhBA,QAAQ,YAAY;KAClB,QAAQ,KAAK;KACb,KAAKK,oBAAoB,GAAG;KAC5B,MAAM,QAAQ,OAAO;IACvB;IACA,eAAe;KACb,QAAQ,KAAK;KACb,KAAKA,oBAAoB,GAAG;KAC5B,OAAO,QAAQ,QAAQ;IACzB;IACA,iBAAiB,QAAQ,UAAU;IACnC,OAAO,UAAU,QAAQ,KAAK,KAAK;IACnC,QAAQ,UAAU,QAAQ,MAAM,KAAK;GAI1B;EACb;EACA,KAAKR,UAAU,IAAI,KAAK,KAAK;EAC7B,OAAO;CACT;CAEA,oBAAoB,KAAmB;EACrC,KAAKA,UAAU,OAAO,GAAG;CAC3B;CAEA,oBAAoB,cAAqD;EACvE,OAAO,KAAKM,cAAc,aAAa,UAAU,EAAE,OACjD,aAAa,OACb,EAAE,gBAAgB,aAAa,eAAe,CAChD;CACF;CAEA,8BAA+D;EAC7D,MAAM,eAAe,KAAKP;EAC1B,IAAI,CAAC,cACH,MAAM,IAAI,MAAM,+BAA+B;EAGjD,OAAO;GACL,cAAc,aAAa;GAC3B,OAAO,aAAa;GACpB,YAAY,aAAa;GACzB,OAAO,KAAKF;EACd;CACF;AACF"}

package/dist/cloudflare/cloudflare-agent-context.d.ts

@@ -0,0 +1,40 @@
+import { ExecutionHost } from "../execution/types.js";
+import { CloudflareAlarmDrainBudget } from "./cloudflare-alarm-budget.js";
+import { CloudflareDurableObjectStorage } from "./cloudflare-host.js";
+import { CloudflareAlarmAgent, CloudflareAlarmDrainSummary } from "./cloudflare-alarm-drainer.js";
+
+//#region src/cloudflare/cloudflare-agent-context.d.ts
+type MaybePromise<T> = Promise<T> | T;
+interface CloudflareAgentContextFactoryOptions<Env> {
+  readonly env: Env;
+  readonly host: ExecutionHost;
+  readonly prefix: string;
+  readonly storage: CloudflareDurableObjectStorage;
+}
+interface CloudflareAgentContextPrefixOptions<Env> {
+  readonly env: Env;
+  readonly storage: CloudflareDurableObjectStorage;
+}
+interface CloudflareAgentContextOptions<Env, Agent extends CloudflareAlarmAgent> {
+  readonly createAgent: (options: CloudflareAgentContextFactoryOptions<Env>) => Agent;
+  readonly defaultPrefix?: string;
+  readonly env: Env;
+  readonly readPrefix?: (options: CloudflareAgentContextPrefixOptions<Env>) => MaybePromise<string | undefined>;
+  readonly storage: CloudflareDurableObjectStorage;
+}
+interface CloudflareAgentContext<Agent extends CloudflareAlarmAgent> {
+  agent(prefix?: string): Agent;
+  drainAlarm(budget?: CloudflareAlarmDrainBudget): Promise<CloudflareAlarmDrainSummary>;
+  host(prefix?: string): ExecutionHost;
+  readonly storage: CloudflareDurableObjectStorage;
+}
+declare function createCloudflareAgentContext<Env, Agent extends CloudflareAlarmAgent>({
+  createAgent,
+  defaultPrefix,
+  env,
+  readPrefix,
+  storage
+}: CloudflareAgentContextOptions<Env, Agent>): CloudflareAgentContext<Agent>;
+//#endregion
+export { CloudflareAgentContext, CloudflareAgentContextFactoryOptions, CloudflareAgentContextOptions, CloudflareAgentContextPrefixOptions, createCloudflareAgentContext };
+//# sourceMappingURL=cloudflare-agent-context.d.ts.map

package/dist/cloudflare/cloudflare-agent-context.js

@@ -0,0 +1,37 @@
+import { createCloudflareDurableObjectHost } from "./cloudflare-host.js";
+import { drainCloudflareAlarm } from "./cloudflare-alarm-drainer.js";
+//#region src/cloudflare/cloudflare-agent-context.ts
+const defaultContextPrefix = "pss-runtime";
+function createCloudflareAgentContext({ createAgent, defaultPrefix = defaultContextPrefix, env, readPrefix, storage }) {
+	const createHost = (prefix = defaultPrefix) => createCloudflareDurableObjectHost({
+		prefix,
+		storage
+	});
+	const createContextAgent = (prefix = defaultPrefix) => createAgent({
+		env,
+		host: createHost(prefix),
+		prefix,
+		storage
+	});
+	return {
+		agent: createContextAgent,
+		drainAlarm: async (budget) => {
+			const prefix = await readPrefix?.({
+				env,
+				storage
+			}) ?? defaultPrefix;
+			return await drainCloudflareAlarm({
+				agent: createContextAgent(prefix),
+				...budget,
+				prefix,
+				storage
+			});
+		},
+		host: createHost,
+		storage
+	};
+}
+//#endregion
+export { createCloudflareAgentContext };
+
+//# sourceMappingURL=cloudflare-agent-context.js.map

package/dist/cloudflare/cloudflare-agent-context.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cloudflare-agent-context.js","names":[],"sources":["../../src/cloudflare/cloudflare-agent-context.ts"],"sourcesContent":["import type { ExecutionHost } from \"../execution\";\nimport type { CloudflareAlarmDrainBudget } from \"./cloudflare-alarm-budget\";\nimport type {\n  CloudflareAlarmAgent,\n  CloudflareAlarmDrainSummary,\n} from \"./cloudflare-alarm-drainer\";\nimport { drainCloudflareAlarm } from \"./cloudflare-alarm-drainer\";\nimport {\n  type CloudflareDurableObjectStorage,\n  createCloudflareDurableObjectHost,\n} from \"./cloudflare-host\";\n\nconst defaultContextPrefix = \"pss-runtime\";\n\ntype MaybePromise<T> = Promise<T> | T;\n\nexport interface CloudflareAgentContextFactoryOptions<Env> {\n  readonly env: Env;\n  readonly host: ExecutionHost;\n  readonly prefix: string;\n  readonly storage: CloudflareDurableObjectStorage;\n}\n\nexport interface CloudflareAgentContextPrefixOptions<Env> {\n  readonly env: Env;\n  readonly storage: CloudflareDurableObjectStorage;\n}\n\nexport interface CloudflareAgentContextOptions<\n  Env,\n  Agent extends CloudflareAlarmAgent,\n> {\n  readonly createAgent: (\n    options: CloudflareAgentContextFactoryOptions<Env>\n  ) => Agent;\n  readonly defaultPrefix?: string;\n  readonly env: Env;\n  readonly readPrefix?: (\n    options: CloudflareAgentContextPrefixOptions<Env>\n  ) => MaybePromise<string | undefined>;\n  readonly storage: CloudflareDurableObjectStorage;\n}\n\nexport interface CloudflareAgentContext<Agent extends CloudflareAlarmAgent> {\n  agent(prefix?: string): Agent;\n  drainAlarm(\n    budget?: CloudflareAlarmDrainBudget\n  ): Promise<CloudflareAlarmDrainSummary>;\n  host(prefix?: string): ExecutionHost;\n  readonly storage: CloudflareDurableObjectStorage;\n}\n\nexport function createCloudflareAgentContext<\n  Env,\n  Agent extends CloudflareAlarmAgent,\n>({\n  createAgent,\n  defaultPrefix = defaultContextPrefix,\n  env,\n  readPrefix,\n  storage,\n}: CloudflareAgentContextOptions<Env, Agent>): CloudflareAgentContext<Agent> {\n  const createHost = (prefix = defaultPrefix) =>\n    createCloudflareDurableObjectHost({ prefix, storage });\n  const createContextAgent = (prefix = defaultPrefix) =>\n    createAgent({\n      env,\n      host: createHost(prefix),\n      prefix,\n      storage,\n    });\n\n  return {\n    agent: createContextAgent,\n    drainAlarm: async (budget) => {\n      const prefix = (await readPrefix?.({ env, storage })) ?? defaultPrefix;\n      return await drainCloudflareAlarm({\n        agent: createContextAgent(prefix),\n        ...budget,\n        prefix,\n        storage,\n      });\n    },\n    host: createHost,\n    storage,\n  };\n}\n"],"mappings":";;;AAYA,MAAM,uBAAuB;AAwC7B,SAAgB,6BAGd,EACA,aACA,gBAAgB,sBAChB,KACA,YACA,WAC2E;CAC3E,MAAM,cAAc,SAAS,kBAC3B,kCAAkC;EAAE;EAAQ;CAAQ,CAAC;CACvD,MAAM,sBAAsB,SAAS,kBACnC,YAAY;EACV;EACA,MAAM,WAAW,MAAM;EACvB;EACA;CACF,CAAC;CAEH,OAAO;EACL,OAAO;EACP,YAAY,OAAO,WAAW;GAC5B,MAAM,SAAU,MAAM,aAAa;IAAE;IAAK;GAAQ,CAAC,KAAM;GACzD,OAAO,MAAM,qBAAqB;IAChC,OAAO,mBAAmB,MAAM;IAChC,GAAG;IACH;IACA;GACF,CAAC;EACH;EACA,MAAM;EACN;CACF;AACF"}

package/dist/cloudflare/cloudflare-alarm-budget.d.ts

@@ -0,0 +1,18 @@
+//#region src/cloudflare/cloudflare-alarm-budget.d.ts
+type CloudflareAlarmContinuationReason = "deadline" | "event-budget" | "failure" | "run-budget" | "session-prompt-budget";
+interface CloudflareAlarmDrainBudget {
+  readonly continuationRunAfterMs?: number;
+  readonly deadlineMs?: number;
+  readonly failureRunAfterMs?: number;
+  readonly maxEvents?: number;
+  readonly maxRuns?: number;
+  readonly maxSessionPrompts?: number;
+  readonly throwOnFailure?: boolean;
+}
+interface FailedScheduledWork {
+  readonly error: string;
+  readonly id: string;
+}
+//#endregion
+export { CloudflareAlarmContinuationReason, CloudflareAlarmDrainBudget, FailedScheduledWork };
+//# sourceMappingURL=cloudflare-alarm-budget.d.ts.map

package/dist/cloudflare/cloudflare-alarm-budget.js

@@ -0,0 +1,77 @@
+//#region src/cloudflare/cloudflare-alarm-budget.ts
+const defaultAlarmDrainBudget = {
+	continuationRunAfterMs: 0,
+	deadlineMs: 3e4,
+	failureRunAfterMs: 1e3,
+	maxEvents: 1e3,
+	maxRuns: 25,
+	maxSessionPrompts: 25,
+	throwOnFailure: false
+};
+function createAlarmDrainState() {
+	return {
+		consumedSessionPrompts: [],
+		droppedEvents: 0,
+		events: [],
+		failedRuns: [],
+		failedSessionPrompts: [],
+		reasons: /* @__PURE__ */ new Set(),
+		resumedRuns: [],
+		runAttempts: 0,
+		sessionPromptAttempts: 0
+	};
+}
+function normalizeAlarmDrainBudget(budget = {}) {
+	return {
+		continuationRunAfterMs: nonNegativeInteger(budget.continuationRunAfterMs ?? defaultAlarmDrainBudget.continuationRunAfterMs),
+		deadlineMs: nonNegativeInteger(budget.deadlineMs ?? defaultAlarmDrainBudget.deadlineMs),
+		failureRunAfterMs: nonNegativeInteger(budget.failureRunAfterMs ?? defaultAlarmDrainBudget.failureRunAfterMs),
+		maxEvents: nonNegativeInteger(budget.maxEvents ?? defaultAlarmDrainBudget.maxEvents),
+		maxRuns: nonNegativeInteger(budget.maxRuns ?? defaultAlarmDrainBudget.maxRuns),
+		maxSessionPrompts: nonNegativeInteger(budget.maxSessionPrompts ?? defaultAlarmDrainBudget.maxSessionPrompts),
+		startedAt: Date.now(),
+		throwOnFailure: budget.throwOnFailure ?? defaultAlarmDrainBudget.throwOnFailure
+	};
+}
+function eventSlotsRemaining(state, budget) {
+	return Math.max(0, budget.maxEvents - state.events.length);
+}
+function shouldStopForDeadline(budget) {
+	return Date.now() - budget.startedAt >= budget.deadlineMs;
+}
+function shouldStopRuns(state, budget) {
+	if (shouldStopForDeadline(budget)) {
+		state.reasons.add("deadline");
+		return true;
+	}
+	if (shouldStopForEventBudget(state, budget)) return true;
+	if (state.runAttempts >= budget.maxRuns) {
+		state.reasons.add("run-budget");
+		return true;
+	}
+	return false;
+}
+function shouldStopSessionPrompts(state, budget) {
+	if (shouldStopForDeadline(budget)) {
+		state.reasons.add("deadline");
+		return true;
+	}
+	if (shouldStopForEventBudget(state, budget)) return true;
+	if (state.sessionPromptAttempts >= budget.maxSessionPrompts) {
+		state.reasons.add("session-prompt-budget");
+		return true;
+	}
+	return false;
+}
+function shouldStopForEventBudget(state, budget) {
+	if (state.events.length < budget.maxEvents) return false;
+	state.reasons.add("event-budget");
+	return true;
+}
+function nonNegativeInteger(value) {
+	return Math.max(0, Math.floor(value));
+}
+//#endregion
+export { createAlarmDrainState, eventSlotsRemaining, normalizeAlarmDrainBudget, shouldStopRuns, shouldStopSessionPrompts };
+
+//# sourceMappingURL=cloudflare-alarm-budget.js.map

package/dist/cloudflare/cloudflare-alarm-budget.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cloudflare-alarm-budget.js","names":[],"sources":["../../src/cloudflare/cloudflare-alarm-budget.ts"],"sourcesContent":["import type { AgentEvent } from \"../index\";\n\nexport type CloudflareAlarmContinuationReason =\n  | \"deadline\"\n  | \"event-budget\"\n  | \"failure\"\n  | \"run-budget\"\n  | \"session-prompt-budget\";\n\nexport interface CloudflareAlarmDrainBudget {\n  readonly continuationRunAfterMs?: number;\n  readonly deadlineMs?: number;\n  readonly failureRunAfterMs?: number;\n  readonly maxEvents?: number;\n  readonly maxRuns?: number;\n  readonly maxSessionPrompts?: number;\n  readonly throwOnFailure?: boolean;\n}\n\nexport interface FailedScheduledWork {\n  readonly error: string;\n  readonly id: string;\n}\n\nexport interface AlarmDrainState {\n  readonly consumedSessionPrompts: string[];\n  droppedEvents: number;\n  readonly events: AgentEvent[];\n  readonly failedRuns: FailedScheduledWork[];\n  readonly failedSessionPrompts: FailedScheduledWork[];\n  readonly reasons: Set<CloudflareAlarmContinuationReason>;\n  readonly resumedRuns: string[];\n  runAttempts: number;\n  sessionPromptAttempts: number;\n}\n\nexport interface NormalizedAlarmDrainBudget {\n  readonly continuationRunAfterMs: number;\n  readonly deadlineMs: number;\n  readonly failureRunAfterMs: number;\n  readonly maxEvents: number;\n  readonly maxRuns: number;\n  readonly maxSessionPrompts: number;\n  readonly startedAt: number;\n  readonly throwOnFailure: boolean;\n}\n\nconst defaultAlarmDrainBudget = {\n  continuationRunAfterMs: 0,\n  deadlineMs: 30_000,\n  failureRunAfterMs: 1000,\n  maxEvents: 1000,\n  maxRuns: 25,\n  maxSessionPrompts: 25,\n  throwOnFailure: false,\n} satisfies Required<CloudflareAlarmDrainBudget>;\n\nexport function createAlarmDrainState(): AlarmDrainState {\n  return {\n    consumedSessionPrompts: [],\n    droppedEvents: 0,\n    events: [],\n    failedRuns: [],\n    failedSessionPrompts: [],\n    reasons: new Set<CloudflareAlarmContinuationReason>(),\n    resumedRuns: [],\n    runAttempts: 0,\n    sessionPromptAttempts: 0,\n  };\n}\n\nexport function normalizeAlarmDrainBudget(\n  budget: CloudflareAlarmDrainBudget = {}\n): NormalizedAlarmDrainBudget {\n  return {\n    continuationRunAfterMs: nonNegativeInteger(\n      budget.continuationRunAfterMs ??\n        defaultAlarmDrainBudget.continuationRunAfterMs\n    ),\n    deadlineMs: nonNegativeInteger(\n      budget.deadlineMs ?? defaultAlarmDrainBudget.deadlineMs\n    ),\n    failureRunAfterMs: nonNegativeInteger(\n      budget.failureRunAfterMs ?? defaultAlarmDrainBudget.failureRunAfterMs\n    ),\n    maxEvents: nonNegativeInteger(\n      budget.maxEvents ?? defaultAlarmDrainBudget.maxEvents\n    ),\n    maxRuns: nonNegativeInteger(\n      budget.maxRuns ?? defaultAlarmDrainBudget.maxRuns\n    ),\n    maxSessionPrompts: nonNegativeInteger(\n      budget.maxSessionPrompts ?? defaultAlarmDrainBudget.maxSessionPrompts\n    ),\n    startedAt: Date.now(),\n    throwOnFailure:\n      budget.throwOnFailure ?? defaultAlarmDrainBudget.throwOnFailure,\n  };\n}\n\nexport function eventSlotsRemaining(\n  state: AlarmDrainState,\n  budget: NormalizedAlarmDrainBudget\n): number {\n  return Math.max(0, budget.maxEvents - state.events.length);\n}\n\nexport function shouldStopForDeadline(\n  budget: NormalizedAlarmDrainBudget\n): boolean {\n  return Date.now() - budget.startedAt >= budget.deadlineMs;\n}\n\nexport function shouldStopRuns(\n  state: AlarmDrainState,\n  budget: NormalizedAlarmDrainBudget\n): boolean {\n  if (shouldStopForDeadline(budget)) {\n    state.reasons.add(\"deadline\");\n    return true;\n  }\n  if (shouldStopForEventBudget(state, budget)) {\n    return true;\n  }\n  if (state.runAttempts >= budget.maxRuns) {\n    state.reasons.add(\"run-budget\");\n    return true;\n  }\n  return false;\n}\n\nexport function shouldStopSessionPrompts(\n  state: AlarmDrainState,\n  budget: NormalizedAlarmDrainBudget\n): boolean {\n  if (shouldStopForDeadline(budget)) {\n    state.reasons.add(\"deadline\");\n    return true;\n  }\n  if (shouldStopForEventBudget(state, budget)) {\n    return true;\n  }\n  if (state.sessionPromptAttempts >= budget.maxSessionPrompts) {\n    state.reasons.add(\"session-prompt-budget\");\n    return true;\n  }\n  return false;\n}\n\nfunction shouldStopForEventBudget(\n  state: AlarmDrainState,\n  budget: NormalizedAlarmDrainBudget\n): boolean {\n  if (state.events.length < budget.maxEvents) {\n    return false;\n  }\n  state.reasons.add(\"event-budget\");\n  return true;\n}\n\nfunction nonNegativeInteger(value: number): number {\n  return Math.max(0, Math.floor(value));\n}\n"],"mappings":";AA+CA,MAAM,0BAA0B;CAC9B,wBAAwB;CACxB,YAAY;CACZ,mBAAmB;CACnB,WAAW;CACX,SAAS;CACT,mBAAmB;CACnB,gBAAgB;AAClB;AAEA,SAAgB,wBAAyC;CACvD,OAAO;EACL,wBAAwB,CAAC;EACzB,eAAe;EACf,QAAQ,CAAC;EACT,YAAY,CAAC;EACb,sBAAsB,CAAC;EACvB,yBAAS,IAAI,IAAuC;EACpD,aAAa,CAAC;EACd,aAAa;EACb,uBAAuB;CACzB;AACF;AAEA,SAAgB,0BACd,SAAqC,CAAC,GACV;CAC5B,OAAO;EACL,wBAAwB,mBACtB,OAAO,0BACL,wBAAwB,sBAC5B;EACA,YAAY,mBACV,OAAO,cAAc,wBAAwB,UAC/C;EACA,mBAAmB,mBACjB,OAAO,qBAAqB,wBAAwB,iBACtD;EACA,WAAW,mBACT,OAAO,aAAa,wBAAwB,SAC9C;EACA,SAAS,mBACP,OAAO,WAAW,wBAAwB,OAC5C;EACA,mBAAmB,mBACjB,OAAO,qBAAqB,wBAAwB,iBACtD;EACA,WAAW,KAAK,IAAI;EACpB,gBACE,OAAO,kBAAkB,wBAAwB;CACrD;AACF;AAEA,SAAgB,oBACd,OACA,QACQ;CACR,OAAO,KAAK,IAAI,GAAG,OAAO,YAAY,MAAM,OAAO,MAAM;AAC3D;AAEA,SAAgB,sBACd,QACS;CACT,OAAO,KAAK,IAAI,IAAI,OAAO,aAAa,OAAO;AACjD;AAEA,SAAgB,eACd,OACA,QACS;CACT,IAAI,sBAAsB,MAAM,GAAG;EACjC,MAAM,QAAQ,IAAI,UAAU;EAC5B,OAAO;CACT;CACA,IAAI,yBAAyB,OAAO,MAAM,GACxC,OAAO;CAET,IAAI,MAAM,eAAe,OAAO,SAAS;EACvC,MAAM,QAAQ,IAAI,YAAY;EAC9B,OAAO;CACT;CACA,OAAO;AACT;AAEA,SAAgB,yBACd,OACA,QACS;CACT,IAAI,sBAAsB,MAAM,GAAG;EACjC,MAAM,QAAQ,IAAI,UAAU;EAC5B,OAAO;CACT;CACA,IAAI,yBAAyB,OAAO,MAAM,GACxC,OAAO;CAET,IAAI,MAAM,yBAAyB,OAAO,mBAAmB;EAC3D,MAAM,QAAQ,IAAI,uBAAuB;EACzC,OAAO;CACT;CACA,OAAO;AACT;AAEA,SAAS,yBACP,OACA,QACS;CACT,IAAI,MAAM,OAAO,SAAS,OAAO,WAC/B,OAAO;CAET,MAAM,QAAQ,IAAI,cAAc;CAChC,OAAO;AACT;AAEA,SAAS,mBAAmB,OAAuB;CACjD,OAAO,KAAK,IAAI,GAAG,KAAK,MAAM,KAAK,CAAC;AACtC"}

package/dist/cloudflare/cloudflare-alarm-drainer.d.ts

@@ -0,0 +1,45 @@
+import { AgentEvent } from "../session/events.js";
+import { AgentRun } from "../session/run.js";
+import { CloudflareAlarmContinuationReason, CloudflareAlarmDrainBudget, FailedScheduledWork } from "./cloudflare-alarm-budget.js";
+import { CloudflareDurableObjectStorage } from "./cloudflare-host.js";
+
+//#region src/cloudflare/cloudflare-alarm-drainer.d.ts
+interface CloudflareAlarmDrainSummary {
+  readonly consumedSessionPrompts: readonly string[];
+  readonly continuationReasons: readonly CloudflareAlarmContinuationReason[];
+  readonly continuationScheduled: boolean;
+  readonly droppedEvents: number;
+  readonly events: readonly AgentEvent[];
+  readonly failedRuns: readonly FailedScheduledWork[];
+  readonly failedSessionPrompts: readonly FailedScheduledWork[];
+  readonly markers: readonly string[];
+  readonly remainingRuns: number;
+  readonly remainingSessionPrompts: number;
+  readonly resumedRuns: readonly string[];
+}
+interface CloudflareAlarmAgent {
+  resume(runId: string): Promise<AgentRun | null>;
+}
+declare class CloudflareAlarmDrainFailureError extends Error {
+  readonly summary: CloudflareAlarmDrainSummary;
+  constructor(summary: CloudflareAlarmDrainSummary);
+}
+declare function drainCloudflareAlarm({
+  agent,
+  continuationRunAfterMs,
+  deadlineMs,
+  failureRunAfterMs,
+  maxEvents,
+  maxRuns,
+  maxSessionPrompts,
+  prefix,
+  storage,
+  throwOnFailure
+}: {
+  readonly agent: CloudflareAlarmAgent;
+  readonly prefix: string;
+  readonly storage: CloudflareDurableObjectStorage;
+} & CloudflareAlarmDrainBudget): Promise<CloudflareAlarmDrainSummary>;
+//#endregion
+export { CloudflareAlarmAgent, CloudflareAlarmDrainFailureError, CloudflareAlarmDrainSummary, drainCloudflareAlarm };
+//# sourceMappingURL=cloudflare-alarm-drainer.d.ts.map

package/dist/cloudflare/cloudflare-alarm-drainer.js

@@ -0,0 +1,103 @@
+import { createAlarmDrainState, normalizeAlarmDrainBudget, shouldStopRuns, shouldStopSessionPrompts } from "./cloudflare-alarm-budget.js";
+import { listScheduledCloudflareRuns, listScheduledCloudflareSessionPrompts, rescheduleCloudflareAlarm } from "./cloudflare-host.js";
+import { resumeScheduledRun, resumeScheduledSessionPrompt } from "./cloudflare-alarm-work.js";
+//#region src/cloudflare/cloudflare-alarm-drainer.ts
+var CloudflareAlarmDrainFailureError = class extends Error {
+	summary;
+	constructor(summary) {
+		super("Cloudflare alarm drain completed with failed scheduled work.");
+		this.name = "CloudflareAlarmDrainFailureError";
+		this.summary = summary;
+	}
+};
+async function drainCloudflareAlarm({ agent, continuationRunAfterMs, deadlineMs, failureRunAfterMs, maxEvents, maxRuns, maxSessionPrompts, prefix, storage, throwOnFailure }) {
+	const budget = normalizeAlarmDrainBudget({
+		continuationRunAfterMs,
+		deadlineMs,
+		failureRunAfterMs,
+		maxEvents,
+		maxRuns,
+		maxSessionPrompts,
+		throwOnFailure
+	});
+	const state = createAlarmDrainState();
+	await drainRuns({
+		agent,
+		budget,
+		prefix,
+		state,
+		storage
+	});
+	await drainSessionPrompts({
+		agent,
+		budget,
+		prefix,
+		state,
+		storage
+	});
+	const summary = await summarizeDrain({
+		budget,
+		prefix,
+		state,
+		storage
+	});
+	if (budget.throwOnFailure && hasFailures(summary)) throw new CloudflareAlarmDrainFailureError(summary);
+	return summary;
+}
+async function drainRuns(options) {
+	for (const runId of await listScheduledCloudflareRuns(options.storage, { prefix: options.prefix })) {
+		if (shouldStopRuns(options.state, options.budget)) break;
+		options.state.runAttempts += 1;
+		await resumeScheduledRun({
+			...options,
+			runId
+		});
+	}
+}
+async function drainSessionPrompts(options) {
+	const prompts = await listScheduledCloudflareSessionPrompts(options.storage, { prefix: options.prefix });
+	for (const prompt of prompts) {
+		if (shouldStopSessionPrompts(options.state, options.budget)) break;
+		options.state.sessionPromptAttempts += 1;
+		await resumeScheduledSessionPrompt({
+			...options,
+			prompt
+		});
+	}
+}
+async function summarizeDrain({ budget, prefix, state, storage }) {
+	const remainingRuns = await listScheduledCloudflareRuns(storage, { prefix });
+	const remainingPrompts = await listScheduledCloudflareSessionPrompts(storage, { prefix });
+	if (state.failedRuns.length > 0 || state.failedSessionPrompts.length > 0) state.reasons.add("failure");
+	const continuationScheduled = shouldRearm(state, remainingRuns.length, remainingPrompts.length);
+	if (continuationScheduled) await rescheduleCloudflareAlarm(storage, { runAfterMs: state.reasons.has("failure") ? budget.failureRunAfterMs : budget.continuationRunAfterMs });
+	return {
+		consumedSessionPrompts: state.consumedSessionPrompts,
+		continuationReasons: [...state.reasons],
+		continuationScheduled,
+		droppedEvents: state.droppedEvents,
+		events: state.events,
+		failedRuns: state.failedRuns,
+		failedSessionPrompts: state.failedSessionPrompts,
+		markers: markersFor(state.reasons),
+		remainingRuns: remainingRuns.length,
+		remainingSessionPrompts: remainingPrompts.length,
+		resumedRuns: state.resumedRuns
+	};
+}
+function shouldRearm(state, remainingRuns, remainingPrompts) {
+	return state.reasons.size > 0 && (remainingRuns > 0 || remainingPrompts > 0 || state.reasons.has("failure"));
+}
+function hasFailures(summary) {
+	return summary.failedRuns.length > 0 || summary.failedSessionPrompts.length > 0;
+}
+function markersFor(reasons) {
+	const markers = ["alarm:resume", "resume:session-prompt"];
+	if (reasons.size > 0) markers.push("alarm:continuation-rearm");
+	if (reasons.has("failure")) markers.push("alarm:failure");
+	return markers;
+}
+//#endregion
+export { CloudflareAlarmDrainFailureError, drainCloudflareAlarm };
+
+//# sourceMappingURL=cloudflare-alarm-drainer.js.map

package/dist/cloudflare/cloudflare-alarm-drainer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cloudflare-alarm-drainer.js","names":[],"sources":["../../src/cloudflare/cloudflare-alarm-drainer.ts"],"sourcesContent":["import type { AgentEvent, AgentRun } from \"../index\";\nimport {\n  type AlarmDrainState,\n  type CloudflareAlarmContinuationReason,\n  type CloudflareAlarmDrainBudget,\n  createAlarmDrainState,\n  type FailedScheduledWork,\n  type NormalizedAlarmDrainBudget,\n  normalizeAlarmDrainBudget,\n  shouldStopRuns,\n  shouldStopSessionPrompts,\n} from \"./cloudflare-alarm-budget\";\nimport {\n  resumeScheduledRun,\n  resumeScheduledSessionPrompt,\n} from \"./cloudflare-alarm-work\";\nimport {\n  type CloudflareDurableObjectStorage,\n  listScheduledCloudflareRuns,\n  listScheduledCloudflareSessionPrompts,\n  rescheduleCloudflareAlarm,\n} from \"./cloudflare-host\";\n\nexport type {\n  CloudflareAlarmContinuationReason,\n  CloudflareAlarmDrainBudget,\n  FailedScheduledWork,\n} from \"./cloudflare-alarm-budget\";\n\nexport interface CloudflareAlarmDrainSummary {\n  readonly consumedSessionPrompts: readonly string[];\n  readonly continuationReasons: readonly CloudflareAlarmContinuationReason[];\n  readonly continuationScheduled: boolean;\n  readonly droppedEvents: number;\n  readonly events: readonly AgentEvent[];\n  readonly failedRuns: readonly FailedScheduledWork[];\n  readonly failedSessionPrompts: readonly FailedScheduledWork[];\n  readonly markers: readonly string[];\n  readonly remainingRuns: number;\n  readonly remainingSessionPrompts: number;\n  readonly resumedRuns: readonly string[];\n}\n\nexport interface CloudflareAlarmAgent {\n  resume(runId: string): Promise<AgentRun | null>;\n}\n\nexport class CloudflareAlarmDrainFailureError extends Error {\n  readonly summary: CloudflareAlarmDrainSummary;\n\n  constructor(summary: CloudflareAlarmDrainSummary) {\n    super(\"Cloudflare alarm drain completed with failed scheduled work.\");\n    this.name = \"CloudflareAlarmDrainFailureError\";\n    this.summary = summary;\n  }\n}\n\nexport async function drainCloudflareAlarm({\n  agent,\n  continuationRunAfterMs,\n  deadlineMs,\n  failureRunAfterMs,\n  maxEvents,\n  maxRuns,\n  maxSessionPrompts,\n  prefix,\n  storage,\n  throwOnFailure,\n}: {\n  readonly agent: CloudflareAlarmAgent;\n  readonly prefix: string;\n  readonly storage: CloudflareDurableObjectStorage;\n} & CloudflareAlarmDrainBudget): Promise<CloudflareAlarmDrainSummary> {\n  const budget = normalizeAlarmDrainBudget({\n    continuationRunAfterMs,\n    deadlineMs,\n    failureRunAfterMs,\n    maxEvents,\n    maxRuns,\n    maxSessionPrompts,\n    throwOnFailure,\n  });\n  const state = createAlarmDrainState();\n\n  await drainRuns({ agent, budget, prefix, state, storage });\n  await drainSessionPrompts({ agent, budget, prefix, state, storage });\n\n  const summary = await summarizeDrain({ budget, prefix, state, storage });\n  if (budget.throwOnFailure && hasFailures(summary)) {\n    throw new CloudflareAlarmDrainFailureError(summary);\n  }\n  return summary;\n}\n\nasync function drainRuns(options: DrainLoopOptions): Promise<void> {\n  for (const runId of await listScheduledCloudflareRuns(options.storage, {\n    prefix: options.prefix,\n  })) {\n    if (shouldStopRuns(options.state, options.budget)) {\n      break;\n    }\n    options.state.runAttempts += 1;\n    await resumeScheduledRun({ ...options, runId });\n  }\n}\n\nasync function drainSessionPrompts(options: DrainLoopOptions): Promise<void> {\n  const prompts = await listScheduledCloudflareSessionPrompts(options.storage, {\n    prefix: options.prefix,\n  });\n  for (const prompt of prompts) {\n    if (shouldStopSessionPrompts(options.state, options.budget)) {\n      break;\n    }\n    options.state.sessionPromptAttempts += 1;\n    await resumeScheduledSessionPrompt({ ...options, prompt });\n  }\n}\n\nasync function summarizeDrain({\n  budget,\n  prefix,\n  state,\n  storage,\n}: {\n  readonly budget: NormalizedAlarmDrainBudget;\n  readonly prefix: string;\n  readonly state: AlarmDrainState;\n  readonly storage: CloudflareDurableObjectStorage;\n}): Promise<CloudflareAlarmDrainSummary> {\n  const remainingRuns = await listScheduledCloudflareRuns(storage, { prefix });\n  const remainingPrompts = await listScheduledCloudflareSessionPrompts(\n    storage,\n    {\n      prefix,\n    }\n  );\n  if (state.failedRuns.length > 0 || state.failedSessionPrompts.length > 0) {\n    state.reasons.add(\"failure\");\n  }\n  const continuationScheduled = shouldRearm(\n    state,\n    remainingRuns.length,\n    remainingPrompts.length\n  );\n  if (continuationScheduled) {\n    await rescheduleCloudflareAlarm(storage, {\n      runAfterMs: state.reasons.has(\"failure\")\n        ? budget.failureRunAfterMs\n        : budget.continuationRunAfterMs,\n    });\n  }\n  return {\n    consumedSessionPrompts: state.consumedSessionPrompts,\n    continuationReasons: [...state.reasons],\n    continuationScheduled,\n    droppedEvents: state.droppedEvents,\n    events: state.events,\n    failedRuns: state.failedRuns,\n    failedSessionPrompts: state.failedSessionPrompts,\n    markers: markersFor(state.reasons),\n    remainingRuns: remainingRuns.length,\n    remainingSessionPrompts: remainingPrompts.length,\n    resumedRuns: state.resumedRuns,\n  };\n}\n\ninterface DrainLoopOptions {\n  readonly agent: CloudflareAlarmAgent;\n  readonly budget: NormalizedAlarmDrainBudget;\n  readonly prefix: string;\n  readonly state: AlarmDrainState;\n  readonly storage: CloudflareDurableObjectStorage;\n}\n\nfunction shouldRearm(\n  state: AlarmDrainState,\n  remainingRuns: number,\n  remainingPrompts: number\n): boolean {\n  return (\n    state.reasons.size > 0 &&\n    (remainingRuns > 0 || remainingPrompts > 0 || state.reasons.has(\"failure\"))\n  );\n}\n\nfunction hasFailures(summary: CloudflareAlarmDrainSummary): boolean {\n  return (\n    summary.failedRuns.length > 0 || summary.failedSessionPrompts.length > 0\n  );\n}\n\nfunction markersFor(\n  reasons: ReadonlySet<CloudflareAlarmContinuationReason>\n): readonly string[] {\n  const markers = [\"alarm:resume\", \"resume:session-prompt\"];\n  if (reasons.size > 0) {\n    markers.push(\"alarm:continuation-rearm\");\n  }\n  if (reasons.has(\"failure\")) {\n    markers.push(\"alarm:failure\");\n  }\n  return markers;\n}\n"],"mappings":";;;;AA+CA,IAAa,mCAAb,cAAsD,MAAM;CAC1D;CAEA,YAAY,SAAsC;EAChD,MAAM,8DAA8D;EACpE,KAAK,OAAO;EACZ,KAAK,UAAU;CACjB;AACF;AAEA,eAAsB,qBAAqB,EACzC,OACA,wBACA,YACA,mBACA,WACA,SACA,mBACA,QACA,SACA,kBAKoE;CACpE,MAAM,SAAS,0BAA0B;EACvC;EACA;EACA;EACA;EACA;EACA;EACA;CACF,CAAC;CACD,MAAM,QAAQ,sBAAsB;CAEpC,MAAM,UAAU;EAAE;EAAO;EAAQ;EAAQ;EAAO;CAAQ,CAAC;CACzD,MAAM,oBAAoB;EAAE;EAAO;EAAQ;EAAQ;EAAO;CAAQ,CAAC;CAEnE,MAAM,UAAU,MAAM,eAAe;EAAE;EAAQ;EAAQ;EAAO;CAAQ,CAAC;CACvE,IAAI,OAAO,kBAAkB,YAAY,OAAO,GAC9C,MAAM,IAAI,iCAAiC,OAAO;CAEpD,OAAO;AACT;AAEA,eAAe,UAAU,SAA0C;CACjE,KAAK,MAAM,SAAS,MAAM,4BAA4B,QAAQ,SAAS,EACrE,QAAQ,QAAQ,OAClB,CAAC,GAAG;EACF,IAAI,eAAe,QAAQ,OAAO,QAAQ,MAAM,GAC9C;EAEF,QAAQ,MAAM,eAAe;EAC7B,MAAM,mBAAmB;GAAE,GAAG;GAAS;EAAM,CAAC;CAChD;AACF;AAEA,eAAe,oBAAoB,SAA0C;CAC3E,MAAM,UAAU,MAAM,sCAAsC,QAAQ,SAAS,EAC3E,QAAQ,QAAQ,OAClB,CAAC;CACD,KAAK,MAAM,UAAU,SAAS;EAC5B,IAAI,yBAAyB,QAAQ,OAAO,QAAQ,MAAM,GACxD;EAEF,QAAQ,MAAM,yBAAyB;EACvC,MAAM,6BAA6B;GAAE,GAAG;GAAS;EAAO,CAAC;CAC3D;AACF;AAEA,eAAe,eAAe,EAC5B,QACA,QACA,OACA,WAMuC;CACvC,MAAM,gBAAgB,MAAM,4BAA4B,SAAS,EAAE,OAAO,CAAC;CAC3E,MAAM,mBAAmB,MAAM,sCAC7B,SACA,EACE,OACF,CACF;CACA,IAAI,MAAM,WAAW,SAAS,KAAK,MAAM,qBAAqB,SAAS,GACrE,MAAM,QAAQ,IAAI,SAAS;CAE7B,MAAM,wBAAwB,YAC5B,OACA,cAAc,QACd,iBAAiB,MACnB;CACA,IAAI,uBACF,MAAM,0BAA0B,SAAS,EACvC,YAAY,MAAM,QAAQ,IAAI,SAAS,IACnC,OAAO,oBACP,OAAO,uBACb,CAAC;CAEH,OAAO;EACL,wBAAwB,MAAM;EAC9B,qBAAqB,CAAC,GAAG,MAAM,OAAO;EACtC;EACA,eAAe,MAAM;EACrB,QAAQ,MAAM;EACd,YAAY,MAAM;EAClB,sBAAsB,MAAM;EAC5B,SAAS,WAAW,MAAM,OAAO;EACjC,eAAe,cAAc;EAC7B,yBAAyB,iBAAiB;EAC1C,aAAa,MAAM;CACrB;AACF;AAUA,SAAS,YACP,OACA,eACA,kBACS;CACT,OACE,MAAM,QAAQ,OAAO,MACpB,gBAAgB,KAAK,mBAAmB,KAAK,MAAM,QAAQ,IAAI,SAAS;AAE7E;AAEA,SAAS,YAAY,SAA+C;CAClE,OACE,QAAQ,WAAW,SAAS,KAAK,QAAQ,qBAAqB,SAAS;AAE3E;AAEA,SAAS,WACP,SACmB;CACnB,MAAM,UAAU,CAAC,gBAAgB,uBAAuB;CACxD,IAAI,QAAQ,OAAO,GACjB,QAAQ,KAAK,0BAA0B;CAEzC,IAAI,QAAQ,IAAI,SAAS,GACvB,QAAQ,KAAK,eAAe;CAE9B,OAAO;AACT"}