[wrangler] Add AI agent detection to analytics events (#11820)

MattieTK · web-flow · commit b0e54b26f261 · 2026-01-12T13:18:59.000Z
* [wrangler] Add AI agent detection to analytics events

* fix: skip process ancestry checks to avoid wmic errors on Windows

The am-i-vibing library's detectAgenticEnvironment function uses
process-ancestry to check for running processes. When processAncestry
is undefined, it calls getProcessAncestry() which uses execSync('ps ...')
to walk up the entire process tree.

For each provider with processChecks (6 providers), it would call
getProcessAncestry() separately, causing:
- Massive slowdowns (tests took 78+ seconds instead of 3 seconds)
- Timeouts in CI environments (especially on Linux)
- Race conditions where output wasn't captured properly

By passing an empty array [], we skip process tree traversal entirely.
Environment variable detection is sufficient for identifying most
agentic environments (Claude Code, Cursor, Windsurf, etc.) and is
instantaneous.
diff --git a/.changeset/agentic-analytics.md b/.changeset/agentic-analytics.md
@@ -0,0 +1,9 @@
+---
+"wrangler": patch
+---
+
+Add AI agent detection to analytics events
+
+Wrangler now detects when commands are executed by AI coding agents (such as Claude Code, Cursor, GitHub Copilot, etc.) using the `am-i-vibing` library. This information is included as an `agent` property in all analytics events, helping Cloudflare understand how developers interact with Wrangler through AI assistants.
+
+The `agent` property will contain the agent ID (e.g., `"claude-code"`, `"cursor-agent"`) when detected, or `null` when running outside an agentic environment.
diff --git a/packages/wrangler/package.json b/packages/wrangler/package.json
@@ -112,6 +112,7 @@
 		"@types/yargs": "^17.0.22",
 		"@vitest/ui": "catalog:default",
 		"@webcontainer/env": "^1.1.0",
+		"am-i-vibing": "^0.1.0",
 		"capnweb": "^0.1.0",
 		"chalk": "^5.2.0",
 		"chokidar": "^4.0.1",
diff --git a/packages/wrangler/src/__tests__/metrics.test.ts b/packages/wrangler/src/__tests__/metrics.test.ts
@@ -1,5 +1,6 @@
 import * as fs from "node:fs";
 import { writeWranglerConfig } from "@cloudflare/workers-utils/test-helpers";
+import { detectAgenticEnvironment } from "am-i-vibing";
 import { http, HttpResponse } from "msw";
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
 import { CI } from "../is-ci";
@@ -26,6 +27,7 @@ import { runInTempDir } from "./helpers/run-in-tmp";
 import { runWrangler } from "./helpers/run-wrangler";
 import type { MockInstance } from "vitest";
 
+vi.mock("am-i-vibing");
 vi.mock("../metrics/helpers");
 vi.mock("../metrics/send-event");
 vi.mock("../package-manager");
@@ -82,6 +84,16 @@ describe("metrics", () => {
 		});
 
 		describe("sendAdhocEvent()", () => {
+			beforeEach(() => {
+				// Default: no agent detected
+				vi.mocked(detectAgenticEnvironment).mockReturnValue({
+					isAgentic: false,
+					id: null,
+					name: null,
+					type: null,
+				});
+			});
+
 			it("should send a request to the default URL", async () => {
 				const requests = mockMetricRequest();
 
@@ -92,7 +104,7 @@ describe("metrics", () => {
 				await Promise.all(dispatcher.requests);
 				expect(requests.count).toBe(1);
 				expect(std.debug).toMatchInlineSnapshot(
-					`"Metrics dispatcher: Posting data {\\"deviceId\\":\\"f82b1f46-eb7b-4154-aa9f-ce95f23b2288\\",\\"event\\":\\"some-event\\",\\"timestamp\\":1733961600000,\\"properties\\":{\\"category\\":\\"Workers\\",\\"wranglerVersion\\":\\"1.2.3\\",\\"wranglerMajorVersion\\":1,\\"wranglerMinorVersion\\":2,\\"wranglerPatchVersion\\":3,\\"os\\":\\"foo:bar\\",\\"a\\":1,\\"b\\":2}}"`
+					`"Metrics dispatcher: Posting data {\\"deviceId\\":\\"f82b1f46-eb7b-4154-aa9f-ce95f23b2288\\",\\"event\\":\\"some-event\\",\\"timestamp\\":1733961600000,\\"properties\\":{\\"category\\":\\"Workers\\",\\"wranglerVersion\\":\\"1.2.3\\",\\"wranglerMajorVersion\\":1,\\"wranglerMinorVersion\\":2,\\"wranglerPatchVersion\\":3,\\"os\\":\\"foo:bar\\",\\"agent\\":null,\\"a\\":1,\\"b\\":2}}"`
 				);
 				expect(std.out).toMatchInlineSnapshot(`""`);
 				expect(std.warn).toMatchInlineSnapshot(`""`);
@@ -123,7 +135,7 @@ describe("metrics", () => {
 
 				expect(requests.count).toBe(0);
 				expect(std.debug).toMatchInlineSnapshot(
-					`"Metrics dispatcher: Dispatching disabled - would have sent {\\"deviceId\\":\\"f82b1f46-eb7b-4154-aa9f-ce95f23b2288\\",\\"event\\":\\"some-event\\",\\"timestamp\\":1733961600000,\\"properties\\":{\\"category\\":\\"Workers\\",\\"wranglerVersion\\":\\"1.2.3\\",\\"wranglerMajorVersion\\":1,\\"wranglerMinorVersion\\":2,\\"wranglerPatchVersion\\":3,\\"os\\":\\"foo:bar\\",\\"a\\":1,\\"b\\":2}}."`
+					`"Metrics dispatcher: Dispatching disabled - would have sent {\\"deviceId\\":\\"f82b1f46-eb7b-4154-aa9f-ce95f23b2288\\",\\"event\\":\\"some-event\\",\\"timestamp\\":1733961600000,\\"properties\\":{\\"category\\":\\"Workers\\",\\"wranglerVersion\\":\\"1.2.3\\",\\"wranglerMajorVersion\\":1,\\"wranglerMinorVersion\\":2,\\"wranglerPatchVersion\\":3,\\"os\\":\\"foo:bar\\",\\"agent\\":null,\\"a\\":1,\\"b\\":2}}."`
 				);
 				expect(std.out).toMatchInlineSnapshot(`""`);
 				expect(std.warn).toMatchInlineSnapshot(`""`);
@@ -143,7 +155,7 @@ describe("metrics", () => {
 				await Promise.all(dispatcher.requests);
 
 				expect(std.debug).toMatchInlineSnapshot(`
-					"Metrics dispatcher: Posting data {\\"deviceId\\":\\"f82b1f46-eb7b-4154-aa9f-ce95f23b2288\\",\\"event\\":\\"some-event\\",\\"timestamp\\":1733961600000,\\"properties\\":{\\"category\\":\\"Workers\\",\\"wranglerVersion\\":\\"1.2.3\\",\\"wranglerMajorVersion\\":1,\\"wranglerMinorVersion\\":2,\\"wranglerPatchVersion\\":3,\\"os\\":\\"foo:bar\\",\\"a\\":1,\\"b\\":2}}
+					"Metrics dispatcher: Posting data {\\"deviceId\\":\\"f82b1f46-eb7b-4154-aa9f-ce95f23b2288\\",\\"event\\":\\"some-event\\",\\"timestamp\\":1733961600000,\\"properties\\":{\\"category\\":\\"Workers\\",\\"wranglerVersion\\":\\"1.2.3\\",\\"wranglerMajorVersion\\":1,\\"wranglerMinorVersion\\":2,\\"wranglerPatchVersion\\":3,\\"os\\":\\"foo:bar\\",\\"agent\\":null,\\"a\\":1,\\"b\\":2}}
 					Metrics dispatcher: Failed to send request: Failed to fetch"
 				`);
 				expect(std.out).toMatchInlineSnapshot(`""`);
@@ -162,12 +174,47 @@ describe("metrics", () => {
 
 				expect(requests.count).toBe(0);
 				expect(std.debug).toMatchInlineSnapshot(
-					`"Metrics dispatcher: Source Key not provided. Be sure to initialize before sending events {\\"deviceId\\":\\"f82b1f46-eb7b-4154-aa9f-ce95f23b2288\\",\\"event\\":\\"some-event\\",\\"timestamp\\":1733961600000,\\"properties\\":{\\"category\\":\\"Workers\\",\\"wranglerVersion\\":\\"1.2.3\\",\\"wranglerMajorVersion\\":1,\\"wranglerMinorVersion\\":2,\\"wranglerPatchVersion\\":3,\\"os\\":\\"foo:bar\\",\\"a\\":1,\\"b\\":2}}"`
+					`"Metrics dispatcher: Source Key not provided. Be sure to initialize before sending events {\\"deviceId\\":\\"f82b1f46-eb7b-4154-aa9f-ce95f23b2288\\",\\"event\\":\\"some-event\\",\\"timestamp\\":1733961600000,\\"properties\\":{\\"category\\":\\"Workers\\",\\"wranglerVersion\\":\\"1.2.3\\",\\"wranglerMajorVersion\\":1,\\"wranglerMinorVersion\\":2,\\"wranglerPatchVersion\\":3,\\"os\\":\\"foo:bar\\",\\"agent\\":null,\\"a\\":1,\\"b\\":2}}"`
 				);
 				expect(std.out).toMatchInlineSnapshot(`""`);
 				expect(std.warn).toMatchInlineSnapshot(`""`);
 				expect(std.err).toMatchInlineSnapshot(`""`);
 			});
+
+			it("should include agent ID when detected", async () => {
+				vi.mocked(detectAgenticEnvironment).mockReturnValue({
+					isAgentic: true,
+					id: "claude-code",
+					name: "Claude Code",
+					type: "agent",
+				});
+
+				const requests = mockMetricRequest();
+				const dispatcher = getMetricsDispatcher({
+					sendMetrics: true,
+				});
+				dispatcher.sendAdhocEvent("some-event", { a: 1 });
+				await Promise.all(dispatcher.requests);
+
+				expect(requests.count).toBe(1);
+				expect(std.debug).toContain('"agent":"claude-code"');
+			});
+
+			it("should set agent to null if detection throws", async () => {
+				vi.mocked(detectAgenticEnvironment).mockImplementation(() => {
+					throw new Error("Detection failed");
+				});
+
+				const requests = mockMetricRequest();
+				const dispatcher = getMetricsDispatcher({
+					sendMetrics: true,
+				});
+				dispatcher.sendAdhocEvent("some-event", { a: 1 });
+				await Promise.all(dispatcher.requests);
+
+				expect(requests.count).toBe(1);
+				expect(std.debug).toContain('"agent":null');
+			});
 		});
 
 		it("should keep track of all requests made", async () => {
@@ -207,10 +254,18 @@ describe("metrics", () => {
 				hasAssets: false,
 				argsUsed: [],
 				argsCombination: "",
+				agent: null,
 				command: "wrangler docs",
 				args: {},
 			};
 			beforeEach(() => {
+				// Default: no agent detected
+				vi.mocked(detectAgenticEnvironment).mockReturnValue({
+					isAgentic: false,
+					id: null,
+					name: null,
+					type: null,
+				});
 				globalThis.ALGOLIA_APP_ID = "FAKE-ID";
 				globalThis.ALGOLIA_PUBLIC_KEY = "FAKE-KEY";
 				msw.use(
@@ -488,6 +543,23 @@ describe("metrics", () => {
 				expect(std.debug).toContain('"errorMessage":"yargs validation error"');
 			});
 
+			it("should include agent ID in command events when detected", async () => {
+				vi.mocked(detectAgenticEnvironment).mockReturnValue({
+					isAgentic: true,
+					id: "cursor-agent",
+					name: "Cursor Agent",
+					type: "agent",
+				});
+
+				writeWranglerConfig();
+				const requests = mockMetricRequest();
+
+				await runWrangler("docs arg");
+
+				expect(requests.count).toBe(2);
+				expect(std.debug).toContain('"agent":"cursor-agent"');
+			});
+
 			describe("banner", () => {
 				beforeEach(() => {
 					vi.mocked(getWranglerVersion).mockReturnValue("1.2.3");
diff --git a/packages/wrangler/src/metrics/metrics-dispatcher.ts b/packages/wrangler/src/metrics/metrics-dispatcher.ts
@@ -1,4 +1,5 @@
 import { configFormat } from "@cloudflare/workers-utils";
+import { detectAgenticEnvironment } from "am-i-vibing";
 import chalk from "chalk";
 import { fetch } from "undici";
 import isInteractive from "../is-interactive";
@@ -64,6 +65,19 @@ export function getMetricsDispatcher(options: MetricsConfigOptions) {
 	const amplitude_session_id = Date.now();
 	let amplitude_event_id = 0;
 
+	// Detect agent environment once when dispatcher is created
+	// Pass empty array for processAncestry to skip process tree checks entirely.
+	// Process tree traversal uses execSync('ps ...') which is slow and can cause
+	// timeouts, especially in CI environments. Environment variable detection
+	// is sufficient for identifying most agentic environments.
+	let agent: string | null = null;
+	try {
+		const agentDetection = detectAgenticEnvironment(process.env, []);
+		agent = agentDetection.id;
+	} catch {
+		// Silent failure - agent remains null
+	}
+
 	return {
 		/**
 		 * This doesn't have a session id and is not tied to the command events.
@@ -82,6 +96,7 @@ export function getMetricsDispatcher(options: MetricsConfigOptions) {
 					wranglerMinorVersion,
 					wranglerPatchVersion,
 					os: getOS(),
+					agent,
 					...properties,
 				},
 			});
@@ -140,6 +155,7 @@ export function getMetricsDispatcher(options: MetricsConfigOptions) {
 					hasAssets: options.hasAssets ?? false,
 					argsUsed: sanitizedArgsKeys,
 					argsCombination: sanitizedArgsKeys.join(", "),
+					agent,
 				};
 
 				// get the args where we don't want to redact their values
diff --git a/packages/wrangler/src/metrics/types.ts b/packages/wrangler/src/metrics/types.ts
@@ -65,6 +65,11 @@ export type CommonEventProperties = {
 	 * Same as argsUsed except concatenated for convenience in Amplitude
 	 */
 	argsCombination: string;
+	/**
+	 * The detected AI agent environment ID, if any (e.g., "claude-code", "cursor-agent").
+	 * Null if not running in an agentic environment.
+	 */
+	agent: string | null;
 };
 
 /** We send a metrics event at the start and end of a command run */
diff --git a/packages/wrangler/telemetry.md b/packages/wrangler/telemetry.md
@@ -26,6 +26,7 @@ Telemetry in Wrangler allows us to better identify bugs and gain visibility on u
 - The format of the Wrangler configuration file (e.g. `toml`, `jsonc`)
 - Total session duration of the command run (e.g. 3 seconds, etc.)
 - Whether the Wrangler client is running in CI or in an interactive instance
+- Whether the command was executed by an AI coding agent (e.g. Claude Code, Cursor, GitHub Copilot), and if so, which agent
 - Error _type_ (e.g. `APIError` or `UserError`), and sanitised error messages that will not include user information like filepaths or stack traces (e.g. `Asset too large`).
 - General machine information such as OS and OS Version
 
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
