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;
    }
  }