openaiのpythonの使い方
概要
- openaiのライブラリのバージョン1.0.0以降でインターフェースの大幅な変更があった
- 以前の0.X.X系のバージョンの使い方は/openai-python-legacy/
- 環境変数
OPENAI_API_KEYにAPIキーを設定しておけば、暗黙的にAPIキーを参照してくれる
インストール
$ pip install openai
APIキーの設定
import os
os.environ["OPENAI_API_KEY"] = "sk-xxxxxxxxxxxxxxxxxxxxxxxx"
利用可能なモデルの確認
from openai import OpenAI
client = OpenAI()
models = pd.DataFrame([dict(model) for model in client.models.list()])
models["date"] = pd.to_datetime(models["created"], unit="s")
models.sort_values(by=["date"], ascending=False)
embedding
from openai import OpenAI
client = OpenAI()
def get_embedding(text, model="text-embedding-3-small"):
vec = client.embeddings.create(input = [text], model=model).data[0].embedding
return vec
chat completion
from openai import OpenAI
client = OpenAI()
response = client.chat.completions.create(
model="gpt-3.5-turbo",
messages=[
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": "Who won the world series in 2020?"},
{"role": "assistant", "content": "The Los Angeles Dodgers won the World Series in 2020."},
{"role": "user", "content": "Where was it played?"}
]
)
print(response.choices[0].message.content)
# => "The 2020 World Series was played at Globe Life Field in Arlington, Texas."
chat completion(streaming)
from openai import OpenAI
client = OpenAI()
response = client.chat.completions.create(
model="gpt-3.5-turbo",
messages=[
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": "Who won the world series in 2020?"},
{"role": "assistant", "content": "The Los Angeles Dodgers won the World Series in 2020."},
{"role": "user", "content": "Where was it played?"}
]
stream=True,
n=1,
)
for chunk in response:
if chunk.choices:
content = chunk.choices[0].message.content
if content:
print(content, end="")
# => "The 2020 World Series was played at Globe Life Field in Arlington, Texas."