feat(mcp): prompts

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
@@ -23,15 +23,21 @@ const client = createClient({
 });
 
 // Create server instance
-const server = new McpServer({
-  name: "phoenix-mcp-server",
-  version: "1.0.0",
-  capabilities: {
-    resources: {},
-    tools: {},
+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,140 @@
+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 = await Promise.all(
+      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 };
+  });
+
+  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: { type: string; text: string } }[] =
+      [];
+    if (template && template.type === "chat") {
+      messages = template.messages
+        .map((message) => {
+          if (message.role === "system") {
+            return undefined;
+          }
+          const messageContent = message.content;
+          if (typeof messageContent === "string") {
+            return {
+              role: message.role,
+              content: {
+                type: "text",
+                text: messageContent,
+              },
+            };
+          }
+          if (Array.isArray(messageContent)) {
+            if (
+              messageContent.length === 1 &&
+              messageContent[0].type === "text"
+            ) {
+              const messageContentText = messageContent[0].text;
+              return {
+                role: message.role,
+                content: {
+                  type: "text",
+                  text: messageContentText,
+                },
+              };
+            }
+          }
+          return undefined;
+        })
+        .filter((message) => message !== undefined);
+    }
+    return {
+      description: promptVersionResponse.data?.data.description,
+      messages: messages,
+    };
+  });
+};