微信小程序接入ChatGPT训练知识库：从零开始的实战指南

2次阅读

没有评论

共计 1817 个字符，预计需要花费 5 分钟才能阅读完成。

最近在开发一个需要智能对话功能的微信小程序时，发现直接集成 ChatGPT 知识库会遇到几个典型问题：

API 调用复杂度高，特别是处理长对话上下文时
网络延迟导致用户体验不佳
敏感数据如何安全传输和存储
微信小程序的环境限制（如域名白名单、请求大小限制等）

这些痛点让很多开发者在尝试 AI 功能时望而却步。经过多次实践，我总结出一套相对成熟的解决方案。

目前主流的接入方案有三种：

直接调用 OpenAI API
优点：响应快，功能全
缺点：需要处理翻墙问题，有地域限制
通过云函数中转
优点：规避地域限制，可以添加业务逻辑
缺点：增加了一层网络延迟
自建代理服务
优点：完全可控，可以做深度定制
缺点：维护成本高

综合考虑后，我选择了 云函数中转 方案，既保证了可用性，又不需要太高维护成本。

// pages/chat/chat.js
Page({
  data: {messages: [],
    inputValue: ''
  },

  sendMessage: function() {
    const that = this;
    wx.request({
      url: 'https://your-domain.com/api/chat',
      method: 'POST',
      data: {
        question: this.data.inputValue,
        history: this.data.messages
      },
      success(res) {
        that.setData({
          messages: [...that.data.messages, 
            {role: 'user', content: that.data.inputValue},
            {role: 'assistant', content: res.data.answer}
          ],
          inputValue: ''
        });
      }
    });
  }
});

// server/api/chat.js
const axios = require('axios');

exports.main = async (event, context) => {
  try {
    const response = await axios.post(
      'https://api.openai.com/v1/chat/completions',
      {
        model: "gpt-3.5-turbo",
        messages: prepareMessages(event.history, event.question)
      },
      {
        headers: {'Authorization': `Bearer ${process.env.OPENAI_KEY}`,
          'Content-Type': 'application/json'
        }
      }
    );

    return {answer: response.data.choices[0].message.content
    };
  } catch (error) {console.error(error);
    return {error: 'AI 服务暂时不可用'};
  }
};

function prepareMessages(history, newQuestion) {
  // 处理历史消息，避免超过 token 限制
  const messages = [...history];
  messages.push({role: 'user', content: newQuestion});
  return messages.slice(-6); // 保留最近 6 条消息
}