agents 0.19.0 → 0.20.1

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 { n as withInvocationScope, r as writeSpanAttributes, t as tracer } from "./cloudflare-BduZwmYK.js";
 import { SUB_PREFIX, 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-CcjiFpTf.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,
@@ -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,
@@ -6895,7 +6904,8 @@ var Agent = class Agent extends Server {
 			transport: {
 				...headerTransportOpts,
 				authProvider,
-				type: transportType
+				type: transportType,
+				skipIssuerMetadataValidation: resolvedOptions?.transport?.skipIssuerMetadataValidation
 			},
 			retry: resolvedOptions?.retry
 		});
@@ -7222,6 +7232,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, callable, camelCaseToKebabCase, createHeaderBasedEmailResolver, getAgentByName, getCurrentAgent, getSubAgentByName, isDurableObjectCodeUpdateReset, isDurableObjectMemoryLimitReset, isDurableObjectStorageReset, isPlatformTransientError, normalizeServerId, parseSubAgentPath, routeAgentEmail, routeAgentRequest, routeSubAgentRequest, unstable_callable };
 
 //# sourceMappingURL=index.js.map