4.3 Pydantic連携

PydanticモデルでGuard作成

Pydanticは、Pythonの型ヒントを活用したデータ検証ライブラリ。Guardrails AIはPydanticモデルから直接Guardを作成できる。

from pydantic import BaseModel, Field
from typing import Annotated, Optional
from guardrails import Guard
from guardrails.validators import ValidLength, ValidRange

class UserProfile(BaseModel):
    name: Annotated[str, ValidLength(min=1, max=50)] = Field(
        description="ユーザー名"
    )
    age: Annotated[int, ValidRange(min=0, max=150)] = Field(
        description="年齢"
    )
    email: str = Field(description="メールアドレス")
    bio: Optional[str] = Field(default=None, description="自己紹介")

guard = Guard.from_pydantic(UserProfile)

Pydanticの型とGuardrails

ネストしたモデル

class Address(BaseModel):
    city: str
    postal_code: str

class User(BaseModel):
    name: str
    address: Address  # ネストしたモデル
    tags: list[str]   # 配列

guard = Guard.from_pydantic(User)

Field descriptionの重要性