instrumentation ++

Author: tech-sushant

src/tools/bstack-sdk.ts

@@ -3,6 +3,8 @@ import { BrowserStackConfig } from "../lib/types.js";
 import { RunTestsOnBrowserStackParamsShape } from "./sdk-utils/common/schema.js";
 import { runTestsOnBrowserStackHandler } from "./sdk-utils/handler.js";
 import { RUN_ON_BROWSERSTACK_DESCRIPTION } from "./sdk-utils/common/constants.js";
+import { handleMCPError } from "../lib/utils.js";
+import { trackMCP } from "../lib/instrumentation.js";
 
 export function registerRunBrowserStackTestsTool(
   server: McpServer,
@@ -15,7 +17,12 @@ export function registerRunBrowserStackTestsTool(
     RUN_ON_BROWSERSTACK_DESCRIPTION,
     RunTestsOnBrowserStackParamsShape,
     async (args) => {
-      return runTestsOnBrowserStackHandler(args, config);
+      try {
+        trackMCP("runTestsOnBrowserStack", server.server.getClientVersion()!, config);
+        return await runTestsOnBrowserStackHandler(args, config);
+      } catch (error) {
+        return handleMCPError("runTestsOnBrowserStack", server, config, error);
+      }
     },
   );
 

src/tools/build-insights.ts

@@ -4,6 +4,7 @@ import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import logger from "../logger.js";
 import { BrowserStackConfig } from "../lib/types.js";
 import { fetchFromBrowserStackAPI, handleMCPError } from "../lib/utils.js";
+import { trackMCP } from "../lib/instrumentation.js";
 
 // Tool function that fetches build insights from two APIs
 export async function fetchBuildInsightsTool(
@@ -78,6 +79,7 @@ export default function addBuildInsightsTools(
     },
     async (args) => {
       try {
+        trackMCP("fetchBuildInsights", server.server.getClientVersion()!, config);
         return await fetchBuildInsightsTool(args, config);
       } catch (error) {
         return handleMCPError("fetchBuildInsights", server, config, error);

src/tools/rca-agent.ts

@@ -9,6 +9,7 @@ import { getRCAData } from "./rca-agent-utils/rca-data.js";
 import { formatRCAData } from "./rca-agent-utils/format-rca.js";
 import { TestStatus } from "./rca-agent-utils/types.js";
 import { handleMCPError } from "../lib/utils.js";
+import { trackMCP } from "../index.js";
 import {
   FETCH_RCA_PARAMS,
   GET_BUILD_ID_PARAMS,
@@ -134,6 +135,7 @@ export default function addRCATools(
     FETCH_RCA_PARAMS,
     async (args) => {
       try {
+        trackMCP("fetchRCA", server.server.getClientVersion()!, config);
         return await fetchRCADataTool(args, config);
       } catch (error) {
         return handleMCPError("fetchRCA", server, config, error);
@@ -147,6 +149,7 @@ export default function addRCATools(
     GET_BUILD_ID_PARAMS,
     async (args) => {
       try {
+        trackMCP("getBuildId", server.server.getClientVersion()!, config);
         return await getBuildIdTool(args, config);
       } catch (error) {
         return handleMCPError("getBuildId", server, config, error);
@@ -160,6 +163,7 @@ export default function addRCATools(
     LIST_TEST_IDS_PARAMS,
     async (args) => {
       try {
+        trackMCP("listTestIds", server.server.getClientVersion()!, config);
         return await listTestIdsTool(args, config);
       } catch (error) {
         return handleMCPError("listTestIds", server, config, error);