【Python】簡易なコマンドラインチャットインターフェースの例

# どこからこのファイルを読んでも動くようにパス指定
import os
__dir__ = os.path.dirname(os.path.abspath(__file__))
# pip install openai でインストール
import openai

# このファイルと同ディレクトリ内に api-key.txt という名前で OpenAI のAPIキーを書いたファイルを置きます
with open(f'{__dir__}/api-key.txt', 'r') as f:
    openai.api_key = f.read()


# https://qiita.com/sakasegawa/items/db2cff79bd14faf2c8e0 より引用した関数
def completion(new_message_text: str, settings_text: str = '', past_messages=None):
    """
    This function generates a response message using OpenAI's GPT-3 model by taking in a new message text,
    optional settings text and a list of past messages as inputs.

    Args:
    new_message_text (str): The new message text which the model will use to generate a response message.
    settings_text (str, optional): The optional settings text that will be added as a system message to the past_messages list. Defaults to ''.
    past_messages (list, optional): The optional list of past messages that the model will use to generate a response message. Defaults to [].

    Returns:
    tuple: A tuple containing the response message text and the updated list of past messages after appending the new and response messages.
    """
    if past_messages is None:
        past_messages = []
    if len(past_messages) == 0 and len(settings_text) != 0:
        system = {"role": "system", "content": settings_text}
        past_messages.append(system)
    new_message = {"role": "user", "content": new_message_text}
    past_messages.append(new_message)

    result = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=past_messages
    )
    response_message = {"role": "assistant", "content": result.choices[0].message.content}
    past_messages.append(response_message)
    response_message_text = result.choices[0].message.content
    return response_message_text, past_messages


if __name__ == '__main__':
    # このファイルから見て system_settings/tsunko.txt の部分にAIアシスタントの設定などを記述を置きます
    with open(f'{__dir__}/system_settings/tsunko.txt', 'r', encoding='utf-8') as f:
        system_settings = f.read()
    past_messages = []
    print("何か入力してください。\\で行を終えると複数行入力になります")
    send_msg = ''
    while True:
        send_input = input('あなた: ')
        # 複数行対応。末尾が \ なら次の行も今のメッセージに含める
        while send_input.endswith('\\'):
            send_input = input('　　　  ')
            send_msg += send_input[:-1] + '\n'
        send_msg += send_input

        if send_msg == 'exit':
            exit(0)
        # チャットAPIなどにメッセージを投げる
        response, messages = completion(send_msg, system_settings, past_messages)
        if '\n' in response:  # 返事が複数行ならインデントを揃えて表示
            print('あいて: ')
            for line in response.splitlines():
                print('　　　  ' + line)
        else:
            print('あいて: ' + response)
        send_msg = ''