15天学会AI应用开发（十六）LangChain实现对话记忆功能

pip install langchain langchain-core langchain-ollama langchain_community

ollama pull qwen2:1.5b

ollama serve

from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
from langchain_core.runnables.history import RunnableWithMessageHistory
from langchain_community.chat_message_histories import ChatMessageHistory
from langchain_ollama import OllamaLLM

# 连接本地离线大模型（Ollama）
llm = OllamaLLM(model="qwen2:1.5b")

 # 1、对话模板（留一个位置放历史）
prompt = ChatPromptTemplate.from_messages([
    ("system", "你是友好助手，记住前面的对话。"),
    MessagesPlaceholder(variable_name="history"),
    ("human", "{input}")
])
# 2、内存存储历史（多用户可按 session_id 分）
store = {}
# 获取指定会话的历史记录
def get_history(session_id: str):
    if session_id not in store:
        store[session_id] = ChatMessageHistory()
    return store[session_id]
# 3、带记忆的执行链
chain = prompt | llm
chain_with_mem = RunnableWithMessageHistory(
    chain,
    get_history,
    input_messages_key="input",
    history_messages_key="history"
)
# 4、测试对话
print("=== 带记忆AI聊天（输入 退出 结束）===")
while True:
    user_input = input("你：")
    if user_input in ["退出", "exit", "quit"]:
        print("对话结束")
        break
    # 调用模型（传入历史）
    response = chain_with_mem.invoke({"input": user_input}, config={"configurable": {"session_id": "session_001"}})
    print(f"AI：{response}\n")

=== 带记忆AI聊天（输入 退出 结束）===
你：你叫什么名字？
AI：我叫小冰。
你：不对，你叫小明
AI：抱歉，我记错了。
你：你叫什么名字？
AI：小冰
你：不对，你叫小明
AI：抱歉，我记错了。请问有什么我能帮助您的吗？
你：你叫什么名字？
AI：小明