如何高效保存ChatGPT聊天记录：从API调用到本地存储的完整解决方案

2次阅读

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

在使用 ChatGPT API 开发应用时，很多开发者都会遇到聊天记录保存的痛点。最常见的问题包括：

API 调用限制 ：免费账号有每分钟请求次数限制，高频调用可能导致临时封禁
非结构化数据 ：API 返回的原始数据包含大量元信息，直接存储会造成冗余
上下文丢失 ：默认不保存历史对话，多轮对话时无法维持连贯性
隐私风险 ：敏感对话若以明文存储可能违反 GDPR 等数据保护法规

优点：零存储成本，实时获取最新数据
缺点：受网络影响大，无法离线访问，历史数据查询困难

JSON 文件存储
适合快速原型开发
单文件结构简单易读
但大文件加载性能差
数据库存储
SQLite 适合轻量级应用
PostgreSQL/MySQL 适合企业级系统
支持复杂查询和索引优化

import json
from datetime import datetime
import openai

class ChatRecorder:
    def __init__(self, file_path='chat_history.json'):
        self.file_path = file_path
        self.session_id = datetime.now().strftime('%Y%m%d_%H%M%S')

    def save_conversation(self, messages):
        """保存对话到 JSON 文件"""
        try:
            entry = {'timestamp': str(datetime.now()),
                'session_id': self.session_id,
                'messages': messages
            }

            # 读取现有数据或初始化空列表
            try:
                with open(self.file_path, 'r') as f:
                    data = json.load(f)
            except (FileNotFoundError, json.JSONDecodeError):
                data = []

            data.append(entry)

            with open(self.file_path, 'w') as f:
                json.dump(data, f, indent=2)

            return True
        except Exception as e:
            print(f"保存失败: {str(e)}")
            return False

import sqlite3
from cryptography.fernet import Fernet

class SecureChatDB:
    def __init__(self, db_path='chats.db', encryption_key=None):
        self.conn = sqlite3.connect(db_path)
        self.cursor = self.conn.cursor()
        self._init_db()

        # 加密初始化
        self.cipher = Fernet(encryption_key) if encryption_key else None

    def _init_db(self):
        """初始化数据库表结构"""
        self.cursor.execute('''CREATE TABLE IF NOT EXISTS conversations
                             (id INTEGER PRIMARY KEY AUTOINCREMENT,
                              session_id TEXT NOT NULL,
                              timestamp DATETIME DEFAULT CURRENT_TIMESTAMP,
                              content BLOB)''')
        self.conn.commit()

    def save_message(self, session_id, message):
        """保存加密后的消息"""
        try:
            content = json.dumps(message).encode('utf-8')
            if self.cipher:
                content = self.cipher.encrypt(content)

            self.cursor.execute("INSERT INTO conversations (session_id, content) VALUES (?, ?)",
                               (session_id, content))
            self.conn.commit()
            return True
        except Exception as e:
            print(f"数据库操作失败: {e}")
            return False