添加内存(状态)
可以使用Memory对象初始化链,这将在调用链时保留数据。这使得链具有状态。
入门
from langchain.chains import ConversationChain
from langchain.memory import ConversationBufferMemory
conversation = ConversationChain(
llm=chat,
memory=ConversationBufferMemory()
)
conversation.run("简要回答。彩虹的前三种颜色是什么?")
# -> 彩虹的前三种颜色是红色、橙色和黄色。
conversation.run("接下来的四种颜色是什么?")
# -> 彩虹的接下来的四种颜色是绿色、蓝色、靛蓝色和紫色。
'The next four colors of a rainbow are green, blue, indigo, and violet.'
基本上,BaseMemory
定义了langchain
存储内存的接口。它通过load_memory_variables
方法读取存储的数据,并通过save_context
方法存储新数据。您可以在Memory部分了解更多信息。