agents 0.0.0-d045755 → 0.0.0-d08612f

package/dist/client-DfIOsabL.d.ts

@@ -0,0 +1,834 @@
+import { t as MCPObservabilityEvent } from "./mcp-CzbSsLfc.js";
+import { t as AgentsOAuthProvider } from "./do-oauth-client-provider-B-ryFIPr.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,
+  CompatibilityCallToolResultSchema,
+  ElicitRequest,
+  ElicitResult,
+  GetPromptRequest,
+  Prompt,
+  ReadResourceRequest,
+  Resource,
+  ResourceTemplate,
+  ServerCapabilities,
+  Tool
+} from "@modelcontextprotocol/sdk/types.js";
+import * as ai0 from "ai";
+import { ToolSet } from "ai";
+import { RequestOptions } from "@modelcontextprotocol/sdk/shared/protocol.js";
+
+//#region src/core/events.d.ts
+interface Disposable {
+  dispose(): void;
+}
+type Event<T> = (listener: (e: T) => void) => Disposable;
+declare class Emitter<T> implements Disposable {
+  private _listeners;
+  readonly event: Event<T>;
+  fire(data: T): void;
+  dispose(): void;
+}
+//#endregion
+//#region src/mcp/types.d.ts
+type MaybePromise<T> = T | Promise<T>;
+type BaseTransportType = "sse" | "streamable-http";
+type TransportType = BaseTransportType | "auto";
+interface CORSOptions {
+  origin?: string;
+  methods?: string;
+  headers?: string;
+  maxAge?: number;
+  exposeHeaders?: string;
+}
+interface ServeOptions {
+  binding?: string;
+  corsOptions?: CORSOptions;
+  transport?: BaseTransportType;
+  jurisdiction?: DurableObjectJurisdiction;
+}
+//#endregion
+//#region src/mcp/client-connection.d.ts
+/**
+ * Connection state machine for MCP client connections.
+ *
+ * State transitions:
+ * - Non-OAuth: init() → CONNECTING → DISCOVERING → READY
+ * - OAuth: init() → AUTHENTICATING → (callback) → CONNECTING → DISCOVERING → READY
+ * - Any state can transition to FAILED on error
+ */
+declare const MCPConnectionState: {
+  /** Waiting for OAuth authorization to complete */
+  readonly AUTHENTICATING: "authenticating";
+  /** Establishing transport connection to MCP server */
+  readonly CONNECTING: "connecting";
+  /** Transport connection established */
+  readonly CONNECTED: "connected";
+  /** Discovering server capabilities (tools, resources, prompts) */
+  readonly DISCOVERING: "discovering";
+  /** Fully connected and ready to use */
+  readonly READY: "ready";
+  /** Connection failed at some point */
+  readonly FAILED: "failed";
+};
+/**
+ * Connection state type for MCP client connections.
+ */
+type MCPConnectionState =
+  (typeof MCPConnectionState)[keyof typeof MCPConnectionState];
+type MCPTransportOptions = (
+  | SSEClientTransportOptions
+  | StreamableHTTPClientTransportOptions
+) & {
+  authProvider?: AgentsOAuthProvider;
+  type?: TransportType;
+};
+/**
+ * Result of a discovery operation.
+ * success indicates whether discovery completed successfully.
+ * error is present when success is false.
+ */
+type MCPDiscoveryResult = {
+  success: boolean;
+  error?: string;
+};
+declare class MCPClientConnection {
+  url: URL;
+  options: {
+    transport: MCPTransportOptions;
+    client: ConstructorParameters<typeof Client>[1];
+  };
+  client: Client;
+  connectionState: MCPConnectionState;
+  lastConnectedTransport: BaseTransportType | undefined;
+  instructions?: string;
+  tools: Tool[];
+  prompts: Prompt[];
+  resources: Resource[];
+  resourceTemplates: ResourceTemplate[];
+  serverCapabilities: ServerCapabilities | undefined;
+  /** Tracks in-flight discovery to allow cancellation */
+  private _discoveryAbortController;
+  private readonly _onObservabilityEvent;
+  readonly onObservabilityEvent: Event<MCPObservabilityEvent>;
+  constructor(
+    url: URL,
+    info: ConstructorParameters<typeof Client>[0],
+    options?: {
+      transport: MCPTransportOptions;
+      client: ConstructorParameters<typeof Client>[1];
+    }
+  );
+  /**
+   * Initialize a client connection, if authentication is required, the connection will be in the AUTHENTICATING state
+   * Sets connection state based on the result and emits observability events
+   *
+   * @returns Error message if connection failed, undefined otherwise
+   */
+  init(): Promise<string | undefined>;
+  /**
+   * Finish OAuth by probing transports based on configured type.
+   * - Explicit: finish on that transport
+   * - Auto: try streamable-http, then sse on 404/405/Not Implemented
+   */
+  private finishAuthProbe;
+  /**
+   * Complete OAuth authorization
+   */
+  completeAuthorization(code: string): Promise<void>;
+  /**
+   * Discover server capabilities and register tools, resources, prompts, and templates.
+   * This method does the work but does not manage connection state - that's handled by discover().
+   */
+  discoverAndRegister(): Promise<void>;
+  /**
+   * Discover server capabilities with timeout and cancellation support.
+   * If called while a previous discovery is in-flight, the previous discovery will be aborted.
+   *
+   * @param options Optional configuration
+   * @param options.timeoutMs Timeout in milliseconds (default: 15000)
+   * @returns Result indicating success/failure with optional error message
+   */
+  discover(options?: { timeoutMs?: number }): Promise<MCPDiscoveryResult>;
+  /**
+   * Cancel any in-flight discovery operation.
+   * Called when closing the connection.
+   */
+  cancelDiscovery(): void;
+  /**
+   * Notification handler registration for tools
+   * Should only be called if serverCapabilities.tools exists
+   */
+  registerTools(): Promise<Tool[]>;
+  /**
+   * Notification handler registration for resources
+   * Should only be called if serverCapabilities.resources exists
+   */
+  registerResources(): Promise<Resource[]>;
+  /**
+   * Notification handler registration for prompts
+   * Should only be called if serverCapabilities.prompts exists
+   */
+  registerPrompts(): Promise<Prompt[]>;
+  registerResourceTemplates(): Promise<ResourceTemplate[]>;
+  fetchTools(): Promise<
+    {
+      inputSchema: {
+        [x: string]: unknown;
+        type: "object";
+        properties?:
+          | {
+              [x: string]: object;
+            }
+          | undefined;
+        required?: string[] | undefined;
+      };
+      name: string;
+      description?: string | undefined;
+      outputSchema?:
+        | {
+            [x: string]: unknown;
+            type: "object";
+            properties?:
+              | {
+                  [x: string]: object;
+                }
+              | undefined;
+            required?: string[] | undefined;
+          }
+        | undefined;
+      annotations?:
+        | {
+            title?: string | undefined;
+            readOnlyHint?: boolean | undefined;
+            destructiveHint?: boolean | undefined;
+            idempotentHint?: boolean | undefined;
+            openWorldHint?: boolean | undefined;
+          }
+        | undefined;
+      _meta?:
+        | {
+            [x: string]: unknown;
+          }
+        | undefined;
+      icons?:
+        | {
+            src: string;
+            mimeType?: string | undefined;
+            sizes?: string[] | undefined;
+          }[]
+        | undefined;
+      title?: string | undefined;
+    }[]
+  >;
+  fetchResources(): Promise<
+    {
+      uri: string;
+      name: string;
+      description?: string | undefined;
+      mimeType?: string | undefined;
+      _meta?:
+        | {
+            [x: string]: unknown;
+          }
+        | undefined;
+      icons?:
+        | {
+            src: string;
+            mimeType?: string | undefined;
+            sizes?: string[] | undefined;
+          }[]
+        | undefined;
+      title?: string | undefined;
+    }[]
+  >;
+  fetchPrompts(): Promise<
+    {
+      name: string;
+      description?: string | undefined;
+      arguments?:
+        | {
+            name: string;
+            description?: string | undefined;
+            required?: boolean | undefined;
+          }[]
+        | undefined;
+      _meta?:
+        | {
+            [x: string]: unknown;
+          }
+        | undefined;
+      icons?:
+        | {
+            src: string;
+            mimeType?: string | undefined;
+            sizes?: string[] | undefined;
+          }[]
+        | undefined;
+      title?: string | undefined;
+    }[]
+  >;
+  fetchResourceTemplates(): Promise<
+    {
+      uriTemplate: string;
+      name: string;
+      description?: string | undefined;
+      mimeType?: string | undefined;
+      _meta?:
+        | {
+            [x: string]: unknown;
+          }
+        | undefined;
+      icons?:
+        | {
+            src: string;
+            mimeType?: string | undefined;
+            sizes?: string[] | undefined;
+          }[]
+        | undefined;
+      title?: string | undefined;
+    }[]
+  >;
+  /**
+   * Handle elicitation request from server
+   * Automatically uses the Agent's built-in elicitation handling if available
+   */
+  handleElicitationRequest(_request: ElicitRequest): Promise<ElicitResult>;
+  /**
+   * Get the transport for the client
+   * @param transportType - The transport type to get
+   * @returns The transport for the client
+   */
+  getTransport(
+    transportType: BaseTransportType
+  ): SSEClientTransport | StreamableHTTPClientTransport;
+  private tryConnect;
+  private _capabilityErrorHandler;
+}
+//#endregion
+//#region src/mcp/client-storage.d.ts
+/**
+ * Represents a row in the cf_agents_mcp_servers table
+ */
+type MCPServerRow = {
+  id: string;
+  name: string;
+  server_url: string;
+  client_id: string | null;
+  auth_url: string | null;
+  callback_url: string;
+  server_options: string | null;
+};
+//#endregion
+//#region src/mcp/client.d.ts
+/**
+ * Options that can be stored in the server_options column
+ * This is what gets JSON.stringify'd and stored in the database
+ */
+type MCPServerOptions = {
+  client?: ConstructorParameters<typeof Client>[1];
+  transport?: {
+    headers?: HeadersInit;
+    type?: TransportType;
+  };
+};
+/**
+ * Options for registering an MCP server
+ */
+type RegisterServerOptions = {
+  url: string;
+  name: string;
+  callbackUrl: string;
+  client?: ConstructorParameters<typeof Client>[1];
+  transport?: MCPTransportOptions;
+  authUrl?: string;
+  clientId?: string;
+};
+/**
+ * Result of attempting to connect to an MCP server.
+ * Discriminated union ensures error is present only on failure.
+ */
+type MCPConnectionResult =
+  | {
+      state: typeof MCPConnectionState.FAILED;
+      error: string;
+    }
+  | {
+      state: typeof MCPConnectionState.AUTHENTICATING;
+      authUrl: string;
+      clientId?: string;
+    }
+  | {
+      state: typeof MCPConnectionState.CONNECTED;
+    };
+/**
+ * Result of discovering server capabilities.
+ * success indicates whether discovery completed successfully.
+ * state is the current connection state at time of return.
+ * error is present when success is false.
+ */
+type MCPDiscoverResult = {
+  success: boolean;
+  state: MCPConnectionState;
+  error?: string;
+};
+type MCPClientOAuthCallbackConfig = {
+  successRedirect?: string;
+  errorRedirect?: string;
+  customHandler?: (result: MCPClientOAuthResult) => Response;
+};
+type MCPClientOAuthResult = {
+  serverId: string;
+  authSuccess: boolean;
+  authError?: string;
+};
+type MCPClientManagerOptions = {
+  storage: DurableObjectStorage;
+};
+/**
+ * Utility class that aggregates multiple MCP clients into one
+ */
+declare class MCPClientManager {
+  private _name;
+  private _version;
+  mcpConnections: Record<string, MCPClientConnection>;
+  private _didWarnAboutUnstableGetAITools;
+  private _oauthCallbackConfig?;
+  private _connectionDisposables;
+  private _storage;
+  private _isRestored;
+  /** @internal Protected for testing purposes. */
+  protected readonly _onObservabilityEvent: Emitter<MCPObservabilityEvent>;
+  readonly onObservabilityEvent: Event<MCPObservabilityEvent>;
+  private readonly _onServerStateChanged;
+  /**
+   * Event that fires whenever any MCP server state changes (registered, connected, removed, etc.)
+   * This is useful for broadcasting server state to clients.
+   */
+  readonly onServerStateChanged: Event<void>;
+  /**
+   * @param _name Name of the MCP client
+   * @param _version Version of the MCP Client
+   * @param options Storage adapter for persisting MCP server state
+   */
+  constructor(
+    _name: string,
+    _version: string,
+    options: MCPClientManagerOptions
+  );
+  private sql;
+  private saveServerToStorage;
+  private removeServerFromStorage;
+  private getServersFromStorage;
+  private clearServerAuthUrl;
+  jsonSchema: typeof ai0.jsonSchema | undefined;
+  /**
+   * Create an auth provider for a server
+   * @internal
+   */
+  private createAuthProvider;
+  /**
+   * Restore MCP server connections from storage
+   * This method is called on Agent initialization to restore previously connected servers
+   *
+   * @param clientName Name to use for OAuth client (typically the agent instance name)
+   */
+  restoreConnectionsFromStorage(clientName: string): Promise<void>;
+  /**
+   * Internal method to restore a single server connection and discovery
+   */
+  private _restoreServer;
+  /**
+   * Connect to and register an MCP server
+   *
+   * @deprecated This method is maintained for backward compatibility.
+   * For new code, use registerServer() and connectToServer() separately.
+   *
+   * @param url Server URL
+   * @param options Connection options
+   * @returns Object with server ID, auth URL (if OAuth), and client ID (if OAuth)
+   */
+  connect(
+    url: string,
+    options?: {
+      reconnect?: {
+        id: string;
+        oauthClientId?: string;
+        oauthCode?: string;
+      };
+      transport?: MCPTransportOptions;
+      client?: ConstructorParameters<typeof Client>[1];
+    }
+  ): Promise<{
+    id: string;
+    authUrl?: string;
+    clientId?: string;
+  }>;
+  /**
+   * Create an in-memory connection object and set up observability
+   * Does NOT save to storage - use registerServer() for that
+   * @returns The connection object (existing or newly created)
+   */
+  private createConnection;
+  /**
+   * Register an MCP server connection without connecting
+   * Creates the connection object, sets up observability, and saves to storage
+   *
+   * @param id Server ID
+   * @param options Registration options including URL, name, callback URL, and connection config
+   * @returns Server ID
+   */
+  registerServer(id: string, options: RegisterServerOptions): Promise<string>;
+  /**
+   * Connect to an already registered MCP server and initialize the connection.
+   *
+   * For OAuth servers, returns `{ state: "authenticating", authUrl, clientId? }`.
+   * The user must complete the OAuth flow via the authUrl, which triggers a
+   * callback handled by `handleCallbackRequest()`.
+   *
+   * For non-OAuth servers, establishes the transport connection and returns
+   * `{ state: "connected" }`. Call `discoverIfConnected()` afterwards to
+   * discover capabilities and transition to "ready" state.
+   *
+   * @param id Server ID (must be registered first via registerServer())
+   * @returns Connection result with current state and OAuth info (if applicable)
+   */
+  connectToServer(id: string): Promise<MCPConnectionResult>;
+  private extractServerIdFromState;
+  isCallbackRequest(req: Request): boolean;
+  handleCallbackRequest(req: Request): Promise<
+    | {
+        serverId: string;
+        authSuccess: boolean;
+        authError: string;
+      }
+    | {
+        serverId: string;
+        authSuccess: boolean;
+        authError?: undefined;
+      }
+  >;
+  /**
+   * Discover server capabilities if connection is in CONNECTED or READY state.
+   * Transitions to DISCOVERING then READY (or CONNECTED on error).
+   * Can be called to refresh server capabilities (e.g., from a UI refresh button).
+   *
+   * If called while a previous discovery is in-flight for the same server,
+   * the previous discovery will be aborted.
+   *
+   * @param serverId The server ID to discover
+   * @param options Optional configuration
+   * @param options.timeoutMs Timeout in milliseconds (default: 30000)
+   * @returns Result with current state and optional error, or undefined if connection not found
+   */
+  discoverIfConnected(
+    serverId: string,
+    options?: {
+      timeoutMs?: number;
+    }
+  ): Promise<MCPDiscoverResult | undefined>;
+  /**
+   * Establish connection in the background after OAuth completion
+   * This method connects to the server and discovers its capabilities
+   * @param serverId The server ID to establish connection for
+   */
+  establishConnection(serverId: string): Promise<void>;
+  /**
+   * Configure OAuth callback handling
+   * @param config OAuth callback configuration
+   */
+  configureOAuthCallback(config: MCPClientOAuthCallbackConfig): void;
+  /**
+   * Get the current OAuth callback configuration
+   * @returns The current OAuth callback configuration
+   */
+  getOAuthCallbackConfig(): MCPClientOAuthCallbackConfig | undefined;
+  /**
+   * @returns namespaced list of tools
+   */
+  listTools(): NamespacedData["tools"];
+  /**
+   * Lazy-loads the jsonSchema function from the AI SDK.
+   *
+   * This defers importing the "ai" package until it's actually needed, which helps reduce
+   * initial bundle size and startup time. The jsonSchema function is required for converting
+   * MCP tools into AI SDK tool definitions via getAITools().
+   *
+   * @internal This method is for internal use only. It's automatically called before operations
+   * that need jsonSchema (like getAITools() or OAuth flows). External consumers should not need
+   * to call this directly.
+   */
+  ensureJsonSchema(): Promise<void>;
+  /**
+   * @returns a set of tools that you can use with the AI SDK
+   */
+  getAITools(): ToolSet;
+  /**
+   * @deprecated this has been renamed to getAITools(), and unstable_getAITools will be removed in the next major version
+   * @returns a set of tools that you can use with the AI SDK
+   */
+  unstable_getAITools(): ToolSet;
+  /**
+   * Closes all active in-memory connections to MCP servers.
+   *
+   * Note: This only closes the transport connections - it does NOT remove
+   * servers from storage. Servers will still be listed and their callback
+   * URLs will still match incoming OAuth requests.
+   *
+   * Use removeServer() instead if you want to fully clean up a server
+   * (closes connection AND removes from storage).
+   */
+  closeAllConnections(): Promise<void>;
+  /**
+   * Closes a connection to an MCP server
+   * @param id The id of the connection to close
+   */
+  closeConnection(id: string): Promise<void>;
+  /**
+   * Remove an MCP server - closes connection if active and removes from storage.
+   */
+  removeServer(serverId: string): Promise<void>;
+  /**
+   * List all MCP servers from storage
+   */
+  listServers(): MCPServerRow[];
+  /**
+   * Dispose the manager and all resources.
+   */
+  dispose(): Promise<void>;
+  /**
+   * @returns namespaced list of prompts
+   */
+  listPrompts(): NamespacedData["prompts"];
+  /**
+   * @returns namespaced list of tools
+   */
+  listResources(): NamespacedData["resources"];
+  /**
+   * @returns namespaced list of resource templates
+   */
+  listResourceTemplates(): NamespacedData["resourceTemplates"];
+  /**
+   * Namespaced version of callTool
+   */
+  callTool(
+    params: CallToolRequest["params"] & {
+      serverId: string;
+    },
+    resultSchema?:
+      | typeof CallToolResultSchema
+      | typeof CompatibilityCallToolResultSchema,
+    options?: RequestOptions
+  ): Promise<
+    | {
+        [x: string]: unknown;
+        content: (
+          | {
+              type: "text";
+              text: string;
+              _meta?: Record<string, unknown> | undefined;
+            }
+          | {
+              type: "image";
+              data: string;
+              mimeType: string;
+              _meta?: Record<string, unknown> | undefined;
+            }
+          | {
+              type: "audio";
+              data: string;
+              mimeType: string;
+              _meta?: Record<string, unknown> | undefined;
+            }
+          | {
+              type: "resource";
+              resource:
+                | {
+                    uri: string;
+                    text: string;
+                    mimeType?: string | undefined;
+                    _meta?: Record<string, unknown> | undefined;
+                  }
+                | {
+                    uri: string;
+                    blob: string;
+                    mimeType?: string | undefined;
+                    _meta?: Record<string, unknown> | undefined;
+                  };
+              _meta?: Record<string, unknown> | undefined;
+            }
+          | {
+              uri: string;
+              name: string;
+              type: "resource_link";
+              description?: string | undefined;
+              mimeType?: string | undefined;
+              _meta?:
+                | {
+                    [x: string]: unknown;
+                  }
+                | undefined;
+              icons?:
+                | {
+                    src: string;
+                    mimeType?: string | undefined;
+                    sizes?: string[] | undefined;
+                  }[]
+                | undefined;
+              title?: string | undefined;
+            }
+        )[];
+        _meta?: Record<string, unknown> | undefined;
+        structuredContent?: Record<string, unknown> | undefined;
+        isError?: boolean | undefined;
+      }
+    | {
+        [x: string]: unknown;
+        toolResult: unknown;
+        _meta?: Record<string, unknown> | undefined;
+      }
+  >;
+  /**
+   * Namespaced version of readResource
+   */
+  readResource(
+    params: ReadResourceRequest["params"] & {
+      serverId: string;
+    },
+    options: RequestOptions
+  ): Promise<{
+    [x: string]: unknown;
+    contents: (
+      | {
+          uri: string;
+          text: string;
+          mimeType?: string | undefined;
+          _meta?: Record<string, unknown> | undefined;
+        }
+      | {
+          uri: string;
+          blob: string;
+          mimeType?: string | undefined;
+          _meta?: Record<string, unknown> | undefined;
+        }
+    )[];
+    _meta?: Record<string, unknown> | undefined;
+  }>;
+  /**
+   * Namespaced version of getPrompt
+   */
+  getPrompt(
+    params: GetPromptRequest["params"] & {
+      serverId: string;
+    },
+    options: RequestOptions
+  ): Promise<{
+    [x: string]: unknown;
+    messages: {
+      role: "user" | "assistant";
+      content:
+        | {
+            type: "text";
+            text: string;
+            _meta?: Record<string, unknown> | undefined;
+          }
+        | {
+            type: "image";
+            data: string;
+            mimeType: string;
+            _meta?: Record<string, unknown> | undefined;
+          }
+        | {
+            type: "audio";
+            data: string;
+            mimeType: string;
+            _meta?: Record<string, unknown> | undefined;
+          }
+        | {
+            type: "resource";
+            resource:
+              | {
+                  uri: string;
+                  text: string;
+                  mimeType?: string | undefined;
+                  _meta?: Record<string, unknown> | undefined;
+                }
+              | {
+                  uri: string;
+                  blob: string;
+                  mimeType?: string | undefined;
+                  _meta?: Record<string, unknown> | undefined;
+                };
+            _meta?: Record<string, unknown> | undefined;
+          }
+        | {
+            uri: string;
+            name: string;
+            type: "resource_link";
+            description?: string | undefined;
+            mimeType?: string | undefined;
+            _meta?:
+              | {
+                  [x: string]: unknown;
+                }
+              | undefined;
+            icons?:
+              | {
+                  src: string;
+                  mimeType?: string | undefined;
+                  sizes?: string[] | undefined;
+                }[]
+              | undefined;
+            title?: string | undefined;
+          };
+    }[];
+    _meta?: Record<string, unknown> | undefined;
+    description?: string | undefined;
+  }>;
+}
+type NamespacedData = {
+  tools: (Tool & {
+    serverId: string;
+  })[];
+  prompts: (Prompt & {
+    serverId: string;
+  })[];
+  resources: (Resource & {
+    serverId: string;
+  })[];
+  resourceTemplates: (ResourceTemplate & {
+    serverId: string;
+  })[];
+};
+declare function getNamespacedData<T extends keyof NamespacedData>(
+  mcpClients: Record<string, MCPClientConnection>,
+  type: T
+): NamespacedData[T];
+//#endregion
+export {
+  MCPConnectionResult as a,
+  RegisterServerOptions as c,
+  BaseTransportType as d,
+  CORSOptions as f,
+  TransportType as h,
+  MCPClientOAuthResult as i,
+  getNamespacedData as l,
+  ServeOptions as m,
+  MCPClientManagerOptions as n,
+  MCPDiscoverResult as o,
+  MaybePromise as p,
+  MCPClientOAuthCallbackConfig as r,
+  MCPServerOptions as s,
+  MCPClientManager as t,
+  MCPConnectionState as u
+};
+//# sourceMappingURL=client-DfIOsabL.d.ts.map