构建实体记忆清单失败

Question

rt，使用老师视频里的代码运行，不会返回entities

Answer 1

coolpi21 2024-04-07 12:04:13

老师的回答像哑谜！

简单点说，就是你的下一次 input 是否能命中上一次 entites 的集合，如果能，就会返回结果。具体可以参考老师提供的源码部分。

举个例子

_input = {
  "input": "胡八一和王胖子雪利杨经常在一起冒险，合称盗墓铁三角"
}
memory.load_memory_variables(_input) 
// 返回结果 
{'history': '', 'entities': {'胡八一': '', '王胖子': '', '雪利杨': '', '盗墓铁三角': ''}}

所以，当你下一次 input 值为 "盗墓铁三角是谁？" 就会有结果了。

memory.save_context(
  _input,
  {
    "output": "听起来很刺激，我也想加入他们"
  }
)
memory.load_memory_variables({"input":"盗墓铁三角是谁？"})
// 返回结果 
{
    'history': 'Human: 胡八一和王胖子雪利杨经常在一起冒险，合称盗墓铁三角\nAI: 听起来很刺激，我也想加入他们', 
    'entities': {'盗墓铁三角': '盗墓铁三角是由胡八一、王胖子和雪利杨组成的冒险团队。'}
}

Answer 2

tomiezhang 2024-04-05 09:54:25

参考代码：https://github.com/langchain-ai/langchain/blob/master/libs/langchain/langchain/memory/entity.py#L372