# 创建 agent 时加入 middleware
agent = create_agent(
    model=deepseek_llm,
    tools=[get_user_info, update_user_info],
    checkpointer=InMemorySaver(),
    system_prompt="你是一个用户信息管理助手...",
    state_schema=CustomAgentState,
    middleware=[                     # ← 关键：添加 middleware 参数
        SummarizationMiddleware(
            model=deepseek_llm,
            trigger=("messages", 5),      # 当消息数达到 5 条时触发摘要
            keep=("messages", 2),         # 摘要后保留最后 2 条消息
            summary_prompt="请总结以下内容: {messages}"
        )
    ]
)