Workers Observability migrate responses to TSV's

apps/workers-observability/package.json

@@ -13,7 +13,7 @@
 	},
 	"dependencies": {
 		"@cloudflare/workers-oauth-provider": "0.0.5",
-		"@fast-csv/format": "^5.0.2",
+		"@fast-csv/format": "5.0.2",
 		"@hono/zod-validator": "0.4.3",
 		"@modelcontextprotocol/sdk": "1.10.2",
 		"@repo/mcp-common": "workspace:*",

apps/workers-observability/src/tools/observability.ts

@@ -64,7 +64,7 @@ This tool provides three primary views of your Worker data:
 				const response = await queryWorkersObservability(agent.props.accessToken, accountId, query)
 
 				if (query.view === 'calculations') {
-					let data = 'DISPLAY THE CALCULATIONS IN A GRAPHICAL WAY'
+					let data = ''
 					for (const calculation of response?.calculations || []) {
 						const alias = calculation.alias || calculation.calculation
 						const aggregates = calculation.aggregates.map((agg) => {
@@ -128,7 +128,7 @@ This tool provides three primary views of your Worker data:
 					}
 				}
 
-				if(query.view === 'events') {
+				if (query.view === 'events') {
 					const events = response?.events?.events
 					return {
 						content: [
@@ -140,16 +140,13 @@ This tool provides three primary views of your Worker data:
 					}
 				}
 
-				if(query.view === 'invocations') {
+				if (query.view === 'invocations') {
 					const invocations = Object.entries(response?.invocations || {}).map(([_, logs]) => {
 						const invocationLog = logs.find((log) => log.$metadata.type === 'cf-worker-event')
 						return invocationLog?.$metadata ?? logs[0]?.$metadata
-					});
+					})
 
-					const tsv = await writeToString(
-						invocations,
-						{ headers: true, delimiter: '\t' }
-					)
+					const tsv = await writeToString(invocations, { headers: true, delimiter: '\t' })
 					return {
 						content: [
 							{

packages/mcp-common/src/api/workers-observability.ts

@@ -8,6 +8,7 @@ import { V4Schema } from '../v4-api'
 
 import type { z } from 'zod'
 import type { zKeysRequest, zQueryRunRequest, zValuesRequest } from '../types/workers-logs-schemas'
+import { env } from 'cloudflare:workers'
 
 type QueryRunRequest = z.infer<typeof zQueryRunRequest>
 
@@ -23,6 +24,8 @@ export async function queryWorkersObservability(
 	accountId: string,
 	query: QueryRunRequest
 ): Promise<z.infer<typeof zReturnedQueryRunResult> | null> {
+	// @ts-expect-error We don't have actual env in this package
+	const environment = env.ENVIRONMENT
 	const data = await fetchCloudflareApi({
 		endpoint: '/workers/observability/telemetry/query',
 		accountId,
@@ -32,6 +35,7 @@ export async function queryWorkersObservability(
 			method: 'POST',
 			headers: {
 				'Content-Type': 'application/json',
+				'workers-observability-origin': `workers-observability-mcp-${environment}`,
 			},
 			body: JSON.stringify({ ...query, timeframe: fixTimeframe(query.timeframe) }),
 		},

packages/mcp-common/src/cloudflare-api.ts

@@ -40,6 +40,16 @@ export async function fetchCloudflareApi<T>({
 }): Promise<T> {
 	const url = `https://api.cloudflare.com/client/v4/accounts/${accountId}${endpoint}`
 
+	// @ts-expect-error We don't have actual env in this package
+	if (env.DEV_DISABLE_OAUTH) {
+		options.headers = {
+			...options.headers,
+			// @ts-expect-error We don't have actual env in this package
+			'X-Auth-Email': env.DEV_CLOUDFLARE_EMAIL,
+			// @ts-expect-error We don't have actual env in this package
+			'X-Auth-Key': env.DEV_CLOUDFLARE_API_TOKEN,
+		}
+	}
 	const response = await fetch(url, {
 		...options,
 		headers: {