Powerdrill MCP Server

Implementation Reference

• src/index.ts:133-184 (registration)

  Registration of the mcp_powerdrill_get_dataset_overview tool, including inline schema and handler function.
  server.tool( 'mcp_powerdrill_get_dataset_overview', { datasetId: z.string().describe('The ID of the dataset to get overview information for') }, async (args, extra) => { try { const { datasetId } = args; // Initialize Powerdrill client const client = new (await import('./utils/powerdrillClient.js')).PowerdrillClient(); // Fetch dataset overview const response = await client.getDatasetOverview(datasetId); // Check if response is valid if (response.code !== 0 || !response.data) { throw new Error(`Invalid API response: ${JSON.stringify(response)}`); } // Format the response as MCP content return { content: [ { type: "text", text: JSON.stringify({ id: response.data.id, name: response.data.name, description: response.data.description, summary: response.data.summary, exploration_questions: response.data.exploration_questions, keywords: response.data.keywords }) } ] }; } catch (error: any) { console.error(`Error getting dataset overview: ${error.message}`); // Return error response return { content: [ { type: "text", text: `Error getting dataset overview: ${error.message}` } ], isError: true }; } } );
• src/index.ts:135-137 (schema)

  Input schema for the tool using Zod validation.
  { datasetId: z.string().describe('The ID of the dataset to get overview information for') },
• src/index.ts:138-183 (handler)

  The handler function that processes the tool call: extracts datasetId, initializes PowerdrillClient, fetches overview, formats response.
  async (args, extra) => { try { const { datasetId } = args; // Initialize Powerdrill client const client = new (await import('./utils/powerdrillClient.js')).PowerdrillClient(); // Fetch dataset overview const response = await client.getDatasetOverview(datasetId); // Check if response is valid if (response.code !== 0 || !response.data) { throw new Error(`Invalid API response: ${JSON.stringify(response)}`); } // Format the response as MCP content return { content: [ { type: "text", text: JSON.stringify({ id: response.data.id, name: response.data.name, description: response.data.description, summary: response.data.summary, exploration_questions: response.data.exploration_questions, keywords: response.data.keywords }) } ] }; } catch (error: any) { console.error(`Error getting dataset overview: ${error.message}`); // Return error response return { content: [ { type: "text", text: `Error getting dataset overview: ${error.message}` } ], isError: true }; } }
• src/utils/powerdrillClient.ts:136-144 (helper)

  Helper method in PowerdrillClient class that makes the API call to retrieve dataset overview.
  async getDatasetOverview(datasetId: string) { try { const response = await this.client.get(`/datasets/${datasetId}/overview?user_id=${this.config.userId}`); return response.data; } catch (error: any) { console.error(`Error retrieving dataset overview for ${datasetId}:`, error.message); throw error; } }