@agent.tool 基礎

from datetime import datetime
from zoneinfo import ZoneInfo
from pydantic_ai import Agent
 
agent = Agent('google-gla:gemini-3-flash-preview', instructions='短く日本語で答えてください。')
 
@agent.tool_plain
def current_time_jst() -> str:
    """現在の日本時刻 (JST) を ISO8601 形式で返します。"""
    return datetime.now(ZoneInfo('Asia/Tokyo')).isoformat()
 
result = agent.run_sync('今、日本では何時ですか?')
print(result.output)
# LLM は instructions と質問を見て、内部で current_time_jst() を呼び、その結果を踏まえて応答する

from pydantic_ai import Agent
 
agent = Agent('google-gla:gemini-3-flash-preview', instructions='答えは日本語で。')
 
# モック天気データ
WEATHER_DB = {
    'Tokyo': '晴れ、22度',
    'Osaka': '曇り、19度',
    'Sapporo': '雪、-2度',
}
 
@agent.tool_plain
def get_weather(city: str) -> str:
    """指定した都市の現在の天気を返します。city は英字の都市名 (例: Tokyo, Osaka)。"""
    return WEATHER_DB.get(city, f'{city} の天気データは未登録です')
 
result = agent.run_sync('東京と札幌の天気を教えて')
print(result.output)
# LLM は get_weather('Tokyo') と get_weather('Sapporo') を 2 回呼び、まとめて答える

from dataclasses import dataclass
import httpx
from pydantic_ai import Agent, RunContext
 
@dataclass
class WeatherDeps:
    http: httpx.AsyncClient
    api_base: str
 
agent_w = Agent(
    'google-gla:gemini-3-flash-preview',
    deps_type=WeatherDeps,
    instructions='答えは日本語で。',
)
 
@agent_w.tool
async def get_weather(ctx: RunContext[WeatherDeps], city: str) -> str:
    """指定都市の現在天気を外部 API から取得します。"""
    url = f'{ctx.deps.api_base}/weather?city={city}'
    response = await ctx.deps.http.get(url)
    response.raise_for_status()
    return response.json()['summary']
 
# 利用側
import asyncio
async def main():
    async with httpx.AsyncClient() as http:
        deps = WeatherDeps(http=http, api_base='https://api.example.com')
        result = await agent_w.run('Tokyo の天気は?', deps=deps)
        print(result.output)
 
asyncio.run(main())