OpenAI Responses API 入門 (4) - Function Calling

# パッケージのインストール
!pip install -U openai openai-agents

import os
from google.colab import userdata

# 環境変数の準備 (左端の鍵アイコンでOPENAI_API_KEYを設定)
os.environ["OPENAI_API_KEY"] = userdata.get("OPENAI_API_KEY")

from openai import OpenAI

# クライアントの準備
client = OpenAI()

import requests

# 関数の定義
def get_weather(latitude, longitude):
    response = requests.get(f"https://api.open-meteo.com/v1/forecast?latitude={latitude}&longitude={longitude}&current=temperature_2m,wind_speed_10m&hourly=temperature_2m,relative_humidity_2m,wind_speed_10m")
    data = response.json()
    return data['current']['temperature_2m']

# ツールリストの準備
tools = [{
    "type": "function",
    "name": "get_weather",
    "description": "指定された座標の現在の温度を摂氏で取得します。",
    "parameters": {
        "type": "object",
        "properties": {
            "latitude": {"type": "number"},
            "longitude": {"type": "number"}
        },
        "required": ["latitude", "longitude"],
        "additionalProperties": False
    },
    "strict": True
}]

# メッセージリストの準備
messages = [
    {"role": "user", "content": "今日の大阪の天気は？"}
]

# 推論の実行
response = client.responses.create(
    model="gpt-4o",
    input=messages,
    tools=tools
)
print(response)

Response(
    id='resp_67d49a5cae0c8191b056da7a35406d420cbb9363ec0da1ce',  
    created_at=1741986396.0,  
    error=None,  
    incomplete_details=None,  
    instructions=None,  
    metadata={},  
    model='gpt-4o-2024-08-06',  
    object='response',  
    output=[
        ResponseFunctionToolCall(
            id='fc_67d49a5d6908819194e157dae6b4d96a0cbb9363ec0da1ce',  
            arguments='{"latitude":34.6937,"longitude":135.5022}',  
            call_id='call_7zR220GxgU7h7FIr5Wpgymu4',  
            name='get_weather',  
            type='function_call',  
            status='completed'
        )
    ],  
    parallel_tool_calls=True,  
    temperature=1.0,  
    tool_choice='auto',  
    tools=[
        FunctionTool(
            name='get_weather',  
            parameters={
                'type': 'object',  
                'properties': {
                    'latitude': {'type': 'number'},  
                    'longitude': {'type': 'number'}
                },  
                'required': ['latitude', 'longitude'],  
                'additionalProperties': False
            },  
            strict=True,  
            type='function',  
            description='指定された座標の現在の温度を摂氏で取得します。'
        )
    ],  
    top_p=1.0,  
    max_output_tokens=None,  
    previous_response_id=None,  
    reasoning=Reasoning(
        effort=None,  
        generate_summary=None
    ),  
    status='completed',  
    text=ResponseTextConfig(
        format=ResponseFormatText(type='text')
    ),  
    truncation='disabled',  
    usage=ResponseUsage(
        input_tokens=277,  
        output_tokens=25,  
        output_tokens_details=OutputTokensDetails(
            reasoning_tokens=0
        ),  
        total_tokens=302,  
        input_tokens_details={'cached_tokens': 0}
    ),  
    user=None,  
    store=True  
)

import json

# 関数の実行
tool_call = response.output[0]
args = json.loads(tool_call.arguments)
result = get_weather(args["latitude"], args["longitude"])
print(result)

# メッセージリストに関数の実行結果を追加
messages.append(tool_call)  # Function Callingの応答
messages.append({           # 関数の実行結果を追加
    "type": "function_call_output",
    "call_id": tool_call.call_id,
    "output": str(result)
})

# 推論の実行
response_2 = client.responses.create(
    model="gpt-4o",
    input=messages,
    tools=tools,
)
print(response_2.output_text)