@posthog/mcp 0.10.4 → 0.10.6

package/dist/extensions/capture.d.ts

@@ -1,4 +1,5 @@
 import type { MCPServerLike, McpEvent } from '../types';
+import type { LoggerFn } from './logger';
 /**
  * Materializes an `McpEvent` against the server's tracking data + session info,
  * then hands it to the `McpEventSink` (wrapping the user's posthog-node client) for the
@@ -9,5 +10,5 @@ import type { MCPServerLike, McpEvent } from '../types';
  * Auto-capture callers (tool calls, listings, identify) intentionally ignore the
  * return value, keeping the tool path isolated from analytics latency/errors.
  */
-export declare function captureEvent(server: MCPServerLike, eventInput: McpEvent): Promise<void> | undefined;
+export declare function captureEvent(server: MCPServerLike, eventInput: McpEvent, logger: LoggerFn): Promise<void> | undefined;
 //# sourceMappingURL=capture.d.ts.map

package/dist/extensions/capture.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"capture.d.ts","sourceRoot":"","sources":["../../src/extensions/capture.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAA;AAMvD;;;;;;;;;GASG;AACH,wBAAgB,YAAY,CAAC,MAAM,EAAE,aAAa,EAAE,UAAU,EAAE,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC,GAAG,SAAS,CAkEnG"}
+{"version":3,"file":"capture.d.ts","sourceRoot":"","sources":["../../src/extensions/capture.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAA;AAGvD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAA;AAGxC;;;;;;;;;GASG;AACH,wBAAgB,YAAY,CAAC,MAAM,EAAE,aAAa,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC,GAAG,SAAS,CAkErH"}

package/dist/extensions/capture.js

@@ -28,11 +28,10 @@ __webpack_require__.d(__webpack_exports__, {
 });
 const external_event_types_js_namespaceObject = require("./event-types.js");
 const external_internal_js_namespaceObject = require("./internal.js");
-const external_logger_js_namespaceObject = require("./logger.js");
 const external_session_js_namespaceObject = require("./session.js");
-function captureEvent(server, eventInput) {
+function captureEvent(server, eventInput, logger) {
     const data = (0, external_internal_js_namespaceObject.getServerTrackingData)(server);
-    if (!data) return void (0, external_logger_js_namespaceObject.log)('Warning: Server tracking data not found. Event will not be published.');
+    if (!data) return void logger('Warning: Server tracking data not found. Event will not be published.');
     const sink = data.sink;
     if (!sink) return;
     const sessionInfo = (0, external_session_js_namespaceObject.getSessionInfo)(server, data, eventInput.sessionId);

package/dist/extensions/capture.mjs

@@ -1,10 +1,9 @@
 import { MCPAnalyticsEventType } from "./event-types.mjs";
 import { getServerTrackingData } from "./internal.mjs";
-import { log } from "./logger.mjs";
 import { getSessionInfo } from "./session.mjs";
-function captureEvent(server, eventInput) {
+function captureEvent(server, eventInput, logger) {
     const data = getServerTrackingData(server);
-    if (!data) return void log('Warning: Server tracking data not found. Event will not be published.');
+    if (!data) return void logger('Warning: Server tracking data not found. Event will not be published.');
     const sink = data.sink;
     if (!sink) return;
     const sessionInfo = getSessionInfo(server, data, eventInput.sessionId);

package/dist/extensions/compatibility.d.ts

@@ -1,4 +1,5 @@
 import type { HighLevelMCPServerLike, MCPServerLike } from '../types';
+import type { LoggerFn } from './logger';
 type ServerRecord = Record<string, unknown>;
 /**
  * PostHog MCP analytics Compatibility Module
@@ -12,12 +13,12 @@ type ServerRecord = Record<string, unknown>;
  * - Client version detection
  * - Server info structure
  */
-export declare function logCompatibilityWarning(): void;
+export declare function logCompatibilityWarning(logger: LoggerFn): void;
 export declare function isHighLevelServer(server: unknown): server is ServerRecord & {
     server: ServerRecord;
 };
 export declare function isLowLevelServer(server: unknown): server is ServerRecord;
-export declare function isCompatibleServerType(server: unknown): MCPServerLike | HighLevelMCPServerLike;
+export declare function isCompatibleServerType(server: unknown, logger: LoggerFn): MCPServerLike | HighLevelMCPServerLike;
 export declare function getMCPCompatibleErrorMessage(error: unknown): string;
 export {};
 //# sourceMappingURL=compatibility.d.ts.map

package/dist/extensions/compatibility.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"compatibility.d.ts","sourceRoot":"","sources":["../../src/extensions/compatibility.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,sBAAsB,EAAE,aAAa,EAAE,MAAM,UAAU,CAAA;AAGrE,KAAK,YAAY,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;AAE3C;;;;;;;;;;;GAWG;AAGH,wBAAgB,uBAAuB,IAAI,IAAI,CAI9C;AAGD,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,IAAI,YAAY,GAAG;IAAE,MAAM,EAAE,YAAY,CAAA;CAAE,CAIpG;AAGD,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,IAAI,YAAY,CAExE;AAGD,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,OAAO,GAAG,aAAa,GAAG,sBAAsB,CAgC9F;AAsDD,wBAAgB,4BAA4B,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,CAanE"}
+{"version":3,"file":"compatibility.d.ts","sourceRoot":"","sources":["../../src/extensions/compatibility.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,sBAAsB,EAAE,aAAa,EAAE,MAAM,UAAU,CAAA;AACrE,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAA;AAExC,KAAK,YAAY,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;AAE3C;;;;;;;;;;;GAWG;AAGH,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,QAAQ,GAAG,IAAI,CAI9D;AAGD,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,IAAI,YAAY,GAAG;IAAE,MAAM,EAAE,YAAY,CAAA;CAAE,CAIpG;AAGD,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,IAAI,YAAY,CAExE;AAGD,wBAAgB,sBAAsB,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,GAAG,aAAa,GAAG,sBAAsB,CAgChH;AAsDD,wBAAgB,4BAA4B,CAAC,KAAK,EAAE,OAAO,GAAG,MAAM,CAanE"}

package/dist/extensions/compatibility.js

@@ -24,15 +24,14 @@ var __webpack_require__ = {};
 var __webpack_exports__ = {};
 __webpack_require__.r(__webpack_exports__);
 __webpack_require__.d(__webpack_exports__, {
-    isLowLevelServer: ()=>isLowLevelServer,
-    isHighLevelServer: ()=>isHighLevelServer,
-    logCompatibilityWarning: ()=>logCompatibilityWarning,
+    getMCPCompatibleErrorMessage: ()=>getMCPCompatibleErrorMessage,
     isCompatibleServerType: ()=>isCompatibleServerType,
-    getMCPCompatibleErrorMessage: ()=>getMCPCompatibleErrorMessage
+    isHighLevelServer: ()=>isHighLevelServer,
+    isLowLevelServer: ()=>isLowLevelServer,
+    logCompatibilityWarning: ()=>logCompatibilityWarning
 });
-const external_logger_js_namespaceObject = require("./logger.js");
-function logCompatibilityWarning() {
-    (0, external_logger_js_namespaceObject.log)('PostHog MCP analytics SDK Compatibility: This version only supports Model Context Protocol TypeScript SDK v1.11 and above. Please upgrade if using an older version.');
+function logCompatibilityWarning(logger) {
+    logger('PostHog MCP analytics SDK Compatibility: This version only supports Model Context Protocol TypeScript SDK v1.11 and above. Please upgrade if using an older version.');
 }
 function isHighLevelServer(server) {
     return !!server && 'object' == typeof server && 'server' in server && !!server.server && 'object' == typeof server.server;
@@ -40,51 +39,51 @@ function isHighLevelServer(server) {
 function isLowLevelServer(server) {
     return !!server && 'object' == typeof server && !('server' in server);
 }
-function isCompatibleServerType(server) {
+function isCompatibleServerType(server, logger) {
     if (!server || 'object' != typeof server) {
-        logCompatibilityWarning();
+        logCompatibilityWarning(logger);
         throw new Error("PostHog MCP analytics SDK compatibility error: Server must be an object. Ensure you're using MCP SDK v1.11 or higher.");
     }
     if (isHighLevelServer(server)) {
         if (!server._registeredTools || 'object' != typeof server._registeredTools) {
-            logCompatibilityWarning();
+            logCompatibilityWarning(logger);
             throw new Error('PostHog MCP analytics SDK compatibility error: High-level server must have _registeredTools object. This requires MCP SDK v1.11 or higher.');
         }
         if ('function' != typeof server.tool) {
-            logCompatibilityWarning();
+            logCompatibilityWarning(logger);
             throw new Error('PostHog MCP analytics SDK compatibility error: High-level server must have tool() method. This requires MCP SDK v1.11 or higher.');
         }
         const targetServer = server.server;
-        validateLowLevelServer(targetServer);
+        validateLowLevelServer(targetServer, logger);
         return server;
     }
-    validateLowLevelServer(server);
+    validateLowLevelServer(server, logger);
     return server;
 }
-function validateLowLevelServer(server) {
+function validateLowLevelServer(server, logger) {
     if (!server || 'object' != typeof server) {
-        logCompatibilityWarning();
+        logCompatibilityWarning(logger);
         throw new Error("PostHog MCP analytics SDK compatibility error: Server must be an object. Ensure you're using MCP SDK v1.11 or higher.");
     }
     const serverRecord = server;
     if ('function' != typeof serverRecord.setRequestHandler) {
-        logCompatibilityWarning();
+        logCompatibilityWarning(logger);
         throw new Error('PostHog MCP analytics SDK compatibility error: Server must have a setRequestHandler method. This requires MCP SDK v1.11 or higher.');
     }
     if (!(serverRecord._requestHandlers && serverRecord._requestHandlers instanceof Map)) {
-        logCompatibilityWarning();
+        logCompatibilityWarning(logger);
         throw new Error('PostHog MCP analytics SDK compatibility error: Server._requestHandlers is not accessible. This requires MCP SDK v1.11 or higher.');
     }
     if ('function' != typeof serverRecord._requestHandlers.get) {
-        logCompatibilityWarning();
+        logCompatibilityWarning(logger);
         throw new Error('PostHog MCP analytics SDK compatibility error: Server._requestHandlers must be a Map with a get method. This requires MCP SDK v1.11 or higher.');
     }
     if ('function' != typeof serverRecord.getClientVersion) {
-        logCompatibilityWarning();
+        logCompatibilityWarning(logger);
         throw new Error('PostHog MCP analytics SDK compatibility error: Server.getClientVersion must be a function. This requires MCP SDK v1.11 or higher.');
     }
     if (!serverRecord._serverInfo || 'object' != typeof serverRecord._serverInfo || !('name' in serverRecord._serverInfo)) {
-        logCompatibilityWarning();
+        logCompatibilityWarning(logger);
         throw new Error('PostHog MCP analytics SDK compatibility error: Server._serverInfo is not accessible or missing name. This requires MCP SDK v1.11 or higher.');
     }
 }

package/dist/extensions/compatibility.mjs

@@ -1,6 +1,5 @@
-import { log } from "./logger.mjs";
-function logCompatibilityWarning() {
-    log('PostHog MCP analytics SDK Compatibility: This version only supports Model Context Protocol TypeScript SDK v1.11 and above. Please upgrade if using an older version.');
+function logCompatibilityWarning(logger) {
+    logger('PostHog MCP analytics SDK Compatibility: This version only supports Model Context Protocol TypeScript SDK v1.11 and above. Please upgrade if using an older version.');
 }
 function isHighLevelServer(server) {
     return !!server && 'object' == typeof server && 'server' in server && !!server.server && 'object' == typeof server.server;
@@ -8,51 +7,51 @@ function isHighLevelServer(server) {
 function isLowLevelServer(server) {
     return !!server && 'object' == typeof server && !('server' in server);
 }
-function isCompatibleServerType(server) {
+function isCompatibleServerType(server, logger) {
     if (!server || 'object' != typeof server) {
-        logCompatibilityWarning();
+        logCompatibilityWarning(logger);
         throw new Error("PostHog MCP analytics SDK compatibility error: Server must be an object. Ensure you're using MCP SDK v1.11 or higher.");
     }
     if (isHighLevelServer(server)) {
         if (!server._registeredTools || 'object' != typeof server._registeredTools) {
-            logCompatibilityWarning();
+            logCompatibilityWarning(logger);
             throw new Error('PostHog MCP analytics SDK compatibility error: High-level server must have _registeredTools object. This requires MCP SDK v1.11 or higher.');
         }
         if ('function' != typeof server.tool) {
-            logCompatibilityWarning();
+            logCompatibilityWarning(logger);
             throw new Error('PostHog MCP analytics SDK compatibility error: High-level server must have tool() method. This requires MCP SDK v1.11 or higher.');
         }
         const targetServer = server.server;
-        validateLowLevelServer(targetServer);
+        validateLowLevelServer(targetServer, logger);
         return server;
     }
-    validateLowLevelServer(server);
+    validateLowLevelServer(server, logger);
     return server;
 }
-function validateLowLevelServer(server) {
+function validateLowLevelServer(server, logger) {
     if (!server || 'object' != typeof server) {
-        logCompatibilityWarning();
+        logCompatibilityWarning(logger);
         throw new Error("PostHog MCP analytics SDK compatibility error: Server must be an object. Ensure you're using MCP SDK v1.11 or higher.");
     }
     const serverRecord = server;
     if ('function' != typeof serverRecord.setRequestHandler) {
-        logCompatibilityWarning();
+        logCompatibilityWarning(logger);
         throw new Error('PostHog MCP analytics SDK compatibility error: Server must have a setRequestHandler method. This requires MCP SDK v1.11 or higher.');
     }
     if (!(serverRecord._requestHandlers && serverRecord._requestHandlers instanceof Map)) {
-        logCompatibilityWarning();
+        logCompatibilityWarning(logger);
         throw new Error('PostHog MCP analytics SDK compatibility error: Server._requestHandlers is not accessible. This requires MCP SDK v1.11 or higher.');
     }
     if ('function' != typeof serverRecord._requestHandlers.get) {
-        logCompatibilityWarning();
+        logCompatibilityWarning(logger);
         throw new Error('PostHog MCP analytics SDK compatibility error: Server._requestHandlers must be a Map with a get method. This requires MCP SDK v1.11 or higher.');
     }
     if ('function' != typeof serverRecord.getClientVersion) {
-        logCompatibilityWarning();
+        logCompatibilityWarning(logger);
         throw new Error('PostHog MCP analytics SDK compatibility error: Server.getClientVersion must be a function. This requires MCP SDK v1.11 or higher.');
     }
     if (!serverRecord._serverInfo || 'object' != typeof serverRecord._serverInfo || !('name' in serverRecord._serverInfo)) {
-        logCompatibilityWarning();
+        logCompatibilityWarning(logger);
         throw new Error('PostHog MCP analytics SDK compatibility error: Server._serverInfo is not accessible or missing name. This requires MCP SDK v1.11 or higher.');
     }
 }

package/dist/extensions/context-parameters.d.ts

@@ -1,5 +1,6 @@
 import type { MCPAnalyticsOptions } from '../types';
 import { type AnalyticsInjectableJsonSchema } from './analytics-parameters';
+import { type LoggerFn } from './logger';
 export interface ContextInjectableTool {
     inputSchema?: AnalyticsInjectableJsonSchema;
     name?: string;
@@ -17,6 +18,6 @@ export declare function getContextDescription(context: MCPAnalyticsOptions['cont
  * - Complex schemas (oneOf/allOf/anyOf/$ref) that can't safely have properties added
  * - Schemas with additionalProperties: false
  */
-export declare function addContextParameterToTool<TTool extends ContextInjectableTool>(tool: TTool, contextDescriptionOverride?: string): TTool;
-export declare function addContextParameterToTools<TTool extends ContextInjectableTool>(tools: TTool[], contextDescriptionOverride?: string): TTool[];
+export declare function addContextParameterToTool<TTool extends ContextInjectableTool>(tool: TTool, contextDescriptionOverride?: string, logger?: LoggerFn): TTool;
+export declare function addContextParameterToTools<TTool extends ContextInjectableTool>(tools: TTool[], contextDescriptionOverride?: string, logger?: LoggerFn): TTool[];
 //# sourceMappingURL=context-parameters.d.ts.map

package/dist/extensions/context-parameters.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"context-parameters.d.ts","sourceRoot":"","sources":["../../src/extensions/context-parameters.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAA;AACnD,OAAO,EAGL,KAAK,6BAA6B,EACnC,MAAM,wBAAwB,CAAA;AAI/B,MAAM,WAAW,qBAAqB;IACpC,WAAW,CAAC,EAAE,6BAA6B,CAAA;IAC3C,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;CACvB;AAED,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,mBAAmB,CAAC,SAAS,CAAC,GAAG,OAAO,CAEjF;AAED,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,mBAAmB,CAAC,SAAS,CAAC,GAAG,MAAM,GAAG,SAAS,CAEjG;AAED;;;;;;;;;GASG;AACH,wBAAgB,yBAAyB,CAAC,KAAK,SAAS,qBAAqB,EAC3E,IAAI,EAAE,KAAK,EACX,0BAA0B,CAAC,EAAE,MAAM,GAClC,KAAK,CA6DP;AAED,wBAAgB,0BAA0B,CAAC,KAAK,SAAS,qBAAqB,EAC5E,KAAK,EAAE,KAAK,EAAE,EACd,0BAA0B,CAAC,EAAE,MAAM,GAClC,KAAK,EAAE,CAQT"}
+{"version":3,"file":"context-parameters.d.ts","sourceRoot":"","sources":["../../src/extensions/context-parameters.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAA;AACnD,OAAO,EAGL,KAAK,6BAA6B,EACnC,MAAM,wBAAwB,CAAA;AAE/B,OAAO,EAAO,KAAK,QAAQ,EAAE,MAAM,UAAU,CAAA;AAE7C,MAAM,WAAW,qBAAqB;IACpC,WAAW,CAAC,EAAE,6BAA6B,CAAA;IAC3C,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;CACvB;AAED,wBAAgB,gBAAgB,CAAC,OAAO,EAAE,mBAAmB,CAAC,SAAS,CAAC,GAAG,OAAO,CAEjF;AAED,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,mBAAmB,CAAC,SAAS,CAAC,GAAG,MAAM,GAAG,SAAS,CAEjG;AAED;;;;;;;;;GASG;AACH,wBAAgB,yBAAyB,CAAC,KAAK,SAAS,qBAAqB,EAC3E,IAAI,EAAE,KAAK,EACX,0BAA0B,CAAC,EAAE,MAAM,EACnC,MAAM,GAAE,QAAc,GACrB,KAAK,CA6DP;AAED,wBAAgB,0BAA0B,CAAC,KAAK,SAAS,qBAAqB,EAC5E,KAAK,EAAE,KAAK,EAAE,EACd,0BAA0B,CAAC,EAAE,MAAM,EACnC,MAAM,GAAE,QAAc,GACrB,KAAK,EAAE,CAET"}

package/dist/extensions/context-parameters.js

@@ -38,15 +38,14 @@ function isContextEnabled(context) {
 function getContextDescription(context) {
     return 'object' == typeof context ? context.description : void 0;
 }
-function addContextParameterToTool(tool, contextDescriptionOverride) {
+function addContextParameterToTool(tool, contextDescriptionOverride, logger = external_logger_js_namespaceObject.log) {
     const modifiedTool = {
         ...tool
     };
     const toolName = tool.name || 'unknown';
     const schema = modifiedTool.inputSchema;
     if (!(0, external_analytics_parameters_js_namespaceObject.canInjectAnalyticsParameter)(schema, 'context')) {
-        if ((0, external_analytics_parameters_js_namespaceObject.hasAnalyticsParameter)(schema, 'context')) (0, external_logger_js_namespaceObject.log)(`WARN: Tool "${toolName}" already has 'context' parameter. Skipping context injection.`);
-        else (0, external_logger_js_namespaceObject.log)(`WARN: Tool "${toolName}" has complex schema (oneOf/allOf/anyOf/$ref). Skipping context injection.`);
+        logger((0, external_analytics_parameters_js_namespaceObject.hasAnalyticsParameter)(schema, 'context') ? `WARN: Tool "${toolName}" already has 'context' parameter. Skipping context injection.` : `WARN: Tool "${toolName}" has complex schema (oneOf/allOf/anyOf/$ref). Skipping context injection.`);
         return modifiedTool;
     }
     if (!modifiedTool.inputSchema) modifiedTool.inputSchema = {
@@ -70,11 +69,8 @@ function addContextParameterToTool(tool, contextDescriptionOverride) {
     ];
     return modifiedTool;
 }
-function addContextParameterToTools(tools, contextDescriptionOverride) {
-    return tools.map((tool)=>{
-        if ('get_more_tools' === tool.name) return tool;
-        return addContextParameterToTool(tool, contextDescriptionOverride);
-    });
+function addContextParameterToTools(tools, contextDescriptionOverride, logger = external_logger_js_namespaceObject.log) {
+    return tools.map((tool)=>addContextParameterToTool(tool, contextDescriptionOverride, logger));
 }
 exports.addContextParameterToTool = __webpack_exports__.addContextParameterToTool;
 exports.addContextParameterToTools = __webpack_exports__.addContextParameterToTools;

package/dist/extensions/context-parameters.mjs

@@ -7,14 +7,14 @@ function isContextEnabled(context) {
 function getContextDescription(context) {
     return 'object' == typeof context ? context.description : void 0;
 }
-function addContextParameterToTool(tool, contextDescriptionOverride) {
+function addContextParameterToTool(tool, contextDescriptionOverride, logger = log) {
     const modifiedTool = {
         ...tool
     };
     const toolName = tool.name || 'unknown';
     const schema = modifiedTool.inputSchema;
     if (!canInjectAnalyticsParameter(schema, 'context')) {
-        hasAnalyticsParameter(schema, 'context') ? log(`WARN: Tool "${toolName}" already has 'context' parameter. Skipping context injection.`) : log(`WARN: Tool "${toolName}" has complex schema (oneOf/allOf/anyOf/$ref). Skipping context injection.`);
+        logger(hasAnalyticsParameter(schema, 'context') ? `WARN: Tool "${toolName}" already has 'context' parameter. Skipping context injection.` : `WARN: Tool "${toolName}" has complex schema (oneOf/allOf/anyOf/$ref). Skipping context injection.`);
         return modifiedTool;
     }
     if (!modifiedTool.inputSchema) modifiedTool.inputSchema = {
@@ -38,10 +38,7 @@ function addContextParameterToTool(tool, contextDescriptionOverride) {
     ];
     return modifiedTool;
 }
-function addContextParameterToTools(tools, contextDescriptionOverride) {
-    return tools.map((tool)=>{
-        if ('get_more_tools' === tool.name) return tool;
-        return addContextParameterToTool(tool, contextDescriptionOverride);
-    });
+function addContextParameterToTools(tools, contextDescriptionOverride, logger = log) {
+    return tools.map((tool)=>addContextParameterToTool(tool, contextDescriptionOverride, logger));
 }
 export { addContextParameterToTool, addContextParameterToTools, getContextDescription, isContextEnabled };

package/dist/extensions/conversation-id.d.ts

@@ -1,12 +1,13 @@
 import { type AnalyticsInjectableJsonSchema } from './analytics-parameters';
+import { type LoggerFn } from './logger';
 export declare const CONVERSATION_ID_PARAM_NAME = "conversation_id";
 export interface ConversationIdInjectableTool {
     inputSchema?: AnalyticsInjectableJsonSchema;
     name?: string;
     [key: string]: unknown;
 }
-export declare function addConversationIdToTool<TTool extends ConversationIdInjectableTool>(tool: TTool): TTool;
-export declare function addConversationIdToTools<TTool extends ConversationIdInjectableTool>(tools: TTool[], missingCapabilityToolName?: string): TTool[];
+export declare function addConversationIdToTool<TTool extends ConversationIdInjectableTool>(tool: TTool, logger?: LoggerFn): TTool;
+export declare function addConversationIdToTools<TTool extends ConversationIdInjectableTool>(tools: TTool[], missingCapabilityToolName?: string, skipMissingCapabilityTool?: boolean, logger?: LoggerFn): TTool[];
 export type ConversationIdResolution = {
     minted: false;
     conversationId: string | undefined;

package/dist/extensions/conversation-id.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"conversation-id.d.ts","sourceRoot":"","sources":["../../src/extensions/conversation-id.ts"],"names":[],"mappings":"AAMA,OAAO,EAGL,KAAK,6BAA6B,EACnC,MAAM,wBAAwB,CAAA;AAK/B,eAAO,MAAM,0BAA0B,oBAAoB,CAAA;AAE3D,MAAM,WAAW,4BAA4B;IAC3C,WAAW,CAAC,EAAE,6BAA6B,CAAA;IAC3C,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;CACvB;AAED,wBAAgB,uBAAuB,CAAC,KAAK,SAAS,4BAA4B,EAAE,IAAI,EAAE,KAAK,GAAG,KAAK,CA0CtG;AAED,wBAAgB,wBAAwB,CAAC,KAAK,SAAS,4BAA4B,EACjF,KAAK,EAAE,KAAK,EAAE,EACd,yBAAyB,GAAE,MAA4B,GACtD,KAAK,EAAE,CAOT;AAED,MAAM,MAAM,wBAAwB,GAChC;IAAE,MAAM,EAAE,KAAK,CAAC;IAAC,cAAc,EAAE,MAAM,GAAG,SAAS,CAAA;CAAE,GACrD;IAAE,MAAM,EAAE,IAAI,CAAC;IAAC,cAAc,EAAE,MAAM,CAAA;CAAE,CAAA;AAE5C;;;;;GAKG;AACH,wBAAgB,qBAAqB,CACnC,OAAO,EAAE,OAAO,EAChB,IAAI,EAAE,OAAO,EACb,QAAQ,EAAE,MAAM,GAAG,SAAS,EAC5B,yBAAyB,GAAE,MAA4B,GACtD,wBAAwB,CAS1B;AAED;;;;GAIG;AACH,wBAAgB,iCAAiC,CAAC,MAAM,EAAE,OAAO,GAAG,OAAO,CAS1E;AAED,wBAAgB,qBAAqB,CAAC,IAAI,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS,CAUvE;AAED,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,OAAO,GAAG,OAAO,CAM1D;AAED,wBAAgB,6BAA6B,CAAC,cAAc,EAAE,MAAM,GAAG;IACrE,IAAI,EAAE,MAAM,CAAA;IACZ,IAAI,EAAE,MAAM,CAAA;CACb,CAKA;AAED,wBAAgB,8BAA8B,CAAC,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,MAAM,GAAG,OAAO,CAS/F"}
+{"version":3,"file":"conversation-id.d.ts","sourceRoot":"","sources":["../../src/extensions/conversation-id.ts"],"names":[],"mappings":"AAMA,OAAO,EAGL,KAAK,6BAA6B,EACnC,MAAM,wBAAwB,CAAA;AAE/B,OAAO,EAAO,KAAK,QAAQ,EAAE,MAAM,UAAU,CAAA;AAG7C,eAAO,MAAM,0BAA0B,oBAAoB,CAAA;AAE3D,MAAM,WAAW,4BAA4B;IAC3C,WAAW,CAAC,EAAE,6BAA6B,CAAA;IAC3C,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;CACvB;AAED,wBAAgB,uBAAuB,CAAC,KAAK,SAAS,4BAA4B,EAChF,IAAI,EAAE,KAAK,EACX,MAAM,GAAE,QAAc,GACrB,KAAK,CA4CP;AAED,wBAAgB,wBAAwB,CAAC,KAAK,SAAS,4BAA4B,EACjF,KAAK,EAAE,KAAK,EAAE,EACd,yBAAyB,GAAE,MAA4B,EACvD,yBAAyB,UAAO,EAChC,MAAM,GAAE,QAAc,GACrB,KAAK,EAAE,CAOT;AAED,MAAM,MAAM,wBAAwB,GAChC;IAAE,MAAM,EAAE,KAAK,CAAC;IAAC,cAAc,EAAE,MAAM,GAAG,SAAS,CAAA;CAAE,GACrD;IAAE,MAAM,EAAE,IAAI,CAAC;IAAC,cAAc,EAAE,MAAM,CAAA;CAAE,CAAA;AAE5C;;;;;GAKG;AACH,wBAAgB,qBAAqB,CACnC,OAAO,EAAE,OAAO,EAChB,IAAI,EAAE,OAAO,EACb,QAAQ,EAAE,MAAM,GAAG,SAAS,EAC5B,yBAAyB,GAAE,MAA4B,GACtD,wBAAwB,CAS1B;AAED;;;;GAIG;AACH,wBAAgB,iCAAiC,CAAC,MAAM,EAAE,OAAO,GAAG,OAAO,CAS1E;AAED,wBAAgB,qBAAqB,CAAC,IAAI,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS,CAUvE;AAED,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,OAAO,GAAG,OAAO,CAM1D;AAED,wBAAgB,6BAA6B,CAAC,cAAc,EAAE,MAAM,GAAG;IACrE,IAAI,EAAE,MAAM,CAAA;IACZ,IAAI,EAAE,MAAM,CAAA;CACb,CAKA;AAED,wBAAgB,8BAA8B,CAAC,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,MAAM,GAAG,OAAO,CAS/F"}

package/dist/extensions/conversation-id.js

@@ -40,15 +40,14 @@ const external_constants_js_namespaceObject = require("./constants.js");
 const external_logger_js_namespaceObject = require("./logger.js");
 const external_tools_js_namespaceObject = require("./tools.js");
 const CONVERSATION_ID_PARAM_NAME = 'conversation_id';
-function addConversationIdToTool(tool) {
+function addConversationIdToTool(tool, logger = external_logger_js_namespaceObject.log) {
     const modifiedTool = {
         ...tool
     };
     const toolName = tool.name || 'unknown';
     const schema = modifiedTool.inputSchema;
     if (!(0, external_analytics_parameters_js_namespaceObject.canInjectAnalyticsParameter)(schema, CONVERSATION_ID_PARAM_NAME)) {
-        if ((0, external_analytics_parameters_js_namespaceObject.hasAnalyticsParameter)(schema, CONVERSATION_ID_PARAM_NAME)) (0, external_logger_js_namespaceObject.log)(`WARN: Tool "${toolName}" already has '${CONVERSATION_ID_PARAM_NAME}' parameter. Skipping conversation_id injection.`);
-        else (0, external_logger_js_namespaceObject.log)(`WARN: Tool "${toolName}" has complex schema (oneOf/allOf/anyOf/$ref). Skipping conversation_id injection.`);
+        logger((0, external_analytics_parameters_js_namespaceObject.hasAnalyticsParameter)(schema, CONVERSATION_ID_PARAM_NAME) ? `WARN: Tool "${toolName}" already has '${CONVERSATION_ID_PARAM_NAME}' parameter. Skipping conversation_id injection.` : `WARN: Tool "${toolName}" has complex schema (oneOf/allOf/anyOf/$ref). Skipping conversation_id injection.`);
         return modifiedTool;
     }
     if (!modifiedTool.inputSchema) modifiedTool.inputSchema = {
@@ -66,10 +65,10 @@ function addConversationIdToTool(tool) {
     };
     return modifiedTool;
 }
-function addConversationIdToTools(tools, missingCapabilityToolName = external_tools_js_namespaceObject.GET_MORE_TOOLS_NAME) {
+function addConversationIdToTools(tools, missingCapabilityToolName = external_tools_js_namespaceObject.GET_MORE_TOOLS_NAME, skipMissingCapabilityTool = true, logger = external_logger_js_namespaceObject.log) {
     return tools.map((tool)=>{
-        if (tool.name === missingCapabilityToolName) return tool;
-        return addConversationIdToTool(tool);
+        if (skipMissingCapabilityTool && tool.name === missingCapabilityToolName) return tool;
+        return addConversationIdToTool(tool, logger);
     });
 }
 function resolveConversationId(enabled, args, toolName, missingCapabilityToolName = external_tools_js_namespaceObject.GET_MORE_TOOLS_NAME) {

package/dist/extensions/conversation-id.mjs

@@ -4,14 +4,14 @@ import { DEFAULT_CONVERSATION_ID_DESCRIPTION } from "./constants.mjs";
 import { log } from "./logger.mjs";
 import { GET_MORE_TOOLS_NAME } from "./tools.mjs";
 const CONVERSATION_ID_PARAM_NAME = 'conversation_id';
-function addConversationIdToTool(tool) {
+function addConversationIdToTool(tool, logger = log) {
     const modifiedTool = {
         ...tool
     };
     const toolName = tool.name || 'unknown';
     const schema = modifiedTool.inputSchema;
     if (!canInjectAnalyticsParameter(schema, CONVERSATION_ID_PARAM_NAME)) {
-        hasAnalyticsParameter(schema, CONVERSATION_ID_PARAM_NAME) ? log(`WARN: Tool "${toolName}" already has '${CONVERSATION_ID_PARAM_NAME}' parameter. Skipping conversation_id injection.`) : log(`WARN: Tool "${toolName}" has complex schema (oneOf/allOf/anyOf/$ref). Skipping conversation_id injection.`);
+        logger(hasAnalyticsParameter(schema, CONVERSATION_ID_PARAM_NAME) ? `WARN: Tool "${toolName}" already has '${CONVERSATION_ID_PARAM_NAME}' parameter. Skipping conversation_id injection.` : `WARN: Tool "${toolName}" has complex schema (oneOf/allOf/anyOf/$ref). Skipping conversation_id injection.`);
         return modifiedTool;
     }
     if (!modifiedTool.inputSchema) modifiedTool.inputSchema = {
@@ -29,10 +29,10 @@ function addConversationIdToTool(tool) {
     };
     return modifiedTool;
 }
-function addConversationIdToTools(tools, missingCapabilityToolName = GET_MORE_TOOLS_NAME) {
+function addConversationIdToTools(tools, missingCapabilityToolName = GET_MORE_TOOLS_NAME, skipMissingCapabilityTool = true, logger = log) {
     return tools.map((tool)=>{
-        if (tool.name === missingCapabilityToolName) return tool;
-        return addConversationIdToTool(tool);
+        if (skipMissingCapabilityTool && tool.name === missingCapabilityToolName) return tool;
+        return addConversationIdToTool(tool, logger);
     });
 }
 function resolveConversationId(enabled, args, toolName, missingCapabilityToolName = GET_MORE_TOOLS_NAME) {

package/dist/extensions/instrument-highlevel.d.ts

@@ -1,3 +1,4 @@
 import type { HighLevelMCPServerLike } from '../types';
-export declare function instrumentHighLevelServer(server: HighLevelMCPServerLike): void;
+import type { LoggerFn } from './logger';
+export declare function instrumentHighLevelServer(server: HighLevelMCPServerLike, logger: LoggerFn): void;
 //# sourceMappingURL=instrument-highlevel.d.ts.map

package/dist/extensions/instrument-highlevel.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"instrument-highlevel.d.ts","sourceRoot":"","sources":["../../src/extensions/instrument-highlevel.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAEV,sBAAsB,EAIvB,MAAM,UAAU,CAAA;AAoQjB,wBAAgB,yBAAyB,CAAC,MAAM,EAAE,sBAAsB,GAAG,IAAI,CA6B9E"}
+{"version":3,"file":"instrument-highlevel.d.ts","sourceRoot":"","sources":["../../src/extensions/instrument-highlevel.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAEV,sBAAsB,EAIvB,MAAM,UAAU,CAAA;AASjB,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAA;AA8PxC,wBAAgB,yBAAyB,CAAC,MAAM,EAAE,sBAAsB,EAAE,MAAM,EAAE,QAAQ,GAAG,IAAI,CA2BhG"}

package/dist/extensions/instrument-highlevel.js

@@ -30,7 +30,6 @@ const external_analytics_parameters_js_namespaceObject = require("./analytics-pa
 const external_context_parameters_js_namespaceObject = require("./context-parameters.js");
 const external_event_types_js_namespaceObject = require("./event-types.js");
 const external_internal_js_namespaceObject = require("./internal.js");
-const external_logger_js_namespaceObject = require("./logger.js");
 const external_mcp_sdk_compat_js_namespaceObject = require("./mcp-sdk-compat.js");
 const external_tools_js_namespaceObject = require("./tools.js");
 const external_instrumentation_js_namespaceObject = require("./instrumentation.js");
@@ -40,31 +39,31 @@ const MCP_ANALYTICS_PROCESSED = Symbol('__posthog_mcp_analytics_processed__');
 function isCallbackUpdate(value) {
     return !!value && 'object' == typeof value && 'callback' in value && 'function' == typeof value.callback;
 }
-function addTracingToToolRegistry(tools, server) {
+function addTracingToToolRegistry(tools, server, logger) {
     return Object.fromEntries(Object.entries(tools).map(([name, tool])=>[
             name,
-            addTracingToToolCallbackInternal(tool, name, server)
+            addTracingToToolCallbackInternal(tool, name, server, ()=>tool.inputSchema, logger)
         ]));
 }
-function setupListenerToRegisteredTools(server) {
+function setupListenerToRegisteredTools(server, logger) {
     try {
         const data = (0, external_internal_js_namespaceObject.getServerTrackingData)(server.server);
-        if (!data) return void (0, external_logger_js_namespaceObject.log)('Warning: Cannot setup listener - no tracking data found');
+        if (!data) return void logger('Warning: Cannot setup listener - no tracking data found');
         const handler = {
             set (target, property, value) {
                 try {
                     if ('string' == typeof property && value && 'object' == typeof value && (0, external_mcp_sdk_compat_js_namespaceObject.hasToolFunction)(value)) {
                         if ('string' == typeof value.description) data.toolDescriptions.set(property, value.description);
                         if (value[MCP_ANALYTICS_PROCESSED]) {
-                            (0, external_logger_js_namespaceObject.log)(`Tool ${String(property)} already processed, skipping proxy wrapping`);
+                            logger(`Tool ${String(property)} already processed, skipping proxy wrapping`);
                             return Reflect.set(target, property, value);
                         }
                         if (wrappedCallbacks.has((0, external_mcp_sdk_compat_js_namespaceObject.getToolFunction)(value))) {
-                            (0, external_logger_js_namespaceObject.log)(`Tool ${String(property)} callback already wrapped, skipping proxy wrapping`);
+                            logger(`Tool ${String(property)} callback already wrapped, skipping proxy wrapping`);
                             return Reflect.set(target, property, value);
                         }
                         const getCurrentInputSchema = ()=>value.inputSchema;
-                        const nextValue = addTracingToToolCallbackInternal(value, property, server, getCurrentInputSchema);
+                        const nextValue = addTracingToToolCallbackInternal(value, property, server, getCurrentInputSchema, logger);
                         if ('function' == typeof nextValue.update) {
                             const originalUpdate = nextValue.update;
                             nextValue.update = function(...updateArgs) {
@@ -73,7 +72,7 @@ function setupListenerToRegisteredTools(server) {
                                     if (isCallbackUpdate(updateObj)) {
                                         const wrappedTool = addTracingToToolCallbackInternal({
                                             callback: updateObj.callback
-                                        }, property, server, getCurrentInputSchema);
+                                        }, property, server, getCurrentInputSchema, logger);
                                         updateObj.callback = (0, external_mcp_sdk_compat_js_namespaceObject.getToolFunction)(wrappedTool);
                                     }
                                 }
@@ -84,7 +83,7 @@ function setupListenerToRegisteredTools(server) {
                     }
                     return Reflect.set(target, property, value);
                 } catch (error) {
-                    (0, external_logger_js_namespaceObject.log)(`Warning: Error in proxy set handler for tool ${String(property)} - ${error}`);
+                    logger(`Warning: Error in proxy set handler for tool ${String(property)} - ${error}`);
                     return Reflect.set(target, property, value);
                 }
             },
@@ -100,21 +99,20 @@ function setupListenerToRegisteredTools(server) {
         };
         const originalTools = server._registeredTools || {};
         server._registeredTools = new Proxy(originalTools, handler);
-        (0, external_logger_js_namespaceObject.log)('Successfully set up listener for new tool registrations');
+        logger('Successfully set up listener for new tool registrations');
     } catch (error) {
-        (0, external_logger_js_namespaceObject.log)(`Warning: Failed to setup listener for registered tools - ${error}`);
+        logger(`Warning: Failed to setup listener for registered tools - ${error}`);
     }
 }
-function addTracingToToolCallbackInternal(tool, toolName, server, getCurrentInputSchema = ()=>tool.inputSchema) {
+function addTracingToToolCallbackInternal(tool, toolName, server, getCurrentInputSchema, logger) {
     const originalCallback = (0, external_mcp_sdk_compat_js_namespaceObject.getToolFunction)(tool);
     const options = (0, external_internal_js_namespaceObject.getServerTrackingData)(server.server)?.options;
-    const missingToolName = (0, external_tools_js_namespaceObject.resolveMissingCapabilityToolName)(options);
     if (wrappedCallbacks.has(originalCallback)) {
-        (0, external_logger_js_namespaceObject.log)(`Tool ${toolName} callback already wrapped, skipping re-wrap`);
+        logger(`Tool ${toolName} callback already wrapped, skipping re-wrap`);
         return tool;
     }
     if (tool[MCP_ANALYTICS_PROCESSED]) {
-        (0, external_logger_js_namespaceObject.log)(`Tool ${toolName} already processed, skipping re-wrap`);
+        logger(`Tool ${toolName} already processed, skipping re-wrap`);
         return tool;
     }
     const wrappedCallback = async (...params)=>{
@@ -129,8 +127,8 @@ function addTracingToToolCallbackInternal(tool, toolName, server, getCurrentInpu
         }
         const inputSchema = getCurrentInputSchema();
         const cleanedArgs = (0, external_analytics_parameters_js_namespaceObject.stripOwnedAnalyticsArguments)(args, {
-            context: toolName !== missingToolName && (0, external_context_parameters_js_namespaceObject.isContextEnabled)(options?.context) && (0, external_analytics_parameters_js_namespaceObject.analyticsOwnsParameter)(inputSchema, 'context'),
-            conversationId: toolName !== missingToolName && options?.enableConversationId === true && (0, external_analytics_parameters_js_namespaceObject.analyticsOwnsParameter)(inputSchema, 'conversation_id')
+            context: (0, external_context_parameters_js_namespaceObject.isContextEnabled)(options?.context) && (0, external_analytics_parameters_js_namespaceObject.analyticsOwnsParameter)(inputSchema, 'context'),
+            conversationId: options?.enableConversationId === true && (0, external_analytics_parameters_js_namespaceObject.analyticsOwnsParameter)(inputSchema, 'conversation_id')
         });
         try {
             if (void 0 === cleanedArgs) {
@@ -150,13 +148,15 @@ function addTracingToToolCallbackInternal(tool, toolName, server, getCurrentInpu
     wrappedTool[MCP_ANALYTICS_PROCESSED] = true;
     return wrappedTool;
 }
-async function handleToolCallRequest(highLevelServer, server, originalCallToolHandler, request, extra) {
+async function handleToolCallRequest(highLevelServer, server, originalCallToolHandler, request, extra, logger) {
     const data = (0, external_internal_js_namespaceObject.getServerTrackingData)(server);
     if (!data) {
-        (0, external_logger_js_namespaceObject.log)('Warning: PostHog MCP analytics is unable to find server tracking data. Please ensure you have called instrument(server, options) before using tool calls.');
+        logger('Warning: PostHog MCP analytics is unable to find server tracking data. Please ensure you have called instrument(server, options) before using tool calls.');
         return await originalCallToolHandler(request, extra);
     }
-    if (request.params?.name === (0, external_tools_js_namespaceObject.resolveMissingCapabilityToolName)(data.options)) {
+    const toolName = request.params?.name;
+    const isMissingCapabilityCandidate = data.options.reportMissing && toolName === (0, external_tools_js_namespaceObject.resolveMissingCapabilityToolName)(data.options);
+    if (isMissingCapabilityCandidate && await (0, external_instrumentation_js_namespaceObject.isToolAdvertised)(server, toolName, extra, data.logger) === false) {
         const context = (0, external_tracing_helpers_js_namespaceObject.getContextArgument)(request) || '';
         return await (0, external_instrumentation_js_namespaceObject.captureToolCall)({
             server,
@@ -167,10 +167,9 @@ async function handleToolCallRequest(highLevelServer, server, originalCallToolHa
             explicitContextIntent: context,
             execute: async ()=>(0, external_tools_js_namespaceObject.handleReportMissing)({
                     context
-                })
+                }, data.logger)
         });
     }
-    const toolName = request.params?.name;
     const registeredTool = toolName ? highLevelServer._registeredTools[toolName] : void 0;
     return await (0, external_instrumentation_js_namespaceObject.captureToolCall)({
         server,
@@ -186,24 +185,24 @@ async function handleToolCallRequest(highLevelServer, server, originalCallToolHa
         }
     });
 }
-function instrumentHighLevelServer(server) {
+function instrumentHighLevelServer(server, logger) {
     try {
         const lowLevelServer = server.server;
         const mcpAnalyticsData = (0, external_internal_js_namespaceObject.getServerTrackingData)(lowLevelServer);
         const handlers = {
-            initialize: external_instrumentation_js_namespaceObject.handleInitializeRequest,
-            'tools/list': external_instrumentation_js_namespaceObject.handleListToolsRequest,
-            'tools/call': (trackedServer, originalHandler, request, extra)=>handleToolCallRequest(server, trackedServer, originalHandler, request, extra)
+            initialize: (trackedServer, originalHandler, request, extra)=>(0, external_instrumentation_js_namespaceObject.handleInitializeRequest)(trackedServer, originalHandler, request, extra, logger),
+            'tools/list': (trackedServer, originalHandler, request, extra)=>(0, external_instrumentation_js_namespaceObject.handleListToolsRequest)(trackedServer, originalHandler, request, extra, logger),
+            'tools/call': (trackedServer, originalHandler, request, extra)=>handleToolCallRequest(server, trackedServer, originalHandler, request, extra, logger)
         };
         (0, external_instrumentation_js_namespaceObject.patchRequestHandlers)(lowLevelServer, handlers);
-        server._registeredTools = addTracingToToolRegistry(server._registeredTools, server);
+        server._registeredTools = addTracingToToolRegistry(server._registeredTools, server, logger);
         if (mcpAnalyticsData) {
             seedToolDescriptionsFromRegistry(mcpAnalyticsData.toolDescriptions, server._registeredTools);
             seedToolCategoriesFromRegistry(mcpAnalyticsData.toolCategories, server._registeredTools);
         }
-        setupListenerToRegisteredTools(server);
+        setupListenerToRegisteredTools(server, logger);
     } catch (error) {
-        (0, external_logger_js_namespaceObject.log)(`Warning: Failed to setup tool call instrumentation - ${error}`);
+        logger(`Warning: Failed to setup tool call instrumentation - ${error}`);
     }
 }
 function seedToolDescriptionsFromRegistry(cache, tools) {