Add get_ai_data Radar tool

Author: andre-j3sus

apps/radar/README.md

@@ -12,6 +12,7 @@ Currently available tools:
 
 | **Category**           | **Tool**                        | **Description**                                                                                                       |
 | ---------------------- | ------------------------------- | --------------------------------------------------------------------------------------------------------------------- |
+| **AI**                 | `get_ai_data`                   | Retrieves AI-related data, including traffic from AI user agents, as well as popular models and model tasks.          |
 | **Autonomous Systems** | `list_autonomous_systems`       | Lists ASes; filter by location and sort by population size                                                            |
 |                        | `get_as_details`                | Retrieves detailed info for a specific ASN                                                                            |
 | **Domains**            | `get_domains_ranking`           | Gets top or trending domains                                                                                          |
@@ -22,8 +23,8 @@ Currently available tools:
 | **HTTP**               | `get_http_data`                 | Retrieves HTTP request data, including timeseries, and breakdowns by dimensions like `deviceType`.                    |
 | **IP Addresses**       | `get_ip_details`                | Provides details about a specific IP address                                                                          |
 | **Internet Services**  | `get_internet_services_ranking` | Gets top Internet services                                                                                            |
-| **Internet Speed**     | `get_internet_speed_data`       | Retrieves summary of bandwidth, latency, jitter, and packet loss, from the previous 90 days of Cloudflare Speed Test. |
 | **Internet Quality**   | `get_internet_quality_data`     | Retrieves a summary or time series of bandwidth, latency, or DNS response time from the Radar Internet Quality Index. |
+| **Internet Speed**     | `get_internet_speed_data`       | Retrieves summary of bandwidth, latency, jitter, and packet loss, from the previous 90 days of Cloudflare Speed Test. |
 | **Layer 3 Attacks**    | `get_l3_attack_data`            | Retrieves L3 attack data, including timeseries, top attacks, and breakdowns by dimensions like `protocol`.            |
 | **Layer 7 Attacks**    | `get_l7_attack_data`            | Retrieves L7 attack data, including timeseries, top attacks, and breakdowns by dimensions like `mitigationProduct`.   |
 | **Traffic Anomalies**  | `get_traffic_anomalies`         | Lists traffic anomalies and outages; filter by AS, location, start date, and end date                                 |

apps/radar/src/tools/radar.ts

@@ -4,6 +4,7 @@ import { getCloudflareClient } from '@repo/mcp-common/src/cloudflare-api'
 import { PaginationLimitParam, PaginationOffsetParam } from '@repo/mcp-common/src/types/shared'
 
 import {
+	AiDimensionParam,
 	AsnArrayParam,
 	AsnParam,
 	AsOrderByParam,
@@ -635,20 +636,24 @@ export function registerRadarTools(agent: RadarMCP) {
 		'get_internet_quality_data',
 		'Retrieves a summary or time series of bandwidth, latency, or DNS response time percentiles from the Radar Internet Quality Index (IQI).',
 		{
+			dateRange: DateRangeArrayParam.optional(),
+			dateStart: DateStartArrayParam.optional(),
 			dateEnd: DateEndArrayParam.optional(),
 			asn: AsnArrayParam,
 			continent: ContinentArrayParam,
 			location: LocationArrayParam,
 			format: z.enum(['summary', 'timeseriesGroups']),
 			metric: InternetQualityMetricParam,
 		},
-		async ({ dateEnd, asn, location, continent, format, metric }) => {
+		async ({ dateRange, dateStart, dateEnd, asn, location, continent, format, metric }) => {
 			try {
 				const client = getCloudflareClient(agent.props.accessToken)
 				const r = await client.radar.quality.iqi[format]({
 					asn,
 					continent,
 					location,
+					dateRange,
+					dateStart,
 					dateEnd,
 					metric,
 				})
@@ -675,4 +680,51 @@ export function registerRadarTools(agent: RadarMCP) {
 			}
 		}
 	)
+
+	agent.server.tool(
+		'get_ai_data',
+		'Retrieves AI-related data, including traffic from AI user agents, as well as popular models and model tasks specifically from Cloudflare Workers AI.',
+		{
+			dateRange: DateRangeArrayParam.optional(),
+			dateStart: DateStartArrayParam.optional(),
+			dateEnd: DateEndArrayParam.optional(),
+			asn: AsnArrayParam,
+			continent: ContinentArrayParam,
+			location: LocationArrayParam,
+			dimension: AiDimensionParam,
+		},
+		async ({ dateRange, dateStart, dateEnd, asn, location, continent, dimension }) => {
+			try {
+				const client = getCloudflareClient(agent.props.accessToken)
+				const r = await resolveAndInvoke(client.radar.ai, dimension, {
+					asn,
+					continent,
+					location,
+					dateRange,
+					dateStart,
+					dateEnd,
+				})
+
+				return {
+					content: [
+						{
+							type: 'text',
+							text: JSON.stringify({
+								result: r,
+							}),
+						},
+					],
+				}
+			} catch (error) {
+				return {
+					content: [
+						{
+							type: 'text',
+							text: `Error getting AI data: ${error instanceof Error && error.message}`,
+						},
+					],
+				}
+			}
+		}
+	)
 }

apps/radar/src/types/radar.ts

@@ -293,6 +293,17 @@ export const EmailSecurityDimensionParam = z
 	])
 	.describe('Dimension indicating the type and format of Email Security data to retrieve.')
 
+export const AiDimensionParam = z
+	.enum([
+		'bots/summary/userAgent',
+		'bots/timeseriesGroups/userAgent',
+		'inference/summary/model',
+		'inference/summary/task',
+		'inference/timeseriesGroups/model',
+		'inference/timeseriesGroups/task',
+	])
+	.describe('Dimension indicating the type and format of AI data to retrieve.')
+
 export const InternetSpeedDimensionParam = z
 	.enum(['summary', 'top/locations', 'top/ases'])
 	.describe('Dimension indicating the type and format of Internet speed data to retrieve.')