ChatGPT+FastAPIによる簡易問い合わせAPI 作成 #Python

ChatGPTとFastAPIを使って簡単な問い合わせAPIを作成したのでメモする。

事前準備

ChatGPT APIキー取得
- OpenAIアカウントを作成し、APIキーを取得する
Python OpenAI、FastAPI系ライブラリをインストールする
```
pip install openai fastapi uvicorn
```

コード

main.py

リクエストボディに指定された質問文をChatGPTライブラリに渡し、回答をレスポンスとして返却する

from pydantic import BaseModel
from fastapi import FastAPI
import openai

openai.api_key = "{取得したAPIキーをセットする}"

app = FastAPI()


# リクエストボディ定義
class Question(BaseModel):
    content: str



@app.post("/question")
def question(question: Question):
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=[
            {"role": "user", "content": question.content},
        ],
    )
    return {"answer": response.choices[0]["message"]["content"].strip()}

動作確認

API起動
```
uvicorn main:app --reload --port 8091
```

リクエスト

POST /question HTTP/1.1
Host: localhost:8091
Content-Type: application/json
Content-Length: 39

{
    "content":"メジャーリーグは何チームありますか"
}

レスポンス

{
    "answer": "現在、メジャーリーグには30チームあります。15チームがアメリカンリーグに属し、残りの15チームがナショナルリーグに属しています。"
}

参考情報

https://platform.openai.com/docs/guides/chat