Skip to main content

添加内存(状态)

可以使用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部分了解更多信息。