ChatGPT保存记录插件开发指南：从技术选型到生产环境部署

1次阅读

没有评论

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

在开发基于 ChatGPT 的应用时，保存对话记录是一个常见需求。开发者需要保存记录的主要原因包括：

审计与合规 ：企业应用需要保留对话记录以满足合规要求
用户体验 ：用户希望查看历史对话，避免重复提问
数据分析 ：通过分析对话内容优化产品体验
上下文保持 ：为多轮对话提供历史上下文

选择合适的存储方案是插件开发的第一步。以下是几种常见方案的对比：

优点：实现简单，零成本
缺点：扩展性差，不适合生产环境

优点：轻量级，单文件数据库
缺点：并发性能有限

优点：文档型数据库，天然适合 JSON 格式的对话数据
缺点：需要额外的数据库服务器

优点：全托管服务，自动扩展
缺点：按使用量收费，可能产生意外成本

我们使用 Node.js 和 Express 框架作为基础：

初始化项目：

npm init -y
npm install express body-parser axios

创建基础服务器：

const express = require('express');
const app = express();
const bodyParser = require('body-parser');

app.use(bodyParser.json());

app.listen(3000, () => {console.log('Server running on port 3000');
});

设计 RESTful API 接口：

POST /conversations – 创建新对话
GET /conversations/:id – 获取特定对话
GET /conversations – 获取所有对话列表
PUT /conversations/:id – 更新对话

const mongoose = require('mongoose');

// 定义对话模型
const conversationSchema = new mongoose.Schema({
  userId: String,
  messages: [{
    role: String,
    content: String,
    timestamp: {type: Date, default: Date.now}
  }],
  createdAt: {type: Date, default: Date.now}
});

const Conversation = mongoose.model('Conversation', conversationSchema);

// 保存对话端点
app.post('/api/conversations', async (req, res) => {
  try {const { userId, messages} = req.body;

    const conversation = new Conversation({
      userId,
      messages
    });

    await conversation.save();

    res.status(201).json({
      success: true,
      data: conversation
    });
  } catch (error) {res.status(500).json({
      success: false,
      error: error.message
    });
  }
});

const authenticate = (req, res, next) => {const authHeader = req.headers['authorization'];
  const token = authHeader && authHeader.split(' ')[1];

  if (!token) {return res.status(401).json({error: 'Unauthorized'});
  }

  // 验证 token 逻辑
  next();};

app.use('/api', authenticate);

使用 AES-256 加密敏感内容：

const crypto = require('crypto');

const encrypt = (text, key) => {const iv = crypto.randomBytes(16);
  const cipher = crypto.createCipheriv('aes-256-cbc', Buffer.from(key), iv);
  let encrypted = cipher.update(text);
  encrypted = Buffer.concat([encrypted, cipher.final()]);
  return {iv: iv.toString('hex'), encryptedData: encrypted.toString('hex') };
};

实现批量写入：收集多条消息后一次性写入数据库
使用 Redis 缓存频繁访问的对话
对数据库查询添加适当索引

实现重试机制和断路器模式：

const retry = async (fn, retries = 3, delay = 1000) => {
  try {return await fn();
  } catch (error) {if (retries <= 0) throw error;
    await new Promise(res => setTimeout(res, delay));
    return retry(fn, retries - 1, delay * 2);
  }
};