Union型

from pydantic import BaseModel
from pydantic_ai import Agent
 
class StockOK(BaseModel):
    product: str
    quantity: int
 
class StockError(BaseModel):
    product: str
    reason: str  # '不存在' / '在庫切れ' など
 
agent = Agent(
    'google-gla:gemini-3-flash-preview',
    output_type=StockOK | StockError,
    instructions='在庫システム照会の結果を返してください。在庫があれば StockOK、なければ StockError。',
)
 
# 利用側は isinstance で分岐
result = agent.run_sync('商品 X-1 の在庫は?')
if isinstance(result.output, StockOK):
    print(f'在庫あり: {result.output.product} × {result.output.quantity}')
else:
    print(f'❌ {result.output.product}: {result.output.reason}')

from typing import Literal
from pydantic import BaseModel, Field
from pydantic_ai import Agent
 
class SearchAction(BaseModel):
    type: Literal['search'] = 'search'
    query: str
 
class CreateAction(BaseModel):
    type: Literal['create'] = 'create'
    title: str
    body: str
 
class DeleteAction(BaseModel):
    type: Literal['delete'] = 'delete'
    id: int
 
Action = SearchAction | CreateAction | DeleteAction
 
agent_a = Agent(
    'google-gla:gemini-3-flash-preview',
    output_type=Action,
    instructions=(
        'ユーザーの自然文を、3 種類のアクションのいずれかに変換してください。\\n'
        '- search: 検索したい場合\\n'
        '- create: 何かを作成したい場合\\n'
        '- delete: 削除したい場合'
    ),
)
 
result = agent_a.run_sync('「PydanticAI入門」というタイトルで記事を書きたい')
match result.output:
    case SearchAction(query=q):
        print(f'検索: {q}')
    case CreateAction(title=t, body=b):
        print(f'作成: {t} / {b[:40]}...')
    case DeleteAction(id=i):
        print(f'削除: id={i}')

from datetime import date
from pydantic import BaseModel
from pydantic_ai import Agent
 
class Booking(BaseModel):
    customer: str
    booking_date: date
    party_size: int
 
agent_b = Agent(
    'google-gla:gemini-3-flash-preview',
    output_type=Booking | str,
    instructions=(
        '予約情報が抽出できれば Booking、'
        '不足や曖昧で抽出できなければ追加質問の文字列を返してください。'
    ),
)
 
# 不完全な入力で曖昧
result = agent_b.run_sync('明日予約お願いします')
if isinstance(result.output, Booking):
    print('予約完了:', result.output)
else:
    print('追加質問:', result.output)  # str

agent_multi = Agent(
    'google-gla:gemini-3-flash-preview',
    output_type=list[SearchAction | CreateAction | DeleteAction],
    instructions='ユーザーの文章から、すべてのアクションを順番に抽出してください。',
)
 
result = agent_multi.run_sync('まず「Python」で検索して、次に「Note 1」を新規作成、最後に id=42 を削除して')
for action in result.output:
    match action:
        case SearchAction(query=q):
            print(f'1. search: {q}')
        case CreateAction(title=t):
            print(f'2. create: {t}')
        case DeleteAction(id=i):
            print(f'3. delete: id={i}')