站在巨人的肩膀上,用Node+ChatGPT模块实现一个接口

本文最后更新于:1 年前

目录

前言

准备工作

功能实现

写在最后


前言

蹭一下最近比较火的人工智能ChatGPT的热度,最近看到许多小伙伴都在调戏ChatGPT,看到这我就坐不住了,这种事怎么能少了我,于是闲(划)暇(水)之余用node做了个api

准备工作

功能实现

登录ChatGPT后打开控制台,复制cookie里的session-token

初始化node项目,下载chatgpt模块

新建node server服务,代码如下:

import http from "http";
http
  .createServer((req, res) => {
    res.setHeader("Access-Control-Allow-Origin", "*"); //设置响应头解决跨域
    if (req.url !== "/sendMsg") return sendRes(res, "not find", 404);
    let _data = "";
    req.on("data", (d) => {
      _data += d;
    });
    req.on("end", () => {
      res.writeHead(200, {
        "Content-Type": "text/plain",
        "Access-Control-Allow-Origin": "*",
      });
      res.write('椰丝');
      res.end();
    });
  })
  .listen(1024, () => {
    console.log("服务开启!");
  });

使用postman或者apiPost发起post请求

完成一个最简单的接口后,咱们接入ChatGPT

/**
 * @name:
 * @description: 封装了一下ChatGPT
 * @param {*} msg 发送的消息
 * @param {*} sessionToken 浏览器cookie拿到的令牌
 * @return {Promise}
 */
const sendChatGPTMsg = async ({ msg, sessionToken }) => {
  const { promise, resolve, reject } = defer();
  const api = new ChatGPTAPI({
    sessionToken,
    markdown: false,
  });
  await api.ensureAuth().catch(reject); // 校验令牌
  api.sendMessage(msg).catch(reject).then(resolve);
  return promise;
};

这个defer是把promise处理了一下

/**
 * @name:
 * @description: promise扁平处理
 * @return {*}
 */
const defer = () => {
  let resolve, reject;
  return {
    promise: new Promise((_resolve, _reject) => {
      resolve = _resolve;
      reject = _reject;
    }),
    resolve,
    reject,
  };
};

这个时候,我们在req.on(“end”)中调用这个api

http
  .createServer((req, res) => {
    res.setHeader("Access-Control-Allow-Origin", "*"); //设置响应头解决跨域
    if (req.url !== "/sendMsg") return sendRes(res, "not find", 404);
    let _data = "";
    req.on("data", (d) => {
      _data += d;
    });
    req.on("end", () => {
      const data = JSON.parse(_data);
      sendChatGPTMsg(data)
        .then((r) => {
          res.writeHead(200, {
            "Content-Type": "text/plain",
            "Access-Control-Allow-Origin": "*",
          });
          console.log(r)
          res.write(r);
          res.end();
        })
    });
  })
  .listen(1024, () => {
    console.log("服务开启!");
  });

并且在ApiPost中将之前浏览器中复制的sessionToken补充全,再填上想问的问题,发送请求,就会有以下效果

至此,一个ChatGPT的api就实现完了,我们可以用它接入飞书,企微的webhook机器人,或者接入内网穿透,参照这篇文章:没有云服务器?内网穿透了解一下_DieHunter1024的博客-CSDN博客

实现远程调用api

写在最后

感谢你看到了最后,如果文章对你有帮助的话,还请点赞支持一下博主,非常感谢

源码:chatGPT-api: 基于chatGPT模块实现一个api