【强烈推荐】LangGraph核心概念解析：State管理+实战案例，一篇搞定大模型开发

关注

原创最新推荐文章于 2026-08-23 16:22:16 发布 925 阅读 AI 写同款· GEO 优化 ›

AI Agent开发实战：从核心概念到复杂工作流构建指南最近在推进大模型应用落地时，团队反复讨论一个核心问题：如何让大模型从“被动问答”升级为“主动执行复杂任务”？传统的提示工程（Prompt Engineering）在面对多步骤、需调用工具、有状态记忆的复杂场景时，显得力不从心。这正是 AI Agent（智能体）技术要解决的核心痛点。它不再是简单的聊天机器人，而是能够感知环境、规划决策、调用工具并持续学习的“数字员工”。本文将为你系统拆解 AI Agent 的开发全貌。无论你是刚接触大模型的开发者，还是希望将 Agent 能力集成到现有业务中的工程师，都能阅读详情

本文详细介绍了LangGraph框架的核心概念——State管理。通过计算器示例和聊天机器人实战，演示了如何使用字典和TypedDict定义State，以及Reducer机制控制状态更新。特别讲解了add_messages函数如何智能管理对话历史，并提供了完整的聊天机器人实现代码。读者可掌握State、Node、Edge等基础概念，理解LangGraph工作原理，并能独立开发大模型应用。

一、先把环境准备好

咱们先把开发环境搞定，后面代码都能直接跑起来。打开终端，依次执行：

# 安装核心包
pip install langgraph langchain-openai
# 安装可视化工具（可选，但强烈推荐）
pip install pyppeteer ipython

创建一个langgraph_demo.py文件，把API Key配置好：

import os
import getpass
# 设置OpenAI API Key
if not os.environ.get("OPENAI_API_KEY"):
os.environ["OPENAI_API_KEY"] = getpass.getpass("请输入你的OpenAI API Key: ")

好了，准备工作完成。接下来进入正题。

二、State到底是个啥？

很多教程上来就讲概念，我换个思路。咱们先看一个实际场景：

假设你要做一个简单的计算器程序，流程是这样的：

用户输入一个数字，比如10
第一步：给这个数字加1
第二步：用上一步的结果减2
最后输出结果

传统写法可能是这样：

def calculator(x):
x = x + 1  # 第一步
y = x - 2  # 第二步
return x, y
result = calculator(10)
print(result)  # (11, 9)

这没问题，但如果流程复杂了呢？比如有10个步骤，每个步骤可能用到前面多个结果，代码就会很乱。

LangGraph的解决方案是：用一个共享的"状态"（State）来存储所有中间结果。每个处理步骤只需要：

读取State中需要的数据
处理完后，把结果写回State
不用管其他步骤的数据

这就是State的核心思想：一个在整个流程中共享的数据容器。

三、第一个例子：用字典做State

咱们把刚才的计算器用LangGraph实现一遍。完整代码如下，每行都加了注释：

from langgraph.graph import StateGraph, START, END
# 定义第一个节点：加法
def addition(state):
"""
这个函数会收到当前的state（一个字典）
从里面取出x的值，加1后返回
"""
print(f"加法节点收到的state: {state}")
current_x = state["x"]
new_x = current_x + 1
# 只返回更新的部分，不用担心其他数据会丢
return {"x": new_x}
# 定义第二个节点：减法
def subtraction(state):
"""
这个函数也会收到state
注意：这里的state已经包含了上一个节点更新后的x值
"""
print(f"减法节点收到的state: {state}")
current_x = state["x"]
y = current_x - 2
# 这里我们新增了一个y，x会保留
return {"y": y}
# 开始构建图
# StateGraph(dict)表示我们的state是一个普通字典
builder = StateGraph(dict)
# 添加两个节点
# 第一个参数是节点名称，第二个参数是对应的函数
builder.add_node("addition", addition)
builder.add_node("subtraction", subtraction)
# 定义节点之间的连接关系（边）
# START -> addition -> subtraction -> END
builder.add_edge(START, "addition")        # 起点连到addition
builder.add_edge("addition", "subtraction") # addition连到subtraction
builder.add_edge("subtraction", END)        # subtraction连到终点
# 编译成可执行的图
graph = builder.compile()
# 运行！传入初始状态
initial_state = {"x": 10}
final_state = graph.invoke(initial_state)
print(f"\n最终状态: {final_state}")

运行这段代码，你会看到：

加法节点收到的state: {‘x’: 10}
减法节点收到的state: {‘x’: 11}

最终状态: {‘x’: 11, ‘y’: 9}

看到了吗？关键点来了：

addition节点只返回了{"x": 11}，但y没有丢
subtraction节点返回了{"y": 9}，x也还在
最终状态包含了所有的更新

这就是LangGraph的魔法：节点只需要返回它要更新的部分，其他数据会自动保留。

看看图长什么样

代码能跑是一回事，理解结构又是另一回事。咱们把这个图可视化出来：

from IPython.display import Image, display
# 生成图的可视化
# xray=True表示显示详细信息
display(Image(graph.get_graph(xray=True).draw_mermaid_png()))

你会看到一个流程图，清楚地显示了节点的连接关系。开发的时候这个功能特别有用，能一眼看出逻辑对不对。

四、进阶：用TypedDict让代码更严谨

用字典虽然灵活，但有个大问题：如果你不小心写错了key的名字，程序运行到那里才会报错。

比如这样：

def buggy_function(state):
# 假设state里没有"z"这个key
return {"result": state["z"] + 1}  # 💥运行时报错
为了避免这种问题，实际开发中我们用
TypedDict
来定义State的结构：
from typing_extensions import TypedDict
# 定义State的结构
class State(TypedDict):
x: int  # x是整数
y: int  # y也是整数
# 现在用State来构建图
builder = StateGraph(State)
# 节点函数保持不变
def addition(state: State):  # 加上类型标注
return {"x": state["x"] + 1}
def subtraction(state: State):
return {"y": state["x"] - 2}
# 后面的代码完全一样
builder.add_node("addition", addition)
builder.add_node("subtraction", subtraction)
builder.add_edge(START, "addition")
builder.add_edge("addition", "subtraction")
builder.add_edge("subtraction", END)
graph = builder.compile()
# 运行
result = graph.invoke({"x": 10})
print(result)  # {'x': 11, 'y': 9}

用了TypedDict后，IDE会给你代码提示，写错了key名编辑器就会提醒你。虽然代码多了几行，但避免了很多低级错误。

五、Reducer：状态更新的幕后功臣

现在来解答一个关键问题：为什么节点只返回部分数据，State就能自动合并？

答案是Reducer函数。每个State的key背后都有一个Reducer，默认行为是"覆盖"。我画个图帮你理解：

当前State: {“x”: 10}
节点返回: {“x”: 11}
Reducer执行: 把x从10改成11
结果State: {“x”: 11}

当前State: {“x”: 11}
节点返回: {“y”: 9}
Reducer执行: x不变，新增y
结果State: {“x”: 11, “y”: 9}

默认的Reducer很简单，就是更新或新增。但我们可以自定义更复杂的行为。

六、实战：维护消息列表

聊天机器人需要记住对话历史，每次都要往消息列表里追加新消息。如果用默认的Reducer（覆盖），每次都会丢失之前的对话。

这时候就需要operator.add这个Reducer：

import operator
from typing import Annotated, List
from typing_extensions import TypedDict
# 注意这里的写法
class State(TypedDict):
# Annotated[类型, Reducer函数]
messages: Annotated[List[dict], operator.add]
# 测试一下
def node1(state):
# 返回一个包含新消息的列表
return {"messages": [{"role": "user", "content": "你好"}]}
def node2(state):
print(f"node2收到的消息: {state['messages']}")
# 再追加一条消息
return {"messages": [{"role": "assistant", "content": "你好啊"}]}
builder = StateGraph(State)
builder.add_node("node1", node1)
builder.add_node("node2", node2)
builder.add_edge(START, "node1")
builder.add_edge("node1", "node2")
builder.add_edge("node2", END)
graph = builder.compile()
# 初始状态包含一个空列表
result = graph.invoke({"messages": []})
print(f"\n最终消息列表: {result['messages']}")

运行结果：

node2收到的消息: [{'role': 'user', 'content': '你好'}]
最终消息列表: [
{'role': 'user', 'content': '你好'},
{'role': 'assistant', 'content': '你好啊'}
]

看到了吗？用了operator.add之后，每次返回的消息列表会追加到现有列表中，而不是覆盖。这就是Reducer的威力。

七、接入真实的AI模型

铺垫了这么多，终于到激动人心的部分了——接入GPT做一个真正能聊天的机器人！

完整代码如下，建议你创建一个新文件chatbot.py：

import os
import getpass
from typing import Annotated
from typing_extensions import TypedDict
from langgraph.graph import StateGraph, START, END
from langgraph.graph.message import add_messages
from langchain_openai import ChatOpenAI
from langchain_core.messages import HumanMessage
from IPython.display import Image, display
# 配置API Key
if not os.environ.get("OPENAI_API_KEY"):
os.environ["OPENAI_API_KEY"] = getpass.getpass("请输入OpenAI API Key: ")
# 定义State结构
class State(TypedDict):
# add_messages是LangGraph提供的专门处理消息的Reducer
messages: Annotated[list, add_messages]
# 创建GPT模型实例
llm = ChatOpenAI(model="gpt-4o")
# 定义聊天节点
def chatbot(state: State):
"""
这个函数接收包含对话历史的state
把所有消息发给GPT，获取回复
"""
# state["messages"]是一个消息列表
# 直接传给模型就行
response = llm.invoke(state["messages"])
# 返回模型的回复，add_messages会自动追加到列表
return {"messages": [response]}
# 构建图
builder = StateGraph(State)
builder.add_node("chatbot", chatbot)
builder.add_edge(START, "chatbot")
builder.add_edge("chatbot", END)
# 编译
graph = builder.compile()
# 可视化看看（可选）
# display(Image(graph.get_graph(xray=True).draw_mermaid_png()))
# 单次对话测试
def test_single():
user_message = HumanMessage(content="你好，请介绍一下你自己")
result = graph.invoke({"messages": [user_message]})
print("用户:", user_message.content)
print("AI:", result["messages"][-1].content)
# 连续对话测试
def chat_loop():
print("聊天机器人已启动（输入'退出'结束对话）\n")
while True:
user_input = input("你: ").strip()
if user_input == "退出":
print("再见！")
break
if not user_input:
continue
# 每次都创建新的对话
# 如果要保持对话历史，需要把messages保存下来
for event in graph.stream({"messages": [HumanMessage(content=user_input)]}):
for value in event.values():
print(f"AI: {value['messages'][-1].content}\n")
# 运行
if __name__ == "__main__":
# test_single()  # 测试单次对话
chat_loop()      # 开启连续对话

运行这段代码，你就有了一个能对话的AI助手！

不过这个版本有个问题：每次对话都是独立的，没有记忆。要实现带记忆的版本，需要这样改：

def chat_with_memory():
"""带对话历史的聊天"""
print("聊天机器人已启动（输入'退出'结束对话）\n")
# 用这个列表保存整个对话历史
conversation_history = []
while True:
user_input = input("你: ").strip()
if user_input == "退出":
print("再见！")
break
if not user_input:
continue
# 把用户消息加入历史
conversation_history.append(HumanMessage(content=user_input))
# 用完整的历史调用模型
result = graph.invoke({"messages": conversation_history})
# 获取AI回复
ai_message = result["messages"][-1]
print(f"AI: {ai_message.content}\n")
# 把AI回复也加入历史
conversation_history.append(ai_message)
# 运行带记忆的版本
if __name__ == "__main__":
chat_with_memory()

现在你可以这样聊：

你: 我叫小明
AI: 你好小明！很高兴认识你…

你: 我刚才说我叫什么？
AI: 你刚才说你叫小明。

看，AI能记住之前的对话了！

八、add_messages的秘密

可能你注意到了，我们用的是add_messages而不是operator.add。这两个有什么区别？

operator.add很简单粗暴：直接把两个列表拼起来。

add_messages更聪明：

每条消息都有唯一ID
如果新消息的ID和已有消息相同，就更新那条消息
如果ID不同，就追加新消息

看个例子：

from langgraph.graph.message import add_messages
from langchain_core.messages import HumanMessage, AIMessage
# 场景1：追加新消息
msg1 = [HumanMessage(content="你好", id="1")]
msg2 = [AIMessage(content="你好啊", id="2")]
result = add_messages(msg1, msg2)
print("场景1 - 追加:", [m.content for m in result])
# 输出: ['你好', '你好啊']
# 场景2：更新已有消息
msg1 = [HumanMessage(content="你好", id="1")]
msg2 = [HumanMessage(content="你好呀", id="1")]  # 注意ID相同
result = add_messages(msg1, msg2)
print("场景2 - 更新:", [m.content for m in result])
# 输出: ['你好呀']  # 第一条消息被更新了

这个特性在人机协作场景特别有用。比如用户可以修改AI的回复，然后继续对话。

九、源码解析：add_messages是怎么实现的？

有些同学喜欢刨根问底，咱们快速过一下核心逻辑（不想看可以跳过）：

def add_messages(left, right):
"""
left: 已有的消息列表
right: 新来的消息列表
"""
# 1. 确保都是列表
if not isinstance(left, list):
left = [left]
if not isinstance(right, list):
right = [right]
# 2. 给每条消息分配ID（如果没有的话）
for m in left:
if m.id is None:
m.id = str(uuid.uuid4())
for m in right:
if m.id is None:
m.id = str(uuid.uuid4())
# 3. 建立ID到消息的映射
left_idx_by_id = {m.id: i for i, m in enumerate(left)}
# 4. 合并：ID相同就更新，不同就追加
merged = left.copy()
for m in right:
if m.id in left_idx_by_id:
# ID存在，更新
merged[left_idx_by_id[m.id]] = m
else:
# ID不存在，追加
merged.append(m)
return merged

核心就是用ID来判断是更新还是追加。简单但很实用。

十、完整项目：支持多轮对话的智能助手

最后把所有知识点整合一下，做个功能完整的聊天助手：

import os
import getpass
from typing import Annotated
from typing_extensions import TypedDict
from langgraph.graph import StateGraph, START, END
from langgraph.graph.message import add_messages
from langchain_openai import ChatOpenAI
from langchain_core.messages import HumanMessage, SystemMessage
# ===== 配置部分 =====
if not os.environ.get("OPENAI_API_KEY"):
os.environ["OPENAI_API_KEY"] = getpass.getpass("请输入OpenAI API Key: ")
# ===== State定义 =====
class ChatState(TypedDict):
messages: Annotated[list, add_messages]
# ===== 创建模型 =====
llm = ChatOpenAI(
model="gpt-4o",
temperature=0.7,  # 控制回复的创造性
)
# ===== 定义节点 =====
def chatbot_node(state: ChatState):
"""处理用户消息，生成回复"""
messages = state["messages"]
response = llm.invoke(messages)
return {"messages": [response]}
# ===== 构建图 =====
builder = StateGraph(ChatState)
builder.add_node("chatbot", chatbot_node)
builder.add_edge(START, "chatbot")
builder.add_edge("chatbot", END)
graph = builder.compile()
# ===== 主程序 =====
def main():
print("=" * 50)
print("智能聊天助手 v1.0")
print("=" * 50)
print("提示：输入'退出'结束对话\n")
# 设置系统提示词
conversation = [
SystemMessage(content="你是一个友好、专业的AI助手。回答要简洁明了。")
]
while True:
# 获取用户输入
user_input = input("👤 你: ").strip()
# 退出判断
if user_input.lower() in ["退出", "exit", "quit"]:
print("\n👋 感谢使用，再见！")
break
# 跳过空输入
if not user_input:
continue
# 添加用户消息
conversation.append(HumanMessage(content=user_input))
try:
# 调用图进行处理
result = graph.invoke({"messages": conversation})
# 获取AI回复
ai_response = result["messages"][-1]
conversation.append(ai_response)
# 显示回复
print(f"🤖 AI: {ai_response.content}\n")
except Exception as e:
print(f"❌ 出错了: {str(e)}\n")
# 出错时移除最后添加的用户消息
conversation.pop()
if __name__ == "__main__":
main()

这个版本包含了：

✅ 完整的对话历史管理
✅ 系统提示词设置
✅ 异常处理
✅ 友好的交互界面

直接运行就能用！

十一、常见问题答疑

Q1: State可以存储什么类型的数据？

A: 理论上任何Python对象都行：字典、列表、自定义类、甚至函数。但建议用简单的数据结构，方便序列化和持久化。

Q2: 节点函数可以不返回任何东西吗？

A: 可以。如果节点只是做一些副作用操作（比如打印日志），可以返回空字典{}或者None。

Q3: 怎么调试State的变化？

A: 在每个节点函数里加print(state)，就能看到每一步State的变化情况。

Q4: TypedDict是必须的吗？

A: 不是必须的，但强烈推荐。它能帮你在开发阶段就发现很多类型错误。

Q5: 对话历史会越来越长，怎么办？

A: 这是个好问题。可以定期裁剪历史，只保留最近N条消息。或者用摘要技术压缩历史信息。后面的文章会详细讲。

十二、总结

今天我们深入学习了LangGraph的核心——State管理：

State是什么：在整个流程中共享的数据容器
基本用法：节点读取State，处理后返回更新的部分
TypedDict：让代码更严谨，避免运行时错误
Reducer机制：控制State的更新方式（覆盖、追加等）
add_messages：智能管理消息列表的专用Reducer
实战应用：构建带记忆的聊天机器人

掌握了这些，你就能灵活构建各种LangGraph应用了。

如何学习大模型 AI ？

由于新岗位的生产效率，要优于被取代岗位的生产效率，所以实际上整个社会的生产效率是提升的。

但是具体到个人，只能说是：

“最先掌握AI的人，将会比较晚掌握AI的人有竞争优势”。

这句话，放在计算机、互联网、移动互联网的开局时期，都是一样的道理。

我在一线互联网企业工作十余年里，指导过不少同行后辈。帮助很多人得到了学习和成长。

我意识到有很多经验和知识值得分享给大家，也可以通过我们的能力和经验解答大家在人工智能学习中的很多困惑，所以在工作繁忙的情况下还是坚持各种整理和分享。但苦于知识传播途径有限，很多互联网行业朋友无法获得正确的资料得到学习提升，故此将并将重要的AI大模型资料包括AI大模型入门学习思维导图、精品AI大模型学习书籍手册、视频教程、实战学习等录播视频免费分享出来。

这份完整版的大模型 AI 学习资料已经上传优快云，朋友们如果需要可以微信扫描下方优快云官方认证二维码免费领取【保证100%免费】