😎

[API構築6] line-chatbotにChatGPT APIを組み込む

2023/07/19に公開

OpenAI

概要

LINEチャットbotを作成するで作ったbotに、ChatGPTでの返答機能を追加する。

SDK導入

前回調査で一番公式に近いと判断したgo用のOpenAI API用ライブラリであるsashabaranov/go-openaiを導入する。

go get github.com/sashabaranov/go-openai

コーディング

botサーバ自体は以下を利用している。

今回返答を行う箇所を以下のように修正した。

func getResMessage(message string) (string, error) {
	client := openai.NewClient(os.Getenv("OPENAI_API_KEY"))
	resp, err := client.CreateChatCompletion(
		context.Background(),
		openai.ChatCompletionRequest{
			Model: openai.GPT3Dot5Turbo,
			Messages: []openai.ChatCompletionMessage{
				{
					Role:    openai.ChatMessageRoleUser,
					Content: message,
				},
			},
		},
	)

	if err != nil {
		return "", err
	}

	return resp.Choices[0].Message.Content, nil
}