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"]。模型生成内容存在随机性,实际笑话可能与示例不同。

相关推荐
LOVE️YOU9 分钟前
Python 在定义函数时,究竟定义的是什么?
python
Maiko Star20 分钟前
* LangChain 提示词模板详解:ChatPromptTemplate 的使用与高级特性
java·人工智能·langchain
段一凡-华北理工大学21 分钟前
大模型应用开发 100 天:Python + LLM 从入门到精通 day26~Prompt 调试与优化——A/B 测试与效果评估
windows·python·大模型·prompt·智能体·提示词工程·高炉智能化
Minecraft红客23 分钟前
以撒的结合
python·游戏·电脑·娱乐
Circ.34 分钟前
Python 调试 Litellm 本地大模型接口:模型列表与对话接口实操踩坑记录
开发语言·python
风早爽太1 小时前
用 ‌Render‌ 快速部署网站和常见问题解决
python·fastapi
波力海苔夹心脆6751 小时前
C# 序列化与反序列化详解:System.Text.Json、Newtonsoft.Json、XmlSerializer 用法、特性选项与安全实践
经验分享·后端·c#·json·.net
JavaGuide2 小时前
NVIDIA 又开源了!这次给 AI Agent 加上权限管控
前端·后端
excel2 小时前
prisma 如何处理数据库竞态
前端·数据库·后端