ChatGPT记忆功能实战指南：从零搭建到生产环境优化

1次阅读

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

在集成 ChatGPT 记忆功能时，开发者常遇到以下核心问题：

Token 限制 ：GPT 模型有严格的 Token 上限（如 GPT-3.5 的 4096 Tokens），长对话容易触发截断
状态持久化 ：跨会话的状态保存需要解决存储介质选择和同步逻辑
上下文管理 ：多轮对话中如何准确还原历史上下文（Context Preservation）

方案	容量上限	持久性	适用场景
LocalStorage	5-10MB	标签页级别	简单客户端状态存储
IndexedDB	50MB+	浏览器级别	结构化数据存储
服务端缓存 (Redis)	无硬性限制	全局持久化	多端同步场景

from typing import List, Dict
from uuid import uuid4

class ConversationChain:
    def __init__(self):
        self.chain: Dict[str, List[dict]] = {}

    def add_message(self, conv_id: str, role: str, content: str) -> str:
        """
        :param conv_id: 会话 ID
        :param role: 'user' 或 'assistant'
        :param content: 消息内容
        :return: 消息 ID
        """
        msg_id = str(uuid4())
        if conv_id not in self.chain:
            self.chain[conv_id] = []
        self.chain[conv_id].append({
            'id': msg_id,
            'role': role,
            'content': content
        })
        return msg_id

from flask import request, jsonify
import jwt
from functools import wraps

SECRET_KEY = 'your-secret-key'

def token_required(f):
    @wraps(f)
    def decorated(*args, **kwargs):
        token = request.headers.get('Authorization')
        if not token:
            return jsonify({'error': 'Token missing'}), 401
        try:
            jwt.decode(token.split()[1], SECRET_KEY, algorithms=['HS256'])
        except jwt.ExpiredSignatureError:
            return jsonify({'error': 'Token expired'}), 401
        except jwt.InvalidTokenError:
            return jsonify({'error': 'Invalid token'}), 401
        return f(*args, **kwargs)
    return decorated