Gemini API로 구조화된 출력 생성


Gemini는 기본적으로 구조화되지 않은 텍스트를 생성하지만 일부 애플리케이션에는 구조화된 텍스트가 필요합니다. 이러한 사용 사례의 경우 Gemini가 자동화된 처리에 적합한 구조화된 데이터 형식인 JSON으로 응답하도록 제한할 수 있습니다. enum에 지정된 옵션 중 하나로 응답하도록 모델을 제한할 수도 있습니다.

다음은 모델의 구조화된 출력이 필요한 몇 가지 사용 사례입니다.

  • 신문 기사에서 회사 정보를 가져와 회사 데이터베이스를 만듭니다.
  • 이력서에서 표준화된 정보를 가져옵니다.
  • 레시피에서 재료를 추출하고 각 재료의 식료품 웹사이트 링크를 표시합니다.

프롬프트에서 Gemini에 JSON 형식의 출력을 생성하도록 요청할 수 있지만 모델이 JSON만 생성하는 것은 아닙니다. 더 확정적인 응답을 위해 Gemini가 항상 예상 구조로 응답하도록 responseSchema 필드에 특정 JSON 스키마를 전달할 수 있습니다.

이 가이드에서는 원하는 SDK를 통해 generateContent 메서드를 사용하거나 REST API를 직접 사용하여 JSON을 생성하는 방법을 보여줍니다. 이 예에서는 텍스트 전용 입력을 보여줍니다. 하지만 Gemini는 이미지, 동영상, 오디오를 포함하는 다중 모드 요청에 대한 JSON 응답을 생성할 수도 있습니다.

시작하기 전에: 프로젝트 및 API 키 설정

Gemini API를 호출하기 전에 프로젝트를 설정하고 API 키를 구성해야 합니다.

JSON 생성

모델이 JSON을 출력하도록 구성되면 모든 프롬프트에 JSON 형식의 출력으로 응답합니다.

스키마를 제공하여 JSON 응답의 구조를 제어할 수 있습니다. 모델에 스키마를 제공하는 방법에는 두 가지가 있습니다.

  • 프롬프트의 텍스트로
  • 모델 구성을 통해 제공되는 구조화된 스키마

두 가지 접근 방식 모두 Gemini 1.5 Flash와 Gemini 1.5 Pro에서 작동합니다.

프롬프트에 스키마를 텍스트로 제공

다음 예에서는 모델이 특정 JSON 형식으로 쿠키 레시피를 반환하도록 요청합니다.

모델은 프롬프트의 텍스트에서 형식 사양을 가져오므로 사양을 나타내는 방법을 어느 정도 유연하게 선택할 수 있습니다. JSON 스키마를 나타내는 적절한 형식은 무엇이든 사용할 수 있습니다.

// Make sure to include these imports:
// import { GoogleGenerativeAI } from "@google/generative-ai";
const genAI = new GoogleGenerativeAI(process.env.API_KEY);

const model = genAI.getGenerativeModel({
  model: "gemini-1.5-flash",
});

const prompt = `List a few popular cookie recipes using this JSON schema:

Recipe = {'recipeName': string}
Return: Array<Recipe>`;

const result = await model.generateContent(prompt);
console.log(result.response.text());
controlled_generation.js

출력은 다음과 같이 표시될 수 있습니다.

[{"recipeName": "Chocolate Chip Cookies"}, {"recipeName": "Oatmeal Raisin Cookies"}, {"recipeName": "Snickerdoodles"}, {"recipeName": "Sugar Cookies"}, {"recipeName": "Peanut Butter Cookies"}]

모델 구성을 통해 스키마 제공

다음 예에서는 다음을 실행합니다.

  1. JSON으로 응답하도록 스키마를 통해 구성된 모델을 인스턴스화합니다.
  2. 모델에 쿠키 레시피를 반환하라는 메시지를 표시합니다.
// Make sure to include these imports:
// import { GoogleGenerativeAI, SchemaType } from "@google/generative-ai";
const genAI = new GoogleGenerativeAI(process.env.API_KEY);

const schema = {
  description: "List of recipes",
  type: SchemaType.ARRAY,
  items: {
    type: SchemaType.OBJECT,
    properties: {
      recipeName: {
        type: SchemaType.STRING,
        description: "Name of the recipe",
        nullable: false,
      },
    },
    required: ["recipeName"],
  },
};

const model = genAI.getGenerativeModel({
  model: "gemini-1.5-pro",
  generationConfig: {
    responseMimeType: "application/json",
    responseSchema: schema,
  },
});

const result = await model.generateContent(
  "List a few popular cookie recipes.",
);
console.log(result.response.text());
controlled_generation.js

출력은 다음과 같이 표시될 수 있습니다.

[{"recipeName": "Chocolate Chip Cookies"}, {"recipeName": "Oatmeal Raisin Cookies"}, {"recipeName": "Snickerdoodles"}, {"recipeName": "Sugar Cookies"}, {"recipeName": "Peanut Butter Cookies"}]