LangGraph 基础图创建思路

LangGraph 是一个用于构建有状态、多参与者应用的工作流库。创建基础图的思路可以概括为以下步骤:

1. 定义状态(State)

状态是工作流中流动的数据容器,通常用 TypedDict 或 Pydantic 模型定义:

from typing import TypedDict, Annotated

from langgraph.graph import StateGraph, END

class State(TypedDict):

messages: Annotated[list, "对话消息列表"]

summary: str

step: int

2. 创建状态图实例

workflow = StateGraph(State) # 传入状态类型

3. 定义节点函数

节点是工作流的基本执行单元,每个节点:

  • 接收当前状态作为输入
  • 执行特定逻辑
  • 返回状态更新(可以是部分更新)

def node1(state: State):

print(f"节点1: 处理消息 {state['messages']}")

return {"step": 1} # 只更新step字段

def node2(state: State):

return {"step": 2, "summary": "处理完成"}

4. 添加节点到图中

workflow.add_node("node_1", node1)

workflow.add_node("node_2", node2)

5. 设置入口点

workflow.set_entry_point("node_1") # 指定从哪里开始

6. 连接节点(添加边)

定义节点间的流转关系:

简单线性流

workflow.add_edge("node_1", "node_2")

workflow.add_edge("node_2", END) # 结束

或使用条件边

workflow.add_conditional_edges(

"node_1",

decide_next, # 决策函数

{"continue": "node_2", "end": END}

)

7. 编译和执行

编译图

app = workflow.compile()

执行

result = app.invoke({"messages": [], "summary": "", "step": 0})

完整示例

from typing import TypedDict

from langgraph.graph import StateGraph, END

1. 定义状态

class MyState(TypedDict):

input: str

processed: str

step: int

2. 创建图

graph = StateGraph(MyState)

3. 定义节点

def input_node(state):

print(f"输入: {state['input']}")

return {"step": 1}

def process_node(state):

processed = state['input'].upper()

return {"processed": processed, "step": 2}

4. 添加节点

graph.add_node("input_node", input_node)

graph.add_node("process_node", process_node)

5. 设置入口

graph.set_entry_point("input_node")

6. 连接节点

graph.add_edge("input_node", "process_node")

graph.add_edge("process_node", END)

7. 编译执行

app = graph.compile()

result = app.invoke({"input": "hello", "processed": "", "step": 0})

print(result) # {'input': 'hello', 'processed': 'HELLO', 'step': 2}

示例执行结果

C:\Users\xiayu\miniconda3\envs\langchain03\python.exe C:\Users\xiayu\PyCharmMiscProject\AI-Agent-Dev-Practices-Code\第8章代码\test.py

输入: hello

{'input': 'hello', 'processed': 'HELLO', 'step': 2}

Process finished with exit code 0

设计思路要点

状态驱动:所有数据都通过状态传递

函数式更新:节点返回状态的部分更新,系统会自动合并

声明式连接:用边明确声明节点间关系

可组合:小图可以组合成大图

可视化:可以导出为可视化图表

常见模式

线性流

A → B → C → END

分支流

→ B

A →

→ C

循环流

A → B → (条件判断) → A 或 END

并行流

→ B

A → → D

→ C

这种设计让复杂的工作流变得清晰、可维护,特别适合构建AI应用、数据处理流水线等场景。

复制代码
相关推荐
Li emily3 小时前
解决了加密货币api多币种订阅时的数据乱序问题
人工智能·python·api·fastapi
山川绿水3 小时前
bugku——PWN——overflow2
人工智能·web安全·网络安全
程序员cxuan3 小时前
微信读书官方发了 skills,把我给秀麻了。
人工智能·后端·程序员
fake_ss1984 小时前
AI时代学习全栈项目开发的新范式
java·人工智能·学习·架构·个人开发·学习方法
nassi_4 小时前
对AI工程问题的一些思考
大数据·人工智能·hadoop
AI技术控4 小时前
《Transformers are Inherently Succinct》论文解读:从“能表达什么”到“多紧凑地表达”
人工智能·python·深度学习·机器学习·自然语言处理
蔡俊锋4 小时前
AI记忆压缩术:从305GB到7.4GB的魔法
人工智能·ai·ai 记忆
Upsy-Daisy4 小时前
AI Agent 项目学习笔记(二):Spring AI 与 ChatClient 主链路解析
人工智能·笔记·学习
zhangxingchao4 小时前
AI应用开发六:企业知识库
前端·人工智能·后端
Terrence Shen5 小时前
关于传统软件工程后端技术和当代AI智能体agent构建的harness engineering的一点思考
人工智能·软件工程