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

package/dist/agent-namespace.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent-namespace.js","names":[],"sources":["../src/agent-namespace.ts"],"sourcesContent":["export function randomAgentNamespace(): string {\n  return agentNamespace(crypto.randomUUID());\n}\n\nexport function agentNamespace(namespace: string): string {\n  return `agent:${namespacePart(namespace)}`;\n}\n\nexport function namespacePart(value: string): string {\n  return encodeURIComponent(value);\n}\n\nexport function parentSessionNamespace({\n  generation,\n  sessionKey,\n  sessionNamespace,\n}: {\n  readonly generation: number;\n  readonly sessionKey: string;\n  readonly sessionNamespace: string;\n}): string {\n  return `${sessionNamespace}:session:${namespacePart(\n    sessionKey\n  )}:generation:${generation}`;\n}\n"],"mappings":";AAAA,SAAgB,uBAA+B;CAC7C,OAAO,eAAe,OAAO,WAAW,CAAC;AAC3C;AAEA,SAAgB,eAAe,WAA2B;CACxD,OAAO,SAAS,cAAc,SAAS;AACzC;AAEA,SAAgB,cAAc,OAAuB;CACnD,OAAO,mBAAmB,KAAK;AACjC;AAEA,SAAgB,uBAAuB,EACrC,YACA,YACA,oBAKS;CACT,OAAO,GAAG,iBAAiB,WAAW,cACpC,UACF,EAAE,cAAc;AAClB"}
+{"version":3,"file":"agent-namespace.js","names":[],"sources":["../src/agent-namespace.ts"],"sourcesContent":["export function randomAgentNamespace(): string {\n  return agentNamespace(crypto.randomUUID());\n}\n\nexport function agentNamespace(namespace: string): string {\n  return `agent:${namespacePart(namespace)}`;\n}\n\nexport function namespacePart(value: string): string {\n  return encodeURIComponent(value);\n}\n\nexport function parentSessionNamespace({\n  generation,\n  sessionKey,\n  sessionNamespace,\n}: {\n  readonly generation: number;\n  readonly sessionKey: string;\n  readonly sessionNamespace: string;\n}): string {\n  return `${sessionNamespace}:session:${namespacePart(\n    sessionKey\n  )}:generation:${generation}`;\n}\n\nexport function ownsAgentNamespace(\n  ownerNamespace: string | undefined,\n  sessionNamespace: string\n): boolean {\n  return (\n    ownerNamespace === sessionNamespace ||\n    ownerNamespace?.startsWith(`${sessionNamespace}:session:`) === true\n  );\n}\n\nexport function stableAgentNamespace({\n  namespace,\n}: {\n  readonly namespace?: string;\n}): string {\n  return namespace ? agentNamespace(namespace) : randomAgentNamespace();\n}\n"],"mappings":";AAAA,SAAgB,uBAA+B;CAC7C,OAAO,eAAe,OAAO,WAAW,CAAC;AAC3C;AAEA,SAAgB,eAAe,WAA2B;CACxD,OAAO,SAAS,cAAc,SAAS;AACzC;AAEA,SAAgB,cAAc,OAAuB;CACnD,OAAO,mBAAmB,KAAK;AACjC;AAgBA,SAAgB,mBACd,gBACA,kBACS;CACT,OACE,mBAAmB,oBACnB,gBAAgB,WAAW,GAAG,iBAAiB,UAAU,MAAM;AAEnE;AAEA,SAAgB,qBAAqB,EACnC,aAGS;CACT,OAAO,YAAY,eAAe,SAAS,IAAI,qBAAqB;AACtE"}

package/dist/agent-options.d.ts

@@ -0,0 +1,29 @@
+import { AgentToolChoice, RuntimeLlm } from "./llm.js";
+import { AgentHost } from "./execution/types.js";
+import { AgentPlugin } from "./plugins.js";
+import { LanguageModel, ToolSet } from "ai";
+
+//#region src/agent-options.d.ts
+interface AgentLanguageModelOptions {
+  readonly host?: AgentHost;
+  readonly instructions?: string;
+  readonly model: LanguageModel;
+  readonly namespace?: string;
+  readonly plugins?: readonly AgentPlugin[];
+  readonly toolChoice?: AgentToolChoice;
+  readonly tools?: ToolSet;
+}
+interface AgentRuntimeModelOptions {
+  readonly host?: AgentHost;
+  readonly instructions?: never;
+  readonly model: RuntimeLlm;
+  readonly namespace?: string;
+  readonly plugins?: readonly AgentPlugin[];
+  readonly toolChoice?: never;
+  readonly tools?: never;
+}
+type AgentOptions = AgentLanguageModelOptions | AgentRuntimeModelOptions;
+type AgentConstructionOptions = AgentOptions;
+//#endregion
+export { AgentConstructionOptions, AgentOptions };
+//# sourceMappingURL=agent-options.d.ts.map

package/dist/agent-options.js

@@ -0,0 +1,16 @@
+//#region src/agent-options.ts
+function assertAgentOptions(options) {
+	if (options === null || typeof options !== "object") throw new TypeError("Agent options are required. Provide { model }.");
+	if (!("model" in options && options.model != null)) throw new TypeError("Agent: missing options.model.");
+	if (typeof options.model !== "function" && (typeof options.model !== "object" || options.model === null)) throw new TypeError("Agent: invalid options.model.");
+}
+function hasRuntimeModel(options) {
+	return typeof options.model === "function";
+}
+function hasLanguageModel(options) {
+	return typeof options.model !== "function";
+}
+//#endregion
+export { assertAgentOptions, hasLanguageModel, hasRuntimeModel };
+
+//# sourceMappingURL=agent-options.js.map

package/dist/agent-options.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-options.js","names":[],"sources":["../src/agent-options.ts"],"sourcesContent":["import type { LanguageModel, ToolSet } from \"ai\";\nimport type { AgentHost } from \"./execution/types\";\nimport type { AgentToolChoice, RuntimeLlm } from \"./llm\";\nimport type { AgentPlugin } from \"./plugins\";\n\ninterface AgentLanguageModelOptions {\n  readonly host?: AgentHost;\n  readonly instructions?: string;\n  readonly model: LanguageModel;\n  readonly namespace?: string;\n  readonly plugins?: readonly AgentPlugin[];\n  readonly toolChoice?: AgentToolChoice;\n  readonly tools?: ToolSet;\n}\n\ninterface AgentRuntimeModelOptions {\n  readonly host?: AgentHost;\n  readonly instructions?: never;\n  readonly model: RuntimeLlm;\n  readonly namespace?: string;\n  readonly plugins?: readonly AgentPlugin[];\n  readonly toolChoice?: never;\n  readonly tools?: never;\n}\n\nexport type AgentModelOptions = Pick<\n  AgentLanguageModelOptions,\n  \"instructions\" | \"model\" | \"toolChoice\"\n>;\nexport type AgentOptions = AgentLanguageModelOptions | AgentRuntimeModelOptions;\n\nexport type AgentConstructionOptions = AgentOptions;\n\nexport function assertAgentOptions(\n  options: unknown\n): asserts options is AgentConstructionOptions {\n  if (options === null || typeof options !== \"object\") {\n    throw new TypeError(\"Agent options are required. Provide { model }.\");\n  }\n\n  const hasModel = \"model\" in options && options.model != null;\n\n  if (!hasModel) {\n    throw new TypeError(\"Agent: missing options.model.\");\n  }\n\n  if (\n    typeof options.model !== \"function\" &&\n    (typeof options.model !== \"object\" || options.model === null)\n  ) {\n    throw new TypeError(\"Agent: invalid options.model.\");\n  }\n}\n\nexport function hasRuntimeModel(\n  options: AgentConstructionOptions\n): options is AgentRuntimeModelOptions {\n  return typeof options.model === \"function\";\n}\n\nexport function hasLanguageModel(\n  options: AgentConstructionOptions\n): options is AgentLanguageModelOptions {\n  return typeof options.model !== \"function\";\n}\n"],"mappings":";AAiCA,SAAgB,mBACd,SAC6C;CAC7C,IAAI,YAAY,QAAQ,OAAO,YAAY,UACzC,MAAM,IAAI,UAAU,gDAAgD;CAKtE,IAAI,EAFa,WAAW,WAAW,QAAQ,SAAS,OAGtD,MAAM,IAAI,UAAU,+BAA+B;CAGrD,IACE,OAAO,QAAQ,UAAU,eACxB,OAAO,QAAQ,UAAU,YAAY,QAAQ,UAAU,OAExD,MAAM,IAAI,UAAU,+BAA+B;AAEvD;AAEA,SAAgB,gBACd,SACqC;CACrC,OAAO,OAAO,QAAQ,UAAU;AAClC;AAEA,SAAgB,iBACd,SACsC;CACtC,OAAO,OAAO,QAAQ,UAAU;AAClC"}

package/dist/agent-resume.js

@@ -0,0 +1,63 @@
+import { ownsAgentNamespace } from "./agent-namespace.js";
+//#region src/agent-resume.ts
+async function resumeAgentRun({ host, ownerNamespace, resumeNotification, runId }) {
+	const run = await host.store.runs.get(runId);
+	if (!run) return null;
+	if (!canAccessRun(run, ownerNamespace)) return null;
+	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 claimRun(host, run) {
+	const claim = await host.store.runs.claim(run.runId, {
+		attempt: (run.lease?.attempt ?? 0) + 1,
+		leaseId: crypto.randomUUID(),
+		leaseMs: 3e5,
+		nowMs: Date.now()
+	});
+	return claim.ok ? claim.record : null;
+}
+//#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 type {\n  ExecutionHost,\n  NotificationRecord,\n  RunRecord,\n} from \"./execution/types\";\nimport type { AgentRun } from \"./session/run\";\n\ninterface ResumeAgentRunInput {\n  readonly host: ExecutionHost;\n  readonly ownerNamespace: string;\n  resumeNotification(notification: NotificationRecord): Promise<AgentRun>;\n  readonly runId: string;\n}\n\nexport async function resumeAgentRun({\n  host,\n  ownerNamespace,\n  resumeNotification,\n  runId,\n}: ResumeAgentRunInput): Promise<AgentRun | null> {\n  const run = await host.store.runs.get(runId);\n  if (!run) {\n    return null;\n  }\n  if (!canAccessRun(run, ownerNamespace)) {\n    return null;\n  }\n\n  if (run.kind === \"notification\" && run.dedupeKey) {\n    const idempotencyKey = run.dedupeKey;\n    const claimed = await claimRun(host, run);\n    if (!claimed) {\n      return null;\n    }\n\n    const notification = await claimNotificationForRun({\n      host,\n      idempotencyKey,\n      ownerNamespace,\n    });\n    if (!notification) {\n      return null;\n    }\n\n    try {\n      const notificationRun = await resumeNotification(notification);\n      await completeNotificationRun(host, claimed.runId);\n      return notificationRun;\n    } catch (error) {\n      await host.store.notifications.releaseByIdempotencyKey(idempotencyKey);\n      throw error;\n    }\n  }\n\n  return null;\n}\n\nasync function claimNotificationForRun({\n  host,\n  idempotencyKey,\n  ownerNamespace,\n}: {\n  readonly host: ExecutionHost;\n  readonly idempotencyKey: string;\n  readonly ownerNamespace: string;\n}): Promise<NotificationRecord | null> {\n  const current =\n    await host.store.notifications.getByIdempotencyKey(idempotencyKey);\n  if (!ownsAgentNamespace(current?.ownerNamespace, ownerNamespace)) {\n    return null;\n  }\n\n  const claim =\n    await host.store.notifications.claimByIdempotencyKey(idempotencyKey);\n  if (claim.ok) {\n    if (ownsAgentNamespace(claim.record.ownerNamespace, ownerNamespace)) {\n      return claim.record;\n    }\n    await host.store.notifications.releaseByIdempotencyKey(idempotencyKey);\n    return null;\n  }\n\n  if (\n    claim.reason === \"already-claimed\" &&\n    ownsAgentNamespace(claim.record?.ownerNamespace, ownerNamespace)\n  ) {\n    return claim.record ?? null;\n  }\n\n  return null;\n}\n\nfunction canAccessRun(run: RunRecord, ownerNamespace: string): boolean {\n  if (run.ownerNamespace) {\n    return ownsAgentNamespace(run.ownerNamespace, ownerNamespace);\n  }\n\n  return (\n    run.sessionKey.startsWith(`parent:${ownerNamespace}:`) ||\n    run.parentRunId?.startsWith(`${ownerNamespace}:session:`) === true\n  );\n}\n\nexport async function completeNotificationRun(\n  host: ExecutionHost,\n  runId: string\n): Promise<void> {\n  const run = await host.store.runs.get(runId);\n  if (run?.kind !== \"notification\" || run.status === \"completed\") {\n    return;\n  }\n\n  await host.store.runs.update({ ...run, status: \"completed\" });\n}\n\nasync function claimRun(\n  host: ExecutionHost,\n  run: RunRecord\n): Promise<RunRecord | null> {\n  const claim = await host.store.runs.claim(run.runId, {\n    attempt: (run.lease?.attempt ?? 0) + 1,\n    leaseId: crypto.randomUUID(),\n    leaseMs: 300_000,\n    nowMs: Date.now(),\n  });\n  return claim.ok ? claim.record : null;\n}\n"],"mappings":";;AAeA,eAAsB,eAAe,EACnC,MACA,gBACA,oBACA,SACgD;CAChD,MAAM,MAAM,MAAM,KAAK,MAAM,KAAK,IAAI,KAAK;CAC3C,IAAI,CAAC,KACH,OAAO;CAET,IAAI,CAAC,aAAa,KAAK,cAAc,GACnC,OAAO;CAGT,IAAI,IAAI,SAAS,kBAAkB,IAAI,WAAW;EAChD,MAAM,iBAAiB,IAAI;EAC3B,MAAM,UAAU,MAAM,SAAS,MAAM,GAAG;EACxC,IAAI,CAAC,SACH,OAAO;EAGT,MAAM,eAAe,MAAM,wBAAwB;GACjD;GACA;GACA;EACF,CAAC;EACD,IAAI,CAAC,cACH,OAAO;EAGT,IAAI;GACF,MAAM,kBAAkB,MAAM,mBAAmB,YAAY;GAC7D,MAAM,wBAAwB,MAAM,QAAQ,KAAK;GACjD,OAAO;EACT,SAAS,OAAO;GACd,MAAM,KAAK,MAAM,cAAc,wBAAwB,cAAc;GACrE,MAAM;EACR;CACF;CAEA,OAAO;AACT;AAEA,eAAe,wBAAwB,EACrC,MACA,gBACA,kBAKqC;CAGrC,IAAI,CAAC,oBAAmB,MADhB,KAAK,MAAM,cAAc,oBAAoB,cAAc,IAClC,gBAAgB,cAAc,GAC7D,OAAO;CAGT,MAAM,QACJ,MAAM,KAAK,MAAM,cAAc,sBAAsB,cAAc;CACrE,IAAI,MAAM,IAAI;EACZ,IAAI,mBAAmB,MAAM,OAAO,gBAAgB,cAAc,GAChE,OAAO,MAAM;EAEf,MAAM,KAAK,MAAM,cAAc,wBAAwB,cAAc;EACrE,OAAO;CACT;CAEA,IACE,MAAM,WAAW,qBACjB,mBAAmB,MAAM,QAAQ,gBAAgB,cAAc,GAE/D,OAAO,MAAM,UAAU;CAGzB,OAAO;AACT;AAEA,SAAS,aAAa,KAAgB,gBAAiC;CACrE,IAAI,IAAI,gBACN,OAAO,mBAAmB,IAAI,gBAAgB,cAAc;CAG9D,OACE,IAAI,WAAW,WAAW,UAAU,eAAe,EAAE,KACrD,IAAI,aAAa,WAAW,GAAG,eAAe,UAAU,MAAM;AAElE;AAEA,eAAsB,wBACpB,MACA,OACe;CACf,MAAM,MAAM,MAAM,KAAK,MAAM,KAAK,IAAI,KAAK;CAC3C,IAAI,KAAK,SAAS,kBAAkB,IAAI,WAAW,aACjD;CAGF,MAAM,KAAK,MAAM,KAAK,OAAO;EAAE,GAAG;EAAK,QAAQ;CAAY,CAAC;AAC9D;AAEA,eAAe,SACb,MACA,KAC2B;CAC3B,MAAM,QAAQ,MAAM,KAAK,MAAM,KAAK,MAAM,IAAI,OAAO;EACnD,UAAU,IAAI,OAAO,WAAW,KAAK;EACrC,SAAS,OAAO,WAAW;EAC3B,SAAS;EACT,OAAO,KAAK,IAAI;CAClB,CAAC;CACD,OAAO,MAAM,KAAK,MAAM,SAAS;AACnC"}

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

@@ -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>;
+  dispose(): Promise<void>;
+  interrupt(): void;
+  send(input: AgentInput): Promise<AgentRun>;
+  steer(input: AgentInput): Promise<AgentRun>;
+}
+//#endregion
+export { SessionHandle };
+//# sourceMappingURL=agent-session-entry.d.ts.map

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 { AgentConstructionOptions, 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);
+  readonly host: AgentHost;
+  readonly namespace?: string;
+  constructor(options: AgentConstructionOptions);
   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,35 +1,33 @@
-import { agentNamespace, parentSessionNamespace, randomAgentNamespace } from "./agent-namespace.js";
-import { assertSubagents } from "./agent-validation.js";
-import { ChildSessionCleanups } from "./child-session-cleanups.js";
+import { executionHost } from "./execution/host.js";
+import { sessionStoreForHost } from "./agent-host-session-store.js";
+import { stableAgentNamespace } from "./agent-namespace.js";
+import { assertAgentOptions, hasLanguageModel, hasRuntimeModel } from "./agent-options.js";
+import { resumeAgentRun } from "./agent-resume.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 {
+var Agent = class {
 	#baseTools;
-	#hooks;
 	#llm;
 	#modelOptions;
-	#childSessionCleanups = new ChildSessionCleanups();
-	#sessionGenerations = /* @__PURE__ */ new Map();
 	#sessions = /* @__PURE__ */ new Map();
 	#sessionNamespace;
 	#store;
-	#subagents;
-	description;
-	name;
+	#host;
+	#plugins;
+	host;
+	namespace;
 	constructor(options) {
 		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;
-		else {
+		this.namespace = options.namespace;
+		this.#sessionNamespace = stableAgentNamespace({ namespace: options.namespace });
+		this.#host = options.host ?? createInMemoryExecutionHost();
+		this.host = this.#host;
+		this.#store = sessionStoreForHost(this.#host);
+		this.#plugins = options.plugins ?? [];
+		if (hasRuntimeModel(options)) this.#llm = options.model;
+		else if (hasLanguageModel(options)) {
 			this.#baseTools = options.tools;
 			this.#modelOptions = {
 				instructions: options.instructions,
@@ -41,85 +39,65 @@ 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
+		});
+	}
 	session(key) {
+		return this.#sessionEntry(key).publicHandle;
+	}
+	#sessionEntry(key) {
 		const existing = this.#sessions.get(key);
 		if (existing) return existing;
 		let session;
-		const getSession = () => {
-			if (!session) throw new Error("Agent session is not initialized.");
-			return session;
-		};
-		const parentAgentNamespace = parentSessionNamespace({
-			generation: this.#sessionGenerations.get(key) ?? 0,
-			sessionKey: key,
-			sessionNamespace: this.#sessionNamespace
-		});
-		session = new AgentSession(this.#llm ?? createLlm(this.#createLlmOptionsForSession(key, parentAgentNamespace, (input, placement) => getSession().enqueueRuntimeInput(input, placement), (event) => getSession().emitObserverEvent(event))), {
+		session = new AgentSession(this.#llm ?? createLlm(this.#createLlmOptionsForSession()), {
 			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();
+					this.#evictSessionHandle(key);
+					await session.delete();
+				},
+				dispose: () => {
+					session.kill();
+					this.#evictSessionHandle(key);
+					return Promise.resolve();
+				},
+				interrupt: () => session.interrupt(),
+				send: (input) => session.send(input),
+				steer: (input) => session.steer(input)
+			}
 		};
-		this.#sessions.set(key, handle);
-		return handle;
+		this.#sessions.set(key, entry);
+		return entry;
+	}
+	#evictSessionHandle(key) {
+		this.#sessions.delete(key);
+	}
+	#resumeNotification(notification) {
+		return this.#sessionEntry(notification.sessionKey).notify(notification.input, { observerEvents: notification.observerEvents });
 	}
-	#createLlmOptionsForSession(key, parentAgentNamespace, enqueueRuntimeInput, emitObserverEvent) {
+	#createLlmOptionsForSession() {
 		const modelOptions = this.#modelOptions;
 		if (!modelOptions) throw new Error("Agent: missing model options.");
-		const tools = this.#subagents.length === 0 ? this.#baseTools : {
-			...this.#baseTools,
-			...createSubagentTools({
-				parentAgentNamespace,
-				parentSession: {
-					emitObserverEvent,
-					enqueueRuntimeInput
-				},
-				parentSessionKey: key,
-				registerChildSession: (sessionKey, cleanup) => this.#childSessionCleanups.register(sessionKey, cleanup),
-				subagents: this.#subagents
-			})
-		};
 		return {
 			instructions: modelOptions.instructions,
 			model: modelOptions.model,
 			toolChoice: modelOptions.toolChoice,
-			tools
+			tools: this.#baseTools
 		};
 	}
-	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","#sessions","#sessionNamespace","#store","#host","#plugins","#resumeNotification","#sessionEntry","#createLlmOptionsForSession","#evictSessionHandle"],"sources":["../src/agent.ts"],"sourcesContent":["import type { ToolSet } from \"ai\";\nimport { sessionStoreForHost } from \"./agent-host-session-store\";\nimport { stableAgentNamespace } from \"./agent-namespace\";\nimport {\n  type AgentConstructionOptions,\n  type AgentModelOptions,\n  assertAgentOptions,\n  hasLanguageModel,\n  hasRuntimeModel,\n} from \"./agent-options\";\nimport { resumeAgentRun } from \"./agent-resume\";\nimport type { AgentSessionEntry, SessionHandle } from \"./agent-session-entry\";\nimport { executionHost } from \"./execution/host\";\nimport { createInMemoryExecutionHost } from \"./execution/memory\";\nimport type { AgentHost, NotificationRecord } from \"./execution/types\";\nimport { createLlm, type RuntimeLlm } from \"./llm\";\nimport type { AgentPlugin } from \"./plugins\";\nimport type { AgentRun } from \"./session/run\";\nimport { type AgentInput, AgentSession } from \"./session/session\";\nimport type { SessionStore } from \"./session/store/types\";\n\nexport type { AgentOptions } from \"./agent-options\";\nexport type { SessionHandle } from \"./agent-session-entry\";\nexport type { AgentHost } from \"./execution/types\";\n\nexport class Agent {\n  readonly #baseTools?: ToolSet;\n  readonly #llm?: RuntimeLlm;\n  readonly #modelOptions?: AgentModelOptions;\n  readonly #sessions = new Map<string, AgentSessionEntry>();\n  readonly #sessionNamespace: string;\n  readonly #store: SessionStore;\n  readonly #host: AgentHost;\n  readonly #plugins: readonly AgentPlugin[];\n  readonly host: AgentHost;\n  readonly namespace?: string;\n  constructor(options: AgentConstructionOptions) {\n    assertAgentOptions(options);\n\n    this.namespace = options.namespace;\n    this.#sessionNamespace = stableAgentNamespace({\n      namespace: options.namespace,\n    });\n    this.#host = options.host ?? createInMemoryExecutionHost();\n    this.host = this.#host;\n    this.#store = sessionStoreForHost(this.#host);\n    this.#plugins = options.plugins ?? [];\n    if (hasRuntimeModel(options)) {\n      this.#llm = options.model;\n    } else if (hasLanguageModel(options)) {\n      this.#baseTools = options.tools;\n      this.#modelOptions = {\n        instructions: options.instructions,\n        model: options.model,\n        toolChoice: options.toolChoice,\n      };\n    }\n  }\n\n  send(input: AgentInput): Promise<AgentRun> {\n    return this.session(\"default\").send(input);\n  }\n\n  async resume(runId: string): Promise<AgentRun | null> {\n    const host = executionHost(this.#host);\n    if (!host) {\n      throw new Error(\"Agent host does not support durable run resume.\");\n    }\n\n    return await resumeAgentRun({\n      host,\n      ownerNamespace: this.#sessionNamespace,\n      resumeNotification: (notification) =>\n        this.#resumeNotification(notification),\n      runId,\n    });\n  }\n\n  session(key: string): SessionHandle {\n    return this.#sessionEntry(key).publicHandle;\n  }\n\n  #sessionEntry(key: string): AgentSessionEntry {\n    const existing = this.#sessions.get(key);\n    if (existing) {\n      return existing;\n    }\n\n    let session: AgentSession | undefined;\n    const llm = this.#llm ?? createLlm(this.#createLlmOptionsForSession());\n    session = new AgentSession(\n      llm,\n      { key, store: this.#store },\n      this.#plugins,\n      {\n        executionHost: executionHost(this.#host),\n      }\n    );\n    const publicHandle: SessionHandle = {\n      delete: async () => {\n        session.kill();\n        this.#evictSessionHandle(key);\n        await session.delete();\n      },\n      dispose: () => {\n        session.kill();\n        this.#evictSessionHandle(key);\n        return Promise.resolve();\n      },\n      interrupt: () => session.interrupt(),\n      send: (input) => session.send(input),\n      steer: (input) => session.steer(input),\n    };\n    const entry: AgentSessionEntry = {\n      notify: (input, options) => session.notify(input, options),\n      publicHandle,\n    };\n    this.#sessions.set(key, entry);\n    return entry;\n  }\n\n  #evictSessionHandle(key: string): void {\n    this.#sessions.delete(key);\n  }\n\n  #resumeNotification(notification: NotificationRecord): Promise<AgentRun> {\n    return this.#sessionEntry(notification.sessionKey).notify(\n      notification.input,\n      { observerEvents: notification.observerEvents }\n    );\n  }\n\n  #createLlmOptionsForSession(): Parameters<typeof createLlm>[0] {\n    const modelOptions = this.#modelOptions;\n    if (!modelOptions) {\n      throw new Error(\"Agent: missing model options.\");\n    }\n\n    return {\n      instructions: modelOptions.instructions,\n      model: modelOptions.model,\n      toolChoice: modelOptions.toolChoice,\n      tools: this.#baseTools,\n    };\n  }\n}\n"],"mappings":";;;;;;;;;AAyBA,IAAa,QAAb,MAAmB;CACjB;CACA;CACA;CACA,4BAAqB,IAAI,IAA+B;CACxD;CACA;CACA;CACA;CACA;CACA;CACA,YAAY,SAAmC;EAC7C,mBAAmB,OAAO;EAE1B,KAAK,YAAY,QAAQ;EACzB,KAAKI,oBAAoB,qBAAqB,EAC5C,WAAW,QAAQ,UACrB,CAAC;EACD,KAAKE,QAAQ,QAAQ,QAAQ,4BAA4B;EACzD,KAAK,OAAO,KAAKA;EACjB,KAAKD,SAAS,oBAAoB,KAAKC,KAAK;EAC5C,KAAKC,WAAW,QAAQ,WAAW,CAAC;EACpC,IAAI,gBAAgB,OAAO,GACzB,KAAKN,OAAO,QAAQ;OACf,IAAI,iBAAiB,OAAO,GAAG;GACpC,KAAKD,aAAa,QAAQ;GAC1B,KAAKE,gBAAgB;IACnB,cAAc,QAAQ;IACtB,OAAO,QAAQ;IACf,YAAY,QAAQ;GACtB;EACF;CACF;CAEA,KAAK,OAAsC;EACzC,OAAO,KAAK,QAAQ,SAAS,EAAE,KAAK,KAAK;CAC3C;CAEA,MAAM,OAAO,OAAyC;EACpD,MAAM,OAAO,cAAc,KAAKI,KAAK;EACrC,IAAI,CAAC,MACH,MAAM,IAAI,MAAM,iDAAiD;EAGnE,OAAO,MAAM,eAAe;GAC1B;GACA,gBAAgB,KAAKF;GACrB,qBAAqB,iBACnB,KAAKI,oBAAoB,YAAY;GACvC;EACF,CAAC;CACH;CAEA,QAAQ,KAA4B;EAClC,OAAO,KAAKC,cAAc,GAAG,EAAE;CACjC;CAEA,cAAc,KAAgC;EAC5C,MAAM,WAAW,KAAKN,UAAU,IAAI,GAAG;EACvC,IAAI,UACF,OAAO;EAGT,IAAI;EAEJ,UAAU,IAAI,aADF,KAAKF,QAAQ,UAAU,KAAKS,4BAA4B,CAAC,GAGnE;GAAE;GAAK,OAAO,KAAKL;EAAO,GAC1B,KAAKE,UACL,EACE,eAAe,cAAc,KAAKD,KAAK,EACzC,CACF;EAgBA,MAAM,QAA2B;GAC/B,SAAS,OAAO,YAAY,QAAQ,OAAO,OAAO,OAAO;GACzD,cAAA;IAhBA,QAAQ,YAAY;KAClB,QAAQ,KAAK;KACb,KAAKK,oBAAoB,GAAG;KAC5B,MAAM,QAAQ,OAAO;IACvB;IACA,eAAe;KACb,QAAQ,KAAK;KACb,KAAKA,oBAAoB,GAAG;KAC5B,OAAO,QAAQ,QAAQ;IACzB;IACA,iBAAiB,QAAQ,UAAU;IACnC,OAAO,UAAU,QAAQ,KAAK,KAAK;IACnC,QAAQ,UAAU,QAAQ,MAAM,KAAK;GAI1B;EACb;EACA,KAAKR,UAAU,IAAI,KAAK,KAAK;EAC7B,OAAO;CACT;CAEA,oBAAoB,KAAmB;EACrC,KAAKA,UAAU,OAAO,GAAG;CAC3B;CAEA,oBAAoB,cAAqD;EACvE,OAAO,KAAKM,cAAc,aAAa,UAAU,EAAE,OACjD,aAAa,OACb,EAAE,gBAAgB,aAAa,eAAe,CAChD;CACF;CAEA,8BAA+D;EAC7D,MAAM,eAAe,KAAKP;EAC1B,IAAI,CAAC,cACH,MAAM,IAAI,MAAM,+BAA+B;EAGjD,OAAO;GACL,cAAc,aAAa;GAC3B,OAAO,aAAa;GACpB,YAAY,aAAa;GACzB,OAAO,KAAKF;EACd;CACF;AACF"}

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

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

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

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

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

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

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

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

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

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

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

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