@posthog/mcp 0.10.4 → 0.10.6

package/dist/extensions/instrumentation.js

@@ -24,11 +24,12 @@ var __webpack_require__ = {};
 var __webpack_exports__ = {};
 __webpack_require__.r(__webpack_exports__);
 __webpack_require__.d(__webpack_exports__, {
-    readToolMetaCategory: ()=>readToolMetaCategory,
+    isToolAdvertised: ()=>isToolAdvertised,
     cacheToolDescriptions: ()=>cacheToolDescriptions,
-    cacheToolCategories: ()=>cacheToolCategories,
     handleInitializeRequest: ()=>handleInitializeRequest,
     captureToolCall: ()=>captureToolCall,
+    readToolMetaCategory: ()=>readToolMetaCategory,
+    cacheToolCategories: ()=>cacheToolCategories,
     handleListToolsRequest: ()=>handleListToolsRequest,
     patchRequestHandlers: ()=>patchRequestHandlers
 });
@@ -41,7 +42,6 @@ const external_event_types_js_namespaceObject = require("./event-types.js");
 const external_exceptions_js_namespaceObject = require("./exceptions.js");
 const external_intent_js_namespaceObject = require("./intent.js");
 const external_internal_js_namespaceObject = require("./internal.js");
-const external_logger_js_namespaceObject = require("./logger.js");
 const external_mcp_payloads_js_namespaceObject = require("./mcp-payloads.js");
 const external_mcp_sdk_compat_js_namespaceObject = require("./mcp-sdk-compat.js");
 const external_session_js_namespaceObject = require("./session.js");
@@ -50,26 +50,26 @@ const external_tools_js_namespaceObject = require("./tools.js");
 const external_tracing_helpers_js_namespaceObject = require("./tracing-helpers.js");
 async function captureToolCall(params) {
     const { server, data, request, extra, execute, parameterOwnership, eventType, explicitContextIntent, takeCapturedError } = params;
-    const ownership = getActiveAnalyticsParameterOwnership(data, request.params?.name, parameterOwnership);
-    const conversation = (0, external_conversation_id_js_namespaceObject.resolveConversationId)(ownership.conversationId, request.params?.arguments, request.params?.name, (0, external_tools_js_namespaceObject.resolveMissingCapabilityToolName)(data.options));
+    const resolvedEventType = eventType ?? external_event_types_js_namespaceObject.MCPAnalyticsEventType.mcpToolsCall;
+    const ownership = getActiveAnalyticsParameterOwnership(data, request.params?.name, parameterOwnership, resolvedEventType === external_event_types_js_namespaceObject.MCPAnalyticsEventType.mcpMissingCapability);
+    const conversation = (0, external_conversation_id_js_namespaceObject.resolveConversationId)(ownership.conversationId, request.params?.arguments, request.params?.name, resolvedEventType === external_event_types_js_namespaceObject.MCPAnalyticsEventType.mcpMissingCapability ? (0, external_tools_js_namespaceObject.resolveMissingCapabilityToolName)(data.options) : '');
     const downstreamRequest = cloneRequestWithoutOwnedAnalyticsArguments(request, ownership);
     const startTime = new Date();
-    const event = await prepareToolCallEvent(server, data, request, downstreamRequest, extra, startTime, conversation, ownership, eventType ?? external_event_types_js_namespaceObject.MCPAnalyticsEventType.mcpToolsCall);
+    const event = await prepareToolCallEvent(server, data, request, downstreamRequest, extra, startTime, conversation, ownership, resolvedEventType);
     if (event && explicitContextIntent) (0, external_intent_js_namespaceObject.setExplicitContextIntent)(event, explicitContextIntent);
     let result;
     try {
         result = await execute(downstreamRequest);
     } catch (error) {
-        publishFailedToolEvent(server, event, error, startTime, conversation);
+        publishFailedToolEvent(server, event, error, startTime, conversation, data.logger);
         throw error;
     }
     const finalResult = applyConversationPromptBack(event, result, conversation);
-    publishSuccessfulToolEvent(server, event, finalResult, startTime, takeCapturedError);
+    publishSuccessfulToolEvent(server, event, finalResult, startTime, data.logger, takeCapturedError);
     return finalResult;
 }
-function getActiveAnalyticsParameterOwnership(data, toolName, override) {
+function getActiveAnalyticsParameterOwnership(data, toolName, override, isMissingCapabilityTool) {
     const ownership = override ?? (toolName ? data.toolAnalyticsParameterOwnership.get(toolName) : void 0);
-    const isMissingCapabilityTool = toolName === (0, external_tools_js_namespaceObject.resolveMissingCapabilityToolName)(data.options);
     return {
         context: !isMissingCapabilityTool && (0, external_context_parameters_js_namespaceObject.isContextEnabled)(data.options.context) && ownership?.context === true,
         conversationId: !isMissingCapabilityTool && true === data.options.enableConversationId && ownership?.conversationId === true
@@ -107,7 +107,7 @@ async function prepareToolCallEvent(server, data, request, downstreamRequest, ex
         (0, external_intent_js_namespaceObject.setEventIntent)(event, await (0, external_intent_js_namespaceObject.resolveToolCallIntent)(data, request, ownership.context, extra));
         return event;
     } catch (error) {
-        (0, external_logger_js_namespaceObject.log)(`Warning: PostHog MCP analytics instrumentation failed for tool ${request.params?.name}, the tool will still run - ${error}`);
+        data.logger(`Warning: PostHog MCP analytics instrumentation failed for tool ${request.params?.name}, the tool will still run - ${error}`);
         return null;
     }
 }
@@ -117,7 +117,7 @@ function applyConversationPromptBack(event, result, conversation) {
     if (event) event.conversationId = void 0;
     return result;
 }
-function publishSuccessfulToolEvent(server, event, result, startTime, takeCapturedError) {
+function publishSuccessfulToolEvent(server, event, result, startTime, logger, takeCapturedError) {
     if (!event) return;
     try {
         if ((0, external_tracing_helpers_js_namespaceObject.isToolResultError)(result)) {
@@ -127,38 +127,71 @@ function publishSuccessfulToolEvent(server, event, result, startTime, takeCaptur
         } else event.isError = false;
         event.response = result;
         event.duration = Date.now() - startTime.getTime();
-        (0, external_capture_js_namespaceObject.captureEvent)(server, event);
+        (0, external_capture_js_namespaceObject.captureEvent)(server, event, logger);
     } catch (error) {
-        (0, external_logger_js_namespaceObject.log)(`Warning: PostHog MCP analytics failed to publish tool event - ${error}`);
+        logger(`Warning: PostHog MCP analytics failed to publish tool event - ${error}`);
     }
 }
-function publishFailedToolEvent(server, event, error, startTime, conversation) {
+function publishFailedToolEvent(server, event, error, startTime, conversation, logger) {
     if (!event) return;
     try {
         if (conversation.minted) event.conversationId = void 0;
         event.isError = true;
         event.error = (0, external_exceptions_js_namespaceObject.captureException)(error);
         event.duration = Date.now() - startTime.getTime();
-        (0, external_capture_js_namespaceObject.captureEvent)(server, event);
+        (0, external_capture_js_namespaceObject.captureEvent)(server, event, logger);
     } catch (publishError) {
-        (0, external_logger_js_namespaceObject.log)(`Warning: PostHog MCP analytics failed to publish failed tool event - ${publishError}`);
+        logger(`Warning: PostHog MCP analytics failed to publish failed tool event - ${publishError}`);
     }
 }
+const originalRequestHandlers = new WeakMap();
+function rememberOriginalRequestHandler(server, handlerName, originalHandler) {
+    let handlers = originalRequestHandlers.get(server);
+    if (!handlers) {
+        handlers = new Map();
+        originalRequestHandlers.set(server, handlers);
+    }
+    handlers.set(handlerName, originalHandler);
+}
 function patchRequestHandlers(server, patches) {
     for (const [handlerName, patch] of Object.entries(patches)){
         const originalHandler = server._requestHandlers.get(handlerName);
-        if (originalHandler) server._requestHandlers.set(handlerName, (request, extra)=>patch(server, originalHandler, request, extra));
+        if (originalHandler) {
+            rememberOriginalRequestHandler(server, handlerName, originalHandler);
+            server._requestHandlers.set(handlerName, (request, extra)=>patch(server, originalHandler, request, extra));
+        }
     }
     const originalSetRequestHandler = server.setRequestHandler.bind(server);
     server.setRequestHandler = (requestSchema, originalHandler)=>{
         const shape = (0, external_mcp_sdk_compat_js_namespaceObject.getObjectShape)(requestSchema);
         const handlerName = shape?.method ? (0, external_mcp_sdk_compat_js_namespaceObject.getLiteralValue)(shape.method) : void 0;
         const patch = 'string' == typeof handlerName ? patches[handlerName] : void 0;
-        if (!patch) return originalSetRequestHandler(requestSchema, originalHandler);
-        return originalSetRequestHandler(requestSchema, (request, extra)=>patch(server, originalHandler, request, extra));
+        if (!patch || 'string' != typeof handlerName) return originalSetRequestHandler(requestSchema, originalHandler);
+        const result = originalSetRequestHandler(requestSchema, originalHandler);
+        const registeredHandler = server._requestHandlers.get(handlerName);
+        if (registeredHandler) {
+            rememberOriginalRequestHandler(server, handlerName, registeredHandler);
+            server._requestHandlers.set(handlerName, (request, extra)=>patch(server, registeredHandler, request, extra));
+        }
+        return result;
     };
 }
-async function handleListToolsRequest(server, originalListToolsHandler, request, extra) {
+async function isToolAdvertised(server, toolName, extra, logger) {
+    const listHandler = originalRequestHandlers.get(server)?.get('tools/list');
+    if (!listHandler || !server._requestHandlers.has('tools/list')) return;
+    try {
+        const response = await listHandler({
+            method: 'tools/list',
+            params: {}
+        }, extra);
+        if (!response || !Array.isArray(response.tools)) return;
+        return response.tools.some((tool)=>tool?.name === toolName);
+    } catch (error) {
+        logger(`Warning: PostHog MCP analytics could not determine whether "${toolName}" is advertised; delegating to the server - ${error}`);
+        return;
+    }
+}
+async function handleListToolsRequest(server, originalListToolsHandler, request, extra, logger) {
     const data = (0, external_internal_js_namespaceObject.getServerTrackingData)(server);
     const startTime = new Date();
     const event = {
@@ -169,19 +202,19 @@ async function handleListToolsRequest(server, originalListToolsHandler, request,
     };
     (0, external_client_identity_js_namespaceObject.stampMetaClientInfo)(event, request);
     if (data) await (0, external_tracing_helpers_js_namespaceObject.applyResolvedMetadata)(event, data, request, extra);
-    const tools = await getTracedToolsList(server, originalListToolsHandler, request, extra, event);
+    const tools = await getTracedToolsList(server, originalListToolsHandler, request, extra, event, logger);
     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 {
             tools
         };
     }
     if (0 === tools.length) {
-        (0, external_logger_js_namespaceObject.log)('Warning: No tools found in the original list. This is likely due to the tools not being registered before PostHog MCP analytics.instrument().');
+        data.logger('Warning: No tools found in the original list. This is likely due to the tools not being registered before PostHog MCP analytics.instrument().');
         event.error = (0, external_exceptions_js_namespaceObject.captureException)('No tools were sent to MCP client.');
         event.isError = true;
         event.duration = Date.now() - startTime.getTime();
-        (0, external_capture_js_namespaceObject.captureEvent)(server, event);
+        (0, external_capture_js_namespaceObject.captureEvent)(server, event, data.logger);
         return {
             tools
         };
@@ -192,7 +225,7 @@ async function handleListToolsRequest(server, originalListToolsHandler, request,
     event.listedToolNames = collectListedToolNames(tools);
     event.isError = false;
     event.duration = Date.now() - startTime.getTime();
-    (0, external_capture_js_namespaceObject.captureEvent)(server, event);
+    (0, external_capture_js_namespaceObject.captureEvent)(server, event, data.logger);
     return {
         tools
     };
@@ -205,18 +238,27 @@ function collectListedToolNames(tools) {
     const names = tools.map((tool)=>tool?.name).filter((name)=>'string' == typeof name);
     return names.length > 0 ? names : void 0;
 }
-async function getTracedToolsList(server, originalListToolsHandler, request, extra, event) {
+async function getTracedToolsList(server, originalListToolsHandler, request, extra, event, logger) {
     try {
         const data = (0, external_internal_js_namespaceObject.getServerTrackingData)(server);
         const originalResponse = await originalListToolsHandler(request, extra);
-        let tools = originalResponse.tools || [];
+        let tools = [
+            ...originalResponse.tools || []
+        ];
         if (data) cacheToolAnalyticsParameterOwnership(data.toolAnalyticsParameterOwnership, tools);
-        if (data && (0, external_context_parameters_js_namespaceObject.isContextEnabled)(data.options.context)) tools = (0, external_context_parameters_js_namespaceObject.addContextParameterToTools)(tools, (0, external_context_parameters_js_namespaceObject.getContextDescription)(data.options.context));
-        if (data?.options.enableConversationId) tools = (0, external_conversation_id_js_namespaceObject.addConversationIdToTools)(tools, (0, external_tools_js_namespaceObject.resolveMissingCapabilityToolName)(data.options));
-        if (data?.options.reportMissing) {
+        if (data && (0, external_context_parameters_js_namespaceObject.isContextEnabled)(data.options.context)) tools = (0, external_context_parameters_js_namespaceObject.addContextParameterToTools)(tools, (0, external_context_parameters_js_namespaceObject.getContextDescription)(data.options.context), data.logger);
+        if (data) {
             const missingToolName = (0, external_tools_js_namespaceObject.resolveMissingCapabilityToolName)(data.options);
-            const alreadyPresent = tools.some((tool)=>tool?.name === missingToolName);
-            if (!alreadyPresent) tools.push((0, external_tools_js_namespaceObject.getReportMissingToolDescriptor)(missingToolName));
+            let injectedMissingCapabilityTool = false;
+            if (data.options.reportMissing) {
+                const alreadyPresent = tools.some((tool)=>tool?.name === missingToolName);
+                if (alreadyPresent) data.logger(`Warning: Cannot inject missing-capability tool "${missingToolName}" because a real tool already uses that name. The real tool will not be intercepted.`);
+                else {
+                    tools.push((0, external_tools_js_namespaceObject.getReportMissingToolDescriptor)(missingToolName));
+                    injectedMissingCapabilityTool = true;
+                }
+            }
+            if (data.options.enableConversationId) tools = (0, external_conversation_id_js_namespaceObject.addConversationIdToTools)(tools, missingToolName, injectedMissingCapabilityTool, data.logger);
         }
         if (data) {
             cacheToolDescriptions(data.toolDescriptions, tools);
@@ -224,11 +266,11 @@ async function getTracedToolsList(server, originalListToolsHandler, request, ext
         }
         return tools;
     } catch (error) {
-        (0, external_logger_js_namespaceObject.log)(`Warning: Original list tools handler failed, this suggests an error PostHog MCP analytics did not cause - ${error}`);
+        logger(`Warning: Original list tools handler failed, this suggests an error PostHog MCP analytics did not cause - ${error}`);
         event.error = (0, external_exceptions_js_namespaceObject.captureException)(error);
         event.isError = true;
         event.duration = event.timestamp ? Date.now() - event.timestamp.getTime() : 0;
-        (0, external_capture_js_namespaceObject.captureEvent)(server, event);
+        (0, external_capture_js_namespaceObject.captureEvent)(server, event, logger);
         throw error;
     }
 }
@@ -273,7 +315,7 @@ function mintStatelessSessionOnInitialize(server, data, request, extra) {
         (0, external_internal_js_namespaceObject.setServerTrackingData)(server, data);
         return sessionId;
     } catch (error) {
-        (0, external_logger_js_namespaceObject.log)(`Warning: PostHog MCP analytics failed to mint a stateless session id - ${error}`);
+        data.logger(`Warning: PostHog MCP analytics failed to mint a stateless session id - ${error}`);
         return;
     }
 }
@@ -289,7 +331,7 @@ function upgradeMintedTokenToNegotiated(server, data, mintedSessionId, negotiate
         });
         (0, external_session_token_js_namespaceObject.writeSessionIdToTransport)(transport, token);
     } catch (error) {
-        (0, external_logger_js_namespaceObject.log)(`Warning: PostHog MCP analytics failed to upgrade the stateless session token - ${error}`);
+        data.logger(`Warning: PostHog MCP analytics failed to upgrade the stateless session token - ${error}`);
     }
 }
 function readInitializeClientInfo(request) {
@@ -301,10 +343,10 @@ function readInitializeClientInfo(request) {
         version: 'string' == typeof version ? version : void 0
     };
 }
-async function handleInitializeRequest(server, originalInitializeHandler, request, extra) {
+async function handleInitializeRequest(server, originalInitializeHandler, 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 originalInitializeHandler(request, extra);
     }
     const mintedSessionId = mintStatelessSessionOnInitialize(server, data, request, extra);
@@ -326,7 +368,7 @@ async function handleInitializeRequest(server, originalInitializeHandler, reques
     data.sessionInfo.protocolVersion = negotiatedProtocolVersion;
     (0, external_internal_js_namespaceObject.setServerTrackingData)(server, data);
     if (mintedSessionId) upgradeMintedTokenToNegotiated(server, data, mintedSessionId, negotiatedProtocolVersion);
-    (0, external_capture_js_namespaceObject.captureEvent)(server, event);
+    (0, external_capture_js_namespaceObject.captureEvent)(server, event, data.logger);
     return result;
 }
 function readProtocolVersion(result, request) {
@@ -340,6 +382,7 @@ exports.cacheToolDescriptions = __webpack_exports__.cacheToolDescriptions;
 exports.captureToolCall = __webpack_exports__.captureToolCall;
 exports.handleInitializeRequest = __webpack_exports__.handleInitializeRequest;
 exports.handleListToolsRequest = __webpack_exports__.handleListToolsRequest;
+exports.isToolAdvertised = __webpack_exports__.isToolAdvertised;
 exports.patchRequestHandlers = __webpack_exports__.patchRequestHandlers;
 exports.readToolMetaCategory = __webpack_exports__.readToolMetaCategory;
 for(var __webpack_i__ in __webpack_exports__)if (-1 === [
@@ -348,6 +391,7 @@ for(var __webpack_i__ in __webpack_exports__)if (-1 === [
     "captureToolCall",
     "handleInitializeRequest",
     "handleListToolsRequest",
+    "isToolAdvertised",
     "patchRequestHandlers",
     "readToolMetaCategory"
 ].indexOf(__webpack_i__)) exports[__webpack_i__] = __webpack_exports__[__webpack_i__];

package/dist/extensions/instrumentation.mjs

@@ -7,7 +7,6 @@ import { MCPAnalyticsEventType } from "./event-types.mjs";
 import { captureException } from "./exceptions.mjs";
 import { resolveToolCallIntent, setEventIntent, setExplicitContextIntent } from "./intent.mjs";
 import { getServerTrackingData, handleIdentify, setServerTrackingData } from "./internal.mjs";
-import { log } from "./logger.mjs";
 import { buildCapturedMcpParameters } from "./mcp-payloads.mjs";
 import { getLiteralValue, getObjectShape } from "./mcp-sdk-compat.mjs";
 import { getSessionId, newSessionId } from "./session.mjs";
@@ -16,26 +15,26 @@ import { getReportMissingToolDescriptor, resolveMissingCapabilityToolName } from
 import { applyResolvedMetadata, isToolResultError } from "./tracing-helpers.mjs";
 async function captureToolCall(params) {
     const { server, data, request, extra, execute, parameterOwnership, eventType, explicitContextIntent, takeCapturedError } = params;
-    const ownership = getActiveAnalyticsParameterOwnership(data, request.params?.name, parameterOwnership);
-    const conversation = resolveConversationId(ownership.conversationId, request.params?.arguments, request.params?.name, resolveMissingCapabilityToolName(data.options));
+    const resolvedEventType = eventType ?? MCPAnalyticsEventType.mcpToolsCall;
+    const ownership = getActiveAnalyticsParameterOwnership(data, request.params?.name, parameterOwnership, resolvedEventType === MCPAnalyticsEventType.mcpMissingCapability);
+    const conversation = resolveConversationId(ownership.conversationId, request.params?.arguments, request.params?.name, resolvedEventType === MCPAnalyticsEventType.mcpMissingCapability ? resolveMissingCapabilityToolName(data.options) : '');
     const downstreamRequest = cloneRequestWithoutOwnedAnalyticsArguments(request, ownership);
     const startTime = new Date();
-    const event = await prepareToolCallEvent(server, data, request, downstreamRequest, extra, startTime, conversation, ownership, eventType ?? MCPAnalyticsEventType.mcpToolsCall);
+    const event = await prepareToolCallEvent(server, data, request, downstreamRequest, extra, startTime, conversation, ownership, resolvedEventType);
     if (event && explicitContextIntent) setExplicitContextIntent(event, explicitContextIntent);
     let result;
     try {
         result = await execute(downstreamRequest);
     } catch (error) {
-        publishFailedToolEvent(server, event, error, startTime, conversation);
+        publishFailedToolEvent(server, event, error, startTime, conversation, data.logger);
         throw error;
     }
     const finalResult = applyConversationPromptBack(event, result, conversation);
-    publishSuccessfulToolEvent(server, event, finalResult, startTime, takeCapturedError);
+    publishSuccessfulToolEvent(server, event, finalResult, startTime, data.logger, takeCapturedError);
     return finalResult;
 }
-function getActiveAnalyticsParameterOwnership(data, toolName, override) {
+function getActiveAnalyticsParameterOwnership(data, toolName, override, isMissingCapabilityTool) {
     const ownership = override ?? (toolName ? data.toolAnalyticsParameterOwnership.get(toolName) : void 0);
-    const isMissingCapabilityTool = toolName === resolveMissingCapabilityToolName(data.options);
     return {
         context: !isMissingCapabilityTool && isContextEnabled(data.options.context) && ownership?.context === true,
         conversationId: !isMissingCapabilityTool && true === data.options.enableConversationId && ownership?.conversationId === true
@@ -73,7 +72,7 @@ async function prepareToolCallEvent(server, data, request, downstreamRequest, ex
         setEventIntent(event, await resolveToolCallIntent(data, request, ownership.context, extra));
         return event;
     } catch (error) {
-        log(`Warning: PostHog MCP analytics instrumentation failed for tool ${request.params?.name}, the tool will still run - ${error}`);
+        data.logger(`Warning: PostHog MCP analytics instrumentation failed for tool ${request.params?.name}, the tool will still run - ${error}`);
         return null;
     }
 }
@@ -83,7 +82,7 @@ function applyConversationPromptBack(event, result, conversation) {
     if (event) event.conversationId = void 0;
     return result;
 }
-function publishSuccessfulToolEvent(server, event, result, startTime, takeCapturedError) {
+function publishSuccessfulToolEvent(server, event, result, startTime, logger, takeCapturedError) {
     if (!event) return;
     try {
         if (isToolResultError(result)) {
@@ -93,38 +92,71 @@ function publishSuccessfulToolEvent(server, event, result, startTime, takeCaptur
         } else event.isError = false;
         event.response = result;
         event.duration = Date.now() - startTime.getTime();
-        captureEvent(server, event);
+        captureEvent(server, event, logger);
     } catch (error) {
-        log(`Warning: PostHog MCP analytics failed to publish tool event - ${error}`);
+        logger(`Warning: PostHog MCP analytics failed to publish tool event - ${error}`);
     }
 }
-function publishFailedToolEvent(server, event, error, startTime, conversation) {
+function publishFailedToolEvent(server, event, error, startTime, conversation, logger) {
     if (!event) return;
     try {
         if (conversation.minted) event.conversationId = void 0;
         event.isError = true;
         event.error = captureException(error);
         event.duration = Date.now() - startTime.getTime();
-        captureEvent(server, event);
+        captureEvent(server, event, logger);
     } catch (publishError) {
-        log(`Warning: PostHog MCP analytics failed to publish failed tool event - ${publishError}`);
+        logger(`Warning: PostHog MCP analytics failed to publish failed tool event - ${publishError}`);
     }
 }
+const originalRequestHandlers = new WeakMap();
+function rememberOriginalRequestHandler(server, handlerName, originalHandler) {
+    let handlers = originalRequestHandlers.get(server);
+    if (!handlers) {
+        handlers = new Map();
+        originalRequestHandlers.set(server, handlers);
+    }
+    handlers.set(handlerName, originalHandler);
+}
 function patchRequestHandlers(server, patches) {
     for (const [handlerName, patch] of Object.entries(patches)){
         const originalHandler = server._requestHandlers.get(handlerName);
-        if (originalHandler) server._requestHandlers.set(handlerName, (request, extra)=>patch(server, originalHandler, request, extra));
+        if (originalHandler) {
+            rememberOriginalRequestHandler(server, handlerName, originalHandler);
+            server._requestHandlers.set(handlerName, (request, extra)=>patch(server, originalHandler, request, extra));
+        }
     }
     const originalSetRequestHandler = server.setRequestHandler.bind(server);
     server.setRequestHandler = (requestSchema, originalHandler)=>{
         const shape = getObjectShape(requestSchema);
         const handlerName = shape?.method ? getLiteralValue(shape.method) : void 0;
         const patch = 'string' == typeof handlerName ? patches[handlerName] : void 0;
-        if (!patch) return originalSetRequestHandler(requestSchema, originalHandler);
-        return originalSetRequestHandler(requestSchema, (request, extra)=>patch(server, originalHandler, request, extra));
+        if (!patch || 'string' != typeof handlerName) return originalSetRequestHandler(requestSchema, originalHandler);
+        const result = originalSetRequestHandler(requestSchema, originalHandler);
+        const registeredHandler = server._requestHandlers.get(handlerName);
+        if (registeredHandler) {
+            rememberOriginalRequestHandler(server, handlerName, registeredHandler);
+            server._requestHandlers.set(handlerName, (request, extra)=>patch(server, registeredHandler, request, extra));
+        }
+        return result;
     };
 }
-async function handleListToolsRequest(server, originalListToolsHandler, request, extra) {
+async function isToolAdvertised(server, toolName, extra, logger) {
+    const listHandler = originalRequestHandlers.get(server)?.get('tools/list');
+    if (!listHandler || !server._requestHandlers.has('tools/list')) return;
+    try {
+        const response = await listHandler({
+            method: 'tools/list',
+            params: {}
+        }, extra);
+        if (!response || !Array.isArray(response.tools)) return;
+        return response.tools.some((tool)=>tool?.name === toolName);
+    } catch (error) {
+        logger(`Warning: PostHog MCP analytics could not determine whether "${toolName}" is advertised; delegating to the server - ${error}`);
+        return;
+    }
+}
+async function handleListToolsRequest(server, originalListToolsHandler, request, extra, logger) {
     const data = getServerTrackingData(server);
     const startTime = new Date();
     const event = {
@@ -135,19 +167,19 @@ async function handleListToolsRequest(server, originalListToolsHandler, request,
     };
     stampMetaClientInfo(event, request);
     if (data) await applyResolvedMetadata(event, data, request, extra);
-    const tools = await getTracedToolsList(server, originalListToolsHandler, request, extra, event);
+    const tools = await getTracedToolsList(server, originalListToolsHandler, request, extra, event, logger);
     if (!data) {
-        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 {
             tools
         };
     }
     if (0 === tools.length) {
-        log('Warning: No tools found in the original list. This is likely due to the tools not being registered before PostHog MCP analytics.instrument().');
+        data.logger('Warning: No tools found in the original list. This is likely due to the tools not being registered before PostHog MCP analytics.instrument().');
         event.error = captureException('No tools were sent to MCP client.');
         event.isError = true;
         event.duration = Date.now() - startTime.getTime();
-        captureEvent(server, event);
+        captureEvent(server, event, data.logger);
         return {
             tools
         };
@@ -158,7 +190,7 @@ async function handleListToolsRequest(server, originalListToolsHandler, request,
     event.listedToolNames = collectListedToolNames(tools);
     event.isError = false;
     event.duration = Date.now() - startTime.getTime();
-    captureEvent(server, event);
+    captureEvent(server, event, data.logger);
     return {
         tools
     };
@@ -171,18 +203,27 @@ function collectListedToolNames(tools) {
     const names = tools.map((tool)=>tool?.name).filter((name)=>'string' == typeof name);
     return names.length > 0 ? names : void 0;
 }
-async function getTracedToolsList(server, originalListToolsHandler, request, extra, event) {
+async function getTracedToolsList(server, originalListToolsHandler, request, extra, event, logger) {
     try {
         const data = getServerTrackingData(server);
         const originalResponse = await originalListToolsHandler(request, extra);
-        let tools = originalResponse.tools || [];
+        let tools = [
+            ...originalResponse.tools || []
+        ];
         if (data) cacheToolAnalyticsParameterOwnership(data.toolAnalyticsParameterOwnership, tools);
-        if (data && isContextEnabled(data.options.context)) tools = addContextParameterToTools(tools, getContextDescription(data.options.context));
-        if (data?.options.enableConversationId) tools = addConversationIdToTools(tools, resolveMissingCapabilityToolName(data.options));
-        if (data?.options.reportMissing) {
+        if (data && isContextEnabled(data.options.context)) tools = addContextParameterToTools(tools, getContextDescription(data.options.context), data.logger);
+        if (data) {
             const missingToolName = resolveMissingCapabilityToolName(data.options);
-            const alreadyPresent = tools.some((tool)=>tool?.name === missingToolName);
-            if (!alreadyPresent) tools.push(getReportMissingToolDescriptor(missingToolName));
+            let injectedMissingCapabilityTool = false;
+            if (data.options.reportMissing) {
+                const alreadyPresent = tools.some((tool)=>tool?.name === missingToolName);
+                if (alreadyPresent) data.logger(`Warning: Cannot inject missing-capability tool "${missingToolName}" because a real tool already uses that name. The real tool will not be intercepted.`);
+                else {
+                    tools.push(getReportMissingToolDescriptor(missingToolName));
+                    injectedMissingCapabilityTool = true;
+                }
+            }
+            if (data.options.enableConversationId) tools = addConversationIdToTools(tools, missingToolName, injectedMissingCapabilityTool, data.logger);
         }
         if (data) {
             cacheToolDescriptions(data.toolDescriptions, tools);
@@ -190,11 +231,11 @@ async function getTracedToolsList(server, originalListToolsHandler, request, ext
         }
         return tools;
     } catch (error) {
-        log(`Warning: Original list tools handler failed, this suggests an error PostHog MCP analytics did not cause - ${error}`);
+        logger(`Warning: Original list tools handler failed, this suggests an error PostHog MCP analytics did not cause - ${error}`);
         event.error = captureException(error);
         event.isError = true;
         event.duration = event.timestamp ? Date.now() - event.timestamp.getTime() : 0;
-        captureEvent(server, event);
+        captureEvent(server, event, logger);
         throw error;
     }
 }
@@ -239,7 +280,7 @@ function mintStatelessSessionOnInitialize(server, data, request, extra) {
         setServerTrackingData(server, data);
         return sessionId;
     } catch (error) {
-        log(`Warning: PostHog MCP analytics failed to mint a stateless session id - ${error}`);
+        data.logger(`Warning: PostHog MCP analytics failed to mint a stateless session id - ${error}`);
         return;
     }
 }
@@ -255,7 +296,7 @@ function upgradeMintedTokenToNegotiated(server, data, mintedSessionId, negotiate
         });
         writeSessionIdToTransport(transport, token);
     } catch (error) {
-        log(`Warning: PostHog MCP analytics failed to upgrade the stateless session token - ${error}`);
+        data.logger(`Warning: PostHog MCP analytics failed to upgrade the stateless session token - ${error}`);
     }
 }
 function readInitializeClientInfo(request) {
@@ -267,10 +308,10 @@ function readInitializeClientInfo(request) {
         version: 'string' == typeof version ? version : void 0
     };
 }
-async function handleInitializeRequest(server, originalInitializeHandler, request, extra) {
+async function handleInitializeRequest(server, originalInitializeHandler, request, extra, logger) {
     const data = getServerTrackingData(server);
     if (!data) {
-        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 originalInitializeHandler(request, extra);
     }
     const mintedSessionId = mintStatelessSessionOnInitialize(server, data, request, extra);
@@ -292,7 +333,7 @@ async function handleInitializeRequest(server, originalInitializeHandler, reques
     data.sessionInfo.protocolVersion = negotiatedProtocolVersion;
     setServerTrackingData(server, data);
     if (mintedSessionId) upgradeMintedTokenToNegotiated(server, data, mintedSessionId, negotiatedProtocolVersion);
-    captureEvent(server, event);
+    captureEvent(server, event, data.logger);
     return result;
 }
 function readProtocolVersion(result, request) {
@@ -301,4 +342,4 @@ function readProtocolVersion(result, request) {
     const requested = request.params?.protocolVersion;
     return 'string' == typeof requested && requested.length > 0 ? requested : void 0;
 }
-export { cacheToolCategories, cacheToolDescriptions, captureToolCall, handleInitializeRequest, handleListToolsRequest, patchRequestHandlers, readToolMetaCategory };
+export { cacheToolCategories, cacheToolDescriptions, captureToolCall, handleInitializeRequest, handleListToolsRequest, isToolAdvertised, patchRequestHandlers, readToolMetaCategory };

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

@@ -1 +1 @@
-{"version":3,"file":"intent.d.ts","sourceRoot":"","sources":["../../src/extensions/intent.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EACV,6BAA6B,EAC7B,gBAAgB,EAChB,wBAAwB,EACxB,cAAc,EACd,QAAQ,EACT,MAAM,UAAU,CAAA;AAIjB,UAAU,cAAc;IACtB,MAAM,EAAE,MAAM,CAAA;IACd,MAAM,EAAE,wBAAwB,CAAA;CACjC;AAkCD,wBAAsB,qBAAqB,CACzC,IAAI,EAAE,gBAAgB,EACtB,OAAO,EAAE,cAAc,EACvB,oBAAoB,EAAE,OAAO,EAC7B,KAAK,CAAC,EAAE,6BAA6B,GACpC,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC,CAYhC;AAED,wBAAgB,cAAc,CAAC,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,cAAc,GAAG,IAAI,GAAG,IAAI,CAO3F;AAED,wBAAgB,wBAAwB,CAAC,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAQ/E"}
+{"version":3,"file":"intent.d.ts","sourceRoot":"","sources":["../../src/extensions/intent.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EACV,6BAA6B,EAC7B,gBAAgB,EAChB,wBAAwB,EACxB,cAAc,EACd,QAAQ,EACT,MAAM,UAAU,CAAA;AAGjB,UAAU,cAAc;IACtB,MAAM,EAAE,MAAM,CAAA;IACd,MAAM,EAAE,wBAAwB,CAAA;CACjC;AAkCD,wBAAsB,qBAAqB,CACzC,IAAI,EAAE,gBAAgB,EACtB,OAAO,EAAE,cAAc,EACvB,oBAAoB,EAAE,OAAO,EAC7B,KAAK,CAAC,EAAE,6BAA6B,GACpC,OAAO,CAAC,cAAc,GAAG,IAAI,CAAC,CAOhC;AAED,wBAAgB,cAAc,CAAC,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,cAAc,GAAG,IAAI,GAAG,IAAI,CAO3F;AAED,wBAAgB,wBAAwB,CAAC,KAAK,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,CAQ/E"}

package/dist/extensions/intent.js

@@ -29,7 +29,6 @@ __webpack_require__.d(__webpack_exports__, {
     setEventIntent: ()=>setEventIntent
 });
 const external_context_parameters_js_namespaceObject = require("./context-parameters.js");
-const external_logger_js_namespaceObject = require("./logger.js");
 function getContextArgument(request) {
     const context = request.params?.arguments?.context;
     return 'string' == typeof context && context.trim() ? context : void 0;
@@ -48,13 +47,13 @@ async function runIntentFallback(data, request, extra) {
             source: 'inferred'
         } : null;
     } catch (error) {
-        (0, external_logger_js_namespaceObject.log)(`intentFallback callback error: ${error}`);
+        data.logger(`intentFallback callback error: ${error}`);
         return null;
     }
 }
 async function resolveToolCallIntent(data, request, analyticsOwnsContext, extra) {
     const contextArgument = getContextArgument(request);
-    if (analyticsOwnsContext && (0, external_context_parameters_js_namespaceObject.isContextEnabled)(data.options.context) && request.params?.name !== 'get_more_tools' && contextArgument) return {
+    if (analyticsOwnsContext && (0, external_context_parameters_js_namespaceObject.isContextEnabled)(data.options.context) && contextArgument) return {
         intent: contextArgument,
         source: 'context_parameter'
     };

package/dist/extensions/intent.mjs

@@ -1,5 +1,4 @@
 import { isContextEnabled } from "./context-parameters.mjs";
-import { log } from "./logger.mjs";
 function getContextArgument(request) {
     const context = request.params?.arguments?.context;
     return 'string' == typeof context && context.trim() ? context : void 0;
@@ -18,13 +17,13 @@ async function runIntentFallback(data, request, extra) {
             source: 'inferred'
         } : null;
     } catch (error) {
-        log(`intentFallback callback error: ${error}`);
+        data.logger(`intentFallback callback error: ${error}`);
         return null;
     }
 }
 async function resolveToolCallIntent(data, request, analyticsOwnsContext, extra) {
     const contextArgument = getContextArgument(request);
-    if (analyticsOwnsContext && isContextEnabled(data.options.context) && request.params?.name !== 'get_more_tools' && contextArgument) return {
+    if (analyticsOwnsContext && isContextEnabled(data.options.context) && contextArgument) return {
         intent: contextArgument,
         source: 'context_parameter'
     };

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

@@ -1 +1 @@
-{"version":3,"file":"internal.d.ts","sourceRoot":"","sources":["../../src/extensions/internal.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EACV,6BAA6B,EAC7B,gBAAgB,EAChB,cAAc,EACd,aAAa,EAEb,YAAY,EACb,MAAM,UAAU,CAAA;AAMjB;;;;;GAKG;AACH,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAkC;IACzD,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAQ;gBAErB,OAAO,SAAO;IAI1B,GAAG,CAAC,SAAS,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS;IAWhD,GAAG,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,YAAY,GAAG,IAAI;IAapD,GAAG,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAI/B,IAAI,IAAI,MAAM;CAGf;AAID,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,aAAa,GAAG,gBAAgB,GAAG,SAAS,CAEzF;AAED,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,aAAa,EAAE,IAAI,EAAE,gBAAgB,GAAG,IAAI,CAEzF;AAED,wBAAgB,kBAAkB,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,YAAY,GAAG,OAAO,CA4B5E;AAED,wBAAgB,eAAe,CAAC,QAAQ,EAAE,YAAY,GAAG,SAAS,EAAE,IAAI,EAAE,YAAY,GAAG,YAAY,CAapG;AAED;;;;;GAKG;AACH,wBAAsB,cAAc,CAClC,MAAM,EAAE,aAAa,EACrB,IAAI,EAAE,gBAAgB,EACtB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,cAAc,EACvB,KAAK,CAAC,EAAE,6BAA6B,GACpC,OAAO,CAAC,IAAI,CAAC,CAgDf;AAED;;;;GAIG;AACH,wBAAsB,sBAAsB,CAC1C,IAAI,EAAE,gBAAgB,EACtB,OAAO,EAAE,cAAc,EACvB,KAAK,CAAC,EAAE,6BAA6B,GACpC,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC,CAUzC"}
+{"version":3,"file":"internal.d.ts","sourceRoot":"","sources":["../../src/extensions/internal.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EACV,6BAA6B,EAC7B,gBAAgB,EAChB,cAAc,EACd,aAAa,EAEb,YAAY,EACb,MAAM,UAAU,CAAA;AAKjB;;;;;GAKG;AACH,qBAAa,aAAa;IACxB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAkC;IACzD,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAQ;gBAErB,OAAO,SAAO;IAI1B,GAAG,CAAC,SAAS,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS;IAWhD,GAAG,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,YAAY,GAAG,IAAI;IAapD,GAAG,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO;IAI/B,IAAI,IAAI,MAAM;CAGf;AAID,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,aAAa,GAAG,gBAAgB,GAAG,SAAS,CAEzF;AAED,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,aAAa,EAAE,IAAI,EAAE,gBAAgB,GAAG,IAAI,CAEzF;AAED,wBAAgB,kBAAkB,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,YAAY,GAAG,OAAO,CA4B5E;AAED,wBAAgB,eAAe,CAAC,QAAQ,EAAE,YAAY,GAAG,SAAS,EAAE,IAAI,EAAE,YAAY,GAAG,YAAY,CAapG;AAED;;;;;GAKG;AACH,wBAAsB,cAAc,CAClC,MAAM,EAAE,aAAa,EACrB,IAAI,EAAE,gBAAgB,EACtB,SAAS,EAAE,MAAM,EACjB,OAAO,EAAE,cAAc,EACvB,KAAK,CAAC,EAAE,6BAA6B,GACpC,OAAO,CAAC,IAAI,CAAC,CAkDf;AAED;;;;GAIG;AACH,wBAAsB,sBAAsB,CAC1C,IAAI,EAAE,gBAAgB,EACtB,OAAO,EAAE,cAAc,EACvB,KAAK,CAAC,EAAE,6BAA6B,GACpC,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC,CAUzC"}