LangGraph 入门实战(11)--输出模式

LangGraph 流式输出:values、updates、messages 和 debug

LangGraph 的 graph.stream() 可以边执行节点边返回数据。通过 stream_mode,我们可以选择查看完整状态、节点增量、模型消息流或详细调试信息。

1. 安装与配置

bash 复制代码
pip install -U langgraph langchain-deepseek
export DEEPSEEK_API_KEY="你的 API Key"
python index.py

2. 构建工作流

下面创建一个简单流程:先把"小狗"加入主题,再让 DeepSeek 根据新主题生成笑话。

python 复制代码
import os
from typing import Any, TypedDict, cast

from langchain_core.messages import BaseMessage
from langchain_deepseek import ChatDeepSeek
from langgraph.graph import START, StateGraph
from pydantic import SecretStr


deepseek = ChatDeepSeek(
    model="deepseek-v4-flash",
    temperature=0,
    base_url="https://api.deepseek.com",
    api_key=SecretStr(os.environ["DEEPSEEK_API_KEY"]),
)


class State(TypedDict):
    topic: str
    joke: str


def refine_topic(state: State):
    return {"topic": state["topic"] + "和小狗"}


def generate_joke(state: State):
    response = deepseek.invoke(
        [{"role": "user", "content": f"请生成一个关于{state['topic']}的笑话"}]
    )
    return {"joke": response.content}


graph = (
    StateGraph(State)
    .add_node(refine_topic)
    .add_node(generate_joke)
    .add_edge(START, "refine_topic")
    .add_edge("refine_topic", "generate_joke")
    .compile()
)

initial_state: State = {"topic": "小猫", "joke": ""}

执行顺序如下:

text 复制代码
START -> refine_topic -> generate_joke -> END

3. 四种输出模式

模式 返回内容 适用场景
values 每一步合并后的完整状态 观察状态变化
updates 当前节点更新的字段 监听节点结果
messages 模型消息片段和元数据 实现打字机效果
debug 节点输入、结果、时间和任务 ID 排查工作流问题

values:完整状态

python 复制代码
for event in graph.stream(initial_state, stream_mode="values"):
    print(event)

输出:

text 复制代码
{'topic': '小猫', 'joke': ''}
{'topic': '小猫和小狗', 'joke': ''}
{'topic': '小猫和小狗', 'joke': '小猫和小狗去讲笑话,结果它们都笑了。'}

updates:节点增量

python 复制代码
for event in graph.stream(initial_state, stream_mode="updates"):
    print(event)

输出:

text 复制代码
{'refine_topic': {'topic': '小猫和小狗'}}
{'generate_joke': {'joke': '小猫和小狗去讲笑话,结果它们都笑了。'}}

messages:模型消息流

messages 返回 (message, metadata)。下面用 标记每个消息片段的边界:

python 复制代码
for event in graph.stream(initial_state, stream_mode="messages"):
    message, metadata = cast(tuple[BaseMessage, dict[str, Any]], event)
    if message.content:
        print(message.content, end="|", flush=True)
print()

输出:

text 复制代码
小猫|和小狗|去讲笑话|,结果它们都笑了。|

LangGraph 当前的类型声明无法根据 messages 自动推断事件结构,所以这里用 cast 告诉 Pylance 返回值是"消息与元数据"元组。消息片段的实际切分位置由模型决定,metadata["langgraph_node"] 可以用来判断消息来自哪个节点。

debug:调试信息

python 复制代码
for event in graph.stream(initial_state, stream_mode="debug"):
    print(event)

精简输出如下,实际结果还包含时间、任务 ID 和触发来源:

text 复制代码
{'step': 1, 'type': 'task',
 'payload': {'name': 'refine_topic', 'input': {'topic': '小猫', 'joke': ''}}}

{'step': 1, 'type': 'task_result',
 'payload': {'name': 'refine_topic', 'result': {'topic': '小猫和小狗'}}}

小结

日常展示状态用 values,只关心节点变更用 updates,向页面实时输出模型内容用 messages,排查节点执行问题时使用 debug

示例中的四次 graph.stream() 会完整执行四遍工作流,因此也会调用模型四次。实际项目只需选择需要的模式;如果希望一次获得多种结果,也可以传入 stream_mode=["updates", "messages"]。模型生成内容存在随机性,实际笑话可能与示例不同。

相关推荐
雪碧聊技术2 小时前
安装Python(保姆级教程)
python·版本更新·python下载
demo007x2 小时前
Hermes-Agent 技术架构
前端·后端·agent
__zRainy__4 小时前
Node系列 · Node基础:net 模块
后端·node.js
Diligently_4 小时前
Anolis 系统更新与密码设置&磁盘扩容
java·后端·spring
kyson_4 小时前
大文件切片上传:从“能传”到“可靠”的 Vue 3 + Java 全栈实现
后端
张龙6874 小时前
uv 全面指南:比 pip 快 100 倍的 Python 包管理器,从安装到团队落地
后端·python
@HNUSTer4 小时前
Python数据可视化科技图表绘制系列教程(八)
python·数据可视化·科技论文·专业制图·科研图表
用户40966601317515 小时前
Guava 不止有 Lists 和 Maps:Cache、EventBus、Multimap 实战
java·后端