🐷

ChatGPTのように返答するSlackBotを作った話

entaku

2023/02/06に公開

Node.js

OpenAI

tech

この記事は？

はやりのChatGPTのように返答するSlackBotを作成してみました！サンプルコードも載せているのでよかったら参考にしてください〜

ChatGPTのAPI、OpenAI-API

まずChatGPTのように返答する方法ですが、今回openAI APIを利用します

https://platform.openai.com/overview

下図はopenAPIの利用画面ですが、ご覧のようにいくつかのAPIが存在します

今回はこの中でもText completionというライブラリを利用していきます。

APIの作成

それではここからコードを見ていきましょう

今回はCloudFunctionで作成しました

クソコードですが積極的に晒していきたいので、こちらに記載してます　！
https://github.com/entaku0818/chatgpt

まずSlackから来たリクエストを処理します
Slackのコマンドでは送信するリクエストが決まっています。

今回この中からtext と response_urlの2つを使います。

token=gIkuvaNzQIHg97ATvDxqgjtO
&team_id=T0001
&team_domain=example
&enterprise_id=E0001
&enterprise_name=Globular%20Construct%20Inc
&channel_id=C2147483705
&channel_name=test
&user_id=U2147483697
&user_name=Steve
&command=/weather
&text=94070
&response_url=https://hooks.slack.com/commands/1234/5678
&trigger_id=13345224609.738474920.8088930838d88f008e0
&api_app_id=A123456

コード上ではまずresponse_url とtextを取得しています。
またテキストの結果を待たずに200okを返しています。こうしないとSlackcommandが応答を待ってタイムアウトしまうためです。


exports.chatgpt = (req, res) => {
 console.log('req:', req.body);

 const message = req.body.text || "";
 const response_url = req.body.response_url;
 _requestCompletion(res, message,response_url);
 res.status(200).send("");
};

続いてopenaiのAPIを叩きます。
Slackからもらったtextを叩いているだけです。
openaiのAPI_KEYを入れるだけで動きます

function _requestCompletion(res, message, response_url) {
  const apiKey = %%%OPENAI_APIKEY%%%
  const apiUrl = 'https://api.openai.com/v1/completions';
  const prompt = message;
  let headers = {
    'Authorization':'Bearer '+ apiKey,
    'Content-type': 'application/json',
  };

  const body = {
      'model': 'text-davinci-003',
      'max_tokens' : 1024,
      'temperature' : 0.9,
      'prompt': prompt
      };
  const options = {
    method: 'POST',
    body:    JSON.stringify(body),
    headers: headers,
  }
  fetch(apiUrl, options)
    .then(res => res.json())
    .then((result) => {
      let resultText = "";
      result.choices.forEach(element => {
        resultText += element.text
      });
      console.log('Success:', resultText);
      _slackMessage(response_url,resultText) 
    })
    .catch((error) => {
      console.error('Error:', error);
    });
}

最後にSlackAPIです
ここもリクエストでもらった response_url に対して応答返しています。これでBotができています。

async function _slackMessage(url,text) {
  
    let headers = {
      'Content-type': 'application/json',
    };
    const body = {
      "text": text,
      "response_type": "in_channel",
     };
    const options = {
      method: 'POST',
      body:    JSON.stringify(body),
      headers: headers,
    }
  fetch(url, options)
    .then(res => res.json())
    .then((result) => {
      console.log('Success:', result);
    })
    .catch((error) => {
      console.error('Error:', error);
    });
}