Google Colab で OpenAI API の Function Calling を試す

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

import os
os.environ["OPENAI_API_KEY"] = "<OpenAI_APIキー>"

from openai import OpenAI

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

# アシスタントの作成
assistant = client.beta.assistants.create(
    instructions="あなたはお天気ボットです。 提供されている関数を使用して質問に答えます。",
    model="gpt-4-1106-preview",
    tools=[{
        "type": "function",
        "function": {
            "name": "getCurrentTemperature",
            "description": "特定の場所の気温を取得",
            "parameters": {
                "type": "object",
                "properties": {
                    "location": {"type": "string", "description": "場所 (例:東京)"},
                    "unit": {"type": "string", "enum": ["c", "f"]}
                },
                "required": ["location"]
            }
        }
    }]
)

# スレッドの準備
thread = client.beta.threads.create()

# ユーザーメッセージの追加
message = client.beta.threads.messages.create(
    thread_id=thread.id,
    role="user",
    content="群馬県の気温を教えてください。"
)

# スレッドのメッセージリストの確認
messages = client.beta.threads.messages.list(
    thread_id=thread.id,
    order="asc"
)
for message in messages:
    print(message.role, ":", message.content[0].text.value)

user : 群馬県の気温を教えてください。

# アシスタントにリクエスト
run = client.beta.threads.runs.create(
    thread_id=thread.id,
    assistant_id=assistant.id,
)

# 実行状況の確認
run = client.beta.threads.runs.retrieve(
    thread_id=thread.id,
    run_id=run.id
)
print(run.status)

requires_action

import json

# require_actionのパラメータの取得
tool_id = run.required_action.submit_tool_outputs.tool_calls[0].id
tool_function_name = run.required_action.submit_tool_outputs.tool_calls[0].function.name
tool_function_arguments = json.loads(run.required_action.submit_tool_outputs.tool_calls[0].function.arguments)

print("id:", tool_id)
print("name:", tool_function_name)
print("arguments:", tool_function_arguments)

id: call_6avyFMrvuLtLTr1BzYmVL8p7
name: getCurrentTemperature
arguments: {'location': '群馬県', 'unit': 'c'}

# 特定の場所の気温を取得する関数 (ダミー)
def getCurrentTemperature(location, unit):
    print("location:", location)
    print("unit:", unit)
    return "22度"

# 特定の場所の気温を取得
tool_function_output = getCurrentTemperature(
    tool_function_arguments["location"], 
    tool_function_arguments["unit"]
)

# 関数の出力を提出
run = client.beta.threads.runs.submit_tool_outputs(
    thread_id=thread.id,
    run_id=run.id,
    tool_outputs=[
        {
            "tool_call_id": tool_id,
            "output": tool_function_output,
        }
    ]
)

# スレッドのメッセージリストの確認
messages = client.beta.threads.messages.list(
    thread_id=thread.id,
    order="asc"
)
for message in messages:
    print(message.role, ":", message.content[0].text.value)

user : 群馬県の気温を教えてください。
assistant : 群馬県の気温は22度です。