流式传输 API¶
LangGraph SDK 允许您从 LangGraph API 服务器流式输出。
注意
LangGraph SDK 和 LangGraph 服务器是 LangGraph 平台的一部分。
基本用法¶
基本用法示例
from langgraph_sdk import get_client
client = get_client(url=<DEPLOYMENT_URL>, api_key=<API_KEY>)
# Using the graph deployed with the name "agent"
assistant_id = "agent"
# create a thread
thread = await client.threads.create()
thread_id = thread["thread_id"]
# create a streaming run
async for chunk in client.runs.stream(
thread_id,
assistant_id,
input=inputs,
stream_mode="updates"
):
print(chunk.data)
import { Client } from "@langchain/langgraph-sdk";
const client = new Client({ apiUrl: <DEPLOYMENT_URL>, apiKey: <API_KEY> });
// Using the graph deployed with the name "agent"
const assistantID = "agent";
// create a thread
const thread = await client.threads.create();
const threadID = thread["thread_id"];
// create a streaming run
const streamResponse = client.runs.stream(
threadID,
assistantID,
{
input,
streamMode: "updates"
}
);
for await (const chunk of streamResponse) {
console.log(chunk.data);
}
扩展示例:流式更新
这是一个可以在 LangGraph API 服务器中运行的示例图表。有关更多详细信息,请参阅LangGraph 平台快速入门。
# graph.py
from typing import TypedDict
from langgraph.graph import StateGraph, START, END
class State(TypedDict):
topic: str
joke: str
def refine_topic(state: State):
return {"topic": state["topic"] + " and cats"}
def generate_joke(state: State):
return {"joke": f"This is a joke about {state['topic']}"}
graph = (
StateGraph(State)
.add_node(refine_topic)
.add_node(generate_joke)
.add_edge(START, "refine_topic")
.add_edge("refine_topic", "generate_joke")
.add_edge("generate_joke", END)
.compile()
)
一旦您拥有一个正在运行的 LangGraph API 服务器,就可以使用LangGraph SDK与它进行交互
from langgraph_sdk import get_client
client = get_client(url=<DEPLOYMENT_URL>)
# Using the graph deployed with the name "agent"
assistant_id = "agent"
# create a thread
thread = await client.threads.create()
thread_id = thread["thread_id"]
# create a streaming run
async for chunk in client.runs.stream( # (1)!
thread_id,
assistant_id,
input={"topic": "ice cream"},
stream_mode="updates" # (2)!
):
print(chunk.data)
client.runs.stream()
方法返回一个迭代器,用于生成流式输出。- 设置
stream_mode="updates"
以仅流式传输每个节点后图状态的更新。也支持其他流模式。有关详细信息,请参阅支持的流模式。
import { Client } from "@langchain/langgraph-sdk";
const client = new Client({ apiUrl: <DEPLOYMENT_URL> });
// Using the graph deployed with the name "agent"
const assistantID = "agent";
// create a thread
const thread = await client.threads.create();
const threadID = thread["thread_id"];
// create a streaming run
const streamResponse = client.runs.stream( // (1)!
threadID,
assistantID,
{
input: { topic: "ice cream" },
streamMode: "updates" // (2)!
}
);
for await (const chunk of streamResponse) {
console.log(chunk.data);
}
client.runs.stream()
方法返回一个迭代器,用于生成流式输出。- 设置
streamMode: "updates"
以仅流式传输每个节点后图状态的更新。也支持其他流模式。有关详细信息,请参阅支持的流模式。
支持的流模式¶
模式 | 描述 | LangGraph 库方法 |
---|---|---|
values |
在每个 super-step 后流式传输完整的图状态。 | .stream() / .astream() 与 stream_mode="values" |
updates |
在图的每个步骤后流式传输状态更新。如果同一步骤中发生多次更新(例如,运行多个节点),则这些更新会单独流式传输。 | .stream() / .astream() 与 stream_mode="updates" |
messages-tuple |
流式传输 LLM 令牌和 LLM 被调用的图节点的元数据(适用于聊天应用)。 | .stream() / .astream() 与 stream_mode="messages" |
debug |
在图执行的整个过程中流式传输尽可能多的信息。 | .stream() / .astream() 与 stream_mode="debug" |
custom |
从图内部流式传输自定义数据 | .stream() / .astream() 与 stream_mode="custom" |
events |
流式传输所有事件(包括图的状态);主要在迁移大型 LCEL 应用时有用。 | .astream_events() |
流式传输多种模式¶
您可以将列表作为 stream_mode
参数传递,以同时流式传输多种模式。
流式输出将是 (mode, chunk)
元组,其中 mode
是流模式的名称,chunk
是该模式流式传输的数据。
流式传输图状态¶
使用 updates
和 values
流模式,以在图执行时流式传输其状态。
updates
在图的每个步骤后流式传输状态的**更新**。values
在图的每个步骤后流式传输状态的**完整值**。
图示例
from typing import TypedDict
from langgraph.graph import StateGraph, START, END
class State(TypedDict):
topic: str
joke: str
def refine_topic(state: State):
return {"topic": state["topic"] + " and cats"}
def generate_joke(state: State):
return {"joke": f"This is a joke about {state['topic']}"}
graph = (
StateGraph(State)
.add_node(refine_topic)
.add_node(generate_joke)
.add_edge(START, "refine_topic")
.add_edge("refine_topic", "generate_joke")
.add_edge("generate_joke", END)
.compile()
)
有状态运行
以下示例假设您希望将流式运行的**输出持久化**到 checkpointer 数据库中,并且已经创建了一个线程。要创建线程
如果您不需要持久化运行的输出,您可以在流式传输时传递 None
而不是 thread_id
。
使用此方式仅流式传输每个步骤后由节点返回的**状态更新**。流式输出包括节点的名称和更新。
使用此方式流式传输每个步骤后图的**完整状态**。
子图¶
要在流式输出中包含来自子图的输出,您可以在父图的 .stream()
方法中设置 subgraphs=True
。这将流式传输来自父图和任何子图的输出。
for chunk in client.runs.stream(
thread_id,
assistant_id,
input={"foo": "foo"},
stream_subgraphs=True, # (1)!
stream_mode="updates",
):
print(chunk)
- 设置
stream_subgraphs=True
以流式传输子图的输出。
扩展示例:从子图流式传输
这是一个可以在 LangGraph API 服务器中运行的示例图表。有关更多详细信息,请参阅LangGraph 平台快速入门。
# graph.py
from langgraph.graph import START, StateGraph
from typing import TypedDict
# Define subgraph
class SubgraphState(TypedDict):
foo: str # note that this key is shared with the parent graph state
bar: str
def subgraph_node_1(state: SubgraphState):
return {"bar": "bar"}
def subgraph_node_2(state: SubgraphState):
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()
# Define parent graph
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()
一旦您拥有一个正在运行的 LangGraph API 服务器,就可以使用LangGraph SDK与它进行交互
from langgraph_sdk import get_client
client = get_client(url=<DEPLOYMENT_URL>)
# Using the graph deployed with the name "agent"
assistant_id = "agent"
# create a thread
thread = await client.threads.create()
thread_id = thread["thread_id"]
async for chunk in client.runs.stream(
thread_id,
assistant_id,
input={"foo": "foo"},
stream_subgraphs=True, # (1)!
stream_mode="updates",
):
print(chunk)
- 设置
stream_subgraphs=True
以流式传输子图的输出。
import { Client } from "@langchain/langgraph-sdk";
const client = new Client({ apiUrl: <DEPLOYMENT_URL> });
// Using the graph deployed with the name "agent"
const assistantID = "agent";
// create a thread
const thread = await client.threads.create();
const threadID = thread["thread_id"];
// create a streaming run
const streamResponse = client.runs.stream(
threadID,
assistantID,
{
input: { foo: "foo" },
streamSubgraphs: true, // (1)!
streamMode: "updates"
}
);
for await (const chunk of streamResponse) {
console.log(chunk);
}
- 设置
streamSubgraphs: true
以流式传输子图的输出。
注意,我们不仅接收节点更新,还接收命名空间,它们告诉我们正在从哪个图(或子图)流式传输。
调试¶
使用 debug
流模式,以在图执行的整个过程中流式传输尽可能多的信息。流式输出包括节点的名称和完整状态。
LLM 令牌¶
使用 messages-tuple
流模式,以从图的任何部分(包括节点、工具、子图或任务)**逐个令牌**地流式传输大型语言模型 (LLM) 输出。
来自 messages-tuple
模式的流式输出是一个 (message_chunk, metadata)
元组,其中
message_chunk
:来自 LLM 的令牌或消息段。metadata
:一个字典,包含有关图节点和 LLM 调用的详细信息。
图示例
from dataclasses import dataclass
from langchain.chat_models import init_chat_model
from langgraph.graph import StateGraph, START
@dataclass
class MyState:
topic: str
joke: str = ""
llm = init_chat_model(model="openai:gpt-4o-mini")
def call_model(state: MyState):
"""Call the LLM to generate a joke about a topic"""
llm_response = llm.invoke( # (1)!
[
{"role": "user", "content": f"Generate a joke about {state.topic}"}
]
)
return {"joke": llm_response.content}
graph = (
StateGraph(MyState)
.add_node(call_model)
.add_edge(START, "call_model")
.compile()
)
- 请注意,即使 LLM 是使用
.invoke
而不是.stream
运行的,也会发出消息事件。
async for chunk in client.runs.stream(
thread_id,
assistant_id,
input={"topic": "ice cream"},
stream_mode="messages-tuple",
):
if chunk.event != "messages":
continue
message_chunk, metadata = chunk.data # (1)!
if message_chunk["content"]:
print(message_chunk["content"], end="|", flush=True)
- “messages-tuple”流模式返回一个
(message_chunk, metadata)
元组的迭代器,其中message_chunk
是 LLM 流式传输的令牌,metadata
是一个字典,包含有关 LLM 被调用的图节点和其他信息。
const streamResponse = client.runs.stream(
threadID,
assistantID,
{
input: { topic: "ice cream" },
streamMode: "messages-tuple"
}
);
for await (const chunk of streamResponse) {
if (chunk.event !== "messages") {
continue;
}
console.log(chunk.data[0]["content"]); // (1)!
}
- “messages-tuple”流模式返回一个
(message_chunk, metadata)
元组的迭代器,其中message_chunk
是 LLM 流式传输的令牌,metadata
是一个字典,包含有关 LLM 被调用的图节点和其他信息。
过滤 LLM 令牌¶
- 要根据 LLM 调用过滤流式令牌,您可以将
tags
与 LLM 调用关联起来。 - 要仅从特定节点流式传输令牌,请使用
stream_mode="messages"
并根据流式元数据中的langgraph_node
字段过滤输出。
流式传输自定义数据¶
要发送**自定义用户定义数据**
流式传输事件¶
要流式传输所有事件,包括图的状态
无状态运行¶
如果您不想将流式运行的**输出持久化**到 checkpointer 数据库中,您可以创建一个无状态运行,而无需创建线程
from langgraph_sdk import get_client
client = get_client(url=<DEPLOYMENT_URL>, api_key=<API_KEY>)
async for chunk in client.runs.stream(
None, # (1)!
assistant_id,
input=inputs,
stream_mode="updates"
):
print(chunk.data)
- 我们传递
None
而不是thread_id
UUID。
import { Client } from "@langchain/langgraph-sdk";
const client = new Client({ apiUrl: <DEPLOYMENT_URL>, apiKey: <API_KEY> });
// create a streaming run
const streamResponse = client.runs.stream(
null, // (1)!
assistantID,
{
input,
streamMode: "updates"
}
);
for await (const chunk of streamResponse) {
console.log(chunk.data);
}
- 我们传递
None
而不是thread_id
UUID。
连接并流式传输¶
LangGraph 平台允许您连接到一个活动的后台运行并从中流式传输输出。为此,您可以使用 LangGraph SDK 的 client.runs.join_stream
方法
from langgraph_sdk import get_client
client = get_client(url=<DEPLOYMENT_URL>, api_key=<API_KEY>)
async for chunk in client.runs.join_stream(
thread_id,
run_id, # (1)!
):
print(chunk)
- 这是您要连接的现有运行的
run_id
。
输出未缓冲
当您使用 .join_stream
时,输出不会被缓冲,因此在连接之前产生的任何输出都将不会被接收。
API 参考¶
有关 API 用法和实现,请参阅 API 参考。