agents 0.16.2 → 0.17.1

package/dist/mcp/client.d.ts

@@ -1,19 +1,23 @@
 import {
-  _t as MCPServerOptions,
-  bt as getNamespacedData,
-  dt as MCPClientOAuthCallbackConfig,
-  ft as MCPClientOAuthResult,
-  gt as MCPServerFilter,
-  ht as MCPOAuthCallbackResult,
-  lt as MCPClientManager,
-  mt as MCPDiscoverResult,
-  pt as MCPConnectionResult,
-  ut as MCPClientManagerOptions,
-  vt as MCP_SERVER_ID_MAX_LENGTH,
-  xt as normalizeServerId,
-  yt as RegisterServerOptions
-} from "../agent-tool-types-CTw3UJUP.js";
+  At as getNamespacedData,
+  Ct as MCPConnectionResult,
+  Dt as MCPServerOptions,
+  Et as MCPServerFilter,
+  Ot as MCP_SERVER_ID_MAX_LENGTH,
+  St as MCPClientOAuthResult,
+  Tt as MCPOAuthCallbackResult,
+  _t as MCPAITool,
+  bt as MCPClientManagerOptions,
+  jt as normalizeServerId,
+  kt as RegisterServerOptions,
+  vt as MCPAIToolSet,
+  wt as MCPDiscoverResult,
+  xt as MCPClientOAuthCallbackConfig,
+  yt as MCPClientManager
+} from "../agent-tool-types-CNyE1iz_.js";
 export {
+  MCPAITool,
+  MCPAIToolSet,
   MCPClientManager,
   MCPClientManagerOptions,
   MCPClientOAuthCallbackConfig,

package/dist/mcp/client.js

@@ -1,2 +1,2 @@
-import { i as normalizeServerId, n as MCP_SERVER_ID_MAX_LENGTH, r as getNamespacedData, t as MCPClientManager } from "../client-BXJ9n2f7.js";
+import { i as normalizeServerId, n as MCP_SERVER_ID_MAX_LENGTH, r as getNamespacedData, t as MCPClientManager } from "../client-BZ-B3NhC.js";
 export { MCPClientManager, MCP_SERVER_ID_MAX_LENGTH, getNamespacedData, normalizeServerId };

package/dist/mcp/index.d.ts

@@ -1,34 +1,34 @@
 import {
-  At as McpAgent,
-  Bt as WorkerTransportOptions,
-  Ct as RPCClientTransportOptions,
-  Dt as ElicitRequest,
-  Et as RPC_DO_PREFIX,
-  Ft as experimental_createMcpHandler,
-  Ht as StreamableHTTPEdgeClientTransport,
-  It as McpAuthContext,
-  Lt as getMcpAuthContext,
-  Mt as DurableObjectEventStore,
-  Nt as CreateMcpHandlerOptions,
-  Ot as ElicitRequestSchema,
-  Pt as createMcpHandler,
-  Rt as TransportState,
-  St as RPCClientTransport,
-  Tt as RPCServerTransportOptions,
-  Ut as McpClientOptions,
-  Vt as SSEEdgeClientTransport,
-  _t as MCPServerOptions,
-  dt as MCPClientOAuthCallbackConfig,
-  ft as MCPClientOAuthResult,
-  jt as ClearableEventStore,
-  kt as ElicitResult,
-  mt as MCPDiscoverResult,
-  pt as MCPConnectionResult,
-  vt as MCP_SERVER_ID_MAX_LENGTH,
-  wt as RPCServerTransport,
-  xt as normalizeServerId,
-  zt as WorkerTransport
-} from "../agent-tool-types-CTw3UJUP.js";
+  $t as McpClientOptions,
+  Bt as McpAgent,
+  Ct as MCPConnectionResult,
+  Dt as MCPServerOptions,
+  Ft as RPCServerTransportOptions,
+  Gt as experimental_createMcpHandler,
+  Ht as DurableObjectEventStore,
+  It as RPC_DO_PREFIX,
+  Jt as TransportState,
+  Kt as McpAuthContext,
+  Lt as ElicitRequest,
+  Mt as RPCClientTransport,
+  Nt as RPCClientTransportOptions,
+  Ot as MCP_SERVER_ID_MAX_LENGTH,
+  Pt as RPCServerTransport,
+  Qt as StreamableHTTPEdgeClientTransport,
+  Rt as ElicitRequestSchema,
+  St as MCPClientOAuthResult,
+  Ut as CreateMcpHandlerOptions,
+  Vt as ClearableEventStore,
+  Wt as createMcpHandler,
+  Xt as WorkerTransportOptions,
+  Yt as WorkerTransport,
+  Zt as SSEEdgeClientTransport,
+  jt as normalizeServerId,
+  qt as getMcpAuthContext,
+  wt as MCPDiscoverResult,
+  xt as MCPClientOAuthCallbackConfig,
+  zt as ElicitResult
+} from "../agent-tool-types-CNyE1iz_.js";
 export {
   type ClearableEventStore,
   type CreateMcpHandlerOptions,

package/dist/mcp/index.js

@@ -1,5 +1,5 @@
 import "../types.js";
-import { c as RPC_DO_PREFIX, i as normalizeServerId, n as MCP_SERVER_ID_MAX_LENGTH, o as RPCClientTransport, s as RPCServerTransport } from "../client-BXJ9n2f7.js";
+import { c as RPC_DO_PREFIX, i as normalizeServerId, n as MCP_SERVER_ID_MAX_LENGTH, o as RPCClientTransport, s as RPCServerTransport } from "../client-BZ-B3NhC.js";
 import { Agent, getAgentByName, getCurrentAgent } from "../index.js";
 import { AsyncLocalStorage } from "node:async_hooks";
 import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
@@ -588,6 +588,7 @@ var StreamableHTTPServerTransport = class {
 		this._streamResponseIds = /* @__PURE__ */ new Map();
 		const { agent } = getCurrentAgent();
 		if (!agent) throw new Error("McpAgent was not found in Transport constructor");
+		this._agent = agent;
 		this.sessionId = agent.getSessionId();
 		this._eventStore = options.eventStore;
 	}
@@ -747,8 +748,7 @@ var StreamableHTTPServerTransport = class {
 		}
 	}
 	async close() {
-		const { agent } = getCurrentAgent();
-		if (!agent) throw new Error("Agent was not found in close");
+		const agent = this._agent;
 		for (const conn of agent.getConnections()) conn.close(1e3, "Session closed");
 		this.onclose?.();
 	}
@@ -803,8 +803,7 @@ var StreamableHTTPServerTransport = class {
 	* there is at most one to send on.
 	*/
 	async sendStandalone(message) {
-		const { agent } = getCurrentAgent();
-		if (!agent) throw new Error("Agent was not found in send");
+		const agent = this._agent;
 		const eventId = await this._eventStore?.storeEvent(STANDALONE_STREAM_ID, message);
 		const standalone = Array.from(agent.getConnections()).find((conn) => conn.state?._standaloneSse);
 		if (standalone) this.writeSSEEvent(standalone, message, eventId);
@@ -817,8 +816,9 @@ var StreamableHTTPServerTransport = class {
 	* {@link sendOnStream} for the actual store / write / cleanup.
 	*/
 	async sendForRequest(message, requestId) {
-		const { agent, connection: originatingConnection } = getCurrentAgent();
-		if (!agent) throw new Error("Agent was not found in send");
+		const agent = this._agent;
+		const context = getCurrentAgent();
+		const originatingConnection = context.agent === agent ? context.connection : void 0;
 		const matchingConnections = Array.from(agent.getConnections()).filter((conn) => conn.state?.requestIds?.includes(requestId));
 		const liveConnection = matchingConnections.find((conn) => conn.id === originatingConnection?.id) ?? (matchingConnections.length === 1 ? matchingConnections[0] : null);
 		if (!liveConnection && matchingConnections.length > 1) {