Skip to main content
本指南解释了使用子图的机制。子图是一个,它在另一个图中被用作节点 子图对于以下情况非常有用:
  • 构建多智能体系统
  • 在多个图中复用一组节点
  • 分布式开发:当您希望不同团队独立处理图的不同部分时,可以将每个部分定义为子图,只要遵守子图接口(输入和输出模式),父图就可以在不了解子图任何细节的情况下构建。

设置

pip install -U langgraph
为 LangGraph 开发设置 LangSmith 注册 LangSmith 以快速发现问题并改进 LangGraph 项目的性能。LangSmith 允许您使用跟踪数据来调试、测试和监控使用 LangGraph 构建的 LLM 应用程序——阅读更多关于如何开始使用 LangSmith的信息。

定义子图通信

添加子图时,需要定义父图和子图之间的通信方式:
模式使用时机状态模式
在节点内调用子图父图和子图具有不同的状态模式(无共享键),或者您需要在它们之间转换状态您编写一个包装函数,将父状态映射到子图输入,并将子图输出映射回父状态
将子图添加为节点父图和子图共享状态键——子图从与父图相同的通道读取和写入您直接将编译后的子图传递给 add_node——不需要包装函数

在节点内调用子图

当父图和子图具有不同的状态模式(无共享键)时,在节点函数内调用子图。这在您希望为多智能体系统中的每个智能体保留私有消息历史时很常见。 节点函数在调用子图之前将父状态转换为子图状态,并在返回之前将结果转换回父状态。
from typing_extensions import TypedDict
from langgraph.graph.state import StateGraph, START

class SubgraphState(TypedDict):
    bar: str

# 子图

def subgraph_node_1(state: SubgraphState):
    return {"bar": "hi! " + state["bar"]}

subgraph_builder = StateGraph(SubgraphState)
subgraph_builder.add_node(subgraph_node_1)
subgraph_builder.add_edge(START, "subgraph_node_1")
subgraph = subgraph_builder.compile()

# 父图

class State(TypedDict):
    foo: str

def call_subgraph(state: State):
    # 将状态转换为子图状态
    subgraph_output = subgraph.invoke({"bar": state["foo"]})
    # 将响应转换回父状态
    return {"foo": subgraph_output["bar"]}

builder = StateGraph(State)
builder.add_node("node_1", call_subgraph)
builder.add_edge(START, "node_1")
graph = builder.compile()
from typing_extensions import TypedDict
from langgraph.graph.state import StateGraph, START

# 定义子图
class SubgraphState(TypedDict):
    # 注意,这些键均未与父图状态共享
    bar: str
    baz: str

def subgraph_node_1(state: SubgraphState):
    return {"baz": "baz"}

def subgraph_node_2(state: SubgraphState):
    return {"bar": state["bar"] + state["baz"]}

subgraph_builder = StateGraph(SubgraphState)
subgraph_builder.add_node(subgraph_node_1)
subgraph_builder.add_node(subgraph_node_2)
subgraph_builder.add_edge(START, "subgraph_node_1")
subgraph_builder.add_edge("subgraph_node_1", "subgraph_node_2")
subgraph = subgraph_builder.compile()

# 定义父图
class ParentState(TypedDict):
    foo: str

def node_1(state: ParentState):
    return {"foo": "hi! " + state["foo"]}

def node_2(state: ParentState):
    # 将状态转换为子图状态
    response = subgraph.invoke({"bar": state["foo"]})
    # 将响应转换回父状态
    return {"foo": response["bar"]}


builder = StateGraph(ParentState)
builder.add_node("node_1", node_1)
builder.add_node("node_2", node_2)
builder.add_edge(START, "node_1")
builder.add_edge("node_1", "node_2")
graph = builder.compile()

for chunk in graph.stream({"foo": "foo"}, subgraphs=True, version="v2"):
    if chunk["type"] == "updates":
        print(chunk["ns"], chunk["data"])
() {'node_1': {'foo': 'hi! foo'}}
('node_2:577b710b-64ae-31fb-9455-6a4d4cc2b0b9',) {'subgraph_node_1': {'baz': 'baz'}}
('node_2:577b710b-64ae-31fb-9455-6a4d4cc2b0b9',) {'subgraph_node_2': {'bar': 'hi! foobaz'}}
() {'node_2': {'foo': 'hi! foobaz'}}
这是一个包含两级子图的示例:父图 -> 子图 -> 孙图。
# 孙图
from typing_extensions import TypedDict
from langgraph.graph.state import StateGraph, START, END

class GrandChildState(TypedDict):
    my_grandchild_key: str

def grandchild_1(state: GrandChildState) -> GrandChildState:
    # 注意:子图或父图的键在此处不可访问
    return {"my_grandchild_key": state["my_grandchild_key"] + ", how are you"}


grandchild = StateGraph(GrandChildState)
grandchild.add_node("grandchild_1", grandchild_1)

grandchild.add_edge(START, "grandchild_1")
grandchild.add_edge("grandchild_1", END)

grandchild_graph = grandchild.compile()

# 子图
class ChildState(TypedDict):
    my_child_key: str

def call_grandchild_graph(state: ChildState) -> ChildState:
    # 注意:父图或孙图的键在此处不可访问
    grandchild_graph_input = {"my_grandchild_key": state["my_child_key"]}
    grandchild_graph_output = grandchild_graph.invoke(grandchild_graph_input)
    return {"my_child_key": grandchild_graph_output["my_grandchild_key"] + " today?"}

child = StateGraph(ChildState)
# 我们在此传递一个函数,而不是仅传递编译后的图(`grandchild_graph`)
child.add_node("child_1", call_grandchild_graph)
child.add_edge(START, "child_1")
child.add_edge("child_1", END)
child_graph = child.compile()

# 父图
class ParentState(TypedDict):
    my_key: str

def parent_1(state: ParentState) -> ParentState:
    # 注意:子图或孙图的键在此处不可访问
    return {"my_key": "hi " + state["my_key"]}

def parent_2(state: ParentState) -> ParentState:
    return {"my_key": state["my_key"] + " bye!"}

def call_child_graph(state: ParentState) -> ParentState:
    child_graph_input = {"my_child_key": state["my_key"]}
    child_graph_output = child_graph.invoke(child_graph_input)
    return {"my_key": child_graph_output["my_child_key"]}

parent = StateGraph(ParentState)
parent.add_node("parent_1", parent_1)
# 我们在此传递一个函数,而不是仅传递编译后的图(`child_graph`)
parent.add_node("child", call_child_graph)
parent.add_node("parent_2", parent_2)

parent.add_edge(START, "parent_1")
parent.add_edge("parent_1", "child")
parent.add_edge("child", "parent_2")
parent.add_edge("parent_2", END)

parent_graph = parent.compile()

for chunk in parent_graph.stream({"my_key": "Bob"}, subgraphs=True, version="v2"):
    if chunk["type"] == "updates":
        print(chunk["ns"], chunk["data"])
() {'parent_1': {'my_key': 'hi Bob'}}
('child:2e26e9ce-602f-862c-aa66-1ea5a4655e3b', 'child_1:781bb3b1-3971-84ce-810b-acf819a03f9c') {'grandchild_1': {'my_grandchild_key': 'hi Bob, how are you'}}
('child:2e26e9ce-602f-862c-aa66-1ea5a4655e3b',) {'child_1': {'my_child_key': 'hi Bob, how are you today?'}}
() {'child': {'my_key': 'hi Bob, how are you today?'}}
() {'parent_2': {'my_key': 'hi Bob, how are you today? bye!'}}

将子图添加为节点

当父图和子图共享状态键时,您可以直接将编译后的子图传递给 add_node。不需要包装函数——子图会自动从父状态通道读取和写入。例如,在多智能体系统中,智能体通常通过共享的消息键进行通信。 SQL 智能体图 如果您的子图与父图共享状态键,可以按照以下步骤将其添加到图中:
  1. 定义子图工作流(下面示例中的 subgraph_builder)并编译它
  2. 在定义父图工作流时,将编译后的子图传递给 add_node 方法
from typing_extensions import TypedDict
from langgraph.graph.state import StateGraph, START

class State(TypedDict):
    foo: str

# 子图

def subgraph_node_1(state: State):
    return {"foo": "hi! " + state["foo"]}

subgraph_builder = StateGraph(State)
subgraph_builder.add_node(subgraph_node_1)
subgraph_builder.add_edge(START, "subgraph_node_1")
subgraph = subgraph_builder.compile()

# 父图

builder = StateGraph(State)
builder.add_node("node_1", subgraph)
builder.add_edge(START, "node_1")
graph = builder.compile()
from typing_extensions import TypedDict
from langgraph.graph.state import StateGraph, START

# 定义子图
class SubgraphState(TypedDict):
    foo: str  # 与父图状态共享
    bar: str  # SubgraphState 私有

def subgraph_node_1(state: SubgraphState):
    return {"bar": "bar"}

def subgraph_node_2(state: SubgraphState):
    # 注意,此节点使用仅在子图中可用的状态键 ('bar')
    # 并在共享状态键 ('foo') 上发送更新
    return {"foo": state["foo"] + state["bar"]}

subgraph_builder = StateGraph(SubgraphState)
subgraph_builder.add_node(subgraph_node_1)
subgraph_builder.add_node(subgraph_node_2)
subgraph_builder.add_edge(START, "subgraph_node_1")
subgraph_builder.add_edge("subgraph_node_1", "subgraph_node_2")
subgraph = subgraph_builder.compile()

# 定义父图
class ParentState(TypedDict):
    foo: str

def node_1(state: ParentState):
    return {"foo": "hi! " + state["foo"]}

builder = StateGraph(ParentState)
builder.add_node("node_1", node_1)
builder.add_node("node_2", subgraph)
builder.add_edge(START, "node_1")
builder.add_edge("node_1", "node_2")
graph = builder.compile()

for chunk in graph.stream({"foo": "foo"}, version="v2"):
    if chunk["type"] == "updates":
        print(chunk["data"])
{'node_1': {'foo': 'hi! foo'}}
{'node_2': {'foo': 'hi! foobar'}}

子图持久化

使用子图时,您需要决定其内部数据在调用之间如何处理。考虑一个委托给专业子智能体的客户支持机器人:“计费专家”子智能体是应该记住客户的早期问题,还是每次调用时都重新开始? .compile() 上的 checkpointer 参数控制子图持久化:
模式checkpointer=行为
每次调用None (默认)每次调用都重新开始,并继承父图的检查点以支持单次调用内的中断持久执行
每个线程True状态在同一线程上的多次调用中累积。每次调用从上一次调用结束的地方继续。
无状态False完全不进行检查点——像普通函数调用一样运行。不支持中断或持久执行。
每次调用是大多数应用程序的正确选择,包括子智能体处理独立请求的多智能体系统。当子智能体需要多轮对话记忆时(例如,一个研究助手在几次交流中建立上下文),请使用每个线程。
父图必须使用检查点进行编译,子图持久化功能(中断、状态检查、每个线程内存)才能工作。请参阅持久化
下面的示例使用 LangChain 的 create_agent,这是构建智能体的常用方法。create_agent 在底层生成一个 LangGraph 图,因此所有子图持久化概念都直接适用。如果您使用原始的 LangGraph StateGraph 构建,相同的模式和配置选项也适用——有关详细信息,请参阅图 API

有状态

有状态子图继承父图的检查点,从而支持中断持久执行和状态检查。两种有状态模式在状态保留时间上有所不同。

每次调用(默认)

这是大多数应用程序的推荐模式,包括子智能体作为工具调用的多智能体系统。它支持中断、持久执行和并行调用,同时保持每次调用隔离。
当每次对子图的调用都是独立的,并且子智能体不需要记住任何先前调用的内容时,使用每次调用持久化。这是最常见的模式,特别是对于处理一次性请求(如“查找此客户的订单”或“总结此文档”)的多智能体系统。 省略 checkpointer 或将其设置为 None。每次调用都重新开始,但在单次调用内,子图继承父图的检查点,并可以使用 interrupt() 暂停和恢复。 以下示例使用两个子智能体(水果专家、蔬菜专家)包装为外部智能体的工具:
from langchain.agents import create_agent
from langchain.tools import tool
from langgraph.checkpoint.memory import MemorySaver
from langgraph.types import Command, interrupt

@tool
def fruit_info(fruit_name: str) -> str:
    """查找水果信息。"""
    return f"关于 {fruit_name} 的信息"

@tool
def veggie_info(veggie_name: str) -> str:
    """查找蔬菜信息。"""
    return f"关于 {veggie_name} 的信息"

# 子智能体 - 未设置检查点(继承父图)
fruit_agent = create_agent(
    model="gpt-4.1-mini",
    tools=[fruit_info],
    prompt="您是水果专家。使用 fruit_info 工具。用一句话回答。",
)

veggie_agent = create_agent(
    model="gpt-4.1-mini",
    tools=[veggie_info],
    prompt="您是蔬菜专家。使用 veggie_info 工具。用一句话回答。",
)

# 将子智能体包装为外部智能体的工具
@tool
def ask_fruit_expert(question: str) -> str:
    """询问水果专家。用于所有水果问题。"""
    response = fruit_agent.invoke(
        {"messages": [{"role": "user", "content": question}]},
    )
    return response["messages"][-1].content

@tool
def ask_veggie_expert(question: str) -> str:
    """询问蔬菜专家。用于所有蔬菜问题。"""
    response = veggie_agent.invoke(
        {"messages": [{"role": "user", "content": question}]},
    )
    return response["messages"][-1].content

# 带检查点的外部智能体
agent = create_agent(
    model="gpt-4.1-mini",
    tools=[ask_fruit_expert, ask_veggie_expert],
    prompt=(
        "您有两个专家:ask_fruit_expert 和 ask_veggie_expert。"
        "始终将问题委托给相应的专家。"
    ),
    checkpointer=MemorySaver(),
)
每次调用都可以使用 interrupt() 暂停和恢复。在工具函数中添加 interrupt() 以在继续之前需要用户批准:
@tool
def fruit_info(fruit_name: str) -> str:
    """查找水果信息。"""
    interrupt("继续?")
    return f"关于 {fruit_name} 的信息"
config = {"configurable": {"thread_id": "1"}}

# 调用 - 子智能体的工具调用 interrupt()
response = agent.invoke(
    {"messages": [{"role": "user", "content": "告诉我关于苹果的信息"}]},
    config=config,
)
# response 包含 __interrupt__

# 恢复 - 批准中断
response = agent.invoke(Command(resume=True), config=config)
# 子智能体消息计数:4

每个线程

当子智能体需要记住先前的交互时,使用每个线程持久化。例如,一个研究助手在几次交流中建立上下文,或者一个编码助手跟踪其已编辑的文件。子智能体的对话历史和数据在同一线程上的多次调用中累积。每次调用从上一次调用结束的地方继续。 使用 checkpointer=True 编译以启用此行为。
每个线程子图不支持并行工具调用。当 LLM 可以访问每个线程子智能体作为工具时,它可能会尝试并行多次调用该工具(例如,同时询问水果专家关于苹果和香蕉的信息)。这会导致检查点冲突,因为两个调用都写入相同的命名空间。下面的示例使用 LangChain 的 ToolCallLimitMiddleware 来防止这种情况。如果您使用纯 LangGraph StateGraph 构建,您需要自己防止并行工具调用——例如,通过配置模型禁用并行工具调用,或添加逻辑确保同一子图不会被并行多次调用。
以下示例使用使用 checkpointer=True 编译的水果专家子智能体:
from langchain.agents import create_agent
from langchain.agents.middleware import ToolCallLimitMiddleware
from langchain.tools import tool
from langgraph.checkpoint.memory import MemorySaver
from langgraph.types import Command, interrupt

@tool
def fruit_info(fruit_name: str) -> str:
    """查找水果信息。"""
    return f"关于 {fruit_name} 的信息"

# 带 checkpointer=True 的子智能体,用于持久状态
fruit_agent = create_agent(
    model="gpt-4.1-mini",
    tools=[fruit_info],
    prompt="您是水果专家。使用 fruit_info 工具。用一句话回答。",
    checkpointer=True,
)

# 将子智能体包装为外部智能体的工具
@tool
def ask_fruit_expert(question: str) -> str:
    """询问水果专家。用于所有水果问题。"""
    response = fruit_agent.invoke(
        {"messages": [{"role": "user", "content": question}]},
    )
    return response["messages"][-1].content

# 带检查点的外部智能体
# 使用 ToolCallLimitMiddleware 防止对每个线程子智能体的并行调用,
# 这会导致检查点冲突。
agent = create_agent(
    model="gpt-4.1-mini",
    tools=[ask_fruit_expert],
    prompt="您有一个水果专家。始终将水果问题委托给 ask_fruit_expert。",
    middleware=[
        ToolCallLimitMiddleware(tool_name="ask_fruit_expert", run_limit=1),
    ],
    checkpointer=MemorySaver(),
)
每个线程子智能体支持 interrupt(),就像每次调用一样。在工具函数中添加 interrupt() 以需要用户批准:
@tool
def fruit_info(fruit_name: str) -> str:
    """查找水果信息。"""
    interrupt("继续?")
    return f"关于 {fruit_name} 的信息"
config = {"configurable": {"thread_id": "1"}}

# 调用 - 子智能体的工具调用 interrupt()
response = agent.invoke(
    {"messages": [{"role": "user", "content": "告诉我关于苹果的信息"}]},
    config=config,
)
# response 包含 __interrupt__

# 恢复 - 批准中断
response = agent.invoke(Command(resume=True), config=config)
# 子智能体消息计数:4

无状态

当您希望像普通函数调用一样运行子智能体而没有检查点开销时,使用此模式。子图无法暂停/恢复,并且不受益于持久执行。使用 checkpointer=False 编译。
没有检查点,子图就没有持久执行。如果进程在运行过程中崩溃,子图无法恢复,必须从头开始重新运行。
subgraph_builder = StateGraph(...)
subgraph = subgraph_builder.compile(checkpointer=False)

检查点参考

使用 .compile() 上的 checkpointer 参数控制子图持久化:
subgraph = builder.compile(checkpointer=False)  # 或 True / None
功能每次调用(默认)每个线程无状态
checkpointer=NoneTrueFalse
中断 (HITL)
多轮记忆
多次调用(不同子图)
多次调用(同一子图)
状态检查
  • 中断 (HITL):子图可以使用 interrupt() 暂停执行并等待用户输入,然后从暂停处恢复。
  • 多轮记忆:子图在同一线程内的多次调用中保留其状态。每次调用从上一次调用结束的地方继续,而不是重新开始。
  • 多次调用(不同子图):可以在单个节点内调用多个不同的子图实例,而不会出现检查点命名空间冲突。
  • 多次调用(同一子图):可以在单个节点内多次调用同一子图实例。使用有状态持久化时,这些调用会写入相同的检查点命名空间并发生冲突——请改用每次调用持久化。
  • 状态检查:子图的状态可通过 get_state(config, subgraphs=True) 用于调试和监控。

查看子图状态

启用持久化后,您可以使用子图选项检查子图状态。使用无状态检查点(checkpointer=False)时,不会保存子图检查点,因此子图状态不可用。
查看子图状态要求 LangGraph 能够静态发现子图——即,它被添加为节点在节点内调用。当子图在工具函数或其他间接调用(例如,子智能体模式)内调用时,此功能无效。无论嵌套如何,中断都会传播到顶层图。
返回当前调用的子图状态。每次调用都重新开始。
from langgraph.graph import START, StateGraph
from langgraph.checkpoint.memory import MemorySaver
from langgraph.types import interrupt, Command
from typing_extensions import TypedDict

class State(TypedDict):
    foo: str

# 子图
def subgraph_node_1(state: State):
    value = interrupt("提供值:")
    return {"foo": state["foo"] + value}

subgraph_builder = StateGraph(State)
subgraph_builder.add_node(subgraph_node_1)
subgraph_builder.add_edge(START, "subgraph_node_1")
subgraph = subgraph_builder.compile()  # 继承父图检查点

# 父图
builder = StateGraph(State)
builder.add_node("node_1", subgraph)
builder.add_edge(START, "node_1")

checkpointer = MemorySaver()
graph = builder.compile(checkpointer=checkpointer)

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

graph.invoke({"foo": ""}, config)

# 查看当前调用的子图状态
subgraph_state = graph.get_state(config, subgraphs=True).tasks[0].state  

# 恢复子图
graph.invoke(Command(resume="bar"), config)

流式传输子图输出

要将子图的输出包含在流式输出中,您可以在父图的流方法中设置子图选项。这将流式传输父图和任何子图的输出。
使用 version="v2",子图事件使用相同的 StreamPart 格式。ns 字段标识源图:
for chunk in graph.stream(
    {"foo": "foo"},
    subgraphs=True,
    stream_mode="updates",
    version="v2",
):
    print(chunk["type"])  # "updates"
    print(chunk["ns"])    # () 表示根,("node_2:<task_id>",) 表示子图
    print(chunk["data"])  # {"node_name": {"key": "value"}}
from typing_extensions import TypedDict
from langgraph.graph.state import StateGraph, START

# 定义子图
class SubgraphState(TypedDict):
    foo: str
    bar: str

def subgraph_node_1(state: SubgraphState):
    return {"bar": "bar"}

def subgraph_node_2(state: SubgraphState):
    # 注意,此节点使用仅在子图中可用的状态键 ('bar')
    # 并在共享状态键 ('foo') 上发送更新
    return {"foo": state["foo"] + state["bar"]}

subgraph_builder = StateGraph(SubgraphState)
subgraph_builder.add_node(subgraph_node_1)
subgraph_builder.add_node(subgraph_node_2)
subgraph_builder.add_edge(START, "subgraph_node_1")
subgraph_builder.add_edge("subgraph_node_1", "subgraph_node_2")
subgraph = subgraph_builder.compile()

# 定义父图
class ParentState(TypedDict):
    foo: str

def node_1(state: ParentState):
    return {"foo": "hi! " + state["foo"]}

builder = StateGraph(ParentState)
builder.add_node("node_1", node_1)
builder.add_node("node_2", subgraph)
builder.add_edge(START, "node_1")
builder.add_edge("node_1", "node_2")
graph = builder.compile()

for chunk in graph.stream(
    {"foo": "foo"},
    stream_mode="updates",
    subgraphs=True,
    version="v2",
):
    if chunk["type"] == "updates":
        print(chunk["ns"], chunk["data"])
() {'node_1': {'foo': 'hi! foo'}}
('node_2:e58e5673-a661-ebb0-70d4-e298a7fc28b7',) {'subgraph_node_1': {'bar': 'bar'}}
('node_2:e58e5673-a661-ebb0-70d4-e298a7fc28b7',) {'subgraph_node_2': {'foo': 'hi! foobar'}}
() {'node_2': {'foo': 'hi! foobar'}}