Powerdrill MCP Server

Implementation Reference

• src/index.ts:198-268 (handler)

  MCP handler function that parses arguments, calls PowerdrillClient.createJob, processes the response blocks (simplifying TABLE/IMAGE), formats as MCP content, handles errors

  async (args, extra) => {
    try {
      // Initialize Powerdrill client
      const client = new (await import('./utils/powerdrillClient.js')).PowerdrillClient();
  
      // Create job parameters
      const jobParams = {
        question: args.question,
        dataset_id: args.dataset_id,
        datasource_ids: args.datasource_ids,
        session_id: args.session_id,
        stream: args.stream,
        output_language: args.output_language,
        job_mode: args.job_mode
      };
  
      // Create job
      const response = await client.createJob(jobParams);
  
      // Check if response is valid
      if (response.code !== 0 || !response.data) {
        throw new Error(`Invalid API response: ${JSON.stringify(response)}`);
      }
  
      // Process blocks for a cleaner response
      const processedBlocks = response.data.blocks.map((block: any) => {
        // For TABLE and IMAGE types, just include the URL and name
        if (block.type === 'TABLE' || block.type === 'IMAGE') {
          return {
            type: block.type,
            url: block.content.url,
            name: block.content.name,
            expires_at: block.content.expires_at
          };
        }
  
        // For other types, keep the original content
        return {
          type: block.type,
          content: block.content,
          stage: block.stage
        };
      });
  
      // Format the response as MCP content
      return {
        content: [
          {
            type: "text",
            text: JSON.stringify({
              job_id: response.data.job_id,
              blocks: processedBlocks
            }, null, 2)
          }
        ]
      };
    } catch (error: any) {
      console.error(`Error creating job: ${error.message}`);
  
      // Return error response
      return {
        content: [
          {
            type: "text",
            text: `Error creating job: ${error.message}`
          }
        ],
        isError: true
      };
    }
  }

• src/index.ts:189-197 (schema)

  Zod input schema defining parameters for the mcp_powerdrill_create_job tool

  {
    question: z.string().describe('The natural language question or prompt to analyze the data'),
    dataset_id: z.string().describe('The ID of the dataset to analyze'),
    datasource_ids: z.array(z.string()).optional().describe('Optional array of specific data source IDs within the dataset to analyze'),
    session_id: z.string().describe('Session ID to group related jobs'),
    stream: z.boolean().optional().default(false).describe('Whether to stream the results (default: false)'),
    output_language: z.string().optional().default('AUTO').describe('The language for the output (default: AUTO)'),
    job_mode: z.string().optional().default('AUTO').describe('The job mode (default: AUTO)')
  },

• src/index.ts:187-269 (registration)

  Registration of the mcp_powerdrill_create_job tool on the MCP server using server.tool

  server.tool(
    'mcp_powerdrill_create_job',
    {
      question: z.string().describe('The natural language question or prompt to analyze the data'),
      dataset_id: z.string().describe('The ID of the dataset to analyze'),
      datasource_ids: z.array(z.string()).optional().describe('Optional array of specific data source IDs within the dataset to analyze'),
      session_id: z.string().describe('Session ID to group related jobs'),
      stream: z.boolean().optional().default(false).describe('Whether to stream the results (default: false)'),
      output_language: z.string().optional().default('AUTO').describe('The language for the output (default: AUTO)'),
      job_mode: z.string().optional().default('AUTO').describe('The job mode (default: AUTO)')
    },
    async (args, extra) => {
      try {
        // Initialize Powerdrill client
        const client = new (await import('./utils/powerdrillClient.js')).PowerdrillClient();
  
        // Create job parameters
        const jobParams = {
          question: args.question,
          dataset_id: args.dataset_id,
          datasource_ids: args.datasource_ids,
          session_id: args.session_id,
          stream: args.stream,
          output_language: args.output_language,
          job_mode: args.job_mode
        };
  
        // Create job
        const response = await client.createJob(jobParams);
  
        // Check if response is valid
        if (response.code !== 0 || !response.data) {
          throw new Error(`Invalid API response: ${JSON.stringify(response)}`);
        }
  
        // Process blocks for a cleaner response
        const processedBlocks = response.data.blocks.map((block: any) => {
          // For TABLE and IMAGE types, just include the URL and name
          if (block.type === 'TABLE' || block.type === 'IMAGE') {
            return {
              type: block.type,
              url: block.content.url,
              name: block.content.name,
              expires_at: block.content.expires_at
            };
          }
  
          // For other types, keep the original content
          return {
            type: block.type,
            content: block.content,
            stage: block.stage
          };
        });
  
        // Format the response as MCP content
        return {
          content: [
            {
              type: "text",
              text: JSON.stringify({
                job_id: response.data.job_id,
                blocks: processedBlocks
              }, null, 2)
            }
          ]
        };
      } catch (error: any) {
        console.error(`Error creating job: ${error.message}`);
  
        // Return error response
        return {
          content: [
            {
              type: "text",
              text: `Error creating job: ${error.message}`
            }
          ],
          isError: true
        };
      }
    }
  );

• src/utils/powerdrillClient.ts:146-165 (helper)

  PowerdrillClient.createJob helper method that performs the actual API POST request to /jobs endpoint

  /**
   * Create a job to analyze data with natural language questions
   * @param params Parameters for creating a job
   * @returns Promise with the job result
   */
  async createJob(params: CreateJobParams) {
    try {
      // Include user_id in the request body
      const requestBody = {
        ...params,
        user_id: this.config.userId
      };
  
      const response = await this.client.post('/jobs', requestBody);
      return response.data;
    } catch (error: any) {
      console.error('Error creating job:', error.message);
      throw error;
    }
  }

• src/utils/powerdrillClient.ts:21-29 (schema)

  TypeScript interface defining parameters for createJob in PowerdrillClient

  export interface CreateJobParams {
    session_id: string;
    question: string;
    dataset_id: string;
    datasource_ids?: string[];
    stream?: boolean;
    output_language?: string;
    job_mode?: string;
  }