Unstructured Transform MCP server installation for Firebase Genkit

npm install genkit @genkit-ai/mcp @genkit-ai/google-genai

export UNSTRUCTURED_API_KEY="<your-unstructured-api-key>"
export GEMINI_API_KEY="<your-gemini-api-key>"

import { createMcpClient } from "@genkit-ai/mcp";

const transformClient = createMcpClient({
  name: "unstructured_transform",
  mcpServer: {
    url: "https://mcp.transform.unstructured.io",
    requestInit: {
      headers: {
        Authorization: `Bearer ${process.env.UNSTRUCTURED_API_KEY}`,
      },
    },
  },
});

import { setTimeout as sleep } from "node:timers/promises";

import { googleAI } from "@genkit-ai/google-genai";
import { createMcpClient } from "@genkit-ai/mcp";
import { genkit, z } from "genkit";

const ai = genkit({ plugins: [googleAI()] });

const transformClient = createMcpClient({
  name: "unstructured_transform",
  mcpServer: {
    url: "https://mcp.transform.unstructured.io", // root URL - do not append /mcp
    requestInit: {
      headers: {
        Authorization: `Bearer ${process.env.UNSTRUCTURED_API_KEY}`,
      },
    },
  },
});

const waitSeconds = ai.defineTool(
  {
    name: "wait_seconds",
    description:
      "Pause before the next status check. Transform jobs take 30 seconds to a few " +
      "minutes; call this between check_job_status calls instead of polling " +
      "in a tight loop. Use 30 seconds unless told otherwise.",
    inputSchema: z.object({ seconds: z.number().int().min(1).max(120) }),
    outputSchema: z.object({ waitedSeconds: z.number() }),
  },
  async ({ seconds }) => {
    await sleep(seconds * 1000);
    return { waitedSeconds: seconds };
  },
);

await transformClient.ready();
const transformTools = await transformClient.getActiveTools(ai);

const { text } = await ai.generate({
  model: googleAI.model("gemini-3.1-flash-lite"),
  system:
    "You parse documents with the Unstructured Transform MCP server. Pass public " +
    "https:// file URLs straight to start_transform_job. It returns a job_id; poll with " +
    "check_job_status, calling wait_seconds(30) between checks - jobs take 30 " +
    "seconds to a few minutes. When the job completes, call get_job_results " +
    "and report the parsed content back to the user.",
  prompt:
    "Parse this PDF with Unstructured Transform: https://arxiv.org/pdf/1706.03762 " +
    "When it completes, tell me the document title and the first three section " +
    "headings from the parsed output.",
  tools: [...transformTools, waitSeconds],
  maxTurns: 20,
});

console.log(text);
await transformClient.disable();

node index.mjs

The PDF (Attention Is All You Need) has been parsed successfully.

*   Document Title: Attention Is All You Need
*   First Three Section Headings:
    1.  1 Introduction
    2.  2 Background
    3.  3 Model Architecture

const { text } = await ai.generate({
  model: googleAI.model("gemini-3.1-flash-lite"),
  system:
    "You extract structured data with the Unstructured Transform MCP server. The " +
    "extraction tools read the element JSON a parse produces, never the raw file, so " +
    "always parse first: call start_transform_job with stages " +
    '{"partition": {"strategy": "vlm"}} for an image, PowerPoint file, or PDF, poll ' +
    "check_job_status with wait_seconds(30) between checks, then call get_job_results " +
    "and keep each file's output_ref. Do not download the parsed output. If the user " +
    "gave no schema, call suggest_extraction_schema_for_file with that output_ref and " +
    "show the draft first. Then call start_extraction_job with element_json_refs set to " +
    "the output_refs and schema_to_extract set to a JSON Schema passed as a JSON string, " +
    "and poll it the same way. Extraction results come back inline, wrapped with the " +
    "source filename; report that filename with each object and keep the wrapper.",
  prompt:
    "Extract each plant's name, light needs, watering instructions, and humidity level " +
    "from https://docs.unstructured.io/img/pipelines/data-extractor/house-plant-care.png " +
    "as JSON. Draft a schema from the image itself and show it to me first.",
  tools: [...transformTools, waitSeconds],
  maxTurns: 30,
});