Apify Model Context Protocol Server

import { z } from 'zod'; import zodToJsonSchema from 'zod-to-json-schema'; import { HelperTools } from '../const.js'; import type { InternalTool, ToolEntry } from '../types.js'; import { ajv } from '../utils/ajv.js'; import { searchApifyDocsCached } from '../utils/apify-docs.js'; const searchApifyDocsToolArgsSchema = z.object({ query: z.string() .min(1) .describe( `Algolia full-text search query to find relevant documentation pages. Use only keywords, do not use full sentences or questions. For example, "standby actor" will return documentation pages that contain the words "standby" and "actor".`, ), limit: z.number() .optional() .default(5) .describe(`Maximum number of search results to return. Defaults to 5. You can increase this limit if you need more results, but keep in mind that the search results are limited to the most relevant pages.`), offset: z.number() .optional() .default(0) .describe(`Offset for the search results. Defaults to 0. Use this to paginate through the search results. For example, if you want to get the next 5 results, set the offset to 5 and limit to 5.`), }); export const searchApifyDocsTool: ToolEntry = { type: 'internal', tool: { name: HelperTools.DOCS_SEARCH, description: `Apify documentation search tool. This tool allows you to search the Apify documentation using Algolia's full-text search. You can use it to find relevant documentation pages based on keywords. The results will include the URL of the documentation page, a fragment identifier (if available), and a limited piece of content that matches the search query. You can then fetch the full content of the document using the ${HelperTools.DOCS_FETCH} tool by providing the URL. Use this tool when a user asks for help with Apify documentation or when you need to find relevant documentation pages based on keywords. For example, when a user wants to build an Apify Actor, you can search "How to build Actors" to find relevant guidance.`, args: searchApifyDocsToolArgsSchema, inputSchema: zodToJsonSchema(searchApifyDocsToolArgsSchema), ajvValidate: ajv.compile(zodToJsonSchema(searchApifyDocsToolArgsSchema)), call: async (toolArgs) => { const { args } = toolArgs; const parsed = searchApifyDocsToolArgsSchema.parse(args); const query = parsed.query.trim(); const resultsRaw = await searchApifyDocsCached(query); const results = resultsRaw.slice(parsed.offset, parsed.offset + parsed.limit); if (results.length === 0) { return { content: [{ type: 'text', text: `No results found for the query "${query}" with limit ${parsed.limit} and offset ${parsed.offset}. Try a different query or adjust the limit and offset.`, }], }; } const textContent = `You can use the Apify docs fetch tool to retrieve the full content of a document by its URL. The document fragment refers to the section of the content containing the relevant part for the search result item. Search results for "${query}": ${results.map((result) => `- Document URL: ${result.url}${result.fragment ? `\n Document fragment: ${result.fragment}` : ''} Content: ${result.content}`).join('\n\n')}`; return { content: [{ type: 'text', text: textContent, }], }; }, } as InternalTool, };