agents 0.11.2 → 0.11.4

package/dist/index.js

@@ -2,9 +2,9 @@ import { MessageType } from "./types.js";
 import { camelCaseToKebabCase } from "./utils.js";
 import { createHeaderBasedEmailResolver, signAgentHeaders } from "./email.js";
 import { __DO_NOT_USE_WILL_BREAK__agentContext } from "./internal_context.js";
-import { i as _classPrivateFieldInitSpec, n as _classPrivateFieldSet2, t as _classPrivateFieldGet2 } from "./classPrivateFieldGet2-BVdP0e3Z.js";
+import { i as _classPrivateFieldInitSpec, n as _classPrivateFieldSet2, t as _classPrivateFieldGet2 } from "./classPrivateFieldGet2-DAZNVUKb.js";
 import { isErrorRetryable, tryN, validateRetryOptions } from "./retries.js";
-import { o as RPC_DO_PREFIX, r as MCPConnectionState, s as DisposableStore, t as MCPClientManager } from "./client-PEDsNnfY.js";
+import { o as RPC_DO_PREFIX, r as MCPConnectionState, s as DisposableStore, t as MCPClientManager } from "./client-B_xdiZbn.js";
 import { DurableObjectOAuthClientProvider } from "./mcp/do-oauth-client-provider.js";
 import { genericObservability } from "./observability/index.js";
 import { AsyncLocalStorage } from "node:async_hooks";
@@ -19,13 +19,13 @@ let _Symbol$dispose;
 * Type guard for RPC request messages
 */
 function isRPCRequest(msg) {
-	return typeof msg === "object" && msg !== null && "type" in msg && msg.type === MessageType.RPC && "id" in msg && typeof msg.id === "string" && "method" in msg && typeof msg.method === "string" && "args" in msg && Array.isArray(msg.args);
+	return typeof msg === "object" && msg !== null && "type" in msg && msg.type === "rpc" && "id" in msg && typeof msg.id === "string" && "method" in msg && typeof msg.method === "string" && "args" in msg && Array.isArray(msg.args);
 }
 /**
 * Type guard for state update messages
 */
 function isStateUpdateMessage(msg) {
-	return typeof msg === "object" && msg !== null && "type" in msg && msg.type === MessageType.CF_AGENT_STATE && "state" in msg;
+	return typeof msg === "object" && msg !== null && "type" in msg && msg.type === "cf_agent_state" && "state" in msg;
 }
 const callableMetadata = /* @__PURE__ */ new WeakMap();
 /**
@@ -535,7 +535,7 @@ var Agent = class Agent extends Server {
 				if (isStateUpdateMessage(parsed)) {
 					if (this.isConnectionReadonly(connection)) {
 						connection.send(JSON.stringify({
-							type: MessageType.CF_AGENT_STATE_ERROR,
+							type: "cf_agent_state_error",
 							error: "Connection is readonly"
 						}));
 						return;
@@ -545,7 +545,7 @@ var Agent = class Agent extends Server {
 					} catch (e) {
 						console.error("[Agent] State update rejected:", e);
 						connection.send(JSON.stringify({
-							type: MessageType.CF_AGENT_STATE_ERROR,
+							type: "cf_agent_state_error",
 							error: "State update rejected"
 						}));
 					}
@@ -586,7 +586,7 @@ var Agent = class Agent extends Server {
 							id,
 							result,
 							success: true,
-							type: MessageType.RPC
+							type: "rpc"
 						};
 						connection.send(JSON.stringify(response));
 					} catch (e) {
@@ -594,7 +594,7 @@ var Agent = class Agent extends Server {
 							error: e instanceof Error ? e.message : "Unknown error occurred",
 							id: parsed.id,
 							success: false,
-							type: MessageType.RPC
+							type: "rpc"
 						};
 						connection.send(JSON.stringify(response));
 						console.error("RPC error:", e);
@@ -630,16 +630,16 @@ var Agent = class Agent extends Server {
 						connection.send(JSON.stringify({
 							name: this.name,
 							agent: camelCaseToKebabCase(this._ParentClass.name),
-							type: MessageType.CF_AGENT_IDENTITY
+							type: "cf_agent_identity"
 						}));
 					}
 					if (this.state) connection.send(JSON.stringify({
 						state: this.state,
-						type: MessageType.CF_AGENT_STATE
+						type: "cf_agent_state"
 					}));
 					connection.send(JSON.stringify({
 						mcp: this.getMcpServers(),
-						type: MessageType.CF_AGENT_MCP_SERVERS
+						type: "cf_agent_mcp_servers"
 					}));
 				} else this._setConnectionNoProtocol(connection);
 				this._emit("connect", { connectionId: connection.id });
@@ -718,10 +718,23 @@ var Agent = class Agent extends Server {
 	* @param excludeIds Additional connection IDs to exclude (e.g. the source)
 	*/
 	_broadcastProtocol(msg, excludeIds = []) {
+		if (this._isFacet) return;
 		const exclude = [...excludeIds];
 		for (const conn of this.getConnections()) if (!this.isConnectionProtocolEnabled(conn)) exclude.push(conn.id);
 		this.broadcast(msg, exclude);
 	}
+	/**
+	* When running as a facet, the parent DO owns the WebSocket registry
+	* (`ctx.getWebSockets()`). Iterating from the child isolate throws
+	* "Cannot perform I/O on behalf of a different Durable Object".
+	* Downstream callers (e.g. chat-streaming paths) invoke
+	* `this.broadcast()` directly, bypassing `_broadcastProtocol`'s
+	* guard, so override at the base to catch every path.
+	*/
+	broadcast(msg, without) {
+		if (this._isFacet) return;
+		super.broadcast(msg, without);
+	}
 	_setStateInternal(nextState, source = "server") {
 		this.validateStateChange(nextState, source);
 		this._state = nextState;
@@ -731,7 +744,7 @@ var Agent = class Agent extends Server {
     `;
 		this._broadcastProtocol(JSON.stringify({
 			state: nextState,
-			type: MessageType.CF_AGENT_STATE
+			type: "cf_agent_state"
 		}), source !== "server" ? [source.id] : []);
 		const { connection, request, email } = __DO_NOT_USE_WILL_BREAK__agentContext.getStore() || {};
 		this.ctx.waitUntil((async () => {
@@ -1801,7 +1814,7 @@ var Agent = class Agent extends Server {
 		if (this._runFiberRecoveryInProgress) return;
 		this._runFiberRecoveryInProgress = true;
 		try {
-			const rows = this.sql`SELECT id, name, snapshot FROM cf_agents_runs`;
+			const rows = this.sql`SELECT id, name, snapshot, created_at FROM cf_agents_runs`;
 			for (const row of rows) {
 				if (this._runFiberActiveFibers.has(row.id)) continue;
 				let snapshot = null;
@@ -1813,7 +1826,8 @@ var Agent = class Agent extends Server {
 				const ctx = {
 					id: row.id,
 					name: row.name,
-					snapshot
+					snapshot,
+					createdAt: row.created_at
 				};
 				try {
 					if (!await this._handleInternalFiberRecovery(ctx)) await this.onFiberRecovered(ctx);
@@ -1987,14 +2001,26 @@ var Agent = class Agent extends Server {
 		await this._scheduleNextAlarm();
 	}
 	/**
-	* Marks this agent as running inside a facet (sub-agent). Once set,
-	* scheduling methods throw a clear error instead of crashing on
-	* `setAlarm()` (which is not supported in facets).
-	* @internal
+	* Initialize this agent as a facet in a single RPC.
+	*
+	* Runs entirely inside the child's isolate, so every storage write
+	* and `onStart()` I/O is owned by the child DO. This replaces the
+	* previous "construct a Request in the parent DO and `stub.fetch()`
+	* it on the child" handshake, whose native I/O was tied to the
+	* parent and triggered "Cannot perform I/O on behalf of a different
+	* Durable Object" on the child.
+	*
+	* Order matters: set `_isFacet` BEFORE triggering initialization, so
+	* the first `onStart()` run (which calls `broadcastMcpServers`) sees
+	* the flag and skips broadcasts that would touch the parent DO's
+	* WebSocket registry.
+	*
+	* @internal Called by {@link subAgent}.
 	*/
-	async _cf_markAsFacet() {
+	async _cf_initAsFacet(name) {
 		this._isFacet = true;
-		await this.ctx.storage.put("cf_agents_is_facet", true);
+		await Promise.all([this.ctx.storage.put("cf_agents_is_facet", true), this.ctx.storage.put("__ps_name", name)]);
+		await this.__unsafe_ensureInitialized();
 	}
 	/**
 	* Get or create a named sub-agent — a child Durable Object (facet)
@@ -2004,7 +2030,7 @@ var Agent = class Agent extends Server {
 	* entry point. The first call for a given name triggers the child's
 	* `onStart()`. Subsequent calls return the existing instance.
 	*
-	* @experimental Requires the `"experimental"` compatibility flag.
+	* @experimental The API surface may change before stabilizing.
 	*
 	* @param cls The Agent subclass (must be exported from the worker)
 	* @param name Unique name for this child instance
@@ -2018,14 +2044,11 @@ var Agent = class Agent extends Server {
 	*/
 	async subAgent(cls, name) {
 		const ctx = this.ctx;
-		if (!ctx.facets || !ctx.exports) throw new Error("subAgent() requires the \"experimental\" compatibility flag. Add it to your wrangler.jsonc compatibility_flags.");
+		if (!ctx.facets || !ctx.exports) throw new Error("subAgent() is not supported in this runtime — `ctx.facets` / `ctx.exports` are unavailable. Update to the latest `compatibility_date` in your wrangler.jsonc.");
 		if (!ctx.exports[cls.name]) throw new Error(`Sub-agent class "${cls.name}" not found in worker exports. Make sure the class is exported from your worker entry point and that the export name matches the class name.`);
 		const facetKey = `${cls.name}\0${name}`;
 		const stub = ctx.facets.get(facetKey, () => ({ class: ctx.exports[cls.name] }));
-		const req = new Request("http://dummy-example.cloudflare.com/cdn-cgi/partyserver/set-name/");
-		req.headers.set("x-partykit-room", name);
-		await stub.fetch(req).then((res) => res.text());
-		await stub._cf_markAsFacet();
+		await stub._cf_initAsFacet(name);
 		return stub;
 	}
 	/**
@@ -2034,7 +2057,7 @@ var Agent = class Agent extends Server {
 	* Pending RPC calls receive the reason as an error.
 	* Transitively aborts the child's own children.
 	*
-	* @experimental Requires the `"experimental"` compatibility flag.
+	* @experimental The API surface may change before stabilizing.
 	*
 	* @param cls The Agent subclass used when creating the child
 	* @param name Name of the child to abort
@@ -2042,7 +2065,7 @@ var Agent = class Agent extends Server {
 	*/
 	abortSubAgent(cls, name, reason) {
 		const ctx = this.ctx;
-		if (!ctx.facets) throw new Error("abortSubAgent() requires the \"experimental\" compatibility flag.");
+		if (!ctx.facets) throw new Error("abortSubAgent() is not supported in this runtime — `ctx.facets` is unavailable. Update to the latest `compatibility_date` in your wrangler.jsonc.");
 		const facetKey = `${cls.name}\0${name}`;
 		ctx.facets.abort(facetKey, reason);
 	}
@@ -2050,14 +2073,14 @@ var Agent = class Agent extends Server {
 	* Delete a sub-agent: abort it if running, then permanently wipe its
 	* storage. Transitively deletes the child's own children.
 	*
-	* @experimental Requires the `"experimental"` compatibility flag.
+	* @experimental The API surface may change before stabilizing.
 	*
 	* @param cls The Agent subclass used when creating the child
 	* @param name Name of the child to delete
 	*/
 	deleteSubAgent(cls, name) {
 		const ctx = this.ctx;
-		if (!ctx.facets) throw new Error("deleteSubAgent() requires the \"experimental\" compatibility flag.");
+		if (!ctx.facets) throw new Error("deleteSubAgent() is not supported in this runtime — `ctx.facets` is unavailable. Update to the latest `compatibility_date` in your wrangler.jsonc.");
 		const facetKey = `${cls.name}\0${name}`;
 		ctx.facets.delete(facetKey);
 	}
@@ -3011,7 +3034,7 @@ var Agent = class Agent extends Server {
 	broadcastMcpServers() {
 		this._broadcastProtocol(JSON.stringify({
 			mcp: this.getMcpServers(),
-			type: MessageType.CF_AGENT_MCP_SERVERS
+			type: "cf_agent_mcp_servers"
 		}));
 	}
 	/**
@@ -3202,7 +3225,7 @@ var StreamingResponse = class {
 			id: this._id,
 			result: chunk,
 			success: true,
-			type: MessageType.RPC
+			type: "rpc"
 		};
 		this._connection.send(JSON.stringify(response));
 		return true;
@@ -3220,7 +3243,7 @@ var StreamingResponse = class {
 			id: this._id,
 			result: finalChunk,
 			success: true,
-			type: MessageType.RPC
+			type: "rpc"
 		};
 		this._connection.send(JSON.stringify(response));
 		return true;
@@ -3237,7 +3260,7 @@ var StreamingResponse = class {
 			error: message,
 			id: this._id,
 			success: false,
-			type: MessageType.RPC
+			type: "rpc"
 		};
 		this._connection.send(JSON.stringify(response));
 		return true;