@agent-inspect/mcp 6.31.18 → 6.31.20

package/dist/index.cjs

@@ -49,15 +49,94 @@ function baseMetadata(options, overrides = {}) {
   };
   return { ...options.metadata ?? {}, ...metadata };
 }
+var SDK_REQUEST_OPTION_KEYS = [
+  "signal",
+  "timeout",
+  "onprogress",
+  "resetTimeoutOnProgress",
+  "maxTotalTimeout",
+  "relatedRequestId",
+  "resumptionToken",
+  "onresumptiontoken"
+];
+function isSdkRequestOptions(value) {
+  if (!value || typeof value !== "object") return false;
+  const obj = value;
+  return SDK_REQUEST_OPTION_KEYS.some((key) => key in obj);
+}
+function isAgentInspectMetadataBag(value) {
+  if (!value || typeof value !== "object") return false;
+  if (isSdkRequestOptions(value)) return false;
+  const obj = value;
+  if (!("metadata" in obj)) return false;
+  if ("parse" in obj) return false;
+  return true;
+}
+function extractCallMetadata(bag) {
+  if (bag.metadata && typeof bag.metadata === "object") {
+    return bag.metadata;
+  }
+  return {};
+}
+function resolveCallToolArgs(resultSchemaOrOptions, maybeOptions) {
+  if (maybeOptions !== void 0) {
+    let callMetadata = {};
+    let requestOptions;
+    if (isAgentInspectMetadataBag(maybeOptions)) {
+      callMetadata = extractCallMetadata(maybeOptions);
+      requestOptions = void 0;
+    } else if (isSdkRequestOptions(maybeOptions) || maybeOptions === null) {
+      requestOptions = maybeOptions && typeof maybeOptions === "object" ? maybeOptions : void 0;
+    } else if (maybeOptions && typeof maybeOptions === "object") {
+      requestOptions = maybeOptions;
+    }
+    return {
+      resultSchema: resultSchemaOrOptions,
+      requestOptions,
+      callMetadata
+    };
+  }
+  if (isAgentInspectMetadataBag(resultSchemaOrOptions)) {
+    return {
+      resultSchema: void 0,
+      requestOptions: void 0,
+      callMetadata: extractCallMetadata(resultSchemaOrOptions)
+    };
+  }
+  if (isSdkRequestOptions(resultSchemaOrOptions)) {
+    return {
+      resultSchema: void 0,
+      requestOptions: resultSchemaOrOptions,
+      callMetadata: {}
+    };
+  }
+  return {
+    resultSchema: resultSchemaOrOptions,
+    requestOptions: void 0,
+    callMetadata: {}
+  };
+}
+function invokeCallTool(callToolImpl, params, resultSchema, requestOptions) {
+  const fn = callToolImpl;
+  if (requestOptions !== void 0) {
+    return fn(params, resultSchema, requestOptions);
+  }
+  if (resultSchema !== void 0) {
+    return fn(params, resultSchema);
+  }
+  return fn(params);
+}
 function wrapMcpClient(client, options = {}) {
   const maxSummaryLength = options.maxSummaryLength ?? 240;
   const listToolsImpl = typeof client.listTools === "function" ? client.listTools.bind(client) : void 0;
   const callToolImpl = client.callTool.bind(client);
-  const wrappedListTools = listToolsImpl ? async (params) => agentInspect.step(
+  const wrappedListTools = listToolsImpl ? async (params, requestOptions) => agentInspect.step(
     "mcp:tools/list",
     async () => {
-      const result = await listToolsImpl(params);
-      return result;
+      if (requestOptions !== void 0) {
+        return listToolsImpl(params, requestOptions);
+      }
+      return listToolsImpl(params);
     },
     {
       type: "tool",
@@ -67,21 +146,28 @@ function wrapMcpClient(client, options = {}) {
       })
     }
   ) : void 0;
-  const wrappedCallTool = async (params) => agentInspect.step(
-    `mcp:${params.name}`,
-    async () => callToolImpl(params),
-    {
-      type: "tool",
-      metadata: baseMetadata(options, {
-        toolName: params.name,
-        toolCallId: nextToolCallId(options.toolCallIdPrefix),
-        argumentSummary: summarizeMcpValue(
-          params.arguments ?? {},
-          maxSummaryLength
-        )
-      })
-    }
-  );
+  const wrappedCallTool = async (params, resultSchemaOrOptions, maybeOptions) => {
+    const { resultSchema, requestOptions, callMetadata } = resolveCallToolArgs(
+      resultSchemaOrOptions,
+      maybeOptions
+    );
+    return agentInspect.step(
+      `mcp:${params.name}`,
+      async () => invokeCallTool(callToolImpl, params, resultSchema, requestOptions),
+      {
+        type: "tool",
+        metadata: baseMetadata(options, {
+          toolName: params.name,
+          toolCallId: nextToolCallId(options.toolCallIdPrefix),
+          argumentSummary: summarizeMcpValue(
+            params.arguments ?? {},
+            maxSummaryLength
+          ),
+          ...callMetadata
+        })
+      }
+    );
+  };
   return new Proxy(client, {
     get(target, property, receiver) {
       if (property === "listTools" && wrappedListTools !== void 0) {
@@ -104,6 +190,7 @@ function resetMcpToolCallIdsForTests() {
 
 exports.hashServerUrl = hashServerUrl;
 exports.resetMcpToolCallIdsForTests = resetMcpToolCallIdsForTests;
+exports.resolveCallToolArgs = resolveCallToolArgs;
 exports.summarizeMcpValue = summarizeMcpValue;
 exports.wrapMcpClient = wrapMcpClient;
 //# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/summarize.ts","../src/wrap.ts"],"names":["step"],"mappings":";;;;;AAAA,IAAM,0BAAA,GAA6B,GAAA;AAEnC,SAAS,eAAe,KAAA,EAAwB;AAC9C,EAAA,IAAI;AACF,IAAA,OAAO,OAAO,KAAK,CAAA;AAAA,EACrB,CAAA,CAAA,MAAQ;AACN,IAAA,OAAO,kBAAA;AAAA,EACT;AACF;AAEO,SAAS,iBAAA,CACd,KAAA,EACA,SAAA,GAAY,0BAAA,EACJ;AACR,EAAA,IAAI,IAAA;AACJ,EAAA,IAAI;AACF,IAAA,IAAI,OAAO,UAAU,QAAA,EAAU;AAC7B,MAAA,IAAA,GAAO,KAAA;AAAA,IACT,CAAA,MAAO;AAIL,MAAA,MAAM,UAAA,GAAa,IAAA,CAAK,SAAA,CAAU,KAAK,CAAA;AACvC,MAAA,IAAA,GAAO,UAAA,KAAe,KAAA,CAAA,GAAY,cAAA,CAAe,KAAK,CAAA,GAAI,UAAA;AAAA,IAC5D;AAAA,EACF,CAAA,CAAA,MAAQ;AACN,IAAA,IAAA,GAAO,eAAe,KAAK,CAAA;AAAA,EAC7B;AACA,EAAA,IAAI,IAAA,CAAK,MAAA,IAAU,SAAA,EAAW,OAAO,IAAA;AACrC,EAAA,OAAO,CAAA,EAAG,IAAA,CAAK,KAAA,CAAM,CAAA,EAAG,IAAA,CAAK,IAAI,CAAA,EAAG,SAAA,GAAY,CAAC,CAAC,CAAC,CAAA,GAAA,CAAA;AACrD;AAEO,SAAS,cAAc,GAAA,EAAqB;AACjD,EAAA,IAAI,IAAA,GAAO,CAAA;AACX,EAAA,KAAA,IAAS,IAAI,CAAA,EAAG,CAAA,GAAI,GAAA,CAAI,MAAA,EAAQ,KAAK,CAAA,EAAG;AACtC,IAAA,IAAA,GAAQ,IAAA,GAAO,EAAA,GAAK,GAAA,CAAI,UAAA,CAAW,CAAC,CAAA,KAAO,CAAA;AAAA,EAC7C;AACA,EAAA,OAAO,KAAK,QAAA,CAAS,EAAE,CAAA,CAAE,QAAA,CAAS,GAAG,GAAG,CAAA;AAC1C;AC7BA,IAAI,eAAA,GAAkB,CAAA;AAEtB,SAAS,eAAe,MAAA,EAAyB;AAC/C,EAAA,eAAA,IAAmB,CAAA;AACnB,EAAA,MAAM,IAAA,GAAO,MAAA,EAAQ,IAAA,EAAK,IAAK,KAAA;AAC/B,EAAA,OAAO,CAAA,EAAG,IAAI,CAAA,CAAA,EAAI,eAAe,CAAA,CAAA;AACnC;AAEA,SAAS,YAAA,CACP,OAAA,EACA,SAAA,GAAuC,EAAC,EACf;AACzB,EAAA,MAAM,QAAA,GAA6B;AAAA,IACjC,MAAA,EAAQ,EAAE,IAAA,EAAM,YAAA,EAAa;AAAA,IAC7B,GAAI,QAAQ,UAAA,GAAa,EAAE,eAAe,OAAA,CAAQ,UAAA,KAAe,EAAC;AAAA,IAClE,GAAI,OAAA,CAAQ,SAAA,GACR,EAAE,gBAAA,EAAkB,cAAc,OAAA,CAAQ,SAAS,CAAA,EAAE,GACrD,EAAC;AAAA,IACL,GAAI,QAAQ,SAAA,GAAY,EAAE,WAAW,OAAA,CAAQ,SAAA,KAAc,EAAC;AAAA,IAC5D,GAAG;AAAA,GACL;AACA,EAAA,OAAO,EAAE,GAAI,OAAA,CAAQ,YAAY,EAAC,EAAI,GAAG,QAAA,EAAS;AACpD;AAUO,SAAS,aAAA,CACd,MAAA,EACA,OAAA,GAAkC,EAAC,EAChC;AACH,EAAA,MAAM,gBAAA,GAAmB,QAAQ,gBAAA,IAAoB,GAAA;AAErD,EAAA,MAAM,aAAA,GACJ,OAAO,MAAA,CAAO,SAAA,KAAc,aACxB,MAAA,CAAO,SAAA,CAAU,IAAA,CAAK,MAAM,CAAA,GAC5B,MAAA;AACN,EAAA,MAAM,YAAA,GAAe,MAAA,CAAO,QAAA,CAAS,IAAA,CAAK,MAAM,CAAA;AAEhD,EAAA,MAAM,gBAAA,GAAmB,aAAA,GACrB,OAAO,MAAA,KACLA,iBAAA;AAAA,IACE,gBAAA;AAAA,IACA,YAAY;AACV,MAAA,MAAM,MAAA,GAAS,MAAM,aAAA,CAAc,MAAM,CAAA;AACzC,MAAA,OAAO,MAAA;AAAA,IACT,CAAA;AAAA,IACA;AAAA,MACE,IAAA,EAAM,MAAA;AAAA,MACN,QAAA,EAAU,aAAa,OAAA,EAAS;AAAA,QAC9B,QAAA,EAAU,YAAA;AAAA,QACV,UAAA,EAAY,cAAA,CAAe,OAAA,CAAQ,gBAAgB;AAAA,OACpD;AAAA;AACH,GACF,GACF,MAAA;AAEJ,EAAA,MAAM,eAAA,GAAkB,OACtB,MAAA,KAEAA,iBAAA;AAAA,IACE,CAAA,IAAA,EAAO,OAAO,IAAI,CAAA,CAAA;AAAA,IAClB,YAAY,aAAa,MAAM,CAAA;AAAA,IAC/B;AAAA,MACE,IAAA,EAAM,MAAA;AAAA,MACN,QAAA,EAAU,aAAa,OAAA,EAAS;AAAA,QAC9B,UAAU,MAAA,CAAO,IAAA;AAAA,QACjB,UAAA,EAAY,cAAA,CAAe,OAAA,CAAQ,gBAAgB,CAAA;AAAA,QACnD,eAAA,EAAiB,iBAAA;AAAA,UACf,MAAA,CAAO,aAAa,EAAC;AAAA,UACrB;AAAA;AACF,OACD;AAAA;AACH,GACF;AAEF,EAAA,OAAO,IAAI,MAAM,MAAA,EAAQ;AAAA,IACvB,GAAA,CAAI,MAAA,EAAQ,QAAA,EAAU,QAAA,EAAU;AAC9B,MAAA,IAAI,QAAA,KAAa,WAAA,IAAe,gBAAA,KAAqB,MAAA,EAAW;AAC9D,QAAA,OAAO,gBAAA;AAAA,MACT;AACA,MAAA,IAAI,aAAa,UAAA,EAAY;AAC3B,QAAA,OAAO,eAAA;AAAA,MACT;AACA,MAAA,MAAM,KAAA,GAAQ,OAAA,CAAQ,GAAA,CAAI,MAAA,EAAQ,UAAU,QAAQ,CAAA;AACpD,MAAA,IAAI,OAAO,UAAU,UAAA,EAAY;AAC/B,QAAA,OAAO,KAAA,CAAM,KAAK,MAAM,CAAA;AAAA,MAC1B;AACA,MAAA,OAAO,KAAA;AAAA,IACT;AAAA,GACD,CAAA;AACH;AAGO,SAAS,2BAAA,GAAoC;AAClD,EAAA,eAAA,GAAkB,CAAA;AACpB","file":"index.cjs","sourcesContent":["const DEFAULT_MAX_SUMMARY_LENGTH = 240;\n\nfunction fallbackString(value: unknown): string {\n  try {\n    return String(value);\n  } catch {\n    return \"[unserializable]\";\n  }\n}\n\nexport function summarizeMcpValue(\n  value: unknown,\n  maxLength = DEFAULT_MAX_SUMMARY_LENGTH,\n): string {\n  let text: string;\n  try {\n    if (typeof value === \"string\") {\n      text = value;\n    } else {\n      // JSON.stringify returns undefined (not a string) for undefined,\n      // functions, and symbols; falling through with undefined would crash\n      // instrumentation at text.length below.\n      const serialized = JSON.stringify(value);\n      text = serialized === undefined ? fallbackString(value) : serialized;\n    }\n  } catch {\n    text = fallbackString(value);\n  }\n  if (text.length <= maxLength) return text;\n  return `${text.slice(0, Math.max(0, maxLength - 3))}...`;\n}\n\nexport function hashServerUrl(url: string): string {\n  let hash = 0;\n  for (let i = 0; i < url.length; i += 1) {\n    hash = (hash * 31 + url.charCodeAt(i)) >>> 0;\n  }\n  return hash.toString(16).padStart(8, \"0\");\n}\n","import { step } from \"agent-inspect\";\n\nimport { hashServerUrl, summarizeMcpValue } from \"./summarize.js\";\nimport type {\n  McpClientLike,\n  McpClientTracerOptions,\n  McpTraceMetadata,\n} from \"./types.js\";\n\nlet toolCallCounter = 0;\n\nfunction nextToolCallId(prefix?: string): string {\n  toolCallCounter += 1;\n  const base = prefix?.trim() || \"mcp\";\n  return `${base}-${toolCallCounter}`;\n}\n\nfunction baseMetadata(\n  options: McpClientTracerOptions,\n  overrides: Partial<McpTraceMetadata> = {},\n): Record<string, unknown> {\n  const metadata: McpTraceMetadata = {\n    source: { type: \"mcp-client\" },\n    ...(options.serverName ? { mcpServerName: options.serverName } : {}),\n    ...(options.serverUrl\n      ? { mcpServerUrlHash: hashServerUrl(options.serverUrl) }\n      : {}),\n    ...(options.sessionId ? { sessionId: options.sessionId } : {}),\n    ...overrides,\n  };\n  return { ...(options.metadata ?? {}), ...metadata };\n}\n\n/**\n * Wraps an MCP client so tools/list and tools/call emit local tool steps.\n * Telemetry only — does not proxy network behavior beyond the wrapped client.\n *\n * Uses a Proxy over the original client so prototype methods (`connect`,\n * `close`, `listResources`, …), private SDK state, and `instanceof` stay intact.\n * A shallow object spread would drop prototype methods and copy private fields.\n */\nexport function wrapMcpClient<T extends McpClientLike>(\n  client: T,\n  options: McpClientTracerOptions = {},\n): T {\n  const maxSummaryLength = options.maxSummaryLength ?? 240;\n\n  const listToolsImpl =\n    typeof client.listTools === \"function\"\n      ? client.listTools.bind(client)\n      : undefined;\n  const callToolImpl = client.callTool.bind(client);\n\n  const wrappedListTools = listToolsImpl\n    ? async (params?: unknown) =>\n        step(\n          \"mcp:tools/list\",\n          async () => {\n            const result = await listToolsImpl(params);\n            return result;\n          },\n          {\n            type: \"tool\",\n            metadata: baseMetadata(options, {\n              toolName: \"tools/list\",\n              toolCallId: nextToolCallId(options.toolCallIdPrefix),\n            }),\n          },\n        )\n    : undefined;\n\n  const wrappedCallTool = async (\n    params: Parameters<McpClientLike[\"callTool\"]>[0],\n  ) =>\n    step(\n      `mcp:${params.name}`,\n      async () => callToolImpl(params),\n      {\n        type: \"tool\",\n        metadata: baseMetadata(options, {\n          toolName: params.name,\n          toolCallId: nextToolCallId(options.toolCallIdPrefix),\n          argumentSummary: summarizeMcpValue(\n            params.arguments ?? {},\n            maxSummaryLength,\n          ),\n        }),\n      },\n    );\n\n  return new Proxy(client, {\n    get(target, property, receiver) {\n      if (property === \"listTools\" && wrappedListTools !== undefined) {\n        return wrappedListTools;\n      }\n      if (property === \"callTool\") {\n        return wrappedCallTool;\n      }\n      const value = Reflect.get(target, property, receiver);\n      if (typeof value === \"function\") {\n        return value.bind(target);\n      }\n      return value;\n    },\n  }) as T;\n}\n\n/** @internal resets tool call ids for deterministic tests */\nexport function resetMcpToolCallIdsForTests(): void {\n  toolCallCounter = 0;\n}\n"]}
+{"version":3,"sources":["../src/summarize.ts","../src/wrap.ts"],"names":["step"],"mappings":";;;;;AAAA,IAAM,0BAAA,GAA6B,GAAA;AAEnC,SAAS,eAAe,KAAA,EAAwB;AAC9C,EAAA,IAAI;AACF,IAAA,OAAO,OAAO,KAAK,CAAA;AAAA,EACrB,CAAA,CAAA,MAAQ;AACN,IAAA,OAAO,kBAAA;AAAA,EACT;AACF;AAEO,SAAS,iBAAA,CACd,KAAA,EACA,SAAA,GAAY,0BAAA,EACJ;AACR,EAAA,IAAI,IAAA;AACJ,EAAA,IAAI;AACF,IAAA,IAAI,OAAO,UAAU,QAAA,EAAU;AAC7B,MAAA,IAAA,GAAO,KAAA;AAAA,IACT,CAAA,MAAO;AAIL,MAAA,MAAM,UAAA,GAAa,IAAA,CAAK,SAAA,CAAU,KAAK,CAAA;AACvC,MAAA,IAAA,GAAO,UAAA,KAAe,KAAA,CAAA,GAAY,cAAA,CAAe,KAAK,CAAA,GAAI,UAAA;AAAA,IAC5D;AAAA,EACF,CAAA,CAAA,MAAQ;AACN,IAAA,IAAA,GAAO,eAAe,KAAK,CAAA;AAAA,EAC7B;AACA,EAAA,IAAI,IAAA,CAAK,MAAA,IAAU,SAAA,EAAW,OAAO,IAAA;AACrC,EAAA,OAAO,CAAA,EAAG,IAAA,CAAK,KAAA,CAAM,CAAA,EAAG,IAAA,CAAK,IAAI,CAAA,EAAG,SAAA,GAAY,CAAC,CAAC,CAAC,CAAA,GAAA,CAAA;AACrD;AAEO,SAAS,cAAc,GAAA,EAAqB;AACjD,EAAA,IAAI,IAAA,GAAO,CAAA;AACX,EAAA,KAAA,IAAS,IAAI,CAAA,EAAG,CAAA,GAAI,GAAA,CAAI,MAAA,EAAQ,KAAK,CAAA,EAAG;AACtC,IAAA,IAAA,GAAQ,IAAA,GAAO,EAAA,GAAK,GAAA,CAAI,UAAA,CAAW,CAAC,CAAA,KAAO,CAAA;AAAA,EAC7C;AACA,EAAA,OAAO,KAAK,QAAA,CAAS,EAAE,CAAA,CAAE,QAAA,CAAS,GAAG,GAAG,CAAA;AAC1C;AC5BA,IAAI,eAAA,GAAkB,CAAA;AAEtB,SAAS,eAAe,MAAA,EAAyB;AAC/C,EAAA,eAAA,IAAmB,CAAA;AACnB,EAAA,MAAM,IAAA,GAAO,MAAA,EAAQ,IAAA,EAAK,IAAK,KAAA;AAC/B,EAAA,OAAO,CAAA,EAAG,IAAI,CAAA,CAAA,EAAI,eAAe,CAAA,CAAA;AACnC;AAEA,SAAS,YAAA,CACP,OAAA,EACA,SAAA,GAAuC,EAAC,EACf;AACzB,EAAA,MAAM,QAAA,GAA6B;AAAA,IACjC,MAAA,EAAQ,EAAE,IAAA,EAAM,YAAA,EAAa;AAAA,IAC7B,GAAI,QAAQ,UAAA,GAAa,EAAE,eAAe,OAAA,CAAQ,UAAA,KAAe,EAAC;AAAA,IAClE,GAAI,OAAA,CAAQ,SAAA,GACR,EAAE,gBAAA,EAAkB,cAAc,OAAA,CAAQ,SAAS,CAAA,EAAE,GACrD,EAAC;AAAA,IACL,GAAI,QAAQ,SAAA,GAAY,EAAE,WAAW,OAAA,CAAQ,SAAA,KAAc,EAAC;AAAA,IAC5D,GAAG;AAAA,GACL;AACA,EAAA,OAAO,EAAE,GAAI,OAAA,CAAQ,YAAY,EAAC,EAAI,GAAG,QAAA,EAAS;AACpD;AAGA,IAAM,uBAAA,GAA0B;AAAA,EAC9B,QAAA;AAAA,EACA,SAAA;AAAA,EACA,YAAA;AAAA,EACA,wBAAA;AAAA,EACA,iBAAA;AAAA,EACA,kBAAA;AAAA,EACA,iBAAA;AAAA,EACA;AACF,CAAA;AAEA,SAAS,oBAAoB,KAAA,EAA4C;AACvE,EAAA,IAAI,CAAC,KAAA,IAAS,OAAO,KAAA,KAAU,UAAU,OAAO,KAAA;AAChD,EAAA,MAAM,GAAA,GAAM,KAAA;AACZ,EAAA,OAAO,uBAAA,CAAwB,IAAA,CAAK,CAAC,GAAA,KAAQ,OAAO,GAAG,CAAA;AACzD;AAMA,SAAS,0BACP,KAAA,EACiD;AACjD,EAAA,IAAI,CAAC,KAAA,IAAS,OAAO,KAAA,KAAU,UAAU,OAAO,KAAA;AAChD,EAAA,IAAI,mBAAA,CAAoB,KAAK,CAAA,EAAG,OAAO,KAAA;AACvC,EAAA,MAAM,GAAA,GAAM,KAAA;AACZ,EAAA,IAAI,EAAE,UAAA,IAAc,GAAA,CAAA,EAAM,OAAO,KAAA;AAEjC,EAAA,IAAI,OAAA,IAAW,KAAK,OAAO,KAAA;AAC3B,EAAA,OAAO,IAAA;AACT;AAEA,SAAS,oBAAoB,GAAA,EAED;AAC1B,EAAA,IAAI,GAAA,CAAI,QAAA,IAAY,OAAO,GAAA,CAAI,aAAa,QAAA,EAAU;AACpD,IAAA,OAAO,GAAA,CAAI,QAAA;AAAA,EACb;AACA,EAAA,OAAO,EAAC;AACV;AAOO,SAAS,mBAAA,CACd,uBACA,YAAA,EAKA;AAEA,EAAA,IAAI,iBAAiB,MAAA,EAAW;AAC9B,IAAA,IAAI,eAAwC,EAAC;AAC7C,IAAA,IAAI,cAAA;AAEJ,IAAA,IAAI,yBAAA,CAA0B,YAAY,CAAA,EAAG;AAC3C,MAAA,YAAA,GAAe,oBAAoB,YAAY,CAAA;AAE/C,MAAA,cAAA,GAAiB,MAAA;AAAA,IACnB,CAAA,MAAA,IAAW,mBAAA,CAAoB,YAAY,CAAA,IAAK,iBAAiB,IAAA,EAAM;AACrE,MAAA,cAAA,GACE,YAAA,IAAgB,OAAO,YAAA,KAAiB,QAAA,GACnC,YAAA,GACD,MAAA;AAAA,IACR,CAAA,MAAA,IAAW,YAAA,IAAgB,OAAO,YAAA,KAAiB,QAAA,EAAU;AAE3D,MAAA,cAAA,GAAiB,YAAA;AAAA,IACnB;AAEA,IAAA,OAAO;AAAA,MACL,YAAA,EAAc,qBAAA;AAAA,MACd,cAAA;AAAA,MACA;AAAA,KACF;AAAA,EACF;AAGA,EAAA,IAAI,yBAAA,CAA0B,qBAAqB,CAAA,EAAG;AACpD,IAAA,OAAO;AAAA,MACL,YAAA,EAAc,MAAA;AAAA,MACd,cAAA,EAAgB,MAAA;AAAA,MAChB,YAAA,EAAc,oBAAoB,qBAAqB;AAAA,KACzD;AAAA,EACF;AAEA,EAAA,IAAI,mBAAA,CAAoB,qBAAqB,CAAA,EAAG;AAC9C,IAAA,OAAO;AAAA,MACL,YAAA,EAAc,MAAA;AAAA,MACd,cAAA,EAAgB,qBAAA;AAAA,MAChB,cAAc;AAAC,KACjB;AAAA,EACF;AAEA,EAAA,OAAO;AAAA,IACL,YAAA,EAAc,qBAAA;AAAA,IACd,cAAA,EAAgB,MAAA;AAAA,IAChB,cAAc;AAAC,GACjB;AACF;AAEA,SAAS,cAAA,CACP,YAAA,EACA,MAAA,EACA,YAAA,EACA,cAAA,EACkB;AAClB,EAAA,MAAM,EAAA,GAAK,YAAA;AAKX,EAAA,IAAI,mBAAmB,MAAA,EAAW;AAChC,IAAA,OAAO,EAAA,CAAG,MAAA,EAAQ,YAAA,EAAc,cAAc,CAAA;AAAA,EAChD;AACA,EAAA,IAAI,iBAAiB,MAAA,EAAW;AAC9B,IAAA,OAAO,EAAA,CAAG,QAAQ,YAAY,CAAA;AAAA,EAChC;AACA,EAAA,OAAO,GAAG,MAAM,CAAA;AAClB;AAcO,SAAS,aAAA,CACd,MAAA,EACA,OAAA,GAAkC,EAAC,EAChC;AACH,EAAA,MAAM,gBAAA,GAAmB,QAAQ,gBAAA,IAAoB,GAAA;AAErD,EAAA,MAAM,aAAA,GACJ,OAAO,MAAA,CAAO,SAAA,KAAc,aACxB,MAAA,CAAO,SAAA,CAAU,IAAA,CAAK,MAAM,CAAA,GAC5B,MAAA;AACN,EAAA,MAAM,YAAA,GAAe,MAAA,CAAO,QAAA,CAAS,IAAA,CAAK,MAAM,CAAA;AAEhD,EAAA,MAAM,gBAAA,GAAmB,aAAA,GACrB,OAAO,MAAA,EAAkB,cAAA,KACvBA,iBAAA;AAAA,IACE,gBAAA;AAAA,IACA,YAAY;AACV,MAAA,IAAI,mBAAmB,MAAA,EAAW;AAChC,QAAA,OACE,aAAA,CAIA,QAAQ,cAAc,CAAA;AAAA,MAC1B;AACA,MAAA,OAAO,cAAc,MAAM,CAAA;AAAA,IAC7B,CAAA;AAAA,IACA;AAAA,MACE,IAAA,EAAM,MAAA;AAAA,MACN,QAAA,EAAU,aAAa,OAAA,EAAS;AAAA,QAC9B,QAAA,EAAU,YAAA;AAAA,QACV,UAAA,EAAY,cAAA,CAAe,OAAA,CAAQ,gBAAgB;AAAA,OACpD;AAAA;AACH,GACF,GACF,MAAA;AAEJ,EAAA,MAAM,eAAA,GAAkB,OACtB,MAAA,EACA,qBAAA,EACA,YAAA,KACG;AACH,IAAA,MAAM,EAAE,YAAA,EAAc,cAAA,EAAgB,YAAA,EAAa,GAAI,mBAAA;AAAA,MACrD,qBAAA;AAAA,MACA;AAAA,KACF;AAEA,IAAA,OAAOA,iBAAA;AAAA,MACL,CAAA,IAAA,EAAO,OAAO,IAAI,CAAA,CAAA;AAAA,MAClB,YACE,cAAA,CAAe,YAAA,EAAc,MAAA,EAAQ,cAAc,cAAc,CAAA;AAAA,MACnE;AAAA,QACE,IAAA,EAAM,MAAA;AAAA,QACN,QAAA,EAAU,aAAa,OAAA,EAAS;AAAA,UAC9B,UAAU,MAAA,CAAO,IAAA;AAAA,UACjB,UAAA,EAAY,cAAA,CAAe,OAAA,CAAQ,gBAAgB,CAAA;AAAA,UACnD,eAAA,EAAiB,iBAAA;AAAA,YACf,MAAA,CAAO,aAAa,EAAC;AAAA,YACrB;AAAA,WACF;AAAA,UACA,GAAG;AAAA,SACJ;AAAA;AACH,KACF;AAAA,EACF,CAAA;AAEA,EAAA,OAAO,IAAI,MAAM,MAAA,EAAQ;AAAA,IACvB,GAAA,CAAI,MAAA,EAAQ,QAAA,EAAU,QAAA,EAAU;AAC9B,MAAA,IAAI,QAAA,KAAa,WAAA,IAAe,gBAAA,KAAqB,MAAA,EAAW;AAC9D,QAAA,OAAO,gBAAA;AAAA,MACT;AACA,MAAA,IAAI,aAAa,UAAA,EAAY;AAC3B,QAAA,OAAO,eAAA;AAAA,MACT;AACA,MAAA,MAAM,KAAA,GAAQ,OAAA,CAAQ,GAAA,CAAI,MAAA,EAAQ,UAAU,QAAQ,CAAA;AACpD,MAAA,IAAI,OAAO,UAAU,UAAA,EAAY;AAC/B,QAAA,OAAO,KAAA,CAAM,KAAK,MAAM,CAAA;AAAA,MAC1B;AACA,MAAA,OAAO,KAAA;AAAA,IACT;AAAA,GACD,CAAA;AACH;AAGO,SAAS,2BAAA,GAAoC;AAClD,EAAA,eAAA,GAAkB,CAAA;AACpB","file":"index.cjs","sourcesContent":["const DEFAULT_MAX_SUMMARY_LENGTH = 240;\n\nfunction fallbackString(value: unknown): string {\n  try {\n    return String(value);\n  } catch {\n    return \"[unserializable]\";\n  }\n}\n\nexport function summarizeMcpValue(\n  value: unknown,\n  maxLength = DEFAULT_MAX_SUMMARY_LENGTH,\n): string {\n  let text: string;\n  try {\n    if (typeof value === \"string\") {\n      text = value;\n    } else {\n      // JSON.stringify returns undefined (not a string) for undefined,\n      // functions, and symbols; falling through with undefined would crash\n      // instrumentation at text.length below.\n      const serialized = JSON.stringify(value);\n      text = serialized === undefined ? fallbackString(value) : serialized;\n    }\n  } catch {\n    text = fallbackString(value);\n  }\n  if (text.length <= maxLength) return text;\n  return `${text.slice(0, Math.max(0, maxLength - 3))}...`;\n}\n\nexport function hashServerUrl(url: string): string {\n  let hash = 0;\n  for (let i = 0; i < url.length; i += 1) {\n    hash = (hash * 31 + url.charCodeAt(i)) >>> 0;\n  }\n  return hash.toString(16).padStart(8, \"0\");\n}\n","import { step } from \"agent-inspect\";\n\nimport { hashServerUrl, summarizeMcpValue } from \"./summarize.js\";\nimport type {\n  McpClientLike,\n  McpClientTracerOptions,\n  McpRequestOptions,\n  McpTraceMetadata,\n} from \"./types.js\";\n\nlet toolCallCounter = 0;\n\nfunction nextToolCallId(prefix?: string): string {\n  toolCallCounter += 1;\n  const base = prefix?.trim() || \"mcp\";\n  return `${base}-${toolCallCounter}`;\n}\n\nfunction baseMetadata(\n  options: McpClientTracerOptions,\n  overrides: Partial<McpTraceMetadata> = {},\n): Record<string, unknown> {\n  const metadata: McpTraceMetadata = {\n    source: { type: \"mcp-client\" },\n    ...(options.serverName ? { mcpServerName: options.serverName } : {}),\n    ...(options.serverUrl\n      ? { mcpServerUrlHash: hashServerUrl(options.serverUrl) }\n      : {}),\n    ...(options.sessionId ? { sessionId: options.sessionId } : {}),\n    ...overrides,\n  };\n  return { ...(options.metadata ?? {}), ...metadata };\n}\n\n/** SDK RequestOptions keys — must never be swallowed as AgentInspect metadata. */\nconst SDK_REQUEST_OPTION_KEYS = [\n  \"signal\",\n  \"timeout\",\n  \"onprogress\",\n  \"resetTimeoutOnProgress\",\n  \"maxTotalTimeout\",\n  \"relatedRequestId\",\n  \"resumptionToken\",\n  \"onresumptiontoken\",\n] as const;\n\nfunction isSdkRequestOptions(value: unknown): value is McpRequestOptions {\n  if (!value || typeof value !== \"object\") return false;\n  const obj = value as Record<string, unknown>;\n  return SDK_REQUEST_OPTION_KEYS.some((key) => key in obj);\n}\n\n/**\n * AgentInspect per-call metadata bag: `{ metadata: {...} }` without SDK option keys.\n * Must not consume ordinary SDK RequestOptions that happen to be objects.\n */\nfunction isAgentInspectMetadataBag(\n  value: unknown,\n): value is { metadata?: Record<string, unknown> } {\n  if (!value || typeof value !== \"object\") return false;\n  if (isSdkRequestOptions(value)) return false;\n  const obj = value as Record<string, unknown>;\n  if (!(\"metadata\" in obj)) return false;\n  // Zod/result schemas expose parse — never treat those as AI bags.\n  if (\"parse\" in obj) return false;\n  return true;\n}\n\nfunction extractCallMetadata(bag: {\n  metadata?: Record<string, unknown>;\n}): Record<string, unknown> {\n  if (bag.metadata && typeof bag.metadata === \"object\") {\n    return bag.metadata;\n  }\n  return {};\n}\n\n/**\n * Resolve SDK callTool arity:\n *   callTool(params, resultSchema?, requestOptions?)\n * plus AgentInspect overloads that peel `{ metadata }` without dropping RequestOptions.\n */\nexport function resolveCallToolArgs(\n  resultSchemaOrOptions?: unknown,\n  maybeOptions?: unknown,\n): {\n  resultSchema: unknown;\n  requestOptions: McpRequestOptions | undefined;\n  callMetadata: Record<string, unknown>;\n} {\n  // Three-arg form: (params, schema|undefined, options)\n  if (maybeOptions !== undefined) {\n    let callMetadata: Record<string, unknown> = {};\n    let requestOptions: McpRequestOptions | undefined;\n\n    if (isAgentInspectMetadataBag(maybeOptions)) {\n      callMetadata = extractCallMetadata(maybeOptions);\n      // Remaining SDK keys (if any were mixed in) still forward — but AI bags exclude them.\n      requestOptions = undefined;\n    } else if (isSdkRequestOptions(maybeOptions) || maybeOptions === null) {\n      requestOptions =\n        maybeOptions && typeof maybeOptions === \"object\"\n          ? (maybeOptions as McpRequestOptions)\n          : undefined;\n    } else if (maybeOptions && typeof maybeOptions === \"object\") {\n      // Unknown object third arg — forward as RequestOptions to preserve caller intent.\n      requestOptions = maybeOptions as McpRequestOptions;\n    }\n\n    return {\n      resultSchema: resultSchemaOrOptions,\n      requestOptions,\n      callMetadata,\n    };\n  }\n\n  // Two-arg form: (params, schema) OR (params, { metadata }) OR (params, RequestOptions)\n  if (isAgentInspectMetadataBag(resultSchemaOrOptions)) {\n    return {\n      resultSchema: undefined,\n      requestOptions: undefined,\n      callMetadata: extractCallMetadata(resultSchemaOrOptions),\n    };\n  }\n\n  if (isSdkRequestOptions(resultSchemaOrOptions)) {\n    return {\n      resultSchema: undefined,\n      requestOptions: resultSchemaOrOptions,\n      callMetadata: {},\n    };\n  }\n\n  return {\n    resultSchema: resultSchemaOrOptions,\n    requestOptions: undefined,\n    callMetadata: {},\n  };\n}\n\nfunction invokeCallTool(\n  callToolImpl: McpClientLike[\"callTool\"],\n  params: Parameters<McpClientLike[\"callTool\"]>[0],\n  resultSchema: unknown,\n  requestOptions: McpRequestOptions | undefined,\n): Promise<unknown> {\n  const fn = callToolImpl as (\n    p: unknown,\n    schema?: unknown,\n    opts?: unknown,\n  ) => Promise<unknown>;\n  if (requestOptions !== undefined) {\n    return fn(params, resultSchema, requestOptions);\n  }\n  if (resultSchema !== undefined) {\n    return fn(params, resultSchema);\n  }\n  return fn(params);\n}\n\n/**\n * Wraps an MCP client so tools/list and tools/call emit local tool steps.\n * Telemetry only — does not proxy network behavior beyond the wrapped client.\n *\n * Uses a Proxy over the original client so prototype methods (`connect`,\n * `close`, `listResources`, …), private SDK state, and `instanceof` stay intact.\n * A shallow object spread would drop prototype methods and copy private fields.\n *\n * Forwards the real SDK argument positions, including `RequestOptions`\n * (`signal`, `timeout`, progress callbacks). Does not fake cancellation while\n * still dispatching the underlying request.\n */\nexport function wrapMcpClient<T extends McpClientLike>(\n  client: T,\n  options: McpClientTracerOptions = {},\n): T {\n  const maxSummaryLength = options.maxSummaryLength ?? 240;\n\n  const listToolsImpl =\n    typeof client.listTools === \"function\"\n      ? client.listTools.bind(client)\n      : undefined;\n  const callToolImpl = client.callTool.bind(client);\n\n  const wrappedListTools = listToolsImpl\n    ? async (params?: unknown, requestOptions?: McpRequestOptions) =>\n        step(\n          \"mcp:tools/list\",\n          async () => {\n            if (requestOptions !== undefined) {\n              return (\n                listToolsImpl as (\n                  p?: unknown,\n                  opts?: unknown,\n                ) => Promise<unknown>\n              )(params, requestOptions);\n            }\n            return listToolsImpl(params);\n          },\n          {\n            type: \"tool\",\n            metadata: baseMetadata(options, {\n              toolName: \"tools/list\",\n              toolCallId: nextToolCallId(options.toolCallIdPrefix),\n            }),\n          },\n        )\n    : undefined;\n\n  const wrappedCallTool = async (\n    params: Parameters<McpClientLike[\"callTool\"]>[0],\n    resultSchemaOrOptions?: unknown,\n    maybeOptions?: unknown,\n  ) => {\n    const { resultSchema, requestOptions, callMetadata } = resolveCallToolArgs(\n      resultSchemaOrOptions,\n      maybeOptions,\n    );\n\n    return step(\n      `mcp:${params.name}`,\n      async () =>\n        invokeCallTool(callToolImpl, params, resultSchema, requestOptions),\n      {\n        type: \"tool\",\n        metadata: baseMetadata(options, {\n          toolName: params.name,\n          toolCallId: nextToolCallId(options.toolCallIdPrefix),\n          argumentSummary: summarizeMcpValue(\n            params.arguments ?? {},\n            maxSummaryLength,\n          ),\n          ...callMetadata,\n        }),\n      },\n    );\n  };\n\n  return new Proxy(client, {\n    get(target, property, receiver) {\n      if (property === \"listTools\" && wrappedListTools !== undefined) {\n        return wrappedListTools;\n      }\n      if (property === \"callTool\") {\n        return wrappedCallTool;\n      }\n      const value = Reflect.get(target, property, receiver);\n      if (typeof value === \"function\") {\n        return value.bind(target);\n      }\n      return value;\n    },\n  }) as T;\n}\n\n/** @internal resets tool call ids for deterministic tests */\nexport function resetMcpToolCallIdsForTests(): void {\n  toolCallCounter = 0;\n}\n"]}

package/dist/index.d.cts

@@ -15,13 +15,36 @@ interface McpCallToolResult {
     isError?: boolean;
     structuredContent?: unknown;
 }
+/**
+ * Subset of MCP SDK `RequestOptions` forwarded by the wrapper.
+ * Compatible with @modelcontextprotocol/sdk without requiring it at runtime.
+ */
+interface McpRequestOptions {
+    signal?: AbortSignal;
+    timeout?: number;
+    onprogress?: (...args: unknown[]) => void;
+    resetTimeoutOnProgress?: boolean;
+    maxTotalTimeout?: number;
+    relatedRequestId?: string;
+    resumptionToken?: string;
+    onresumptiontoken?: (...args: unknown[]) => void;
+    /** AgentInspect per-call metadata (optional; never consumes SDK option keys). */
+    metadata?: Record<string, unknown>;
+    [key: string]: unknown;
+}
 /**
  * Minimal MCP client surface for local tracing. Compatible with common SDK
  * client shapes without requiring @modelcontextprotocol/sdk at runtime.
+ *
+ * Matches SDK arity: `callTool(params, resultSchema?, requestOptions?)` and
+ * `listTools(params?, requestOptions?)`. AgentInspect also accepts a
+ * `{ metadata }` bag in place of schema/options when it is not RequestOptions-shaped.
  */
 interface McpClientLike {
-    listTools?: (params?: unknown) => Promise<McpListToolsResult>;
-    callTool: (params: McpCallToolParams) => Promise<McpCallToolResult>;
+    listTools?: (params?: unknown, options?: McpRequestOptions) => Promise<McpListToolsResult>;
+    callTool: (params: McpCallToolParams, resultSchemaOrOptions?: unknown, options?: McpRequestOptions | {
+        metadata?: Record<string, unknown>;
+    }) => Promise<McpCallToolResult>;
 }
 interface McpClientTracerOptions {
     serverName?: string;
@@ -50,6 +73,16 @@ interface McpTraceMetadata {
 declare function summarizeMcpValue(value: unknown, maxLength?: number): string;
 declare function hashServerUrl(url: string): string;
 
+/**
+ * Resolve SDK callTool arity:
+ *   callTool(params, resultSchema?, requestOptions?)
+ * plus AgentInspect overloads that peel `{ metadata }` without dropping RequestOptions.
+ */
+declare function resolveCallToolArgs(resultSchemaOrOptions?: unknown, maybeOptions?: unknown): {
+    resultSchema: unknown;
+    requestOptions: McpRequestOptions | undefined;
+    callMetadata: Record<string, unknown>;
+};
 /**
  * Wraps an MCP client so tools/list and tools/call emit local tool steps.
  * Telemetry only — does not proxy network behavior beyond the wrapped client.
@@ -57,9 +90,13 @@ declare function hashServerUrl(url: string): string;
  * Uses a Proxy over the original client so prototype methods (`connect`,
  * `close`, `listResources`, …), private SDK state, and `instanceof` stay intact.
  * A shallow object spread would drop prototype methods and copy private fields.
+ *
+ * Forwards the real SDK argument positions, including `RequestOptions`
+ * (`signal`, `timeout`, progress callbacks). Does not fake cancellation while
+ * still dispatching the underlying request.
  */
 declare function wrapMcpClient<T extends McpClientLike>(client: T, options?: McpClientTracerOptions): T;
 /** @internal resets tool call ids for deterministic tests */
 declare function resetMcpToolCallIdsForTests(): void;
 
-export { type McpCallToolParams, type McpCallToolResult, type McpClientLike, type McpClientTracerOptions, type McpListToolsResult, type McpToolDescriptor, type McpTraceMetadata, hashServerUrl, resetMcpToolCallIdsForTests, summarizeMcpValue, wrapMcpClient };
+export { type McpCallToolParams, type McpCallToolResult, type McpClientLike, type McpClientTracerOptions, type McpListToolsResult, type McpRequestOptions, type McpToolDescriptor, type McpTraceMetadata, hashServerUrl, resetMcpToolCallIdsForTests, resolveCallToolArgs, summarizeMcpValue, wrapMcpClient };

package/dist/index.d.ts

@@ -15,13 +15,36 @@ interface McpCallToolResult {
     isError?: boolean;
     structuredContent?: unknown;
 }
+/**
+ * Subset of MCP SDK `RequestOptions` forwarded by the wrapper.
+ * Compatible with @modelcontextprotocol/sdk without requiring it at runtime.
+ */
+interface McpRequestOptions {
+    signal?: AbortSignal;
+    timeout?: number;
+    onprogress?: (...args: unknown[]) => void;
+    resetTimeoutOnProgress?: boolean;
+    maxTotalTimeout?: number;
+    relatedRequestId?: string;
+    resumptionToken?: string;
+    onresumptiontoken?: (...args: unknown[]) => void;
+    /** AgentInspect per-call metadata (optional; never consumes SDK option keys). */
+    metadata?: Record<string, unknown>;
+    [key: string]: unknown;
+}
 /**
  * Minimal MCP client surface for local tracing. Compatible with common SDK
  * client shapes without requiring @modelcontextprotocol/sdk at runtime.
+ *
+ * Matches SDK arity: `callTool(params, resultSchema?, requestOptions?)` and
+ * `listTools(params?, requestOptions?)`. AgentInspect also accepts a
+ * `{ metadata }` bag in place of schema/options when it is not RequestOptions-shaped.
  */
 interface McpClientLike {
-    listTools?: (params?: unknown) => Promise<McpListToolsResult>;
-    callTool: (params: McpCallToolParams) => Promise<McpCallToolResult>;
+    listTools?: (params?: unknown, options?: McpRequestOptions) => Promise<McpListToolsResult>;
+    callTool: (params: McpCallToolParams, resultSchemaOrOptions?: unknown, options?: McpRequestOptions | {
+        metadata?: Record<string, unknown>;
+    }) => Promise<McpCallToolResult>;
 }
 interface McpClientTracerOptions {
     serverName?: string;
@@ -50,6 +73,16 @@ interface McpTraceMetadata {
 declare function summarizeMcpValue(value: unknown, maxLength?: number): string;
 declare function hashServerUrl(url: string): string;
 
+/**
+ * Resolve SDK callTool arity:
+ *   callTool(params, resultSchema?, requestOptions?)
+ * plus AgentInspect overloads that peel `{ metadata }` without dropping RequestOptions.
+ */
+declare function resolveCallToolArgs(resultSchemaOrOptions?: unknown, maybeOptions?: unknown): {
+    resultSchema: unknown;
+    requestOptions: McpRequestOptions | undefined;
+    callMetadata: Record<string, unknown>;
+};
 /**
  * Wraps an MCP client so tools/list and tools/call emit local tool steps.
  * Telemetry only — does not proxy network behavior beyond the wrapped client.
@@ -57,9 +90,13 @@ declare function hashServerUrl(url: string): string;
  * Uses a Proxy over the original client so prototype methods (`connect`,
  * `close`, `listResources`, …), private SDK state, and `instanceof` stay intact.
  * A shallow object spread would drop prototype methods and copy private fields.
+ *
+ * Forwards the real SDK argument positions, including `RequestOptions`
+ * (`signal`, `timeout`, progress callbacks). Does not fake cancellation while
+ * still dispatching the underlying request.
  */
 declare function wrapMcpClient<T extends McpClientLike>(client: T, options?: McpClientTracerOptions): T;
 /** @internal resets tool call ids for deterministic tests */
 declare function resetMcpToolCallIdsForTests(): void;
 
-export { type McpCallToolParams, type McpCallToolResult, type McpClientLike, type McpClientTracerOptions, type McpListToolsResult, type McpToolDescriptor, type McpTraceMetadata, hashServerUrl, resetMcpToolCallIdsForTests, summarizeMcpValue, wrapMcpClient };
+export { type McpCallToolParams, type McpCallToolResult, type McpClientLike, type McpClientTracerOptions, type McpListToolsResult, type McpRequestOptions, type McpToolDescriptor, type McpTraceMetadata, hashServerUrl, resetMcpToolCallIdsForTests, resolveCallToolArgs, summarizeMcpValue, wrapMcpClient };

package/dist/index.mjs

@@ -47,15 +47,94 @@ function baseMetadata(options, overrides = {}) {
   };
   return { ...options.metadata ?? {}, ...metadata };
 }
+var SDK_REQUEST_OPTION_KEYS = [
+  "signal",
+  "timeout",
+  "onprogress",
+  "resetTimeoutOnProgress",
+  "maxTotalTimeout",
+  "relatedRequestId",
+  "resumptionToken",
+  "onresumptiontoken"
+];
+function isSdkRequestOptions(value) {
+  if (!value || typeof value !== "object") return false;
+  const obj = value;
+  return SDK_REQUEST_OPTION_KEYS.some((key) => key in obj);
+}
+function isAgentInspectMetadataBag(value) {
+  if (!value || typeof value !== "object") return false;
+  if (isSdkRequestOptions(value)) return false;
+  const obj = value;
+  if (!("metadata" in obj)) return false;
+  if ("parse" in obj) return false;
+  return true;
+}
+function extractCallMetadata(bag) {
+  if (bag.metadata && typeof bag.metadata === "object") {
+    return bag.metadata;
+  }
+  return {};
+}
+function resolveCallToolArgs(resultSchemaOrOptions, maybeOptions) {
+  if (maybeOptions !== void 0) {
+    let callMetadata = {};
+    let requestOptions;
+    if (isAgentInspectMetadataBag(maybeOptions)) {
+      callMetadata = extractCallMetadata(maybeOptions);
+      requestOptions = void 0;
+    } else if (isSdkRequestOptions(maybeOptions) || maybeOptions === null) {
+      requestOptions = maybeOptions && typeof maybeOptions === "object" ? maybeOptions : void 0;
+    } else if (maybeOptions && typeof maybeOptions === "object") {
+      requestOptions = maybeOptions;
+    }
+    return {
+      resultSchema: resultSchemaOrOptions,
+      requestOptions,
+      callMetadata
+    };
+  }
+  if (isAgentInspectMetadataBag(resultSchemaOrOptions)) {
+    return {
+      resultSchema: void 0,
+      requestOptions: void 0,
+      callMetadata: extractCallMetadata(resultSchemaOrOptions)
+    };
+  }
+  if (isSdkRequestOptions(resultSchemaOrOptions)) {
+    return {
+      resultSchema: void 0,
+      requestOptions: resultSchemaOrOptions,
+      callMetadata: {}
+    };
+  }
+  return {
+    resultSchema: resultSchemaOrOptions,
+    requestOptions: void 0,
+    callMetadata: {}
+  };
+}
+function invokeCallTool(callToolImpl, params, resultSchema, requestOptions) {
+  const fn = callToolImpl;
+  if (requestOptions !== void 0) {
+    return fn(params, resultSchema, requestOptions);
+  }
+  if (resultSchema !== void 0) {
+    return fn(params, resultSchema);
+  }
+  return fn(params);
+}
 function wrapMcpClient(client, options = {}) {
   const maxSummaryLength = options.maxSummaryLength ?? 240;
   const listToolsImpl = typeof client.listTools === "function" ? client.listTools.bind(client) : void 0;
   const callToolImpl = client.callTool.bind(client);
-  const wrappedListTools = listToolsImpl ? async (params) => step(
+  const wrappedListTools = listToolsImpl ? async (params, requestOptions) => step(
     "mcp:tools/list",
     async () => {
-      const result = await listToolsImpl(params);
-      return result;
+      if (requestOptions !== void 0) {
+        return listToolsImpl(params, requestOptions);
+      }
+      return listToolsImpl(params);
     },
     {
       type: "tool",
@@ -65,21 +144,28 @@ function wrapMcpClient(client, options = {}) {
       })
     }
   ) : void 0;
-  const wrappedCallTool = async (params) => step(
-    `mcp:${params.name}`,
-    async () => callToolImpl(params),
-    {
-      type: "tool",
-      metadata: baseMetadata(options, {
-        toolName: params.name,
-        toolCallId: nextToolCallId(options.toolCallIdPrefix),
-        argumentSummary: summarizeMcpValue(
-          params.arguments ?? {},
-          maxSummaryLength
-        )
-      })
-    }
-  );
+  const wrappedCallTool = async (params, resultSchemaOrOptions, maybeOptions) => {
+    const { resultSchema, requestOptions, callMetadata } = resolveCallToolArgs(
+      resultSchemaOrOptions,
+      maybeOptions
+    );
+    return step(
+      `mcp:${params.name}`,
+      async () => invokeCallTool(callToolImpl, params, resultSchema, requestOptions),
+      {
+        type: "tool",
+        metadata: baseMetadata(options, {
+          toolName: params.name,
+          toolCallId: nextToolCallId(options.toolCallIdPrefix),
+          argumentSummary: summarizeMcpValue(
+            params.arguments ?? {},
+            maxSummaryLength
+          ),
+          ...callMetadata
+        })
+      }
+    );
+  };
   return new Proxy(client, {
     get(target, property, receiver) {
       if (property === "listTools" && wrappedListTools !== void 0) {
@@ -100,6 +186,6 @@ function resetMcpToolCallIdsForTests() {
   toolCallCounter = 0;
 }
 
-export { hashServerUrl, resetMcpToolCallIdsForTests, summarizeMcpValue, wrapMcpClient };
+export { hashServerUrl, resetMcpToolCallIdsForTests, resolveCallToolArgs, summarizeMcpValue, wrapMcpClient };
 //# sourceMappingURL=index.mjs.map
 //# sourceMappingURL=index.mjs.map

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/summarize.ts","../src/wrap.ts"],"names":[],"mappings":";;;AAAA,IAAM,0BAAA,GAA6B,GAAA;AAEnC,SAAS,eAAe,KAAA,EAAwB;AAC9C,EAAA,IAAI;AACF,IAAA,OAAO,OAAO,KAAK,CAAA;AAAA,EACrB,CAAA,CAAA,MAAQ;AACN,IAAA,OAAO,kBAAA;AAAA,EACT;AACF;AAEO,SAAS,iBAAA,CACd,KAAA,EACA,SAAA,GAAY,0BAAA,EACJ;AACR,EAAA,IAAI,IAAA;AACJ,EAAA,IAAI;AACF,IAAA,IAAI,OAAO,UAAU,QAAA,EAAU;AAC7B,MAAA,IAAA,GAAO,KAAA;AAAA,IACT,CAAA,MAAO;AAIL,MAAA,MAAM,UAAA,GAAa,IAAA,CAAK,SAAA,CAAU,KAAK,CAAA;AACvC,MAAA,IAAA,GAAO,UAAA,KAAe,KAAA,CAAA,GAAY,cAAA,CAAe,KAAK,CAAA,GAAI,UAAA;AAAA,IAC5D;AAAA,EACF,CAAA,CAAA,MAAQ;AACN,IAAA,IAAA,GAAO,eAAe,KAAK,CAAA;AAAA,EAC7B;AACA,EAAA,IAAI,IAAA,CAAK,MAAA,IAAU,SAAA,EAAW,OAAO,IAAA;AACrC,EAAA,OAAO,CAAA,EAAG,IAAA,CAAK,KAAA,CAAM,CAAA,EAAG,IAAA,CAAK,IAAI,CAAA,EAAG,SAAA,GAAY,CAAC,CAAC,CAAC,CAAA,GAAA,CAAA;AACrD;AAEO,SAAS,cAAc,GAAA,EAAqB;AACjD,EAAA,IAAI,IAAA,GAAO,CAAA;AACX,EAAA,KAAA,IAAS,IAAI,CAAA,EAAG,CAAA,GAAI,GAAA,CAAI,MAAA,EAAQ,KAAK,CAAA,EAAG;AACtC,IAAA,IAAA,GAAQ,IAAA,GAAO,EAAA,GAAK,GAAA,CAAI,UAAA,CAAW,CAAC,CAAA,KAAO,CAAA;AAAA,EAC7C;AACA,EAAA,OAAO,KAAK,QAAA,CAAS,EAAE,CAAA,CAAE,QAAA,CAAS,GAAG,GAAG,CAAA;AAC1C;AC7BA,IAAI,eAAA,GAAkB,CAAA;AAEtB,SAAS,eAAe,MAAA,EAAyB;AAC/C,EAAA,eAAA,IAAmB,CAAA;AACnB,EAAA,MAAM,IAAA,GAAO,MAAA,EAAQ,IAAA,EAAK,IAAK,KAAA;AAC/B,EAAA,OAAO,CAAA,EAAG,IAAI,CAAA,CAAA,EAAI,eAAe,CAAA,CAAA;AACnC;AAEA,SAAS,YAAA,CACP,OAAA,EACA,SAAA,GAAuC,EAAC,EACf;AACzB,EAAA,MAAM,QAAA,GAA6B;AAAA,IACjC,MAAA,EAAQ,EAAE,IAAA,EAAM,YAAA,EAAa;AAAA,IAC7B,GAAI,QAAQ,UAAA,GAAa,EAAE,eAAe,OAAA,CAAQ,UAAA,KAAe,EAAC;AAAA,IAClE,GAAI,OAAA,CAAQ,SAAA,GACR,EAAE,gBAAA,EAAkB,cAAc,OAAA,CAAQ,SAAS,CAAA,EAAE,GACrD,EAAC;AAAA,IACL,GAAI,QAAQ,SAAA,GAAY,EAAE,WAAW,OAAA,CAAQ,SAAA,KAAc,EAAC;AAAA,IAC5D,GAAG;AAAA,GACL;AACA,EAAA,OAAO,EAAE,GAAI,OAAA,CAAQ,YAAY,EAAC,EAAI,GAAG,QAAA,EAAS;AACpD;AAUO,SAAS,aAAA,CACd,MAAA,EACA,OAAA,GAAkC,EAAC,EAChC;AACH,EAAA,MAAM,gBAAA,GAAmB,QAAQ,gBAAA,IAAoB,GAAA;AAErD,EAAA,MAAM,aAAA,GACJ,OAAO,MAAA,CAAO,SAAA,KAAc,aACxB,MAAA,CAAO,SAAA,CAAU,IAAA,CAAK,MAAM,CAAA,GAC5B,MAAA;AACN,EAAA,MAAM,YAAA,GAAe,MAAA,CAAO,QAAA,CAAS,IAAA,CAAK,MAAM,CAAA;AAEhD,EAAA,MAAM,gBAAA,GAAmB,aAAA,GACrB,OAAO,MAAA,KACL,IAAA;AAAA,IACE,gBAAA;AAAA,IACA,YAAY;AACV,MAAA,MAAM,MAAA,GAAS,MAAM,aAAA,CAAc,MAAM,CAAA;AACzC,MAAA,OAAO,MAAA;AAAA,IACT,CAAA;AAAA,IACA;AAAA,MACE,IAAA,EAAM,MAAA;AAAA,MACN,QAAA,EAAU,aAAa,OAAA,EAAS;AAAA,QAC9B,QAAA,EAAU,YAAA;AAAA,QACV,UAAA,EAAY,cAAA,CAAe,OAAA,CAAQ,gBAAgB;AAAA,OACpD;AAAA;AACH,GACF,GACF,MAAA;AAEJ,EAAA,MAAM,eAAA,GAAkB,OACtB,MAAA,KAEA,IAAA;AAAA,IACE,CAAA,IAAA,EAAO,OAAO,IAAI,CAAA,CAAA;AAAA,IAClB,YAAY,aAAa,MAAM,CAAA;AAAA,IAC/B;AAAA,MACE,IAAA,EAAM,MAAA;AAAA,MACN,QAAA,EAAU,aAAa,OAAA,EAAS;AAAA,QAC9B,UAAU,MAAA,CAAO,IAAA;AAAA,QACjB,UAAA,EAAY,cAAA,CAAe,OAAA,CAAQ,gBAAgB,CAAA;AAAA,QACnD,eAAA,EAAiB,iBAAA;AAAA,UACf,MAAA,CAAO,aAAa,EAAC;AAAA,UACrB;AAAA;AACF,OACD;AAAA;AACH,GACF;AAEF,EAAA,OAAO,IAAI,MAAM,MAAA,EAAQ;AAAA,IACvB,GAAA,CAAI,MAAA,EAAQ,QAAA,EAAU,QAAA,EAAU;AAC9B,MAAA,IAAI,QAAA,KAAa,WAAA,IAAe,gBAAA,KAAqB,MAAA,EAAW;AAC9D,QAAA,OAAO,gBAAA;AAAA,MACT;AACA,MAAA,IAAI,aAAa,UAAA,EAAY;AAC3B,QAAA,OAAO,eAAA;AAAA,MACT;AACA,MAAA,MAAM,KAAA,GAAQ,OAAA,CAAQ,GAAA,CAAI,MAAA,EAAQ,UAAU,QAAQ,CAAA;AACpD,MAAA,IAAI,OAAO,UAAU,UAAA,EAAY;AAC/B,QAAA,OAAO,KAAA,CAAM,KAAK,MAAM,CAAA;AAAA,MAC1B;AACA,MAAA,OAAO,KAAA;AAAA,IACT;AAAA,GACD,CAAA;AACH;AAGO,SAAS,2BAAA,GAAoC;AAClD,EAAA,eAAA,GAAkB,CAAA;AACpB","file":"index.mjs","sourcesContent":["const DEFAULT_MAX_SUMMARY_LENGTH = 240;\n\nfunction fallbackString(value: unknown): string {\n  try {\n    return String(value);\n  } catch {\n    return \"[unserializable]\";\n  }\n}\n\nexport function summarizeMcpValue(\n  value: unknown,\n  maxLength = DEFAULT_MAX_SUMMARY_LENGTH,\n): string {\n  let text: string;\n  try {\n    if (typeof value === \"string\") {\n      text = value;\n    } else {\n      // JSON.stringify returns undefined (not a string) for undefined,\n      // functions, and symbols; falling through with undefined would crash\n      // instrumentation at text.length below.\n      const serialized = JSON.stringify(value);\n      text = serialized === undefined ? fallbackString(value) : serialized;\n    }\n  } catch {\n    text = fallbackString(value);\n  }\n  if (text.length <= maxLength) return text;\n  return `${text.slice(0, Math.max(0, maxLength - 3))}...`;\n}\n\nexport function hashServerUrl(url: string): string {\n  let hash = 0;\n  for (let i = 0; i < url.length; i += 1) {\n    hash = (hash * 31 + url.charCodeAt(i)) >>> 0;\n  }\n  return hash.toString(16).padStart(8, \"0\");\n}\n","import { step } from \"agent-inspect\";\n\nimport { hashServerUrl, summarizeMcpValue } from \"./summarize.js\";\nimport type {\n  McpClientLike,\n  McpClientTracerOptions,\n  McpTraceMetadata,\n} from \"./types.js\";\n\nlet toolCallCounter = 0;\n\nfunction nextToolCallId(prefix?: string): string {\n  toolCallCounter += 1;\n  const base = prefix?.trim() || \"mcp\";\n  return `${base}-${toolCallCounter}`;\n}\n\nfunction baseMetadata(\n  options: McpClientTracerOptions,\n  overrides: Partial<McpTraceMetadata> = {},\n): Record<string, unknown> {\n  const metadata: McpTraceMetadata = {\n    source: { type: \"mcp-client\" },\n    ...(options.serverName ? { mcpServerName: options.serverName } : {}),\n    ...(options.serverUrl\n      ? { mcpServerUrlHash: hashServerUrl(options.serverUrl) }\n      : {}),\n    ...(options.sessionId ? { sessionId: options.sessionId } : {}),\n    ...overrides,\n  };\n  return { ...(options.metadata ?? {}), ...metadata };\n}\n\n/**\n * Wraps an MCP client so tools/list and tools/call emit local tool steps.\n * Telemetry only — does not proxy network behavior beyond the wrapped client.\n *\n * Uses a Proxy over the original client so prototype methods (`connect`,\n * `close`, `listResources`, …), private SDK state, and `instanceof` stay intact.\n * A shallow object spread would drop prototype methods and copy private fields.\n */\nexport function wrapMcpClient<T extends McpClientLike>(\n  client: T,\n  options: McpClientTracerOptions = {},\n): T {\n  const maxSummaryLength = options.maxSummaryLength ?? 240;\n\n  const listToolsImpl =\n    typeof client.listTools === \"function\"\n      ? client.listTools.bind(client)\n      : undefined;\n  const callToolImpl = client.callTool.bind(client);\n\n  const wrappedListTools = listToolsImpl\n    ? async (params?: unknown) =>\n        step(\n          \"mcp:tools/list\",\n          async () => {\n            const result = await listToolsImpl(params);\n            return result;\n          },\n          {\n            type: \"tool\",\n            metadata: baseMetadata(options, {\n              toolName: \"tools/list\",\n              toolCallId: nextToolCallId(options.toolCallIdPrefix),\n            }),\n          },\n        )\n    : undefined;\n\n  const wrappedCallTool = async (\n    params: Parameters<McpClientLike[\"callTool\"]>[0],\n  ) =>\n    step(\n      `mcp:${params.name}`,\n      async () => callToolImpl(params),\n      {\n        type: \"tool\",\n        metadata: baseMetadata(options, {\n          toolName: params.name,\n          toolCallId: nextToolCallId(options.toolCallIdPrefix),\n          argumentSummary: summarizeMcpValue(\n            params.arguments ?? {},\n            maxSummaryLength,\n          ),\n        }),\n      },\n    );\n\n  return new Proxy(client, {\n    get(target, property, receiver) {\n      if (property === \"listTools\" && wrappedListTools !== undefined) {\n        return wrappedListTools;\n      }\n      if (property === \"callTool\") {\n        return wrappedCallTool;\n      }\n      const value = Reflect.get(target, property, receiver);\n      if (typeof value === \"function\") {\n        return value.bind(target);\n      }\n      return value;\n    },\n  }) as T;\n}\n\n/** @internal resets tool call ids for deterministic tests */\nexport function resetMcpToolCallIdsForTests(): void {\n  toolCallCounter = 0;\n}\n"]}
+{"version":3,"sources":["../src/summarize.ts","../src/wrap.ts"],"names":[],"mappings":";;;AAAA,IAAM,0BAAA,GAA6B,GAAA;AAEnC,SAAS,eAAe,KAAA,EAAwB;AAC9C,EAAA,IAAI;AACF,IAAA,OAAO,OAAO,KAAK,CAAA;AAAA,EACrB,CAAA,CAAA,MAAQ;AACN,IAAA,OAAO,kBAAA;AAAA,EACT;AACF;AAEO,SAAS,iBAAA,CACd,KAAA,EACA,SAAA,GAAY,0BAAA,EACJ;AACR,EAAA,IAAI,IAAA;AACJ,EAAA,IAAI;AACF,IAAA,IAAI,OAAO,UAAU,QAAA,EAAU;AAC7B,MAAA,IAAA,GAAO,KAAA;AAAA,IACT,CAAA,MAAO;AAIL,MAAA,MAAM,UAAA,GAAa,IAAA,CAAK,SAAA,CAAU,KAAK,CAAA;AACvC,MAAA,IAAA,GAAO,UAAA,KAAe,KAAA,CAAA,GAAY,cAAA,CAAe,KAAK,CAAA,GAAI,UAAA;AAAA,IAC5D;AAAA,EACF,CAAA,CAAA,MAAQ;AACN,IAAA,IAAA,GAAO,eAAe,KAAK,CAAA;AAAA,EAC7B;AACA,EAAA,IAAI,IAAA,CAAK,MAAA,IAAU,SAAA,EAAW,OAAO,IAAA;AACrC,EAAA,OAAO,CAAA,EAAG,IAAA,CAAK,KAAA,CAAM,CAAA,EAAG,IAAA,CAAK,IAAI,CAAA,EAAG,SAAA,GAAY,CAAC,CAAC,CAAC,CAAA,GAAA,CAAA;AACrD;AAEO,SAAS,cAAc,GAAA,EAAqB;AACjD,EAAA,IAAI,IAAA,GAAO,CAAA;AACX,EAAA,KAAA,IAAS,IAAI,CAAA,EAAG,CAAA,GAAI,GAAA,CAAI,MAAA,EAAQ,KAAK,CAAA,EAAG;AACtC,IAAA,IAAA,GAAQ,IAAA,GAAO,EAAA,GAAK,GAAA,CAAI,UAAA,CAAW,CAAC,CAAA,KAAO,CAAA;AAAA,EAC7C;AACA,EAAA,OAAO,KAAK,QAAA,CAAS,EAAE,CAAA,CAAE,QAAA,CAAS,GAAG,GAAG,CAAA;AAC1C;AC5BA,IAAI,eAAA,GAAkB,CAAA;AAEtB,SAAS,eAAe,MAAA,EAAyB;AAC/C,EAAA,eAAA,IAAmB,CAAA;AACnB,EAAA,MAAM,IAAA,GAAO,MAAA,EAAQ,IAAA,EAAK,IAAK,KAAA;AAC/B,EAAA,OAAO,CAAA,EAAG,IAAI,CAAA,CAAA,EAAI,eAAe,CAAA,CAAA;AACnC;AAEA,SAAS,YAAA,CACP,OAAA,EACA,SAAA,GAAuC,EAAC,EACf;AACzB,EAAA,MAAM,QAAA,GAA6B;AAAA,IACjC,MAAA,EAAQ,EAAE,IAAA,EAAM,YAAA,EAAa;AAAA,IAC7B,GAAI,QAAQ,UAAA,GAAa,EAAE,eAAe,OAAA,CAAQ,UAAA,KAAe,EAAC;AAAA,IAClE,GAAI,OAAA,CAAQ,SAAA,GACR,EAAE,gBAAA,EAAkB,cAAc,OAAA,CAAQ,SAAS,CAAA,EAAE,GACrD,EAAC;AAAA,IACL,GAAI,QAAQ,SAAA,GAAY,EAAE,WAAW,OAAA,CAAQ,SAAA,KAAc,EAAC;AAAA,IAC5D,GAAG;AAAA,GACL;AACA,EAAA,OAAO,EAAE,GAAI,OAAA,CAAQ,YAAY,EAAC,EAAI,GAAG,QAAA,EAAS;AACpD;AAGA,IAAM,uBAAA,GAA0B;AAAA,EAC9B,QAAA;AAAA,EACA,SAAA;AAAA,EACA,YAAA;AAAA,EACA,wBAAA;AAAA,EACA,iBAAA;AAAA,EACA,kBAAA;AAAA,EACA,iBAAA;AAAA,EACA;AACF,CAAA;AAEA,SAAS,oBAAoB,KAAA,EAA4C;AACvE,EAAA,IAAI,CAAC,KAAA,IAAS,OAAO,KAAA,KAAU,UAAU,OAAO,KAAA;AAChD,EAAA,MAAM,GAAA,GAAM,KAAA;AACZ,EAAA,OAAO,uBAAA,CAAwB,IAAA,CAAK,CAAC,GAAA,KAAQ,OAAO,GAAG,CAAA;AACzD;AAMA,SAAS,0BACP,KAAA,EACiD;AACjD,EAAA,IAAI,CAAC,KAAA,IAAS,OAAO,KAAA,KAAU,UAAU,OAAO,KAAA;AAChD,EAAA,IAAI,mBAAA,CAAoB,KAAK,CAAA,EAAG,OAAO,KAAA;AACvC,EAAA,MAAM,GAAA,GAAM,KAAA;AACZ,EAAA,IAAI,EAAE,UAAA,IAAc,GAAA,CAAA,EAAM,OAAO,KAAA;AAEjC,EAAA,IAAI,OAAA,IAAW,KAAK,OAAO,KAAA;AAC3B,EAAA,OAAO,IAAA;AACT;AAEA,SAAS,oBAAoB,GAAA,EAED;AAC1B,EAAA,IAAI,GAAA,CAAI,QAAA,IAAY,OAAO,GAAA,CAAI,aAAa,QAAA,EAAU;AACpD,IAAA,OAAO,GAAA,CAAI,QAAA;AAAA,EACb;AACA,EAAA,OAAO,EAAC;AACV;AAOO,SAAS,mBAAA,CACd,uBACA,YAAA,EAKA;AAEA,EAAA,IAAI,iBAAiB,MAAA,EAAW;AAC9B,IAAA,IAAI,eAAwC,EAAC;AAC7C,IAAA,IAAI,cAAA;AAEJ,IAAA,IAAI,yBAAA,CAA0B,YAAY,CAAA,EAAG;AAC3C,MAAA,YAAA,GAAe,oBAAoB,YAAY,CAAA;AAE/C,MAAA,cAAA,GAAiB,MAAA;AAAA,IACnB,CAAA,MAAA,IAAW,mBAAA,CAAoB,YAAY,CAAA,IAAK,iBAAiB,IAAA,EAAM;AACrE,MAAA,cAAA,GACE,YAAA,IAAgB,OAAO,YAAA,KAAiB,QAAA,GACnC,YAAA,GACD,MAAA;AAAA,IACR,CAAA,MAAA,IAAW,YAAA,IAAgB,OAAO,YAAA,KAAiB,QAAA,EAAU;AAE3D,MAAA,cAAA,GAAiB,YAAA;AAAA,IACnB;AAEA,IAAA,OAAO;AAAA,MACL,YAAA,EAAc,qBAAA;AAAA,MACd,cAAA;AAAA,MACA;AAAA,KACF;AAAA,EACF;AAGA,EAAA,IAAI,yBAAA,CAA0B,qBAAqB,CAAA,EAAG;AACpD,IAAA,OAAO;AAAA,MACL,YAAA,EAAc,MAAA;AAAA,MACd,cAAA,EAAgB,MAAA;AAAA,MAChB,YAAA,EAAc,oBAAoB,qBAAqB;AAAA,KACzD;AAAA,EACF;AAEA,EAAA,IAAI,mBAAA,CAAoB,qBAAqB,CAAA,EAAG;AAC9C,IAAA,OAAO;AAAA,MACL,YAAA,EAAc,MAAA;AAAA,MACd,cAAA,EAAgB,qBAAA;AAAA,MAChB,cAAc;AAAC,KACjB;AAAA,EACF;AAEA,EAAA,OAAO;AAAA,IACL,YAAA,EAAc,qBAAA;AAAA,IACd,cAAA,EAAgB,MAAA;AAAA,IAChB,cAAc;AAAC,GACjB;AACF;AAEA,SAAS,cAAA,CACP,YAAA,EACA,MAAA,EACA,YAAA,EACA,cAAA,EACkB;AAClB,EAAA,MAAM,EAAA,GAAK,YAAA;AAKX,EAAA,IAAI,mBAAmB,MAAA,EAAW;AAChC,IAAA,OAAO,EAAA,CAAG,MAAA,EAAQ,YAAA,EAAc,cAAc,CAAA;AAAA,EAChD;AACA,EAAA,IAAI,iBAAiB,MAAA,EAAW;AAC9B,IAAA,OAAO,EAAA,CAAG,QAAQ,YAAY,CAAA;AAAA,EAChC;AACA,EAAA,OAAO,GAAG,MAAM,CAAA;AAClB;AAcO,SAAS,aAAA,CACd,MAAA,EACA,OAAA,GAAkC,EAAC,EAChC;AACH,EAAA,MAAM,gBAAA,GAAmB,QAAQ,gBAAA,IAAoB,GAAA;AAErD,EAAA,MAAM,aAAA,GACJ,OAAO,MAAA,CAAO,SAAA,KAAc,aACxB,MAAA,CAAO,SAAA,CAAU,IAAA,CAAK,MAAM,CAAA,GAC5B,MAAA;AACN,EAAA,MAAM,YAAA,GAAe,MAAA,CAAO,QAAA,CAAS,IAAA,CAAK,MAAM,CAAA;AAEhD,EAAA,MAAM,gBAAA,GAAmB,aAAA,GACrB,OAAO,MAAA,EAAkB,cAAA,KACvB,IAAA;AAAA,IACE,gBAAA;AAAA,IACA,YAAY;AACV,MAAA,IAAI,mBAAmB,MAAA,EAAW;AAChC,QAAA,OACE,aAAA,CAIA,QAAQ,cAAc,CAAA;AAAA,MAC1B;AACA,MAAA,OAAO,cAAc,MAAM,CAAA;AAAA,IAC7B,CAAA;AAAA,IACA;AAAA,MACE,IAAA,EAAM,MAAA;AAAA,MACN,QAAA,EAAU,aAAa,OAAA,EAAS;AAAA,QAC9B,QAAA,EAAU,YAAA;AAAA,QACV,UAAA,EAAY,cAAA,CAAe,OAAA,CAAQ,gBAAgB;AAAA,OACpD;AAAA;AACH,GACF,GACF,MAAA;AAEJ,EAAA,MAAM,eAAA,GAAkB,OACtB,MAAA,EACA,qBAAA,EACA,YAAA,KACG;AACH,IAAA,MAAM,EAAE,YAAA,EAAc,cAAA,EAAgB,YAAA,EAAa,GAAI,mBAAA;AAAA,MACrD,qBAAA;AAAA,MACA;AAAA,KACF;AAEA,IAAA,OAAO,IAAA;AAAA,MACL,CAAA,IAAA,EAAO,OAAO,IAAI,CAAA,CAAA;AAAA,MAClB,YACE,cAAA,CAAe,YAAA,EAAc,MAAA,EAAQ,cAAc,cAAc,CAAA;AAAA,MACnE;AAAA,QACE,IAAA,EAAM,MAAA;AAAA,QACN,QAAA,EAAU,aAAa,OAAA,EAAS;AAAA,UAC9B,UAAU,MAAA,CAAO,IAAA;AAAA,UACjB,UAAA,EAAY,cAAA,CAAe,OAAA,CAAQ,gBAAgB,CAAA;AAAA,UACnD,eAAA,EAAiB,iBAAA;AAAA,YACf,MAAA,CAAO,aAAa,EAAC;AAAA,YACrB;AAAA,WACF;AAAA,UACA,GAAG;AAAA,SACJ;AAAA;AACH,KACF;AAAA,EACF,CAAA;AAEA,EAAA,OAAO,IAAI,MAAM,MAAA,EAAQ;AAAA,IACvB,GAAA,CAAI,MAAA,EAAQ,QAAA,EAAU,QAAA,EAAU;AAC9B,MAAA,IAAI,QAAA,KAAa,WAAA,IAAe,gBAAA,KAAqB,MAAA,EAAW;AAC9D,QAAA,OAAO,gBAAA;AAAA,MACT;AACA,MAAA,IAAI,aAAa,UAAA,EAAY;AAC3B,QAAA,OAAO,eAAA;AAAA,MACT;AACA,MAAA,MAAM,KAAA,GAAQ,OAAA,CAAQ,GAAA,CAAI,MAAA,EAAQ,UAAU,QAAQ,CAAA;AACpD,MAAA,IAAI,OAAO,UAAU,UAAA,EAAY;AAC/B,QAAA,OAAO,KAAA,CAAM,KAAK,MAAM,CAAA;AAAA,MAC1B;AACA,MAAA,OAAO,KAAA;AAAA,IACT;AAAA,GACD,CAAA;AACH;AAGO,SAAS,2BAAA,GAAoC;AAClD,EAAA,eAAA,GAAkB,CAAA;AACpB","file":"index.mjs","sourcesContent":["const DEFAULT_MAX_SUMMARY_LENGTH = 240;\n\nfunction fallbackString(value: unknown): string {\n  try {\n    return String(value);\n  } catch {\n    return \"[unserializable]\";\n  }\n}\n\nexport function summarizeMcpValue(\n  value: unknown,\n  maxLength = DEFAULT_MAX_SUMMARY_LENGTH,\n): string {\n  let text: string;\n  try {\n    if (typeof value === \"string\") {\n      text = value;\n    } else {\n      // JSON.stringify returns undefined (not a string) for undefined,\n      // functions, and symbols; falling through with undefined would crash\n      // instrumentation at text.length below.\n      const serialized = JSON.stringify(value);\n      text = serialized === undefined ? fallbackString(value) : serialized;\n    }\n  } catch {\n    text = fallbackString(value);\n  }\n  if (text.length <= maxLength) return text;\n  return `${text.slice(0, Math.max(0, maxLength - 3))}...`;\n}\n\nexport function hashServerUrl(url: string): string {\n  let hash = 0;\n  for (let i = 0; i < url.length; i += 1) {\n    hash = (hash * 31 + url.charCodeAt(i)) >>> 0;\n  }\n  return hash.toString(16).padStart(8, \"0\");\n}\n","import { step } from \"agent-inspect\";\n\nimport { hashServerUrl, summarizeMcpValue } from \"./summarize.js\";\nimport type {\n  McpClientLike,\n  McpClientTracerOptions,\n  McpRequestOptions,\n  McpTraceMetadata,\n} from \"./types.js\";\n\nlet toolCallCounter = 0;\n\nfunction nextToolCallId(prefix?: string): string {\n  toolCallCounter += 1;\n  const base = prefix?.trim() || \"mcp\";\n  return `${base}-${toolCallCounter}`;\n}\n\nfunction baseMetadata(\n  options: McpClientTracerOptions,\n  overrides: Partial<McpTraceMetadata> = {},\n): Record<string, unknown> {\n  const metadata: McpTraceMetadata = {\n    source: { type: \"mcp-client\" },\n    ...(options.serverName ? { mcpServerName: options.serverName } : {}),\n    ...(options.serverUrl\n      ? { mcpServerUrlHash: hashServerUrl(options.serverUrl) }\n      : {}),\n    ...(options.sessionId ? { sessionId: options.sessionId } : {}),\n    ...overrides,\n  };\n  return { ...(options.metadata ?? {}), ...metadata };\n}\n\n/** SDK RequestOptions keys — must never be swallowed as AgentInspect metadata. */\nconst SDK_REQUEST_OPTION_KEYS = [\n  \"signal\",\n  \"timeout\",\n  \"onprogress\",\n  \"resetTimeoutOnProgress\",\n  \"maxTotalTimeout\",\n  \"relatedRequestId\",\n  \"resumptionToken\",\n  \"onresumptiontoken\",\n] as const;\n\nfunction isSdkRequestOptions(value: unknown): value is McpRequestOptions {\n  if (!value || typeof value !== \"object\") return false;\n  const obj = value as Record<string, unknown>;\n  return SDK_REQUEST_OPTION_KEYS.some((key) => key in obj);\n}\n\n/**\n * AgentInspect per-call metadata bag: `{ metadata: {...} }` without SDK option keys.\n * Must not consume ordinary SDK RequestOptions that happen to be objects.\n */\nfunction isAgentInspectMetadataBag(\n  value: unknown,\n): value is { metadata?: Record<string, unknown> } {\n  if (!value || typeof value !== \"object\") return false;\n  if (isSdkRequestOptions(value)) return false;\n  const obj = value as Record<string, unknown>;\n  if (!(\"metadata\" in obj)) return false;\n  // Zod/result schemas expose parse — never treat those as AI bags.\n  if (\"parse\" in obj) return false;\n  return true;\n}\n\nfunction extractCallMetadata(bag: {\n  metadata?: Record<string, unknown>;\n}): Record<string, unknown> {\n  if (bag.metadata && typeof bag.metadata === \"object\") {\n    return bag.metadata;\n  }\n  return {};\n}\n\n/**\n * Resolve SDK callTool arity:\n *   callTool(params, resultSchema?, requestOptions?)\n * plus AgentInspect overloads that peel `{ metadata }` without dropping RequestOptions.\n */\nexport function resolveCallToolArgs(\n  resultSchemaOrOptions?: unknown,\n  maybeOptions?: unknown,\n): {\n  resultSchema: unknown;\n  requestOptions: McpRequestOptions | undefined;\n  callMetadata: Record<string, unknown>;\n} {\n  // Three-arg form: (params, schema|undefined, options)\n  if (maybeOptions !== undefined) {\n    let callMetadata: Record<string, unknown> = {};\n    let requestOptions: McpRequestOptions | undefined;\n\n    if (isAgentInspectMetadataBag(maybeOptions)) {\n      callMetadata = extractCallMetadata(maybeOptions);\n      // Remaining SDK keys (if any were mixed in) still forward — but AI bags exclude them.\n      requestOptions = undefined;\n    } else if (isSdkRequestOptions(maybeOptions) || maybeOptions === null) {\n      requestOptions =\n        maybeOptions && typeof maybeOptions === \"object\"\n          ? (maybeOptions as McpRequestOptions)\n          : undefined;\n    } else if (maybeOptions && typeof maybeOptions === \"object\") {\n      // Unknown object third arg — forward as RequestOptions to preserve caller intent.\n      requestOptions = maybeOptions as McpRequestOptions;\n    }\n\n    return {\n      resultSchema: resultSchemaOrOptions,\n      requestOptions,\n      callMetadata,\n    };\n  }\n\n  // Two-arg form: (params, schema) OR (params, { metadata }) OR (params, RequestOptions)\n  if (isAgentInspectMetadataBag(resultSchemaOrOptions)) {\n    return {\n      resultSchema: undefined,\n      requestOptions: undefined,\n      callMetadata: extractCallMetadata(resultSchemaOrOptions),\n    };\n  }\n\n  if (isSdkRequestOptions(resultSchemaOrOptions)) {\n    return {\n      resultSchema: undefined,\n      requestOptions: resultSchemaOrOptions,\n      callMetadata: {},\n    };\n  }\n\n  return {\n    resultSchema: resultSchemaOrOptions,\n    requestOptions: undefined,\n    callMetadata: {},\n  };\n}\n\nfunction invokeCallTool(\n  callToolImpl: McpClientLike[\"callTool\"],\n  params: Parameters<McpClientLike[\"callTool\"]>[0],\n  resultSchema: unknown,\n  requestOptions: McpRequestOptions | undefined,\n): Promise<unknown> {\n  const fn = callToolImpl as (\n    p: unknown,\n    schema?: unknown,\n    opts?: unknown,\n  ) => Promise<unknown>;\n  if (requestOptions !== undefined) {\n    return fn(params, resultSchema, requestOptions);\n  }\n  if (resultSchema !== undefined) {\n    return fn(params, resultSchema);\n  }\n  return fn(params);\n}\n\n/**\n * Wraps an MCP client so tools/list and tools/call emit local tool steps.\n * Telemetry only — does not proxy network behavior beyond the wrapped client.\n *\n * Uses a Proxy over the original client so prototype methods (`connect`,\n * `close`, `listResources`, …), private SDK state, and `instanceof` stay intact.\n * A shallow object spread would drop prototype methods and copy private fields.\n *\n * Forwards the real SDK argument positions, including `RequestOptions`\n * (`signal`, `timeout`, progress callbacks). Does not fake cancellation while\n * still dispatching the underlying request.\n */\nexport function wrapMcpClient<T extends McpClientLike>(\n  client: T,\n  options: McpClientTracerOptions = {},\n): T {\n  const maxSummaryLength = options.maxSummaryLength ?? 240;\n\n  const listToolsImpl =\n    typeof client.listTools === \"function\"\n      ? client.listTools.bind(client)\n      : undefined;\n  const callToolImpl = client.callTool.bind(client);\n\n  const wrappedListTools = listToolsImpl\n    ? async (params?: unknown, requestOptions?: McpRequestOptions) =>\n        step(\n          \"mcp:tools/list\",\n          async () => {\n            if (requestOptions !== undefined) {\n              return (\n                listToolsImpl as (\n                  p?: unknown,\n                  opts?: unknown,\n                ) => Promise<unknown>\n              )(params, requestOptions);\n            }\n            return listToolsImpl(params);\n          },\n          {\n            type: \"tool\",\n            metadata: baseMetadata(options, {\n              toolName: \"tools/list\",\n              toolCallId: nextToolCallId(options.toolCallIdPrefix),\n            }),\n          },\n        )\n    : undefined;\n\n  const wrappedCallTool = async (\n    params: Parameters<McpClientLike[\"callTool\"]>[0],\n    resultSchemaOrOptions?: unknown,\n    maybeOptions?: unknown,\n  ) => {\n    const { resultSchema, requestOptions, callMetadata } = resolveCallToolArgs(\n      resultSchemaOrOptions,\n      maybeOptions,\n    );\n\n    return step(\n      `mcp:${params.name}`,\n      async () =>\n        invokeCallTool(callToolImpl, params, resultSchema, requestOptions),\n      {\n        type: \"tool\",\n        metadata: baseMetadata(options, {\n          toolName: params.name,\n          toolCallId: nextToolCallId(options.toolCallIdPrefix),\n          argumentSummary: summarizeMcpValue(\n            params.arguments ?? {},\n            maxSummaryLength,\n          ),\n          ...callMetadata,\n        }),\n      },\n    );\n  };\n\n  return new Proxy(client, {\n    get(target, property, receiver) {\n      if (property === \"listTools\" && wrappedListTools !== undefined) {\n        return wrappedListTools;\n      }\n      if (property === \"callTool\") {\n        return wrappedCallTool;\n      }\n      const value = Reflect.get(target, property, receiver);\n      if (typeof value === \"function\") {\n        return value.bind(target);\n      }\n      return value;\n    },\n  }) as T;\n}\n\n/** @internal resets tool call ids for deterministic tests */\nexport function resetMcpToolCallIdsForTests(): void {\n  toolCallCounter = 0;\n}\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agent-inspect/mcp",
-  "version": "6.31.18",
+  "version": "6.31.20",
   "license": "MIT",
   "type": "module",
   "description": "Local MCP client tool-call tracing for AgentInspect",
@@ -35,7 +35,7 @@
     "dist"
   ],
   "dependencies": {
-    "agent-inspect": "6.31.18"
+    "agent-inspect": "6.31.20"
   },
   "publishConfig": {
     "access": "public"