@perplexity-ai/mcp-server 0.2.2 → 0.13.0

package/tools/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/tools/types.ts"],"names":[],"mappings":";AAAA,sFAAsF;;AAiDtF,kDASC;AAED,sDA2BC;AAtCD,SAAgB,mBAAmB,CAAC,MAAe;IACjD,OAAO;QACL,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;aACtC;SACF;KACF,CAAC;AACJ,CAAC;AAEM,KAAK,UAAU,qBAAqB,CAAC,QAAkB;IAC5D,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;IAC3B,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACtE,IAAI,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAClC,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;SAC7C,CAAC;IACJ,CAAC;SAAM,IAAI,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACzC,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;SAC7C,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE;wBACR,uEAAuE;wBACvE,GAAG,EAAE,0BAA0B;wBAC/B,QAAQ;wBACR,IAAI,EAAE,IAAI;qBACX;iBACF;aACF;SACF,CAAC;IACJ,CAAC;AACH,CAAC"}

package/tools/types.mjs

@@ -0,0 +1,42 @@
+// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+export function asTextContentResult(result) {
+    return {
+        content: [
+            {
+                type: 'text',
+                text: JSON.stringify(result, null, 2),
+            },
+        ],
+    };
+}
+export async function asBinaryContentResult(response) {
+    const blob = await response.blob();
+    const mimeType = blob.type;
+    const data = Buffer.from(await blob.arrayBuffer()).toString('base64');
+    if (mimeType.startsWith('image/')) {
+        return {
+            content: [{ type: 'image', mimeType, data }],
+        };
+    }
+    else if (mimeType.startsWith('audio/')) {
+        return {
+            content: [{ type: 'audio', mimeType, data }],
+        };
+    }
+    else {
+        return {
+            content: [
+                {
+                    type: 'resource',
+                    resource: {
+                        // We must give a URI, even though this isn't actually an MCP resource.
+                        uri: 'resource://tool-response',
+                        mimeType,
+                        blob: data,
+                    },
+                },
+            ],
+        };
+    }
+}
+//# sourceMappingURL=types.mjs.map

package/tools/types.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.mjs","sourceRoot":"","sources":["../src/tools/types.ts"],"names":[],"mappings":"AAAA,sFAAsF;AAiDtF,MAAM,UAAU,mBAAmB,CAAC,MAAe;IACjD,OAAO;QACL,OAAO,EAAE;YACP;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;aACtC;SACF;KACF,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAAC,QAAkB;IAC5D,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IACnC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC;IAC3B,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACtE,IAAI,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAClC,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;SAC7C,CAAC;IACJ,CAAC;SAAM,IAAI,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QACzC,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;SAC7C,CAAC;IACJ,CAAC;SAAM,CAAC;QACN,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,UAAU;oBAChB,QAAQ,EAAE;wBACR,uEAAuE;wBACvE,GAAG,EAAE,0BAA0B;wBAC/B,QAAQ;wBACR,IAAI,EAAE,IAAI;qBACX;iBACF;aACF;SACF,CAAC;IACJ,CAAC;AACH,CAAC"}

package/tools.d.mts

@@ -0,0 +1,2 @@
+export * from "./tools/index.mjs";
+//# sourceMappingURL=tools.d.mts.map

package/tools.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.d.mts","sourceRoot":"","sources":["src/tools.ts"],"names":[],"mappings":""}

package/tools.d.ts

@@ -0,0 +1,2 @@
+export * from "./tools/index.js";
+//# sourceMappingURL=tools.d.ts.map

package/tools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["src/tools.ts"],"names":[],"mappings":""}

package/tools.js

@@ -0,0 +1,18 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./tools/index.js"), exports);
+//# sourceMappingURL=tools.js.map

package/tools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.js","sourceRoot":"","sources":["src/tools.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,mDAA8B"}

package/tools.mjs

@@ -0,0 +1,2 @@
+export * from "./tools/index.mjs";
+//# sourceMappingURL=tools.mjs.map

package/tools.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.mjs","sourceRoot":"","sources":["src/tools.ts"],"names":[],"mappings":""}

package/dist/index.js

@@ -1,446 +0,0 @@
-#!/usr/bin/env node
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-import { Server } from "@modelcontextprotocol/sdk/server/index.js";
-import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
-import { CallToolRequestSchema, ListToolsRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
-/**
- * Definition of the Perplexity Ask Tool.
- * This tool accepts an array of messages and returns a chat completion response
- * from the Perplexity API, with citations appended to the message if provided.
- */
-const PERPLEXITY_ASK_TOOL = {
-    name: "perplexity_ask",
-    description: "Engages in a conversation using the Sonar API. " +
-        "Accepts an array of messages (each with a role and content) " +
-        "and returns a ask completion response from the Perplexity model.",
-    inputSchema: {
-        type: "object",
-        properties: {
-            messages: {
-                type: "array",
-                items: {
-                    type: "object",
-                    properties: {
-                        role: {
-                            type: "string",
-                            description: "Role of the message (e.g., system, user, assistant)",
-                        },
-                        content: {
-                            type: "string",
-                            description: "The content of the message",
-                        },
-                    },
-                    required: ["role", "content"],
-                },
-                description: "Array of conversation messages",
-            },
-        },
-        required: ["messages"],
-    },
-};
-/**
- * Definition of the Perplexity Research Tool.
- * This tool performs deep research queries using the Perplexity API.
- */
-const PERPLEXITY_RESEARCH_TOOL = {
-    name: "perplexity_research",
-    description: "Performs deep research using the Perplexity API. " +
-        "Accepts an array of messages (each with a role and content) " +
-        "and returns a comprehensive research response with citations.",
-    inputSchema: {
-        type: "object",
-        properties: {
-            messages: {
-                type: "array",
-                items: {
-                    type: "object",
-                    properties: {
-                        role: {
-                            type: "string",
-                            description: "Role of the message (e.g., system, user, assistant)",
-                        },
-                        content: {
-                            type: "string",
-                            description: "The content of the message",
-                        },
-                    },
-                    required: ["role", "content"],
-                },
-                description: "Array of conversation messages",
-            },
-        },
-        required: ["messages"],
-    },
-};
-/**
- * Definition of the Perplexity Reason Tool.
- * This tool performs reasoning queries using the Perplexity API.
- */
-const PERPLEXITY_REASON_TOOL = {
-    name: "perplexity_reason",
-    description: "Performs reasoning tasks using the Perplexity API. " +
-        "Accepts an array of messages (each with a role and content) " +
-        "and returns a well-reasoned response using the sonar-reasoning-pro model.",
-    inputSchema: {
-        type: "object",
-        properties: {
-            messages: {
-                type: "array",
-                items: {
-                    type: "object",
-                    properties: {
-                        role: {
-                            type: "string",
-                            description: "Role of the message (e.g., system, user, assistant)",
-                        },
-                        content: {
-                            type: "string",
-                            description: "The content of the message",
-                        },
-                    },
-                    required: ["role", "content"],
-                },
-                description: "Array of conversation messages",
-            },
-        },
-        required: ["messages"],
-    },
-};
-/**
- * Definition of the Perplexity Search Tool.
- * This tool performs web search using the Perplexity Search API.
- */
-const PERPLEXITY_SEARCH_TOOL = {
-    name: "perplexity_search",
-    description: "Performs web search using the Perplexity Search API. " +
-        "Returns ranked search results with titles, URLs, snippets, and metadata.",
-    inputSchema: {
-        type: "object",
-        properties: {
-            query: {
-                type: "string",
-                description: "Search query string",
-            },
-            max_results: {
-                type: "number",
-                description: "Maximum number of results to return (1-20, default: 10)",
-                minimum: 1,
-                maximum: 20,
-            },
-            max_tokens_per_page: {
-                type: "number",
-                description: "Maximum tokens to extract per webpage (default: 1024)",
-                minimum: 256,
-                maximum: 2048,
-            },
-            country: {
-                type: "string",
-                description: "ISO 3166-1 alpha-2 country code for regional results (e.g., 'US', 'GB')",
-            },
-        },
-        required: ["query"],
-    },
-};
-// Retrieve the Perplexity API key from environment variables
-const PERPLEXITY_API_KEY = process.env.PERPLEXITY_API_KEY;
-if (!PERPLEXITY_API_KEY) {
-    console.error("Error: PERPLEXITY_API_KEY environment variable is required");
-    process.exit(1);
-}
-// Configure timeout for API requests (default: 5 minutes)
-// Can be overridden via PERPLEXITY_TIMEOUT_MS environment variable
-const TIMEOUT_MS = parseInt(process.env.PERPLEXITY_TIMEOUT_MS || "300000", 10);
-/**
- * Performs a chat completion by sending a request to the Perplexity API.
- * Appends citations to the returned message content if they exist.
- *
- * @param {Array<{ role: string; content: string }>} messages - An array of message objects.
- * @param {string} model - The model to use for the completion.
- * @returns {Promise<string>} The chat completion result with appended citations.
- * @throws Will throw an error if the API request fails.
- */
-function performChatCompletion(messages_1) {
-    return __awaiter(this, arguments, void 0, function* (messages, model = "sonar-pro") {
-        // Construct the API endpoint URL and request body
-        const url = new URL("https://api.perplexity.ai/chat/completions");
-        const body = {
-            model: model, // Model identifier passed as parameter
-            messages: messages,
-            // Additional parameters can be added here if required (e.g., max_tokens, temperature, etc.)
-            // See the Sonar API documentation for more details: 
-            // https://docs.perplexity.ai/api-reference/chat-completions
-        };
-        const controller = new AbortController();
-        const timeoutId = setTimeout(() => controller.abort(), TIMEOUT_MS);
-        let response;
-        try {
-            response = yield fetch(url.toString(), {
-                method: "POST",
-                headers: {
-                    "Content-Type": "application/json",
-                    "Authorization": `Bearer ${PERPLEXITY_API_KEY}`,
-                },
-                body: JSON.stringify(body),
-                signal: controller.signal,
-            });
-            clearTimeout(timeoutId);
-        }
-        catch (error) {
-            clearTimeout(timeoutId);
-            if (error instanceof Error && error.name === "AbortError") {
-                throw new Error(`Request timeout: Perplexity API did not respond within ${TIMEOUT_MS}ms. Consider increasing PERPLEXITY_TIMEOUT_MS.`);
-            }
-            throw new Error(`Network error while calling Perplexity API: ${error}`);
-        }
-        // Check for non-successful HTTP status
-        if (!response.ok) {
-            let errorText;
-            try {
-                errorText = yield response.text();
-            }
-            catch (parseError) {
-                errorText = "Unable to parse error response";
-            }
-            throw new Error(`Perplexity API error: ${response.status} ${response.statusText}\n${errorText}`);
-        }
-        // Attempt to parse the JSON response from the API
-        let data;
-        try {
-            data = yield response.json();
-        }
-        catch (jsonError) {
-            throw new Error(`Failed to parse JSON response from Perplexity API: ${jsonError}`);
-        }
-        // Directly retrieve the main message content from the response 
-        let messageContent = data.choices[0].message.content;
-        // If citations are provided, append them to the message content
-        if (data.citations && Array.isArray(data.citations) && data.citations.length > 0) {
-            messageContent += "\n\nCitations:\n";
-            data.citations.forEach((citation, index) => {
-                messageContent += `[${index + 1}] ${citation}\n`;
-            });
-        }
-        return messageContent;
-    });
-}
-/**
- * Formats search results from the Perplexity Search API into a readable string.
- *
- * @param {any} data - The search response data from the API.
- * @returns {string} Formatted search results.
- */
-function formatSearchResults(data) {
-    if (!data.results || !Array.isArray(data.results)) {
-        return "No search results found.";
-    }
-    let formattedResults = `Found ${data.results.length} search results:\n\n`;
-    data.results.forEach((result, index) => {
-        formattedResults += `${index + 1}. **${result.title}**\n`;
-        formattedResults += `   URL: ${result.url}\n`;
-        if (result.snippet) {
-            formattedResults += `   ${result.snippet}\n`;
-        }
-        if (result.date) {
-            formattedResults += `   Date: ${result.date}\n`;
-        }
-        formattedResults += `\n`;
-    });
-    return formattedResults;
-}
-/**
- * Performs a web search using the Perplexity Search API.
- *
- * @param {string} query - The search query string.
- * @param {number} maxResults - Maximum number of results to return (1-20).
- * @param {number} maxTokensPerPage - Maximum tokens to extract per webpage.
- * @param {string} country - Optional ISO country code for regional results.
- * @returns {Promise<string>} The formatted search results.
- * @throws Will throw an error if the API request fails.
- */
-function performSearch(query_1) {
-    return __awaiter(this, arguments, void 0, function* (query, maxResults = 10, maxTokensPerPage = 1024, country) {
-        const url = new URL("https://api.perplexity.ai/search");
-        const body = {
-            query: query,
-            max_results: maxResults,
-            max_tokens_per_page: maxTokensPerPage,
-        };
-        if (country) {
-            body.country = country;
-        }
-        const controller = new AbortController();
-        const timeoutId = setTimeout(() => controller.abort(), TIMEOUT_MS);
-        let response;
-        try {
-            response = yield fetch(url.toString(), {
-                method: "POST",
-                headers: {
-                    "Content-Type": "application/json",
-                    "Authorization": `Bearer ${PERPLEXITY_API_KEY}`,
-                },
-                body: JSON.stringify(body),
-                signal: controller.signal,
-            });
-            clearTimeout(timeoutId);
-        }
-        catch (error) {
-            clearTimeout(timeoutId);
-            if (error instanceof Error && error.name === "AbortError") {
-                throw new Error(`Request timeout: Perplexity Search API did not respond within ${TIMEOUT_MS}ms. Consider increasing PERPLEXITY_TIMEOUT_MS.`);
-            }
-            throw new Error(`Network error while calling Perplexity Search API: ${error}`);
-        }
-        // Check for non-successful HTTP status
-        if (!response.ok) {
-            let errorText;
-            try {
-                errorText = yield response.text();
-            }
-            catch (parseError) {
-                errorText = "Unable to parse error response";
-            }
-            throw new Error(`Perplexity Search API error: ${response.status} ${response.statusText}\n${errorText}`);
-        }
-        let data;
-        try {
-            data = yield response.json();
-        }
-        catch (jsonError) {
-            throw new Error(`Failed to parse JSON response from Perplexity Search API: ${jsonError}`);
-        }
-        return formatSearchResults(data);
-    });
-}
-// Initialize the server with tool metadata and capabilities
-const server = new Server({
-    name: "example-servers/perplexity-ask",
-    version: "0.1.0",
-}, {
-    capabilities: {
-        tools: {},
-    },
-});
-/**
- * Registers a handler for listing available tools.
- * When the client requests a list of tools, this handler returns all available Perplexity tools.
- */
-server.setRequestHandler(ListToolsRequestSchema, () => __awaiter(void 0, void 0, void 0, function* () {
-    return ({
-        tools: [PERPLEXITY_ASK_TOOL, PERPLEXITY_RESEARCH_TOOL, PERPLEXITY_REASON_TOOL, PERPLEXITY_SEARCH_TOOL],
-    });
-}));
-/**
- * Registers a handler for calling a specific tool.
- * Processes requests by validating input and invoking the appropriate tool.
- *
- * @param {object} request - The incoming tool call request.
- * @returns {Promise<object>} The response containing the tool's result or an error.
- */
-server.setRequestHandler(CallToolRequestSchema, (request) => __awaiter(void 0, void 0, void 0, function* () {
-    try {
-        const { name, arguments: args } = request.params;
-        if (!args) {
-            throw new Error("No arguments provided");
-        }
-        switch (name) {
-            case "perplexity_ask": {
-                if (!Array.isArray(args.messages)) {
-                    throw new Error("Invalid arguments for perplexity_ask: 'messages' must be an array");
-                }
-                // Invoke the chat completion function with the provided messages
-                const messages = args.messages;
-                const result = yield performChatCompletion(messages, "sonar-pro");
-                return {
-                    content: [{ type: "text", text: result }],
-                    isError: false,
-                };
-            }
-            case "perplexity_research": {
-                if (!Array.isArray(args.messages)) {
-                    throw new Error("Invalid arguments for perplexity_research: 'messages' must be an array");
-                }
-                // Invoke the chat completion function with the provided messages using the deep research model
-                const messages = args.messages;
-                const result = yield performChatCompletion(messages, "sonar-deep-research");
-                return {
-                    content: [{ type: "text", text: result }],
-                    isError: false,
-                };
-            }
-            case "perplexity_reason": {
-                if (!Array.isArray(args.messages)) {
-                    throw new Error("Invalid arguments for perplexity_reason: 'messages' must be an array");
-                }
-                // Invoke the chat completion function with the provided messages using the reasoning model
-                const messages = args.messages;
-                const result = yield performChatCompletion(messages, "sonar-reasoning-pro");
-                return {
-                    content: [{ type: "text", text: result }],
-                    isError: false,
-                };
-            }
-            case "perplexity_search": {
-                if (typeof args.query !== "string") {
-                    throw new Error("Invalid arguments for perplexity_search: 'query' must be a string");
-                }
-                const { query, max_results, max_tokens_per_page, country } = args;
-                const maxResults = typeof max_results === "number" ? max_results : undefined;
-                const maxTokensPerPage = typeof max_tokens_per_page === "number" ? max_tokens_per_page : undefined;
-                const countryCode = typeof country === "string" ? country : undefined;
-                const result = yield performSearch(query, maxResults, maxTokensPerPage, countryCode);
-                return {
-                    content: [{ type: "text", text: result }],
-                    isError: false,
-                };
-            }
-            default:
-                // Respond with an error if an unknown tool is requested
-                return {
-                    content: [{ type: "text", text: `Unknown tool: ${name}` }],
-                    isError: true,
-                };
-        }
-    }
-    catch (error) {
-        // Return error details in the response
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Error: ${error instanceof Error ? error.message : String(error)}`,
-                },
-            ],
-            isError: true,
-        };
-    }
-}));
-/**
- * Initializes and runs the server using standard I/O for communication.
- * Logs an error and exits if the server fails to start.
- */
-function runServer() {
-    return __awaiter(this, void 0, void 0, function* () {
-        try {
-            const transport = new StdioServerTransport();
-            yield server.connect(transport);
-            console.error("Perplexity MCP Server running on stdio with Ask, Research, Reason, and Search tools");
-        }
-        catch (error) {
-            console.error("Fatal error running server:", error);
-            process.exit(1);
-        }
-    });
-}
-// Start the server and catch any startup errors
-runServer().catch((error) => {
-    console.error("Fatal error running server:", error);
-    process.exit(1);
-});