@himanshusinghs/mongodb-mcp-server 1.5.1 → 1.7.0

package/dist/cjs/transports/base.d.ts

@@ -1,17 +1,33 @@
 import type { UserConfig } from "../common/config/userConfig.js";
-import { Server, type ServerOptions } from "../server.js";
+import { type AnyToolClass, Server, type ServerOptions } from "../server.js";
+import { type SessionOptions } from "../common/session.js";
 import type { LoggerBase } from "../common/logger.js";
+import { CompositeLogger } from "../common/logger.js";
 import { DeviceId } from "../helpers/deviceId.js";
 import { type ConnectionManagerFactoryFn } from "../common/connectionManager.js";
 import { type ConnectionErrorHandler } from "../common/connectionErrorHandler.js";
 import type { CommonProperties } from "../telemetry/types.js";
 import type { AtlasLocalClientFactoryFn } from "../common/atlasLocal.js";
-import type { ToolClass } from "../tools/tool.js";
-import type { ApiClientFactoryFn } from "../common/atlas/apiClient.js";
+import { type ApiClientFactoryFn } from "../common/atlas/apiClient.js";
+/**
+ * Request context containing HTTP headers and query parameters.
+ * This type is primarily used by HTTP-based transports.
+ */
 export type RequestContext = {
     headers?: Record<string, string | string[] | undefined>;
     query?: Record<string, string | string[] | undefined>;
 };
+export type CustomizableSessionOptions<TUserConfig extends UserConfig = UserConfig> = Partial<Pick<SessionOptions<TUserConfig>, "userConfig" | "apiClient" | "atlasLocalClient" | "connectionManager" | "connectionErrorHandler"> & {
+    sessionIdGenerator?: () => string;
+}>;
+export type CustomizableServerOptions<TUserConfig extends UserConfig = UserConfig, TContext = unknown> = Partial<Pick<ServerOptions<TUserConfig, TContext>, "uiRegistry" | "tools" | "toolContext" | "elicitation">> & {
+    /**
+     * An optional key value pair of telemetry properties that are reported to
+     * the telemetry backend. Most, if not all, of the properties are captured
+     * automatically.
+     */
+    telemetryProperties?: Partial<CommonProperties>;
+};
 /**
  * A function to dynamically generate `UserConfig` object, potentially unique to
  * each MCP client session.
@@ -29,10 +45,10 @@ export type RequestContext = {
  * @see {@link RequestContext} to inspect the properties available on
  * `requestContext` object.
  */
-type CreateSessionConfigFn = (context: {
-    userConfig: UserConfig;
+type CreateSessionConfigFn<TUserConfig extends UserConfig = UserConfig> = (context: {
+    userConfig: TUserConfig;
     request?: RequestContext;
-}) => Promise<UserConfig> | UserConfig;
+}) => Promise<TUserConfig> | TUserConfig;
 /**
  * Configuration options for customizing how transport runners are initialized.
  * This includes specifying the base user configuration, providing custom
@@ -47,8 +63,10 @@ type CreateSessionConfigFn = (context: {
  * advanced use-cases (such as embedding the MCP server in another application
  * or supporting custom authentication flows) may require customizing other
  * `TransportRunnerConfig` options as well.
+ *
+ * @template TContext - The type of the custom tool context object
  */
-export type TransportRunnerConfig = {
+export type TransportRunnerConfig<TUserConfig extends UserConfig = UserConfig> = {
     /**
      * Base user configuration for the server.
      *
@@ -65,8 +83,9 @@ export type TransportRunnerConfig = {
      * server library exports) to create a default configuration -
      * `UserConfigSchema.parse({})`.
      */
-    userConfig: UserConfig;
+    userConfig: TUserConfig;
     /**
+     * @deprecated Use `start({ sessionOptions: {connectionManager: MyCustomConnectionManager} })` instead
      * An optional factory function to generates an instance of
      * `ConnectionManager`. When not provided, MongoDB MCP Server uses an
      * internal implementation to manage connection to MongoDB deployments.
@@ -75,19 +94,10 @@ export type TransportRunnerConfig = {
      * connections differently and outside of MongoDB MCP server.
      */
     createConnectionManager?: ConnectionManagerFactoryFn;
-    /**
-     * An optional function to handle connection related errors. When not
-     * provided, MongoDB MCP Server uses an internal implementation to handle
-     * the errors raised by internal implementation of `ConnectionManager`
-     * class.
-     *
-     * Customize this only if you need to handle the Connection errors different
-     * from the internal implementation or if you have provided a different
-     * implementation of `ConnectionManager` that might raise errors unknown to
-     * default internal connection error handler.
-     */
+    /** @deprecated Use `start({ serverOptions: {connectionErrorHandler: MyCustomConnectionErrorHandler} })` instead */
     connectionErrorHandler?: ConnectionErrorHandler;
     /**
+     * @deprecated Use `start({ sessionOptions: {atlasLocalClient: MyCustomAtlasLocalClient} })` instead
      * An optional factory function to create a client for working with Atlas
      * local deployments. When not provided, MongoDB MCP Server uses an internal
      * implementation to create the local Atlas client.
@@ -103,67 +113,17 @@ export type TransportRunnerConfig = {
      */
     additionalLoggers?: LoggerBase[];
     /**
-     * An optional key value pair of telemetry properties that are reported to
-     * the telemetry backend. Most, if not all, of the properties are captured
-     * automatically.
+     * @deprecated This field will be removed in a future version. Use `createServer({ serverOptions: {telemetryProperties: MyCustomTelemetryProperties} })` instead.
      */
     telemetryProperties?: Partial<CommonProperties>;
+    /** @deprecated Use `createServer({ serverOptions: {tools: [...AllTools, MyCustomTool]} })` instead */
+    tools?: AnyToolClass[];
     /**
-     * An optional list of tools constructors to be registered to the MongoDB
-     * MCP Server.
-     *
-     * When not provided, MongoDB MCP Server will register all internal tools.
-     * When specified, **only** the tools in this list will be registered.
-     *
-     * This allows you to:
-     * - Register only custom tools (excluding all internal tools)
-     * - Register a subset of internal tools alongside custom tools
-     * - Register all internal tools plus custom tools
-     *
-     * To include internal tools, import them from `mongodb-mcp-server/tools`:
-     *
-     * ```typescript
-     * import { AllTools, AggregateTool, FindTool } from "mongodb-mcp-server/tools";
-     *
-     * // Register all internal tools plus custom tools
-     * tools: [...AllTools, MyCustomTool]
-     *
-     * // Register only specific MongoDB tools plus custom tools
-     * tools: [AggregateTool, FindTool, MyCustomTool]
-     *
-     * // Register all internal tools of mongodb category
-     * tools: [AllTools.filter((tool) => tool.category === "mongodb")]
-     * ```
-     *
-     * Note: Ensure that each tool has unique names otherwise the server will
-     * throw an error when initializing an MCP Client session. If you're using
-     * only the internal tools, then you don't have to worry about it unless,
-     * you've overridden the tool names.
-     *
-     * To ensure that you provide compliant tool implementations extend your
-     * tool implementation using `ToolBase` class and ensure that they conform
-     * to `ToolClass` type.
-     *
-     * @see {@link ToolClass} for the type that tool classes must conform to
-     * @see {@link ToolBase} for base class for all the tools
-     */
-    tools?: ToolClass[];
-    /**
-     * An optional function to hook into session configuration lifecycle and
-     * provide session specific configuration (`UserConfig`).
-     *
-     * The function is called before each session is created, allowing you to:
-     * - Fetch configuration from external sources (secrets managers, APIs)
-     * - Apply user-specific permissions and limits
-     * - Modify connection strings dynamically
-     * - Validate authentication credentials
-     *
-     * This function is called for each new MCP client connection. For stdio
-     * transport, this is called once at server startup. For HTTP transport,
-     * this is called for each new session.
+     * @deprecated This method will be removed in a future version. Use `createServer({ userConfig: MyCustomUserConfig})` instead.
      */
-    createSessionConfig?: CreateSessionConfigFn;
+    createSessionConfig?: CreateSessionConfigFn<TUserConfig>;
     /**
+     * @deprecated Use `createServer({ sessionOptions: {apiClient: MyCustomApiClient} })` instead
      * An optional factory function to generates an instance of
      * `ApiClient`. When not provided, MongoDB MCP Server uses an
      * internal implementation to create the API client.
@@ -173,25 +133,57 @@ export type TransportRunnerConfig = {
      */
     createApiClient?: ApiClientFactoryFn;
 };
-export declare abstract class TransportRunnerBase {
+export declare abstract class TransportRunnerBase<TUserConfig extends UserConfig = UserConfig, TContext = unknown> {
     logger: LoggerBase;
     deviceId: DeviceId;
-    protected readonly userConfig: UserConfig;
-    private readonly createConnectionManager;
-    private readonly connectionErrorHandler;
-    private readonly createAtlasLocalClient;
-    private readonly telemetryProperties;
-    private readonly tools?;
-    private readonly createSessionConfig?;
-    private readonly createApiClient;
-    protected constructor({ userConfig, createConnectionManager, connectionErrorHandler, createAtlasLocalClient, additionalLoggers, telemetryProperties, tools, createSessionConfig, createApiClient, }: TransportRunnerConfig);
+    /** Base user configuration for the server. */
+    protected readonly userConfig: TUserConfig;
+    /** @deprecated This method will be removed in a future version. Extend `StreamableHttpRunner` and override `createServerForRequest` instead. */
+    protected readonly createConnectionManager: ConnectionManagerFactoryFn;
+    /** @deprecated This method will be removed in a future version. Extend `StreamableHttpRunner` and override `createServerForRequest` instead. */
+    protected readonly connectionErrorHandler: ConnectionErrorHandler;
+    /** @deprecated This method will be removed in a future version. Extend `StreamableHttpRunner` and override `createServerForRequest` instead. */
+    protected readonly createAtlasLocalClient: AtlasLocalClientFactoryFn;
+    /** @deprecated This field will be removed in a future version. Use `start({ serverOptions: {telemetryProperties: MyCustomTelemetryProperties} })` instead. */
+    protected readonly telemetryProperties: Partial<CommonProperties>;
+    /** @deprecated This field will be removed in a future version. Use `start({ serverOptions: {tools: [...AllTools, MyCustomTool]} })` instead. */
+    protected readonly tools?: AnyToolClass[];
+    /** @deprecated This method will be removed in a future version. Extend `StreamableHttpRunner` and override `createServerForRequest` instead. */
+    protected readonly createSessionConfig?: CreateSessionConfigFn<TUserConfig>;
+    /** @deprecated This method will be removed in a future version. Extend `StreamableHttpRunner` and override `createServerForRequest` instead. */
+    protected readonly createApiClient: ApiClientFactoryFn;
+    protected constructor({ userConfig, createConnectionManager, connectionErrorHandler, createAtlasLocalClient, additionalLoggers, telemetryProperties, tools, createSessionConfig, createApiClient, }: TransportRunnerConfig<TUserConfig>);
+    /**
+     * Creates a new MCP server instance with the provided configuration.
+     * This method handles server instantiation but does NOT perform session config resolution.
+     *
+     * @param config - Configuration object containing userConfig and optional serverOptions
+     * @returns A configured Server instance
+     */
+    protected createServer({ userConfig, serverOptions, sessionOptions, logger, }?: {
+        userConfig?: TUserConfig;
+        logger?: CompositeLogger;
+        serverOptions?: CustomizableServerOptions<TUserConfig, TContext>;
+        sessionOptions?: CustomizableSessionOptions<TUserConfig>;
+    }): Promise<Server<TUserConfig, TContext>>;
+    /**
+     * @deprecated Remove all session hooks and use `start({serverOptions, sessionOptions})` or override `StreamableHttpRunner.createServerForRequest` instead. This method will be removed in a future version.
+     *
+     * Creates a new MCP server instance and handles session config resolution.
+     * For new code, prefer using `createServer` with pre-resolved configuration.
+     */
     protected setupServer(request?: RequestContext, { serverOptions, }?: {
-        serverOptions?: Pick<ServerOptions, "uiRegistry">;
-    }): Promise<Server>;
-    abstract start(): Promise<void>;
+        serverOptions?: CustomizableServerOptions<TUserConfig, TContext>;
+    }): Promise<Server<TUserConfig, TContext>>;
+    abstract start({ serverOptions, sessionOptions, }: {
+        /** Upstream `serverOptions` passed from running `runner.start({ serverOptions })` method */
+        serverOptions?: ServerOptions<TUserConfig, TContext>;
+        /** Upstream `sessionOptions` passed from running `runner.start({ sessionOptions })` method */
+        sessionOptions?: SessionOptions<TUserConfig>;
+    }): Promise<void>;
     abstract closeTransport(): Promise<void>;
     close(): Promise<void>;
-    private static getInstructions;
+    protected static getInstructions(config: UserConfig): string;
 }
 export {};
 //# sourceMappingURL=base.d.ts.map

package/dist/cjs/transports/base.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../src/transports/base.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gCAAgC,CAAC;AAEjE,OAAO,EAAE,MAAM,EAAE,KAAK,aAAa,EAAE,MAAM,cAAc,CAAC;AAI1D,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAGtD,OAAO,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAElD,OAAO,EAA8B,KAAK,0BAA0B,EAAE,MAAM,gCAAgC,CAAC;AAC7G,OAAO,EACH,KAAK,sBAAsB,EAE9B,MAAM,qCAAqC,CAAC;AAC7C,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAE9D,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AAGzE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAElD,OAAO,KAAK,EAAa,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAIlF,MAAM,MAAM,cAAc,GAAG;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC,CAAC;IACxD,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC,CAAC;CACzD,CAAC;AAEF;;;;;;;;;;;;;;;;GAgBG;AACH,KAAK,qBAAqB,GAAG,CAAC,OAAO,EAAE;IACnC,UAAU,EAAE,UAAU,CAAC;IACvB,OAAO,CAAC,EAAE,cAAc,CAAC;CAC5B,KAAK,OAAO,CAAC,UAAU,CAAC,GAAG,UAAU,CAAC;AAEvC;;;;;;;;;;;;;;GAcG;AACH,MAAM,MAAM,qBAAqB,GAAG;IAChC;;;;;;;;;;;;;;;OAeG;IACH,UAAU,EAAE,UAAU,CAAC;IAEvB;;;;;;;OAOG;IACH,uBAAuB,CAAC,EAAE,0BAA0B,CAAC;IAErD;;;;;;;;;;OAUG;IACH,sBAAsB,CAAC,EAAE,sBAAsB,CAAC;IAEhD;;;;OAIG;IACH,sBAAsB,CAAC,EAAE,yBAAyB,CAAC;IAEnD;;;;;;;OAOG;IACH,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC;IAEjC;;;;OAIG;IACH,mBAAmB,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAEhD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAsCG;IACH,KAAK,CAAC,EAAE,SAAS,EAAE,CAAC;IAEpB;;;;;;;;;;;;;OAaG;IACH,mBAAmB,CAAC,EAAE,qBAAqB,CAAC;IAE5C;;;;;;;OAOG;IACH,eAAe,CAAC,EAAE,kBAAkB,CAAC;CACxC,CAAC;AAEF,8BAAsB,mBAAmB;IAC9B,MAAM,EAAE,UAAU,CAAC;IACnB,QAAQ,EAAE,QAAQ,CAAC;IAC1B,SAAS,CAAC,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC;IAC1C,OAAO,CAAC,QAAQ,CAAC,uBAAuB,CAA6B;IACrE,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAAyB;IAChE,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAA4B;IACnE,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAA4B;IAChE,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAc;IACrC,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAwB;IAC7D,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAqB;IAErD,SAAS,aAAa,EAClB,UAAU,EACV,uBAAoD,EACpD,sBAAsD,EACtD,sBAAsD,EACtD,iBAAsB,EACtB,mBAAwB,EACxB,KAAK,EACL,mBAAmB,EACnB,eAAsC,GACzC,EAAE,qBAAqB;cAiCR,WAAW,CACvB,OAAO,CAAC,EAAE,cAAc,EACxB,EACI,aAAa,GAChB,GAAE;QACC,aAAa,CAAC,EAAE,IAAI,CAAC,aAAa,EAAE,YAAY,CAAC,CAAC;KAChD,GACP,OAAO,CAAC,MAAM,CAAC;IAqFlB,QAAQ,CAAC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAE/B,QAAQ,CAAC,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;IAElC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAQ5B,OAAO,CAAC,MAAM,CAAC,eAAe;CAiBjC"}
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../src/transports/base.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gCAAgC,CAAC;AAEjE,OAAO,EAAE,KAAK,YAAY,EAAE,MAAM,EAAE,KAAK,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7E,OAAO,EAAW,KAAK,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAGpE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EAAE,eAAe,EAAwC,MAAM,qBAAqB,CAAC;AAE5F,OAAO,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AAElD,OAAO,EAAkC,KAAK,0BAA0B,EAAE,MAAM,gCAAgC,CAAC;AACjH,OAAO,EACH,KAAK,sBAAsB,EAE9B,MAAM,qCAAqC,CAAC;AAC7C,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAE9D,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AAIzE,OAAO,EAAa,KAAK,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAIlF;;;GAGG;AACH,MAAM,MAAM,cAAc,GAAG;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC,CAAC;IACxD,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,EAAE,GAAG,SAAS,CAAC,CAAC;CACzD,CAAC;AAEF,MAAM,MAAM,0BAA0B,CAAC,WAAW,SAAS,UAAU,GAAG,UAAU,IAAI,OAAO,CACzF,IAAI,CACA,cAAc,CAAC,WAAW,CAAC,EAC3B,YAAY,GAAG,WAAW,GAAG,kBAAkB,GAAG,mBAAmB,GAAG,wBAAwB,CACnG,GAAG;IACA,kBAAkB,CAAC,EAAE,MAAM,MAAM,CAAC;CACrC,CACJ,CAAC;AAEF,MAAM,MAAM,yBAAyB,CAAC,WAAW,SAAS,UAAU,GAAG,UAAU,EAAE,QAAQ,GAAG,OAAO,IAAI,OAAO,CAC5G,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,QAAQ,CAAC,EAAE,YAAY,GAAG,OAAO,GAAG,aAAa,GAAG,aAAa,CAAC,CACrG,GAAG;IACA;;;;OAIG;IACH,mBAAmB,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;CACnD,CAAC;AAEF;;;;;;;;;;;;;;;;GAgBG;AACH,KAAK,qBAAqB,CAAC,WAAW,SAAS,UAAU,GAAG,UAAU,IAAI,CAAC,OAAO,EAAE;IAChF,UAAU,EAAE,WAAW,CAAC;IACxB,OAAO,CAAC,EAAE,cAAc,CAAC;CAC5B,KAAK,OAAO,CAAC,WAAW,CAAC,GAAG,WAAW,CAAC;AAEzC;;;;;;;;;;;;;;;;GAgBG;AACH,MAAM,MAAM,qBAAqB,CAAC,WAAW,SAAS,UAAU,GAAG,UAAU,IAAI;IAC7E;;;;;;;;;;;;;;;OAeG;IACH,UAAU,EAAE,WAAW,CAAC;IAExB;;;;;;;;OAQG;IACH,uBAAuB,CAAC,EAAE,0BAA0B,CAAC;IAErD,mHAAmH;IACnH,sBAAsB,CAAC,EAAE,sBAAsB,CAAC;IAEhD;;;;;OAKG;IACH,sBAAsB,CAAC,EAAE,yBAAyB,CAAC;IAEnD;;;;;;;OAOG;IACH,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC;IAEjC;;OAEG;IACH,mBAAmB,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAEhD,sGAAsG;IACtG,KAAK,CAAC,EAAE,YAAY,EAAE,CAAC;IAEvB;;OAEG;IACH,mBAAmB,CAAC,EAAE,qBAAqB,CAAC,WAAW,CAAC,CAAC;IAEzD;;;;;;;;OAQG;IACH,eAAe,CAAC,EAAE,kBAAkB,CAAC;CACxC,CAAC;AAEF,8BAAsB,mBAAmB,CAAC,WAAW,SAAS,UAAU,GAAG,UAAU,EAAE,QAAQ,GAAG,OAAO;IAC9F,MAAM,EAAE,UAAU,CAAC;IACnB,QAAQ,EAAE,QAAQ,CAAC;IAC1B,8CAA8C;IAC9C,SAAS,CAAC,QAAQ,CAAC,UAAU,EAAE,WAAW,CAAC;IAC3C,gJAAgJ;IAChJ,SAAS,CAAC,QAAQ,CAAC,uBAAuB,EAAE,0BAA0B,CAAC;IACvE,gJAAgJ;IAChJ,SAAS,CAAC,QAAQ,CAAC,sBAAsB,EAAE,sBAAsB,CAAC;IAClE,gJAAgJ;IAChJ,SAAS,CAAC,QAAQ,CAAC,sBAAsB,EAAE,yBAAyB,CAAC;IACrE,8JAA8J;IAC9J,SAAS,CAAC,QAAQ,CAAC,mBAAmB,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAClE,gJAAgJ;IAChJ,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,YAAY,EAAE,CAAC;IAC1C,gJAAgJ;IAChJ,SAAS,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,qBAAqB,CAAC,WAAW,CAAC,CAAC;IAC5E,gJAAgJ;IAChJ,SAAS,CAAC,QAAQ,CAAC,eAAe,EAAE,kBAAkB,CAAC;IAEvD,SAAS,aAAa,EAClB,UAAU,EACV,uBAAwD,EACxD,sBAAsD,EACtD,sBAAsD,EACtD,iBAAsB,EACtB,mBAAwB,EACxB,KAAK,EACL,mBAAmB,EACnB,eAAwC,GAC3C,EAAE,qBAAqB,CAAC,WAAW,CAAC;IAiCrC;;;;;;OAMG;cACa,YAAY,CAAC,EACzB,UAA4B,EAC5B,aAAa,EACb,cAAc,EACd,MAAyC,GAC5C,GAAE;QACC,UAAU,CAAC,EAAE,WAAW,CAAC;QACzB,MAAM,CAAC,EAAE,eAAe,CAAC;QACzB,aAAa,CAAC,EAAE,yBAAyB,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QACjE,cAAc,CAAC,EAAE,0BAA0B,CAAC,WAAW,CAAC,CAAC;KACvD,GAAG,OAAO,CAAC,MAAM,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;IA2E/C;;;;;OAKG;cACa,WAAW,CACvB,OAAO,CAAC,EAAE,cAAc,EACxB,EACI,aAAa,GAChB,GAAE;QACC,aAAa,CAAC,EAAE,yBAAyB,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;KAC/D,GACP,OAAO,CAAC,MAAM,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;IAqCzC,QAAQ,CAAC,KAAK,CAAC,EACX,aAAa,EACb,cAAc,GACjB,EAAE;QACC,4FAA4F;QAC5F,aAAa,CAAC,EAAE,aAAa,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QACrD,8FAA8F;QAC9F,cAAc,CAAC,EAAE,cAAc,CAAC,WAAW,CAAC,CAAC;KAChD,GAAG,OAAO,CAAC,IAAI,CAAC;IAEjB,QAAQ,CAAC,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;IAElC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAQ5B,SAAS,CAAC,MAAM,CAAC,eAAe,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM;CAiB/D"}

package/dist/cjs/transports/base.js

@@ -50,8 +50,9 @@ const atlasLocal_js_1 = require("../common/atlasLocal.js");
 const vectorSearchEmbeddingsManager_js_1 = require("../common/search/vectorSearchEmbeddingsManager.js");
 const configOverrides_js_1 = require("../common/config/configOverrides.js");
 const apiClient_js_1 = require("../common/atlas/apiClient.js");
+const apiClient_js_2 = require("../common/atlas/apiClient.js");
 class TransportRunnerBase {
-    constructor({ userConfig, createConnectionManager = connectionManager_js_1.createMCPConnectionManager, connectionErrorHandler = connectionErrorHandler_js_1.connectionErrorHandler, createAtlasLocalClient = atlasLocal_js_1.defaultCreateAtlasLocalClient, additionalLoggers = [], telemetryProperties = {}, tools, createSessionConfig, createApiClient = apiClient_js_1.createAtlasApiClient, }) {
+    constructor({ userConfig, createConnectionManager = connectionManager_js_1.defaultCreateConnectionManager, connectionErrorHandler = connectionErrorHandler_js_1.connectionErrorHandler, createAtlasLocalClient = atlasLocal_js_1.defaultCreateAtlasLocalClient, additionalLoggers = [], telemetryProperties = {}, tools, createSessionConfig, createApiClient = apiClient_js_2.defaultCreateApiClient, }) {
         this.userConfig = userConfig;
         this.createConnectionManager = createConnectionManager;
         this.connectionErrorHandler = connectionErrorHandler;
@@ -75,52 +76,46 @@ class TransportRunnerBase {
         this.logger = new logger_js_1.CompositeLogger(...loggers);
         this.deviceId = deviceId_js_1.DeviceId.create(this.logger);
     }
-    async setupServer(request, { serverOptions, } = {}) {
-        let userConfig = this.userConfig;
-        if (this.createSessionConfig) {
-            userConfig = await this.createSessionConfig({ userConfig, request });
-        }
-        else {
-            userConfig = (0, configOverrides_js_1.applyConfigOverrides)({ baseConfig: this.userConfig, request });
-        }
+    /**
+     * Creates a new MCP server instance with the provided configuration.
+     * This method handles server instantiation but does NOT perform session config resolution.
+     *
+     * @param config - Configuration object containing userConfig and optional serverOptions
+     * @returns A configured Server instance
+     */
+    async createServer({ userConfig = this.userConfig, serverOptions, sessionOptions, logger = new logger_js_1.CompositeLogger(this.logger), } = {}) {
         const mcpServer = new mcp_js_1.McpServer({
             name: packageInfo_js_1.packageInfo.mcpServerName,
             version: packageInfo_js_1.packageInfo.version,
         }, {
             instructions: TransportRunnerBase.getInstructions(userConfig),
         });
-        const logger = new logger_js_1.CompositeLogger(this.logger);
         const exportsManager = exportsManager_js_1.ExportsManager.init(userConfig, logger);
-        const connectionManager = await this.createConnectionManager({
-            logger,
-            userConfig,
-            deviceId: this.deviceId,
-        });
-        let apiClient;
-        if (userConfig.apiClientId && userConfig.apiClientSecret) {
-            apiClient = this.createApiClient({
+        const connectionManager = sessionOptions?.connectionManager ??
+            (await this.createConnectionManager({ logger: logger, deviceId: this.deviceId, userConfig }));
+        const apiClient = userConfig.apiClientId && userConfig.apiClientSecret
+            ? new apiClient_js_1.ApiClient({
                 baseUrl: userConfig.apiBaseUrl,
                 credentials: {
                     clientId: userConfig.apiClientId,
                     clientSecret: userConfig.apiClientSecret,
                 },
-                requestContext: request,
-            }, logger);
-        }
+            }, logger)
+            : undefined;
         const session = new session_js_1.Session({
             userConfig,
-            atlasLocalClient: await this.createAtlasLocalClient({ logger }),
+            atlasLocalClient: sessionOptions?.atlasLocalClient ?? (await this.createAtlasLocalClient({ logger: this.logger })),
             logger,
+            connectionErrorHandler: sessionOptions?.connectionErrorHandler ?? this.connectionErrorHandler,
             exportsManager,
             connectionManager,
             keychain: keychain_js_1.Keychain.root,
             vectorSearchEmbeddingsManager: new vectorSearchEmbeddingsManager_js_1.VectorSearchEmbeddingsManager(userConfig, connectionManager),
-            apiClient,
+            apiClient: sessionOptions?.apiClient ?? apiClient,
         });
         const telemetry = telemetry_js_1.Telemetry.create(session, userConfig, this.deviceId, {
-            commonProperties: this.telemetryProperties,
+            commonProperties: serverOptions?.telemetryProperties ?? this.telemetryProperties,
         });
-        const elicitation = new elicitation_js_1.Elicitation({ server: mcpServer.server });
         let uiRegistry = serverOptions?.uiRegistry;
         if (!uiRegistry && userConfig.previewFeatures.includes("mcpUI")) {
             const uiRegistryModule = await Promise.resolve().then(() => __importStar(require("../ui/registry/registry.js")));
@@ -131,10 +126,11 @@ class TransportRunnerBase {
             session,
             telemetry,
             userConfig,
-            connectionErrorHandler: this.connectionErrorHandler,
-            elicitation,
-            tools: this.tools,
+            connectionErrorHandler: sessionOptions?.connectionErrorHandler ?? this.connectionErrorHandler,
+            elicitation: serverOptions?.elicitation ?? new elicitation_js_1.Elicitation({ server: mcpServer.server }),
+            tools: serverOptions?.tools ?? this.tools,
             uiRegistry,
+            toolContext: serverOptions?.toolContext,
         });
         // We need to create the MCP logger after the server is constructed
         // because it needs the server instance
@@ -143,6 +139,43 @@ class TransportRunnerBase {
         }
         return result;
     }
+    /**
+     * @deprecated Remove all session hooks and use `start({serverOptions, sessionOptions})` or override `StreamableHttpRunner.createServerForRequest` instead. This method will be removed in a future version.
+     *
+     * Creates a new MCP server instance and handles session config resolution.
+     * For new code, prefer using `createServer` with pre-resolved configuration.
+     */
+    async setupServer(request, { serverOptions, } = {}) {
+        let userConfig = this.userConfig;
+        if (this.createSessionConfig) {
+            userConfig = await this.createSessionConfig({ userConfig, request });
+        }
+        else {
+            userConfig = (0, configOverrides_js_1.applyConfigOverrides)({ baseConfig: this.userConfig, request });
+        }
+        return this.createServer({
+            userConfig,
+            serverOptions,
+            sessionOptions: {
+                connectionManager: await this.createConnectionManager({
+                    logger: this.logger,
+                    deviceId: this.deviceId,
+                    userConfig,
+                }),
+                atlasLocalClient: await this.createAtlasLocalClient({ logger: this.logger }),
+                apiClient: userConfig.apiClientId && userConfig.apiClientSecret
+                    ? this.createApiClient({
+                        baseUrl: userConfig.apiBaseUrl,
+                        credentials: {
+                            clientId: userConfig.apiClientId,
+                            clientSecret: userConfig.apiClientSecret,
+                        },
+                        requestContext: request,
+                    }, this.logger)
+                    : undefined,
+            },
+        });
+    }
     async close() {
         try {
             await this.closeTransport();

package/dist/cjs/transports/base.js.map

@@ -1 +1 @@
-{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../src/transports/base.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AACA,6DAAuD;AACvD,4CAA0D;AAC1D,qDAA+C;AAC/C,4DAAsD;AACtD,oEAAoE;AAEpE,mDAA4F;AAC5F,mEAA6D;AAC7D,wDAAkD;AAClD,uDAAiD;AACjD,yEAA6G;AAC7G,mFAG6C;AAE7C,sDAAgD;AAEhD,2DAAwE;AACxE,wGAAkG;AAElG,4EAA2E;AAE3E,+DAAoE;AAmLpE,MAAsB,mBAAmB;IAYrC,YAAsB,EAClB,UAAU,EACV,uBAAuB,GAAG,iDAA0B,EACpD,sBAAsB,GAAG,kDAA6B,EACtD,sBAAsB,GAAG,6CAA6B,EACtD,iBAAiB,GAAG,EAAE,EACtB,mBAAmB,GAAG,EAAE,EACxB,KAAK,EACL,mBAAmB,EACnB,eAAe,GAAG,mCAAoB,GAClB;QACpB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,uBAAuB,GAAG,uBAAuB,CAAC;QACvD,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;QACrD,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;QACrD,IAAI,CAAC,mBAAmB,GAAG,mBAAmB,CAAC;QAC/C,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,mBAAmB,GAAG,mBAAmB,CAAC;QAC/C,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,MAAM,OAAO,GAAiB,CAAC,GAAG,iBAAiB,CAAC,CAAC;QACrD,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC7C,OAAO,CAAC,IAAI,CAAC,IAAI,yBAAa,CAAC,sBAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;QACnD,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3C,OAAO,CAAC,IAAI,CACR,IAAI,sBAAU,CACV,IAAI,CAAC,UAAU,CAAC,OAAO,EACvB,CAAC,GAAG,EAAE,EAAE;gBACJ,8EAA8E;gBAC9E,sCAAsC;gBACtC,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,GAAG,CAAC,CAAC;gBACtD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACpB,CAAC,EACD,sBAAQ,CAAC,IAAI,CAChB,CACJ,CAAC;QACN,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,IAAI,2BAAe,CAAC,GAAG,OAAO,CAAC,CAAC;QAC9C,IAAI,CAAC,QAAQ,GAAG,sBAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACjD,CAAC;IAES,KAAK,CAAC,WAAW,CACvB,OAAwB,EACxB,EACI,aAAa,MAGb,EAAE;QAEN,IAAI,UAAU,GAAe,IAAI,CAAC,UAAU,CAAC;QAE7C,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC3B,UAAU,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC,CAAC;QACzE,CAAC;aAAM,CAAC;YACJ,UAAU,GAAG,IAAA,yCAAoB,EAAC,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,OAAO,EAAE,CAAC,CAAC;QAChF,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,kBAAS,CAC3B;YACI,IAAI,EAAE,4BAAW,CAAC,aAAa;YAC/B,OAAO,EAAE,4BAAW,CAAC,OAAO;SAC/B,EACD;YACI,YAAY,EAAE,mBAAmB,CAAC,eAAe,CAAC,UAAU,CAAC;SAChE,CACJ,CAAC;QAEF,MAAM,MAAM,GAAG,IAAI,2BAAe,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChD,MAAM,cAAc,GAAG,kCAAc,CAAC,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAC/D,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC;YACzD,MAAM;YACN,UAAU;YACV,QAAQ,EAAE,IAAI,CAAC,QAAQ;SAC1B,CAAC,CAAC;QAEH,IAAI,SAAgC,CAAC;QACrC,IAAI,UAAU,CAAC,WAAW,IAAI,UAAU,CAAC,eAAe,EAAE,CAAC;YACvD,SAAS,GAAG,IAAI,CAAC,eAAe,CAC5B;gBACI,OAAO,EAAE,UAAU,CAAC,UAAU;gBAC9B,WAAW,EAAE;oBACT,QAAQ,EAAE,UAAU,CAAC,WAAW;oBAChC,YAAY,EAAE,UAAU,CAAC,eAAe;iBAC3C;gBACD,cAAc,EAAE,OAAO;aAC1B,EACD,MAAM,CACT,CAAC;QACN,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,oBAAO,CAAC;YACxB,UAAU;YACV,gBAAgB,EAAE,MAAM,IAAI,CAAC,sBAAsB,CAAC,EAAE,MAAM,EAAE,CAAC;YAC/D,MAAM;YACN,cAAc;YACd,iBAAiB;YACjB,QAAQ,EAAE,sBAAQ,CAAC,IAAI;YACvB,6BAA6B,EAAE,IAAI,gEAA6B,CAAC,UAAU,EAAE,iBAAiB,CAAC;YAC/F,SAAS;SACZ,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,wBAAS,CAAC,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,CAAC,QAAQ,EAAE;YACnE,gBAAgB,EAAE,IAAI,CAAC,mBAAmB;SAC7C,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG,IAAI,4BAAW,CAAC,EAAE,MAAM,EAAE,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;QAElE,IAAI,UAAU,GAA2B,aAAa,EAAE,UAAU,CAAC;QACnE,IAAI,CAAC,UAAU,IAAI,UAAU,CAAC,eAAe,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC9D,MAAM,gBAAgB,GAAG,wDAAa,4BAA4B,GAAC,CAAC;YACpE,UAAU,GAAG,IAAI,gBAAgB,CAAC,UAAU,EAAE,CAAC;QACnD,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,kBAAM,CAAC;YACtB,SAAS;YACT,OAAO;YACP,SAAS;YACT,UAAU;YACV,sBAAsB,EAAE,IAAI,CAAC,sBAAsB;YACnD,WAAW;YACX,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,UAAU;SACb,CAAC,CAAC;QAEH,mEAAmE;QACnE,uCAAuC;QACvC,IAAI,UAAU,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACrC,MAAM,CAAC,SAAS,CAAC,IAAI,qBAAS,CAAC,MAAM,EAAE,sBAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;QAC3D,CAAC;QAED,OAAO,MAAM,CAAC;IAClB,CAAC;IAMD,KAAK,CAAC,KAAK;QACP,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAChC,CAAC;gBAAS,CAAC;YACP,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QAC1B,CAAC;IACL,CAAC;IAEO,MAAM,CAAC,eAAe,CAAC,MAAkB;QAC7C,IAAI,YAAY,GAAG;;SAElB,CAAC;QACF,IAAI,MAAM,CAAC,gBAAgB,EAAE,CAAC;YAC1B,YAAY,IAAI;;aAEf,CAAC;QACN,CAAC;QAED,IAAI,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;YAC/C,YAAY,IAAI;+EACmD,CAAC;QACxE,CAAC;QAED,OAAO,YAAY,CAAC;IACxB,CAAC;CACJ;AAhLD,kDAgLC"}
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../src/transports/base.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AACA,6DAAuD;AACvD,4CAA6E;AAC7E,qDAAoE;AACpE,4DAAsD;AACtD,oEAAoE;AAEpE,mDAA4F;AAC5F,mEAA6D;AAC7D,wDAAkD;AAClD,uDAAiD;AACjD,yEAAiH;AACjH,mFAG6C;AAE7C,sDAAgD;AAEhD,2DAAwE;AACxE,wGAAkG;AAClG,4EAA2E;AAC3E,+DAAkF;AAClF,+DAAsE;AAmJtE,MAAsB,mBAAmB;IAoBrC,YAAsB,EAClB,UAAU,EACV,uBAAuB,GAAG,qDAA8B,EACxD,sBAAsB,GAAG,kDAA6B,EACtD,sBAAsB,GAAG,6CAA6B,EACtD,iBAAiB,GAAG,EAAE,EACtB,mBAAmB,GAAG,EAAE,EACxB,KAAK,EACL,mBAAmB,EACnB,eAAe,GAAG,qCAAsB,GACP;QACjC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,uBAAuB,GAAG,uBAAuB,CAAC;QACvD,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;QACrD,IAAI,CAAC,sBAAsB,GAAG,sBAAsB,CAAC;QACrD,IAAI,CAAC,mBAAmB,GAAG,mBAAmB,CAAC;QAC/C,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,mBAAmB,GAAG,mBAAmB,CAAC;QAC/C,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,MAAM,OAAO,GAAiB,CAAC,GAAG,iBAAiB,CAAC,CAAC;QACrD,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC7C,OAAO,CAAC,IAAI,CAAC,IAAI,yBAAa,CAAC,sBAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;QACnD,CAAC;QAED,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3C,OAAO,CAAC,IAAI,CACR,IAAI,sBAAU,CACV,IAAI,CAAC,UAAU,CAAC,OAAO,EACvB,CAAC,GAAG,EAAE,EAAE;gBACJ,8EAA8E;gBAC9E,sCAAsC;gBACtC,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,GAAG,CAAC,CAAC;gBACtD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACpB,CAAC,EACD,sBAAQ,CAAC,IAAI,CAChB,CACJ,CAAC;QACN,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,IAAI,2BAAe,CAAC,GAAG,OAAO,CAAC,CAAC;QAC9C,IAAI,CAAC,QAAQ,GAAG,sBAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACjD,CAAC;IAED;;;;;;OAMG;IACO,KAAK,CAAC,YAAY,CAAC,EACzB,UAAU,GAAG,IAAI,CAAC,UAAU,EAC5B,aAAa,EACb,cAAc,EACd,MAAM,GAAG,IAAI,2BAAe,CAAC,IAAI,CAAC,MAAM,CAAC,MAMzC,EAAE;QACF,MAAM,SAAS,GAAG,IAAI,kBAAS,CAC3B;YACI,IAAI,EAAE,4BAAW,CAAC,aAAa;YAC/B,OAAO,EAAE,4BAAW,CAAC,OAAO;SAC/B,EACD;YACI,YAAY,EAAE,mBAAmB,CAAC,eAAe,CAAC,UAAU,CAAC;SAChE,CACJ,CAAC;QAEF,MAAM,cAAc,GAAG,kCAAc,CAAC,IAAI,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAE/D,MAAM,iBAAiB,GACnB,cAAc,EAAE,iBAAiB;YACjC,CAAC,MAAM,IAAI,CAAC,uBAAuB,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;QAElG,MAAM,SAAS,GACX,UAAU,CAAC,WAAW,IAAI,UAAU,CAAC,eAAe;YAChD,CAAC,CAAC,IAAI,wBAAS,CACT;gBACI,OAAO,EAAE,UAAU,CAAC,UAAU;gBAC9B,WAAW,EAAE;oBACT,QAAQ,EAAE,UAAU,CAAC,WAAW;oBAChC,YAAY,EAAE,UAAU,CAAC,eAAe;iBAC3C;aACJ,EACD,MAAM,CACT;YACH,CAAC,CAAC,SAAS,CAAC;QAEpB,MAAM,OAAO,GAAG,IAAI,oBAAO,CAAC;YACxB,UAAU;YACV,gBAAgB,EACZ,cAAc,EAAE,gBAAgB,IAAI,CAAC,MAAM,IAAI,CAAC,sBAAsB,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;YACpG,MAAM;YACN,sBAAsB,EAAE,cAAc,EAAE,sBAAsB,IAAI,IAAI,CAAC,sBAAsB;YAC7F,cAAc;YACd,iBAAiB;YACjB,QAAQ,EAAE,sBAAQ,CAAC,IAAI;YACvB,6BAA6B,EAAE,IAAI,gEAA6B,CAAC,UAAU,EAAE,iBAAiB,CAAC;YAC/F,SAAS,EAAE,cAAc,EAAE,SAAS,IAAI,SAAS;SACpD,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,wBAAS,CAAC,MAAM,CAAC,OAAO,EAAE,UAAU,EAAE,IAAI,CAAC,QAAQ,EAAE;YACnE,gBAAgB,EAAE,aAAa,EAAE,mBAAmB,IAAI,IAAI,CAAC,mBAAmB;SACnF,CAAC,CAAC;QAEH,IAAI,UAAU,GAA2B,aAAa,EAAE,UAAU,CAAC;QACnE,IAAI,CAAC,UAAU,IAAI,UAAU,CAAC,eAAe,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC9D,MAAM,gBAAgB,GAAG,wDAAa,4BAA4B,GAAC,CAAC;YACpE,UAAU,GAAG,IAAI,gBAAgB,CAAC,UAAU,EAAE,CAAC;QACnD,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,kBAAM,CAAwB;YAC7C,SAAS;YACT,OAAO;YACP,SAAS;YACT,UAAU;YACV,sBAAsB,EAAE,cAAc,EAAE,sBAAsB,IAAI,IAAI,CAAC,sBAAsB;YAC7F,WAAW,EAAE,aAAa,EAAE,WAAW,IAAI,IAAI,4BAAW,CAAC,EAAE,MAAM,EAAE,SAAS,CAAC,MAAM,EAAE,CAAC;YACxF,KAAK,EAAE,aAAa,EAAE,KAAK,IAAI,IAAI,CAAC,KAAK;YACzC,UAAU;YACV,WAAW,EAAE,aAAa,EAAE,WAAW;SAC1C,CAAC,CAAC;QAEH,mEAAmE;QACnE,uCAAuC;QACvC,IAAI,UAAU,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACrC,MAAM,CAAC,SAAS,CAAC,IAAI,qBAAS,CAAC,MAAM,EAAE,sBAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;QAC3D,CAAC;QAED,OAAO,MAAM,CAAC;IAClB,CAAC;IAED;;;;;OAKG;IACO,KAAK,CAAC,WAAW,CACvB,OAAwB,EACxB,EACI,aAAa,MAGb,EAAE;QAEN,IAAI,UAAU,GAAgB,IAAI,CAAC,UAAU,CAAC;QAE9C,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC3B,UAAU,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC,CAAC;QACzE,CAAC;aAAM,CAAC;YACJ,UAAU,GAAG,IAAA,yCAAoB,EAAC,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,OAAO,EAAE,CAAC,CAAC;QAChF,CAAC;QAED,OAAO,IAAI,CAAC,YAAY,CAAC;YACrB,UAAU;YACV,aAAa;YACb,cAAc,EAAE;gBACZ,iBAAiB,EAAE,MAAM,IAAI,CAAC,uBAAuB,CAAC;oBAClD,MAAM,EAAE,IAAI,CAAC,MAAM;oBACnB,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,UAAU;iBACb,CAAC;gBACF,gBAAgB,EAAE,MAAM,IAAI,CAAC,sBAAsB,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;gBAC5E,SAAS,EACL,UAAU,CAAC,WAAW,IAAI,UAAU,CAAC,eAAe;oBAChD,CAAC,CAAC,IAAI,CAAC,eAAe,CAChB;wBACI,OAAO,EAAE,UAAU,CAAC,UAAU;wBAC9B,WAAW,EAAE;4BACT,QAAQ,EAAE,UAAU,CAAC,WAAW;4BAChC,YAAY,EAAE,UAAU,CAAC,eAAe;yBAC3C;wBACD,cAAc,EAAE,OAAO;qBAC1B,EACD,IAAI,CAAC,MAAM,CACd;oBACH,CAAC,CAAC,SAAS;aACtB;SACJ,CAAC,CAAC;IACP,CAAC;IAcD,KAAK,CAAC,KAAK;QACP,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAChC,CAAC;gBAAS,CAAC;YACP,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QAC1B,CAAC;IACL,CAAC;IAES,MAAM,CAAC,eAAe,CAAC,MAAkB;QAC/C,IAAI,YAAY,GAAG;;SAElB,CAAC;QACF,IAAI,MAAM,CAAC,gBAAgB,EAAE,CAAC;YAC1B,YAAY,IAAI;;aAEf,CAAC;QACN,CAAC;QAED,IAAI,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;YAC/C,YAAY,IAAI;+EACmD,CAAC;QACxE,CAAC;QAED,OAAO,YAAY,CAAC;IACxB,CAAC;CACJ;AAlPD,kDAkPC"}

package/dist/cjs/transports/dryModeRunner.d.ts

@@ -1,4 +1,5 @@
-import { TransportRunnerBase, type TransportRunnerConfig } from "./base.js";
+import { type CustomizableSessionOptions, TransportRunnerBase, type TransportRunnerConfig } from "./base.js";
+import type { CustomizableServerOptions } from "../lib.js";
 export type DryRunModeTestHelpers = {
     logger: {
         log(this: void, message: string): void;
@@ -10,7 +11,10 @@ export declare class DryRunModeRunner extends TransportRunnerBase {
     private server;
     private consoleLogger;
     constructor({ logger, ...transportRunnerConfig }: DryRunModeRunnerConfig);
-    start(): Promise<void>;
+    start({ serverOptions, sessionOptions, }?: {
+        serverOptions?: CustomizableServerOptions;
+        sessionOptions?: CustomizableSessionOptions;
+    }): Promise<void>;
     closeTransport(): Promise<void>;
     private dumpConfig;
     private dumpTools;

package/dist/cjs/transports/dryModeRunner.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"dryModeRunner.d.ts","sourceRoot":"","sources":["../../../src/transports/dryModeRunner.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,KAAK,qBAAqB,EAAE,MAAM,WAAW,CAAC;AAG5E,MAAM,MAAM,qBAAqB,GAAG;IAChC,MAAM,EAAE;QACJ,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QACvC,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;KAC5C,CAAC;CACL,CAAC;AAEF,KAAK,sBAAsB,GAAG,qBAAqB,GAAG,qBAAqB,CAAC;AAE5E,qBAAa,gBAAiB,SAAQ,mBAAmB;IACrD,OAAO,CAAC,MAAM,CAAqB;IACnC,OAAO,CAAC,aAAa,CAAkC;gBAE3C,EAAE,MAAM,EAAE,GAAG,qBAAqB,EAAE,EAAE,sBAAsB;IAKzD,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAStB,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;IAI9C,OAAO,CAAC,UAAU;IAKlB,OAAO,CAAC,SAAS;CAWpB"}
+{"version":3,"file":"dryModeRunner.d.ts","sourceRoot":"","sources":["../../../src/transports/dryModeRunner.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,0BAA0B,EAAE,mBAAmB,EAAE,KAAK,qBAAqB,EAAE,MAAM,WAAW,CAAC;AAE7G,OAAO,KAAK,EAAE,yBAAyB,EAAE,MAAM,WAAW,CAAC;AAE3D,MAAM,MAAM,qBAAqB,GAAG;IAChC,MAAM,EAAE;QACJ,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;QACvC,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAAC;KAC5C,CAAC;CACL,CAAC;AAEF,KAAK,sBAAsB,GAAG,qBAAqB,GAAG,qBAAqB,CAAC;AAE5E,qBAAa,gBAAiB,SAAQ,mBAAmB;IACrD,OAAO,CAAC,MAAM,CAAqB;IACnC,OAAO,CAAC,aAAa,CAAkC;gBAE3C,EAAE,MAAM,EAAE,GAAG,qBAAqB,EAAE,EAAE,sBAAsB;IAKzD,KAAK,CAAC,EACjB,aAAa,EACb,cAAc,GACjB,GAAE;QACC,aAAa,CAAC,EAAE,yBAAyB,CAAC;QAC1C,cAAc,CAAC,EAAE,0BAA0B,CAAC;KAC1C,GAAG,OAAO,CAAC,IAAI,CAAC;IASP,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;IAI9C,OAAO,CAAC,UAAU;IAKlB,OAAO,CAAC,SAAS;CAWpB"}

package/dist/cjs/transports/dryModeRunner.js

@@ -8,8 +8,8 @@ class DryRunModeRunner extends base_js_1.TransportRunnerBase {
         super(transportRunnerConfig);
         this.consoleLogger = logger;
     }
-    async start() {
-        this.server = await this.setupServer();
+    async start({ serverOptions, sessionOptions, } = {}) {
+        this.server = await this.createServer({ serverOptions, sessionOptions });
         const transport = new inMemoryTransport_js_1.InMemoryTransport();
         await this.server.connect(transport);
         this.dumpConfig();

package/dist/cjs/transports/dryModeRunner.js.map

@@ -1 +1 @@
-{"version":3,"file":"dryModeRunner.js","sourceRoot":"","sources":["../../../src/transports/dryModeRunner.ts"],"names":[],"mappings":";;;AAAA,iEAA2D;AAC3D,uCAA4E;AAY5E,MAAa,gBAAiB,SAAQ,6BAAmB;IAIrD,YAAY,EAAE,MAAM,EAAE,GAAG,qBAAqB,EAA0B;QACpE,KAAK,CAAC,qBAAqB,CAAC,CAAC;QAC7B,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC;IAChC,CAAC;IAEQ,KAAK,CAAC,KAAK;QAChB,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;QACvC,MAAM,SAAS,GAAG,IAAI,wCAAiB,EAAE,CAAC;QAE1C,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACrC,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,IAAI,CAAC,SAAS,EAAE,CAAC;IACrB,CAAC;IAEQ,KAAK,CAAC,cAAc;QACzB,MAAM,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC;IAC/B,CAAC;IAEO,UAAU;QACd,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QACzC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACrE,CAAC;IAEO,SAAS;QACb,MAAM,KAAK,GACP,IAAI,CAAC,MAAM,EAAE,KAAK;aACb,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;aAClC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACZ,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,QAAQ,EAAE,IAAI,CAAC,QAAQ;SAC1B,CAAC,CAAC,IAAI,EAAE,CAAC;QAClB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QACzC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IAC3D,CAAC;CACJ;AAtCD,4CAsCC"}
+{"version":3,"file":"dryModeRunner.js","sourceRoot":"","sources":["../../../src/transports/dryModeRunner.ts"],"names":[],"mappings":";;;AAAA,iEAA2D;AAC3D,uCAA6G;AAa7G,MAAa,gBAAiB,SAAQ,6BAAmB;IAIrD,YAAY,EAAE,MAAM,EAAE,GAAG,qBAAqB,EAA0B;QACpE,KAAK,CAAC,qBAAqB,CAAC,CAAC;QAC7B,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC;IAChC,CAAC;IAEQ,KAAK,CAAC,KAAK,CAAC,EACjB,aAAa,EACb,cAAc,MAId,EAAE;QACF,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,EAAE,aAAa,EAAE,cAAc,EAAE,CAAC,CAAC;QACzE,MAAM,SAAS,GAAG,IAAI,wCAAiB,EAAE,CAAC;QAE1C,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACrC,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,IAAI,CAAC,SAAS,EAAE,CAAC;IACrB,CAAC;IAEQ,KAAK,CAAC,cAAc;QACzB,MAAM,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC;IAC/B,CAAC;IAEO,UAAU;QACd,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QACzC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACrE,CAAC;IAEO,SAAS;QACb,MAAM,KAAK,GACP,IAAI,CAAC,MAAM,EAAE,KAAK;aACb,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;aAClC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACZ,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,QAAQ,EAAE,IAAI,CAAC,QAAQ;SAC1B,CAAC,CAAC,IAAI,EAAE,CAAC;QAClB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QACzC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IAC3D,CAAC;CACJ;AA5CD,4CA4CC"}

package/dist/cjs/transports/interfaces/common.d.ts

@@ -0,0 +1,80 @@
+import type { UserConfig } from "../../common/config/userConfig.js";
+import type { LoggerBase } from "../../common/logger.js";
+import type { CommonProperties } from "../../telemetry/types.js";
+import type { ToolClass } from "../../tools/tool.js";
+export type RunnerConfigCommonParameters = {
+    schemaVersion?: undefined | number;
+    /**
+     * Base user configuration for the server.
+     *
+     * Can be generated by parsing CLI arguments, environment variables or
+     * written manually while conforming the interface requirements of
+     * `UserConfig`.
+     *
+     * To parse CLI arguments and environment variables in order to generate a
+     * `UserConfig` object, you can use `parseUserConfig` function, also
+     * exported as `parseUserConfig` through MCP server library
+     * exports.
+     *
+     * Optionally, you can also use `UserConfigSchema` (available through MCP
+     * server library exports) to create a default configuration -
+     * `UserConfigSchema.parse({})`.
+     */
+    userConfig: UserConfig;
+    /**
+     * An optional list of loggers to be used in addition to the default logger
+     * implementations. When not provided, MongoDB MCP Server will not utilize
+     * any loggers other than the default that it works with.
+     *
+     * Customize this only if the default enabled loggers (disk/stderr/mcp) are
+     * not covering your use-case.
+     */
+    additionalLoggers?: LoggerBase[];
+    /**
+     * An optional key value pair of telemetry properties that are reported to
+     * the telemetry backend. Most, if not all, of the properties are captured
+     * automatically.
+     */
+    telemetryProperties?: Partial<CommonProperties>;
+    /**
+     * An optional list of tools constructors to be registered to the MongoDB
+     * MCP Server.
+     *
+     * When not provided, MongoDB MCP Server will register all internal tools.
+     * When specified, **only** the tools in this list will be registered.
+     *
+     * This allows you to:
+     * - Register only custom tools (excluding all internal tools)
+     * - Register a subset of internal tools alongside custom tools
+     * - Register all internal tools plus custom tools
+     *
+     * To include internal tools, import them from `mongodb-mcp-server/tools`:
+     *
+     * ```typescript
+     * import { AllTools, AggregateTool, FindTool } from "mongodb-mcp-server/tools";
+     *
+     * // Register all internal tools plus custom tools
+     * tools: [...AllTools, MyCustomTool]
+     *
+     * // Register only specific MongoDB tools plus custom tools
+     * tools: [AggregateTool, FindTool, MyCustomTool]
+     *
+     * // Register all internal tools of mongodb category
+     * tools: [AllTools.filter((tool) => tool.category === "mongodb")]
+     * ```
+     *
+     * Note: Ensure that each tool has unique names otherwise the server will
+     * throw an error when initializing an MCP Client session. If you're using
+     * only the internal tools, then you don't have to worry about it unless,
+     * you've overridden the tool names.
+     *
+     * To ensure that you provide compliant tool implementations extend your
+     * tool implementation using `ToolBase` class and ensure that they conform
+     * to `ToolClass` type.
+     *
+     * @see {@link ToolClass} for the type that tool classes must conform to
+     * @see {@link ToolBase} for base class for all the tools
+     */
+    tools?: ToolClass[];
+};
+//# sourceMappingURL=common.d.ts.map

package/dist/cjs/transports/interfaces/common.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"common.d.ts","sourceRoot":"","sources":["../../../../src/transports/interfaces/common.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mCAAmC,CAAC;AACpE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AACjE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAErD,MAAM,MAAM,4BAA4B,GAAG;IACvC,aAAa,CAAC,EAAE,SAAS,GAAG,MAAM,CAAC;IACnC;;;;;;;;;;;;;;;OAeG;IACH,UAAU,EAAE,UAAU,CAAC;IAEvB;;;;;;;OAOG;IACH,iBAAiB,CAAC,EAAE,UAAU,EAAE,CAAC;IAEjC;;;;OAIG;IACH,mBAAmB,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,CAAC;IAEhD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAsCG;IACH,KAAK,CAAC,EAAE,SAAS,EAAE,CAAC;CACvB,CAAC"}

package/dist/cjs/transports/interfaces/common.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=common.js.map

package/dist/cjs/transports/interfaces/common.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"common.js","sourceRoot":"","sources":["../../../../src/transports/interfaces/common.ts"],"names":[],"mappings":""}

package/dist/cjs/transports/interfaces/index.d.ts

@@ -0,0 +1,4 @@
+import type { TransportRunnerConfigLegacy } from "./legacy.js";
+export type { TransportRunnerConfigLegacy, CreateSessionConfigFn, RequestContext } from "./legacy.js";
+export type TransportRunnerConfig = TransportRunnerConfigLegacy;
+//# sourceMappingURL=index.d.ts.map

package/dist/cjs/transports/interfaces/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/transports/interfaces/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,aAAa,CAAC;AAE/D,YAAY,EAAE,2BAA2B,EAAE,qBAAqB,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AACtG,MAAM,MAAM,qBAAqB,GAAG,2BAA2B,CAAC"}

package/dist/cjs/transports/interfaces/index.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=index.js.map

package/dist/cjs/transports/interfaces/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/transports/interfaces/index.ts"],"names":[],"mappings":""}