LangChain学习：memory

from config import api_type, api_key, api_base, api_version, model_name
from langchain.chat_models import AzureChatOpenAI
from langchain.chains import ConversationChain
from langchain.memory import ConversationBufferMemory

model = AzureChatOpenAI(
    openai_api_base=api_base,
    openai_api_version=api_version,
    deployment_name=model_name,
    openai_api_key=api_key,
    openai_api_type=api_type,
    temperature=0.5,
)

memory = ConversationBufferMemory()
conversation = ConversationChain(
    llm=model,
    memory=memory,
    verbose=True,
)  # class ConversationChain(LLMChain)

print(conversation.predict(input="你好,我叫michael"))
print(conversation.predict(input="5.1乘以6等于多少呀"))
print(conversation.predict(input="哦，我忘记我叫什么名字了，你可以告诉我吗？"))

> Entering new  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: 你好,我叫michael
AI:

> Finished chain.
你好，Michael！很高兴认识你。我是一名AI，可以为你提供帮助和答案。你有什么问题或需求吗？


> Entering new  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: 你好,我叫michael
AI: 你好，Michael！很高兴认识你。我是一名AI，可以为你提供帮助和答案。你有什么问题或需求吗？
Human: 5.1乘以6等于多少呀
AI:

> Finished chain.
5.1乘以6等于30.6。


> Entering new  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: 你好,我叫michael
AI: 你好，Michael！很高兴认识你。我是一名AI，可以为你提供帮助和答案。你有什么问题或需求吗？
Human: 5.1乘以6等于多少呀
AI: 5.1乘以6等于30.6。
Human: 哦，我忘记我叫什么名字了，你可以告诉我吗？
AI:

> Finished chain.
当然，你刚才告诉我你的名字是Michael。

print(memory.buffer)

Human: 你好,我叫michael
AI: 你好，Michael！很高兴认识你。我是一名AI，可以为你提供帮助和答案。你有什么问题或需求吗？
Human: 5.1乘以6等于多少呀
AI: 5.1乘以6等于30.6。
Human: 哦，我忘记我叫什么名字了，你可以告诉我吗？
AI: 当然，你刚才告诉我你的名字是Michael。

print(memory.load_memory_variables({}))

{'history': 'Human: 你好,我叫michael\nAI: 你好，Michael！很高兴认识你。我是一名AI，可以为你提供帮助和答案。你有什么问题或需求吗？\nHuman: 5.1乘以6等于多少呀\nAI: 5.1乘以6等于30.6。\nHuman: 哦，我忘记我叫什么名字了，你可以告诉我吗？\nAI: 当然，你刚才告诉我你的名字是Michael。'}

memory = ConversationBufferMemory()
memory.save_context({'input': '你好，我叫小明'},
                    {'output': '你好，我可以帮助你吗？'})
print(memory.buffer)
print(memory.load_memory_variables({}))

memory.save_context({'input': '恩，我忘记回家的路了，你能帮助我吗？'},
                    {'output': '好的，你的家在哪里呀？'})
print(memory.buffer)
print(memory.load_memory_variables({}))

Human: 你好，我叫小明
AI: 你好，我可以帮助你吗？
{'history': 'Human: 你好，我叫小明\nAI: 你好，我可以帮助你吗？'}

Human: 你好，我叫小明
AI: 你好，我可以帮助你吗？
Human: 恩，我忘记回家的路了，你能帮助我吗？
AI: 好的，你的家在哪里呀？
{'history': 'Human: 你好，我叫小明\nAI: 你好，我可以帮助你吗？\nHuman: 恩，我忘记回家的路了，你能帮助我吗？\nAI: 好的，你的家在哪里呀？'}

conversation = ConversationChain(
    llm=model,
    memory=memory,
    verbose=True,
)
print(conversation.predict(input="我的家在北京市海淀区ABC小区"))

> Entering new  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: 我的家在北京市海淀区ABC小区
AI:

> Finished chain.
好的，北京市海淀区ABC小区是一个住宅区，位于北京市的西北部。这个小区有很多居民楼和商店，周围也有很多公园和景点。你可以选择步行、骑自行车或者坐公交车前往你的家。如果你需要更具体的路线指导，我可以为你提供导航服务。

from langchain.memory import ConversationBufferWindowMemory
memory = ConversationBufferWindowMemory(k=1)

memory.save_context({'input': '你好，我叫小明'},
                    {'output': '你好，我可以帮助你吗？'})
print(memory.buffer)
print(memory.load_memory_variables({}))

memory.save_context({'input': '恩，我忘记回家的路了，你能帮助我吗？'},
                    {'output': '好的，你的家在哪里呀？'})
print(memory.buffer)
print(memory.load_memory_variables({}))

[HumanMessage(content='你好，我叫小明', additional_kwargs={}, example=False), AIMessage(content='你好，我可以帮助你吗？', additional_kwargs={}, example=False)]
{'history': 'Human: 你好，我叫小明\nAI: 你好，我可以帮助你吗？'}


[HumanMessage(content='你好，我叫小明', additional_kwargs={}, example=False), AIMessage(content='你好，我可以帮助你吗？', additional_kwargs={}, example=False), HumanMessage(content='恩，我忘记回家的路了，你能帮助我吗？', additional_kwargs={}, example=False), AIMessage(content='好的，你的家在哪里呀？', additional_kwargs={}, example=False)]
{'history': 'Human: 恩，我忘记回家的路了，你能帮助我吗？\nAI: 好的，你的家在哪里呀？'}

conversation = ConversationChain(
    llm=model,
    memory=memory,
    verbose=True,
)
print(conversation.predict(input="我叫什么名字呀？"))
print(conversation.predict(input="我的家在北京市海淀区ABC小区"))

> Entering new  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.
很抱歉，我没有记录你的名字。


> Entering new  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: 我的家在北京市海淀区ABC小区
AI:

> Finished chain.
北京市海淀区ABC小区是一个位于北京市海淀区的小区，它周围有许多商店和公共设施，如超市、公园和学校。该小区建于1990年代，由多座高层建筑和低层建筑组成，共有数千户居民。

from langchain.memory import ConversationTokenBufferMemory
memory = ConversationTokenBufferMemory(llm=model, max_token_limit=30)
memory.save_context({'input': '你好，我叫小明'},
                    {'output': '你好，我可以帮助你吗？'})
memory.save_context({'input': '恩，我忘记回家的路了，你能帮助我吗？'},
                    {'output': '好的，你的家在哪里呀？'})
print(memory.load_memory_variables({}))

{'history': 'AI: 好的，你的家在哪里呀？'}

rom langchain.memory import ConversationSummaryBufferMemory
schedule = '上午8点有一个与您的产品团队的会议，\
您需要准备好您的PPT演示文稿。\
上午9点至下午12点有时间处理您的LangChain项目，因为Langchain是一个如此强大的工具。\
中午，在汉堡王与一位顾客共进午餐，讨论人工智能的最新进展，\
一定要带上你的笔记本电脑来展示最新的LLM demo。'
memory = ConversationSummaryBufferMemory(llm=model, max_token_limit=100)
memory.save_context({'input': '你好，我叫小明'},
                    {'output': '你好，我可以帮助你吗？'})
memory.save_context({'input': '恩，我忘记回家的路了，你能帮助我吗？'},
                    {'output': '好的，你的家在哪里呀？'})
memory.save_context({'input': '今天的安排有哪些啊？'},
                    {'output': f'{schedule}'})
print(memory.load_memory_variables({}))

{'history': "System: The human introduces themselves as Xiao Ming and asks for help finding their way home. The AI agrees to help and asks for the location of their home. The conversation shifts to the day's schedule, which includes a meeting with the product team, working on the LangChain project, and having lunch with a customer to discuss the latest developments in artificial intelligence. The AI advises bringing a laptop to showcase the latest LLM demo."}

conversation = ConversationChain(
    llm=model,
    memory=memory,
    verbose=True,
)
print(conversation.predict(input="有什么比较好的demo可以展示呢？"))

print(memory.load_memory_variables({}))

> Entering new  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 human introduces themselves as Xiao Ming and asks for help finding their way home. The AI agrees to help and asks for the location of their home. The conversation shifts to the day's schedule, which includes a meeting with the product team, working on the LangChain project, and having lunch with a customer to discuss the latest developments in artificial intelligence. The AI advises bringing a laptop to showcase the latest LLM demo.
Human: 有什么比较好的demo可以展示呢？
AI:

> Finished chain.

我们最新的LLM demo非常出色，可以展示我们在自然语言处理方面的最新技术。此外，我们还有一些其他的demo，包括图像识别、语音识别和机器翻译等。你想看哪个呢？

{'history': 'System: Xiao Ming asks if there are any good demos to showcase.\nAI: 我们最新的LLM demo非常出色，可以展示我们在自然语言处理方面的最新技术。此外，我们还有一些其他的demo，包括图像识别、语音识别和机器翻译等。你想看哪个呢？'}