Add CloudflareMCPServer wrapper to track metrics

Author: cmsparks

apps/docs-autorag/package.json

@@ -5,7 +5,7 @@
 	"scripts": {
 		"check:lint": "run-eslint-workers",
 		"check:types": "run-tsc",
-		"deploy": "wrangler deploy",
+		"deploy": "run-wrangler-deploy",
 		"dev": "wrangler dev",
 		"start": "wrangler dev",
 		"cf-typegen": "wrangler types",
@@ -16,6 +16,7 @@
 		"@hono/zod-validator": "0.4.3",
 		"@modelcontextprotocol/sdk": "1.9.0",
 		"@repo/mcp-common": "workspace:*",
+		"@repo/mcp-observability": "workspace:*",
 		"agents": "0.0.62",
 		"cloudflare": "4.2.0",
 		"hono": "4.7.6",

apps/docs-autorag/src/index.ts

@@ -1,5 +1,7 @@
-import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js'
 import { McpAgent } from 'agents/mcp'
+import { env } from 'cloudflare:workers'
+
+import { CloudflareMCPServer } from '@repo/mcp-common/src/server'
 
 import { registerDocsTools } from './tools/docs'
 
@@ -9,9 +11,9 @@ export type Props = never
 export type State = never
 
 export class CloudflareDocumentationMCP extends McpAgent<Env, State, Props> {
-	server = new McpServer({
-		name: 'Remote MCP Server with Cloudflare Documentation',
-		version: '1.0.0',
+	server = new CloudflareMCPServer(undefined, env.MCP_METRICS, {
+		name: env.MCP_SERVER_NAME,
+		version: env.MCP_SERVER_VERSION,
 	})
 
 	constructor(

apps/docs-autorag/worker-configuration.d.ts

@@ -1,11 +1,14 @@
 /* eslint-disable */
-// Generated by Wrangler by running `wrangler types` (hash: f47f9f62d3a1ddb9823a379fdafce5ce)
-// Runtime types generated with workerd@1.20250417.0 2025-03-10 nodejs_compat
+// Generated by Wrangler by running `wrangler types` (hash: 23bfeb1ecdb06845b72c16f50728901d)
+// Runtime types generated with workerd@1.20250422.0 2025-03-10 nodejs_compat
 declare namespace Cloudflare {
 	interface Env {
 		ENVIRONMENT: "development" | "staging" | "production";
 		AUTORAG_NAME: "cloudflare-docs-autorag";
+		MCP_SERVER_NAME: "PLACEHOLDER";
+		MCP_SERVER_VERSION: "PLACEHOLDER";
 		MCP_OBJECT: DurableObjectNamespace<import("./src/index").CloudflareDocumentationMCP>;
+		MCP_METRICS: AnalyticsEngineDataset;
 		AI: Ai;
 	}
 }

apps/docs-autorag/wrangler.jsonc

@@ -30,13 +30,21 @@
 	},
 	"vars": {
 		"ENVIRONMENT": "development",
-		"AUTORAG_NAME": "cloudflare-docs-autorag"
+		"AUTORAG_NAME": "cloudflare-docs-autorag",
+		"MCP_SERVER_NAME": "PLACEHOLDER",
+		"MCP_SERVER_VERSION": "PLACEHOLDER"
 	},
 	"dev": {
 		"port": 8976
 	},
 	"workers_dev": false,
 	"preview_urls": false,
+	"analytics_engine_datasets": [
+		{
+		  "binding": "MCP_METRICS",
+		  "dataset": "mcp-metrics-dev"
+		}
+	],
 	"env": {
 		"staging": {
 			"name": "mcp-cloudflare-docs-autorag-staging",

apps/sandbox-container/package.json

@@ -26,6 +26,7 @@
 		"@n8n/json-schema-to-zod": "^1.1.0",
 		"@repo/eval-tools": "workspace:*",
 		"@repo/mcp-common": "workspace:*",
+		"@repo/mcp-observability": "workspace:*",
 		"@types/node": "^22.13.10",
 		"agents": "^0.0.62",
 		"cron-schedule": "^5.0.4",

apps/sandbox-container/server/containerMcp.ts

@@ -1,4 +1,3 @@
-import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js'
 import { McpAgent } from 'agents/mcp'
 
 import { OPEN_CONTAINER_PORT } from '../shared/consts'
@@ -9,23 +8,27 @@ import { BASE_INSTRUCTIONS } from './prompts'
 import { fileToBase64, stripProtocolFromFilePath } from './utils'
 
 import type { FileList } from '../shared/schema'
-import type { Env, Props } from '.'
+import type { Props } from '.'
+import { CloudflareMCPServer } from "@repo/mcp-common/src/server"
 
-export class ContainerMcpAgent extends McpAgent<Env, Props> {
-	server = new McpServer(
-		{
-			name: 'Container MCP Agent',
-			version: '1.0.0',
-		},
-		{ instructions: BASE_INSTRUCTIONS }
-	)
+export class ContainerMcpAgent extends McpAgent<Env, {}, Props> {
+	server: CloudflareMCPServer
 
 	constructor(
 		public ctx: DurableObjectState,
 		public env: Env
 	) {
 		console.log('creating container DO')
 		super(ctx, env)
+		this.server = new CloudflareMCPServer(
+			this.props.user.id,
+			this.env.MCP_METRICS,
+			{
+				name: this.env.MCP_SERVER_NAME,
+				version: this.env.MCP_SERVER_VERSION,
+			},
+			{ instructions: BASE_INSTRUCTIONS }
+		)
 	}
 
 	async destroyContainer(): Promise<void> {

apps/sandbox-container/server/index.ts

@@ -14,13 +14,12 @@ import type { AccountSchema, UserSchema } from '@repo/mcp-common/src/cloudflare-
 
 export { ContainerManager, ContainerMcpAgent }
 
-export type Env = {
-	CONTAINER_MCP_AGENT: DurableObjectNamespace<ContainerMcpAgent>
-	CONTAINER_MANAGER: DurableObjectNamespace<ContainerManager>
-	ENVIRONMENT: 'dev' | 'prod' | 'test'
-	CLOUDFLARE_CLIENT_ID: string
-	CLOUDFLARE_CLIENT_SECRET: string
-}
+import { MetricsTracker } from "@repo/mcp-observability"
+
+const metrics = new MetricsTracker(env.MCP_METRICS, {
+	name: env.MCP_SERVER_NAME,
+	version: env.MCP_SERVER_VERSION
+})
 
 // Context from the auth process, encrypted & stored in the auth token
 // and provided to the DurableMCP as this.props
@@ -41,6 +40,7 @@ const ContainerScopes = {
 
 export default {
 	fetch: (req: Request, env: Env, ctx: ExecutionContext) => {
+		// @ts-ignore
 		if (env.ENVIRONMENT === 'test') {
 			ctx.props = {}
 			return ContainerMcpAgent.mount('/sse', { binding: 'CONTAINER_MCP_AGENT' }).fetch(
@@ -55,7 +55,7 @@ export default {
 			// @ts-ignore
 			apiHandler: ContainerMcpAgent.mount('/sse', { binding: 'CONTAINER_MCP_AGENT' }),
 			// @ts-ignore
-			defaultHandler: createAuthHandlers({ scopes: ContainerScopes }),
+			defaultHandler: createAuthHandlers({ scopes: ContainerScopes, metrics }),
 			authorizeEndpoint: '/oauth/authorize',
 			tokenEndpoint: '/token',
 			tokenExchangeCallback: (options) =>