ChatGPT Function Calling で enum の配列を扱う

複数のタグがあり、ユーザーが入力した文章に関連したタグがあったら全部取得したいときがあります。そんなときに使える Function Calling のプログラムです。

import json
import openai

openai.api_key = "XXXXXXXXXXXXXXX"

functions = [
    {
        "name": "make_tags",
        "description": "ユーザーのコメントからタグを選択する",
        "parameters": {
            "type": "object",
            "properties": {
                "tags": {
                    "type": "array",
                    "items": {
                        "type": "string",
                        "enum": [
                            "赤",
                            "緑",
                            "青",
                            "大",
                            "中",
                            "小",
                            "太",
                            "細",
                            "美味い",
                            "不味い",
                            "ラーメン",
                            "カレー",
                            "ピザ",
                        ],
                    }
                },
            },
            "required": ["tags"],
        },
    }
]
user_content ="朱色の小さくてほそい旨くないカレー"
response = openai.ChatCompletion.create(
    model="gpt-4-0613",
    messages=[
        {"role": "user", "content": user_content},
    ],
    functions=functions,
    function_call={"name": "make_tags"},
)
item_params = None
try:
    item_params = json.loads(response.choices[0].message.function_call.arguments)
    print("question:", user_content)
    print("result:", item_params)
except:
    print("error")
    print(response.choices[0].message.function_call.arguments)

question: 朱色の小さくてほそい旨くないカレー
result: {'tags': ['赤', '小', '細', '不味い', 'カレー']}