将记忆存储到数据库中

scan7242026-04-27 9:23

from langchain.agents import create_agent

import pymysql

from langgraph.checkpoint.memory import InMemorySaver

from langgraph.checkpoint.mysql.pymysql import PyMySQLSaver

from init_llm import deepseek_llm

DB_URI="mysql+pymsql://root:Zyj@xx@43.11.11.88:3306/langchain_db?charset=utf8mb4"

with PyMySQLSaver.from_conn_string(DB_URI) as checkpointer:

checkpointer.setup()

agent = create_agent(

model=deepseek_llm,

tools=\[\],

checkpointer=checkpointer

)

config={"configurable": {"thread_id":"session01"}}

resp=agent.invoke({"messages": ${"role":"user","content":"我叫张三，你是谁？""}$ },config=config)

print(resp $'messages'$ $-1$ )

resp=agent.invoke({"messages": ${"role":"user","content":"请问我叫什么名字？""}$ },config=config)

print(resp $'messages'$ $-1$ )