Pydanticバリデーション

from typing import Annotated
from pydantic import Field
from pydantic_ai import Agent
 
agent = Agent('google-gla:gemini-3-flash-preview', instructions='答えは日本語で。')
 
@agent.tool_plain
def book_seat(
    seat_no: Annotated[int, Field(ge=1, le=300, description='座席番号 (1〜300)')],
    name: Annotated[str, Field(min_length=1, max_length=50, description='予約者氏名')],
) -> str:
    """座席を予約します。"""
    return f'予約完了: 座席 {seat_no} / {name} さん'
 
# 「399 番の席を予約」と頼んでも、Pydantic が範囲外を弾き、LLM は範囲内に修正してくる
print(agent.run_sync('399 番に田中で予約して').output)

from typing import Literal
from pydantic_ai import Agent
 
agent_w = Agent('google-gla:gemini-3-flash-preview', instructions='答えは日本語で。')
 
@agent_w.tool_plain
def log_weather(
    city: str,
    condition: Literal['sunny', 'cloudy', 'rainy', 'snowy'],
    temp_c: float,
) -> str:
    """天気をログに記録します。condition は 4 種類のいずれかのみ受け付けます。"""
    return f'記録: {city} / {condition} / {temp_c}°C'
 
print(agent_w.run_sync('東京は今日とても良い天気で 22 度でした。記録して。').output)
# LLM は 'good' などではなく必ず 'sunny' のいずれかを選ぶ

from datetime import date
from pydantic import BaseModel, Field
from pydantic_ai import Agent
 
class BookingRequest(BaseModel):
    name: str = Field(min_length=1)
    seat_no: int = Field(ge=1, le=300)
    booking_date: date
 
agent_b = Agent('google-gla:gemini-3-flash-preview', instructions='短く日本語で答えて。')
 
@agent_b.tool_plain
def make_booking(req: BookingRequest) -> str:
    """予約を確定します。"""
    return f'予約完了: {req.name} さん / 座席 {req.seat_no} / {req.booking_date}'
 
print(agent_b.run_sync('明日 田中で 12 番座席を予約して').output)

from datetime import date
from pydantic import BaseModel, Field, model_validator
from pydantic_ai import Agent
 
class DateRange(BaseModel):
    start: date
    end: date
 
    @model_validator(mode='after')
    def check_order(self) -> 'DateRange':
        if self.end < self.start:
            raise ValueError(f'end ({self.end}) は start ({self.start}) 以降である必要があります')
        return self
 
agent_d = Agent('google-gla:gemini-3-flash-preview', instructions='日本語で簡潔に。')
 
@agent_d.tool_plain
def query_logs(range_: DateRange) -> str:
    """指定日付範囲のログ件数を返します。"""
    days = (range_.end - range_.start).days + 1
    return f'{range_.start} 〜 {range_.end} ({days} 日間) のログ: 42 件'
 
# 「先月から今月まで」のような曖昧な指示でも、LLM は妥当な順序の日付を組み立ててくる
print(agent_d.run_sync('2026-04-01 から 2026-04-30 までのログ件数は?').output)

@agent.tool_plain(retries=3)
def my_tool(x: Annotated[int, Field(ge=0)]) -> str:
    ...