agents 0.0.0-de168a2 → 0.0.0-df52d4b

package/dist/mcp/index.d.ts

@@ -1,15 +1,40 @@
+import { M as MCPClientManager } from '../client-DgyzBU_8.js';
+export { S as SSEEdgeClientTransport, a as StreamableHTTPEdgeClientTransport } from '../client-DgyzBU_8.js';
 import { DurableObject } from 'cloudflare:workers';
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
 import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
-import { Connection } from 'partyserver';
+import { ElicitResult } from '@modelcontextprotocol/sdk/types.js';
+export { ElicitRequest, ElicitRequestSchema, ElicitResult } from '@modelcontextprotocol/sdk/types.js';
+import { Connection, WSMessage } from 'partyserver';
+import 'zod';
+import '@modelcontextprotocol/sdk/client/index.js';
+import '@modelcontextprotocol/sdk/shared/protocol.js';
+import 'ai';
+import '@modelcontextprotocol/sdk/client/sse.js';
+import '@modelcontextprotocol/sdk/client/streamableHttp.js';
+import './do-oauth-client-provider.js';
+import '@modelcontextprotocol/sdk/client/auth.js';
+import '@modelcontextprotocol/sdk/shared/auth.js';
 
 interface CORSOptions {
     origin?: string;
     methods?: string;
     headers?: string;
     maxAge?: number;
+    exposeHeaders?: string;
 }
+type MaybePromise<T> = T | Promise<T>;
 declare abstract class McpAgent<Env = unknown, State = unknown, Props extends Record<string, unknown> = Record<string, unknown>> extends DurableObject<Env> {
-    #private;
+    private _status;
+    private _transport?;
+    private _transportType;
+    private _requestIdToConnectionId;
+    /**
+     * Since McpAgent's _aren't_ yet real "Agents", let's only expose a couple of the methods
+     * to the outer class: initialState/state/setState/onStateUpdate/sql
+     */
+    private _agent;
+    get mcp(): MCPClientManager;
     protected constructor(ctx: DurableObjectState, env: Env);
     /**
      * Agents API allowlist
@@ -18,27 +43,69 @@ declare abstract class McpAgent<Env = unknown, State = unknown, Props extends Re
     get state(): State;
     sql<T = Record<string, string | number | boolean | null>>(strings: TemplateStringsArray, ...values: (string | number | boolean | null)[]): T[];
     setState(state: State): void;
+    /**
+     * Elicit user input with a message and schema
+     */
+    elicitInput(params: {
+        message: string;
+        requestedSchema: unknown;
+    }): Promise<ElicitResult>;
     onStateUpdate(state: State | undefined, source: Connection | "server"): void;
     onStart(): Promise<void>;
     /**
      * McpAgent API
      */
-    abstract server: McpServer;
+    abstract server: MaybePromise<McpServer | Server>;
     props: Props;
     initRun: boolean;
     abstract init(): Promise<void>;
+    /**
+     * Handle errors that occur during initialization or operation.
+     * Override this method to provide custom error handling.
+     * @param error - The error that occurred
+     * @returns An error response object with status code and message
+     */
+    onError(error: Error): {
+        status: number;
+        message: string;
+    };
     _init(props: Props): Promise<void>;
+    setInitialized(): Promise<void>;
+    isInitialized(): Promise<boolean>;
+    updateProps(props: Props): Promise<void>;
+    private _initialize;
     fetch(request: Request): Promise<Response>;
     getWebSocket(): WebSocket | null;
-    onMCPMessage(sessionId: string, request: Request): Promise<Response>;
+    getWebSocketForResponseID(id: string): WebSocket | null;
+    onMessage(connection: Connection, event: WSMessage): Promise<void>;
+    /**
+     * Wait for elicitation response through storage polling
+     */
+    private _waitForElicitationResponse;
+    /**
+     * Handle elicitation responses   */
+    private _handleElicitationResponse;
+    onSSEMcpMessage(_sessionId: string, messageBody: unknown): Promise<Error | null>;
     webSocketMessage(ws: WebSocket, event: ArrayBuffer | string): Promise<void>;
     webSocketError(ws: WebSocket, error: unknown): Promise<void>;
     webSocketClose(ws: WebSocket, code: number, reason: string, wasClean: boolean): Promise<void>;
-    static mount(path: string, { binding, corsOptions, }?: {
+    static mount(path: string, { binding, corsOptions }?: {
+        binding?: string;
+        corsOptions?: CORSOptions;
+    }): {
+        fetch<Env>(this: void, request: Request, env: Env, ctx: ExecutionContext): Promise<Response>;
+    };
+    static serveSSE(path: string, { binding, corsOptions }?: {
+        binding?: string;
+        corsOptions?: CORSOptions;
+    }): {
+        fetch<Env>(this: void, request: Request, env: Env, ctx: ExecutionContext): Promise<Response>;
+    };
+    static serve(path: string, { binding, corsOptions }?: {
         binding?: string;
         corsOptions?: CORSOptions;
     }): {
-        fetch: (request: Request, env: Record<string, DurableObjectNamespace<McpAgent>>, ctx: ExecutionContext) => Promise<Response | undefined>;
+        fetch<Env>(this: void, request: Request, env: Env, ctx: ExecutionContext): Promise<Response>;
     };
 }