@tambo-ai/react 0.54.1 → 0.56.0

package/dist/mcp/mcp-client.d.ts

@@ -1,8 +1,33 @@
+import { type OAuthClientProvider } from "@modelcontextprotocol/sdk/client/auth.js";
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { CreateMessageRequest, CreateMessageResult, ElicitRequest, ElicitResult } from "@modelcontextprotocol/sdk/types.js";
 import { JSONSchema7 } from "json-schema";
 export declare enum MCPTransport {
     SSE = "sse",
     HTTP = "http"
 }
+/**
+ * Handlers for MCP requests - these are only used if the server supports the corresponding capabilities
+ * @param elicitation - Handler for elicitation requests
+ * @param sampling - Handler for sampling requests
+ * @example
+ * ```typescript
+ * const mcp = await MCPClient.create(
+ *     'https://api.example.com/mcp',
+ *     MCPTransport.HTTP,
+ *     {},
+ *     undefined,
+ *     undefined,
+ *     {
+ *       elicitation: (e: ElicitRequest) => Promise.resolve({...}),
+ *     },
+ * });
+ * ```
+ */
+interface MCPHandlers {
+    elicitation?: (e: ElicitRequest) => Promise<ElicitResult>;
+    sampling?: (e: CreateMessageRequest) => Promise<CreateMessageResult>;
+}
 /**
  * A client for interacting with MCP (Model Context Protocol) servers.
  * Provides a simple interface for listing and calling tools exposed by the server.
@@ -17,8 +42,11 @@ export declare class MCPClient {
     private client;
     private transport;
     private transportType;
+    sessionId?: string;
     private endpoint;
     private headers;
+    private authProvider?;
+    private handlers;
     /**
      * Tracks an in-flight reconnect so concurrent triggers coalesce
      * (single-flight). When set, additional calls to `reconnect()` or
@@ -54,6 +82,7 @@ export declare class MCPClient {
     static readonly BACKOFF_MULTIPLIER = 2;
     static readonly BACKOFF_MAX_MS = 30000;
     static readonly BACKOFF_JITTER_RATIO = 0.2;
+    elicitation: EventTarget;
     /**
      * Private constructor to enforce using the static create method.
      * @param endpoint - The URL of the MCP server to connect to
@@ -62,16 +91,19 @@ export declare class MCPClient {
      */
     private constructor();
     /**
-     * Creates and initializes a new MCPClient instance.
-     * This is the recommended way to create an MCPClient as it handles both
-     * instantiation and connection setup.
+     * Creates and initializes a new MCPClient instance. This is the recommended
+     * way to create an MCPClient as it handles both instantiation and connection
+     * setup.
      * @param endpoint - The URL of the MCP server to connect to
      * @param transportType - The transport type to use for the MCP client. Defaults to HTTP.
      * @param headers - Optional custom headers to include in requests
+     * @param authProvider - Optional auth provider to use for authentication
+     * @param sessionId - Optional session id to use for the MCP client - if not
+     *   provided, a new session will be created
      * @returns A connected MCPClient instance ready for use
-     * @throws Will throw an error if connection fails
+     * @throws {Error} Will throw an error if connection fails
      */
-    static create(endpoint: string, transportType?: MCPTransport, headers?: Record<string, string>): Promise<MCPClient>;
+    static create(endpoint: string, transportType: MCPTransport | undefined, headers: Record<string, string> | undefined, authProvider: OAuthClientProvider | undefined, sessionId: string | undefined, handlers?: MCPHandlers): Promise<MCPClient>;
     /**
      * Reconnects to the MCP server, optionally retaining the same session ID.
      *
@@ -89,6 +121,7 @@ export declare class MCPClient {
      * @param newSession - Whether to create a new session (true) or reuse existing session ID (false)
      * @param reportErrorOnClose - Whether to report errors when closing the client
      * Note that only StreamableHTTPClientTransport supports session IDs.
+     * @returns A promise that resolves when the reconnect is complete
      */
     reconnect(newSession?: boolean, reportErrorOnClose?: boolean): Promise<void>;
     /**
@@ -99,6 +132,7 @@ export declare class MCPClient {
     private onclose;
     /**
      * Compute the next backoff delay with symmetric jitter.
+     * @returns The next backoff delay in milliseconds
      */
     private computeBackoffDelayMs;
     /**
@@ -107,228 +141,77 @@ export declare class MCPClient {
      */
     private scheduleAutoReconnect;
     private initializeTransport;
+    /**
+     * Initializes the MCP client with the appropriate capabilities and handlers
+     * @returns The initialized MCP client
+     */
     private initializeClient;
     /**
      * Retrieves a complete list of all available tools from the MCP server.
      * Handles pagination automatically by following cursors until all tools are fetched.
      * @returns A complete list of all available tools and their descriptions
-     * @throws Will throw an error if any server request fails during pagination
+     * @throws {Error} Will throw an error if any server request fails during pagination
      */
     listTools(): Promise<MCPToolSpec[]>;
+    getServerCapabilities(): {
+        [x: string]: unknown;
+        tools?: {
+            [x: string]: unknown;
+            listChanged?: boolean | undefined;
+        } | undefined;
+        experimental?: {
+            [x: string]: unknown;
+        } | undefined;
+        logging?: {
+            [x: string]: unknown;
+        } | undefined;
+        completions?: {
+            [x: string]: unknown;
+        } | undefined;
+        prompts?: {
+            [x: string]: unknown;
+            listChanged?: boolean | undefined;
+        } | undefined;
+        resources?: {
+            [x: string]: unknown;
+            listChanged?: boolean | undefined;
+            subscribe?: boolean | undefined;
+        } | undefined;
+    } | undefined;
+    getServerVersion(): {
+        [x: string]: unknown;
+        name: string;
+        version: string;
+        title?: string | undefined;
+        websiteUrl?: string | undefined;
+        icons?: {
+            [x: string]: unknown;
+            src: string;
+            mimeType?: string | undefined;
+            sizes?: string[] | undefined;
+        }[] | undefined;
+    } | undefined;
+    getInstructions(): string | undefined;
     /**
      * Calls a specific tool on the MCP server with the provided arguments.
      * @param name - The name of the tool to call
      * @param args - Arguments to pass to the tool, must match the tool's expected schema
      * @returns The result from the tool execution
-     * @throws Will throw an error if the tool call fails or if arguments are invalid
+     * @throws {Error} Will throw an error if the tool call fails or if arguments are invalid
      */
-    callTool(name: string, args: Record<string, unknown>): Promise<import("zod").objectOutputType<{
-        _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-    } & {
-        content: import("zod").ZodDefault<import("zod").ZodArray<import("zod").ZodUnion<[import("zod").ZodObject<{
-            type: import("zod").ZodLiteral<"text">;
-            text: import("zod").ZodString;
-            _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-        }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
-            type: import("zod").ZodLiteral<"text">;
-            text: import("zod").ZodString;
-            _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-        }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
-            type: import("zod").ZodLiteral<"text">;
-            text: import("zod").ZodString;
-            _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-        }, import("zod").ZodTypeAny, "passthrough">>, import("zod").ZodObject<{
-            type: import("zod").ZodLiteral<"image">;
-            data: import("zod").ZodEffects<import("zod").ZodString, string, string>;
-            mimeType: import("zod").ZodString;
-            _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-        }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
-            type: import("zod").ZodLiteral<"image">;
-            data: import("zod").ZodEffects<import("zod").ZodString, string, string>;
-            mimeType: import("zod").ZodString;
-            _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-        }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
-            type: import("zod").ZodLiteral<"image">;
-            data: import("zod").ZodEffects<import("zod").ZodString, string, string>;
-            mimeType: import("zod").ZodString;
-            _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-        }, import("zod").ZodTypeAny, "passthrough">>, import("zod").ZodObject<{
-            type: import("zod").ZodLiteral<"audio">;
-            data: import("zod").ZodEffects<import("zod").ZodString, string, string>;
-            mimeType: import("zod").ZodString;
-            _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-        }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
-            type: import("zod").ZodLiteral<"audio">;
-            data: import("zod").ZodEffects<import("zod").ZodString, string, string>;
-            mimeType: import("zod").ZodString;
-            _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-        }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
-            type: import("zod").ZodLiteral<"audio">;
-            data: import("zod").ZodEffects<import("zod").ZodString, string, string>;
-            mimeType: import("zod").ZodString;
-            _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-        }, import("zod").ZodTypeAny, "passthrough">>, import("zod").ZodObject<import("zod").objectUtil.extendShape<import("zod").objectUtil.extendShape<{
-            name: import("zod").ZodString;
-            title: import("zod").ZodOptional<import("zod").ZodString>;
-        }, {
-            uri: import("zod").ZodString;
-            description: import("zod").ZodOptional<import("zod").ZodString>;
-            mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-            _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-        }>, {
-            type: import("zod").ZodLiteral<"resource_link">;
-        }>, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<import("zod").objectUtil.extendShape<import("zod").objectUtil.extendShape<{
-            name: import("zod").ZodString;
-            title: import("zod").ZodOptional<import("zod").ZodString>;
-        }, {
-            uri: import("zod").ZodString;
-            description: import("zod").ZodOptional<import("zod").ZodString>;
-            mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-            _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-        }>, {
-            type: import("zod").ZodLiteral<"resource_link">;
-        }>, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<import("zod").objectUtil.extendShape<import("zod").objectUtil.extendShape<{
-            name: import("zod").ZodString;
-            title: import("zod").ZodOptional<import("zod").ZodString>;
-        }, {
-            uri: import("zod").ZodString;
-            description: import("zod").ZodOptional<import("zod").ZodString>;
-            mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-            _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-        }>, {
-            type: import("zod").ZodLiteral<"resource_link">;
-        }>, import("zod").ZodTypeAny, "passthrough">>, import("zod").ZodObject<{
-            type: import("zod").ZodLiteral<"resource">;
-            resource: import("zod").ZodUnion<[import("zod").ZodObject<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                text: import("zod").ZodString;
-            }>, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                text: import("zod").ZodString;
-            }>, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                text: import("zod").ZodString;
-            }>, import("zod").ZodTypeAny, "passthrough">>, import("zod").ZodObject<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                blob: import("zod").ZodEffects<import("zod").ZodString, string, string>;
-            }>, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                blob: import("zod").ZodEffects<import("zod").ZodString, string, string>;
-            }>, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                blob: import("zod").ZodEffects<import("zod").ZodString, string, string>;
-            }>, import("zod").ZodTypeAny, "passthrough">>]>;
-            _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-        }, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{
-            type: import("zod").ZodLiteral<"resource">;
-            resource: import("zod").ZodUnion<[import("zod").ZodObject<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                text: import("zod").ZodString;
-            }>, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                text: import("zod").ZodString;
-            }>, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                text: import("zod").ZodString;
-            }>, import("zod").ZodTypeAny, "passthrough">>, import("zod").ZodObject<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                blob: import("zod").ZodEffects<import("zod").ZodString, string, string>;
-            }>, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                blob: import("zod").ZodEffects<import("zod").ZodString, string, string>;
-            }>, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                blob: import("zod").ZodEffects<import("zod").ZodString, string, string>;
-            }>, import("zod").ZodTypeAny, "passthrough">>]>;
-            _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-        }, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{
-            type: import("zod").ZodLiteral<"resource">;
-            resource: import("zod").ZodUnion<[import("zod").ZodObject<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                text: import("zod").ZodString;
-            }>, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                text: import("zod").ZodString;
-            }>, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                text: import("zod").ZodString;
-            }>, import("zod").ZodTypeAny, "passthrough">>, import("zod").ZodObject<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                blob: import("zod").ZodEffects<import("zod").ZodString, string, string>;
-            }>, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                blob: import("zod").ZodEffects<import("zod").ZodString, string, string>;
-            }>, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<import("zod").objectUtil.extendShape<{
-                uri: import("zod").ZodString;
-                mimeType: import("zod").ZodOptional<import("zod").ZodString>;
-                _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-            }, {
-                blob: import("zod").ZodEffects<import("zod").ZodString, string, string>;
-            }>, import("zod").ZodTypeAny, "passthrough">>]>;
-            _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-        }, import("zod").ZodTypeAny, "passthrough">>]>, "many">>;
-        structuredContent: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-        isError: import("zod").ZodOptional<import("zod").ZodBoolean>;
-    }, import("zod").ZodTypeAny, "passthrough"> | import("zod").objectOutputType<{
-        _meta: import("zod").ZodOptional<import("zod").ZodObject<{}, "passthrough", import("zod").ZodTypeAny, import("zod").objectOutputType<{}, import("zod").ZodTypeAny, "passthrough">, import("zod").objectInputType<{}, import("zod").ZodTypeAny, "passthrough">>>;
-    } & {
-        toolResult: import("zod").ZodUnknown;
-    }, import("zod").ZodTypeAny, "passthrough">>;
+    callTool(name: string, args: Record<string, unknown>): Promise<MCPToolCallResult>;
+    updateElicitationHandler(handler: ((e: ElicitRequest) => Promise<ElicitResult>) | undefined): void;
+    updateSamplingHandler(handler: ((e: CreateMessageRequest) => Promise<CreateMessageResult>) | undefined): void;
 }
+/**
+ * The result of a tool call.
+ * This is the same as the result of a tool call in the OpenAI SDK, but is reified here
+ */
+export type MCPToolCallResult = Awaited<ReturnType<typeof Client.prototype.callTool>>;
 export interface MCPToolSpec {
     name: string;
     description?: string;
     inputSchema?: JSONSchema7;
 }
+export {};
 //# sourceMappingURL=mcp-client.d.ts.map

package/dist/mcp/mcp-client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"mcp-client.d.ts","sourceRoot":"","sources":["../../src/mcp/mcp-client.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE1C,oBAAY,YAAY;IACtB,GAAG,QAAQ;IACX,IAAI,SAAS;CACd;AACD;;;;;;;;;GASG;AACH,qBAAa,SAAS;IACpB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,SAAS,CAAqD;IACtE,OAAO,CAAC,aAAa,CAAe;IACpC,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,OAAO,CAAyB;IACxC;;;;;OAKG;IACH,OAAO,CAAC,YAAY,CAAC,CAAgB;IACrC;;;OAGG;IACH,OAAO,CAAC,cAAc,CAAC,CAAgC;IACvD;;;OAGG;IACH,OAAO,CAAC,eAAe,CAAK;IAE5B;;;;;;;;;;;;;OAaG;IACH,MAAM,CAAC,QAAQ,CAAC,kBAAkB,OAAO;IACzC,MAAM,CAAC,QAAQ,CAAC,kBAAkB,KAAK;IACvC,MAAM,CAAC,QAAQ,CAAC,cAAc,SAAU;IACxC,MAAM,CAAC,QAAQ,CAAC,oBAAoB,OAAO;IAE3C;;;;;OAKG;IACH,OAAO;IAYP;;;;;;;;;OASG;WACU,MAAM,CACjB,QAAQ,EAAE,MAAM,EAChB,aAAa,GAAE,YAAgC,EAC/C,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAC/B,OAAO,CAAC,SAAS,CAAC;IAKrB;;;;;;;;;;;;;;;;;OAiBG;IACG,SAAS,CAAC,UAAU,UAAQ,EAAE,kBAAkB,UAAO;IAmD7D;;;;OAIG;IACH,OAAO,CAAC,OAAO;IAIf;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAY7B;;;OAGG;IACH,OAAO,CAAC,qBAAqB;IAoC7B,OAAO,CAAC,mBAAmB;IAa3B,OAAO,CAAC,gBAAgB;IASxB;;;;;OAKG;IACG,SAAS,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;IAiCzC;;;;;;OAMG;IACG,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAO3D;AASD,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B"}
+{"version":3,"file":"mcp-client.d.ts","sourceRoot":"","sources":["../../src/mcp/mcp-client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,mBAAmB,EAAE,MAAM,0CAA0C,CAAC;AACpF,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAGnE,OAAO,EACL,oBAAoB,EAEpB,mBAAmB,EACnB,aAAa,EAEb,YAAY,EACb,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAE1C,oBAAY,YAAY;IACtB,GAAG,QAAQ;IACX,IAAI,SAAS;CACd;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,UAAU,WAAW;IACnB,WAAW,CAAC,EAAE,CAAC,CAAC,EAAE,aAAa,KAAK,OAAO,CAAC,YAAY,CAAC,CAAC;IAC1D,QAAQ,CAAC,EAAE,CAAC,CAAC,EAAE,oBAAoB,KAAK,OAAO,CAAC,mBAAmB,CAAC,CAAC;CACtE;AAED;;;;;;;;;GASG;AACH,qBAAa,SAAS;IACpB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,SAAS,CAAqD;IACtE,OAAO,CAAC,aAAa,CAAe;IAC7B,SAAS,CAAC,EAAE,MAAM,CAAC;IAC1B,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,OAAO,CAAyB;IACxC,OAAO,CAAC,YAAY,CAAC,CAAsB;IAC3C,OAAO,CAAC,QAAQ,CAAc;IAC9B;;;;;OAKG;IACH,OAAO,CAAC,YAAY,CAAC,CAAgB;IACrC;;;OAGG;IACH,OAAO,CAAC,cAAc,CAAC,CAAgC;IACvD;;;OAGG;IACH,OAAO,CAAC,eAAe,CAAK;IAE5B;;;;;;;;;;;;;OAaG;IACH,MAAM,CAAC,QAAQ,CAAC,kBAAkB,OAAO;IACzC,MAAM,CAAC,QAAQ,CAAC,kBAAkB,KAAK;IACvC,MAAM,CAAC,QAAQ,CAAC,cAAc,SAAU;IACxC,MAAM,CAAC,QAAQ,CAAC,oBAAoB,OAAO;IAEpC,WAAW,EAAE,WAAW,CAAqB;IACpD;;;;;OAKG;IACH,OAAO;IAiBP;;;;;;;;;;;;OAYG;WACU,MAAM,CACjB,QAAQ,EAAE,MAAM,EAChB,aAAa,EAAE,YAAY,YAAoB,EAC/C,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,SAAS,EAC3C,YAAY,EAAE,mBAAmB,GAAG,SAAS,EAC7C,SAAS,EAAE,MAAM,GAAG,SAAS,EAC7B,QAAQ,GAAE,WAAgB,GACzB,OAAO,CAAC,SAAS,CAAC;IAerB;;;;;;;;;;;;;;;;;;OAkBG;IACG,SAAS,CAAC,UAAU,UAAQ,EAAE,kBAAkB,UAAO;IAuD7D;;;;OAIG;IACH,OAAO,CAAC,OAAO;IAIf;;;OAGG;IACH,OAAO,CAAC,qBAAqB;IAY7B;;;OAGG;IACH,OAAO,CAAC,qBAAqB;IAoC7B,OAAO,CAAC,mBAAmB;IAe3B;;;OAGG;IACH,OAAO,CAAC,gBAAgB;IA8BxB;;;;;OAKG;IACG,SAAS,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;IAmCzC,qBAAqB;;;;;;;;;;;;;;;;;;;;;;;;;IAIrB,gBAAgB;;;;;;;;;;;;;IAIhB,eAAe;IAIf;;;;;;OAMG;IACG,QAAQ,CACZ,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC5B,OAAO,CAAC,iBAAiB,CAAC;IAQ7B,wBAAwB,CACtB,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,KAAK,OAAO,CAAC,YAAY,CAAC,CAAC,GAAG,SAAS;IAmBpE,qBAAqB,CACnB,OAAO,EACH,CAAC,CAAC,CAAC,EAAE,oBAAoB,KAAK,OAAO,CAAC,mBAAmB,CAAC,CAAC,GAC3D,SAAS;CAkBhB;AAED;;;GAGG;AACH,MAAM,MAAM,iBAAiB,GAAG,OAAO,CACrC,UAAU,CAAC,OAAO,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,CAC7C,CAAC;AASF,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B"}

package/dist/mcp/mcp-client.js

@@ -4,6 +4,7 @@ exports.MCPClient = exports.MCPTransport = void 0;
 const index_js_1 = require("@modelcontextprotocol/sdk/client/index.js");
 const sse_js_1 = require("@modelcontextprotocol/sdk/client/sse.js");
 const streamableHttp_js_1 = require("@modelcontextprotocol/sdk/client/streamableHttp.js");
+const types_js_1 = require("@modelcontextprotocol/sdk/types.js");
 var MCPTransport;
 (function (MCPTransport) {
     MCPTransport["SSE"] = "sse";
@@ -23,8 +24,11 @@ class MCPClient {
     client;
     transport;
     transportType;
+    sessionId;
     endpoint;
     headers;
+    authProvider;
+    handlers;
     /**
      * Tracks an in-flight reconnect so concurrent triggers coalesce
      * (single-flight). When set, additional calls to `reconnect()` or
@@ -60,32 +64,41 @@ class MCPClient {
     static BACKOFF_MULTIPLIER = 2;
     static BACKOFF_MAX_MS = 30_000;
     static BACKOFF_JITTER_RATIO = 0.2;
+    elicitation = new EventTarget();
     /**
      * Private constructor to enforce using the static create method.
      * @param endpoint - The URL of the MCP server to connect to
      * @param transportType - The transport to use for the MCP client
      * @param headers - Optional custom headers to include in requests
      */
-    constructor(endpoint, transportType, headers = {}) {
+    constructor(endpoint, transportType, headers, authProvider, sessionId, handlers = {}) {
         this.endpoint = endpoint;
-        this.headers = headers;
+        this.headers = headers ?? {};
+        this.authProvider = authProvider;
         this.transportType = transportType;
-        this.transport = this.initializeTransport(undefined);
+        this.handlers = handlers;
+        this.transport = this.initializeTransport(sessionId);
         this.client = this.initializeClient();
     }
     /**
-     * Creates and initializes a new MCPClient instance.
-     * This is the recommended way to create an MCPClient as it handles both
-     * instantiation and connection setup.
+     * Creates and initializes a new MCPClient instance. This is the recommended
+     * way to create an MCPClient as it handles both instantiation and connection
+     * setup.
      * @param endpoint - The URL of the MCP server to connect to
      * @param transportType - The transport type to use for the MCP client. Defaults to HTTP.
      * @param headers - Optional custom headers to include in requests
+     * @param authProvider - Optional auth provider to use for authentication
+     * @param sessionId - Optional session id to use for the MCP client - if not
+     *   provided, a new session will be created
      * @returns A connected MCPClient instance ready for use
-     * @throws Will throw an error if connection fails
+     * @throws {Error} Will throw an error if connection fails
      */
-    static async create(endpoint, transportType = MCPTransport.HTTP, headers) {
-        const mcpClient = new MCPClient(endpoint, transportType, headers);
+    static async create(endpoint, transportType = MCPTransport.HTTP, headers, authProvider, sessionId, handlers = {}) {
+        const mcpClient = new MCPClient(endpoint, transportType, headers, authProvider, sessionId, handlers);
         await mcpClient.client.connect(mcpClient.transport);
+        if ("sessionId" in mcpClient.transport) {
+            mcpClient.sessionId = mcpClient.transport.sessionId;
+        }
         return mcpClient;
     }
     /**
@@ -105,6 +118,7 @@ class MCPClient {
      * @param newSession - Whether to create a new session (true) or reuse existing session ID (false)
      * @param reportErrorOnClose - Whether to report errors when closing the client
      * Note that only StreamableHTTPClientTransport supports session IDs.
+     * @returns A promise that resolves when the reconnect is complete
      */
     async reconnect(newSession = false, reportErrorOnClose = true) {
         // If a reconnect is already running, coalesce into it.
@@ -117,11 +131,7 @@ class MCPClient {
             this.reconnectTimer = undefined;
         }
         const doReconnect = async () => {
-            const sessionId = newSession
-                ? undefined
-                : "sessionId" in this.transport
-                    ? this.transport.sessionId
-                    : undefined;
+            const sessionId = newSession ? undefined : this.sessionId;
             // Prevent re-entrant onclose during deliberate close by detaching
             // the handler from the previous client instance.
             const prevClient = this.client;
@@ -138,6 +148,14 @@ class MCPClient {
             this.transport = this.initializeTransport(sessionId);
             this.client = this.initializeClient();
             await this.client.connect(this.transport);
+            // We may have gotten a session id from the server, so we need to set it
+            if ("sessionId" in this.transport) {
+                this.sessionId = this.transport.sessionId;
+                if (sessionId !== this.sessionId) {
+                    // This is a pretty unusual thing to happen, but it might be possible?
+                    console.warn("Session id mismatch", sessionId, this.sessionId);
+                }
+            }
         };
         this.reconnecting = (async () => {
             try {
@@ -161,6 +179,7 @@ class MCPClient {
     }
     /**
      * Compute the next backoff delay with symmetric jitter.
+     * @returns The next backoff delay in milliseconds
      */
     computeBackoffDelayMs() {
         const base = Math.min(MCPClient.BACKOFF_MAX_MS, MCPClient.BACKOFF_INITIAL_MS *
@@ -206,29 +225,50 @@ class MCPClient {
     initializeTransport(sessionId) {
         if (this.transportType === MCPTransport.SSE) {
             return new sse_js_1.SSEClientTransport(new URL(this.endpoint), {
+                authProvider: this.authProvider,
                 requestInit: { headers: this.headers },
             });
         }
         else {
             return new streamableHttp_js_1.StreamableHTTPClientTransport(new URL(this.endpoint), {
-                sessionId,
+                authProvider: this.authProvider,
                 requestInit: { headers: this.headers },
+                sessionId,
             });
         }
     }
+    /**
+     * Initializes the MCP client with the appropriate capabilities and handlers
+     * @returns The initialized MCP client
+     */
     initializeClient() {
+        const elicitationCapability = this.handlers.elicitation
+            ? { elicitation: {} }
+            : {};
+        const samplingCapability = this.handlers.sampling ? { sampling: {} } : {};
         const client = new index_js_1.Client({
             name: "tambo-mcp-client",
             version: "1.0.0",
+        }, {
+            capabilities: {
+                ...elicitationCapability,
+                ...samplingCapability,
+            },
         });
         client.onclose = this.onclose.bind(this);
+        if (this.handlers.elicitation) {
+            client.setRequestHandler(types_js_1.ElicitRequestSchema, this.handlers.elicitation);
+        }
+        if (this.handlers.sampling) {
+            client.setRequestHandler(types_js_1.CreateMessageRequestSchema, this.handlers.sampling);
+        }
         return client;
     }
     /**
      * Retrieves a complete list of all available tools from the MCP server.
      * Handles pagination automatically by following cursors until all tools are fetched.
      * @returns A complete list of all available tools and their descriptions
-     * @throws Will throw an error if any server request fails during pagination
+     * @throws {Error} Will throw an error if any server request fails during pagination
      */
     async listTools() {
         const allTools = [];
@@ -237,7 +277,9 @@ class MCPClient {
         while (hasMore) {
             const response = await this.client.listTools({ cursor }, {});
             allTools.push(...response.tools.map((tool) => {
-                if (tool.inputSchema.type !== "object") {
+                // make sure the right type is used
+                const inputSchemaType = tool.inputSchema.type;
+                if (inputSchemaType !== "object") {
                     throw new Error(`Input schema for tool ${tool.name} is not an object`);
                 }
                 return {
@@ -255,12 +297,21 @@ class MCPClient {
         }
         return allTools;
     }
+    getServerCapabilities() {
+        return this.client.getServerCapabilities();
+    }
+    getServerVersion() {
+        return this.client.getServerVersion();
+    }
+    getInstructions() {
+        return this.client.getInstructions();
+    }
     /**
      * Calls a specific tool on the MCP server with the provided arguments.
      * @param name - The name of the tool to call
      * @param args - Arguments to pass to the tool, must match the tool's expected schema
      * @returns The result from the tool execution
-     * @throws Will throw an error if the tool call fails or if arguments are invalid
+     * @throws {Error} Will throw an error if the tool call fails or if arguments are invalid
      */
     async callTool(name, args) {
         const result = await this.client.callTool({
@@ -269,6 +320,40 @@ class MCPClient {
         });
         return result;
     }
+    updateElicitationHandler(handler) {
+        // Because we advertise the elicitation capability on initial connection, we can only update
+        // an existing handler, not add it if we haven't set it yet.
+        if (handler && !this.handlers.elicitation) {
+            throw new Error("Elicitation handler must be set on create");
+        }
+        this.handlers = {
+            ...this.handlers,
+            elicitation: handler,
+        };
+        if (!handler) {
+            const method = types_js_1.ElicitRequestSchema.shape.method.value;
+            this.client.removeRequestHandler(method);
+            return;
+        }
+        this.client.setRequestHandler(types_js_1.ElicitRequestSchema, handler);
+    }
+    updateSamplingHandler(handler) {
+        // Because we advertise the sampling capability on initial connection, we can only update
+        // an existing handler, not add it if we haven't set it yet.
+        if (handler && !this.handlers.sampling) {
+            throw new Error("Sampling handler must be set on create");
+        }
+        this.handlers = {
+            ...this.handlers,
+            sampling: handler,
+        };
+        if (!handler) {
+            const method = types_js_1.CreateMessageRequestSchema.shape.method.value;
+            this.client.removeRequestHandler(method);
+            return;
+        }
+        this.client.setRequestHandler(types_js_1.CreateMessageRequestSchema, handler);
+    }
 }
 exports.MCPClient = MCPClient;
 //# sourceMappingURL=mcp-client.js.map