Building your first Discord ChatGPT Bot

1// @see https://docs.aircode.io/guide/functions/
2const axios = require('axios');
3const nacl = require('tweetnacl');
4const { Configuration, OpenAIApi } = require('openai');
5
6const { EmbedBuilder } = require('discord.js');
7
8const {
9  verifyKey,
10  InteractionType,
11  InteractionResponseType,
12} = require('discord-interactions');
13
14const apiKey = process.env.OPENAI_KEY;
15
16const discordToken = process.env.DISCORD_BOT_TOKEN;
17
18// Your public key can be found on your application in the Developer Portal
19const discordPublicKey = process.env.DISCORD_PUBLIC_KEY;
20
21// Slash command's name you use tointeract with bots on Discord after typing /
22const slashCommandName = 'chatgpt';
23
24// Use openAI API to fetch ChatGPT completion
25const getOpenAIChatCompletion = async (question) => {
26  try {
27    const configuration = new Configuration({
28      apiKey,
29    });
30
31    const openai = new OpenAIApi(configuration);
32
33    const completion = await openai.createChatCompletion({
34      // OpenAI models https://platform.openai.com/docs/models
35      model: 'gpt-3.5-turbo',
36      messages: [{ role: 'assistant', content: question }],
37    });
38
39    console.log(
40      'ChatGPT completion data:',
41      completion.data.choices[0].message.content.trim()
42    );
43    return {
44      code: 0,
45      data: completion.data.choices[0].message.content.trim(),
46    };
47  } catch (error) {
48    console.error(`OpenAI api error: ${error.message}`);
49    return {
50      code: 1,
51      message: `OpenAI api error: ${error.message}`,
52    };
53  }
54};
55
56// Build an Embed (rich embed) object for replied message
57const createResponseEmbed = (response) => {
58  const embed = new EmbedBuilder().setColor('#37393E').setDescription(response);
59
60  return { embeds: [embed] };
61};
62
63module.exports = async function (params, context) {
64  console.log('params context', params);
65
66  // Get signature and timestamp from headers
67  const signature = context.headers['X-Signature-Ed25519'];
68  const timestamp = context.headers['X-Signature-Timestamp'];
69  // Get rawBody content
70  const body = context.rawBody;
71
72  // Verify whether the request from discord
73  const isVerified = nacl.sign.detached.verify(
74    Buffer.from(timestamp + body),
75    Buffer.from(signature, 'hex'),
76    Buffer.from(discordPublicKey, 'hex')
77  );
78
79  if (!isVerified) {
80    context.status(401);
81    return 'Invalid request signature';
82  }
83
84  // Interaction type and data
85  const { type, id, data, token } = params;
86
87  console.log('params', params);
88  // Handle verification requests reply with PONG
89  if (type === InteractionType.PING) {
90    return { type: InteractionResponseType.PONG };
91  }
92
93  // Handle the user registered interaction command
94  if (
95    type === InteractionType.APPLICATION_COMMAND &&
96    data.name === slashCommandName
97  ) {
98    const userInput = data.options[0].value;
99
100    // Immediately acknowledge the interaction
101    try {
102      await axios.post(
103        `https://discord.com/api/v9/interactions/${id}/${token}/callback`,
104        {
105          type: InteractionResponseType.DEFERRED_CHANNEL_MESSAGE_WITH_SOURCE,
106        }
107      );
108    } catch (error) {
109      console.error('Error acknowledging the interaction:', error);
110      return;
111    }
112
113    const response = await getOpenAIChatCompletion(userInput);
114    console.log('response form OpenAI ChatCompletion: ', response);
115
116    // Patch message
117    if (response.code === 0) {
118      try {
119        await axios.patch(
120          `https://discord.com/api/v9/webhooks/${process.env.DISCORD_APP_ID}/${token}/messages/@original`,
121          {
122            content: `> ${userInput} \n ${response.data}`,
123          }
124        );
125      } catch (error) {
126        console.error('Error editing the original response:', error.message);
127        return;
128      }
129    }
130  }
131};
132

1// @see https://docs.aircode.io/guide/functions/
2const axios = require('axios');
3const Discord = require('discord.js');
4
5async function registerCommands(appId, commands) {
6	const { DISCORD_BOT_TOKEN } = process.env;
7
8  try {
9    // Use discord sdk register chatgpt command
10    const rest = new Discord.REST({ version: '10' }).setToken(
11      DISCORD_BOT_TOKEN
12    );
13    const data = await rest.put(Discord.Routes.applicationCommands(appId), {
14      body: commands,
15    });
16
17    console.log('Command register response:', data);
18
19    if (data) {
20      return {
21        success: true,
22        message: `WOW, slash command ${data
23          .map((command) => `/${command.name}`)
24          .join(' ')} registered`,
25      };
26    }
27  } catch (err) {
28    console.error('Error from registering commands:', err);
29  }
30}
31
32module.exports = async function (params, context) {
33  console.log('Received params:', params);
34  const slashCommandName = 'chatgpt';
35
36  const ChatGPTCommand = {
37    name: slashCommandName,
38    description: 'Interact with ChatGPT',
39    options: [
40      {
41        name: 'prompt',
42        type: 3,
43        description: 'The message sending to ChatGPT',
44        required: true,
45      },
46    ],
47  };
48
49  const commands = [ChatGPTCommand];
50  // Register ChatGPT command
51  const res = await registerCommands(process.env.DISCORD_APP_ID, commands);
52
53  return res;
54};
55