[WIP] Refactor everything server to be more modular and use recommended APIs.
* Adding prompts (simple, complex, with completions)
* Add prompts/simple.ts
- in addSimplePrompt()
- register a simple prompt with no arguments
* Add prompts/complex.ts
- in addComplexPrompt()
- define promptArgsSchema containing a required city arg of type string and an optional state arg of type state
- register the complex prompt with a prompt callback that combines the city and state into a prompt asking for the weather in that location
* Add prompts/completions.ts
- in addPromptWithCompletions()
- define promptArgsSchema containing department and name string fields with completion handlers
- register the completable prompt with a prompt callback that combines the inputs into a prompt asking to promote the selected name to head of the selected department
* Add prompts/index.ts
- import addSimplePrompt, addComplexPrompt, and addPromptWithCompletions
- export registerPrompts function
- in registerPrompts()
- call addSimplePrompt
- call addComplexPrompt
- call addPromptWithCompletions
* In package.json
- add prettier devDependency
- add prettier:check script
- add prettier:fix script
- in build script, copy docs folder to dist
* All other changes were prettier formatting
This commit is contained in:
parent
1c64b36c78
commit
daec74fbc1
5 changed files with 130 additions and 10 deletions
46
src/everything/prompts/completions.ts
Normal file
46
src/everything/prompts/completions.ts
Normal file
|
|
@ -0,0 +1,46 @@
|
||||||
|
import { z } from "zod";
|
||||||
|
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
||||||
|
import { completable } from "@modelcontextprotocol/sdk/server/completable.js";
|
||||||
|
|
||||||
|
export const addPromptWithCompletions = (server: McpServer) => {
|
||||||
|
const promptArgsSchema = {
|
||||||
|
department: completable(z.string(), (value) => {
|
||||||
|
return ["Engineering", "Sales", "Marketing", "Support"].filter((d) =>
|
||||||
|
d.startsWith(value)
|
||||||
|
);
|
||||||
|
}),
|
||||||
|
name: completable(z.string(), (value, context) => {
|
||||||
|
const department = context?.arguments?.["department"];
|
||||||
|
if (department === "Engineering") {
|
||||||
|
return ["Alice", "Bob", "Charlie"].filter((n) => n.startsWith(value));
|
||||||
|
} else if (department === "Sales") {
|
||||||
|
return ["David", "Eve", "Frank"].filter((n) => n.startsWith(value));
|
||||||
|
} else if (department === "Marketing") {
|
||||||
|
return ["Grace", "Henry", "Iris"].filter((n) => n.startsWith(value));
|
||||||
|
} else if (department === "Support") {
|
||||||
|
return ["John", "Kim", "Lee"].filter((n) => n.startsWith(value));
|
||||||
|
}
|
||||||
|
return [];
|
||||||
|
}),
|
||||||
|
};
|
||||||
|
|
||||||
|
server.registerPrompt(
|
||||||
|
"completable-prompt",
|
||||||
|
{
|
||||||
|
title: "Team Management",
|
||||||
|
description: "Choose a team member to lead their specific department.",
|
||||||
|
argsSchema: promptArgsSchema,
|
||||||
|
},
|
||||||
|
async ({ department, name }) => ({
|
||||||
|
messages: [
|
||||||
|
{
|
||||||
|
role: "user",
|
||||||
|
content: {
|
||||||
|
type: "text",
|
||||||
|
text: `Please promote ${name} to the head of the ${department} team.`,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
],
|
||||||
|
})
|
||||||
|
);
|
||||||
|
};
|
||||||
32
src/everything/prompts/complex.ts
Normal file
32
src/everything/prompts/complex.ts
Normal file
|
|
@ -0,0 +1,32 @@
|
||||||
|
import { z } from "zod";
|
||||||
|
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
||||||
|
|
||||||
|
export const addComplexPrompt = (server: McpServer) => {
|
||||||
|
const promptArgsSchema = {
|
||||||
|
city: z.string().describe("Name of the city"),
|
||||||
|
state: z.string().describe("Name of the state").optional(),
|
||||||
|
};
|
||||||
|
|
||||||
|
server.registerPrompt(
|
||||||
|
"complex-prompt",
|
||||||
|
{
|
||||||
|
title: "Complex Prompt",
|
||||||
|
description: "A prompt with two arguments, one required and one optional",
|
||||||
|
argsSchema: promptArgsSchema,
|
||||||
|
},
|
||||||
|
(args) => {
|
||||||
|
const location = `${args?.city}${args?.state ? `, ${args?.state}` : ""}`;
|
||||||
|
return {
|
||||||
|
messages: [
|
||||||
|
{
|
||||||
|
role: "user",
|
||||||
|
content: {
|
||||||
|
type: "text",
|
||||||
|
text: `What's weather in ${location}?`,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
],
|
||||||
|
};
|
||||||
|
}
|
||||||
|
);
|
||||||
|
};
|
||||||
14
src/everything/prompts/index.ts
Normal file
14
src/everything/prompts/index.ts
Normal file
|
|
@ -0,0 +1,14 @@
|
||||||
|
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
||||||
|
import { addSimplePrompt } from "./simple.js";
|
||||||
|
import { addComplexPrompt } from "./complex.js";
|
||||||
|
import { addPromptWithCompletions } from "./completions.js";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Register the prompts with the MCP server.
|
||||||
|
* @param server
|
||||||
|
*/
|
||||||
|
export const registerPrompts = (server: McpServer) => {
|
||||||
|
addSimplePrompt(server);
|
||||||
|
addComplexPrompt(server);
|
||||||
|
addPromptWithCompletions(server);
|
||||||
|
};
|
||||||
22
src/everything/prompts/simple.ts
Normal file
22
src/everything/prompts/simple.ts
Normal file
|
|
@ -0,0 +1,22 @@
|
||||||
|
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
||||||
|
|
||||||
|
export const addSimplePrompt = (server: McpServer) => {
|
||||||
|
server.registerPrompt(
|
||||||
|
"simple-prompt",
|
||||||
|
{
|
||||||
|
title: "Simple Prompt",
|
||||||
|
description: "A prompt with no arguments",
|
||||||
|
},
|
||||||
|
() => ({
|
||||||
|
messages: [
|
||||||
|
{
|
||||||
|
role: "user",
|
||||||
|
content: {
|
||||||
|
type: "text",
|
||||||
|
text: "This is a simple prompt without arguments.",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
],
|
||||||
|
})
|
||||||
|
);
|
||||||
|
};
|
||||||
|
|
@ -1,15 +1,17 @@
|
||||||
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
||||||
import { registerTools } from "../tools/index.js";
|
|
||||||
import { registerResources } from "../resources/index.js";
|
|
||||||
import { dirname, join } from "path";
|
import { dirname, join } from "path";
|
||||||
import { readFileSync } from "fs";
|
import { readFileSync } from "fs";
|
||||||
import { fileURLToPath } from "url";
|
import { fileURLToPath } from "url";
|
||||||
const __filename = fileURLToPath(import.meta.url);
|
import { registerTools } from "../tools/index.js";
|
||||||
const __dirname = dirname(__filename);
|
import { registerResources } from "../resources/index.js";
|
||||||
const instructions = readInstructions();
|
import { registerPrompts } from "../prompts/index.js";
|
||||||
|
|
||||||
// Create the MCP resource server
|
// Everything Server factory
|
||||||
export const createServer = () => {
|
export const createServer = () => {
|
||||||
|
// Read the server instructions
|
||||||
|
const instructions = readInstructions();
|
||||||
|
|
||||||
|
// Create the server
|
||||||
const server = new McpServer(
|
const server = new McpServer(
|
||||||
{
|
{
|
||||||
name: "mcp-servers/everything",
|
name: "mcp-servers/everything",
|
||||||
|
|
@ -35,6 +37,9 @@ export const createServer = () => {
|
||||||
// Register the resources
|
// Register the resources
|
||||||
registerResources(server);
|
registerResources(server);
|
||||||
|
|
||||||
|
// Register the prompts
|
||||||
|
registerPrompts(server);
|
||||||
|
|
||||||
return {
|
return {
|
||||||
server,
|
server,
|
||||||
cleanup: () => {},
|
cleanup: () => {},
|
||||||
|
|
@ -42,14 +47,15 @@ export const createServer = () => {
|
||||||
};
|
};
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// Read the server instructions from a file
|
||||||
function readInstructions(): string {
|
function readInstructions(): string {
|
||||||
|
const __filename = fileURLToPath(import.meta.url);
|
||||||
|
const __dirname = dirname(__filename);
|
||||||
|
const filePath = join(__dirname, "..", "docs", "server-instructions.md");
|
||||||
let instructions;
|
let instructions;
|
||||||
|
|
||||||
try {
|
try {
|
||||||
instructions = readFileSync(
|
instructions = readFileSync(filePath, "utf-8");
|
||||||
join(__dirname, "..", "docs", "server-instructions.md"),
|
|
||||||
"utf-8"
|
|
||||||
);
|
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
instructions = "Server instructions not loaded: " + e;
|
instructions = "Server instructions not loaded: " + e;
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue