用 JavaScript 开发飞书 ChatGPT 机器人（含全部源码，免费托管，手把手教程）

1// 引入基础依赖包
2const aircode = require('aircode');
3const axios = require('axios');
4
5// 引入 OpenAI 的 SDK
6const openai = require('openai');
7
8// 从环境变量中获取 OpenAI 的 Secret
9const OpenAISecret = process.env.OpenAISecret;
10let chatGPT = null;
11if (OpenAISecret) {
12  // 与 ChatGTP 聊天的方法，传入字符串即可
13  const configuration = new openai.Configuration({ apiKey: OpenAISecret });
14  const client = new openai.OpenAIApi(configuration);
15  chatGPT = async (content) => {
16    return await client.createChatCompletion({
17      // 使用当前 OpenAI 开放的最新 3.5 模型，如果后续 4 发布，则修改此处参数即可
18      // OpenAI models 参数列表 https://platform.openai.com/docs/models
19      model: 'gpt-3.5-turbo',
20      // 让 ChatGPT 充当的角色为 assistant
21      messages: [{ role: 'assistant', content }],
22    });
23  };
24}
25
26// 从环境变量中获取飞书机器人的 App ID 和 App Secret
27const feishuAppId = process.env.feishuAppId;
28const feishuAppSecret = process.env.feishuAppSecret;
29
30// 获取飞书 tenant_access_token 的方法
31const getTenantToken = async () => {
32  const url =
33    'https://open.feishu.cn/open-apis/v3/auth/tenant_access_token/internal/';
34  const res = await axios.post(url, {
35    app_id: feishuAppId,
36    app_secret: feishuAppSecret,
37  });
38  return res.data.tenant_access_token;
39};
40
41// 用飞书机器人回复用户消息的方法
42const feishuReply = async (objs) => {
43  const tenantToken = await getTenantToken();
44  const url = `https://open.feishu.cn/open-apis/im/v1/messages/${objs.msgId}/reply`;
45  let content = objs.content;
46
47  // 实现 at 用户能力
48  if (objs.openId) content = `<at user_id="${objs.openId}"></at> ${content}`;
49  const res = await axios({
50    url,
51    method: 'post',
52    headers: { Authorization: `Bearer ${tenantToken}` },
53    data: { msg_type: 'text', content: JSON.stringify({ text: content }) },
54  });
55  return res.data.data;
56};
57
58// 飞书 ChatGPT 机器人的入口函数
59module.exports = async function (params, context) {
60  // 判断是否开启了事件 Encrypt Key，如果开启提示错误
61  if (params.encrypt)
62    return { error: '请在飞书机器人配置中移除 Encrypt Key。' };
63
64  // 用来做飞书接口校验，飞书接口要求有 challenge 参数时需直接返回
65  if (params.challenge) return { challenge: params.challenge };
66
67  // 判断是否没有开启事件相关权限，如果没有开启，则返回错误
68  if (!params.header || !params.header.event_id) {
69    // 判断当前是否为通过 Debug 环境触发
70    if (context.trigger === 'DEBUG') {
71      return {
72        error:
73          '如机器人已配置好，请先通过与机器人聊天测试，再使用「Mock by online requests」功能调试。',
74      };
75    } else {
76      return {
77        error:
78          '请参考教程配置好飞书机器人的事件权限，相关权限需发布机器人后才能生效。',
79      };
80    }
81  }
82
83  // 所有调用当前函数的参数都可以直接从 params 中获取
84  // 飞书机器人每条用户消息都会有 event_id
85  const eventId = params.header.event_id;
86
87  // 可以使用数据库极其简单地写入数据到数据表中
88  // 实例化一个名字叫做 contents 的表
89  const contentsTable = aircode.db.table('contents');
90
91  // 搜索 contents 表中是否有 eventId 与当前这次一致的
92  const contentObj = await contentsTable.where({ eventId }).findOne();
93
94  // 如果 contentObj 有值，则代表这条 event 出现过
95  // 由于 ChatGPT 返回时间较长，这种情况可能是飞书系统的重试，直接 return 掉，防止重复调用
96  // 当当前环境为 DEBUG 环境时，这条不生效，方便调试
97  if (contentObj && context.trigger !== 'DEBUG') return;
98  const message = params.event.message;
99  const msgType = message.message_type;
100
101  // 获取发送消息的人信息
102  const sender = params.event.sender;
103
104  // 用户发送过来的内容
105  let content = '';
106
107  // 返回给用户的消息
108  let replyContent = '';
109
110  // 目前 ChatGPT 仅支持文本内容
111  if (msgType === 'text') {
112    // 获取用户具体消息，机器人默认将收到的消息直接返回
113    content = JSON.parse(message.content).text;
114
115    // 如果是 at 所有人，则不处理
116    if (content.indexOf('@_all') >= 0) return;
117
118    // 获取用户发送的内容实体，去掉 at 符号等
119    content = content.replace('@_user_1 ', '');
120
121    // 默认将用户发送的内容回复给用户，仅是一个直接返回对话的机器人
122    replyContent = content;
123
124    // 将消息体信息储存到数据库中，以备后续查询历史或做上下文支持使用
125    await contentsTable.save({
126      eventId: params.header.event_id,
127      msgId: message.message_id,
128      openId: sender.sender_id.open_id,
129      content,
130    });
131
132    // 如果配置了 OpenAI Key 则让 ChatGPT 回复
133    if (OpenAISecret) {
134      // 将用户具体消息发送给 ChatGPT
135      try {
136        const result = await chatGPT(content);
137
138        // 将获取到的 ChatGPT 回复给用户
139        replyContent = `${result.data.choices[0].message.content.trim()}`;
140      } catch (err) {
141        err = err.toJSON();
142        if (err.status === 401)
143          return {
144            error:
145              'OpenAI ChatGPT 返回 401 错误，应该是你配置错了 OpenAI 平台的 Secret Token，请重新配置。',
146          };
147        if (err.status === 429)
148          return {
149            error:
150              'OpenAI ChatGPT 的 API 返回 429 错误，可能是你的额度已耗尽，你需要去 OpenAI 付费充值你的 API 配额。付费购买的 ChatGPT Plus 仅是网页版，API 需要另外充值。',
151          };
152        return err.message;
153      }
154    }
155  } else {
156    replyContent = '不好意思，暂时不支持其他类型的文件。';
157  }
158
159  // 将处理后的消息通过飞书机器人发送给用户
160  await feishuReply({
161    msgId: message.message_id,
162    openId: sender.sender_id.open_id,
163    content: replyContent,
164  });
165
166  // 整个函数调用结束，需要有返回
167  return null;
168};
169