instructor gemini

date: 2024-11-26 excerpt: instructor geiminiの使い方

instructor geminiの使い方

概要

geminiでもinstructorを使用することができる
temperatureなどのパラメータを指定する際はclientの初期化時に指定する

インストール

$ pip install "instructor[google-genai]"

使用例

import instructor
from pydantic import BaseModel, Field


class Blog(BaseModel):
    title: str = Field(..., description="ブログのタイトル")
    body: str = Field(..., description="ブログ本文")


client = instructor.from_provider(
    "google/gemini-2.5-pro",
    mode=instructor.Mode.GENAI_STRUCTURED_OUTPUTS,
)

# As a parameter
response = client.chat.completions.create(
    system="あなたはブログ記事を構造化するアシスタント",
    messages=[{"role": "user", "content": "instructorとgeminiの使い方についてブログを書きたい"}],
    response_model=Blog,
    generation_config={
        "temperature": 0.0,
        #"max_output_tokens": 256,
        #"top_p": 1,
        #"top_k": 32,
    },
)

print(response)

参考

Instructor, The Most Popular Library for Simple Structured Outputs