feat(mcp): prompts

Author: mikeldking

js/packages/phoenix-mcp/package.json

@@ -23,7 +23,7 @@
   "license": "Apache-2.0",
   "dependencies": {
     "@arizeai/phoenix-client": "workspace:*",
-    "@modelcontextprotocol/sdk": "^1.8.0",
+    "@modelcontextprotocol/sdk": "^1.9.0",
     "glob": "^11.0.1",
     "minimist": "^1.2.8",
     "zod": "^3.24.2"

js/packages/phoenix-mcp/src/index.ts

@@ -8,7 +8,7 @@ import { initializeDatasetTools } from "./datasetTools.js";
 import { initializeExperimentTools } from "./experimentTools.js";
 import { initializePromptTools } from "./promptTools.js";
 import { initializeReadmeResources } from "./readmeResource.js";
-
+import { initializePrompts } from "./prompts.js";
 const argv = minimist(process.argv.slice(2));
 
 // Initialize Phoenix client
@@ -27,11 +27,11 @@ const server = new McpServer({
   name: "phoenix-mcp-server",
   version: "1.0.0",
   capabilities: {
-    resources: {},
-    tools: {},
+    prompts: {},
   },
 });
 
+initializePrompts({ client, server });
 initializePromptTools({ client, server });
 initializeExperimentTools({ client, server });
 initializeDatasetTools({ client, server });

js/packages/phoenix-mcp/src/prompts.ts

@@ -0,0 +1,110 @@
+import { PhoenixClient } from "@arizeai/phoenix-client";
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import {
+  GetPromptRequestSchema,
+  ListPromptsRequestSchema,
+} from "@modelcontextprotocol/sdk/types.js";
+
+const parseVariablesMustache = (template: string) => {
+  const regex = /\{\{([^{}]+)\}\}/g;
+  const variables = [];
+  let match;
+  while ((match = regex.exec(template))) {
+    variables.push(match[1]);
+  }
+  return variables;
+};
+
+const parseArgumentsFString = (template: string) => {
+  const regex = /\{([^{}]+)\}/g;
+  const variables = [];
+  let match;
+  while ((match = regex.exec(template))) {
+    variables.push(match[1]);
+  }
+  return variables;
+};
+
+export const initializePrompts = async ({
+  client,
+  server,
+}: {
+  client: PhoenixClient;
+  server: McpServer;
+}) => {
+  server.server.setRequestHandler(ListPromptsRequestSchema, async () => {
+    const promptsResponse = await client.GET("/v1/prompts");
+
+    if (!promptsResponse.data) {
+      return {
+        prompts: [],
+      };
+    }
+
+    // Get all the prompts and parse out the arguments
+    const prompts = promptsResponse.data.data.map(async (prompt) => {
+      const promptVersionResponse = await client.GET(
+        "/v1/prompts/{prompt_identifier}/latest",
+        {
+          params: {
+            path: {
+              prompt_identifier: prompt.name,
+            },
+          },
+        }
+      );
+
+      const args: string[] = [];
+      const template = promptVersionResponse.data?.data.template;
+      const format = promptVersionResponse.data?.data.template_format;
+      const parser =
+        format === "F_STRING" ? parseArgumentsFString : parseVariablesMustache;
+      if (template && template.type === "chat") {
+        template.messages.forEach((message) => {
+          const content = message.content;
+          if (typeof content === "string") {
+            args.push(...parser(content));
+          }
+        });
+      }
+
+      return {
+        name: prompt.name,
+        description: prompt.description,
+        arguments: args,
+      };
+    });
+
+    return { prompts: prompts };
+  });
+
+  server.server.setRequestHandler(GetPromptRequestSchema, async (request) => {
+    // const args = request.params.arguments || {};
+    // Get the latest version of the prompt
+    const promptVersionResponse = await client.GET(
+      "/v1/prompts/{prompt_identifier}/latest",
+      {
+        params: {
+          path: {
+            prompt_identifier: request.params.name,
+          },
+        },
+      }
+    );
+
+    const template = promptVersionResponse.data?.data.template;
+    let messages: { role: string; content: string }[] = [];
+    if (template && template.type === "chat") {
+      messages = template.messages.map((message) => {
+        return {
+          role: message.role as string,
+          content: message.content as string,
+        };
+      });
+    }
+    return {
+      description: promptVersionResponse.data?.data.description,
+      messages: messages,
+    };
+  });
+};