feat(tools): add get_random_words tool with Fisher-Yates shuffle
This commit is contained in:
36
tools/getRandomWords.ts
Normal file
36
tools/getRandomWords.ts
Normal file
@@ -0,0 +1,36 @@
|
||||
import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js"
|
||||
import z from "zod"
|
||||
import { readDictionary } from "@/services/dictionaries"
|
||||
import { getRandomWords } from "@/services/randomWord"
|
||||
|
||||
export const addRandomWordsTool = (server: McpServer) => {
|
||||
server.registerTool(
|
||||
"get_random_words",
|
||||
{
|
||||
title: "Get Random Words",
|
||||
description: "Get a list of random words from a dictionary.",
|
||||
inputSchema: z.object({
|
||||
dictionary: z.string(),
|
||||
count: z.number().optional().default(1)
|
||||
})
|
||||
},
|
||||
async (input) => {
|
||||
try {
|
||||
const words: string[] = (await readDictionary(input.dictionary)).split("\n")
|
||||
|
||||
return { content: [{ type: "text", text: getRandomWords(words, input.count).join("\n") }] }
|
||||
} catch (error) {
|
||||
if (error instanceof Error) {
|
||||
return {
|
||||
isError: true,
|
||||
content: []
|
||||
}
|
||||
}
|
||||
return {
|
||||
isError: true,
|
||||
content: []
|
||||
}
|
||||
}
|
||||
}
|
||||
)
|
||||
}
|
||||
Reference in New Issue
Block a user