feat(tool): add tool for getting target metadata

Signed-off-by: TJ Hoplock <[email protected]>

Author: tjhop

pkg/mcp/promapi.go

@@ -296,6 +296,23 @@ func targetsApiCall(ctx context.Context) (string, error) {
 	return string(jsonBytes), nil
 }
 
+func targetsMetadataApiCall(ctx context.Context, matchTarget, metric, limit string) (string, error) {
+	ctx, cancel := context.WithTimeout(ctx, apiTimeout)
+	defer cancel()
+
+	tm, err := apiV1Client.TargetsMetadata(ctx, matchTarget, metric, limit)
+	if err != nil {
+		return "", fmt.Errorf("error getting target metadata from Prometheus: %w", err)
+	}
+
+	jsonBytes, err := json.Marshal(tm)
+	if err != nil {
+		return "", fmt.Errorf("error converting target metadata to JSON: %w", err)
+	}
+
+	return string(jsonBytes), nil
+}
+
 func walReplayApiCall(ctx context.Context) (string, error) {
 	ctx, cancel := context.WithTimeout(ctx, apiTimeout)
 	defer cancel()

pkg/mcp/server.go

@@ -39,6 +39,7 @@ func NewServer(logger *slog.Logger) *server.MCPServer {
 	mcpServer.AddTool(runtimeinfoTool, runtimeinfoToolHandler)
 	mcpServer.AddTool(rulesTool, rulesToolHandler)
 	mcpServer.AddTool(targetsTool, targetsToolHandler)
+	mcpServer.AddTool(targetsMetadataTool, targetsMetadataToolHandler)
 	mcpServer.AddTool(walReplayTool, walReplayToolHandler)
 
 	return mcpServer

pkg/mcp/tools.go

@@ -133,6 +133,19 @@ var (
 		mcp.WithDescription("Get overview of Prometheus target discovery"),
 	)
 
+	targetsMetadataTool = mcp.NewTool("targets_metadata",
+		mcp.WithDescription("Returns metadata about metrics currently scraped by the target "),
+		mcp.WithString("match_target",
+			mcp.Description("[Optional] Label selectors that match targets by their label sets. All targets are selected if left empty."),
+		),
+		mcp.WithString("metric",
+			mcp.Description("[Optional] A metric name to retrieve metadata for. All metric metadata is retrieved if left empty."),
+		),
+		mcp.WithString("limit",
+			mcp.Description("[Optional] Maximum number of targets to match."),
+		),
+	)
+
 	walReplayTool = mcp.NewTool("wal_replay_status",
 		mcp.WithDescription("Get current WAL replay status"),
 	)
@@ -278,7 +291,6 @@ func labelNamesToolHandler(ctx context.Context, request mcp.CallToolRequest) (*m
 	return mcp.NewToolResultText(data), err
 }
 
-// LabelValues performs a query for the values of the given label, time range and matchers.
 func labelValuesToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
 	arguments := request.Params.Arguments
 	label, ok := arguments["label"].(string)
@@ -366,6 +378,28 @@ func targetsToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.
 	return mcp.NewToolResultText(data), err
 }
 
+func targetsMetadataToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	arguments := request.Params.Arguments
+
+	matchTarget := ""
+	if argMatchTarget, ok := arguments["match_target"].(string); ok {
+		matchTarget = argMatchTarget
+	}
+
+	metric := ""
+	if argMetric, ok := arguments["metric"].(string); ok {
+		metric = argMetric
+	}
+
+	limit := ""
+	if argLimit, ok := arguments["limit"].(string); ok {
+		limit = argLimit
+	}
+
+	data, err := targetsMetadataApiCall(ctx, matchTarget, metric, limit)
+	return mcp.NewToolResultText(data), err
+}
+
 func walReplayToolHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
 	data, err := walReplayApiCall(ctx)
 	return mcp.NewToolResultText(data), err