@posthog/mcp 0.0.4 → 0.0.6

package/dist/index.d.cts

@@ -105,6 +105,7 @@ declare const PostHogMCPAnalyticsProperty: {
   readonly ServerVersion: "$mcp_server_version";
   readonly SessionId: "$session_id";
   readonly Source: "$mcp_source";
+  readonly ToolDescription: "$mcp_tool_description";
   readonly ToolName: "$mcp_tool_name";
 };
 type PostHogMCPAnalyticsProperty = (typeof PostHogMCPAnalyticsProperty)[keyof typeof PostHogMCPAnalyticsProperty];

package/dist/index.d.mts

@@ -104,6 +104,7 @@ declare const PostHogMCPAnalyticsProperty: {
   readonly ServerVersion: "$mcp_server_version";
   readonly SessionId: "$session_id";
   readonly Source: "$mcp_source";
+  readonly ToolDescription: "$mcp_tool_description";
   readonly ToolName: "$mcp_tool_name";
 };
 type PostHogMCPAnalyticsProperty = (typeof PostHogMCPAnalyticsProperty)[keyof typeof PostHogMCPAnalyticsProperty];

package/dist/index.mjs

@@ -388,6 +388,7 @@ const PostHogMCPAnalyticsProperty = {
 	ServerVersion: "$mcp_server_version",
 	SessionId: "$session_id",
 	Source: "$mcp_source",
+	ToolDescription: "$mcp_tool_description",
 	ToolName: "$mcp_tool_name"
 };
 //#endregion
@@ -456,6 +457,7 @@ function addCommonEventProperties(event, properties) {
 		properties[PostHogMCPAnalyticsProperty.ResourceName] = event.resourceName;
 		if (event.eventType === MCPAnalyticsEventType.mcpToolsCall) properties[PostHogMCPAnalyticsProperty.ToolName] = event.resourceName;
 	}
+	if (event.toolDescription && event.eventType === MCPAnalyticsEventType.mcpToolsCall) properties[PostHogMCPAnalyticsProperty.ToolDescription] = event.toolDescription;
 	if (event.duration !== void 0) properties[PostHogMCPAnalyticsProperty.DurationMs] = event.duration;
 	if (event.serverName) properties[PostHogMCPAnalyticsProperty.ServerName] = event.serverName;
 	if (event.serverVersion) properties[PostHogMCPAnalyticsProperty.ServerVersion] = event.serverVersion;
@@ -492,10 +494,12 @@ function buildExceptionEvent(event) {
 		properties[PostHogMCPAnalyticsProperty.ResourceName] = event.resourceName;
 		if (event.eventType === MCPAnalyticsEventType.mcpToolsCall) properties[PostHogMCPAnalyticsProperty.ToolName] = event.resourceName;
 	}
+	if (event.toolDescription && event.eventType === MCPAnalyticsEventType.mcpToolsCall) properties[PostHogMCPAnalyticsProperty.ToolDescription] = event.toolDescription;
 	if (event.serverName) properties[PostHogMCPAnalyticsProperty.ServerName] = event.serverName;
 	if (event.serverVersion) properties[PostHogMCPAnalyticsProperty.ServerVersion] = event.serverVersion;
 	if (event.clientName) properties[PostHogMCPAnalyticsProperty.ClientName] = event.clientName;
 	if (event.clientVersion) properties[PostHogMCPAnalyticsProperty.ClientVersion] = event.clientVersion;
+	addCustomEventProperties(event, properties);
 	return {
 		event: PostHogMCPAnalyticsEvent.Exception,
 		distinct_id: distinctId,
@@ -551,6 +555,7 @@ const PROTECTED_FIELDS = new Set([
 	"identifyActorName",
 	"identifyData",
 	"resourceName",
+	"toolDescription",
 	"eventType",
 	"actorId",
 	"tags",
@@ -728,7 +733,7 @@ function sanitizeParameters(obj) {
 }
 //#endregion
 //#region package.json
-var version = "0.0.4";
+var version = "0.0.6";
 //#endregion
 //#region src/modules/session.ts
 function newSessionId() {
@@ -1188,6 +1193,7 @@ function publishEvent(server, eventInput) {
 		identifyActorName: sessionInfo.identifyActorName,
 		identifyActorData: sessionInfo.identifyActorData,
 		resourceName: eventInput.resourceName,
+		toolDescription: eventInput.toolDescription,
 		parameters: eventInput.parameters,
 		response: eventInput.response,
 		userIntent: eventInput.userIntent,
@@ -2136,6 +2142,7 @@ async function getTracedToolsList(server, originalListToolsHandler, request, ext
 		if (data?.options.reportMissing) {
 			if (!tools.some((tool) => tool?.name === "get_more_tools")) tools.push(getReportMissingToolDescriptor());
 		}
+		if (data) cacheToolDescriptions(data.toolDescriptions, tools);
 		return tools;
 	} catch (error) {
 		writeToLog(`Warning: Original list tools handler failed, this suggests an error PostHog MCP analytics did not cause - ${error}`);
@@ -2219,13 +2226,15 @@ async function handleToolCallRequest(server, originalCallToolHandler, request, e
 	}
 	const conversation = resolveConversationId(data.options.enableConversationId ?? false, request.params?.arguments, request.params?.name);
 	const downstreamRequest = conversation.conversationId ? cloneRequestWithoutConversationId(request) : request;
+	const toolName = request.params?.name;
 	const event = {
 		sessionId: getServerSessionId(server, extra),
 		conversationId: conversation.conversationId,
-		resourceName: request.params?.name || "Unknown Tool Name",
+		resourceName: toolName || "Unknown Tool Name",
 		parameters: buildCapturedMcpParameters(downstreamRequest),
 		eventType: MCPAnalyticsEventType.mcpToolsCall,
 		timestamp: /* @__PURE__ */ new Date(),
+		toolDescription: toolName ? data.toolDescriptions.get(toolName) : void 0,
 		redactionFn: data.options.redactSensitiveInformation
 	};
 	try {
@@ -2279,6 +2288,10 @@ function getContextArgument$1(request) {
 function getEventDuration(event) {
 	return event.timestamp ? Date.now() - event.timestamp.getTime() : 0;
 }
+function cacheToolDescriptions(cache, tools) {
+	if (!tools) return;
+	for (const tool of tools) if (tool?.name && typeof tool.description === "string") cache.set(tool.name, tool.description);
+}
 //#endregion
 //#region src/modules/mcp-sdk-compat.ts
 /**
@@ -2366,7 +2379,8 @@ function addTracingToToolRegistry(tools, server) {
 }
 function setupListenerToRegisteredTools(server) {
 	try {
-		if (!getServerTrackingData(server.server)) {
+		const data = getServerTrackingData(server.server);
+		if (!data) {
 			writeToLog("Warning: Cannot setup listener - no tracking data found");
 			return;
 		}
@@ -2374,6 +2388,7 @@ function setupListenerToRegisteredTools(server) {
 			set(target, property, value) {
 				try {
 					if (typeof property === "string" && value && typeof value === "object" && hasToolFunction(value)) {
+						if (typeof value.description === "string") data.toolDescriptions.set(property, value.description);
 						if (value[MCP_ANALYTICS_PROCESSED]) {
 							writeToLog(`Tool ${String(property)} already processed, skipping proxy wrapping`);
 							return Reflect.set(target, property, value);
@@ -2496,13 +2511,15 @@ async function initializeToolCallEvent(server, request, downstreamRequest, extra
 				shouldPublishEvent: false
 			};
 		}
+		const toolName = request.params?.name;
 		const event = {
 			sessionId: getServerSessionId(server, extra),
 			conversationId: conversation.conversationId,
-			resourceName: request.params?.name || "Unknown Tool",
+			resourceName: toolName || "Unknown Tool",
 			parameters: buildCapturedMcpParameters(downstreamRequest),
 			eventType: MCPAnalyticsEventType.mcpToolsCall,
 			timestamp: startTime,
+			toolDescription: toolName ? data.toolDescriptions.get(toolName) : void 0,
 			redactionFn: data.options.redactSensitiveInformation
 		};
 		await handleIdentify(server, data, request, extra);
@@ -2592,16 +2609,20 @@ function publishFailedToolEvent(server, tracing, error, startTime) {
 }
 function setupTracking(server) {
 	try {
-		getServerTrackingData(server.server);
+		const mcpAnalyticsData = getServerTrackingData(server.server);
 		setupToolsCallHandlerWrapping(server);
 		setupInitializeTracing(server);
 		server._registeredTools = addTracingToToolRegistry(server._registeredTools, server);
+		if (mcpAnalyticsData) seedToolDescriptionsFromRegistry(mcpAnalyticsData.toolDescriptions, server._registeredTools);
 		setupListToolsTracing(server);
 		setupListenerToRegisteredTools(server);
 	} catch (error) {
 		writeToLog(`Warning: Failed to setup tool call tracing - ${error}`);
 	}
 }
+function seedToolDescriptionsFromRegistry(cache, tools) {
+	for (const [name, tool] of Object.entries(tools)) if (typeof tool?.description === "string") cache.set(name, tool.description);
+}
 //#endregion
 //#region src/index.ts
 /**
@@ -2737,6 +2758,7 @@ function buildTrackingData(lowLevelServer, options) {
 		sessionId: newSessionId(),
 		lastActivity: /* @__PURE__ */ new Date(),
 		identifiedSessions: /* @__PURE__ */ new Map(),
+		toolDescriptions: /* @__PURE__ */ new Map(),
 		sessionInfo: getSessionInfo(lowLevelServer, void 0),
 		options: {
 			reportMissing: options.reportMissing ?? false,