Toolset

from pydantic_ai import Agent, FunctionToolset, RunContext
from pydantic_ai.models.google import GoogleModel
 
# 関数を 2 通りの方法で登録
weather = FunctionToolset()
 
@weather.tool_plain
def temperature(city: str) -> float:
    """指定都市の現在気温 (摂氏) を返す。"""
    table = {'Tokyo': 22.5, 'Osaka': 24.1, 'Sapporo': 12.0}
    return table.get(city, 20.0)
 
@weather.tool
def conditions(ctx: RunContext, city: str) -> str:
    """指定都市の現在の天気を文字列で返す。"""
    return f'{city}: 晴れ'
 
agent = Agent(
    GoogleModel('gemini-3-flash-preview'),
    toolsets=[weather],
    instructions='天気の質問には toolset を使って答えてください。',
)
 
print(agent.run_sync('東京の気温と天気は?').output)

from pydantic_ai import Agent, CombinedToolset, FunctionToolset
from pydantic_ai.models.google import GoogleModel
 
calc = FunctionToolset()
 
@calc.tool_plain
def add(a: float, b: float) -> float:
    """2 数の和。"""
    return a + b
 
@calc.tool_plain
def multiply(a: float, b: float) -> float:
    """2 数の積。"""
    return a * b
 
# weather (上のセル) と統合
all_tools = CombinedToolset([weather, calc])
 
agent = Agent(
    GoogleModel('gemini-3-flash-preview'),
    toolsets=[all_tools],
    instructions='必要に応じて天気と計算のツールを使ってください。',
)
 
print(agent.run_sync('大阪と札幌の気温の差は?').output)

stocks = FunctionToolset()
 
@stocks.tool_plain
def current(symbol: str) -> float:
    """指定銘柄の現在株価。"""
    return 150.5
 
# 同名ツールが weather にあるとして、prefix で衝突回避
combined = CombinedToolset([
    weather.prefixed('weather'),   # weather_temperature, weather_conditions
    stocks.prefixed('stock'),       # stock_current
])
 
agent = Agent(GoogleModel('gemini-3-flash-preview'), toolsets=[combined])

mock_tools = FunctionToolset()
 
@mock_tools.tool_plain
def temperature(city: str) -> float:
    return 99.9  # テスト用固定値
 
with agent.override(toolsets=[mock_tools]):
    print(agent.run_sync('東京の気温').output)