HF Dataset MCP

Instructions

Implementation Reference

• src/tools/get-rows.ts:43-60 (handler)

  The handler function for the get_rows tool, which executes the request to fetch rows.

  async ({ dataset, config, split, offset, length }) => {
    const data = await fetchDatasetViewer<RowsResponse>("/rows", {
      dataset,
      config,
      split,
      offset: offset ?? 0,
      length: length ?? 100,
    });
  
    return {
      content: [
        {
          type: "text" as const,
          text: JSON.stringify(data, null, 2),
        },
      ],
    };
  }

• src/tools/get-rows.ts:25-42 (schema)

  Input validation schema for the get_rows tool.

  {
    dataset: z.string().describe("Dataset ID (e.g., 'stanfordnlp/imdb')"),
    config: z.string().describe("Configuration name (from list_splits)"),
    split: z.string().describe("Split name (train, test, validation)"),
    offset: z
      .number()
      .int()
      .min(0)
      .optional()
      .describe("Row index to start from (default: 0)"),
    length: z
      .number()
      .int()
      .min(1)
      .max(100)
      .optional()
      .describe("Number of rows to fetch (default: 100, max: 100)"),
  },

• src/tools/get-rows.ts:21-62 (registration)

  Registration function for the get_rows tool.

  export function registerGetRows(server: McpServer) {
    server.tool(
      "get_rows",
      "Fetch a slice of rows from a dataset split",
      {
        dataset: z.string().describe("Dataset ID (e.g., 'stanfordnlp/imdb')"),
        config: z.string().describe("Configuration name (from list_splits)"),
        split: z.string().describe("Split name (train, test, validation)"),
        offset: z
          .number()
          .int()
          .min(0)
          .optional()
          .describe("Row index to start from (default: 0)"),
        length: z
          .number()
          .int()
          .min(1)
          .max(100)
          .optional()
          .describe("Number of rows to fetch (default: 100, max: 100)"),
      },
      async ({ dataset, config, split, offset, length }) => {
        const data = await fetchDatasetViewer<RowsResponse>("/rows", {
          dataset,
          config,
          split,
          offset: offset ?? 0,
          length: length ?? 100,
        });
  
        return {
          content: [
            {
              type: "text" as const,
              text: JSON.stringify(data, null, 2),
            },
          ],
        };
      }
    );
  }