Inline managed auth telemetry schemas

Author: masnwilliams

src/lib/mcp/tools/auth-connections.test.ts

@@ -1,5 +1,7 @@
 import { afterEach, describe, expect, test } from "bun:test";
 import { organizationWideAuthInfo } from "@/lib/mcp/auth-context.test-fixtures";
+import { connectTestMcp } from "@/lib/mcp/mcp-test-fixtures";
+import { registerAuthConnectionTools } from "./auth-connections";
 import { toSafeAuthConnection } from "./managed-auth-state";
 import {
   assertNoSecrets,
@@ -46,6 +48,25 @@ describe("manage_auth_connections programmatic surface", () => {
     expect(schema?.sign_in_option_id).toBeUndefined();
   });
 
+  test("publishes browser telemetry categories without JSON Schema references", async () => {
+    const { client, close } = await connectTestMcp(
+      registerAuthConnectionTools,
+      unusedKernelClient,
+    );
+
+    try {
+      const tool = (await client.listTools()).tools.find(
+        ({ name }) => name === "manage_auth_connections",
+      );
+      const browserTelemetry = tool?.inputSchema.properties?.browser_telemetry;
+
+      expect(browserTelemetry).toBeDefined();
+      expect(JSON.stringify(browserTelemetry)).not.toContain('"$ref"');
+    } finally {
+      await close();
+    }
+  });
+
   test("passes the optional project selector through to the client", async () => {
     const { handler, schema } = captureHandler();
     let selectedProject: string | undefined;

src/lib/mcp/tools/managed-auth-telemetry.ts

@@ -1,19 +1,21 @@
 import { z } from "zod";
 
-const telemetryCategorySchema = z.object({
-  enabled: z.boolean().optional(),
-});
+function telemetryCategorySchema() {
+  return z.object({
+    enabled: z.boolean().optional(),
+  });
+}
 
 const telemetryCategoriesSchema = z.object({
-  captcha: telemetryCategorySchema.optional(),
-  connection: telemetryCategorySchema.optional(),
-  console: telemetryCategorySchema.optional(),
-  control: telemetryCategorySchema.optional(),
-  interaction: telemetryCategorySchema.optional(),
-  network: telemetryCategorySchema.optional(),
-  page: telemetryCategorySchema.optional(),
-  screenshot: telemetryCategorySchema.optional(),
-  system: telemetryCategorySchema.optional(),
+  captcha: telemetryCategorySchema().optional(),
+  connection: telemetryCategorySchema().optional(),
+  console: telemetryCategorySchema().optional(),
+  control: telemetryCategorySchema().optional(),
+  interaction: telemetryCategorySchema().optional(),
+  network: telemetryCategorySchema().optional(),
+  page: telemetryCategorySchema().optional(),
+  screenshot: telemetryCategorySchema().optional(),
+  system: telemetryCategorySchema().optional(),
 });
 
 export const managedAuthBrowserTelemetrySchema = z