langchainでAzure OpenAIのAPIを叩くときの仕様が若干変わってたのでメモ

背景

langchainを久々に触ったら今後若干仕様が変わるとのwarningが出たのでメモ。

使い方

環境変数は以下のように設定(API versionはこれを使ってね！っていうお触れ書きがOpenAI側からあったような気もするが、忘れたので後で追記する)

OPENAI_API_TYPE="azure"
OPENAI_API_VERSION="2023-09-01-preview"
OPENAI_API_KEY="*********************"
AZURE_OPENAI_ENDPOINT="https://xxx.openai.azure.com/

新たにlangchain-openaiをインストール。
langchain.chat_modelsから叩いていたが、langchain v0.2.0から変更になるらしい。

pip install langchain-openai

コード上では以下のように使う。

from langchain-openai import AzureChatOpenAI

llm = AzureChatOpenAI(azure_deployment="deployment_name")