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

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"}

package/dist/cloudflare/cloudflare-alarm-run-drain.d.ts

@@ -0,0 +1,13 @@
+import { AgentEvent } from "../session/events.js";
+import { AgentRun } from "../session/run.js";
+//#region src/cloudflare/cloudflare-alarm-run-drain.d.ts
+interface CloudflareAgentRunDrainOptions {
+  readonly deadlineMs?: number;
+  readonly maxEvents?: number;
+  readonly onEvent?: (event: AgentEvent) => Promise<void> | void;
+  readonly startedAt?: number;
+}
+declare function drainAgentRun(run: AgentRun, options?: CloudflareAgentRunDrainOptions): Promise<AgentEvent[]>;
+//#endregion
+export { CloudflareAgentRunDrainOptions, drainAgentRun };
+//# sourceMappingURL=cloudflare-alarm-run-drain.d.ts.map

package/dist/cloudflare/cloudflare-alarm-run-drain.js

@@ -0,0 +1,81 @@
+//#region src/cloudflare/cloudflare-alarm-run-drain.ts
+async function drainAgentRun(run, options = {}) {
+	return [...(await drainAgentRunWithBudget(run, {
+		deadlineMs: options.deadlineMs,
+		maxEvents: options.maxEvents,
+		onEvent: options.onEvent,
+		startedAt: options.startedAt
+	})).events];
+}
+async function drainAgentRunWithBudget(run, options = {}) {
+	const events = [];
+	let droppedEvents = 0;
+	const deadlineAt = options.deadlineMs === void 0 ? void 0 : (options.startedAt ?? Date.now()) + options.deadlineMs;
+	const iterator = run.events()[Symbol.asyncIterator]();
+	let stoppedReason;
+	try {
+		while (true) {
+			const eventBudgetFull = options.maxEvents !== void 0 && events.length >= options.maxEvents;
+			const nextEvent = await readNextEvent(iterator, deadlineAt);
+			if (nextEvent.deadlineExpired) {
+				stoppedReason = "deadline";
+				break;
+			}
+			if (nextEvent.result.done) break;
+			if (deadlineExpired(deadlineAt)) {
+				droppedEvents += 1;
+				stoppedReason = "deadline";
+				break;
+			}
+			if (eventBudgetFull) {
+				droppedEvents += 1;
+				stoppedReason = "event-budget";
+				break;
+			}
+			events.push(nextEvent.result.value);
+			await options.onEvent?.(nextEvent.result.value);
+		}
+	} finally {
+		if (stoppedReason) stopIterator(iterator);
+	}
+	return stoppedReason ? {
+		droppedEvents,
+		events,
+		stoppedReason
+	} : {
+		droppedEvents,
+		events
+	};
+}
+async function readNextEvent(iterator, deadlineAt) {
+	if (deadlineExpired(deadlineAt)) return { deadlineExpired: true };
+	if (deadlineAt === void 0) return {
+		deadlineExpired: false,
+		result: await iterator.next()
+	};
+	const remainingMs = Math.max(0, deadlineAt - Date.now());
+	let timeout;
+	const nextEvent = iterator.next().then((result) => ({
+		deadlineExpired: false,
+		result
+	}));
+	const deadline = new Promise((resolve) => {
+		timeout = setTimeout(() => resolve({ deadlineExpired: true }), remainingMs);
+	});
+	try {
+		return await Promise.race([nextEvent, deadline]);
+	} finally {
+		if (timeout !== void 0) clearTimeout(timeout);
+	}
+}
+function deadlineExpired(deadlineAt) {
+	return deadlineAt !== void 0 && Date.now() >= deadlineAt;
+}
+function stopIterator(iterator) {
+	const returned = iterator.return?.();
+	if (returned) returned.catch(() => void 0);
+}
+//#endregion
+export { drainAgentRun, drainAgentRunWithBudget };
+
+//# sourceMappingURL=cloudflare-alarm-run-drain.js.map

package/dist/cloudflare/cloudflare-alarm-run-drain.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cloudflare-alarm-run-drain.js","names":[],"sources":["../../src/cloudflare/cloudflare-alarm-run-drain.ts"],"sourcesContent":["import type { AgentEvent, AgentRun } from \"../index\";\n\nexport type AgentRunDrainStopReason = \"deadline\" | \"event-budget\";\n\nexport interface CloudflareAgentRunDrainOptions {\n  readonly deadlineMs?: number;\n  readonly maxEvents?: number;\n  readonly onEvent?: (event: AgentEvent) => Promise<void> | void;\n  readonly startedAt?: number;\n}\n\nexport interface AgentRunDrainResult {\n  readonly droppedEvents: number;\n  readonly events: readonly AgentEvent[];\n  readonly stoppedReason?: AgentRunDrainStopReason;\n}\n\nexport async function drainAgentRun(\n  run: AgentRun,\n  options: CloudflareAgentRunDrainOptions = {}\n): Promise<AgentEvent[]> {\n  return [\n    ...(\n      await drainAgentRunWithBudget(run, {\n        deadlineMs: options.deadlineMs,\n        maxEvents: options.maxEvents,\n        onEvent: options.onEvent,\n        startedAt: options.startedAt,\n      })\n    ).events,\n  ];\n}\n\nexport async function drainAgentRunWithBudget(\n  run: AgentRun,\n  options: CloudflareAgentRunDrainOptions = {}\n): Promise<AgentRunDrainResult> {\n  const events: AgentEvent[] = [];\n  let droppedEvents = 0;\n  const deadlineAt =\n    options.deadlineMs === undefined\n      ? undefined\n      : (options.startedAt ?? Date.now()) + options.deadlineMs;\n  const iterator = run.events()[Symbol.asyncIterator]();\n  let stoppedReason: AgentRunDrainStopReason | undefined;\n\n  try {\n    while (true) {\n      const eventBudgetFull =\n        options.maxEvents !== undefined && events.length >= options.maxEvents;\n      const nextEvent = await readNextEvent(iterator, deadlineAt);\n      if (nextEvent.deadlineExpired) {\n        stoppedReason = \"deadline\";\n        break;\n      }\n      if (nextEvent.result.done) {\n        break;\n      }\n      if (deadlineExpired(deadlineAt)) {\n        droppedEvents += 1;\n        stoppedReason = \"deadline\";\n        break;\n      }\n      if (eventBudgetFull) {\n        droppedEvents += 1;\n        stoppedReason = \"event-budget\";\n        break;\n      }\n      events.push(nextEvent.result.value);\n      await options.onEvent?.(nextEvent.result.value);\n    }\n  } finally {\n    if (stoppedReason) {\n      stopIterator(iterator);\n    }\n  }\n\n  return stoppedReason\n    ? { droppedEvents, events, stoppedReason }\n    : { droppedEvents, events };\n}\n\ntype NextEventResult =\n  | {\n      readonly deadlineExpired: false;\n      readonly result: IteratorResult<AgentEvent>;\n    }\n  | { readonly deadlineExpired: true };\n\nasync function readNextEvent(\n  iterator: AsyncIterator<AgentEvent>,\n  deadlineAt: number | undefined\n): Promise<NextEventResult> {\n  if (deadlineExpired(deadlineAt)) {\n    return { deadlineExpired: true };\n  }\n  if (deadlineAt === undefined) {\n    return { deadlineExpired: false, result: await iterator.next() };\n  }\n\n  const remainingMs = Math.max(0, deadlineAt - Date.now());\n  let timeout: ReturnType<typeof setTimeout> | undefined;\n  const nextEvent = iterator.next().then(\n    (result) =>\n      ({\n        deadlineExpired: false,\n        result,\n      }) satisfies NextEventResult\n  );\n  const deadline = new Promise<NextEventResult>((resolve) => {\n    timeout = setTimeout(() => resolve({ deadlineExpired: true }), remainingMs);\n  });\n\n  try {\n    return await Promise.race([nextEvent, deadline]);\n  } finally {\n    if (timeout !== undefined) {\n      clearTimeout(timeout);\n    }\n  }\n}\n\nfunction deadlineExpired(deadlineAt: number | undefined): boolean {\n  return deadlineAt !== undefined && Date.now() >= deadlineAt;\n}\n\nfunction stopIterator(iterator: AsyncIterator<AgentEvent>): void {\n  const returned = iterator.return?.();\n  if (returned) {\n    returned.catch(() => undefined);\n  }\n}\n"],"mappings":";AAiBA,eAAsB,cACpB,KACA,UAA0C,CAAC,GACpB;CACvB,OAAO,CACL,IACE,MAAM,wBAAwB,KAAK;EACjC,YAAY,QAAQ;EACpB,WAAW,QAAQ;EACnB,SAAS,QAAQ;EACjB,WAAW,QAAQ;CACrB,CAAC,GACD,MACJ;AACF;AAEA,eAAsB,wBACpB,KACA,UAA0C,CAAC,GACb;CAC9B,MAAM,SAAuB,CAAC;CAC9B,IAAI,gBAAgB;CACpB,MAAM,aACJ,QAAQ,eAAe,KAAA,IACnB,KAAA,KACC,QAAQ,aAAa,KAAK,IAAI,KAAK,QAAQ;CAClD,MAAM,WAAW,IAAI,OAAO,EAAE,OAAO,eAAe;CACpD,IAAI;CAEJ,IAAI;EACF,OAAO,MAAM;GACX,MAAM,kBACJ,QAAQ,cAAc,KAAA,KAAa,OAAO,UAAU,QAAQ;GAC9D,MAAM,YAAY,MAAM,cAAc,UAAU,UAAU;GAC1D,IAAI,UAAU,iBAAiB;IAC7B,gBAAgB;IAChB;GACF;GACA,IAAI,UAAU,OAAO,MACnB;GAEF,IAAI,gBAAgB,UAAU,GAAG;IAC/B,iBAAiB;IACjB,gBAAgB;IAChB;GACF;GACA,IAAI,iBAAiB;IACnB,iBAAiB;IACjB,gBAAgB;IAChB;GACF;GACA,OAAO,KAAK,UAAU,OAAO,KAAK;GAClC,MAAM,QAAQ,UAAU,UAAU,OAAO,KAAK;EAChD;CACF,UAAU;EACR,IAAI,eACF,aAAa,QAAQ;CAEzB;CAEA,OAAO,gBACH;EAAE;EAAe;EAAQ;CAAc,IACvC;EAAE;EAAe;CAAO;AAC9B;AASA,eAAe,cACb,UACA,YAC0B;CAC1B,IAAI,gBAAgB,UAAU,GAC5B,OAAO,EAAE,iBAAiB,KAAK;CAEjC,IAAI,eAAe,KAAA,GACjB,OAAO;EAAE,iBAAiB;EAAO,QAAQ,MAAM,SAAS,KAAK;CAAE;CAGjE,MAAM,cAAc,KAAK,IAAI,GAAG,aAAa,KAAK,IAAI,CAAC;CACvD,IAAI;CACJ,MAAM,YAAY,SAAS,KAAK,EAAE,MAC/B,YACE;EACC,iBAAiB;EACjB;CACF,EACJ;CACA,MAAM,WAAW,IAAI,SAA0B,YAAY;EACzD,UAAU,iBAAiB,QAAQ,EAAE,iBAAiB,KAAK,CAAC,GAAG,WAAW;CAC5E,CAAC;CAED,IAAI;EACF,OAAO,MAAM,QAAQ,KAAK,CAAC,WAAW,QAAQ,CAAC;CACjD,UAAU;EACR,IAAI,YAAY,KAAA,GACd,aAAa,OAAO;CAExB;AACF;AAEA,SAAS,gBAAgB,YAAyC;CAChE,OAAO,eAAe,KAAA,KAAa,KAAK,IAAI,KAAK;AACnD;AAEA,SAAS,aAAa,UAA2C;CAC/D,MAAM,WAAW,SAAS,SAAS;CACnC,IAAI,UACF,SAAS,YAAY,KAAA,CAAS;AAElC"}

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

@@ -0,0 +1,110 @@
+import { eventSlotsRemaining } from "./cloudflare-alarm-budget.js";
+import { drainAgentRunWithBudget } from "./cloudflare-alarm-run-drain.js";
+import { ackScheduledCloudflareRun, ackScheduledCloudflareSessionPrompt, createCloudflareDurableObjectHost } from "./cloudflare-host.js";
+//#region src/cloudflare/cloudflare-alarm-work.ts
+async function resumeScheduledRun({ agent, budget, prefix, runId, state, storage }) {
+	try {
+		const run = await agent.resume(runId);
+		if (!run) {
+			if (await shouldRetryScheduledRun(storage, prefix, runId)) {
+				state.failedRuns.push({
+					error: "Run was not claimable during this alarm.",
+					id: runId
+				});
+				return;
+			}
+			await ackScheduledCloudflareRun(storage, runId, { prefix });
+			return;
+		}
+		state.resumedRuns.push(runId);
+		appendRunDrainEvents(state, await drainAgentRunWithBudget(run, {
+			deadlineMs: budget.deadlineMs,
+			maxEvents: eventSlotsRemaining(state, budget),
+			startedAt: budget.startedAt
+		}));
+		if (shouldContinueRunDrain(state)) return;
+		await ackScheduledCloudflareRun(storage, runId, { prefix });
+	} catch (error) {
+		state.failedRuns.push({
+			error: describeError(error),
+			id: runId
+		});
+	}
+}
+async function resumeScheduledSessionPrompt({ agent, budget, prefix, prompt, state, storage }) {
+	try {
+		const runId = await scheduledSessionPromptRunId(storage, prefix, prompt);
+		if (!runId) {
+			state.failedSessionPrompts.push({
+				error: "Session prompt did not include or resolve to a run id.",
+				id: sessionPromptId(prompt)
+			});
+			return;
+		}
+		const run = await agent.resume(runId);
+		if (!run) {
+			if (await shouldRetryScheduledSessionPrompt(storage, prefix, prompt)) {
+				state.failedSessionPrompts.push({
+					error: "Session prompt was not claimable during this alarm.",
+					id: sessionPromptId(prompt)
+				});
+				return;
+			}
+			await ackScheduledCloudflareSessionPrompt(storage, prompt, { prefix });
+			return;
+		}
+		appendRunDrainEvents(state, await drainAgentRunWithBudget(run, {
+			deadlineMs: budget.deadlineMs,
+			maxEvents: eventSlotsRemaining(state, budget),
+			startedAt: budget.startedAt
+		}));
+		if (shouldContinueRunDrain(state)) return;
+		state.consumedSessionPrompts.push(sessionPromptId(prompt));
+		await ackScheduledCloudflareSessionPrompt(storage, prompt, { prefix });
+	} catch (error) {
+		state.failedSessionPrompts.push({
+			error: describeError(error),
+			id: sessionPromptId(prompt)
+		});
+	}
+}
+async function shouldRetryScheduledRun(storage, prefix, runId) {
+	const run = await createCloudflareDurableObjectHost({
+		prefix,
+		storage
+	}).store.runs.get(runId);
+	return run ? isRetryableRunStatus(run.status) : true;
+}
+async function shouldRetryScheduledSessionPrompt(storage, prefix, prompt) {
+	const runId = await scheduledSessionPromptRunId(storage, prefix, prompt);
+	return runId ? await shouldRetryScheduledRun(storage, prefix, runId) : prompt.idempotencyKey !== void 0;
+}
+async function scheduledSessionPromptRunId(storage, prefix, prompt) {
+	if (prompt.runId) return prompt.runId;
+	if (!prompt.idempotencyKey) return;
+	return (await createCloudflareDurableObjectHost({
+		prefix,
+		storage
+	}).store.notifications.getByIdempotencyKey(prompt.idempotencyKey))?.runId;
+}
+function sessionPromptId(prompt) {
+	return prompt.idempotencyKey ?? prompt.runId ?? "<missing-run-id>";
+}
+function isRetryableRunStatus(status) {
+	return status === "leased" || status === "queued" || status === "running" || status === "suspended";
+}
+function describeError(error) {
+	return error instanceof Error ? error.message : String(error);
+}
+function appendRunDrainEvents(state, result) {
+	state.events.push(...result.events);
+	state.droppedEvents += result.droppedEvents;
+	if (result.stoppedReason) state.reasons.add(result.stoppedReason);
+}
+function shouldContinueRunDrain(state) {
+	return state.reasons.has("deadline") || state.reasons.has("event-budget");
+}
+//#endregion
+export { resumeScheduledRun, resumeScheduledSessionPrompt };
+
+//# sourceMappingURL=cloudflare-alarm-work.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"cloudflare-alarm-work.js","names":[],"sources":["../../src/cloudflare/cloudflare-alarm-work.ts"],"sourcesContent":["import type { RunStatus } from \"../execution\";\nimport {\n  type AlarmDrainState,\n  eventSlotsRemaining,\n  type NormalizedAlarmDrainBudget,\n} from \"./cloudflare-alarm-budget\";\nimport type { CloudflareAlarmAgent } from \"./cloudflare-alarm-drainer\";\nimport {\n  type AgentRunDrainResult,\n  drainAgentRunWithBudget,\n} from \"./cloudflare-alarm-run-drain\";\nimport {\n  ackScheduledCloudflareRun,\n  ackScheduledCloudflareSessionPrompt,\n  type CloudflareDurableObjectStorage,\n  type CloudflareScheduledSessionPrompt,\n  createCloudflareDurableObjectHost,\n} from \"./cloudflare-host\";\n\nexport async function resumeScheduledRun({\n  agent,\n  budget,\n  prefix,\n  runId,\n  state,\n  storage,\n}: {\n  readonly agent: CloudflareAlarmAgent;\n  readonly budget: NormalizedAlarmDrainBudget;\n  readonly prefix: string;\n  readonly runId: string;\n  readonly state: AlarmDrainState;\n  readonly storage: CloudflareDurableObjectStorage;\n}): Promise<void> {\n  try {\n    const run = await agent.resume(runId);\n    if (!run) {\n      if (await shouldRetryScheduledRun(storage, prefix, runId)) {\n        state.failedRuns.push({\n          error: \"Run was not claimable during this alarm.\",\n          id: runId,\n        });\n        return;\n      }\n      await ackScheduledCloudflareRun(storage, runId, { prefix });\n      return;\n    }\n    state.resumedRuns.push(runId);\n    appendRunDrainEvents(\n      state,\n      await drainAgentRunWithBudget(run, {\n        deadlineMs: budget.deadlineMs,\n        maxEvents: eventSlotsRemaining(state, budget),\n        startedAt: budget.startedAt,\n      })\n    );\n    if (shouldContinueRunDrain(state)) {\n      return;\n    }\n    await ackScheduledCloudflareRun(storage, runId, { prefix });\n  } catch (error) {\n    state.failedRuns.push({ error: describeError(error), id: runId });\n  }\n}\n\nexport async function resumeScheduledSessionPrompt({\n  agent,\n  budget,\n  prefix,\n  prompt,\n  state,\n  storage,\n}: {\n  readonly agent: CloudflareAlarmAgent;\n  readonly budget: NormalizedAlarmDrainBudget;\n  readonly prefix: string;\n  readonly prompt: CloudflareScheduledSessionPrompt;\n  readonly state: AlarmDrainState;\n  readonly storage: CloudflareDurableObjectStorage;\n}): Promise<void> {\n  try {\n    const runId = await scheduledSessionPromptRunId(storage, prefix, prompt);\n    if (!runId) {\n      state.failedSessionPrompts.push({\n        error: \"Session prompt did not include or resolve to a run id.\",\n        id: sessionPromptId(prompt),\n      });\n      return;\n    }\n\n    const run = await agent.resume(runId);\n    if (!run) {\n      if (await shouldRetryScheduledSessionPrompt(storage, prefix, prompt)) {\n        state.failedSessionPrompts.push({\n          error: \"Session prompt was not claimable during this alarm.\",\n          id: sessionPromptId(prompt),\n        });\n        return;\n      }\n      await ackScheduledCloudflareSessionPrompt(storage, prompt, { prefix });\n      return;\n    }\n    appendRunDrainEvents(\n      state,\n      await drainAgentRunWithBudget(run, {\n        deadlineMs: budget.deadlineMs,\n        maxEvents: eventSlotsRemaining(state, budget),\n        startedAt: budget.startedAt,\n      })\n    );\n    if (shouldContinueRunDrain(state)) {\n      return;\n    }\n    state.consumedSessionPrompts.push(sessionPromptId(prompt));\n    await ackScheduledCloudflareSessionPrompt(storage, prompt, { prefix });\n  } catch (error) {\n    state.failedSessionPrompts.push({\n      error: describeError(error),\n      id: sessionPromptId(prompt),\n    });\n  }\n}\n\nasync function shouldRetryScheduledRun(\n  storage: CloudflareDurableObjectStorage,\n  prefix: string,\n  runId: string\n): Promise<boolean> {\n  const run = await createCloudflareDurableObjectHost({\n    prefix,\n    storage,\n  }).store.runs.get(runId);\n  return run ? isRetryableRunStatus(run.status) : true;\n}\n\nasync function shouldRetryScheduledSessionPrompt(\n  storage: CloudflareDurableObjectStorage,\n  prefix: string,\n  prompt: CloudflareScheduledSessionPrompt\n): Promise<boolean> {\n  const runId = await scheduledSessionPromptRunId(storage, prefix, prompt);\n  return runId\n    ? await shouldRetryScheduledRun(storage, prefix, runId)\n    : prompt.idempotencyKey !== undefined;\n}\n\nasync function scheduledSessionPromptRunId(\n  storage: CloudflareDurableObjectStorage,\n  prefix: string,\n  prompt: CloudflareScheduledSessionPrompt\n): Promise<string | undefined> {\n  if (prompt.runId) {\n    return prompt.runId;\n  }\n  if (!prompt.idempotencyKey) {\n    return;\n  }\n\n  const notification = await createCloudflareDurableObjectHost({\n    prefix,\n    storage,\n  }).store.notifications.getByIdempotencyKey(prompt.idempotencyKey);\n  return notification?.runId;\n}\n\nfunction sessionPromptId(prompt: CloudflareScheduledSessionPrompt): string {\n  return prompt.idempotencyKey ?? prompt.runId ?? \"<missing-run-id>\";\n}\n\nfunction isRetryableRunStatus(status: RunStatus | undefined): boolean {\n  return (\n    status === \"leased\" ||\n    status === \"queued\" ||\n    status === \"running\" ||\n    status === \"suspended\"\n  );\n}\n\nfunction describeError(error: unknown): string {\n  return error instanceof Error ? error.message : String(error);\n}\n\nfunction appendRunDrainEvents(\n  state: AlarmDrainState,\n  result: AgentRunDrainResult\n): void {\n  state.events.push(...result.events);\n  state.droppedEvents += result.droppedEvents;\n  if (result.stoppedReason) {\n    state.reasons.add(result.stoppedReason);\n  }\n}\n\nfunction shouldContinueRunDrain(state: AlarmDrainState): boolean {\n  return state.reasons.has(\"deadline\") || state.reasons.has(\"event-budget\");\n}\n"],"mappings":";;;;AAmBA,eAAsB,mBAAmB,EACvC,OACA,QACA,QACA,OACA,OACA,WAQgB;CAChB,IAAI;EACF,MAAM,MAAM,MAAM,MAAM,OAAO,KAAK;EACpC,IAAI,CAAC,KAAK;GACR,IAAI,MAAM,wBAAwB,SAAS,QAAQ,KAAK,GAAG;IACzD,MAAM,WAAW,KAAK;KACpB,OAAO;KACP,IAAI;IACN,CAAC;IACD;GACF;GACA,MAAM,0BAA0B,SAAS,OAAO,EAAE,OAAO,CAAC;GAC1D;EACF;EACA,MAAM,YAAY,KAAK,KAAK;EAC5B,qBACE,OACA,MAAM,wBAAwB,KAAK;GACjC,YAAY,OAAO;GACnB,WAAW,oBAAoB,OAAO,MAAM;GAC5C,WAAW,OAAO;EACpB,CAAC,CACH;EACA,IAAI,uBAAuB,KAAK,GAC9B;EAEF,MAAM,0BAA0B,SAAS,OAAO,EAAE,OAAO,CAAC;CAC5D,SAAS,OAAO;EACd,MAAM,WAAW,KAAK;GAAE,OAAO,cAAc,KAAK;GAAG,IAAI;EAAM,CAAC;CAClE;AACF;AAEA,eAAsB,6BAA6B,EACjD,OACA,QACA,QACA,QACA,OACA,WAQgB;CAChB,IAAI;EACF,MAAM,QAAQ,MAAM,4BAA4B,SAAS,QAAQ,MAAM;EACvE,IAAI,CAAC,OAAO;GACV,MAAM,qBAAqB,KAAK;IAC9B,OAAO;IACP,IAAI,gBAAgB,MAAM;GAC5B,CAAC;GACD;EACF;EAEA,MAAM,MAAM,MAAM,MAAM,OAAO,KAAK;EACpC,IAAI,CAAC,KAAK;GACR,IAAI,MAAM,kCAAkC,SAAS,QAAQ,MAAM,GAAG;IACpE,MAAM,qBAAqB,KAAK;KAC9B,OAAO;KACP,IAAI,gBAAgB,MAAM;IAC5B,CAAC;IACD;GACF;GACA,MAAM,oCAAoC,SAAS,QAAQ,EAAE,OAAO,CAAC;GACrE;EACF;EACA,qBACE,OACA,MAAM,wBAAwB,KAAK;GACjC,YAAY,OAAO;GACnB,WAAW,oBAAoB,OAAO,MAAM;GAC5C,WAAW,OAAO;EACpB,CAAC,CACH;EACA,IAAI,uBAAuB,KAAK,GAC9B;EAEF,MAAM,uBAAuB,KAAK,gBAAgB,MAAM,CAAC;EACzD,MAAM,oCAAoC,SAAS,QAAQ,EAAE,OAAO,CAAC;CACvE,SAAS,OAAO;EACd,MAAM,qBAAqB,KAAK;GAC9B,OAAO,cAAc,KAAK;GAC1B,IAAI,gBAAgB,MAAM;EAC5B,CAAC;CACH;AACF;AAEA,eAAe,wBACb,SACA,QACA,OACkB;CAClB,MAAM,MAAM,MAAM,kCAAkC;EAClD;EACA;CACF,CAAC,EAAE,MAAM,KAAK,IAAI,KAAK;CACvB,OAAO,MAAM,qBAAqB,IAAI,MAAM,IAAI;AAClD;AAEA,eAAe,kCACb,SACA,QACA,QACkB;CAClB,MAAM,QAAQ,MAAM,4BAA4B,SAAS,QAAQ,MAAM;CACvE,OAAO,QACH,MAAM,wBAAwB,SAAS,QAAQ,KAAK,IACpD,OAAO,mBAAmB,KAAA;AAChC;AAEA,eAAe,4BACb,SACA,QACA,QAC6B;CAC7B,IAAI,OAAO,OACT,OAAO,OAAO;CAEhB,IAAI,CAAC,OAAO,gBACV;CAOF,QAAO,MAJoB,kCAAkC;EAC3D;EACA;CACF,CAAC,EAAE,MAAM,cAAc,oBAAoB,OAAO,cAAc,IAC3C;AACvB;AAEA,SAAS,gBAAgB,QAAkD;CACzE,OAAO,OAAO,kBAAkB,OAAO,SAAS;AAClD;AAEA,SAAS,qBAAqB,QAAwC;CACpE,OACE,WAAW,YACX,WAAW,YACX,WAAW,aACX,WAAW;AAEf;AAEA,SAAS,cAAc,OAAwB;CAC7C,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAC9D;AAEA,SAAS,qBACP,OACA,QACM;CACN,MAAM,OAAO,KAAK,GAAG,OAAO,MAAM;CAClC,MAAM,iBAAiB,OAAO;CAC9B,IAAI,OAAO,eACT,MAAM,QAAQ,IAAI,OAAO,aAAa;AAE1C;AAEA,SAAS,uBAAuB,OAAiC;CAC/D,OAAO,MAAM,QAAQ,IAAI,UAAU,KAAK,MAAM,QAAQ,IAAI,cAAc;AAC1E"}

package/dist/cloudflare/cloudflare-checkpoint-store.js

@@ -0,0 +1,39 @@
+import { getRun, putRun, readList, storeKey, withTransaction } from "./cloudflare-store-utils.js";
+//#region src/cloudflare/cloudflare-checkpoint-store.ts
+var DurableObjectCheckpointStore = class {
+	#prefix;
+	#storage;
+	constructor(storage, prefix) {
+		this.#prefix = prefix;
+		this.#storage = storage;
+	}
+	async append(checkpoint, options) {
+		return await withTransaction(this.#storage, async (storage) => {
+			const run = await getRun(storage, this.#prefix, checkpoint.runId);
+			const currentVersion = run?.checkpointVersion ?? 0;
+			if (currentVersion !== options.expectedVersion) return {
+				currentVersion,
+				ok: false,
+				reason: "stale-version"
+			};
+			const checkpoints = await readList(storage, storeKey(this.#prefix, "checkpoints", checkpoint.runId));
+			checkpoints.push(structuredClone(checkpoint));
+			await storage.put(storeKey(this.#prefix, "checkpoints", checkpoint.runId), checkpoints);
+			if (run) await putRun(storage, this.#prefix, {
+				...run,
+				checkpointVersion: checkpoint.version
+			});
+			return {
+				ok: true,
+				version: checkpoint.version
+			};
+		});
+	}
+	async latest(runId) {
+		return (await readList(this.#storage, storeKey(this.#prefix, "checkpoints", runId))).at(-1) ?? null;
+	}
+};
+//#endregion
+export { DurableObjectCheckpointStore };
+
+//# sourceMappingURL=cloudflare-checkpoint-store.js.map

package/dist/cloudflare/cloudflare-checkpoint-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cloudflare-checkpoint-store.js","names":["#prefix","#storage"],"sources":["../../src/cloudflare/cloudflare-checkpoint-store.ts"],"sourcesContent":["import type {\n  CheckpointStore,\n  CheckpointWriteResult,\n  RunCheckpoint,\n} from \"../execution\";\nimport {\n  getRun,\n  putRun,\n  readList,\n  storeKey,\n  withTransaction,\n} from \"./cloudflare-store-utils\";\nimport type { CloudflareDurableObjectStorage } from \"./durable-object-storage\";\n\nexport class DurableObjectCheckpointStore implements CheckpointStore {\n  readonly #prefix: string;\n  readonly #storage: CloudflareDurableObjectStorage;\n\n  constructor(storage: CloudflareDurableObjectStorage, prefix: string) {\n    this.#prefix = prefix;\n    this.#storage = storage;\n  }\n\n  async append(\n    checkpoint: RunCheckpoint,\n    options: { readonly expectedVersion: number }\n  ): Promise<CheckpointWriteResult> {\n    return await withTransaction(this.#storage, async (storage) => {\n      const run = await getRun(storage, this.#prefix, checkpoint.runId);\n      const currentVersion = run?.checkpointVersion ?? 0;\n      if (currentVersion !== options.expectedVersion) {\n        return {\n          currentVersion,\n          ok: false,\n          reason: \"stale-version\",\n        };\n      }\n\n      const checkpoints = await readList<RunCheckpoint>(\n        storage,\n        storeKey(this.#prefix, \"checkpoints\", checkpoint.runId)\n      );\n      checkpoints.push(structuredClone(checkpoint));\n      await storage.put(\n        storeKey(this.#prefix, \"checkpoints\", checkpoint.runId),\n        checkpoints\n      );\n      if (run) {\n        await putRun(storage, this.#prefix, {\n          ...run,\n          checkpointVersion: checkpoint.version,\n        });\n      }\n      return { ok: true, version: checkpoint.version };\n    });\n  }\n\n  async latest(runId: string): Promise<RunCheckpoint | null> {\n    const checkpoints = await readList<RunCheckpoint>(\n      this.#storage,\n      storeKey(this.#prefix, \"checkpoints\", runId)\n    );\n    return checkpoints.at(-1) ?? null;\n  }\n}\n"],"mappings":";;AAcA,IAAa,+BAAb,MAAqE;CACnE;CACA;CAEA,YAAY,SAAyC,QAAgB;EACnE,KAAKA,UAAU;EACf,KAAKC,WAAW;CAClB;CAEA,MAAM,OACJ,YACA,SACgC;EAChC,OAAO,MAAM,gBAAgB,KAAKA,UAAU,OAAO,YAAY;GAC7D,MAAM,MAAM,MAAM,OAAO,SAAS,KAAKD,SAAS,WAAW,KAAK;GAChE,MAAM,iBAAiB,KAAK,qBAAqB;GACjD,IAAI,mBAAmB,QAAQ,iBAC7B,OAAO;IACL;IACA,IAAI;IACJ,QAAQ;GACV;GAGF,MAAM,cAAc,MAAM,SACxB,SACA,SAAS,KAAKA,SAAS,eAAe,WAAW,KAAK,CACxD;GACA,YAAY,KAAK,gBAAgB,UAAU,CAAC;GAC5C,MAAM,QAAQ,IACZ,SAAS,KAAKA,SAAS,eAAe,WAAW,KAAK,GACtD,WACF;GACA,IAAI,KACF,MAAM,OAAO,SAAS,KAAKA,SAAS;IAClC,GAAG;IACH,mBAAmB,WAAW;GAChC,CAAC;GAEH,OAAO;IAAE,IAAI;IAAM,SAAS,WAAW;GAAQ;EACjD,CAAC;CACH;CAEA,MAAM,OAAO,OAA8C;EAKzD,QAAO,MAJmB,SACxB,KAAKC,UACL,SAAS,KAAKD,SAAS,eAAe,KAAK,CAC7C,GACmB,GAAG,EAAE,KAAK;CAC/B;AACF"}

package/dist/cloudflare/cloudflare-durable-object-fetch.d.ts

@@ -0,0 +1,21 @@
+import { CloudflareDurableObjectNamespace, CloudflareDurableObjectStub } from "./cloudflare-host.js";
+
+//#region src/cloudflare/cloudflare-durable-object-fetch.d.ts
+interface CloudflareDurableObjectStubOptions<Stub extends CloudflareDurableObjectStub = CloudflareDurableObjectStub> {
+  readonly namespace?: CloudflareDurableObjectNamespace<Stub>;
+  readonly objectName: string;
+}
+interface CloudflareDurableObjectFetchOptions<Stub extends CloudflareDurableObjectStub = CloudflareDurableObjectStub> extends CloudflareDurableObjectStubOptions<Stub> {
+  readonly request: Request;
+}
+declare function getCloudflareDurableObjectStub<Stub extends CloudflareDurableObjectStub = CloudflareDurableObjectStub>({
+  namespace,
+  objectName
+}: CloudflareDurableObjectStubOptions<Stub>): Stub | undefined;
+declare function fetchCloudflareDurableObject<Stub extends CloudflareDurableObjectStub = CloudflareDurableObjectStub>({
+  request,
+  ...options
+}: CloudflareDurableObjectFetchOptions<Stub>): Promise<Response | undefined>;
+//#endregion
+export { CloudflareDurableObjectFetchOptions, CloudflareDurableObjectStubOptions, fetchCloudflareDurableObject, getCloudflareDurableObjectStub };
+//# sourceMappingURL=cloudflare-durable-object-fetch.d.ts.map

package/dist/cloudflare/cloudflare-durable-object-fetch.js

@@ -0,0 +1,11 @@
+//#region src/cloudflare/cloudflare-durable-object-fetch.ts
+function getCloudflareDurableObjectStub({ namespace, objectName }) {
+	return namespace?.get(namespace.idFromName(objectName));
+}
+async function fetchCloudflareDurableObject({ request, ...options }) {
+	return await getCloudflareDurableObjectStub(options)?.fetch(request);
+}
+//#endregion
+export { fetchCloudflareDurableObject, getCloudflareDurableObjectStub };
+
+//# sourceMappingURL=cloudflare-durable-object-fetch.js.map

package/dist/cloudflare/cloudflare-durable-object-fetch.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cloudflare-durable-object-fetch.js","names":[],"sources":["../../src/cloudflare/cloudflare-durable-object-fetch.ts"],"sourcesContent":["import type {\n  CloudflareDurableObjectNamespace,\n  CloudflareDurableObjectStub,\n} from \"./cloudflare-host\";\n\nexport interface CloudflareDurableObjectStubOptions<\n  Stub extends CloudflareDurableObjectStub = CloudflareDurableObjectStub,\n> {\n  readonly namespace?: CloudflareDurableObjectNamespace<Stub>;\n  readonly objectName: string;\n}\n\nexport interface CloudflareDurableObjectFetchOptions<\n  Stub extends CloudflareDurableObjectStub = CloudflareDurableObjectStub,\n> extends CloudflareDurableObjectStubOptions<Stub> {\n  readonly request: Request;\n}\n\nexport function getCloudflareDurableObjectStub<\n  Stub extends CloudflareDurableObjectStub = CloudflareDurableObjectStub,\n>({\n  namespace,\n  objectName,\n}: CloudflareDurableObjectStubOptions<Stub>): Stub | undefined {\n  return namespace?.get(namespace.idFromName(objectName));\n}\n\nexport async function fetchCloudflareDurableObject<\n  Stub extends CloudflareDurableObjectStub = CloudflareDurableObjectStub,\n>({\n  request,\n  ...options\n}: CloudflareDurableObjectFetchOptions<Stub>): Promise<Response | undefined> {\n  return await getCloudflareDurableObjectStub(options)?.fetch(request);\n}\n"],"mappings":";AAkBA,SAAgB,+BAEd,EACA,WACA,cAC6D;CAC7D,OAAO,WAAW,IAAI,UAAU,WAAW,UAAU,CAAC;AACxD;AAEA,eAAsB,6BAEpB,EACA,SACA,GAAG,WACwE;CAC3E,OAAO,MAAM,+BAA+B,OAAO,GAAG,MAAM,OAAO;AACrE"}

package/dist/cloudflare/cloudflare-event-store.js

@@ -0,0 +1,33 @@
+import { readList, storeKey, withTransaction } from "./cloudflare-store-utils.js";
+//#region src/cloudflare/cloudflare-event-store.ts
+var DurableObjectEventStore = class {
+	#prefix;
+	#storage;
+	constructor(storage, prefix) {
+		this.#prefix = prefix;
+		this.#storage = storage;
+	}
+	async append(runId, event) {
+		return await withTransaction(this.#storage, async (storage) => {
+			const events = await readList(storage, storeKey(this.#prefix, "events", runId));
+			const cursor = { offset: events.length + 1 };
+			events.push({
+				cursor,
+				event: structuredClone(event),
+				runId
+			});
+			await storage.put(storeKey(this.#prefix, "events", runId), events);
+			return cursor;
+		});
+	}
+	async *read(runId, cursor) {
+		await Promise.resolve();
+		const events = await readList(this.#storage, storeKey(this.#prefix, "events", runId));
+		const start = cursor?.offset ?? 0;
+		for (const event of events.slice(start)) yield event;
+	}
+};
+//#endregion
+export { DurableObjectEventStore };
+
+//# sourceMappingURL=cloudflare-event-store.js.map

package/dist/cloudflare/cloudflare-event-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cloudflare-event-store.js","names":["#prefix","#storage"],"sources":["../../src/cloudflare/cloudflare-event-store.ts"],"sourcesContent":["import type { EventCursor, EventStore, StoredAgentEvent } from \"../execution\";\nimport type { AgentEvent } from \"../index\";\nimport { readList, storeKey, withTransaction } from \"./cloudflare-store-utils\";\nimport type { CloudflareDurableObjectStorage } from \"./durable-object-storage\";\n\nexport class DurableObjectEventStore implements EventStore {\n  readonly #prefix: string;\n  readonly #storage: CloudflareDurableObjectStorage;\n\n  constructor(storage: CloudflareDurableObjectStorage, prefix: string) {\n    this.#prefix = prefix;\n    this.#storage = storage;\n  }\n\n  async append(runId: string, event: AgentEvent): Promise<EventCursor> {\n    return await withTransaction(this.#storage, async (storage) => {\n      const events = await readList<StoredAgentEvent>(\n        storage,\n        storeKey(this.#prefix, \"events\", runId)\n      );\n      const cursor = { offset: events.length + 1 };\n      events.push({ cursor, event: structuredClone(event), runId });\n      await storage.put(storeKey(this.#prefix, \"events\", runId), events);\n      return cursor;\n    });\n  }\n\n  async *read(\n    runId: string,\n    cursor?: EventCursor\n  ): AsyncIterable<StoredAgentEvent> {\n    await Promise.resolve();\n    const events = await readList<StoredAgentEvent>(\n      this.#storage,\n      storeKey(this.#prefix, \"events\", runId)\n    );\n    const start = cursor?.offset ?? 0;\n    for (const event of events.slice(start)) {\n      yield event;\n    }\n  }\n}\n"],"mappings":";;AAKA,IAAa,0BAAb,MAA2D;CACzD;CACA;CAEA,YAAY,SAAyC,QAAgB;EACnE,KAAKA,UAAU;EACf,KAAKC,WAAW;CAClB;CAEA,MAAM,OAAO,OAAe,OAAyC;EACnE,OAAO,MAAM,gBAAgB,KAAKA,UAAU,OAAO,YAAY;GAC7D,MAAM,SAAS,MAAM,SACnB,SACA,SAAS,KAAKD,SAAS,UAAU,KAAK,CACxC;GACA,MAAM,SAAS,EAAE,QAAQ,OAAO,SAAS,EAAE;GAC3C,OAAO,KAAK;IAAE;IAAQ,OAAO,gBAAgB,KAAK;IAAG;GAAM,CAAC;GAC5D,MAAM,QAAQ,IAAI,SAAS,KAAKA,SAAS,UAAU,KAAK,GAAG,MAAM;GACjE,OAAO;EACT,CAAC;CACH;CAEA,OAAO,KACL,OACA,QACiC;EACjC,MAAM,QAAQ,QAAQ;EACtB,MAAM,SAAS,MAAM,SACnB,KAAKC,UACL,SAAS,KAAKD,SAAS,UAAU,KAAK,CACxC;EACA,MAAM,QAAQ,QAAQ,UAAU;EAChC,KAAK,MAAM,SAAS,OAAO,MAAM,KAAK,GACpC,MAAM;CAEV;AACF"}

package/dist/cloudflare/cloudflare-execution-session-store.js

@@ -0,0 +1,40 @@
+import { getSession, putSession, storeKey, withTransaction } from "./cloudflare-store-utils.js";
+//#region src/cloudflare/cloudflare-execution-session-store.ts
+var DurableObjectExecutionSessionStore = class {
+	#prefix;
+	#storage;
+	constructor(storage, prefix) {
+		this.#prefix = prefix;
+		this.#storage = storage;
+	}
+	async commit(sessionKey, next, options) {
+		return await withTransaction(this.#storage, async (storage) => {
+			const currentVersion = (await getSession(storage, this.#prefix, sessionKey))?.version ?? null;
+			if (options.expectedVersion !== currentVersion) return {
+				ok: false,
+				reason: "conflict"
+			};
+			const version = String((await storage.get(storeKey(this.#prefix, "session-version", sessionKey)) ?? 0) + 1);
+			await storage.put(storeKey(this.#prefix, "session-version", sessionKey), Number(version));
+			await putSession(storage, this.#prefix, sessionKey, {
+				state: next.state,
+				version
+			});
+			return {
+				ok: true,
+				version
+			};
+		});
+	}
+	async delete(sessionKey) {
+		await this.#storage.delete(storeKey(this.#prefix, "session", sessionKey));
+		await this.#storage.delete(storeKey(this.#prefix, "session-version", sessionKey));
+	}
+	async load(sessionKey) {
+		return await getSession(this.#storage, this.#prefix, sessionKey);
+	}
+};
+//#endregion
+export { DurableObjectExecutionSessionStore };
+
+//# sourceMappingURL=cloudflare-execution-session-store.js.map

package/dist/cloudflare/cloudflare-execution-session-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cloudflare-execution-session-store.js","names":["#prefix","#storage"],"sources":["../../src/cloudflare/cloudflare-execution-session-store.ts"],"sourcesContent":["import type {\n  CommitResult,\n  ExpectedSessionVersion,\n  SessionStore,\n  SessionStoreCommit,\n  StoredSession,\n} from \"../index\";\nimport {\n  getSession,\n  putSession,\n  storeKey,\n  withTransaction,\n} from \"./cloudflare-store-utils\";\nimport type { CloudflareDurableObjectStorage } from \"./durable-object-storage\";\n\nexport class DurableObjectExecutionSessionStore implements SessionStore {\n  readonly #prefix: string;\n  readonly #storage: CloudflareDurableObjectStorage;\n\n  constructor(storage: CloudflareDurableObjectStorage, prefix: string) {\n    this.#prefix = prefix;\n    this.#storage = storage;\n  }\n\n  async commit(\n    sessionKey: string,\n    next: SessionStoreCommit,\n    options: { readonly expectedVersion: ExpectedSessionVersion }\n  ): Promise<CommitResult> {\n    return await withTransaction(this.#storage, async (storage) => {\n      const stored = await getSession(storage, this.#prefix, sessionKey);\n      const currentVersion = stored?.version ?? null;\n      if (options.expectedVersion !== currentVersion) {\n        return { ok: false, reason: \"conflict\" };\n      }\n\n      const version = String(\n        ((await storage.get<number>(\n          storeKey(this.#prefix, \"session-version\", sessionKey)\n        )) ?? 0) + 1\n      );\n      await storage.put(\n        storeKey(this.#prefix, \"session-version\", sessionKey),\n        Number(version)\n      );\n      await putSession(storage, this.#prefix, sessionKey, {\n        state: next.state,\n        version,\n      });\n      return { ok: true, version };\n    });\n  }\n\n  async delete(sessionKey: string): Promise<void> {\n    await this.#storage.delete(storeKey(this.#prefix, \"session\", sessionKey));\n    await this.#storage.delete(\n      storeKey(this.#prefix, \"session-version\", sessionKey)\n    );\n  }\n\n  async load(sessionKey: string): Promise<StoredSession | null> {\n    return await getSession(this.#storage, this.#prefix, sessionKey);\n  }\n}\n"],"mappings":";;AAeA,IAAa,qCAAb,MAAwE;CACtE;CACA;CAEA,YAAY,SAAyC,QAAgB;EACnE,KAAKA,UAAU;EACf,KAAKC,WAAW;CAClB;CAEA,MAAM,OACJ,YACA,MACA,SACuB;EACvB,OAAO,MAAM,gBAAgB,KAAKA,UAAU,OAAO,YAAY;GAE7D,MAAM,kBAAiB,MADF,WAAW,SAAS,KAAKD,SAAS,UAAU,IAClC,WAAW;GAC1C,IAAI,QAAQ,oBAAoB,gBAC9B,OAAO;IAAE,IAAI;IAAO,QAAQ;GAAW;GAGzC,MAAM,UAAU,QACZ,MAAM,QAAQ,IACd,SAAS,KAAKA,SAAS,mBAAmB,UAAU,CACtD,KAAM,KAAK,CACb;GACA,MAAM,QAAQ,IACZ,SAAS,KAAKA,SAAS,mBAAmB,UAAU,GACpD,OAAO,OAAO,CAChB;GACA,MAAM,WAAW,SAAS,KAAKA,SAAS,YAAY;IAClD,OAAO,KAAK;IACZ;GACF,CAAC;GACD,OAAO;IAAE,IAAI;IAAM;GAAQ;EAC7B,CAAC;CACH;CAEA,MAAM,OAAO,YAAmC;EAC9C,MAAM,KAAKC,SAAS,OAAO,SAAS,KAAKD,SAAS,WAAW,UAAU,CAAC;EACxE,MAAM,KAAKC,SAAS,OAClB,SAAS,KAAKD,SAAS,mBAAmB,UAAU,CACtD;CACF;CAEA,MAAM,KAAK,YAAmD;EAC5D,OAAO,MAAM,WAAW,KAAKC,UAAU,KAAKD,SAAS,UAAU;CACjE;AACF"}

package/dist/cloudflare/cloudflare-execution-store.js

@@ -0,0 +1,35 @@
+import { DurableObjectCheckpointStore } from "./cloudflare-checkpoint-store.js";
+import { DurableObjectEventStore } from "./cloudflare-event-store.js";
+import { DurableObjectExecutionSessionStore } from "./cloudflare-execution-session-store.js";
+import { DurableObjectNotificationInbox } from "./cloudflare-notification-store.js";
+import { DurableObjectRunStore } from "./cloudflare-run-store.js";
+//#region src/cloudflare/cloudflare-execution-store.ts
+var DurableObjectExecutionStore = class DurableObjectExecutionStore {
+	checkpoints;
+	events;
+	notifications;
+	runs;
+	sessions;
+	#prefix;
+	#storage;
+	constructor({ prefix = "pss-runtime", storage }) {
+		this.#prefix = prefix;
+		this.#storage = storage;
+		this.checkpoints = new DurableObjectCheckpointStore(storage, prefix);
+		this.events = new DurableObjectEventStore(storage, prefix);
+		this.notifications = new DurableObjectNotificationInbox(storage, prefix);
+		this.runs = new DurableObjectRunStore(storage, prefix);
+		this.sessions = new DurableObjectExecutionSessionStore(storage, prefix);
+	}
+	async transaction(fn) {
+		if (this.#storage.transaction) return await this.#storage.transaction((storage) => fn(new DurableObjectExecutionStore({
+			prefix: this.#prefix,
+			storage
+		})));
+		return await fn(this);
+	}
+};
+//#endregion
+export { DurableObjectExecutionStore };
+
+//# sourceMappingURL=cloudflare-execution-store.js.map

package/dist/cloudflare/cloudflare-execution-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cloudflare-execution-store.js","names":["#prefix","#storage"],"sources":["../../src/cloudflare/cloudflare-execution-store.ts"],"sourcesContent":["import type {\n  CheckpointStore,\n  EventStore,\n  ExecutionStore,\n  ExecutionStoreTransaction,\n  NotificationInbox,\n  RunStore,\n} from \"../execution\";\nimport type { SessionStore } from \"../index\";\nimport { DurableObjectCheckpointStore } from \"./cloudflare-checkpoint-store\";\nimport { DurableObjectEventStore } from \"./cloudflare-event-store\";\nimport { DurableObjectExecutionSessionStore } from \"./cloudflare-execution-session-store\";\nimport { DurableObjectNotificationInbox } from \"./cloudflare-notification-store\";\nimport { DurableObjectRunStore } from \"./cloudflare-run-store\";\nimport type { CloudflareDurableObjectStorage } from \"./durable-object-storage\";\n\nexport class DurableObjectExecutionStore implements ExecutionStore {\n  readonly checkpoints: CheckpointStore;\n  readonly events: EventStore;\n  readonly notifications: NotificationInbox;\n  readonly runs: RunStore;\n  readonly sessions: SessionStore;\n  readonly #prefix: string;\n  readonly #storage: CloudflareDurableObjectStorage;\n\n  constructor({\n    prefix = \"pss-runtime\",\n    storage,\n  }: {\n    readonly prefix?: string;\n    readonly storage: CloudflareDurableObjectStorage;\n  }) {\n    this.#prefix = prefix;\n    this.#storage = storage;\n    this.checkpoints = new DurableObjectCheckpointStore(storage, prefix);\n    this.events = new DurableObjectEventStore(storage, prefix);\n    this.notifications = new DurableObjectNotificationInbox(storage, prefix);\n    this.runs = new DurableObjectRunStore(storage, prefix);\n    this.sessions = new DurableObjectExecutionSessionStore(storage, prefix);\n  }\n\n  async transaction<T>(\n    fn: (tx: ExecutionStoreTransaction) => Promise<T>\n  ): Promise<T> {\n    if (this.#storage.transaction) {\n      return await this.#storage.transaction((storage) =>\n        fn(new DurableObjectExecutionStore({ prefix: this.#prefix, storage }))\n      );\n    }\n\n    return await fn(this);\n  }\n}\n"],"mappings":";;;;;;AAgBA,IAAa,8BAAb,MAAa,4BAAsD;CACjE;CACA;CACA;CACA;CACA;CACA;CACA;CAEA,YAAY,EACV,SAAS,eACT,WAIC;EACD,KAAKA,UAAU;EACf,KAAKC,WAAW;EAChB,KAAK,cAAc,IAAI,6BAA6B,SAAS,MAAM;EACnE,KAAK,SAAS,IAAI,wBAAwB,SAAS,MAAM;EACzD,KAAK,gBAAgB,IAAI,+BAA+B,SAAS,MAAM;EACvE,KAAK,OAAO,IAAI,sBAAsB,SAAS,MAAM;EACrD,KAAK,WAAW,IAAI,mCAAmC,SAAS,MAAM;CACxE;CAEA,MAAM,YACJ,IACY;EACZ,IAAI,KAAKA,SAAS,aAChB,OAAO,MAAM,KAAKA,SAAS,aAAa,YACtC,GAAG,IAAI,4BAA4B;GAAE,QAAQ,KAAKD;GAAS;EAAQ,CAAC,CAAC,CACvE;EAGF,OAAO,MAAM,GAAG,IAAI;CACtB;AACF"}