前端 AI 对话历史存储与上下文回溯方案设计

2. 上下文窗口裁剪核心逻辑

这是前端与 AI 交互的关键。我们需要一个函数，从完整的 messages 中提取出符合 Token 限制或条数限制的'有效上下文'。

class HistoryManager {
  private db: IDBDatabase;
  // 设定保留的对话轮数（一轮 = User + Assistant）
  private readonly CONTEXT_WINDOW_SIZE = 10;

  constructor(db: IDBDatabase) {
    this.db = db;
  }

  /**
   * 核心方法：获取用于 API 调用的有效上下文
   * @param messages 完整的历史消息列表
   * @param systemPrompt 系统提示词
   * @returns 经过裁剪的、可用于发送的消息数组
   */
  public getValidContext(messages: Message[], systemPrompt: string): Message[] {
    // 1. 始终保留系统提示词
    const systemMessage: Message = {
      id: 'sys',
      role: 'system',
      content: systemPrompt,
      timestamp: 0,
    };

    // 2. 滑动窗口算法：只取最近的 N 条记录
    // 这里简单按条数裁剪，生产环境建议按 Token 数估算
    // filter out system message just in case
    const historyWithoutSystem = messages.filter((m) => m.role !== 'system');

    // 截取最后 N 条
    const recentHistory = historyWithoutSystem.slice(-this.CONTEXT_WINDOW_SIZE * 2);

    // 3. 组装最终上下文：System Prompt + 最近对话
    return [systemMessage, ...recentHistory];
  }

  // ... 其他方法
}

3. 持久化存储实战（IndexedDB 模拟）

以下代码展示了如何将对话保存到 IndexedDB 中，确保刷新不丢失。

// 数据库操作封装
class ChatDB {
  private dbName = 'AI_Chat_DB';
  private storeName = 'conversations';
  public db: IDBDatabase | null = null;

  async init(): Promise<void> {
    return new Promise((resolve, reject) => {
      const request = indexedDB.open(this.dbName, 1);
      request.onerror = () => reject(request.error);
      request.onsuccess = () => {
        this.db = request.result;
        resolve();
      };
      // 创建表结构
      request.onupgradeneeded = (event: any) => {
        const db = event.target.result;
        if (!db.objectStoreNames.contains(this.storeName)) {
          // 以 id 为主键
          db.createObjectStore(this.storeName, { keyPath: 'id' });
        }
      };
    });
  }

  /**
   * 保存或更新会话
   * 实战中建议做'防抖'处理，避免频繁写入
   */
  async saveConversation(conversation: Conversation): Promise<void> {
    if (!this.db) await this.init();
    return new Promise((resolve, reject) => {
      const transaction = this.db!.transaction([this.storeName], 'readwrite');
      const store = transaction.objectStore(this.storeName);
      const request = store.put(conversation);
      // put 操作是 idempotent 的
      request.onsuccess = () => resolve();
      request.onerror = () => reject(request.error);
    });
  }
}

4. 综合应用示例

在实际的业务组件中，我们的调用流程如下：

// 业务逻辑伪代码
async function handleUserSend(
  userInput: string,
  currentConversation: Conversation
) {
  // 1. 构造用户消息
  const userMsg: Message = {
    id: crypto.randomUUID(),
    role: 'user',
    content: userInput,
    timestamp: Date.now(),
  };

  // 2. 更新本地状态（乐观更新 UI）
  currentConversation.messages.push(userMsg);

  // 3. 计算上下文（裁剪逻辑）
  const manager = new HistoryManager(dbInstance);
  const context = manager.getValidContext(
    currentConversation.messages,
    '你是一个资深的前端架构师，请用简洁的语言回答问题。'
  );

  // 4. 发送给 LLM API
  const aiResponse = await fetchAIResponse(context); // 假设这是你的 API 调用函数

  // 5. 追加 AI 回复并持久化
  currentConversation.messages.push(aiResponse);
  currentConversation.updatedAt = Date.now();

  // 6. 存入 IndexedDB
  const chatDb = new ChatDB();
  await chatDb.saveConversation(currentConversation);
}

总结与思考

在前端侧设计 AI 对话历史存储，绝不仅仅是'存数据'那么简单。这本质上是在前端构建一个简易的'上下文管理引擎'。

通过这次重构，我有几点深刻的体会：

1. 前端的价值在于交互与体验：后端的大模型是无状态的，前端必须承担起状态管理的责任。IndexedDB 虽然 API 繁琐，但在处理大量结构化数据时，其性能优势是 localStorage 无法比拟的。
2. 成本控制发生在客户端：很多开发者抱怨 Token 消耗快，其实往往是因为前端没有做好上下文裁剪。通过滑动窗口策略，前端可以有效控制 API 调用的 Token 消耗，直接为企业节省真金白银的成本。
3. 未来的演进方向：目前的方案是基于'条数'的简单裁剪。更高级的方案是引入向量数据库（Vector DB），将历史对话向量化存储。当用户提问时，通过语义检索提取最相关的历史片段注入上下文，实现真正的'长时记忆'。这也是我目前正在探索的方向。

技术选型没有银弹，只有最适合业务场景的方案。希望这套方案能为正在转型 AI 开发的前端同行们提供一些务实的参考。