开发一个 Discord ChatGPT 机器人

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

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