Amazon ConnectとLexでのコールセンター向けAIチャットボットで、Function Callingを利用し、発話内容から必要な情報を補正しつつJSON形式で抽出してみた

import json
import os
import openai
from decimal import Decimal
openai.api_key = os.environ['API_Key']

def function_calling(prompt):
    messages = [
        {'role': 'user', 'content': prompt},
    ]
    functions = [
        {
            'name': 'get_artist_and_song_title',
            'description': 'アーティスト名と曲名を取得します',
            'parameters': {
                'type': 'object',
                'properties': {
                    'artist': {
                        'type': 'string',
                        'description': 'アーティスト名。アーティスト名が省略されている場合、フルネームにしてください。 例:Ado '
                    },
                    'song': {
                        'type': 'string',
                        'description': '曲名。 曲名が省略されている場合、フルネームにしてください。例:うっせぇわ '
                    },
                },
                'required': ['artist', 'song']
            }
        }
    ]
    response = openai.ChatCompletion.create(
        # model='gpt-3.5-turbo-0613',
        model='gpt-4-0613',
        messages=messages,
        functions=functions,
        temperature=0,
        max_tokens=512,
        # function_call='auto'
        function_call={'name': 'get_artist_and_song_title'}
    )

    print(json.dumps(response['choices'][0], default=decimal_to_int, ensure_ascii=False))
    return json.loads(response['choices'][0]['message']['function_call']['arguments'])

def decimal_to_int(obj):
    if isinstance(obj, Decimal):
        return int(obj)

def elicit_slot(slot_to_elicit, intent_name, slots):
    return {
        'sessionState': {
            'dialogAction': {
                'type': 'ElicitSlot',
                'slotToElicit': slot_to_elicit,
            },
            'intent': {
                'name': intent_name,
                'slots': slots,
                'state': 'InProgress'
            }
        }
    }

def confirm_intent(message_content, intent_name, slots):
    return {
        'messages': [{'contentType': 'PlainText', 'content': message_content}],
        'sessionState': {
            'dialogAction': {
                'type': 'ConfirmIntent',
            },
            'intent': {
                'name': intent_name,
                'slots': slots,
                'state': 'Fulfilled'
            }
        }
    }

def close(fulfillment_state, message_content, intent_name, slots):
    return {
        'messages': [{'contentType': 'PlainText', 'content': message_content}],
        'sessionState': {
            'dialogAction': {
                'type': 'Close',
            },
            'intent': {
                'name': intent_name,
                'slots': slots,
                'state': fulfillment_state
            }
        }
    }

def free_intent(event):
    print('Received event:' + json.dumps(event,default=decimal_to_int, ensure_ascii=False))
    intent_name = event['sessionState']['intent']['name']
    slots = event['sessionState']['intent']['slots']
    user_prompt = event['inputTranscript']

    if slots['freeinput'] is None:
        return elicit_slot('freeinput', intent_name, slots)

    confirmation_status = event['sessionState']['intent']['confirmationState']

    if confirmation_status == 'Confirmed':
        return close('Fulfilled', 'それでは、電話を切ります', intent_name, slots)

    elif confirmation_status == 'Denied':
        return close('Failed', 'お力になれず、申し訳ありません。電話を切ります', intent_name, slots)

    # confirmation_status == 'None'
    get_artist_and_song = function_calling(user_prompt)
    artist_name = get_artist_and_song['artist']
    song_name = get_artist_and_song['song']

    return confirm_intent(
        f'それでは、回答します。アーティスト名は{artist_name}、曲名は{song_name}。以上が回答になります。回答に納得したかたは、はい、とお伝え下さい。納得いかない場合、いいえ、とお伝え下さい',
        intent_name, slots)


def lambda_handler(event, context):
    print('Received event:' + json.dumps(event,default=decimal_to_int, ensure_ascii=False))

    intent_name = event['sessionState']['intent']['name']

    if intent_name == 'free':
        return free_intent(event)

{
  "artist": "official 髭 男 dism",
  "song": "pretender"
}

# 省略...

functions = [
    {
        "name": "get_artist_and_song_title",
        "description": "アーティスト名と曲名を取得します。スペースを含むアーティスト名はスペースを削除してください。",
        "parameters": {
            "type": "object",
            "properties": {
                "artist": {
                    "type": "string",
                    "description": "アーティスト名。アーティスト名が省略されている場合、フルネームにしてください。例:Ado"
                },
                "song": {
                    "type": "string",
                    "description": "曲名。 曲名が省略されている場合、フルネームにしてください。例:うっせぇわ "
                },
            },
            "required": ["artist", "song"]
        },
    },

# 省略...

{
  "artist": "official髭男dism",
  "song": "pretender"
}