@posthog/mcp 0.4.1 → 0.4.2

package/dist/extensions/instrumentation.d.ts

@@ -47,6 +47,13 @@ export declare function captureToolCall(params: TraceToolCallParams): Promise<un
  * SDK-managed tools (context parameter, conversation id, `get_more_tools`) are
  * injected. Idempotent per server. Works for both low-level and high-level
  * servers — the high-level wrapper passes its underlying `server`.
+ *
+ * The handler can reach us two ways, and we cover both:
+ *  1. It already exists when instrument() runs (tools registered first) — wrap
+ *     the entry in `_requestHandlers` directly.
+ *  2. It's registered after instrument() (e.g. `@rekog/mcp-nest` registers it at
+ *     the low level once it gets the server) — intercept `setRequestHandler` and
+ *     wrap the handler as it lands, mirroring how `tools/call` already works.
  */
 export declare function instrumentToolsListHandler(server: MCPServerLike): void;
 export declare function cacheToolDescriptions(cache: Map<string, string>, tools: ListToolsResult['tools'] | undefined): void;

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

@@ -1 +1 @@
-{"version":3,"file":"instrumentation.d.ts","sourceRoot":"","sources":["../../src/extensions/instrumentation.ts"],"names":[],"mappings":"AAIA,OAAO,EAGL,KAAK,eAAe,EACrB,MAAM,oCAAoC,CAAA;AAC3C,OAAO,KAAK,EAAE,6BAA6B,EAAE,gBAAgB,EAAE,cAAc,EAAE,aAAa,EAAY,MAAM,UAAU,CAAA;AAWxH,OAAO,EAAE,qBAAqB,EAAE,MAAM,eAAe,CAAA;AAoBrD;;;;GAIG;AACH,KAAK,YAAY,GAAG,CAAC,iBAAiB,EAAE,cAAc,KAAK,OAAO,CAAC,OAAO,CAAC,CAAA;AAE3E,UAAU,mBAAmB;IAC3B,MAAM,EAAE,aAAa,CAAA;IACrB,IAAI,EAAE,gBAAgB,CAAA;IACtB,OAAO,EAAE,cAAc,CAAA;IACvB,KAAK,CAAC,EAAE,6BAA6B,CAAA;IACrC,OAAO,EAAE,YAAY,CAAA;IACrB;;;;OAIG;IACH,SAAS,CAAC,EAAE,qBAAqB,CAAA;IACjC;;;;OAIG;IACH,qBAAqB,CAAC,EAAE,MAAM,CAAA;IAC9B;;;;;OAKG;IACH,iBAAiB,CAAC,EAAE,MAAM,OAAO,CAAA;CAClC;AAED;;;;;;;GAOG;AACH,wBAAsB,eAAe,CAAC,MAAM,EAAE,mBAAmB,GAAG,OAAO,CAAC,OAAO,CAAC,CAuCnF;AAwHD;;;;;GAKG;AACH,wBAAgB,0BAA0B,CAAC,MAAM,EAAE,aAAa,GAAG,IAAI,CAsBtE;AAuGD,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,KAAK,EAAE,eAAe,CAAC,OAAO,CAAC,GAAG,SAAS,GAAG,IAAI,CASnH;AAED;;;;GAIG;AACH,wBAAgB,oBAAoB,CAAC,IAAI,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS,CAGtE;AAED,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,KAAK,EAAE,eAAe,CAAC,OAAO,CAAC,GAAG,SAAS,GAAG,IAAI,CAUjH;AAID;;;GAGG;AACH,wBAAgB,2BAA2B,CAAC,MAAM,EAAE,aAAa,GAAG,IAAI,CAiCvE"}
+{"version":3,"file":"instrumentation.d.ts","sourceRoot":"","sources":["../../src/extensions/instrumentation.ts"],"names":[],"mappings":"AAIA,OAAO,EAA2B,KAAK,eAAe,EAAE,MAAM,oCAAoC,CAAA;AAClG,OAAO,KAAK,EAAE,6BAA6B,EAAE,gBAAgB,EAAE,cAAc,EAAE,aAAa,EAAY,MAAM,UAAU,CAAA;AAWxH,OAAO,EAAE,qBAAqB,EAAE,MAAM,eAAe,CAAA;AAqBrD;;;;GAIG;AACH,KAAK,YAAY,GAAG,CAAC,iBAAiB,EAAE,cAAc,KAAK,OAAO,CAAC,OAAO,CAAC,CAAA;AAE3E,UAAU,mBAAmB;IAC3B,MAAM,EAAE,aAAa,CAAA;IACrB,IAAI,EAAE,gBAAgB,CAAA;IACtB,OAAO,EAAE,cAAc,CAAA;IACvB,KAAK,CAAC,EAAE,6BAA6B,CAAA;IACrC,OAAO,EAAE,YAAY,CAAA;IACrB;;;;OAIG;IACH,SAAS,CAAC,EAAE,qBAAqB,CAAA;IACjC;;;;OAIG;IACH,qBAAqB,CAAC,EAAE,MAAM,CAAA;IAC9B;;;;;OAKG;IACH,iBAAiB,CAAC,EAAE,MAAM,OAAO,CAAA;CAClC;AAED;;;;;;;GAOG;AACH,wBAAsB,eAAe,CAAC,MAAM,EAAE,mBAAmB,GAAG,OAAO,CAAC,OAAO,CAAC,CAuCnF;AAwHD;;;;;;;;;;;;GAYG;AACH,wBAAgB,0BAA0B,CAAC,MAAM,EAAE,aAAa,GAAG,IAAI,CAiCtE;AAuGD,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,KAAK,EAAE,eAAe,CAAC,OAAO,CAAC,GAAG,SAAS,GAAG,IAAI,CASnH;AAED;;;;GAIG;AACH,wBAAgB,oBAAoB,CAAC,IAAI,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS,CAGtE;AAED,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,KAAK,EAAE,eAAe,CAAC,OAAO,CAAC,GAAG,SAAS,GAAG,IAAI,CAUjH;AAID;;;GAGG;AACH,wBAAgB,2BAA2B,CAAC,MAAM,EAAE,aAAa,GAAG,IAAI,CAiCvE"}

package/dist/extensions/instrumentation.js

@@ -41,6 +41,7 @@ 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");
 const external_tools_js_namespaceObject = require("./tools.js");
 const external_tracing_helpers_js_namespaceObject = require("./tracing-helpers.js");
@@ -122,11 +123,17 @@ const listToolsTracingSetup = new WeakMap();
 function instrumentToolsListHandler(server) {
     if (!server._capabilities?.tools) return;
     if (listToolsTracingSetup.get(server)) return;
-    const originalListToolsHandler = server._requestHandlers.get('tools/list');
-    if (!originalListToolsHandler) return;
+    listToolsTracingSetup.set(server, true);
     try {
-        server.setRequestHandler(types_js_namespaceObject.ListToolsRequestSchema, async (request, extra)=>await handleListToolsRequest(server, originalListToolsHandler, request, extra));
-        listToolsTracingSetup.set(server, true);
+        const existingHandler = server._requestHandlers.get('tools/list');
+        if (existingHandler) server._requestHandlers.set('tools/list', (request, extra)=>handleListToolsRequest(server, existingHandler, request, extra));
+        const originalSetRequestHandler = server.setRequestHandler.bind(server);
+        server.setRequestHandler = (requestSchema, handler)=>{
+            const shape = (0, external_mcp_sdk_compat_js_namespaceObject.getObjectShape)(requestSchema);
+            const method = shape?.method ? (0, external_mcp_sdk_compat_js_namespaceObject.getLiteralValue)(shape.method) : void 0;
+            if ('tools/list' === method) return originalSetRequestHandler(requestSchema, (request, extra)=>handleListToolsRequest(server, handler, request, extra));
+            return originalSetRequestHandler(requestSchema, handler);
+        };
     } catch (error) {
         (0, external_logger_js_namespaceObject.log)(`Warning: Failed to override list tools handler - ${error}`);
     }

package/dist/extensions/instrumentation.mjs

@@ -1,4 +1,4 @@
-import { InitializeRequestSchema, ListToolsRequestSchema } from "@modelcontextprotocol/sdk/types.js";
+import { InitializeRequestSchema } from "@modelcontextprotocol/sdk/types.js";
 import { addContextParameterToTools, getContextDescription, isContextEnabled } from "./context-parameters.mjs";
 import { addConversationIdToTools, canInjectConversationIdPromptBack, cloneRequestWithoutConversationId, injectConversationIdPromptBack, resolveConversationId } from "./conversation-id.mjs";
 import { captureEvent } from "./capture.mjs";
@@ -8,6 +8,7 @@ import { resolveToolCallIntent, setEventIntent, setExplicitContextIntent } from
 import { getServerTrackingData, handleIdentify } from "./internal.mjs";
 import { log } from "./logger.mjs";
 import { buildCapturedMcpParameters } from "./mcp-payloads.mjs";
+import { getLiteralValue, getObjectShape } from "./mcp-sdk-compat.mjs";
 import { getServerSessionId } from "./session.mjs";
 import { getReportMissingToolDescriptor, resolveMissingCapabilityToolName } from "./tools.mjs";
 import { applyResolvedMetadata, isToolResultError } from "./tracing-helpers.mjs";
@@ -89,11 +90,17 @@ const listToolsTracingSetup = new WeakMap();
 function instrumentToolsListHandler(server) {
     if (!server._capabilities?.tools) return;
     if (listToolsTracingSetup.get(server)) return;
-    const originalListToolsHandler = server._requestHandlers.get('tools/list');
-    if (!originalListToolsHandler) return;
+    listToolsTracingSetup.set(server, true);
     try {
-        server.setRequestHandler(ListToolsRequestSchema, async (request, extra)=>await handleListToolsRequest(server, originalListToolsHandler, request, extra));
-        listToolsTracingSetup.set(server, true);
+        const existingHandler = server._requestHandlers.get('tools/list');
+        if (existingHandler) server._requestHandlers.set('tools/list', (request, extra)=>handleListToolsRequest(server, existingHandler, request, extra));
+        const originalSetRequestHandler = server.setRequestHandler.bind(server);
+        server.setRequestHandler = (requestSchema, handler)=>{
+            const shape = getObjectShape(requestSchema);
+            const method = shape?.method ? getLiteralValue(shape.method) : void 0;
+            if ('tools/list' === method) return originalSetRequestHandler(requestSchema, (request, extra)=>handleListToolsRequest(server, handler, request, extra));
+            return originalSetRequestHandler(requestSchema, handler);
+        };
     } catch (error) {
         log(`Warning: Failed to override list tools handler - ${error}`);
     }

package/dist/version.d.ts

@@ -1,2 +1,2 @@
-export declare const version = "0.4.1";
+export declare const version = "0.4.2";
 //# sourceMappingURL=version.d.ts.map

package/dist/version.js

@@ -26,7 +26,7 @@ __webpack_require__.r(__webpack_exports__);
 __webpack_require__.d(__webpack_exports__, {
     version: ()=>version
 });
-const version = '0.4.1';
+const version = '0.4.2';
 exports.version = __webpack_exports__.version;
 for(var __webpack_i__ in __webpack_exports__)if (-1 === [
     "version"

package/dist/version.mjs

@@ -1,2 +1,2 @@
-const version = '0.4.1';
+const version = '0.4.2';
 export { version };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@posthog/mcp",
-  "version": "0.4.1",
+  "version": "0.4.2",
   "bugs": {
     "url": "https://github.com/PostHog/posthog-js/issues"
   },
@@ -28,7 +28,7 @@
   "module": "dist/index.mjs",
   "types": "dist/index.d.ts",
   "dependencies": {
-    "@posthog/core": "^1.36.0"
+    "@posthog/core": "^1.37.3"
   },
   "devDependencies": {
     "@babel/preset-env": "^7.27.1",
@@ -40,7 +40,7 @@
     "jest": "29.7.0",
     "zod": "^3.25.0",
     "@posthog-tooling/tsconfig-base": "1.1.1",
-    "posthog-node": "^5.38.2"
+    "posthog-node": "^5.38.5"
   },
   "peerDependencies": {
     "@modelcontextprotocol/sdk": ">=1.26.0",

package/src/extensions/instrumentation.ts

@@ -2,11 +2,7 @@
 // Copyright (c) 2025 MCPcat
 // Licensed under the MIT License: https://github.com/MCPCat/mcpcat-typescript-sdk/blob/main/LICENSE
 
-import {
-  InitializeRequestSchema,
-  ListToolsRequestSchema,
-  type ListToolsResult,
-} from '@modelcontextprotocol/sdk/types.js'
+import { InitializeRequestSchema, type ListToolsResult } from '@modelcontextprotocol/sdk/types.js'
 import type { CompatibleRequestHandlerExtra, MCPAnalyticsData, MCPRequestLike, MCPServerLike, McpEvent } from '../types'
 import { addContextParameterToTools, getContextDescription, isContextEnabled } from './context-parameters'
 import {
@@ -24,6 +20,7 @@ import { resolveToolCallIntent, setEventIntent, setExplicitContextIntent } from
 import { getServerTrackingData, handleIdentify } from './internal'
 import { log } from './logger'
 import { buildCapturedMcpParameters } from './mcp-payloads'
+import { getLiteralValue, getObjectShape } from './mcp-sdk-compat'
 import { getServerSessionId } from './session'
 import { getReportMissingToolDescriptor, resolveMissingCapabilityToolName } from './tools'
 import { applyResolvedMetadata, isToolResultError } from './tracing-helpers'
@@ -244,6 +241,13 @@ const listToolsTracingSetup = new WeakMap<MCPServerLike, boolean>()
  * SDK-managed tools (context parameter, conversation id, `get_more_tools`) are
  * injected. Idempotent per server. Works for both low-level and high-level
  * servers — the high-level wrapper passes its underlying `server`.
+ *
+ * The handler can reach us two ways, and we cover both:
+ *  1. It already exists when instrument() runs (tools registered first) — wrap
+ *     the entry in `_requestHandlers` directly.
+ *  2. It's registered after instrument() (e.g. `@rekog/mcp-nest` registers it at
+ *     the low level once it gets the server) — intercept `setRequestHandler` and
+ *     wrap the handler as it lands, mirroring how `tools/call` already works.
  */
 export function instrumentToolsListHandler(server: MCPServerLike): void {
   if (!(server as MCPServerWithCapabilities)._capabilities?.tools) {
@@ -252,18 +256,29 @@ export function instrumentToolsListHandler(server: MCPServerLike): void {
   if (listToolsTracingSetup.get(server)) {
     return
   }
-
-  const originalListToolsHandler = server._requestHandlers.get('tools/list')
-  if (!originalListToolsHandler) {
-    return
-  }
+  listToolsTracingSetup.set(server, true)
 
   try {
-    server.setRequestHandler(
-      ListToolsRequestSchema,
-      async (request, extra) => await handleListToolsRequest(server, originalListToolsHandler, request, extra)
-    )
-    listToolsTracingSetup.set(server, true)
+    // A handler already registered before instrument() — wrap it in place.
+    const existingHandler = server._requestHandlers.get('tools/list')
+    if (existingHandler) {
+      server._requestHandlers.set('tools/list', (request, extra) =>
+        handleListToolsRequest(server, existingHandler, request, extra)
+      )
+    }
+
+    // A handler registered after instrument() — intercept the registration.
+    const originalSetRequestHandler = server.setRequestHandler.bind(server)
+    server.setRequestHandler = ((requestSchema: unknown, handler: MCPRequestHandler) => {
+      const shape = getObjectShape(requestSchema)
+      const method = shape?.method ? getLiteralValue(shape.method) : undefined
+      if (method === 'tools/list') {
+        return originalSetRequestHandler(requestSchema, (request, extra) =>
+          handleListToolsRequest(server, handler, request, extra)
+        )
+      }
+      return originalSetRequestHandler(requestSchema, handler)
+    }) as MCPServerLike['setRequestHandler']
   } catch (error) {
     log(`Warning: Failed to override list tools handler - ${error}`)
   }

package/src/version.ts

@@ -1 +1 @@
-export const version = '0.4.1'
+export const version = '0.4.2'