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

package/dist/agent-resume.js

@@ -0,0 +1,143 @@
+import { readDurableBackgroundChildRunState } from "./subagent-background-child-run-state.js";
+import { ownsAgentNamespace } from "./agent-namespace.js";
+import { StoredAgentRun } from "./execution/run.js";
+import { BufferedAgentRun } from "./session/run.js";
+import { buildDurableResumeGroups } from "./subagent-background-resume-group.js";
+import { runBackgroundJob } from "./subagent-background-runner.js";
+//#region src/agent-resume.ts
+const defaultResumeLeaseMs = 3e5;
+async function resumeAgentRun({ host, ownerNamespace, resumeNotification, runId, subagents }) {
+	const run = await host.store.runs.get(runId);
+	if (!run) return null;
+	if (!canAccessRun(run, ownerNamespace)) return null;
+	if (run.kind === "background-subagent") return await resumeBackgroundSubagentRun({
+		host,
+		run,
+		subagents
+	});
+	if (run.kind === "notification" && run.dedupeKey) {
+		const idempotencyKey = run.dedupeKey;
+		const claimed = await claimRun(host, run);
+		if (!claimed) return null;
+		const notification = await claimNotificationForRun({
+			host,
+			idempotencyKey,
+			ownerNamespace
+		});
+		if (!notification) return null;
+		try {
+			const notificationRun = await resumeNotification(notification);
+			await completeNotificationRun(host, claimed.runId);
+			return notificationRun;
+		} catch (error) {
+			await host.store.notifications.releaseByIdempotencyKey(idempotencyKey);
+			throw error;
+		}
+	}
+	return null;
+}
+async function claimNotificationForRun({ host, idempotencyKey, ownerNamespace }) {
+	if (!ownsAgentNamespace((await host.store.notifications.getByIdempotencyKey(idempotencyKey))?.ownerNamespace, ownerNamespace)) return null;
+	const claim = await host.store.notifications.claimByIdempotencyKey(idempotencyKey);
+	if (claim.ok) {
+		if (ownsAgentNamespace(claim.record.ownerNamespace, ownerNamespace)) return claim.record;
+		await host.store.notifications.releaseByIdempotencyKey(idempotencyKey);
+		return null;
+	}
+	if (claim.reason === "already-claimed" && ownsAgentNamespace(claim.record?.ownerNamespace, ownerNamespace)) return claim.record ?? null;
+	return null;
+}
+function canAccessRun(run, ownerNamespace) {
+	if (run.ownerNamespace) return ownsAgentNamespace(run.ownerNamespace, ownerNamespace);
+	return run.sessionKey.startsWith(`parent:${ownerNamespace}:`) || run.parentRunId?.startsWith(`${ownerNamespace}:session:`) === true;
+}
+async function completeNotificationRun(host, runId) {
+	const run = await host.store.runs.get(runId);
+	if (run?.kind !== "notification" || run.status === "completed") return;
+	await host.store.runs.update({
+		...run,
+		status: "completed"
+	});
+}
+async function resumeBackgroundSubagentRun({ host, run, subagents }) {
+	const claimed = await claimRun(host, run);
+	if (!claimed) return null;
+	const state = readDurableBackgroundChildRunState(await host.store.checkpoints.latest(run.runId));
+	if (!state) throw new AgentResumeError(run.runId, "missing background run state");
+	const subagent = subagents.find((candidate) => candidate.name === state.subagent);
+	if (!subagent) throw new AgentResumeError(run.runId, `missing subagent ${JSON.stringify(state.subagent)}`);
+	const childSession = subagent.session(claimed.sessionKey);
+	const job = {
+		abort: () => childSession.interrupt(),
+		childRunId: claimed.runId,
+		childRunLeaseId: claimed.lease?.leaseId,
+		cleanup: () => Promise.resolve(),
+		dedupeKey: claimed.dedupeKey,
+		delegateToolCallId: state.delegateToolCallId,
+		description: state.description,
+		executionHost: host,
+		groupId: state.groupId,
+		id: claimed.publicTaskId ?? claimed.runId,
+		ownerNamespace: claimed.ownerNamespace,
+		parentRunId: claimed.parentRunId,
+		parentSessionKey: state.parentSessionKey,
+		promise: Promise.resolve(),
+		sessionKey: claimed.sessionKey,
+		settled: false,
+		status: "running",
+		subagent: state.subagent
+	};
+	const jobs = new Map([[job.id, job]]);
+	job.promise = runBackgroundJob({
+		childSession,
+		groups: await buildDurableResumeGroups({
+			currentJob: job,
+			host,
+			run: claimed,
+			state,
+			jobs
+		}),
+		jobs,
+		job,
+		parentSession: durableParentSession(host, run.runId),
+		prompt: state.prompt
+	}).finally(() => {
+		job.settled = true;
+	});
+	await job.promise;
+	return new StoredAgentRun({
+		eventStore: host.store.events,
+		runId: run.runId
+	});
+}
+async function claimRun(host, run) {
+	const claim = await host.store.runs.claim(run.runId, {
+		attempt: (run.lease?.attempt ?? 0) + 1,
+		leaseId: crypto.randomUUID(),
+		leaseMs: defaultResumeLeaseMs,
+		nowMs: Date.now()
+	});
+	return claim.ok ? claim.record : null;
+}
+function durableParentSession(host, runId) {
+	return {
+		emitObserverEvent: (event) => host.store.events.append(runId, event).then(),
+		enqueueRuntimeInput: () => void 0,
+		notify: () => Promise.resolve(emptyRun())
+	};
+}
+function emptyRun() {
+	const run = new BufferedAgentRun();
+	run.close();
+	return run;
+}
+var AgentResumeError = class extends Error {
+	constructor(runId, reason) {
+		super(`Cannot resume agent run ${runId}: ${reason}`);
+		this.name = "AgentResumeError";
+	}
+};
+//#endregion
+export { resumeAgentRun };
+
+//# sourceMappingURL=agent-resume.js.map

package/dist/agent-resume.js.map

@@ -0,0 +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"}

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

@@ -0,0 +1,13 @@
+import { AgentInput } from "./session/input.js";
+import { AgentRun } from "./session/run.js";
+//#region src/agent-session-entry.d.ts
+interface SessionHandle {
+  delete(): Promise<void>;
+  interrupt(): void;
+  kill(): Promise<void>;
+  send(input: AgentInput): Promise<AgentRun>;
+  steer(input: AgentInput): Promise<AgentRun>;
+}
+//#endregion
+export { SessionHandle };
+//# sourceMappingURL=agent-session-entry.d.ts.map

package/dist/agent-validation.js

@@ -1,8 +1,8 @@
 //#region src/agent-validation.ts
 const subagentNamePattern = /^[a-z][a-z0-9_-]{0,51}$/;
-function assertSubagents(options, agentClass, hasCustomLlm) {
+function assertSubagents(options, agentClass, hasRuntimeModel) {
 	if (!("subagents" in options) || options.subagents === void 0) return;
-	if (hasCustomLlm) throw new TypeError("Agent: subagents require options.model.");
+	if (hasRuntimeModel) throw new TypeError("Agent: subagents require an AI SDK model.");
 	if (!Array.isArray(options.subagents)) throw new TypeError("Agent: subagents must be an array.");
 	assertSubagentTools(options.subagents, agentClass, options.tools ?? {});
 }

package/dist/agent-validation.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent-validation.js","names":[],"sources":["../src/agent-validation.ts"],"sourcesContent":["import type { ToolSet } from \"ai\";\nimport type { Agent, AgentOptions } from \"./agent\";\n\nconst subagentNamePattern = /^[a-z][a-z0-9_-]{0,51}$/;\n\nexport function assertSubagents(\n  options: AgentOptions,\n  agentClass: new (options: AgentOptions) => Agent,\n  hasCustomLlm: boolean\n): void {\n  if (!(\"subagents\" in options) || options.subagents === undefined) {\n    return;\n  }\n\n  if (hasCustomLlm) {\n    throw new TypeError(\"Agent: subagents require options.model.\");\n  }\n\n  if (!Array.isArray(options.subagents)) {\n    throw new TypeError(\"Agent: subagents must be an array.\");\n  }\n\n  assertSubagentTools(options.subagents, agentClass, options.tools ?? {});\n}\n\nfunction assertSubagentTools(\n  subagents: readonly Agent[],\n  agentClass: new (options: AgentOptions) => Agent,\n  tools: ToolSet\n): void {\n  const toolNames = new Set(Object.keys(tools));\n  const generatedToolNames = new Set<string>();\n  for (const [index, subagent] of subagents.entries()) {\n    const name = assertSubagentMetadata(subagent, index, agentClass);\n    const toolName = `delegate_to_${name.replaceAll(\"-\", \"_\")}`;\n    if (toolNames.has(toolName)) {\n      throw new TypeError(\n        `Agent: subagent tool ${toolName} collides with an existing tool.`\n      );\n    }\n\n    if (generatedToolNames.has(toolName)) {\n      throw new TypeError(`Agent: duplicate subagent tool name ${toolName}.`);\n    }\n\n    generatedToolNames.add(toolName);\n  }\n\n  for (const reservedToolName of [\"background_output\", \"background_cancel\"]) {\n    if (toolNames.has(reservedToolName)) {\n      throw new TypeError(\n        `Agent: ${reservedToolName} collides with a reserved subagent tool.`\n      );\n    }\n  }\n}\n\nfunction assertSubagentMetadata(\n  subagent: Agent,\n  index: number,\n  agentClass: new (options: AgentOptions) => Agent\n): string {\n  if (!(subagent instanceof agentClass)) {\n    throw new TypeError(`Agent: subagents[${index}] must be an Agent.`);\n  }\n\n  if (!isValidSubagentName(subagent.name)) {\n    throw new TypeError(\n      `Agent: subagents[${index}].name is required or too long.`\n    );\n  }\n\n  if (!isNonEmptyText(subagent.description)) {\n    throw new TypeError(`Agent: subagents[${index}].description is required.`);\n  }\n\n  return subagent.name;\n}\n\nfunction isNonEmptyText(value: string | undefined): value is string {\n  return typeof value === \"string\" && value.trim().length > 0;\n}\n\nfunction isValidSubagentName(value: string | undefined): value is string {\n  return typeof value === \"string\" && subagentNamePattern.test(value);\n}\n"],"mappings":";AAGA,MAAM,sBAAsB;AAE5B,SAAgB,gBACd,SACA,YACA,cACM;CACN,IAAI,EAAE,eAAe,YAAY,QAAQ,cAAc,KAAA,GACrD;CAGF,IAAI,cACF,MAAM,IAAI,UAAU,yCAAyC;CAG/D,IAAI,CAAC,MAAM,QAAQ,QAAQ,SAAS,GAClC,MAAM,IAAI,UAAU,oCAAoC;CAG1D,oBAAoB,QAAQ,WAAW,YAAY,QAAQ,SAAS,CAAC,CAAC;AACxE;AAEA,SAAS,oBACP,WACA,YACA,OACM;CACN,MAAM,YAAY,IAAI,IAAI,OAAO,KAAK,KAAK,CAAC;CAC5C,MAAM,qCAAqB,IAAI,IAAY;CAC3C,KAAK,MAAM,CAAC,OAAO,aAAa,UAAU,QAAQ,GAAG;EAEnD,MAAM,WAAW,eADJ,uBAAuB,UAAU,OAAO,UAClB,EAAE,WAAW,KAAK,GAAG;EACxD,IAAI,UAAU,IAAI,QAAQ,GACxB,MAAM,IAAI,UACR,wBAAwB,SAAS,iCACnC;EAGF,IAAI,mBAAmB,IAAI,QAAQ,GACjC,MAAM,IAAI,UAAU,uCAAuC,SAAS,EAAE;EAGxE,mBAAmB,IAAI,QAAQ;CACjC;CAEA,KAAK,MAAM,oBAAoB,CAAC,qBAAqB,mBAAmB,GACtE,IAAI,UAAU,IAAI,gBAAgB,GAChC,MAAM,IAAI,UACR,UAAU,iBAAiB,yCAC7B;AAGN;AAEA,SAAS,uBACP,UACA,OACA,YACQ;CACR,IAAI,EAAE,oBAAoB,aACxB,MAAM,IAAI,UAAU,oBAAoB,MAAM,oBAAoB;CAGpE,IAAI,CAAC,oBAAoB,SAAS,IAAI,GACpC,MAAM,IAAI,UACR,oBAAoB,MAAM,gCAC5B;CAGF,IAAI,CAAC,eAAe,SAAS,WAAW,GACtC,MAAM,IAAI,UAAU,oBAAoB,MAAM,2BAA2B;CAG3E,OAAO,SAAS;AAClB;AAEA,SAAS,eAAe,OAA4C;CAClE,OAAO,OAAO,UAAU,YAAY,MAAM,KAAK,EAAE,SAAS;AAC5D;AAEA,SAAS,oBAAoB,OAA4C;CACvE,OAAO,OAAO,UAAU,YAAY,oBAAoB,KAAK,KAAK;AACpE"}
+{"version":3,"file":"agent-validation.js","names":[],"sources":["../src/agent-validation.ts"],"sourcesContent":["import type { ToolSet } from \"ai\";\nimport type { Agent } from \"./agent\";\nimport type { AgentOptions } from \"./agent-options\";\n\nconst subagentNamePattern = /^[a-z][a-z0-9_-]{0,51}$/;\n\nexport function assertSubagents(\n  options: AgentOptions,\n  agentClass: new (options: AgentOptions) => Agent,\n  hasRuntimeModel: boolean\n): void {\n  if (!(\"subagents\" in options) || options.subagents === undefined) {\n    return;\n  }\n\n  if (hasRuntimeModel) {\n    throw new TypeError(\"Agent: subagents require an AI SDK model.\");\n  }\n\n  if (!Array.isArray(options.subagents)) {\n    throw new TypeError(\"Agent: subagents must be an array.\");\n  }\n\n  assertSubagentTools(options.subagents, agentClass, options.tools ?? {});\n}\n\nfunction assertSubagentTools(\n  subagents: readonly Agent[],\n  agentClass: new (options: AgentOptions) => Agent,\n  tools: ToolSet\n): void {\n  const toolNames = new Set(Object.keys(tools));\n  const generatedToolNames = new Set<string>();\n  for (const [index, subagent] of subagents.entries()) {\n    const name = assertSubagentMetadata(subagent, index, agentClass);\n    const toolName = `delegate_to_${name.replaceAll(\"-\", \"_\")}`;\n    if (toolNames.has(toolName)) {\n      throw new TypeError(\n        `Agent: subagent tool ${toolName} collides with an existing tool.`\n      );\n    }\n\n    if (generatedToolNames.has(toolName)) {\n      throw new TypeError(`Agent: duplicate subagent tool name ${toolName}.`);\n    }\n\n    generatedToolNames.add(toolName);\n  }\n\n  for (const reservedToolName of [\"background_output\", \"background_cancel\"]) {\n    if (toolNames.has(reservedToolName)) {\n      throw new TypeError(\n        `Agent: ${reservedToolName} collides with a reserved subagent tool.`\n      );\n    }\n  }\n}\n\nfunction assertSubagentMetadata(\n  subagent: Agent,\n  index: number,\n  agentClass: new (options: AgentOptions) => Agent\n): string {\n  if (!(subagent instanceof agentClass)) {\n    throw new TypeError(`Agent: subagents[${index}] must be an Agent.`);\n  }\n\n  if (!isValidSubagentName(subagent.name)) {\n    throw new TypeError(\n      `Agent: subagents[${index}].name is required or too long.`\n    );\n  }\n\n  if (!isNonEmptyText(subagent.description)) {\n    throw new TypeError(`Agent: subagents[${index}].description is required.`);\n  }\n\n  return subagent.name;\n}\n\nfunction isNonEmptyText(value: string | undefined): value is string {\n  return typeof value === \"string\" && value.trim().length > 0;\n}\n\nfunction isValidSubagentName(value: string | undefined): value is string {\n  return typeof value === \"string\" && subagentNamePattern.test(value);\n}\n"],"mappings":";AAIA,MAAM,sBAAsB;AAE5B,SAAgB,gBACd,SACA,YACA,iBACM;CACN,IAAI,EAAE,eAAe,YAAY,QAAQ,cAAc,KAAA,GACrD;CAGF,IAAI,iBACF,MAAM,IAAI,UAAU,2CAA2C;CAGjE,IAAI,CAAC,MAAM,QAAQ,QAAQ,SAAS,GAClC,MAAM,IAAI,UAAU,oCAAoC;CAG1D,oBAAoB,QAAQ,WAAW,YAAY,QAAQ,SAAS,CAAC,CAAC;AACxE;AAEA,SAAS,oBACP,WACA,YACA,OACM;CACN,MAAM,YAAY,IAAI,IAAI,OAAO,KAAK,KAAK,CAAC;CAC5C,MAAM,qCAAqB,IAAI,IAAY;CAC3C,KAAK,MAAM,CAAC,OAAO,aAAa,UAAU,QAAQ,GAAG;EAEnD,MAAM,WAAW,eADJ,uBAAuB,UAAU,OAAO,UAClB,EAAE,WAAW,KAAK,GAAG;EACxD,IAAI,UAAU,IAAI,QAAQ,GACxB,MAAM,IAAI,UACR,wBAAwB,SAAS,iCACnC;EAGF,IAAI,mBAAmB,IAAI,QAAQ,GACjC,MAAM,IAAI,UAAU,uCAAuC,SAAS,EAAE;EAGxE,mBAAmB,IAAI,QAAQ;CACjC;CAEA,KAAK,MAAM,oBAAoB,CAAC,qBAAqB,mBAAmB,GACtE,IAAI,UAAU,IAAI,gBAAgB,GAChC,MAAM,IAAI,UACR,UAAU,iBAAiB,yCAC7B;AAGN;AAEA,SAAS,uBACP,UACA,OACA,YACQ;CACR,IAAI,EAAE,oBAAoB,aACxB,MAAM,IAAI,UAAU,oBAAoB,MAAM,oBAAoB;CAGpE,IAAI,CAAC,oBAAoB,SAAS,IAAI,GACpC,MAAM,IAAI,UACR,oBAAoB,MAAM,gCAC5B;CAGF,IAAI,CAAC,eAAe,SAAS,WAAW,GACtC,MAAM,IAAI,UAAU,oBAAoB,MAAM,2BAA2B;CAG3E,OAAO,SAAS;AAClB;AAEA,SAAS,eAAe,OAA4C;CAClE,OAAO,OAAO,UAAU,YAAY,MAAM,KAAK,EAAE,SAAS;AAC5D;AAEA,SAAS,oBAAoB,OAA4C;CACvE,OAAO,OAAO,UAAU,YAAY,oBAAoB,KAAK,KAAK;AACpE"}

package/dist/agent.d.ts

@@ -1,55 +1,19 @@
-import { AgentToolChoice, Llm } from "./llm.js";
 import { AgentInput } from "./session/input.js";
-import { AgentHooks } from "./hooks.js";
+import { AgentHost } from "./execution/types.js";
+import { AgentOptions } from "./agent-options.js";
 import { AgentRun } from "./session/run.js";
-import { SessionStore } from "./session/store/types.js";
-import { LanguageModel, ToolSet } from "ai";
+import { SessionHandle } from "./agent-session-entry.js";
 
 //#region src/agent.d.ts
-interface AgentLanguageModelOptions {
-  description?: string;
-  hooks?: AgentHooks;
-  instructions?: string;
-  llm?: never;
-  model: LanguageModel;
-  name?: string;
-  sessions?: AgentSessionOptions;
-  subagents?: readonly Agent[];
-  toolChoice?: AgentToolChoice;
-  tools?: ToolSet;
-}
-interface AgentLlmOptions {
-  description?: string;
-  hooks?: AgentHooks;
-  instructions?: never;
-  llm: Llm;
-  model?: never;
-  name?: string;
-  sessions?: AgentSessionOptions;
-  subagents?: never;
-  toolChoice?: never;
-  tools?: never;
-}
-interface AgentSessionOptions {
-  namespace?: string;
-  store?: SessionStore;
-}
-interface SessionHandle {
-  delete(): Promise<void>;
-  interrupt(): void;
-  kill(): void;
-  send(input: AgentInput): Promise<AgentRun>;
-  steer(input: AgentInput): Promise<AgentRun>;
-}
-type AgentOptions = AgentLanguageModelOptions | AgentLlmOptions;
 declare class Agent {
   #private;
   readonly description?: string;
   readonly name?: string;
   constructor(options: AgentOptions);
   send(input: AgentInput): Promise<AgentRun>;
+  resume(runId: string): Promise<AgentRun | null>;
   session(key: string): SessionHandle;
 }
 //#endregion
-export { Agent, AgentOptions, AgentSessionOptions, SessionHandle };
+export { Agent };
 //# sourceMappingURL=agent.d.ts.map

package/dist/agent.js

@@ -1,14 +1,19 @@
-import { agentNamespace, parentSessionNamespace, randomAgentNamespace } from "./agent-namespace.js";
+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 { 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 { MemorySessionStore } from "./session/store/memory.js";
 import { createSubagentTools } from "./subagents.js";
 //#region src/agent.ts
 var Agent = class Agent {
 	#baseTools;
-	#hooks;
 	#llm;
 	#modelOptions;
 	#childSessionCleanups = new ChildSessionCleanups();
@@ -16,6 +21,8 @@ var Agent = class Agent {
 	#sessions = /* @__PURE__ */ new Map();
 	#sessionNamespace;
 	#store;
+	#host;
+	#plugins;
 	#subagents;
 	description;
 	name;
@@ -23,12 +30,16 @@ var Agent = class Agent {
 		assertAgentOptions(options);
 		this.description = options.description;
 		this.name = options.name;
-		this.#sessionNamespace = stableAgentNamespace(options);
-		this.#store = options.sessions?.store ?? new MemorySessionStore();
-		this.#hooks = options.hooks;
-		assertSubagents(options, Agent, hasCustomLlm(options));
-		this.#subagents = hasCustomLlm(options) ? [] : options.subagents ?? [];
-		if (hasCustomLlm(options)) this.#llm = options.llm;
+		this.#sessionNamespace = stableAgentNamespace({
+			name: options.name,
+			namespace: options.namespace
+		});
+		this.#host = options.host ?? createInMemoryExecutionHost();
+		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 {
 			this.#baseTools = options.tools;
 			this.#modelOptions = {
@@ -41,7 +52,21 @@ var Agent = class Agent {
 	send(input) {
 		return this.session("default").send(input);
 	}
+	async resume(runId) {
+		const host = executionHost(this.#host);
+		if (!host) throw new Error("Agent host does not support durable run resume.");
+		return await resumeAgentRun({
+			host,
+			ownerNamespace: this.#sessionNamespace,
+			resumeNotification: (notification) => this.#resumeNotification(notification),
+			runId,
+			subagents: this.#subagents
+		});
+	}
 	session(key) {
+		return this.#sessionEntry(key).publicHandle;
+	}
+	#sessionEntry(key) {
 		const existing = this.#sessions.get(key);
 		if (existing) return existing;
 		let session;
@@ -54,40 +79,65 @@ var Agent = class Agent {
 			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))), {
+		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)), {
 			key,
 			store: this.#store
-		}, this.#hooks);
-		const handle = {
-			delete: async () => {
-				await session.delete();
-				this.#sessions.delete(key);
-				this.#sessionGenerations.set(key, (this.#sessionGenerations.get(key) ?? 0) + 1);
-				await this.#deleteChildSessions(key);
-			},
-			interrupt: () => session.interrupt(),
-			kill: () => {
-				session.kill();
-				this.#sessionGenerations.set(key, (this.#sessionGenerations.get(key) ?? 0) + 1);
-				this.#deleteChildSessions(key).catch(() => void 0);
-				this.#sessions.delete(key);
-			},
-			send: (input) => session.send(input),
-			steer: (input) => session.steer(input)
+		}, this.#plugins, { executionHost: executionHost(this.#host) });
+		const entry = {
+			notify: (input, options) => session.notify(input, options),
+			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 () => {
+					session.kill();
+					await this.#cancelDurableChildRunsBeforeLocalCleanup(key, parentAgentNamespace);
+					this.#evictSessionHandle(key);
+					await this.#childSessionCleanups.delete(key);
+				},
+				send: (input) => session.send(input),
+				steer: (input) => session.steer(input)
+			}
 		};
-		this.#sessions.set(key, handle);
-		return handle;
+		this.#sessions.set(key, entry);
+		return entry;
 	}
-	#createLlmOptionsForSession(key, parentAgentNamespace, enqueueRuntimeInput, emitObserverEvent) {
+	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) {
 		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
+					enqueueRuntimeInput,
+					notify
 				},
 				parentSessionKey: key,
 				registerChildSession: (sessionKey, cleanup) => this.#childSessionCleanups.register(sessionKey, cleanup),
@@ -101,25 +151,7 @@ var Agent = class Agent {
 			tools
 		};
 	}
-	async #deleteChildSessions(parentSessionKey) {
-		await this.#childSessionCleanups.delete(parentSessionKey);
-	}
 };
-function stableAgentNamespace(options) {
-	const namespace = options.sessions?.namespace ?? options.name;
-	return namespace ? agentNamespace(namespace) : randomAgentNamespace();
-}
-function assertAgentOptions(options) {
-	if (options === null || typeof options !== "object") throw new TypeError("Agent options are required. Provide either { model } or { llm }.");
-	const hasLlm = hasCustomLlm(options);
-	const hasModel = "model" in options && options.model != null;
-	if (hasLlm && hasModel) throw new TypeError("Agent: provide either options.llm or options.model.");
-	if ("llm" in options && options.llm !== void 0 && !hasLlm) throw new TypeError("Agent: invalid options.llm.");
-	if (!(hasLlm || hasModel)) throw new TypeError("Agent: missing options.model.");
-}
-function hasCustomLlm(options) {
-	return "llm" in options && typeof options.llm === "function";
-}
 //#endregion
 export { Agent };
 

package/dist/agent.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent.js","names":["#baseTools","#hooks","#llm","#modelOptions","#childSessionCleanups","#sessionGenerations","#sessions","#sessionNamespace","#store","#subagents","#createLlmOptionsForSession","#deleteChildSessions"],"sources":["../src/agent.ts"],"sourcesContent":["import type { LanguageModel, ToolSet } from \"ai\";\nimport {\n  agentNamespace,\n  parentSessionNamespace,\n  randomAgentNamespace,\n} from \"./agent-namespace\";\nimport { assertSubagents } from \"./agent-validation\";\nimport { ChildSessionCleanups } from \"./child-session-cleanups\";\nimport type { AgentHooks } from \"./hooks\";\nimport { type AgentToolChoice, createLlm, type Llm } from \"./llm\";\nimport type { UserInput } from \"./session/events\";\nimport type { AgentRun } from \"./session/run\";\nimport { type AgentInput, AgentSession } from \"./session/session\";\nimport { MemorySessionStore } from \"./session/store/memory\";\nimport type { SessionStore } from \"./session/store/types\";\nimport { createSubagentTools } from \"./subagents\";\n\ninterface AgentLanguageModelOptions {\n  description?: string;\n  hooks?: AgentHooks;\n  instructions?: string;\n  llm?: never;\n  model: LanguageModel;\n  name?: string;\n  sessions?: AgentSessionOptions;\n  subagents?: readonly Agent[];\n  toolChoice?: AgentToolChoice;\n  tools?: ToolSet;\n}\n\ninterface AgentLlmOptions {\n  description?: string;\n  hooks?: AgentHooks;\n  instructions?: never;\n  llm: Llm;\n  model?: never;\n  name?: string;\n  sessions?: AgentSessionOptions;\n  subagents?: never;\n  toolChoice?: never;\n  tools?: never;\n}\n\nexport interface AgentSessionOptions {\n  namespace?: string;\n  store?: SessionStore;\n}\n\nexport interface SessionHandle {\n  delete(): Promise<void>;\n  interrupt(): void;\n  kill(): void;\n  send(input: AgentInput): Promise<AgentRun>;\n  steer(input: AgentInput): Promise<AgentRun>;\n}\n\nexport type AgentOptions = AgentLanguageModelOptions | AgentLlmOptions;\ntype AgentModelOptions = Pick<\n  AgentLanguageModelOptions,\n  \"instructions\" | \"model\" | \"toolChoice\"\n>;\n\nexport class Agent {\n  readonly #baseTools?: ToolSet;\n  readonly #hooks?: AgentHooks;\n  readonly #llm?: Llm;\n  readonly #modelOptions?: AgentModelOptions;\n  readonly #childSessionCleanups = new ChildSessionCleanups();\n  readonly #sessionGenerations = new Map<string, number>();\n  readonly #sessions = new Map<string, SessionHandle>();\n  readonly #sessionNamespace: string;\n  readonly #store: SessionStore;\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(options);\n    this.#store = options.sessions?.store ?? new MemorySessionStore();\n    this.#hooks = options.hooks;\n    assertSubagents(options, Agent, hasCustomLlm(options));\n    this.#subagents = hasCustomLlm(options) ? [] : (options.subagents ?? []);\n    if (hasCustomLlm(options)) {\n      this.#llm = options.llm;\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  session(key: string): SessionHandle {\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        )\n      );\n    session = new AgentSession(llm, { key, store: this.#store }, this.#hooks);\n    const handle: SessionHandle = {\n      delete: async () => {\n        await session.delete();\n        this.#sessions.delete(key);\n        this.#sessionGenerations.set(\n          key,\n          (this.#sessionGenerations.get(key) ?? 0) + 1\n        );\n        await this.#deleteChildSessions(key);\n      },\n      interrupt: () => session.interrupt(),\n      kill: () => {\n        session.kill();\n        this.#sessionGenerations.set(\n          key,\n          (this.#sessionGenerations.get(key) ?? 0) + 1\n        );\n        this.#deleteChildSessions(key).catch(() => undefined);\n        this.#sessions.delete(key);\n      },\n      send: (input) => session.send(input),\n      steer: (input) => session.steer(input),\n    };\n    this.#sessions.set(key, handle);\n    return handle;\n  }\n\n  #createLlmOptionsForSession(\n    key: string,\n    parentAgentNamespace: string,\n    enqueueRuntimeInput: AgentSession[\"enqueueRuntimeInput\"],\n    emitObserverEvent: AgentSession[\"emitObserverEvent\"]\n  ): Parameters<typeof createLlm>[0] {\n    const modelOptions = this.#modelOptions;\n    if (!modelOptions) {\n      throw new Error(\"Agent: missing model options.\");\n    }\n    const tools =\n      this.#subagents.length === 0\n        ? this.#baseTools\n        : {\n            ...this.#baseTools,\n            ...createSubagentTools({\n              parentAgentNamespace,\n              parentSession: { emitObserverEvent, enqueueRuntimeInput },\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  async #deleteChildSessions(parentSessionKey: string): Promise<void> {\n    await this.#childSessionCleanups.delete(parentSessionKey);\n  }\n}\n\nfunction stableAgentNamespace(options: AgentOptions): string {\n  const namespace = options.sessions?.namespace ?? options.name;\n  return namespace ? agentNamespace(namespace) : randomAgentNamespace();\n}\n\nfunction assertAgentOptions(options: unknown): asserts options is AgentOptions {\n  if (options === null || typeof options !== \"object\") {\n    throw new TypeError(\n      \"Agent options are required. Provide either { model } or { llm }.\"\n    );\n  }\n\n  const hasLlm = hasCustomLlm(options);\n  const hasModel = \"model\" in options && options.model != null;\n\n  if (hasLlm && hasModel) {\n    throw new TypeError(\"Agent: provide either options.llm or options.model.\");\n  }\n\n  if (\"llm\" in options && options.llm !== undefined && !hasLlm) {\n    throw new TypeError(\"Agent: invalid options.llm.\");\n  }\n\n  if (!(hasLlm || hasModel)) {\n    throw new TypeError(\"Agent: missing options.model.\");\n  }\n}\n\nfunction hasCustomLlm(options: object): options is AgentLlmOptions {\n  return \"llm\" in options && typeof options.llm === \"function\";\n}\n"],"mappings":";;;;;;;;AA8DA,IAAa,QAAb,MAAa,MAAM;CACjB;CACA;CACA;CACA;CACA,wBAAiC,IAAI,qBAAqB;CAC1D,sCAA+B,IAAI,IAAoB;CACvD,4BAAqB,IAAI,IAA2B;CACpD;CACA;CACA;CACA;CACA;CAEA,YAAY,SAAuB;EACjC,mBAAmB,OAAO;EAE1B,KAAK,cAAc,QAAQ;EAC3B,KAAK,OAAO,QAAQ;EACpB,KAAKO,oBAAoB,qBAAqB,OAAO;EACrD,KAAKC,SAAS,QAAQ,UAAU,SAAS,IAAI,mBAAmB;EAChE,KAAKP,SAAS,QAAQ;EACtB,gBAAgB,SAAS,OAAO,aAAa,OAAO,CAAC;EACrD,KAAKQ,aAAa,aAAa,OAAO,IAAI,CAAC,IAAK,QAAQ,aAAa,CAAC;EACtE,IAAI,aAAa,OAAO,GACtB,KAAKP,OAAO,QAAQ;OACf;GACL,KAAKF,aAAa,QAAQ;GAC1B,KAAKG,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,QAAQ,KAA4B;EAClC,MAAM,WAAW,KAAKG,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;EAYD,UAAU,IAAI,aAVZ,KAAKL,QACL,UACE,KAAKQ,4BACH,KACA,uBACC,OAAkB,cACjB,WAAW,EAAE,oBAAoB,OAAO,SAAS,IAClD,UAAU,WAAW,EAAE,kBAAkB,KAAK,CACjD,CACF,GAC8B;GAAE;GAAK,OAAO,KAAKF;EAAO,GAAG,KAAKP,MAAM;EACxE,MAAM,SAAwB;GAC5B,QAAQ,YAAY;IAClB,MAAM,QAAQ,OAAO;IACrB,KAAKK,UAAU,OAAO,GAAG;IACzB,KAAKD,oBAAoB,IACvB,MACC,KAAKA,oBAAoB,IAAI,GAAG,KAAK,KAAK,CAC7C;IACA,MAAM,KAAKM,qBAAqB,GAAG;GACrC;GACA,iBAAiB,QAAQ,UAAU;GACnC,YAAY;IACV,QAAQ,KAAK;IACb,KAAKN,oBAAoB,IACvB,MACC,KAAKA,oBAAoB,IAAI,GAAG,KAAK,KAAK,CAC7C;IACA,KAAKM,qBAAqB,GAAG,EAAE,YAAY,KAAA,CAAS;IACpD,KAAKL,UAAU,OAAO,GAAG;GAC3B;GACA,OAAO,UAAU,QAAQ,KAAK,KAAK;GACnC,QAAQ,UAAU,QAAQ,MAAM,KAAK;EACvC;EACA,KAAKA,UAAU,IAAI,KAAK,MAAM;EAC9B,OAAO;CACT;CAEA,4BACE,KACA,sBACA,qBACA,mBACiC;EACjC,MAAM,eAAe,KAAKH;EAC1B,IAAI,CAAC,cACH,MAAM,IAAI,MAAM,+BAA+B;EAEjD,MAAM,QACJ,KAAKM,WAAW,WAAW,IACvB,KAAKT,aACL;GACE,GAAG,KAAKA;GACR,GAAG,oBAAoB;IACrB;IACA,eAAe;KAAE;KAAmB;IAAoB;IACxD,kBAAkB;IAClB,uBAAuB,YAAY,YACjC,KAAKI,sBAAsB,SAAS,YAAY,OAAO;IACzD,WAAW,KAAKK;GAClB,CAAC;EACH;EAEN,OAAO;GACL,cAAc,aAAa;GAC3B,OAAO,aAAa;GACpB,YAAY,aAAa;GACzB;EACF;CACF;CAEA,MAAME,qBAAqB,kBAAyC;EAClE,MAAM,KAAKP,sBAAsB,OAAO,gBAAgB;CAC1D;AACF;AAEA,SAAS,qBAAqB,SAA+B;CAC3D,MAAM,YAAY,QAAQ,UAAU,aAAa,QAAQ;CACzD,OAAO,YAAY,eAAe,SAAS,IAAI,qBAAqB;AACtE;AAEA,SAAS,mBAAmB,SAAmD;CAC7E,IAAI,YAAY,QAAQ,OAAO,YAAY,UACzC,MAAM,IAAI,UACR,kEACF;CAGF,MAAM,SAAS,aAAa,OAAO;CACnC,MAAM,WAAW,WAAW,WAAW,QAAQ,SAAS;CAExD,IAAI,UAAU,UACZ,MAAM,IAAI,UAAU,qDAAqD;CAG3E,IAAI,SAAS,WAAW,QAAQ,QAAQ,KAAA,KAAa,CAAC,QACpD,MAAM,IAAI,UAAU,6BAA6B;CAGnD,IAAI,EAAE,UAAU,WACd,MAAM,IAAI,UAAU,+BAA+B;AAEvD;AAEA,SAAS,aAAa,SAA6C;CACjE,OAAO,SAAS,WAAW,OAAO,QAAQ,QAAQ;AACpD"}
+{"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"}

package/dist/execution/host.js

@@ -0,0 +1,14 @@
+//#region src/execution/host.ts
+function executionHost(host) {
+	if (isExecutionHost(host)) return host;
+}
+function isExecutionHost(host) {
+	return "scheduler" in host && "store" in host && isExecutionStore(host.store);
+}
+function isExecutionStore(store) {
+	return typeof store === "object" && store !== null && "checkpoints" in store && "events" in store && "notifications" in store && "runs" in store && "sessions" in store;
+}
+//#endregion
+export { executionHost };
+
+//# sourceMappingURL=host.js.map

package/dist/execution/host.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"host.js","names":[],"sources":["../../src/execution/host.ts"],"sourcesContent":["import type { AgentHost, ExecutionHost } from \"./types\";\n\nexport function executionHost(host: AgentHost): ExecutionHost | undefined {\n  if (isExecutionHost(host)) {\n    return host;\n  }\n\n  return;\n}\n\nfunction isExecutionHost(host: AgentHost): host is ExecutionHost {\n  return \"scheduler\" in host && \"store\" in host && isExecutionStore(host.store);\n}\n\nfunction isExecutionStore(store: unknown): store is ExecutionHost[\"store\"] {\n  return (\n    typeof store === \"object\" &&\n    store !== null &&\n    \"checkpoints\" in store &&\n    \"events\" in store &&\n    \"notifications\" in store &&\n    \"runs\" in store &&\n    \"sessions\" in store\n  );\n}\n"],"mappings":";AAEA,SAAgB,cAAc,MAA4C;CACxE,IAAI,gBAAgB,IAAI,GACtB,OAAO;AAIX;AAEA,SAAS,gBAAgB,MAAwC;CAC/D,OAAO,eAAe,QAAQ,WAAW,QAAQ,iBAAiB,KAAK,KAAK;AAC9E;AAEA,SAAS,iBAAiB,OAAiD;CACzE,OACE,OAAO,UAAU,YACjB,UAAU,QACV,iBAAiB,SACjB,YAAY,SACZ,mBAAmB,SACnB,UAAU,SACV,cAAc;AAElB"}

package/dist/execution/index.d.ts

@@ -0,0 +1,4 @@
+import { RuntimeToolExecutionCheckpoint, RuntimeToolExecutionContext, RuntimeToolExecutionDecision, RuntimeToolRetryPolicy, ToolExecutionNeedsRecoveryError } from "../llm-tool-execution.js";
+import { AgentHostCapabilities, CheckpointPhase, CheckpointStore, CheckpointWriteResult, ClaimRunOptions, ClaimRunResult, EventCursor, EventStore, ExecutionHost, ExecutionScheduler, ExecutionStore, ExecutionStoreTransaction, NotificationClaimResult, NotificationInbox, NotificationRecord, NotificationStatus, NotificationWriteResult, ResumeSessionOptions, RunCheckpoint, RunKind, RunLease, RunRecord, RunStatus, RunStore, StoredAgentEvent } from "./types.js";
+import { createInMemoryExecutionHost } from "./memory.js";
+export { type AgentHostCapabilities, type CheckpointPhase, type CheckpointStore, type CheckpointWriteResult, type ClaimRunOptions, type ClaimRunResult, type EventCursor, type EventStore, type ExecutionHost, type ExecutionScheduler, type ExecutionStore, type ExecutionStoreTransaction, type NotificationClaimResult, type NotificationInbox, type NotificationRecord, type NotificationStatus, type NotificationWriteResult, type ResumeSessionOptions, type RunCheckpoint, type RunKind, type RunLease, type RunRecord, type RunStatus, type RunStore, type RuntimeToolExecutionCheckpoint, type RuntimeToolExecutionContext, type RuntimeToolExecutionDecision, type RuntimeToolRetryPolicy, type StoredAgentEvent, ToolExecutionNeedsRecoveryError, createInMemoryExecutionHost };

package/dist/execution/index.js

@@ -0,0 +1,3 @@
+import { createInMemoryExecutionHost } from "./memory.js";
+import { ToolExecutionNeedsRecoveryError } from "../llm-tool-execution.js";
+export { ToolExecutionNeedsRecoveryError, createInMemoryExecutionHost };

package/dist/execution/memory-notifications.js

@@ -0,0 +1,54 @@
+//#region src/execution/memory-notifications.ts
+var InMemoryNotificationInbox = class {
+	#state;
+	constructor(state) {
+		this.#state = state;
+	}
+	claimByIdempotencyKey(idempotencyKey) {
+		const record = this.#state().notificationsByKey.get(idempotencyKey);
+		if (!record) return Promise.resolve({
+			ok: false,
+			reason: "not-found"
+		});
+		if (record.status !== "pending") return Promise.resolve({
+			ok: false,
+			reason: "already-claimed",
+			record: structuredClone(record)
+		});
+		this.#state().notificationsByKey.set(idempotencyKey, {
+			...record,
+			status: "acked"
+		});
+		return Promise.resolve({
+			ok: true,
+			record: structuredClone(record)
+		});
+	}
+	enqueue(record) {
+		const existing = this.#state().notificationsByKey.get(record.idempotencyKey);
+		if (existing) return Promise.resolve({
+			existingNotificationId: existing.notificationId,
+			ok: false,
+			reason: "duplicate"
+		});
+		this.#state().notificationsByKey.set(record.idempotencyKey, structuredClone(record));
+		return Promise.resolve({ ok: true });
+	}
+	getByIdempotencyKey(idempotencyKey) {
+		const record = this.#state().notificationsByKey.get(idempotencyKey);
+		return Promise.resolve(record ? structuredClone(record) : null);
+	}
+	releaseByIdempotencyKey(idempotencyKey) {
+		const record = this.#state().notificationsByKey.get(idempotencyKey);
+		if (record?.status !== "acked") return Promise.resolve();
+		this.#state().notificationsByKey.set(idempotencyKey, {
+			...record,
+			status: "pending"
+		});
+		return Promise.resolve();
+	}
+};
+//#endregion
+export { InMemoryNotificationInbox };
+
+//# sourceMappingURL=memory-notifications.js.map

package/dist/execution/memory-notifications.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory-notifications.js","names":["#state"],"sources":["../../src/execution/memory-notifications.ts"],"sourcesContent":["import type { ExecutionState } from \"./memory-state\";\nimport type {\n  NotificationClaimResult,\n  NotificationInbox,\n  NotificationRecord,\n  NotificationWriteResult,\n} from \"./types\";\n\nexport class InMemoryNotificationInbox implements NotificationInbox {\n  readonly #state: () => ExecutionState;\n\n  constructor(state: () => ExecutionState) {\n    this.#state = state;\n  }\n\n  claimByIdempotencyKey(\n    idempotencyKey: string\n  ): Promise<NotificationClaimResult> {\n    const record = this.#state().notificationsByKey.get(idempotencyKey);\n    if (!record) {\n      return Promise.resolve({ ok: false, reason: \"not-found\" });\n    }\n\n    if (record.status !== \"pending\") {\n      return Promise.resolve({\n        ok: false,\n        reason: \"already-claimed\",\n        record: structuredClone(record),\n      });\n    }\n\n    this.#state().notificationsByKey.set(idempotencyKey, {\n      ...record,\n      status: \"acked\",\n    });\n    return Promise.resolve({\n      ok: true,\n      record: structuredClone(record),\n    });\n  }\n\n  enqueue(record: NotificationRecord): Promise<NotificationWriteResult> {\n    const existing = this.#state().notificationsByKey.get(\n      record.idempotencyKey\n    );\n    if (existing) {\n      return Promise.resolve({\n        existingNotificationId: existing.notificationId,\n        ok: false,\n        reason: \"duplicate\",\n      });\n    }\n\n    this.#state().notificationsByKey.set(\n      record.idempotencyKey,\n      structuredClone(record)\n    );\n    return Promise.resolve({ ok: true });\n  }\n\n  getByIdempotencyKey(\n    idempotencyKey: string\n  ): Promise<NotificationRecord | null> {\n    const record = this.#state().notificationsByKey.get(idempotencyKey);\n    return Promise.resolve(record ? structuredClone(record) : null);\n  }\n\n  releaseByIdempotencyKey(idempotencyKey: string): Promise<void> {\n    const record = this.#state().notificationsByKey.get(idempotencyKey);\n    if (record?.status !== \"acked\") {\n      return Promise.resolve();\n    }\n\n    this.#state().notificationsByKey.set(idempotencyKey, {\n      ...record,\n      status: \"pending\",\n    });\n    return Promise.resolve();\n  }\n}\n"],"mappings":";AAQA,IAAa,4BAAb,MAAoE;CAClE;CAEA,YAAY,OAA6B;EACvC,KAAKA,SAAS;CAChB;CAEA,sBACE,gBACkC;EAClC,MAAM,SAAS,KAAKA,OAAO,EAAE,mBAAmB,IAAI,cAAc;EAClE,IAAI,CAAC,QACH,OAAO,QAAQ,QAAQ;GAAE,IAAI;GAAO,QAAQ;EAAY,CAAC;EAG3D,IAAI,OAAO,WAAW,WACpB,OAAO,QAAQ,QAAQ;GACrB,IAAI;GACJ,QAAQ;GACR,QAAQ,gBAAgB,MAAM;EAChC,CAAC;EAGH,KAAKA,OAAO,EAAE,mBAAmB,IAAI,gBAAgB;GACnD,GAAG;GACH,QAAQ;EACV,CAAC;EACD,OAAO,QAAQ,QAAQ;GACrB,IAAI;GACJ,QAAQ,gBAAgB,MAAM;EAChC,CAAC;CACH;CAEA,QAAQ,QAA8D;EACpE,MAAM,WAAW,KAAKA,OAAO,EAAE,mBAAmB,IAChD,OAAO,cACT;EACA,IAAI,UACF,OAAO,QAAQ,QAAQ;GACrB,wBAAwB,SAAS;GACjC,IAAI;GACJ,QAAQ;EACV,CAAC;EAGH,KAAKA,OAAO,EAAE,mBAAmB,IAC/B,OAAO,gBACP,gBAAgB,MAAM,CACxB;EACA,OAAO,QAAQ,QAAQ,EAAE,IAAI,KAAK,CAAC;CACrC;CAEA,oBACE,gBACoC;EACpC,MAAM,SAAS,KAAKA,OAAO,EAAE,mBAAmB,IAAI,cAAc;EAClE,OAAO,QAAQ,QAAQ,SAAS,gBAAgB,MAAM,IAAI,IAAI;CAChE;CAEA,wBAAwB,gBAAuC;EAC7D,MAAM,SAAS,KAAKA,OAAO,EAAE,mBAAmB,IAAI,cAAc;EAClE,IAAI,QAAQ,WAAW,SACrB,OAAO,QAAQ,QAAQ;EAGzB,KAAKA,OAAO,EAAE,mBAAmB,IAAI,gBAAgB;GACnD,GAAG;GACH,QAAQ;EACV,CAAC;EACD,OAAO,QAAQ,QAAQ;CACzB;AACF"}