index.ts•5.05 kB
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js"
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js"
import { z } from "zod"
import { promises as fs } from 'fs'
import { join } from 'path'
import {
type Routine,
createRoutine,
loadRoutines,
deleteRoutine,
updateRoutine,
} from "./routines"
const server = new McpServer({
name: "mcp-routine",
version: "1.0.0",
capabilities: {
resources: {},
tools: {},
},
})
const routineFilename = process.env.FILENAME ?? join(process.cwd(), "mcp-routines.json")
server.tool(
"create-routine",
"Create a custom routine from recently run actions. Inspect recently run tools ",
{
name: z.string().describe("Name of the custom tool to be created"),
description: z.string().describe("Description of the tool, provide as much context as possible so that when user calls the tool again the LLM can follow the instruction to complete the task with different set of inputs"),
steps: z.array(
z.object({
description: z.string().describe("Description of the step to help the LLM understand the purpose of the tool call"),
tool: z.string().describe("The tool used with name, input schema used"),
params: z.object({}).passthrough().describe("Parameters used to call the tool, based on the context some of these should be swapped out with dynamic values"),
}).strict().describe("Each step of the routine is a tool call that will get executed and returned the result to feed into the next step")
),
},
async ({ name, description, steps }) => {
await createRoutine({
filename: routineFilename,
routine: { name, description, steps },
})
return {
content: [
{
type: "text",
text: `Successfully created routine "${name}" with ${steps.length} steps. Always tell user to refresh their MCP connection to see the new tool.`
}
]
}
}
)
server.tool(
"load-routines",
"Load routines",
{},
async () => {
const routines = await loadRoutines(routineFilename)
return {
content: [
{
type: "text",
text: JSON.stringify(routines, null, 2)
}
]
}
}
)
server.tool(
"update-routine",
"Update a routine by name. Factor in the existing schema and update only the portion specified by the user. Always confirm with user that they want to update it. User may supply a name that's not exactly as how it's stored. Use the load-routines tool to get the list of all routines.",
{
name: z.string().describe("Exact name of the routine to be updated."),
description: z.string().describe("Description of the routine."),
steps: z.array(
z.object({
description: z.string().describe("Description of the step to help the LLM understand the purpose of the tool call"),
tool: z.string().describe("The tool used with name, input schema used"),
params: z.object({}).passthrough().describe("Parameters used to call the tool, based on the context some of these should be swapped out with dynamic values"),
})
).describe("Steps of the routine."),
},
async ({ name, description, steps }) => {
await updateRoutine({
filename: routineFilename,
routine: { name, description, steps },
})
return {
content: [
{
type: "text",
text: `Successfully updated routine ${name}. Always tell user to refresh their MCP tools.`
}
]
}
}
)
server.tool(
"delete-routine",
"Delete a routine by name. Always confirm with user that they want to delete it. User may supply a name that's not exactly as how it's stored. Use the load-routines tool to get the list of all routines.",
{
name: z.string().describe("Exact name of the routine to be deleted.")
},
async ({ name }) => {
await deleteRoutine({
name,
filename:routineFilename
})
return {
content: [
{
type: "text",
text: `Successfully deleted routine ${name}. Always tell user to refresh their MCP tools.`
}
]
}
}
)
const routines = await loadRoutines(routineFilename)
// Load all routines as MCP tools.
for (let routine of routines) {
server.tool(
routine.name,
routine.description,
{},
() => {
return {
content: [
{
type: "text",
text: `
Perform the following steps by calling the right MCP tools as specified in the steps below.\n
The provided params are only examples. DO NOT assume they are the user inputs, request user to supply the params.\n
${JSON.stringify(routine, null, 2)}
`
}
]
}
}
)
}
async function main() {
const transport = new StdioServerTransport()
await server.connect(transport)
console.info("MCP routine server is running on stdio")
}
main().catch((error) => {
console.error("Fatal error in main():", error)
process.exit(1)
})