import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { z } from "zod";
import type { WikiPagesResponse } from "../types.js";
import { queryWikipedia, jsonResponse, textResponse, errorResponse } from "../api.js";
export function registerCategoriesTool(server: McpServer): void {
server.tool(
"wiki_get_categories",
"Get the categories of a Wikipedia article",
{
title: z.string().describe("Article title"),
limit: z.number().optional().default(50).describe("Maximum number of categories (default 50)"),
},
async ({ title, limit }) => {
try {
const data = await queryWikipedia<WikiPagesResponse>({
action: "query",
titles: title,
prop: "categories",
cllimit: String(limit),
});
const pages = data.query?.pages ?? {};
const page = Object.values(pages)[0];
if (page?.missing !== undefined) {
return textResponse(`Article "${title}" not found`);
}
const categories = page?.categories?.map((cat) =>
cat.title.replace(/^Category:/, "")
) ?? [];
return jsonResponse({
title: page?.title,
categories,
});
} catch (error) {
return errorResponse(error);
}
}
);
}
