Structured output

import { z } from "zod";

const schema = z.object({
  totalFiles: z.number().describe("Number of files found"),
  fileTypes: z.array(z.string()).describe("File extensions in the project"),
  summary: z.string().describe("Short explanation of the project contents"),
});

const result = await agent.run({
  prompt: "Analyze the current directory.",
  schema,
});

console.log(result.totalFiles);

import { z } from "zod";
import { MCPAgent } from "@mcp-use/agent";

const ProjectSummary = z.object({
  packageName: z.string().nullable(),
  scripts: z.array(z.string()),
  hasTests: z.boolean(),
  notes: z.string(),
});

type ProjectSummary = z.infer<typeof ProjectSummary>;

async function main() {
  const agent = new MCPAgent({
    llm: "openai/gpt-4o",
    mcpServers: {
      filesystem: {
        command: "npx",
        args: ["-y", "@modelcontextprotocol/server-filesystem", "./"],
      },
    },
  });

  try {
    const summary: ProjectSummary = await agent.run({
      prompt: "Inspect package.json and summarize the project scripts.",
      schema: ProjectSummary,
      maxSteps: 8,
    });

    console.log(summary);
  } finally {
    await agent.close();
  }
}

main().catch(console.error);

for await (const event of agent.streamEvents({
  prompt: "Inspect the project and return a summary.",
  schema: ProjectSummary,
})) {
  if (event.event === "on_structured_output_progress") {
    console.log(event.data?.message);
  }

  if (event.event === "on_structured_output") {
    const summary = ProjectSummary.parse(event.data?.output);
    console.log(summary);
  }

  if (event.event === "on_structured_output_error") {
    throw new Error(String(event.data?.error));
  }
}