@posthog/mcp 0.10.2 → 0.10.4

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

@@ -0,0 +1,17 @@
+import type { AnalyticsParameterOwnership } from '../types';
+export interface AnalyticsInjectableJsonSchema {
+    $ref?: unknown;
+    additionalProperties?: boolean;
+    allOf?: unknown;
+    anyOf?: unknown;
+    oneOf?: unknown;
+    properties?: Record<string, unknown>;
+    required?: string[];
+    type?: string;
+}
+export declare function hasAnalyticsParameter(schema: AnalyticsInjectableJsonSchema | undefined, parameterName: string): boolean;
+export declare function canInjectAnalyticsParameter(schema: AnalyticsInjectableJsonSchema | undefined, parameterName: string): boolean;
+export declare function analyticsOwnsParameter(inputSchema: unknown, parameterName: string): boolean;
+export declare function getAnalyticsParameterOwnership(inputSchema: unknown): AnalyticsParameterOwnership;
+export declare function stripOwnedAnalyticsArguments(args: unknown, ownership: AnalyticsParameterOwnership): unknown;
+//# sourceMappingURL=analytics-parameters.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"analytics-parameters.d.ts","sourceRoot":"","sources":["../../src/extensions/analytics-parameters.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,UAAU,CAAA;AAgB3D,MAAM,WAAW,6BAA6B;IAC5C,IAAI,CAAC,EAAE,OAAO,CAAA;IACd,oBAAoB,CAAC,EAAE,OAAO,CAAA;IAC9B,KAAK,CAAC,EAAE,OAAO,CAAA;IACf,KAAK,CAAC,EAAE,OAAO,CAAA;IACf,KAAK,CAAC,EAAE,OAAO,CAAA;IACf,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IACpC,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAA;IACnB,IAAI,CAAC,EAAE,MAAM,CAAA;CACd;AAED,wBAAgB,qBAAqB,CACnC,MAAM,EAAE,6BAA6B,GAAG,SAAS,EACjD,aAAa,EAAE,MAAM,GACpB,OAAO,CAET;AAED,wBAAgB,2BAA2B,CACzC,MAAM,EAAE,6BAA6B,GAAG,SAAS,EACjD,aAAa,EAAE,MAAM,GACpB,OAAO,CAIT;AAED,wBAAgB,sBAAsB,CAAC,WAAW,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,GAAG,OAAO,CAyB3F;AAED,wBAAgB,8BAA8B,CAAC,WAAW,EAAE,OAAO,GAAG,2BAA2B,CAKhG;AAED,wBAAgB,4BAA4B,CAAC,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,2BAA2B,GAAG,OAAO,CAW3G"}

package/dist/extensions/analytics-parameters.js

@@ -0,0 +1,98 @@
+"use strict";
+var __webpack_require__ = {};
+(()=>{
+    __webpack_require__.d = (exports1, definition)=>{
+        for(var key in definition)if (__webpack_require__.o(definition, key) && !__webpack_require__.o(exports1, key)) Object.defineProperty(exports1, key, {
+            enumerable: true,
+            get: definition[key]
+        });
+    };
+})();
+(()=>{
+    __webpack_require__.o = (obj, prop)=>Object.prototype.hasOwnProperty.call(obj, prop);
+})();
+(()=>{
+    __webpack_require__.r = (exports1)=>{
+        if ('undefined' != typeof Symbol && Symbol.toStringTag) Object.defineProperty(exports1, Symbol.toStringTag, {
+            value: 'Module'
+        });
+        Object.defineProperty(exports1, '__esModule', {
+            value: true
+        });
+    };
+})();
+var __webpack_exports__ = {};
+__webpack_require__.r(__webpack_exports__);
+__webpack_require__.d(__webpack_exports__, {
+    hasAnalyticsParameter: ()=>hasAnalyticsParameter,
+    analyticsOwnsParameter: ()=>analyticsOwnsParameter,
+    getAnalyticsParameterOwnership: ()=>getAnalyticsParameterOwnership,
+    canInjectAnalyticsParameter: ()=>canInjectAnalyticsParameter,
+    stripOwnedAnalyticsArguments: ()=>stripOwnedAnalyticsArguments
+});
+const external_mcp_sdk_compat_js_namespaceObject = require("./mcp-sdk-compat.js");
+const JSON_SCHEMA_KEYWORDS = [
+    '$defs',
+    '$ref',
+    'additionalProperties',
+    'allOf',
+    'anyOf',
+    'definitions',
+    'oneOf',
+    'properties',
+    'required',
+    'type'
+];
+function hasAnalyticsParameter(schema, parameterName) {
+    return !!schema?.properties && Object.prototype.hasOwnProperty.call(schema.properties, parameterName);
+}
+function canInjectAnalyticsParameter(schema, parameterName) {
+    return !hasAnalyticsParameter(schema, parameterName) && !schema?.$ref && !schema?.oneOf && !schema?.allOf && !schema?.anyOf;
+}
+function analyticsOwnsParameter(inputSchema, parameterName) {
+    if (!inputSchema || 'object' != typeof inputSchema) return canInjectAnalyticsParameter(void 0, parameterName);
+    if ((0, external_mcp_sdk_compat_js_namespaceObject.isZodRawShapeCompat)(inputSchema)) return canInjectAnalyticsParameter({
+        properties: inputSchema
+    }, parameterName);
+    const shape = (0, external_mcp_sdk_compat_js_namespaceObject.getObjectShape)(inputSchema);
+    if (shape) return canInjectAnalyticsParameter({
+        properties: shape
+    }, parameterName);
+    const schema = inputSchema;
+    if (JSON_SCHEMA_KEYWORDS.some((keyword)=>Object.prototype.hasOwnProperty.call(schema, keyword))) return canInjectAnalyticsParameter(schema, parameterName);
+    if ('_def' in schema || '_zod' in schema || '~standard' in schema) return canInjectAnalyticsParameter(void 0, parameterName);
+    return canInjectAnalyticsParameter(schema, parameterName);
+}
+function getAnalyticsParameterOwnership(inputSchema) {
+    return {
+        context: analyticsOwnsParameter(inputSchema, 'context'),
+        conversationId: analyticsOwnsParameter(inputSchema, 'conversation_id')
+    };
+}
+function stripOwnedAnalyticsArguments(args, ownership) {
+    let cleanedArgs = args;
+    if (ownership.context && cleanedArgs && 'object' == typeof cleanedArgs && 'context' in cleanedArgs) {
+        const { context: _context, ...rest } = cleanedArgs;
+        cleanedArgs = rest;
+    }
+    if (ownership.conversationId && cleanedArgs && 'object' == typeof cleanedArgs && 'conversation_id' in cleanedArgs) {
+        const { conversation_id: _conversationId, ...rest } = cleanedArgs;
+        cleanedArgs = rest;
+    }
+    return cleanedArgs;
+}
+exports.analyticsOwnsParameter = __webpack_exports__.analyticsOwnsParameter;
+exports.canInjectAnalyticsParameter = __webpack_exports__.canInjectAnalyticsParameter;
+exports.getAnalyticsParameterOwnership = __webpack_exports__.getAnalyticsParameterOwnership;
+exports.hasAnalyticsParameter = __webpack_exports__.hasAnalyticsParameter;
+exports.stripOwnedAnalyticsArguments = __webpack_exports__.stripOwnedAnalyticsArguments;
+for(var __webpack_i__ in __webpack_exports__)if (-1 === [
+    "analyticsOwnsParameter",
+    "canInjectAnalyticsParameter",
+    "getAnalyticsParameterOwnership",
+    "hasAnalyticsParameter",
+    "stripOwnedAnalyticsArguments"
+].indexOf(__webpack_i__)) exports[__webpack_i__] = __webpack_exports__[__webpack_i__];
+Object.defineProperty(exports, '__esModule', {
+    value: true
+});

package/dist/extensions/analytics-parameters.mjs

@@ -0,0 +1,52 @@
+import { getObjectShape, isZodRawShapeCompat } from "./mcp-sdk-compat.mjs";
+const JSON_SCHEMA_KEYWORDS = [
+    '$defs',
+    '$ref',
+    'additionalProperties',
+    'allOf',
+    'anyOf',
+    'definitions',
+    'oneOf',
+    'properties',
+    'required',
+    'type'
+];
+function hasAnalyticsParameter(schema, parameterName) {
+    return !!schema?.properties && Object.prototype.hasOwnProperty.call(schema.properties, parameterName);
+}
+function canInjectAnalyticsParameter(schema, parameterName) {
+    return !hasAnalyticsParameter(schema, parameterName) && !schema?.$ref && !schema?.oneOf && !schema?.allOf && !schema?.anyOf;
+}
+function analyticsOwnsParameter(inputSchema, parameterName) {
+    if (!inputSchema || 'object' != typeof inputSchema) return canInjectAnalyticsParameter(void 0, parameterName);
+    if (isZodRawShapeCompat(inputSchema)) return canInjectAnalyticsParameter({
+        properties: inputSchema
+    }, parameterName);
+    const shape = getObjectShape(inputSchema);
+    if (shape) return canInjectAnalyticsParameter({
+        properties: shape
+    }, parameterName);
+    const schema = inputSchema;
+    if (JSON_SCHEMA_KEYWORDS.some((keyword)=>Object.prototype.hasOwnProperty.call(schema, keyword))) return canInjectAnalyticsParameter(schema, parameterName);
+    if ('_def' in schema || '_zod' in schema || '~standard' in schema) return canInjectAnalyticsParameter(void 0, parameterName);
+    return canInjectAnalyticsParameter(schema, parameterName);
+}
+function getAnalyticsParameterOwnership(inputSchema) {
+    return {
+        context: analyticsOwnsParameter(inputSchema, 'context'),
+        conversationId: analyticsOwnsParameter(inputSchema, 'conversation_id')
+    };
+}
+function stripOwnedAnalyticsArguments(args, ownership) {
+    let cleanedArgs = args;
+    if (ownership.context && cleanedArgs && 'object' == typeof cleanedArgs && 'context' in cleanedArgs) {
+        const { context: _context, ...rest } = cleanedArgs;
+        cleanedArgs = rest;
+    }
+    if (ownership.conversationId && cleanedArgs && 'object' == typeof cleanedArgs && 'conversation_id' in cleanedArgs) {
+        const { conversation_id: _conversationId, ...rest } = cleanedArgs;
+        cleanedArgs = rest;
+    }
+    return cleanedArgs;
+}
+export { analyticsOwnsParameter, canInjectAnalyticsParameter, getAnalyticsParameterOwnership, hasAnalyticsParameter, stripOwnedAnalyticsArguments };

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

@@ -1,15 +1,7 @@
 import type { MCPAnalyticsOptions } from '../types';
-interface JsonSchema {
-    additionalProperties?: boolean;
-    allOf?: unknown;
-    anyOf?: unknown;
-    oneOf?: unknown;
-    properties?: Record<string, unknown>;
-    required?: string[];
-    type?: string;
-}
+import { type AnalyticsInjectableJsonSchema } from './analytics-parameters';
 export interface ContextInjectableTool {
-    inputSchema?: JsonSchema;
+    inputSchema?: AnalyticsInjectableJsonSchema;
     name?: string;
     [key: string]: unknown;
 }
@@ -22,10 +14,9 @@ export declare function getContextDescription(context: MCPAnalyticsOptions['cont
  *
  * Skips injection (with warning) for:
  * - Tools that already have a 'context' parameter
- * - Complex schemas (oneOf/allOf/anyOf) that can't safely have properties added
+ * - 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 {};
 //# 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;AAInD,UAAU,UAAU;IAClB,oBAAoB,CAAC,EAAE,OAAO,CAAA;IAC9B,KAAK,CAAC,EAAE,OAAO,CAAA;IACf,KAAK,CAAC,EAAE,OAAO,CAAA;IACf,KAAK,CAAC,EAAE,OAAO,CAAA;IACf,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IACpC,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAA;IACnB,IAAI,CAAC,EAAE,MAAM,CAAA;CACd;AAED,MAAM,WAAW,qBAAqB;IACpC,WAAW,CAAC,EAAE,UAAU,CAAA;IACxB,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,CAgEP;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;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"}

package/dist/extensions/context-parameters.js

@@ -29,6 +29,7 @@ __webpack_require__.d(__webpack_exports__, {
     getContextDescription: ()=>getContextDescription,
     isContextEnabled: ()=>isContextEnabled
 });
+const external_analytics_parameters_js_namespaceObject = require("./analytics-parameters.js");
 const external_constants_js_namespaceObject = require("./constants.js");
 const external_logger_js_namespaceObject = require("./logger.js");
 function isContextEnabled(context) {
@@ -43,12 +44,9 @@ function addContextParameterToTool(tool, contextDescriptionOverride) {
     };
     const toolName = tool.name || 'unknown';
     const schema = modifiedTool.inputSchema;
-    if (schema?.properties?.context) {
-        (0, external_logger_js_namespaceObject.log)(`WARN: Tool "${toolName}" already has 'context' parameter. Skipping context injection.`);
-        return modifiedTool;
-    }
-    if (schema?.oneOf || schema?.allOf || schema?.anyOf) {
-        (0, external_logger_js_namespaceObject.log)(`WARN: Tool "${toolName}" has complex schema (oneOf/allOf/anyOf). Skipping context injection.`);
+    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.`);
         return modifiedTool;
     }
     if (!modifiedTool.inputSchema) modifiedTool.inputSchema = {

package/dist/extensions/context-parameters.mjs

@@ -1,3 +1,4 @@
+import { canInjectAnalyticsParameter, hasAnalyticsParameter } from "./analytics-parameters.mjs";
 import { DEFAULT_CONTEXT_PARAMETER_DESCRIPTION } from "./constants.mjs";
 import { log } from "./logger.mjs";
 function isContextEnabled(context) {
@@ -12,12 +13,8 @@ function addContextParameterToTool(tool, contextDescriptionOverride) {
     };
     const toolName = tool.name || 'unknown';
     const schema = modifiedTool.inputSchema;
-    if (schema?.properties?.context) {
-        log(`WARN: Tool "${toolName}" already has 'context' parameter. Skipping context injection.`);
-        return modifiedTool;
-    }
-    if (schema?.oneOf || schema?.allOf || schema?.anyOf) {
-        log(`WARN: Tool "${toolName}" has complex schema (oneOf/allOf/anyOf). Skipping context injection.`);
+    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.`);
         return modifiedTool;
     }
     if (!modifiedTool.inputSchema) modifiedTool.inputSchema = {

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

@@ -1,15 +1,7 @@
+import { type AnalyticsInjectableJsonSchema } from './analytics-parameters';
 export declare const CONVERSATION_ID_PARAM_NAME = "conversation_id";
-interface JsonSchema {
-    additionalProperties?: boolean;
-    allOf?: unknown;
-    anyOf?: unknown;
-    oneOf?: unknown;
-    properties?: Record<string, unknown>;
-    required?: string[];
-    type?: string;
-}
 export interface ConversationIdInjectableTool {
-    inputSchema?: JsonSchema;
+    inputSchema?: AnalyticsInjectableJsonSchema;
     name?: string;
     [key: string]: unknown;
 }
@@ -36,17 +28,10 @@ export declare function resolveConversationId(enabled: boolean, args: unknown, t
  */
 export declare function canInjectConversationIdPromptBack(result: unknown): boolean;
 export declare function extractConversationId(args: unknown): string | undefined;
-export declare function cloneRequestWithoutConversationId<TRequest extends {
-    params?: {
-        arguments?: unknown;
-        [k: string]: unknown;
-    };
-}>(request: TRequest): TRequest;
 export declare function stripConversationId(args: unknown): unknown;
 export declare function buildConversationIdPromptBack(conversationId: string): {
     type: 'text';
     text: string;
 };
 export declare function injectConversationIdPromptBack(result: unknown, conversationId: string): unknown;
-export {};
 //# sourceMappingURL=conversation-id.d.ts.map

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":"AAUA,eAAO,MAAM,0BAA0B,oBAAoB,CAAA;AAE3D,UAAU,UAAU;IAClB,oBAAoB,CAAC,EAAE,OAAO,CAAA;IAC9B,KAAK,CAAC,EAAE,OAAO,CAAA;IACf,KAAK,CAAC,EAAE,OAAO,CAAA;IACf,KAAK,CAAC,EAAE,OAAO,CAAA;IACf,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IACpC,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAA;IACnB,IAAI,CAAC,EAAE,MAAM,CAAA;CACd;AAED,MAAM,WAAW,4BAA4B;IAC3C,WAAW,CAAC,EAAE,UAAU,CAAA;IACxB,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,CA2CtG;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,iCAAiC,CAC/C,QAAQ,SAAS;IAAE,MAAM,CAAC,EAAE;QAAE,SAAS,CAAC,EAAE,OAAO,CAAC;QAAC,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAA;KAAE,CAAA;CAAE,EAC3E,OAAO,EAAE,QAAQ,GAAG,QAAQ,CAe7B;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;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"}

package/dist/extensions/conversation-id.js

@@ -27,15 +27,15 @@ __webpack_require__.d(__webpack_exports__, {
     CONVERSATION_ID_PARAM_NAME: ()=>CONVERSATION_ID_PARAM_NAME,
     resolveConversationId: ()=>resolveConversationId,
     buildConversationIdPromptBack: ()=>buildConversationIdPromptBack,
-    addConversationIdToTool: ()=>addConversationIdToTool,
+    injectConversationIdPromptBack: ()=>injectConversationIdPromptBack,
     addConversationIdToTools: ()=>addConversationIdToTools,
     extractConversationId: ()=>extractConversationId,
-    injectConversationIdPromptBack: ()=>injectConversationIdPromptBack,
-    cloneRequestWithoutConversationId: ()=>cloneRequestWithoutConversationId,
+    addConversationIdToTool: ()=>addConversationIdToTool,
     stripConversationId: ()=>stripConversationId,
     canInjectConversationIdPromptBack: ()=>canInjectConversationIdPromptBack
 });
 const core_namespaceObject = require("@posthog/core");
+const external_analytics_parameters_js_namespaceObject = require("./analytics-parameters.js");
 const external_constants_js_namespaceObject = require("./constants.js");
 const external_logger_js_namespaceObject = require("./logger.js");
 const external_tools_js_namespaceObject = require("./tools.js");
@@ -46,12 +46,9 @@ function addConversationIdToTool(tool) {
     };
     const toolName = tool.name || 'unknown';
     const schema = modifiedTool.inputSchema;
-    if (schema?.properties?.[CONVERSATION_ID_PARAM_NAME]) {
-        (0, external_logger_js_namespaceObject.log)(`WARN: Tool "${toolName}" already has '${CONVERSATION_ID_PARAM_NAME}' parameter. Skipping conversation_id injection.`);
-        return modifiedTool;
-    }
-    if (schema?.oneOf || schema?.allOf || schema?.anyOf) {
-        (0, external_logger_js_namespaceObject.log)(`WARN: Tool "${toolName}" has complex schema (oneOf/allOf/anyOf). Skipping conversation_id injection.`);
+    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.`);
         return modifiedTool;
     }
     if (!modifiedTool.inputSchema) modifiedTool.inputSchema = {
@@ -103,18 +100,6 @@ function extractConversationId(args) {
     const trimmed = value.trim();
     return trimmed.length > 0 ? trimmed : void 0;
 }
-function cloneRequestWithoutConversationId(request) {
-    if (!request.params || 'object' != typeof request.params) return request;
-    const args = request.params.arguments;
-    if (!(args && 'object' == typeof args)) return request;
-    return {
-        ...request,
-        params: {
-            ...request.params,
-            arguments: stripConversationId(args)
-        }
-    };
-}
 function stripConversationId(args) {
     if (!args || 'object' != typeof args || !(CONVERSATION_ID_PARAM_NAME in args)) return args;
     const { [CONVERSATION_ID_PARAM_NAME]: _omit, ...rest } = args;
@@ -142,7 +127,6 @@ exports.addConversationIdToTool = __webpack_exports__.addConversationIdToTool;
 exports.addConversationIdToTools = __webpack_exports__.addConversationIdToTools;
 exports.buildConversationIdPromptBack = __webpack_exports__.buildConversationIdPromptBack;
 exports.canInjectConversationIdPromptBack = __webpack_exports__.canInjectConversationIdPromptBack;
-exports.cloneRequestWithoutConversationId = __webpack_exports__.cloneRequestWithoutConversationId;
 exports.extractConversationId = __webpack_exports__.extractConversationId;
 exports.injectConversationIdPromptBack = __webpack_exports__.injectConversationIdPromptBack;
 exports.resolveConversationId = __webpack_exports__.resolveConversationId;
@@ -153,7 +137,6 @@ for(var __webpack_i__ in __webpack_exports__)if (-1 === [
     "addConversationIdToTools",
     "buildConversationIdPromptBack",
     "canInjectConversationIdPromptBack",
-    "cloneRequestWithoutConversationId",
     "extractConversationId",
     "injectConversationIdPromptBack",
     "resolveConversationId",

package/dist/extensions/conversation-id.mjs

@@ -1,4 +1,5 @@
 import { uuidv7 } from "@posthog/core";
+import { canInjectAnalyticsParameter, hasAnalyticsParameter } from "./analytics-parameters.mjs";
 import { DEFAULT_CONVERSATION_ID_DESCRIPTION } from "./constants.mjs";
 import { log } from "./logger.mjs";
 import { GET_MORE_TOOLS_NAME } from "./tools.mjs";
@@ -9,12 +10,8 @@ function addConversationIdToTool(tool) {
     };
     const toolName = tool.name || 'unknown';
     const schema = modifiedTool.inputSchema;
-    if (schema?.properties?.[CONVERSATION_ID_PARAM_NAME]) {
-        log(`WARN: Tool "${toolName}" already has '${CONVERSATION_ID_PARAM_NAME}' parameter. Skipping conversation_id injection.`);
-        return modifiedTool;
-    }
-    if (schema?.oneOf || schema?.allOf || schema?.anyOf) {
-        log(`WARN: Tool "${toolName}" has complex schema (oneOf/allOf/anyOf). Skipping conversation_id injection.`);
+    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.`);
         return modifiedTool;
     }
     if (!modifiedTool.inputSchema) modifiedTool.inputSchema = {
@@ -66,18 +63,6 @@ function extractConversationId(args) {
     const trimmed = value.trim();
     return trimmed.length > 0 ? trimmed : void 0;
 }
-function cloneRequestWithoutConversationId(request) {
-    if (!request.params || 'object' != typeof request.params) return request;
-    const args = request.params.arguments;
-    if (!(args && 'object' == typeof args)) return request;
-    return {
-        ...request,
-        params: {
-            ...request.params,
-            arguments: stripConversationId(args)
-        }
-    };
-}
 function stripConversationId(args) {
     if (!args || 'object' != typeof args || !(CONVERSATION_ID_PARAM_NAME in args)) return args;
     const { [CONVERSATION_ID_PARAM_NAME]: _omit, ...rest } = args;
@@ -100,4 +85,4 @@ function injectConversationIdPromptBack(result, conversationId) {
         ]
     };
 }
-export { CONVERSATION_ID_PARAM_NAME, addConversationIdToTool, addConversationIdToTools, buildConversationIdPromptBack, canInjectConversationIdPromptBack, cloneRequestWithoutConversationId, extractConversationId, injectConversationIdPromptBack, resolveConversationId, stripConversationId };
+export { CONVERSATION_ID_PARAM_NAME, addConversationIdToTool, addConversationIdToTools, buildConversationIdPromptBack, canInjectConversationIdPromptBack, extractConversationId, injectConversationIdPromptBack, resolveConversationId, stripConversationId };

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;AAqPjB,wBAAgB,yBAAyB,CAAC,MAAM,EAAE,sBAAsB,GAAG,IAAI,CAwB9E"}
+{"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"}

package/dist/extensions/instrument-highlevel.js

@@ -26,7 +26,8 @@ __webpack_require__.r(__webpack_exports__);
 __webpack_require__.d(__webpack_exports__, {
     instrumentHighLevelServer: ()=>instrumentHighLevelServer
 });
-const external_conversation_id_js_namespaceObject = require("./conversation-id.js");
+const external_analytics_parameters_js_namespaceObject = require("./analytics-parameters.js");
+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");
@@ -62,7 +63,8 @@ function setupListenerToRegisteredTools(server) {
                             (0, external_logger_js_namespaceObject.log)(`Tool ${String(property)} callback already wrapped, skipping proxy wrapping`);
                             return Reflect.set(target, property, value);
                         }
-                        const nextValue = addTracingToToolCallbackInternal(value, property, server);
+                        const getCurrentInputSchema = ()=>value.inputSchema;
+                        const nextValue = addTracingToToolCallbackInternal(value, property, server, getCurrentInputSchema);
                         if ('function' == typeof nextValue.update) {
                             const originalUpdate = nextValue.update;
                             nextValue.update = function(...updateArgs) {
@@ -71,7 +73,7 @@ function setupListenerToRegisteredTools(server) {
                                     if (isCallbackUpdate(updateObj)) {
                                         const wrappedTool = addTracingToToolCallbackInternal({
                                             callback: updateObj.callback
-                                        }, property, server);
+                                        }, property, server, getCurrentInputSchema);
                                         updateObj.callback = (0, external_mcp_sdk_compat_js_namespaceObject.getToolFunction)(wrappedTool);
                                     }
                                 }
@@ -103,9 +105,10 @@ function setupListenerToRegisteredTools(server) {
         (0, external_logger_js_namespaceObject.log)(`Warning: Failed to setup listener for registered tools - ${error}`);
     }
 }
-function addTracingToToolCallbackInternal(tool, toolName, server) {
+function addTracingToToolCallbackInternal(tool, toolName, server, getCurrentInputSchema = ()=>tool.inputSchema) {
     const originalCallback = (0, external_mcp_sdk_compat_js_namespaceObject.getToolFunction)(tool);
-    const missingToolName = (0, external_tools_js_namespaceObject.resolveMissingCapabilityToolName)((0, external_internal_js_namespaceObject.getServerTrackingData)(server.server)?.options);
+    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`);
         return tool;
@@ -124,14 +127,11 @@ function addTracingToToolCallbackInternal(tool, toolName, server) {
             args = void 0;
             extra = params[0];
         }
-        const removeContextFromArgs = (input)=>{
-            if (input && 'object' == typeof input && 'context' in input) {
-                const { context: _context, ...rest } = input;
-                return rest;
-            }
-            return input;
-        };
-        const cleanedArgs = toolName === missingToolName ? args : (0, external_conversation_id_js_namespaceObject.stripConversationId)(removeContextFromArgs(args));
+        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')
+        });
         try {
             if (void 0 === cleanedArgs) {
                 const handler = originalCallback;
@@ -150,7 +150,7 @@ function addTracingToToolCallbackInternal(tool, toolName, server) {
     wrappedTool[MCP_ANALYTICS_PROCESSED] = true;
     return wrappedTool;
 }
-async function handleToolCallRequest(server, originalCallToolHandler, request, extra) {
+async function handleToolCallRequest(highLevelServer, server, originalCallToolHandler, request, extra) {
     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.');
@@ -170,12 +170,15 @@ async function handleToolCallRequest(server, originalCallToolHandler, request, e
                 })
         });
     }
+    const toolName = request.params?.name;
+    const registeredTool = toolName ? highLevelServer._registeredTools[toolName] : void 0;
     return await (0, external_instrumentation_js_namespaceObject.captureToolCall)({
         server,
         data,
         request,
         extra,
-        execute: ()=>originalCallToolHandler(request, extra),
+        execute: (downstreamRequest)=>originalCallToolHandler(downstreamRequest, extra),
+        parameterOwnership: registeredTool ? (0, external_analytics_parameters_js_namespaceObject.getAnalyticsParameterOwnership)(registeredTool.inputSchema) : void 0,
         takeCapturedError: ()=>{
             const captured = extra?.__mcp_analytics_error;
             if (extra) extra.__mcp_analytics_error = void 0;
@@ -190,7 +193,7 @@ function instrumentHighLevelServer(server) {
         const handlers = {
             initialize: external_instrumentation_js_namespaceObject.handleInitializeRequest,
             'tools/list': external_instrumentation_js_namespaceObject.handleListToolsRequest,
-            'tools/call': handleToolCallRequest
+            'tools/call': (trackedServer, originalHandler, request, extra)=>handleToolCallRequest(server, trackedServer, originalHandler, request, extra)
         };
         (0, external_instrumentation_js_namespaceObject.patchRequestHandlers)(lowLevelServer, handlers);
         server._registeredTools = addTracingToToolRegistry(server._registeredTools, server);

package/dist/extensions/instrument-highlevel.mjs

@@ -1,4 +1,5 @@
-import { stripConversationId } from "./conversation-id.mjs";
+import { analyticsOwnsParameter, getAnalyticsParameterOwnership, stripOwnedAnalyticsArguments } from "./analytics-parameters.mjs";
+import { isContextEnabled } from "./context-parameters.mjs";
 import { MCPAnalyticsEventType } from "./event-types.mjs";
 import { getServerTrackingData } from "./internal.mjs";
 import { log } from "./logger.mjs";
@@ -34,7 +35,8 @@ function setupListenerToRegisteredTools(server) {
                             log(`Tool ${String(property)} callback already wrapped, skipping proxy wrapping`);
                             return Reflect.set(target, property, value);
                         }
-                        const nextValue = addTracingToToolCallbackInternal(value, property, server);
+                        const getCurrentInputSchema = ()=>value.inputSchema;
+                        const nextValue = addTracingToToolCallbackInternal(value, property, server, getCurrentInputSchema);
                         if ('function' == typeof nextValue.update) {
                             const originalUpdate = nextValue.update;
                             nextValue.update = function(...updateArgs) {
@@ -43,7 +45,7 @@ function setupListenerToRegisteredTools(server) {
                                     if (isCallbackUpdate(updateObj)) {
                                         const wrappedTool = addTracingToToolCallbackInternal({
                                             callback: updateObj.callback
-                                        }, property, server);
+                                        }, property, server, getCurrentInputSchema);
                                         updateObj.callback = getToolFunction(wrappedTool);
                                     }
                                 }
@@ -75,9 +77,10 @@ function setupListenerToRegisteredTools(server) {
         log(`Warning: Failed to setup listener for registered tools - ${error}`);
     }
 }
-function addTracingToToolCallbackInternal(tool, toolName, server) {
+function addTracingToToolCallbackInternal(tool, toolName, server, getCurrentInputSchema = ()=>tool.inputSchema) {
     const originalCallback = getToolFunction(tool);
-    const missingToolName = resolveMissingCapabilityToolName(getServerTrackingData(server.server)?.options);
+    const options = getServerTrackingData(server.server)?.options;
+    const missingToolName = resolveMissingCapabilityToolName(options);
     if (wrappedCallbacks.has(originalCallback)) {
         log(`Tool ${toolName} callback already wrapped, skipping re-wrap`);
         return tool;
@@ -96,14 +99,11 @@ function addTracingToToolCallbackInternal(tool, toolName, server) {
             args = void 0;
             extra = params[0];
         }
-        const removeContextFromArgs = (input)=>{
-            if (input && 'object' == typeof input && 'context' in input) {
-                const { context: _context, ...rest } = input;
-                return rest;
-            }
-            return input;
-        };
-        const cleanedArgs = toolName === missingToolName ? args : stripConversationId(removeContextFromArgs(args));
+        const inputSchema = getCurrentInputSchema();
+        const cleanedArgs = stripOwnedAnalyticsArguments(args, {
+            context: toolName !== missingToolName && isContextEnabled(options?.context) && analyticsOwnsParameter(inputSchema, 'context'),
+            conversationId: toolName !== missingToolName && options?.enableConversationId === true && analyticsOwnsParameter(inputSchema, 'conversation_id')
+        });
         try {
             if (void 0 === cleanedArgs) {
                 const handler = originalCallback;
@@ -122,7 +122,7 @@ function addTracingToToolCallbackInternal(tool, toolName, server) {
     wrappedTool[MCP_ANALYTICS_PROCESSED] = true;
     return wrappedTool;
 }
-async function handleToolCallRequest(server, originalCallToolHandler, request, extra) {
+async function handleToolCallRequest(highLevelServer, server, originalCallToolHandler, request, extra) {
     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.');
@@ -142,12 +142,15 @@ async function handleToolCallRequest(server, originalCallToolHandler, request, e
                 })
         });
     }
+    const toolName = request.params?.name;
+    const registeredTool = toolName ? highLevelServer._registeredTools[toolName] : void 0;
     return await captureToolCall({
         server,
         data,
         request,
         extra,
-        execute: ()=>originalCallToolHandler(request, extra),
+        execute: (downstreamRequest)=>originalCallToolHandler(downstreamRequest, extra),
+        parameterOwnership: registeredTool ? getAnalyticsParameterOwnership(registeredTool.inputSchema) : void 0,
         takeCapturedError: ()=>{
             const captured = extra?.__mcp_analytics_error;
             if (extra) extra.__mcp_analytics_error = void 0;
@@ -162,7 +165,7 @@ function instrumentHighLevelServer(server) {
         const handlers = {
             initialize: handleInitializeRequest,
             'tools/list': handleListToolsRequest,
-            'tools/call': handleToolCallRequest
+            'tools/call': (trackedServer, originalHandler, request, extra)=>handleToolCallRequest(server, trackedServer, originalHandler, request, extra)
         };
         patchRequestHandlers(lowLevelServer, handlers);
         server._registeredTools = addTracingToToolRegistry(server._registeredTools, server);