跳至主要内容

langchain记忆和优化方案



在使用LangChain开发聊天机器人时,平衡上下文长度和历史记忆可通过以下策略实现,结合技术方法和LangChain工具:

---

 1. 摘要压缩法(Summary Compression)
- 实现方式:
  - 使用`ConversationSummaryMemory`模块,自动生成历史对话的摘要。
  - 在每轮对话后,用LLM生成简洁摘要(如“用户喜欢咖啡,曾询问订单状态”),替代原始长文本。
- 优点:显著减少token占用,保留核心信息。
- 示例代码:
  python
  from langchain.memory import ConversationSummaryMemory
  from langchain.llms import OpenAI

  llm = OpenAI()
  memory = ConversationSummaryMemory(llm=llm)
  memory.save_context({"input": "我喜欢喝咖啡"}, {"output": "已记录您的偏好"})
  print(memory.load_memory_variables({})) 输出摘要

---

 2. 关键实体提取(Entity Extraction)
- 实现方式:
  - 使用`EntityMemory`模块,自动识别并存储用户、地点、偏好等实体。
  - 结合数据库(如Redis)持久化存储关键信息。
- 优点:快速检索结构化数据,无需遍历历史。
- 示例代码:
  python
  from langchain.memory import EntityMemory
  memory = EntityMemory(llm=llm)
  memory.save_context({"input": "我叫张三"}, {"output": "你好,张三!"})
  print(memory.load_memory_variables({"input": "我的名字是什么?"})) 返回"张三"

---

 3. 滑动窗口+向量检索(Sliding Window + Vector Search)
- 实现方式:
  - 用`ConversationBufferWindowMemory`保留最近N轮对话(如3轮)。
  - 旧对话分块存入向量数据库(如FAISS),通过语义检索召回相关历史。
- 优点:动态控制长度,按需召回重要信息。
- 示例流程:
  1. 最近对话:`buffer_window_memory.load_memory()`
  2. 历史对话向量化存储:
     python
     from langchain.embeddings import OpenAIEmbeddings
     from langchain.vectorstores import FAISS

     embeddings = OpenAIEmbeddings()
     vector_store = FAISS.from_texts("历史对话片段", embeddings)
  3. 检索相关历史:
     python
     retriever = vector_store.as_retriever()
     docs = retriever.get_relevant_documents("用户的问题")

---

 4. 动态上下文管理(Dynamic Context Management)
- 实现方式:
  - 根据当前对话意图(通过意图识别模型或关键词匹配),动态加载相关历史。
  - 例如:用户问“订单状态”,则从数据库加载订单号、时间等字段。
- 工具建议:结合`langchain.chains.LLMRouterChain`实现意图路由。

---

 5. 综合方案示例
python
from langchain.memory import (
    ConversationSummaryMemory,
    ConversationBufferWindowMemory,
    EntityMemory,
    CombinedMemory
)
from langchain.llms import OpenAI

llm = OpenAI()
 组合多种记忆模块
memory = CombinedMemory(memories=
    ConversationBufferWindowMemory(k=3), 保留最近3轮对话
    ConversationSummaryMemory(llm=llm), 生成全局摘要
    EntityMemory(llm=llm) 提取关键实体
)

 模拟对话
memory.save_context({"input": "我叫李四"}, {"output": "你好,李四!"})
memory.save_context({"input": "喜欢咖啡"}, {"output": "已记录您的偏好"})
memory.save_context({"input": "订单状态"}, {"output": "请提供订单号"})

 调用时,自动整合摘要、实体和最近对话
context = memory.load_memory_variables({"input": "我之前喜欢什么饮料?"})
print(context)
 输出:{'history': '...', 'entities': {'李四', '咖啡'}, ...}

---

 6. 优化方向
- Token计算:使用`langchain.llms.base.get_num_tokens`监控长度,接近限制时触发摘要。
- 提示词优化:改进摘要生成提示(如“请用100字总结对话,保留用户偏好和关键事实”)。
- 评估测试:通过人工测试或自动化评估(如BLEU分数)验证记忆准确性。

通过结合摘要、实体提取、滑动窗口和向量检索,可以在控制上下文长度的同时,精准召回关键历史信息。

此博客中的热门博文

Elasticsearch 读写原理指南

### 1. 什么是 segment,里面装了什么? 在 Lucene(也是 Elasticsearch)里,索引被切分成若干 **segment(段)**,每个 segment 是一个完整的、只读的倒排索引单元。一个 segment 包含: * **倒排词典** —— 用 **FST(Finite‑State Transducer)** 以高度压缩的形式保存每个字段出现的所有 term 以及 term→ord 的映射。对应的磁盘文件是 `*.tim`(新版)或 `*.tis/*.tii`(旧版)。 * **倒排列表(postings)** —— 保存每个 term 出现的文档 ID、频次、位置信息等,文件名通常是 `*.doc`、`*.pos`、`*.pay`。 * **存储字段**(_source、store:true 的字段)—— 以二进制块的形式写入 `*.fdt` / `*.fdx`。 * **doc‑values、norms、向量** 等辅助结构,分别保存在 `*.dv`、`*.norm`、`*.tv` 等文件里。 * **deleted‑docs bitmap**(`*.del`),标记哪些文档已被删除或被更新。 所有这些文件在 segment **写入磁盘后即成为只读**,后续的查询只能读取,永远不会在原文件上进行增删改。 --- ### 2. 原始文档和 FST 为什么都在 segment 里? * **原始文档**:Elasticsearch 默认把完整的 JSON(_source)以及任何 `store:true` 的字段写入 segment 的 `*.fdt/*.fdx` 文件。每个 segment 保存自己的那部分文档,旧的 segment 在合并前仍然保留,直到合并后被删除。 * **FST**:每个字段的词典在每个 segment 中单独维护,采用 FST 进行前缀共享和字节压缩。这样即使同一个 term 在多个 segment 中出现,也会在每个 segment 里拥有独立的映射,查询时只需要在对应 segment 的 FST 中定位即可。 --- ### 3. 查询时到底是怎么遍历 segment 的? 1. **请求入口**      客户端的搜索请求先到达 **协调节点**,协调节点把请求 ...

LLM缓存详解

 可以把“大模型缓存”理解成: 把已经算过的结果(或中间结果)存下来,下次尽量复用 。但这里面其实分几层,不只是简单的“问题→答案”缓存。 1️⃣ 常见的几种缓存类型 (1)KV Cache(推理内部缓存) Transformer 在生成时,会把前面 token 的 Key/Value 向量 缓存下来。 本质:避免重复计算 attention 作用: 同一请求内部加速 特点: 👉 只对“同一上下文继续生成”有效 👉 不跨用户、不跨请求 这类缓存是你体感“流式输出越来越快”的原因之一。 (2)Prompt Cache(提示词缓存) 缓存的是: 相同(或高度相似)的 prompt → 对应的中间表示 / 输出 典型场景: 系统提示词(system prompt)很长 多轮对话里前文基本不变 👉 这里能省掉 前缀计算成本(prefill) (3)Embedding / 语义缓存(Semantic Cache) 这个才是你问题的关键 👇 不是按“字符串完全一致”,而是: 把问题转成向量 → 找“语义相似”的历史问题 → 直接复用答案 2️⃣ 为什么命中缓存成本低很多? 因为大模型推理成本主要在两块: (1)Prefill(吃 prompt) 复杂度 ~ O(n²) 很贵(尤其长 prompt) (2)Decode(逐 token 生成) 每个 token 都要算一遍模型 而缓存命中后: KV cache:不用重复 attention Prompt cache:不用重新 encode 语义缓存: 直接跳过模型推理 👉 相当于从: 几十~几百毫秒 + GPU算力 变成: 一次向量检索(毫秒级)+ 直接返回 所以成本差一个数量级是正常的。 3️⃣ “每个人问法不同,怎么命中缓存?” 这是核心难点,也是工程重点👇 ❌ 不能靠字符串匹配 比如: “今天天气怎么样” “今天外面热不热” 字符串完全不同 → 必须 miss ✅ 用语义相似度(Embedding) 流程一般是: 把问题转 embedding(向量) 在向量数据库里找 TopK 相似问题 如果相似度 > 阈值(比如 0.9) 直接返回缓存答案 一个简单示意 Q1: 北京天气怎么样 → embedding A Q2: 北京今天热吗 → embedding B cosine(A, B) ≈ 0.95...

事务的ACID是什么

 事务的 ACID 是数据库事务必须满足的四个基本性质,用来保证在并发和故障情况下数据的正确性与可靠性: A(Atomicity,原子性) 一个事务中的操作要么 全部成功 ,要么 全部失败回滚 ,不存在“只做了一半”的中间状态。 C(Consistency,一致性) 事务执行前后,数据库都必须处于 一致的合法状态 ,满足约束(如主键、外键、唯一性、业务规则等)。 I(Isolation,隔离性) 并发执行的多个事务之间 相互隔离 ,一个事务未提交的中间结果对其他事务不可见(具体强弱由隔离级别决定)。 D(Durability,持久性) 一旦事务提交成功,其结果会被 永久保存 ,即使系统崩溃也不会丢失(通常依赖 WAL/redo log 等机制)。 一句话记忆: 要么全做完、前后不破坏规则、互不干扰、做完不丢。