feat(wizard): Add User-Agent header to wizard HTTP requests (#291)

Co-authored-by: Claude <noreply@anthropic.com>

Author: robbie-c

src/lib/agent-interface.ts

@@ -11,6 +11,7 @@ import { analytics } from '../utils/analytics';
 import {
   WIZARD_INTERACTION_EVENT_NAME,
   WIZARD_REMARK_EVENT_NAME,
+  WIZARD_USER_AGENT,
 } from './constants';
 import { getLlmGatewayUrlFromHost } from '../utils/urls';
 import { LINTING_TOOLS } from './safe-tools';
@@ -357,6 +358,7 @@ export async function initializeAgent(
         url: config.posthogMcpUrl,
         headers: {
           Authorization: `Bearer ${config.posthogApiKey}`,
+          'User-Agent': WIZARD_USER_AGENT,
         },
       },
       ...Object.fromEntries(

src/lib/api.ts

@@ -1,6 +1,7 @@
 import axios, { AxiosError } from 'axios';
 import { z } from 'zod';
 import { analytics } from '../utils/analytics';
+import { WIZARD_USER_AGENT } from './constants';
 
 export const ApiUserSchema = z.object({
   distinct_id: z.string(),
@@ -48,6 +49,7 @@ export async function fetchUserData(
     const response = await axios.get(`${baseUrl}/api/users/@me/`, {
       headers: {
         Authorization: `Bearer ${accessToken}`,
+        'User-Agent': WIZARD_USER_AGENT,
       },
     });
 
@@ -71,6 +73,7 @@ export async function fetchProjectData(
     const response = await axios.get(`${baseUrl}/api/projects/${projectId}/`, {
       headers: {
         Authorization: `Bearer ${accessToken}`,
+        'User-Agent': WIZARD_USER_AGENT,
       },
     });
 

src/lib/constants.ts

@@ -57,3 +57,11 @@ export const OAUTH_PORT = 8239;
 
 export const WIZARD_INTERACTION_EVENT_NAME = 'wizard interaction';
 export const WIZARD_REMARK_EVENT_NAME = 'wizard remark';
+
+/**
+ * User-Agent string for the wizard when making HTTP requests.
+ * Used for direct PostHog API calls and passed to the MCP server
+ * so it can identify requests originating from the wizard.
+ */
+import packageJson from '../../package.json';
+export const WIZARD_USER_AGENT = `posthog/wizard; version: ${packageJson.version}`;

src/utils/oauth.ts

@@ -5,7 +5,7 @@ import chalk from 'chalk';
 import opn from 'opn';
 import { z } from 'zod';
 import clack from './clack';
-import { ISSUES_URL, OAUTH_PORT } from '../lib/constants';
+import { ISSUES_URL, OAUTH_PORT, WIZARD_USER_AGENT } from '../lib/constants';
 import { abort } from './clack-utils';
 import { analytics } from './analytics';
 import type { CloudRegion } from './types';
@@ -187,6 +187,7 @@ async function exchangeCodeForToken(
     {
       headers: {
         'Content-Type': 'application/json',
+        'User-Agent': WIZARD_USER_AGENT,
       },
     },
   );

tsconfig.json

@@ -7,6 +7,7 @@
     "types": ["node", "jest"],
     "typeRoots": ["./node_modules/@types", "./types"],
     "declaration": true,
+    "resolveJsonModule": true,
     "outDir": "dist"
   },
   "include": [