Langchain 的 ConversationSummaryBufferMemory

ConversationSummaryBufferMemory 在内存中保留最近交互的缓冲区，但不仅仅是完全刷新旧的交互，而是将它们编译成摘要并使用两者。但与之前的实现不同的是，它使用令牌长度而不是交互次数来确定何时刷新交互。

我们首先来了解一下如何使用这些实用程序，

示例代码，

from langchain.memory import ConversationSummaryBufferMemory
from langchain.llms import OpenAI

llm = OpenAI()

memory = ConversationSummaryBufferMemory(
    llm=llm, max_token_limit=40, return_messages=True
)
memory.save_context({"input": "嗨"}, {"output": "最近怎么样？"})
memory.save_context({"input": "没什么特别的，你呢？"}, {"output": "没什么特别的。"})

memory.load_memory_variables({})

输出结果，

{'history': [SystemMessage(content='\nThe human and AI are engaging in conversation. The human greets the AI and the AI asks how the human is doing.', additional_kwargs={}),
  HumanMessage(content='没什么特别的，你呢？', additional_kwargs={}, example=False),
  AIMessage(content='没什么特别的。', additional_kwargs={}, example=False)]}

示例代码，

from langchain.chains import ConversationChain

conversation_with_summary = ConversationChain(
    llm=llm,
    # We set a very low max_token_limit for the purposes of testing.
    memory=ConversationSummaryBufferMemory(llm=OpenAI(), max_token_limit=250),
    verbose=True,
)
conversation_with_summary.predict(input="嗨，最近怎么样？")

输出结果，

> Entering new ConversationChain chain...
Prompt after formatting:
The following is a friendly conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.

Current conversation:

Human: 嗨，最近怎么样？
AI:

> Finished chain.
' 嗨！最近还不错，我在学习新的技能，并且正在尝试新的任务。我也在尝试改进我的语言处理能力，以便更好地与人交流。你呢？'

示例代码，

conversation_with_summary.predict(input="只是在写一些文档！")

输出结果，

Entering new ConversationChain chain...
Prompt after formatting:
The following is a friendly conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.

Current conversation:
Human: 嗨，最近怎么样？
AI:  嗨！最近还不错，我在学习新的技能，并且正在尝试新的任务。我也在尝试改进我的语言处理能力，以便更好地与人交流。你呢？
Human: 只是在写一些文档！
AI:

> Finished chain.
' 哦，看起来很有趣！你在写什么文档？'

示例代码，

conversation_with_summary.predict(input="你听说过 LangChain 吗？")

输出结果，

> Entering new ConversationChain chain...
Prompt after formatting:
The following is a friendly conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.

Current conversation:
Human: 嗨，最近怎么样？
AI:  嗨！最近还不错，我在学习新的技能，并且正在尝试新的任务。我也在尝试改进我的语言处理能力，以便更好地与人交流。你呢？
Human: 只是在写一些文档！
AI:  哦，看起来很有趣！你在写什么文档？
Human: 你听说过 LangChain 吗？
AI:

> Finished chain.
' 是的，我知道LangChain。它是一个基于区块链的语言学习平台，旨在帮助人们更好地学习外语。你正在写关于LangChain的文档吗？'

示例代码，

# 我们可以看到，摘要和缓冲区都已更新
conversation_with_summary.predict(
    input="哈哈，不对，不过很多人都把它混淆了。"
)

输出结果，

> Entering new ConversationChain chain...
Prompt after formatting:
The following is a friendly conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.

Current conversation:
System: 
The AI thinks artificial intelligence is a force for good because it will help humans reach their full potential. It has been learning new skills and trying new tasks, as well as attempting to improve its language processing so that it can communicate better with humans. The human asks how the AI has been doing recently.
Human: 只是在写一些文档！
AI:  哦，看起来很有趣！你在写什么文档？
Human: 你听说过 LangChain 吗？
AI:  是的，我知道LangChain。它是一个基于区块链的语言学习平台，旨在帮助人们更好地学习外语。你正在写关于LangChain的文档吗？
Human: 哈哈，不对，不过很多人都把它混淆了。
AI:

> Finished chain.
' 哦，我明白了。那你在写什么文档呢？'

完结！

Langchain 的 ConversationSummaryBufferMemory

Langchain 的 ConversationSummaryBufferMemory

猜你喜欢