agents 0.20.0 → 0.21.0

package/dist/index.d.ts

@@ -4,42 +4,48 @@ import {
   $ as RoutingRetryOptions,
   A as AgentGetOptions,
   B as EmailSendBinding,
+  Bn as parseSubAgentPath,
   C as DetachedRunAgentToolResult,
   D as AddRpcMcpServerOptions,
   E as AddMcpServerOptions,
   F as Connection,
-  Ft as normalizeServerId,
+  Fn as buildAgentPath,
   G as FiberStatus,
   H as FiberInspection,
+  Hn as withInvocationScope,
   I as ConnectionContext,
+  In as buildAgentUrl,
   J as MCPServerMessage,
   K as ListFibersOptions,
   L as DEFAULT_AGENT_STATIC_OPTIONS,
   M as AgentOptions,
-  Mt as MCP_SERVER_ID_MAX_LENGTH,
+  Mn as BuildAgentPathOptions,
   N as AgentStaticOptions,
+  Nn as SUB_PREFIX,
   O as Agent,
   P as CallableMetadata,
+  Pn as SubAgentPathMatch,
   Q as RPCResponse,
   R as DeleteFibersOptions,
   S as DetachedAgentToolConfig,
   T as RunAgentToolResult,
   U as FiberRecoveryContext,
+  Ut as MCPAITool,
   V as FiberContext,
+  Vn as routeSubAgentRequest,
   W as FiberRecoveryResult,
+  Wt as MCPAIToolSet,
   X as QueueItem,
   Y as MCPServersState,
   Z as RPCRequest,
   _ as AgentToolRunState,
   a as AgentToolEvent,
+  an as normalizeServerId,
   at as StartFiberResult,
   b as AgentToolTerminalStatus,
-  bt as MCPAIToolSet,
   c as AgentToolFailure,
-  cn as SubAgentPathMatch,
   ct as SubAgentClass,
   d as AgentToolMilestone,
-  dn as routeSubAgentRequest,
   dt as callable,
   et as Schedule,
   f as AgentToolProgress,
@@ -51,13 +57,14 @@ import {
   i as AgentToolDisplayMetadata,
   it as StartFiberOptions,
   j as AgentNamespace,
+  jn as AgentPathStep,
   k as AgentContext,
   l as AgentToolInterruptedReason,
-  ln as getSubAgentByName,
   lt as SubAgentStub,
   m as AgentToolRunInfo,
   mt as routeAgentEmail,
   n as AGENT_TOOL_PROGRESS_PART,
+  nn as MCP_SERVER_ID_MAX_LENGTH,
   nt as SendEmailOptions,
   o as AgentToolEventMessage,
   ot as StateUpdateMessage,
@@ -67,20 +74,18 @@ import {
   r as AgentToolChildAdapter,
   rt as SqlError,
   s as AgentToolEventState,
-  sn as SUB_PREFIX,
   st as StreamingResponse,
   t as AGENT_TOOL_MILESTONE_PART,
   tt as ScheduleCriteria,
   u as AgentToolLifecycleResult,
-  un as parseSubAgentPath,
   ut as WSMessage,
   v as AgentToolRunStatus,
   w as RunAgentToolOptions,
   x as ChatCapableAgentClass,
   y as AgentToolStoredChunk,
-  yt as MCPAITool,
-  z as EmailRoutingOptions
-} from "./agent-tool-types-Btk9ETS-.js";
+  z as EmailRoutingOptions,
+  zn as getSubAgentByName
+} from "./agent-tool-types-CzGGB-20.js";
 import { n as camelCaseToKebabCase } from "./utils-CGtGDSgA.js";
 import {
   i as isDurableObjectStorageReset,
@@ -108,6 +113,7 @@ export {
   type AgentMcpOAuthProvider,
   AgentNamespace,
   AgentOptions,
+  type AgentPathStep,
   AgentStaticOptions,
   type AgentToolChildAdapter,
   type AgentToolDisplayMetadata,
@@ -128,6 +134,7 @@ export {
   type AgentToolStoredChunk,
   type AgentToolTerminalStatus,
   type AgentsOAuthProvider,
+  type BuildAgentPathOptions,
   CallableMetadata,
   type ChatCapableAgentClass,
   type Connection,
@@ -174,6 +181,9 @@ export {
   type TransportType,
   type WSMessage,
   __DO_NOT_USE_WILL_BREAK__agentContext,
+  withInvocationScope as __DO_NOT_USE_WILL_BREAK__withInvocationScope,
+  buildAgentPath,
+  buildAgentUrl,
   callable,
   camelCaseToKebabCase,
   createHeaderBasedEmailResolver,

package/dist/index.js

@@ -4,12 +4,12 @@ import { MessageType } from "./types.js";
 import { camelCaseToKebabCase, isInternalJsStubProp } from "./utils.js";
 import { createHeaderBasedEmailResolver, signAgentHeaders } from "./email.js";
 import { i as _classPrivateFieldInitSpec, n as _classPrivateFieldSet2, t as _classPrivateFieldGet2 } from "./classPrivateFieldGet2-DZBYAB34.js";
-import { SUB_PREFIX, getSubAgentByName, parseSubAgentPath, routeSubAgentRequest } from "./sub-routing.js";
+import { n as withInvocationScope, r as writeSpanAttributes, t as tracer } from "./cloudflare-BduZwmYK.js";
+import { SUB_PREFIX, buildAgentPath, buildAgentUrl, getSubAgentByName, parseSubAgentPath, routeSubAgentRequest } from "./sub-routing.js";
 import { isDurableObjectCodeUpdateReset, isDurableObjectMemoryLimitReset, isDurableObjectStorageReset, isErrorRetryable, isPlatformTransientError, tryN, validateRetryOptions } from "./retries.js";
 import { a as MCPConnectionState, c as RPC_DO_PREFIX, i as normalizeServerId, l as DisposableStore, n as MCP_SERVER_ID_MAX_LENGTH, t as MCPClientManager } from "./client-zqKcsyFa.js";
 import { DurableObjectOAuthClientProvider } from "./mcp/do-oauth-client-provider.js";
 import { genericObservability } from "./observability/index.js";
-import { n as writeSpanAttributes, t as tracer } from "./cloudflare-BldFV0Pa.js";
 import { AsyncLocalStorage } from "node:async_hooks";
 import { parseCronExpression } from "cron-schedule";
 import { nanoid } from "nanoid";
@@ -18,6 +18,15 @@ import { RpcTarget, exports } from "cloudflare:workers";
 import { Server, getServerByName, routePartykitRequest } from "partyserver";
 //#region src/index.ts
 let _Symbol$dispose;
+/**
+* Enters an agent invocation: the context every handler reads, plus the span
+* scope that stops invocation-bounded spans from outliving it. Scopes do not
+* nest, so the outermost live entry point owns the boundary — pass
+* `detached` for work that deliberately runs on past its caller.
+*/
+function runInInvocation(store, body, options) {
+	return __DO_NOT_USE_WILL_BREAK__agentContext.run(store, () => withInvocationScope(body, options));
+}
 function isClosedWebSocketSendError(error) {
 	return error instanceof TypeError && error.message.includes("WebSocket send() after close");
 }
@@ -380,7 +389,7 @@ function withAgentContext(method) {
 	return function(...args) {
 		const { agent } = getCurrentAgent();
 		if (agent === this) return method.apply(this, args);
-		return __DO_NOT_USE_WILL_BREAK__agentContext.run({
+		return runInInvocation({
 			agent: this,
 			connection: void 0,
 			request: void 0,
@@ -497,7 +506,7 @@ var Agent = class Agent extends Server {
 			"cloudflare.agents.storage.system": "durable_object",
 			"cloudflare.agents.storage.phase": storagePhase,
 			...attributes
-		}, (span) => run((finishAttributes) => writeSpanAttributes(span, finishAttributes)));
+		}, (span) => run((finishAttributes) => writeSpanAttributes(span, finishAttributes)), __DO_NOT_USE_WILL_BREAK__agentContext.getStore()?.connection === void 0 ? void 0 : { boundToInvocation: true });
 	}
 	/**
 	* Execute SQL queries against the Agent's database
@@ -824,7 +833,7 @@ var Agent = class Agent extends Server {
 		}
 		const _onRequest = this.onRequest.bind(this);
 		this.onRequest = (request) => {
-			return __DO_NOT_USE_WILL_BREAK__agentContext.run({
+			return runInInvocation({
 				agent: this,
 				connection: void 0,
 				request,
@@ -839,7 +848,7 @@ var Agent = class Agent extends Server {
 		this.onMessage = async (connection, message) => {
 			if (await this._cf_forwardSubAgentWebSocketMessage(connection, message)) return;
 			this._ensureConnectionWrapped(connection);
-			return __DO_NOT_USE_WILL_BREAK__agentContext.run({
+			return runInInvocation({
 				agent: this,
 				connection,
 				request: void 0,
@@ -932,7 +941,7 @@ var Agent = class Agent extends Server {
 			const subAgentOuterUrl = ctx.request.headers.get(SUB_AGENT_OUTER_URL_HEADER);
 			if (subAgentOuterUrl) this._unsafe_setConnectionFlag(connection, CF_SUB_AGENT_OUTER_URL_KEY, subAgentOuterUrl);
 			if (await this._cf_forwardSubAgentWebSocketConnect(connection, ctx.request, { gate: false })) return;
-			return __DO_NOT_USE_WILL_BREAK__agentContext.run({
+			return runInInvocation({
 				agent: this,
 				connection,
 				request: ctx.request,
@@ -979,7 +988,7 @@ var Agent = class Agent extends Server {
 		const _onClose = this.onClose.bind(this);
 		this.onClose = async (connection, code, reason, wasClean) => {
 			if (await this._cf_forwardSubAgentWebSocketClose(connection, code, reason, wasClean)) return;
-			return __DO_NOT_USE_WILL_BREAK__agentContext.run({
+			return runInInvocation({
 				agent: this,
 				connection,
 				request: void 0,
@@ -995,7 +1004,7 @@ var Agent = class Agent extends Server {
 		};
 		const _onStart = this.onStart.bind(this);
 		const startAgent = async (props, update) => {
-			return __DO_NOT_USE_WILL_BREAK__agentContext.run({
+			return runInInvocation({
 				agent: this,
 				connection: void 0,
 				request: void 0,
@@ -1098,7 +1107,7 @@ var Agent = class Agent extends Server {
 		const { connection, request, email } = __DO_NOT_USE_WILL_BREAK__agentContext.getStore() || {};
 		this.ctx.waitUntil((async () => {
 			try {
-				await __DO_NOT_USE_WILL_BREAK__agentContext.run({
+				await runInInvocation({
 					agent: this,
 					connection,
 					request,
@@ -1106,7 +1115,7 @@ var Agent = class Agent extends Server {
 				}, async () => {
 					this._emit("state:update");
 					await this._callStatePersistenceHook(nextState, source);
-				});
+				}, { detached: true });
 			} catch (e) {
 				try {
 					await this.onError(e);
@@ -1371,7 +1380,7 @@ var Agent = class Agent extends Server {
 			forward: (rcptTo, headers) => payload._bridge.forward(rcptTo, headers),
 			reply: (options) => payload._bridge.reply(options)
 		};
-		return __DO_NOT_USE_WILL_BREAK__agentContext.run({
+		return runInInvocation({
 			agent: this,
 			connection: void 0,
 			request: void 0,
@@ -1620,7 +1629,7 @@ var Agent = class Agent extends Server {
 						continue;
 					}
 					const { connection, request, email } = __DO_NOT_USE_WILL_BREAK__agentContext.getStore() || {};
-					await __DO_NOT_USE_WILL_BREAK__agentContext.run({
+					await runInInvocation({
 						agent: this,
 						connection,
 						request,
@@ -1655,7 +1664,7 @@ var Agent = class Agent extends Server {
 						} finally {
 							this.dequeue(row.id);
 						}
-					});
+					}, { detached: true });
 				}
 			}
 		} finally {
@@ -3295,7 +3304,7 @@ var Agent = class Agent extends Server {
 			console.error(`callback ${row.callback} not found`);
 			return;
 		}
-		await __DO_NOT_USE_WILL_BREAK__agentContext.run({
+		await runInInvocation({
 			agent: this,
 			connection: void 0,
 			request: void 0,
@@ -4091,7 +4100,7 @@ var Agent = class Agent extends Server {
 			headers: forwardedHeaders
 		};
 		if (req.headers.get("Upgrade")?.toLowerCase() === "websocket") forwardedHeaders.set(SUB_AGENT_OUTER_URL_HEADER, req.url);
-		if (req.body && req.method !== "GET" && req.method !== "HEAD") forwardedInit.body = await req.arrayBuffer();
+		if (req.body && req.method !== "GET" && req.method !== "HEAD") forwardedInit.body = req.body;
 		const forwarded = new Request(rewritten, forwardedInit);
 		return fetcher.fetch(forwarded);
 	}
@@ -4766,12 +4775,12 @@ var Agent = class Agent extends Server {
 			await invoke();
 			return;
 		}
-		await __DO_NOT_USE_WILL_BREAK__agentContext.run({
+		await runInInvocation({
 			agent: this,
 			connection: void 0,
 			request: void 0,
 			email: void 0
-		}, invoke);
+		}, invoke, { detached: true });
 	}
 	/**
 	* Arm the self-scheduling detached reconcile backbone. Existing schedules are
@@ -5727,7 +5736,7 @@ var Agent = class Agent extends Server {
 		}));
 		this._recordSubAgent(className, name, identity);
 		try {
-			await __DO_NOT_USE_WILL_BREAK__agentContext.run({
+			await runInInvocation({
 				agent: this,
 				connection: void 0,
 				request: void 0,
@@ -6046,7 +6055,8 @@ var Agent = class Agent extends Server {
 		};
 		const instance = await workflow.create({
 			id: workflowId,
-			params: augmentedParams
+			params: augmentedParams,
+			retention: options?.retention
 		});
 		const id = nanoid();
 		const metadataJson = options?.metadata ? JSON.stringify(options.metadata) : null;
@@ -7223,6 +7233,6 @@ var StreamingResponse = class {
 	}
 };
 //#endregion
-export { AGENT_TOOL_MILESTONE_PART, AGENT_TOOL_PROGRESS_PART, Agent, DEFAULT_AGENT_STATIC_OPTIONS, DurableObjectOAuthClientProvider, MCP_SERVER_ID_MAX_LENGTH, MessageType, SUB_PREFIX, SqlError, StreamingResponse, __DO_NOT_USE_WILL_BREAK__agentContext, callable, camelCaseToKebabCase, createHeaderBasedEmailResolver, getAgentByName, getCurrentAgent, getSubAgentByName, isDurableObjectCodeUpdateReset, isDurableObjectMemoryLimitReset, isDurableObjectStorageReset, isPlatformTransientError, normalizeServerId, parseSubAgentPath, routeAgentEmail, routeAgentRequest, routeSubAgentRequest, unstable_callable };
+export { AGENT_TOOL_MILESTONE_PART, AGENT_TOOL_PROGRESS_PART, Agent, DEFAULT_AGENT_STATIC_OPTIONS, DurableObjectOAuthClientProvider, MCP_SERVER_ID_MAX_LENGTH, MessageType, SUB_PREFIX, SqlError, StreamingResponse, __DO_NOT_USE_WILL_BREAK__agentContext, withInvocationScope as __DO_NOT_USE_WILL_BREAK__withInvocationScope, buildAgentPath, buildAgentUrl, callable, camelCaseToKebabCase, createHeaderBasedEmailResolver, getAgentByName, getCurrentAgent, getSubAgentByName, isDurableObjectCodeUpdateReset, isDurableObjectMemoryLimitReset, isDurableObjectStorageReset, isPlatformTransientError, normalizeServerId, parseSubAgentPath, routeAgentEmail, routeAgentRequest, routeSubAgentRequest, unstable_callable };
 
 //# sourceMappingURL=index.js.map