@fitchmultz/pi-mcp-adapter 8.0.1 → 8.0.3

package/dist/client-operations.d.ts

@@ -0,0 +1,5 @@
+import type { Client } from "@modelcontextprotocol/client";
+export type ClientOperation = "request/refresh" | "sampling callback" | "elicitation callback" | "subscription";
+export declare const clientOperations: WeakMap<Client, Map<Promise<unknown>, ClientOperation>>;
+export declare function drainClientCallbacks(client: Client): Promise<void>;
+export declare function drainClientOperations(client: Client): Promise<void>;

package/dist/client-operations.js

@@ -0,0 +1,19 @@
+export const clientOperations = new WeakMap();
+export async function drainClientCallbacks(client) {
+    for (;;) {
+        const pending = [...(clientOperations.get(client) ?? [])]
+            .filter(([, kind]) => kind === "sampling callback" || kind === "elicitation callback")
+            .map(([operation]) => operation);
+        if (!pending.length)
+            return;
+        await Promise.allSettled(pending);
+    }
+}
+export async function drainClientOperations(client) {
+    for (;;) {
+        const pending = [...(clientOperations.get(client)?.keys() ?? [])];
+        if (!pending.length)
+            return;
+        await Promise.allSettled(pending);
+    }
+}

package/dist/commands.d.ts

@@ -1,7 +1,7 @@
 import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 import type { McpExtensionState } from "./state.ts";
 import { type McpAuthResult, type McpConfig } from "./types.ts";
-import { type McpOAuthRuntime } from "./mcp-auth-flow.ts";
+import { type McpOAuthRuntime } from "./mcp-oauth-runtime.ts";
 export declare function showStatus(state: McpExtensionState, ctx: ExtensionContext): Promise<void>;
 export declare function showPrompts(state: McpExtensionState, ctx: ExtensionContext): Promise<void>;
 export declare function showTools(state: McpExtensionState, ctx: ExtensionContext): Promise<void>;

package/dist/commands.js

@@ -3,7 +3,7 @@ import { ensureCompatibilityImports, getMcpDiscoverySummary, getProjectConfigPat
 import { markKeepAliveAfterConnect, notifyToolMetadataUpdated, updateMetadataCache, updateStatusBar, getFailureAgeSeconds, getFailureMessage, clearFailure, recordFailure } from "./init.js";
 import { loadMetadataCache, reconstructPromptMetadata } from "./metadata-cache.js";
 import { buildToolMetadata } from "./tool-metadata.js";
-import { supportsOAuth, authenticate, removeAuth } from "./mcp-auth-flow.js";
+import { supportsOAuth } from "./mcp-oauth-runtime.js";
 import { getAuthStorageOptions, inspectAuthForUrl } from "./mcp-auth.js";
 import { loadOnboardingState, markSetupCompleted as persistSetupCompleted, markSharedConfigHintShown } from "./onboarding-state.js";
 import { openPath, resolveServerUrl, sanitizeTerminalText } from "./utils.js";
@@ -229,6 +229,7 @@ export async function authenticateServer(serverName, config, ctx, signal, runtim
         }
         ui.setStatus("mcp-auth", `Authenticating ${serverName}...`);
         const authStorageOptions = getAuthStorageOptions(config.settings?.oauthDir, cwd);
+        const { authenticate } = await import("./mcp-auth-flow.js");
         const status = await authenticate(serverName, serverUrl, definition, {
             ...(authStorageOptions.baseDir ? { authStorageOptions } : {}),
             onAuthorizationUrl: (authorizationUrl) => {
@@ -272,6 +273,7 @@ export async function logoutServer(serverName, state, ctx) {
     }
     const signal = state.owner?.signal;
     try {
+        const { removeAuth } = await import("./mcp-auth-flow.js");
         await removeAuth(serverName, { authStorageOptions: state.authStorageOptions, signal, runtime: state.oauthRuntime });
     }
     catch (error) {

package/dist/connection-runtime.d.ts

@@ -0,0 +1,503 @@
+import { Client } from "@modelcontextprotocol/client";
+import type { McpOAuthProvider } from "./mcp-oauth-provider.ts";
+export declare class ManagedClient extends Client {
+    serverName: string;
+    oauthProvider: McpOAuthProvider | undefined;
+    private own;
+    ownNative<T>(run: () => Promise<T>): Promise<T>;
+    ownSend(run: () => Promise<void>): Promise<void>;
+    ownCallback<T>(kind: "sampling callback" | "elicitation callback", run: () => Promise<T>): Promise<T>;
+    private track;
+    callTool(...args: Parameters<Client["callTool"]>): Promise<{
+        [x: string]: unknown;
+        _meta?: {
+            [x: string]: unknown;
+            "io.modelcontextprotocol/serverInfo"?: {
+                version: string;
+                websiteUrl?: string | undefined;
+                description?: string | undefined;
+                icons?: {
+                    src: string;
+                    mimeType?: string | undefined;
+                    sizes?: string[] | undefined;
+                    theme?: "dark" | "light" | undefined;
+                }[] | undefined;
+                name: string;
+                title?: string | undefined;
+            } | undefined;
+        } | undefined;
+        content: ({
+            type: "text";
+            text: string;
+            annotations?: {
+                audience?: ("assistant" | "user")[] | undefined;
+                priority?: number | undefined;
+                lastModified?: string | undefined;
+            } | undefined;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        } | {
+            type: "image";
+            data: string;
+            mimeType: string;
+            annotations?: {
+                audience?: ("assistant" | "user")[] | undefined;
+                priority?: number | undefined;
+                lastModified?: string | undefined;
+            } | undefined;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        } | {
+            type: "audio";
+            data: string;
+            mimeType: string;
+            annotations?: {
+                audience?: ("assistant" | "user")[] | undefined;
+                priority?: number | undefined;
+                lastModified?: string | undefined;
+            } | undefined;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        } | {
+            uri: string;
+            description?: string | undefined;
+            mimeType?: string | undefined;
+            size?: number | undefined;
+            annotations?: {
+                audience?: ("assistant" | "user")[] | undefined;
+                priority?: number | undefined;
+                lastModified?: string | undefined;
+            } | undefined;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+            icons?: {
+                src: string;
+                mimeType?: string | undefined;
+                sizes?: string[] | undefined;
+                theme?: "dark" | "light" | undefined;
+            }[] | undefined;
+            name: string;
+            title?: string | undefined;
+            type: "resource_link";
+        } | {
+            type: "resource";
+            resource: {
+                uri: string;
+                mimeType?: string | undefined;
+                _meta?: {
+                    [x: string]: unknown;
+                } | undefined;
+                text: string;
+            } | {
+                uri: string;
+                mimeType?: string | undefined;
+                _meta?: {
+                    [x: string]: unknown;
+                } | undefined;
+                blob: string;
+            };
+            annotations?: {
+                audience?: ("assistant" | "user")[] | undefined;
+                priority?: number | undefined;
+                lastModified?: string | undefined;
+            } | undefined;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+        })[];
+        structuredContent?: unknown;
+        isError?: boolean | undefined;
+    }>;
+    readResource(...args: Parameters<Client["readResource"]>): Promise<{
+        [x: string]: unknown;
+        _meta?: {
+            [x: string]: unknown;
+            "io.modelcontextprotocol/serverInfo"?: {
+                version: string;
+                websiteUrl?: string | undefined;
+                description?: string | undefined;
+                icons?: {
+                    src: string;
+                    mimeType?: string | undefined;
+                    sizes?: string[] | undefined;
+                    theme?: "dark" | "light" | undefined;
+                }[] | undefined;
+                name: string;
+                title?: string | undefined;
+            } | undefined;
+        } | undefined;
+        contents: ({
+            uri: string;
+            mimeType?: string | undefined;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+            text: string;
+        } | {
+            uri: string;
+            mimeType?: string | undefined;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+            blob: string;
+        })[];
+    }>;
+    getPrompt(...args: Parameters<Client["getPrompt"]>): Promise<{
+        [x: string]: unknown;
+        _meta?: {
+            [x: string]: unknown;
+            "io.modelcontextprotocol/serverInfo"?: {
+                version: string;
+                websiteUrl?: string | undefined;
+                description?: string | undefined;
+                icons?: {
+                    src: string;
+                    mimeType?: string | undefined;
+                    sizes?: string[] | undefined;
+                    theme?: "dark" | "light" | undefined;
+                }[] | undefined;
+                name: string;
+                title?: string | undefined;
+            } | undefined;
+        } | undefined;
+        description?: string | undefined;
+        messages: {
+            role: "assistant" | "user";
+            content: {
+                type: "text";
+                text: string;
+                annotations?: {
+                    audience?: ("assistant" | "user")[] | undefined;
+                    priority?: number | undefined;
+                    lastModified?: string | undefined;
+                } | undefined;
+                _meta?: {
+                    [x: string]: unknown;
+                } | undefined;
+            } | {
+                type: "image";
+                data: string;
+                mimeType: string;
+                annotations?: {
+                    audience?: ("assistant" | "user")[] | undefined;
+                    priority?: number | undefined;
+                    lastModified?: string | undefined;
+                } | undefined;
+                _meta?: {
+                    [x: string]: unknown;
+                } | undefined;
+            } | {
+                type: "audio";
+                data: string;
+                mimeType: string;
+                annotations?: {
+                    audience?: ("assistant" | "user")[] | undefined;
+                    priority?: number | undefined;
+                    lastModified?: string | undefined;
+                } | undefined;
+                _meta?: {
+                    [x: string]: unknown;
+                } | undefined;
+            } | {
+                uri: string;
+                description?: string | undefined;
+                mimeType?: string | undefined;
+                size?: number | undefined;
+                annotations?: {
+                    audience?: ("assistant" | "user")[] | undefined;
+                    priority?: number | undefined;
+                    lastModified?: string | undefined;
+                } | undefined;
+                _meta?: {
+                    [x: string]: unknown;
+                } | undefined;
+                icons?: {
+                    src: string;
+                    mimeType?: string | undefined;
+                    sizes?: string[] | undefined;
+                    theme?: "dark" | "light" | undefined;
+                }[] | undefined;
+                name: string;
+                title?: string | undefined;
+                type: "resource_link";
+            } | {
+                type: "resource";
+                resource: {
+                    uri: string;
+                    mimeType?: string | undefined;
+                    _meta?: {
+                        [x: string]: unknown;
+                    } | undefined;
+                    text: string;
+                } | {
+                    uri: string;
+                    mimeType?: string | undefined;
+                    _meta?: {
+                        [x: string]: unknown;
+                    } | undefined;
+                    blob: string;
+                };
+                annotations?: {
+                    audience?: ("assistant" | "user")[] | undefined;
+                    priority?: number | undefined;
+                    lastModified?: string | undefined;
+                } | undefined;
+                _meta?: {
+                    [x: string]: unknown;
+                } | undefined;
+            };
+        }[];
+    }>;
+    listTools(...args: Parameters<Client["listTools"]>): Promise<{
+        [x: string]: unknown;
+        _meta?: {
+            [x: string]: unknown;
+            "io.modelcontextprotocol/serverInfo"?: {
+                version: string;
+                websiteUrl?: string | undefined;
+                description?: string | undefined;
+                icons?: {
+                    src: string;
+                    mimeType?: string | undefined;
+                    sizes?: string[] | undefined;
+                    theme?: "dark" | "light" | undefined;
+                }[] | undefined;
+                name: string;
+                title?: string | undefined;
+            } | undefined;
+        } | undefined;
+        nextCursor?: string | undefined;
+        tools: {
+            description?: string | undefined;
+            inputSchema: {
+                [x: string]: unknown;
+                type: "object";
+                properties?: {
+                    [x: string]: string | number | boolean | (string | number | boolean | (string | number | boolean | (string | number | boolean | (string | number | boolean | (string | number | boolean | (string | number | boolean | (string | number | boolean | (string | number | boolean | (string | number | boolean | (string | number | boolean | /*elided*/ any | {
+                        [x: string]: string | number | boolean | /*elided*/ any | /*elided*/ any | null;
+                    } | null)[] | {
+                        [x: string]: string | number | boolean | any | any | null;
+                    } | null)[] | {
+                        [x: string]: string | number | boolean | any | any | null;
+                    } | null)[] | {
+                        [x: string]: string | number | boolean | any | any | null;
+                    } | null)[] | {
+                        [x: string]: string | number | boolean | any | any | null;
+                    } | null)[] | {
+                        [x: string]: string | number | boolean | any | any | null;
+                    } | null)[] | {
+                        [x: string]: string | number | boolean | any | any | null;
+                    } | null)[] | {
+                        [x: string]: string | number | boolean | any | any | null;
+                    } | null)[] | {
+                        [x: string]: string | number | boolean | any | any | null;
+                    } | null)[] | {
+                        [x: string]: string | number | boolean | any | any | null;
+                    } | null)[] | {
+                        [x: string]: string | number | boolean | any | any | null;
+                    } | null;
+                } | undefined;
+                required?: string[] | undefined;
+            };
+            outputSchema?: {
+                [x: string]: unknown;
+                $schema?: string | undefined;
+            } | undefined;
+            annotations?: {
+                title?: string | undefined;
+                readOnlyHint?: boolean | undefined;
+                destructiveHint?: boolean | undefined;
+                idempotentHint?: boolean | undefined;
+                openWorldHint?: boolean | undefined;
+            } | undefined;
+            execution?: {
+                taskSupport?: "forbidden" | "optional" | "required" | undefined;
+            } | undefined;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+            icons?: {
+                src: string;
+                mimeType?: string | undefined;
+                sizes?: string[] | undefined;
+                theme?: "dark" | "light" | undefined;
+            }[] | undefined;
+            name: string;
+            title?: string | undefined;
+        }[];
+    }>;
+    listResources(...args: Parameters<Client["listResources"]>): Promise<{
+        [x: string]: unknown;
+        _meta?: {
+            [x: string]: unknown;
+            "io.modelcontextprotocol/serverInfo"?: {
+                version: string;
+                websiteUrl?: string | undefined;
+                description?: string | undefined;
+                icons?: {
+                    src: string;
+                    mimeType?: string | undefined;
+                    sizes?: string[] | undefined;
+                    theme?: "dark" | "light" | undefined;
+                }[] | undefined;
+                name: string;
+                title?: string | undefined;
+            } | undefined;
+        } | undefined;
+        nextCursor?: string | undefined;
+        resources: {
+            uri: string;
+            description?: string | undefined;
+            mimeType?: string | undefined;
+            size?: number | undefined;
+            annotations?: {
+                audience?: ("assistant" | "user")[] | undefined;
+                priority?: number | undefined;
+                lastModified?: string | undefined;
+            } | undefined;
+            _meta?: {
+                [x: string]: unknown;
+            } | undefined;
+            icons?: {
+                src: string;
+                mimeType?: string | undefined;
+                sizes?: string[] | undefined;
+                theme?: "dark" | "light" | undefined;
+            }[] | undefined;
+            name: string;
+            title?: string | undefined;
+        }[];
+    }>;
+    listPrompts(...args: Parameters<Client["listPrompts"]>): Promise<{
+        [x: string]: unknown;
+        _meta?: {
+            [x: string]: unknown;
+            "io.modelcontextprotocol/serverInfo"?: {
+                version: string;
+                websiteUrl?: string | undefined;
+                description?: string | undefined;
+                icons?: {
+                    src: string;
+                    mimeType?: string | undefined;
+                    sizes?: string[] | undefined;
+                    theme?: "dark" | "light" | undefined;
+                }[] | undefined;
+                name: string;
+                title?: string | undefined;
+            } | undefined;
+        } | undefined;
+        nextCursor?: string | undefined;
+        prompts: {
+            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;
+                theme?: "dark" | "light" | undefined;
+            }[] | undefined;
+            name: string;
+            title?: string | undefined;
+        }[];
+    }>;
+    complete(...args: Parameters<Client["complete"]>): Promise<{
+        [x: string]: unknown;
+        _meta?: {
+            [x: string]: unknown;
+            "io.modelcontextprotocol/serverInfo"?: {
+                version: string;
+                websiteUrl?: string | undefined;
+                description?: string | undefined;
+                icons?: {
+                    src: string;
+                    mimeType?: string | undefined;
+                    sizes?: string[] | undefined;
+                    theme?: "dark" | "light" | undefined;
+                }[] | undefined;
+                name: string;
+                title?: string | undefined;
+            } | undefined;
+        } | undefined;
+        completion: {
+            [x: string]: unknown;
+            values: string[];
+            total?: number | undefined;
+            hasMore?: boolean | undefined;
+        };
+    }>;
+    setLoggingLevel(...args: Parameters<Client["setLoggingLevel"]>): Promise<{
+        _meta?: {
+            [x: string]: unknown;
+            "io.modelcontextprotocol/serverInfo"?: {
+                version: string;
+                websiteUrl?: string | undefined;
+                description?: string | undefined;
+                icons?: {
+                    src: string;
+                    mimeType?: string | undefined;
+                    sizes?: string[] | undefined;
+                    theme?: "dark" | "light" | undefined;
+                }[] | undefined;
+                name: string;
+                title?: string | undefined;
+            } | undefined;
+        } | undefined;
+    }>;
+    listen(...args: Parameters<Client["listen"]>): Promise<import("@modelcontextprotocol/client").McpSubscription>;
+    subscribeResource(...args: Parameters<Client["subscribeResource"]>): Promise<{
+        _meta?: {
+            [x: string]: unknown;
+            "io.modelcontextprotocol/serverInfo"?: {
+                version: string;
+                websiteUrl?: string | undefined;
+                description?: string | undefined;
+                icons?: {
+                    src: string;
+                    mimeType?: string | undefined;
+                    sizes?: string[] | undefined;
+                    theme?: "dark" | "light" | undefined;
+                }[] | undefined;
+                name: string;
+                title?: string | undefined;
+            } | undefined;
+        } | undefined;
+    }>;
+    unsubscribeResource(...args: Parameters<Client["unsubscribeResource"]>): Promise<{
+        _meta?: {
+            [x: string]: unknown;
+            "io.modelcontextprotocol/serverInfo"?: {
+                version: string;
+                websiteUrl?: string | undefined;
+                description?: string | undefined;
+                icons?: {
+                    src: string;
+                    mimeType?: string | undefined;
+                    sizes?: string[] | undefined;
+                    theme?: "dark" | "light" | undefined;
+                }[] | undefined;
+                name: string;
+                title?: string | undefined;
+            } | undefined;
+        } | undefined;
+    }>;
+}
+export { StdioClientTransport } from "@modelcontextprotocol/client/stdio";
+export { SdkHttpError, StreamableHTTPClientTransport, SSEClientTransport, auth, extractWWWAuthenticateParams, UnauthorizedError } from "@modelcontextprotocol/client";
+export { UnixSocketClientTransport } from "./unix-socket-transport.ts";
+export { trackToolTransport, trackToolTransportFailure } from "./session-recovery.ts";
+export { McpOAuthProvider } from "./mcp-oauth-provider.ts";
+export { extractOAuthConfig, isOAuthChallenge, recordOAuthChallenge, getOAuthRequest, bindOAuthRequestIssuer } from "./mcp-auth-flow.ts";
+export { registerSamplingHandler } from "./sampling-handler.ts";
+export { registerElicitationHandler } from "./elicitation-handler.ts";
+export { serverStreamResultPatchNotificationSchema } from "./ui-stream-types.ts";

package/dist/connection-runtime.js

@@ -0,0 +1,98 @@
+import { AsyncLocalStorage } from "node:async_hooks";
+import { Client } from "@modelcontextprotocol/client";
+import { clientOperations } from "./client-operations.js";
+import { throwIfAborted } from "./abort.js";
+// Track complete SDK operations, including pagination and pre-request cache awaits.
+export class ManagedClient extends Client {
+    serverName = "";
+    oauthProvider;
+    own(operation, kind) {
+        let pending = clientOperations.get(this);
+        if (!pending)
+            clientOperations.set(this, pending = new Map());
+        pending.set(operation, kind);
+        const settled = () => { pending.delete(operation); };
+        void operation.then(settled, settled);
+        return operation;
+    }
+    ownNative(run) {
+        return this.own(run(), "request/refresh");
+    }
+    ownSend(run) {
+        // SDK-generated callback replies also perform HTTP/auth work, without calling callTool().
+        return this.own(Promise.resolve().then(run), "request/refresh");
+    }
+    ownCallback(kind, run) {
+        // Insert before dispatch. Cancellation/close of the SDK response does not settle this promise.
+        return this.own(Promise.resolve().then(run), kind);
+    }
+    async track(start, signal) {
+        const controller = this.oauthProvider ? new AbortController() : undefined;
+        let removeAbortListener = () => { };
+        const operation = this.oauthProvider
+            ? this.oauthProvider.runWithSignal(controller.signal, () => {
+                // Native legacy cancellation notifications must retain their operation's auth context.
+                const abort = AsyncLocalStorage.bind(() => controller.abort(signal?.reason));
+                if (signal?.aborted)
+                    abort();
+                else
+                    signal?.addEventListener("abort", abort, { once: true });
+                removeAbortListener = () => signal?.removeEventListener("abort", abort);
+                return start(controller.signal);
+            })
+            : start(signal);
+        void operation.then(removeAbortListener, () => { controller?.abort(); removeAbortListener(); });
+        return this.own(operation, "request/refresh");
+    }
+    callTool(...args) {
+        return this.track(signal => super.callTool(args[0], { ...args[1], ...(signal ? { signal } : {}) }), args[1]?.signal);
+    }
+    readResource(...args) {
+        return this.track(signal => super.readResource(args[0], { ...args[1], ...(signal ? { signal } : {}) }), args[1]?.signal);
+    }
+    getPrompt(...args) {
+        return this.track(signal => super.getPrompt(args[0], { ...args[1], ...(signal ? { signal } : {}) }), args[1]?.signal);
+    }
+    listTools(...args) {
+        return this.track(signal => super.listTools(args[0], { ...args[1], ...(signal ? { signal } : {}) }), args[1]?.signal);
+    }
+    listResources(...args) {
+        return this.track(signal => super.listResources(args[0], { ...args[1], ...(signal ? { signal } : {}) }), args[1]?.signal);
+    }
+    listPrompts(...args) {
+        return this.track(signal => super.listPrompts(args[0], { ...args[1], ...(signal ? { signal } : {}) }), args[1]?.signal);
+    }
+    complete(...args) {
+        return this.track(signal => super.complete(args[0], { ...args[1], ...(signal ? { signal } : {}) }), args[1]?.signal);
+    }
+    setLoggingLevel(...args) {
+        return this.track(signal => super.setLoggingLevel(args[0], { ...args[1], ...(signal ? { signal } : {}) }), args[1]?.signal);
+    }
+    listen(...args) {
+        return this.track(async (signal) => {
+            const subscription = await super.listen(args[0], { ...args[1], ...(signal ? { signal } : {}) });
+            this.own(subscription.closed, "subscription");
+            return subscription;
+        }, args[1]?.signal);
+    }
+    subscribeResource(...args) {
+        return this.track(async (signal) => {
+            throwIfAborted(signal);
+            return super.subscribeResource(args[0], { ...args[1], ...(signal ? { signal } : {}) });
+        }, args[1]?.signal);
+    }
+    unsubscribeResource(...args) {
+        return this.track(signal => super.unsubscribeResource(args[0], { ...args[1], ...(signal ? { signal } : {}) }), args[1]?.signal);
+    }
+}
+// Loaded only by the manager's connection path. Keep native constructors and
+// synchronous connection setup together, after the one asynchronous import.
+export { StdioClientTransport } from "@modelcontextprotocol/client/stdio";
+export { SdkHttpError, StreamableHTTPClientTransport, SSEClientTransport, auth, extractWWWAuthenticateParams, UnauthorizedError } from "@modelcontextprotocol/client";
+export { UnixSocketClientTransport } from "./unix-socket-transport.js";
+export { trackToolTransport, trackToolTransportFailure } from "./session-recovery.js";
+export { McpOAuthProvider } from "./mcp-oauth-provider.js";
+export { extractOAuthConfig, isOAuthChallenge, recordOAuthChallenge, getOAuthRequest, bindOAuthRequestIssuer } from "./mcp-auth-flow.js";
+export { registerSamplingHandler } from "./sampling-handler.js";
+export { registerElicitationHandler } from "./elicitation-handler.js";
+export { serverStreamResultPatchNotificationSchema } from "./ui-stream-types.js";

package/dist/direct-tools.js

@@ -1,7 +1,6 @@
 import { abortable, throwIfAborted } from "./abort.js";
 import { parseDirectToolSelectors } from "./metadata-cache.js";
 export { getMissingConfiguredDirectToolServers } from "./metadata-cache.js";
-import { executeCall } from "./proxy-modes.js";
 import { isServerDisabled } from "./types.js";
 import { nativeMcpResult } from "./tool-registrar.js";
 export function directToolSelection(config, serverName, envOverride) {
@@ -46,6 +45,7 @@ export function createDirectToolExecutor(getState, getInitPromise, spec, beforeE
                 details: { error: "not_initialized" },
             }, !!spec.outputSchema);
         }
+        const { executeCall } = await import("./proxy-modes.js");
         const result = await executeCall(state, spec.originalName, params, spec.serverName, undefined, signal, { toolCallId }, beforeExecute ? (callSignal, operation) => beforeExecute(toolCallId, { ...ctx, signal: callSignal }, operation) : undefined, { exactOriginalName: true, ...(spec.resourceUri ? { resourceUri: spec.resourceUri } : {}) });
         const { mode: _mode, ...details } = result.details;
         return nativeMcpResult({ ...result, details }, !!spec.outputSchema);