手把手做一个公众号GPT智能客服(七)GPT 接入微信机器人

// /controller/openai.js

const { Configuration, OpenAIApi } = require("openai")

const apiKey = "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"

const configuration = new Configuration({
  apiKey
})

const openAIApi = new OpenAIApi(configuration)

/**
 * 保存用户发送到chatgpt的消息信息
 * prompts 数据结构：
 * {
 *   wxopenid: [
 *     { role: "user", content: Content }
 *   ]
 * }
 */
const prompts = {}

// 处理给chatgpt的数据保存记忆体
function savePrompt(fromusername, content) {
  if (prompts[fromusername]) {
    // 存在添加数据
    prompts[fromusername].push(
      { role: 'user', content }
    )
  } else {
    // 不存在就新建数据
    prompts[fromusername] = [
      { role: 'user', content }
    ]
  }
  return prompts[fromusername]
}

async function getAnswer(fromusername, prompt) {
  const messages = savePrompt(fromusername, prompt)

  const completion = await openAIApi.createChatCompletion({
    model: "gpt-3.5-turbo",
    messages: messages,
  })

  return completion.data.choices[0].message.content
}

module.exports = getAnswer

// /controller/switcher.js

const { customerController } = require('./customer')

// +++++++++++++++++++
const getAnswer = require('./openai')
// +++++++++++++++++++

const { getModel, decreaseBalance, getBalance } = require('../model/users')

const { formatMsg } = require('../utils/format')

async function messageController(req, res, next) {
  // 接收到微信方发来的消息并处理
  const msg = formatMsg(req.body)
  
  // 暂存用户发送来的问题
  const content = msg['content']

  // 回复信息
  const {
    fromusername
  } = msg

  msg['createtime'] = Math.floor((new Date().getTime()) / 1000)

  // 生成问题答案逻辑
  const model = await getModel(fromusername)
  if (model && model === 'chatgpt') {
    const balance = await getBalance(fromusername)
    if (balance > 0) {
      msg['content'] = '答案正在准备中...'
      res.render('reply', msg)

      // 扣除账户金额 1
      await decreaseBalance(fromusername)

      // +++++++++++++++++++
      // 获取 chatgpt 的回答
      const answer = await getAnswer(fromusername, content)
      // +++++++++++++++++++

      // 发送客服消息
      await customerController(fromusername, answer)
    } else {
      msg['content'] = '您的账户余额不足，请充值~'
      res.render('reply', msg)
    }
  } else {
    // 转人工服务
  }
}

module.exports = messageController

// /model/users.js

//...

// 更改用户模式
async function changeModel(wxOpenId, model) {
  const result = await userModel.findOneAndUpdate({ wxOpenId }, { model})
  return result
}

//...

module.exports = {
  //...
  changeModel
}

// /controller/swither.js

//...

const { getBalance, rechargeAccount, getUser, createAccount, changeModel } = require('../model/users')

async function switcherController(req, res) {
  //...

  switch (requestBody.msgtype) {
    case 'event':
      //...
      switch (requestBody.eventkey) {
        //...
        case 'model_chatgpt':
          await changeModel(fromusername, 'chatgpt')
          res.render('reply', {
            ...requestBody,
            content: '你的身份已经切换为 ChatGPT 模式，机器人为您提供答疑服务 ☕️'
          })
          break
        case 'model_human':
          await changeModel(fromusername, 'human')
          res.render('reply', {
            ...requestBody,
            content: '你的身份已经切换为人工服务模式，很高兴为你答疑解惑 ☕️'
          })
          break
        //...
      }

      //...
  }
}

//...

//...

async function switcherController(req, res) {
  //...

  switch (requestBody.msgtype) {
    case 'voice':
    case 'text':
    //...
  }
}

//...

// /controller/message.js

//...

async function messageController(req, res, next) {
  //...
  
  // 暂存用户发送来的问题
  const content = msg['recognition'] ? msg['recognition'] : msg['content']
	
  //...
}

//...