结构化响应

您可以通过向 .respond() 方法提供模式（JSON 或 zod），来强制 LLM 返回特定的响应格式。这可以保证模型的输出符合您提供的模式。

使用 zod 模式强制执行

如果您希望模型生成满足给定模式的 JSON，建议使用 zod 来提供模式。当提供 zod 模式时，预测结果将包含一个额外的字段 parsed，其中包含已解析、验证和类型化的结果。

定义 zod 模式

import { z } from "zod";

// A zod schema for a book
const bookSchema = z.object({
  title: z.string(),
  author: z.string(),
  year: z.number().int(),
});

生成结构化响应

const result = await model.respond("Tell me about The Hobbit.",
  { structured: bookSchema },
  maxTokens: 100, // Recommended to avoid getting stuck
);

const book = result.parsed;
console.info(book);
//           ^
// Note that `book` is now correctly typed as { title: string, author: string, year: number }

使用 JSON 模式强制执行

您还可以使用 JSON 模式强制执行结构化响应。

定义 JSON 模式

// A JSON schema for a book
const schema = {
  type: "object",
  properties: {
    title: { type: "string" },
    author: { type: "string" },
    year: { type: "integer" },
  },
  required: ["title", "author", "year"],
};

生成结构化响应

const result = await model.respond("Tell me about The Hobbit.", {
  structured: {
    type: "json",
    jsonSchema: schema,
  },
  maxTokens: 100, // Recommended to avoid getting stuck
});

const book = JSON.parse(result.content);
console.info(book);