agents 0.0.0-a215bb2 → 0.0.0-a3b7e23

package/dist/{client-C-nwz-3N.d.ts → client-DnzF-4rM.d.ts}

@@ -1,6 +1,14 @@
-import { MCPObservabilityEvent } from "./mcp-BEwaCsxO.js";
-import { AgentsOAuthProvider } from "./do-oauth-client-provider-CCwGwnrA.js";
+import { t as MCPObservabilityEvent } from "./mcp-Dw5vDrY8.js";
+import { t as AgentsOAuthProvider } from "./do-oauth-client-provider-DGc5pP0l.js";
 import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import {
+  SSEClientTransport,
+  SSEClientTransportOptions
+} from "@modelcontextprotocol/sdk/client/sse.js";
+import {
+  StreamableHTTPClientTransport,
+  StreamableHTTPClientTransportOptions
+} from "@modelcontextprotocol/sdk/client/streamableHttp.js";
 import {
   CallToolRequest,
   CallToolResultSchema,
@@ -15,14 +23,6 @@ import {
   ServerCapabilities,
   Tool
 } from "@modelcontextprotocol/sdk/types.js";
-import {
-  SSEClientTransport,
-  SSEClientTransportOptions
-} from "@modelcontextprotocol/sdk/client/sse.js";
-import {
-  StreamableHTTPClientTransport,
-  StreamableHTTPClientTransportOptions
-} from "@modelcontextprotocol/sdk/client/streamableHttp.js";
 import * as ai0 from "ai";
 import { ToolSet } from "ai";
 import * as zod0 from "zod";
@@ -34,36 +34,6 @@ interface Disposable {
 }
 type Event<T> = (listener: (e: T) => void) => Disposable;
 //#endregion
-//#region src/mcp/sse-edge.d.ts
-declare class SSEEdgeClientTransport extends SSEClientTransport {
-  private authProvider;
-  /**
-   * Creates a new EdgeSSEClientTransport, which overrides fetch to be compatible with the CF workers environment
-   */
-  constructor(url: URL, options: SSEClientTransportOptions);
-  authHeaders(): Promise<
-    | {
-        Authorization: string;
-      }
-    | undefined
-  >;
-}
-//#endregion
-//#region src/mcp/streamable-http-edge.d.ts
-declare class StreamableHTTPEdgeClientTransport extends StreamableHTTPClientTransport {
-  private authProvider;
-  /**
-   * Creates a new StreamableHTTPEdgeClientTransport, which overrides fetch to be compatible with the CF workers environment
-   */
-  constructor(url: URL, options: StreamableHTTPClientTransportOptions);
-  authHeaders(): Promise<
-    | {
-        Authorization: string;
-      }
-    | undefined
-  >;
-}
-//#endregion
 //#region src/mcp/types.d.ts
 type MaybePromise<T> = T | Promise<T>;
 type BaseTransportType = "sse" | "streamable-http";
@@ -79,6 +49,7 @@ interface ServeOptions {
   binding?: string;
   corsOptions?: CORSOptions;
   transport?: BaseTransportType;
+  jurisdiction?: DurableObjectJurisdiction;
 }
 //#endregion
 //#region src/mcp/client-connection.d.ts
@@ -168,13 +139,13 @@ declare class MCPClientConnection {
           | undefined;
         required?: string[] | undefined;
       };
-      description?: string | undefined;
       _meta?:
         | {
             [x: string]: unknown;
           }
         | undefined;
       title?: string | undefined;
+      description?: string | undefined;
       icons?:
         | {
             [x: string]: unknown;
@@ -212,7 +183,6 @@ declare class MCPClientConnection {
       [x: string]: unknown;
       name: string;
       uri: string;
-      description?: string | undefined;
       _meta?:
         | {
             [x: string]: unknown;
@@ -220,6 +190,7 @@ declare class MCPClientConnection {
         | undefined;
       mimeType?: string | undefined;
       title?: string | undefined;
+      description?: string | undefined;
       icons?:
         | {
             [x: string]: unknown;
@@ -234,7 +205,6 @@ declare class MCPClientConnection {
     {
       [x: string]: unknown;
       name: string;
-      description?: string | undefined;
       _meta?:
         | {
             [x: string]: unknown;
@@ -249,6 +219,7 @@ declare class MCPClientConnection {
           }[]
         | undefined;
       title?: string | undefined;
+      description?: string | undefined;
       icons?:
         | {
             [x: string]: unknown;
@@ -264,7 +235,6 @@ declare class MCPClientConnection {
       [x: string]: unknown;
       name: string;
       uriTemplate: string;
-      description?: string | undefined;
       _meta?:
         | {
             [x: string]: unknown;
@@ -272,6 +242,7 @@ declare class MCPClientConnection {
         | undefined;
       mimeType?: string | undefined;
       title?: string | undefined;
+      description?: string | undefined;
       icons?:
         | {
             [x: string]: unknown;
@@ -294,7 +265,7 @@ declare class MCPClientConnection {
    */
   getTransport(
     transportType: BaseTransportType
-  ): StreamableHTTPEdgeClientTransport | SSEEdgeClientTransport;
+  ): StreamableHTTPClientTransport | SSEClientTransport;
   private tryConnect;
   private _capabilityErrorHandler;
 }
@@ -360,12 +331,12 @@ declare class MCPClientManager {
     | {
         serverId: string;
         authSuccess: boolean;
-        authError?: undefined;
+        authError: string;
       }
     | {
         serverId: string;
         authSuccess: boolean;
-        authError: string;
+        authError?: undefined;
       }
   >;
   /**
@@ -5298,16 +5269,15 @@ declare function getNamespacedData<T extends keyof NamespacedData>(
 ): NamespacedData[T];
 //#endregion
 export {
-  BaseTransportType,
-  MCPClientManager,
-  MCPClientOAuthCallbackConfig,
-  MCPClientOAuthResult,
-  MCPConnectionState,
-  MaybePromise,
-  SSEEdgeClientTransport,
-  ServeOptions,
-  StreamableHTTPEdgeClientTransport,
-  TransportType,
-  getNamespacedData
+  MCPConnectionState as a,
+  MaybePromise as c,
+  getNamespacedData as i,
+  ServeOptions as l,
+  MCPClientOAuthCallbackConfig as n,
+  BaseTransportType as o,
+  MCPClientOAuthResult as r,
+  CORSOptions as s,
+  MCPClientManager as t,
+  TransportType as u
 };
-//# sourceMappingURL=client-C-nwz-3N.d.ts.map
+//# sourceMappingURL=client-DnzF-4rM.d.ts.map

package/dist/client.d.ts

@@ -1,12 +1,12 @@
-import "./serializable-gtr9YMhp.js";
+import "./serializable-CymX8ovI.js";
 import {
-  AgentClient,
-  AgentClientFetchOptions,
-  AgentClientOptions,
-  StreamOptions,
-  agentFetch,
-  camelCaseToKebabCase
-} from "./client-CmMi85Sj.js";
+  a as agentFetch,
+  i as StreamOptions,
+  n as AgentClientFetchOptions,
+  o as camelCaseToKebabCase,
+  r as AgentClientOptions,
+  t as AgentClient
+} from "./client-CbWe9FBd.js";
 export {
   AgentClient,
   AgentClientFetchOptions,

package/dist/client.js

@@ -1,4 +1,4 @@
-import "./ai-types-UZlfLOYP.js";
-import { AgentClient, agentFetch, camelCaseToKebabCase } from "./client-DjR-lC16.js";
+import "./ai-types-B3aQaFv3.js";
+import { n as agentFetch, r as camelCaseToKebabCase, t as AgentClient } from "./client-BfiZ3HQd.js";
 
 export { AgentClient, agentFetch, camelCaseToKebabCase };

package/dist/codemode/ai.js

@@ -1,8 +1,8 @@
-import "../ai-types-UZlfLOYP.js";
-import "../client-DjR-lC16.js";
-import "../client-CZBVDDoO.js";
-import "../do-oauth-client-provider-B2jr6UNq.js";
-import { getAgentByName } from "../src-L3cHuAag.js";
+import "../ai-types-B3aQaFv3.js";
+import "../client-BfiZ3HQd.js";
+import "../client-DZhjV_XA.js";
+import "../do-oauth-client-provider-CswoD5Lu.js";
+import { s as getAgentByName } from "../src-CwIW7tU2.js";
 import { generateObject, tool } from "ai";
 import { openai } from "@ai-sdk/openai";
 import { z } from "zod/v3";

package/dist/{do-oauth-client-provider-B2jr6UNq.js → do-oauth-client-provider-CswoD5Lu.js}

@@ -89,5 +89,5 @@ var DurableObjectOAuthClientProvider = class {
 };
 
 //#endregion
-export { DurableObjectOAuthClientProvider };
-//# sourceMappingURL=do-oauth-client-provider-B2jr6UNq.js.map
+export { DurableObjectOAuthClientProvider as t };
+//# sourceMappingURL=do-oauth-client-provider-CswoD5Lu.js.map

package/dist/{do-oauth-client-provider-B2jr6UNq.js.map → do-oauth-client-provider-CswoD5Lu.js.map}

@@ -1 +1 @@
-{"version":3,"file":"do-oauth-client-provider-B2jr6UNq.js","names":["storage: DurableObjectStorage","clientName: string","baseRedirectUrl: string"],"sources":["../src/mcp/do-oauth-client-provider.ts"],"sourcesContent":["import type { OAuthClientProvider } from \"@modelcontextprotocol/sdk/client/auth.js\";\nimport type {\n  OAuthClientInformation,\n  OAuthClientInformationFull,\n  OAuthClientMetadata,\n  OAuthTokens\n} from \"@modelcontextprotocol/sdk/shared/auth.js\";\nimport { nanoid } from \"nanoid\";\n\n// A slight extension to the standard OAuthClientProvider interface because `redirectToAuthorization` doesn't give us the interface we need\n// This allows us to track authentication for a specific server and associated dynamic client registration\nexport interface AgentsOAuthProvider extends OAuthClientProvider {\n  authUrl: string | undefined;\n  clientId: string | undefined;\n  serverId: string | undefined;\n}\n\nexport class DurableObjectOAuthClientProvider implements AgentsOAuthProvider {\n  private _authUrl_: string | undefined;\n  private _serverId_: string | undefined;\n  private _clientId_: string | undefined;\n\n  constructor(\n    public storage: DurableObjectStorage,\n    public clientName: string,\n    public baseRedirectUrl: string\n  ) {}\n\n  get clientMetadata(): OAuthClientMetadata {\n    return {\n      client_name: this.clientName,\n      client_uri: this.clientUri,\n      grant_types: [\"authorization_code\", \"refresh_token\"],\n      redirect_uris: [this.redirectUrl],\n      response_types: [\"code\"],\n      token_endpoint_auth_method: \"none\"\n    };\n  }\n\n  get clientUri() {\n    return new URL(this.redirectUrl).origin;\n  }\n\n  get redirectUrl() {\n    return `${this.baseRedirectUrl}/${this.serverId}`;\n  }\n\n  get clientId() {\n    if (!this._clientId_) {\n      throw new Error(\"Trying to access clientId before it was set\");\n    }\n    return this._clientId_;\n  }\n\n  set clientId(clientId_: string) {\n    this._clientId_ = clientId_;\n  }\n\n  get serverId() {\n    if (!this._serverId_) {\n      throw new Error(\"Trying to access serverId before it was set\");\n    }\n    return this._serverId_;\n  }\n\n  set serverId(serverId_: string) {\n    this._serverId_ = serverId_;\n  }\n\n  keyPrefix(clientId: string) {\n    return `/${this.clientName}/${this.serverId}/${clientId}`;\n  }\n\n  clientInfoKey(clientId: string) {\n    return `${this.keyPrefix(clientId)}/client_info/`;\n  }\n\n  async clientInformation(): Promise<OAuthClientInformation | undefined> {\n    if (!this._clientId_) {\n      return undefined;\n    }\n    return (\n      (await this.storage.get<OAuthClientInformation>(\n        this.clientInfoKey(this.clientId)\n      )) ?? undefined\n    );\n  }\n\n  async saveClientInformation(\n    clientInformation: OAuthClientInformationFull\n  ): Promise<void> {\n    await this.storage.put(\n      this.clientInfoKey(clientInformation.client_id),\n      clientInformation\n    );\n    this.clientId = clientInformation.client_id;\n  }\n\n  tokenKey(clientId: string) {\n    return `${this.keyPrefix(clientId)}/token`;\n  }\n\n  async tokens(): Promise<OAuthTokens | undefined> {\n    if (!this._clientId_) {\n      return undefined;\n    }\n    return (\n      (await this.storage.get<OAuthTokens>(this.tokenKey(this.clientId))) ??\n      undefined\n    );\n  }\n\n  async saveTokens(tokens: OAuthTokens): Promise<void> {\n    await this.storage.put(this.tokenKey(this.clientId), tokens);\n  }\n\n  get authUrl() {\n    return this._authUrl_;\n  }\n\n  /**\n   * Because this operates on the server side (but we need browser auth), we send this url back to the user\n   * and require user interact to initiate the redirect flow\n   */\n  async redirectToAuthorization(authUrl: URL): Promise<void> {\n    // Generate secure random token for state parameter\n    const stateToken = nanoid();\n    authUrl.searchParams.set(\"state\", stateToken);\n    this._authUrl_ = authUrl.toString();\n  }\n\n  codeVerifierKey(clientId: string) {\n    return `${this.keyPrefix(clientId)}/code_verifier`;\n  }\n\n  async saveCodeVerifier(verifier: string): Promise<void> {\n    const key = this.codeVerifierKey(this.clientId);\n\n    // Don't overwrite existing verifier to preserve first PKCE verifier\n    const existing = await this.storage.get<string>(key);\n    if (existing) {\n      return;\n    }\n\n    await this.storage.put(key, verifier);\n  }\n\n  async codeVerifier(): Promise<string> {\n    const codeVerifier = await this.storage.get<string>(\n      this.codeVerifierKey(this.clientId)\n    );\n    if (!codeVerifier) {\n      throw new Error(\"No code verifier found\");\n    }\n    return codeVerifier;\n  }\n}\n"],"mappings":";;;AAiBA,IAAa,mCAAb,MAA6E;CAK3E,YACE,AAAOA,SACP,AAAOC,YACP,AAAOC,iBACP;EAHO;EACA;EACA;;CAGT,IAAI,iBAAsC;AACxC,SAAO;GACL,aAAa,KAAK;GAClB,YAAY,KAAK;GACjB,aAAa,CAAC,sBAAsB,gBAAgB;GACpD,eAAe,CAAC,KAAK,YAAY;GACjC,gBAAgB,CAAC,OAAO;GACxB,4BAA4B;GAC7B;;CAGH,IAAI,YAAY;AACd,SAAO,IAAI,IAAI,KAAK,YAAY,CAAC;;CAGnC,IAAI,cAAc;AAChB,SAAO,GAAG,KAAK,gBAAgB,GAAG,KAAK;;CAGzC,IAAI,WAAW;AACb,MAAI,CAAC,KAAK,WACR,OAAM,IAAI,MAAM,8CAA8C;AAEhE,SAAO,KAAK;;CAGd,IAAI,SAAS,WAAmB;AAC9B,OAAK,aAAa;;CAGpB,IAAI,WAAW;AACb,MAAI,CAAC,KAAK,WACR,OAAM,IAAI,MAAM,8CAA8C;AAEhE,SAAO,KAAK;;CAGd,IAAI,SAAS,WAAmB;AAC9B,OAAK,aAAa;;CAGpB,UAAU,UAAkB;AAC1B,SAAO,IAAI,KAAK,WAAW,GAAG,KAAK,SAAS,GAAG;;CAGjD,cAAc,UAAkB;AAC9B,SAAO,GAAG,KAAK,UAAU,SAAS,CAAC;;CAGrC,MAAM,oBAAiE;AACrE,MAAI,CAAC,KAAK,WACR;AAEF,SACG,MAAM,KAAK,QAAQ,IAClB,KAAK,cAAc,KAAK,SAAS,CAClC,IAAK;;CAIV,MAAM,sBACJ,mBACe;AACf,QAAM,KAAK,QAAQ,IACjB,KAAK,cAAc,kBAAkB,UAAU,EAC/C,kBACD;AACD,OAAK,WAAW,kBAAkB;;CAGpC,SAAS,UAAkB;AACzB,SAAO,GAAG,KAAK,UAAU,SAAS,CAAC;;CAGrC,MAAM,SAA2C;AAC/C,MAAI,CAAC,KAAK,WACR;AAEF,SACG,MAAM,KAAK,QAAQ,IAAiB,KAAK,SAAS,KAAK,SAAS,CAAC,IAClE;;CAIJ,MAAM,WAAW,QAAoC;AACnD,QAAM,KAAK,QAAQ,IAAI,KAAK,SAAS,KAAK,SAAS,EAAE,OAAO;;CAG9D,IAAI,UAAU;AACZ,SAAO,KAAK;;;;;;CAOd,MAAM,wBAAwB,SAA6B;EAEzD,MAAM,aAAa,QAAQ;AAC3B,UAAQ,aAAa,IAAI,SAAS,WAAW;AAC7C,OAAK,YAAY,QAAQ,UAAU;;CAGrC,gBAAgB,UAAkB;AAChC,SAAO,GAAG,KAAK,UAAU,SAAS,CAAC;;CAGrC,MAAM,iBAAiB,UAAiC;EACtD,MAAM,MAAM,KAAK,gBAAgB,KAAK,SAAS;AAI/C,MADiB,MAAM,KAAK,QAAQ,IAAY,IAAI,CAElD;AAGF,QAAM,KAAK,QAAQ,IAAI,KAAK,SAAS;;CAGvC,MAAM,eAAgC;EACpC,MAAM,eAAe,MAAM,KAAK,QAAQ,IACtC,KAAK,gBAAgB,KAAK,SAAS,CACpC;AACD,MAAI,CAAC,aACH,OAAM,IAAI,MAAM,yBAAyB;AAE3C,SAAO"}
+{"version":3,"file":"do-oauth-client-provider-CswoD5Lu.js","names":["storage: DurableObjectStorage","clientName: string","baseRedirectUrl: string"],"sources":["../src/mcp/do-oauth-client-provider.ts"],"sourcesContent":["import type { OAuthClientProvider } from \"@modelcontextprotocol/sdk/client/auth.js\";\nimport type {\n  OAuthClientInformation,\n  OAuthClientInformationFull,\n  OAuthClientMetadata,\n  OAuthTokens\n} from \"@modelcontextprotocol/sdk/shared/auth.js\";\nimport { nanoid } from \"nanoid\";\n\n// A slight extension to the standard OAuthClientProvider interface because `redirectToAuthorization` doesn't give us the interface we need\n// This allows us to track authentication for a specific server and associated dynamic client registration\nexport interface AgentsOAuthProvider extends OAuthClientProvider {\n  authUrl: string | undefined;\n  clientId: string | undefined;\n  serverId: string | undefined;\n}\n\nexport class DurableObjectOAuthClientProvider implements AgentsOAuthProvider {\n  private _authUrl_: string | undefined;\n  private _serverId_: string | undefined;\n  private _clientId_: string | undefined;\n\n  constructor(\n    public storage: DurableObjectStorage,\n    public clientName: string,\n    public baseRedirectUrl: string\n  ) {}\n\n  get clientMetadata(): OAuthClientMetadata {\n    return {\n      client_name: this.clientName,\n      client_uri: this.clientUri,\n      grant_types: [\"authorization_code\", \"refresh_token\"],\n      redirect_uris: [this.redirectUrl],\n      response_types: [\"code\"],\n      token_endpoint_auth_method: \"none\"\n    };\n  }\n\n  get clientUri() {\n    return new URL(this.redirectUrl).origin;\n  }\n\n  get redirectUrl() {\n    return `${this.baseRedirectUrl}/${this.serverId}`;\n  }\n\n  get clientId() {\n    if (!this._clientId_) {\n      throw new Error(\"Trying to access clientId before it was set\");\n    }\n    return this._clientId_;\n  }\n\n  set clientId(clientId_: string) {\n    this._clientId_ = clientId_;\n  }\n\n  get serverId() {\n    if (!this._serverId_) {\n      throw new Error(\"Trying to access serverId before it was set\");\n    }\n    return this._serverId_;\n  }\n\n  set serverId(serverId_: string) {\n    this._serverId_ = serverId_;\n  }\n\n  keyPrefix(clientId: string) {\n    return `/${this.clientName}/${this.serverId}/${clientId}`;\n  }\n\n  clientInfoKey(clientId: string) {\n    return `${this.keyPrefix(clientId)}/client_info/`;\n  }\n\n  async clientInformation(): Promise<OAuthClientInformation | undefined> {\n    if (!this._clientId_) {\n      return undefined;\n    }\n    return (\n      (await this.storage.get<OAuthClientInformation>(\n        this.clientInfoKey(this.clientId)\n      )) ?? undefined\n    );\n  }\n\n  async saveClientInformation(\n    clientInformation: OAuthClientInformationFull\n  ): Promise<void> {\n    await this.storage.put(\n      this.clientInfoKey(clientInformation.client_id),\n      clientInformation\n    );\n    this.clientId = clientInformation.client_id;\n  }\n\n  tokenKey(clientId: string) {\n    return `${this.keyPrefix(clientId)}/token`;\n  }\n\n  async tokens(): Promise<OAuthTokens | undefined> {\n    if (!this._clientId_) {\n      return undefined;\n    }\n    return (\n      (await this.storage.get<OAuthTokens>(this.tokenKey(this.clientId))) ??\n      undefined\n    );\n  }\n\n  async saveTokens(tokens: OAuthTokens): Promise<void> {\n    await this.storage.put(this.tokenKey(this.clientId), tokens);\n  }\n\n  get authUrl() {\n    return this._authUrl_;\n  }\n\n  /**\n   * Because this operates on the server side (but we need browser auth), we send this url back to the user\n   * and require user interact to initiate the redirect flow\n   */\n  async redirectToAuthorization(authUrl: URL): Promise<void> {\n    // Generate secure random token for state parameter\n    const stateToken = nanoid();\n    authUrl.searchParams.set(\"state\", stateToken);\n    this._authUrl_ = authUrl.toString();\n  }\n\n  codeVerifierKey(clientId: string) {\n    return `${this.keyPrefix(clientId)}/code_verifier`;\n  }\n\n  async saveCodeVerifier(verifier: string): Promise<void> {\n    const key = this.codeVerifierKey(this.clientId);\n\n    // Don't overwrite existing verifier to preserve first PKCE verifier\n    const existing = await this.storage.get<string>(key);\n    if (existing) {\n      return;\n    }\n\n    await this.storage.put(key, verifier);\n  }\n\n  async codeVerifier(): Promise<string> {\n    const codeVerifier = await this.storage.get<string>(\n      this.codeVerifierKey(this.clientId)\n    );\n    if (!codeVerifier) {\n      throw new Error(\"No code verifier found\");\n    }\n    return codeVerifier;\n  }\n}\n"],"mappings":";;;AAiBA,IAAa,mCAAb,MAA6E;CAK3E,YACE,AAAOA,SACP,AAAOC,YACP,AAAOC,iBACP;EAHO;EACA;EACA;;CAGT,IAAI,iBAAsC;AACxC,SAAO;GACL,aAAa,KAAK;GAClB,YAAY,KAAK;GACjB,aAAa,CAAC,sBAAsB,gBAAgB;GACpD,eAAe,CAAC,KAAK,YAAY;GACjC,gBAAgB,CAAC,OAAO;GACxB,4BAA4B;GAC7B;;CAGH,IAAI,YAAY;AACd,SAAO,IAAI,IAAI,KAAK,YAAY,CAAC;;CAGnC,IAAI,cAAc;AAChB,SAAO,GAAG,KAAK,gBAAgB,GAAG,KAAK;;CAGzC,IAAI,WAAW;AACb,MAAI,CAAC,KAAK,WACR,OAAM,IAAI,MAAM,8CAA8C;AAEhE,SAAO,KAAK;;CAGd,IAAI,SAAS,WAAmB;AAC9B,OAAK,aAAa;;CAGpB,IAAI,WAAW;AACb,MAAI,CAAC,KAAK,WACR,OAAM,IAAI,MAAM,8CAA8C;AAEhE,SAAO,KAAK;;CAGd,IAAI,SAAS,WAAmB;AAC9B,OAAK,aAAa;;CAGpB,UAAU,UAAkB;AAC1B,SAAO,IAAI,KAAK,WAAW,GAAG,KAAK,SAAS,GAAG;;CAGjD,cAAc,UAAkB;AAC9B,SAAO,GAAG,KAAK,UAAU,SAAS,CAAC;;CAGrC,MAAM,oBAAiE;AACrE,MAAI,CAAC,KAAK,WACR;AAEF,SACG,MAAM,KAAK,QAAQ,IAClB,KAAK,cAAc,KAAK,SAAS,CAClC,IAAK;;CAIV,MAAM,sBACJ,mBACe;AACf,QAAM,KAAK,QAAQ,IACjB,KAAK,cAAc,kBAAkB,UAAU,EAC/C,kBACD;AACD,OAAK,WAAW,kBAAkB;;CAGpC,SAAS,UAAkB;AACzB,SAAO,GAAG,KAAK,UAAU,SAAS,CAAC;;CAGrC,MAAM,SAA2C;AAC/C,MAAI,CAAC,KAAK,WACR;AAEF,SACG,MAAM,KAAK,QAAQ,IAAiB,KAAK,SAAS,KAAK,SAAS,CAAC,IAClE;;CAIJ,MAAM,WAAW,QAAoC;AACnD,QAAM,KAAK,QAAQ,IAAI,KAAK,SAAS,KAAK,SAAS,EAAE,OAAO;;CAG9D,IAAI,UAAU;AACZ,SAAO,KAAK;;;;;;CAOd,MAAM,wBAAwB,SAA6B;EAEzD,MAAM,aAAa,QAAQ;AAC3B,UAAQ,aAAa,IAAI,SAAS,WAAW;AAC7C,OAAK,YAAY,QAAQ,UAAU;;CAGrC,gBAAgB,UAAkB;AAChC,SAAO,GAAG,KAAK,UAAU,SAAS,CAAC;;CAGrC,MAAM,iBAAiB,UAAiC;EACtD,MAAM,MAAM,KAAK,gBAAgB,KAAK,SAAS;AAI/C,MADiB,MAAM,KAAK,QAAQ,IAAY,IAAI,CAElD;AAGF,QAAM,KAAK,QAAQ,IAAI,KAAK,SAAS;;CAGvC,MAAM,eAAgC;EACpC,MAAM,eAAe,MAAM,KAAK,QAAQ,IACtC,KAAK,gBAAgB,KAAK,SAAS,CACpC;AACD,MAAI,CAAC,aACH,OAAM,IAAI,MAAM,yBAAyB;AAE3C,SAAO"}

package/dist/{do-oauth-client-provider-CCwGwnrA.d.ts → do-oauth-client-provider-DGc5pP0l.d.ts}

@@ -51,5 +51,5 @@ declare class DurableObjectOAuthClientProvider implements AgentsOAuthProvider {
   codeVerifier(): Promise<string>;
 }
 //#endregion
-export { AgentsOAuthProvider, DurableObjectOAuthClientProvider };
-//# sourceMappingURL=do-oauth-client-provider-CCwGwnrA.d.ts.map
+export { DurableObjectOAuthClientProvider as n, AgentsOAuthProvider as t };
+//# sourceMappingURL=do-oauth-client-provider-DGc5pP0l.d.ts.map

package/dist/{index-CMEWpbHx.d.ts → index-Db_90HY3.d.ts}

@@ -1,10 +1,10 @@
 import {
-  MCPClientManager,
-  MCPConnectionState,
-  TransportType
-} from "./client-C-nwz-3N.js";
-import { Observability } from "./index-W4JUkafc.js";
-import { MessageType } from "./ai-types-BWW4umHY.js";
+  a as MCPConnectionState,
+  t as MCPClientManager,
+  u as TransportType
+} from "./client-DnzF-4rM.js";
+import { t as Observability } from "./index-DhJCaDWd.js";
+import { n as MessageType } from "./ai-types-D5YoPrBZ.js";
 import {
   Connection,
   Connection as Connection$1,
@@ -171,6 +171,7 @@ declare class Agent<
 > extends Server<Env, Props> {
   private _state;
   private _disposables;
+  private _mcpStateRestored;
   private _ParentClass;
   readonly mcp: MCPClientManager;
   /**
@@ -346,6 +347,7 @@ declare class Agent<
    * @returns A map of method names to their metadata
    */
   private _isCallable;
+  private _ensureMcpStateRestored;
   /**
    * Connect to a new MCP Server
    *
@@ -372,16 +374,6 @@ declare class Agent<
     id: string;
     authUrl: string | undefined;
   }>;
-  /**
-   * Handle potential OAuth callback requests after DO hibernation.
-   * Detects OAuth callbacks, restores state from database, and processes the callback.
-   * Returns a Response if this was an OAuth callback, otherwise returns undefined.
-   */
-  private _handlePotentialOAuthCallback;
-  /**
-   * Process an OAuth callback request (assumes state is already restored)
-   */
-  private _processOAuthCallback;
   private _connectToMcpServerInternal;
   removeMcpServer(id: string): Promise<void>;
   getMcpServers(): MCPServersState;
@@ -534,35 +526,35 @@ declare class StreamingResponse {
 }
 //#endregion
 export {
-  Agent,
-  AgentContext,
-  AgentEmail,
-  AgentNamespace,
-  AgentOptions,
-  CallableMetadata,
-  type Connection$1 as Connection,
-  type ConnectionContext,
-  EmailResolver,
-  EmailRoutingOptions,
-  EmailSendOptions,
-  MCPServer,
-  MCPServerMessage,
-  MCPServersState,
-  QueueItem,
-  RPCRequest,
-  RPCResponse,
-  Schedule,
-  StateUpdateMessage,
-  StreamingResponse,
-  type WSMessage,
-  callable,
-  createAddressBasedEmailResolver,
-  createCatchAllEmailResolver,
-  createHeaderBasedEmailResolver,
-  getAgentByName,
-  getCurrentAgent,
-  routeAgentEmail,
-  routeAgentRequest,
-  unstable_callable
+  unstable_callable as A,
+  createAddressBasedEmailResolver as C,
+  getCurrentAgent as D,
+  getAgentByName as E,
+  routeAgentEmail as O,
+  callable as S,
+  createHeaderBasedEmailResolver as T,
+  RPCResponse as _,
+  AgentOptions as a,
+  StreamingResponse as b,
+  ConnectionContext as c,
+  EmailSendOptions as d,
+  MCPServer as f,
+  RPCRequest as g,
+  QueueItem as h,
+  AgentNamespace as i,
+  routeAgentRequest as k,
+  EmailResolver as l,
+  MCPServersState as m,
+  AgentContext as n,
+  CallableMetadata as o,
+  MCPServerMessage as p,
+  AgentEmail as r,
+  Connection$1 as s,
+  Agent as t,
+  EmailRoutingOptions as u,
+  Schedule as v,
+  createCatchAllEmailResolver as w,
+  WSMessage as x,
+  StateUpdateMessage as y
 };
-//# sourceMappingURL=index-CMEWpbHx.d.ts.map
+//# sourceMappingURL=index-Db_90HY3.d.ts.map

package/dist/{index-W4JUkafc.d.ts → index-DhJCaDWd.d.ts}

@@ -1,4 +1,4 @@
-import { BaseEvent, MCPObservabilityEvent } from "./mcp-BEwaCsxO.js";
+import { n as BaseEvent, t as MCPObservabilityEvent } from "./mcp-Dw5vDrY8.js";
 
 //#region src/observability/agent.d.ts
 
@@ -50,5 +50,9 @@ interface Observability {
  */
 declare const genericObservability: Observability;
 //#endregion
-export { Observability, ObservabilityEvent, genericObservability };
-//# sourceMappingURL=index-W4JUkafc.d.ts.map
+export {
+  ObservabilityEvent as n,
+  genericObservability as r,
+  Observability as t
+};
+//# sourceMappingURL=index-DhJCaDWd.d.ts.map

package/dist/index.d.ts

@@ -1,40 +1,40 @@
-import { TransportType } from "./client-C-nwz-3N.js";
-import "./mcp-BEwaCsxO.js";
-import "./do-oauth-client-provider-CCwGwnrA.js";
-import "./index-W4JUkafc.js";
-import "./ai-types-BWW4umHY.js";
+import { u as TransportType } from "./client-DnzF-4rM.js";
+import "./mcp-Dw5vDrY8.js";
+import "./do-oauth-client-provider-DGc5pP0l.js";
+import "./index-DhJCaDWd.js";
+import "./ai-types-D5YoPrBZ.js";
 import {
-  Agent,
-  AgentContext,
-  AgentEmail,
-  AgentNamespace,
-  AgentOptions,
-  CallableMetadata,
-  Connection,
-  ConnectionContext,
-  EmailResolver,
-  EmailRoutingOptions,
-  EmailSendOptions,
-  MCPServer,
-  MCPServerMessage,
-  MCPServersState,
-  QueueItem,
-  RPCRequest,
-  RPCResponse,
-  Schedule,
-  StateUpdateMessage,
-  StreamingResponse,
-  WSMessage,
-  callable,
-  createAddressBasedEmailResolver,
-  createCatchAllEmailResolver,
-  createHeaderBasedEmailResolver,
-  getAgentByName,
-  getCurrentAgent,
-  routeAgentEmail,
-  routeAgentRequest,
-  unstable_callable
-} from "./index-CMEWpbHx.js";
+  A as unstable_callable,
+  C as createAddressBasedEmailResolver,
+  D as getCurrentAgent,
+  E as getAgentByName,
+  O as routeAgentEmail,
+  S as callable,
+  T as createHeaderBasedEmailResolver,
+  _ as RPCResponse,
+  a as AgentOptions,
+  b as StreamingResponse,
+  c as ConnectionContext,
+  d as EmailSendOptions,
+  f as MCPServer,
+  g as RPCRequest,
+  h as QueueItem,
+  i as AgentNamespace,
+  k as routeAgentRequest,
+  l as EmailResolver,
+  m as MCPServersState,
+  n as AgentContext,
+  o as CallableMetadata,
+  p as MCPServerMessage,
+  r as AgentEmail,
+  s as Connection,
+  t as Agent,
+  u as EmailRoutingOptions,
+  v as Schedule,
+  w as createCatchAllEmailResolver,
+  x as WSMessage,
+  y as StateUpdateMessage
+} from "./index-Db_90HY3.js";
 export {
   Agent,
   AgentContext,

package/dist/index.js

@@ -1,7 +1,7 @@
-import "./ai-types-UZlfLOYP.js";
-import "./client-DjR-lC16.js";
-import "./client-CZBVDDoO.js";
-import "./do-oauth-client-provider-B2jr6UNq.js";
-import { Agent, StreamingResponse, callable, createAddressBasedEmailResolver, createCatchAllEmailResolver, createHeaderBasedEmailResolver, getAgentByName, getCurrentAgent, routeAgentEmail, routeAgentRequest, unstable_callable } from "./src-L3cHuAag.js";
+import "./ai-types-B3aQaFv3.js";
+import "./client-BfiZ3HQd.js";
+import "./client-DZhjV_XA.js";
+import "./do-oauth-client-provider-CswoD5Lu.js";
+import { a as createCatchAllEmailResolver, c as getCurrentAgent, d as unstable_callable, i as createAddressBasedEmailResolver, l as routeAgentEmail, n as StreamingResponse, o as createHeaderBasedEmailResolver, r as callable, s as getAgentByName, t as Agent, u as routeAgentRequest } from "./src-CwIW7tU2.js";
 
 export { Agent, StreamingResponse, callable, createAddressBasedEmailResolver, createCatchAllEmailResolver, createHeaderBasedEmailResolver, getAgentByName, getCurrentAgent, routeAgentEmail, routeAgentRequest, unstable_callable };

package/dist/mcp/client.d.ts

@@ -1,4 +1,4 @@
-import { MCPClientManager, MCPClientOAuthCallbackConfig, MCPClientOAuthResult, getNamespacedData } from "../client-C-nwz-3N.js";
-import "../mcp-BEwaCsxO.js";
-import "../do-oauth-client-provider-CCwGwnrA.js";
+import { i as getNamespacedData, n as MCPClientOAuthCallbackConfig, r as MCPClientOAuthResult, t as MCPClientManager } from "../client-DnzF-4rM.js";
+import "../mcp-Dw5vDrY8.js";
+import "../do-oauth-client-provider-DGc5pP0l.js";
 export { MCPClientManager, MCPClientOAuthCallbackConfig, MCPClientOAuthResult, getNamespacedData };

package/dist/mcp/client.js

@@ -1,3 +1,3 @@
-import { MCPClientManager, getNamespacedData } from "../client-CZBVDDoO.js";
+import { n as getNamespacedData, t as MCPClientManager } from "../client-DZhjV_XA.js";
 
 export { MCPClientManager, getNamespacedData };

package/dist/mcp/do-oauth-client-provider.d.ts

@@ -1,2 +1,2 @@
-import { AgentsOAuthProvider, DurableObjectOAuthClientProvider } from "../do-oauth-client-provider-CCwGwnrA.js";
+import { n as DurableObjectOAuthClientProvider, t as AgentsOAuthProvider } from "../do-oauth-client-provider-DGc5pP0l.js";
 export { AgentsOAuthProvider, DurableObjectOAuthClientProvider };

package/dist/mcp/do-oauth-client-provider.js

@@ -1,3 +1,3 @@
-import { DurableObjectOAuthClientProvider } from "../do-oauth-client-provider-B2jr6UNq.js";
+import { t as DurableObjectOAuthClientProvider } from "../do-oauth-client-provider-CswoD5Lu.js";
 
 export { DurableObjectOAuthClientProvider };