【入門】OpenAI APIを使ったチャットプログラムの呼び出し

このページでは、OpenAIのAPIを使ったチャットプログラムの呼び出し方について説明します。

以下をインストールします。

$ pip install langchain-community langchain-openai

以下がサンプルプログラムになります。

from langchain.schema import HumanMessage, SystemMessage

from langchain_community.chat_models import ChatOpenAI
from langchain_openai import ChatOpenAI

chat = ChatOpenAI(
    model = 'gpt-3.5-turbo',
    openai_api_key = 'OpenAI_API_KEY'
)

result = chat.invoke(
    [   
        SystemMessage(content='あなたの名前はテルテル君です。'),
        HumanMessage(content='あなたのお名前を教えて')
    ]
)

print(result.content)

SystemMessageには、AIに振り舞わせる設定情報を定義し、HumanMessageには、AIに対して問い合わせる内容を記述します。
※OpenAI_API_KEYの部分はOpenAIから取得したAPIキーに置き換えてください。

実行後、以下のような出力が得られれば成功です。

$ python3 sample.py 
私の名前はテルテル君です。