Files
langgraph/docs/docs/how-tos/streaming.md
T
d59091672f feat: add docs translations (#5552)
Co-authored-by: Eugene Yurtsev <eyurtsev@gmail.com>
Co-authored-by: Tat Dat Duong <david@duong.cz>
2025-07-30 02:18:30 +00:00

55 KiB

Stream outputs

You can stream outputs from a LangGraph agent or workflow.

Supported stream modes

:::python Pass one or more of the following stream modes as a list to the @[stream()][CompiledStateGraph.stream] or @[astream()][CompiledStateGraph.astream] methods: :::

:::js Pass one or more of the following stream modes as a list to the @[stream()][CompiledStateGraph.stream] method: :::

Mode Description
values Streams the full value of the state after each step of the graph.
updates Streams the updates to the state after each step of the graph. If multiple updates are made in the same step (e.g., multiple nodes are run), those updates are streamed separately.
custom Streams custom data from inside your graph nodes.
messages Streams 2-tuples (LLM token, metadata) from any graph nodes where an LLM is invoked.
debug Streams as much information as possible throughout the execution of the graph.

Stream from an agent

Agent progress

:::python To stream agent progress, use the @[stream()][CompiledStateGraph.stream] or @[astream()][CompiledStateGraph.astream] methods with stream_mode="updates". This emits an event after every agent step. :::

:::js To stream agent progress, use the @[stream()][CompiledStateGraph.stream] method with streamMode: "updates". This emits an event after every agent step. :::

For example, if you have an agent that calls a tool once, you should see the following updates:

  • LLM node: AI message with tool call requests
  • Tool node: Tool message with execution result
  • LLM node: Final AI response

:::python === "Sync"

```python
agent = create_react_agent(
    model="anthropic:claude-3-7-sonnet-latest",
    tools=[get_weather],
)
# highlight-next-line
for chunk in agent.stream(
    {"messages": [{"role": "user", "content": "what is the weather in sf"}]},
    # highlight-next-line
    stream_mode="updates"
):
    print(chunk)
    print("\n")
```

=== "Async"

```python
agent = create_react_agent(
    model="anthropic:claude-3-7-sonnet-latest",
    tools=[get_weather],
)
# highlight-next-line
async for chunk in agent.astream(
    {"messages": [{"role": "user", "content": "what is the weather in sf"}]},
    # highlight-next-line
    stream_mode="updates"
):
    print(chunk)
    print("\n")
```

:::

:::js

const agent = createReactAgent({
  llm: model,
  tools: [getWeather],
});

for await (const chunk of await agent.stream(
  { messages: [{ role: "user", content: "what is the weather in sf" }] },
  { streamMode: "updates" }
)) {
  console.log(chunk);
  console.log("\n");
}

:::

LLM tokens

:::python To stream tokens as they are produced by the LLM, use stream_mode="messages":

=== "Sync"

```python
agent = create_react_agent(
    model="anthropic:claude-3-7-sonnet-latest",
    tools=[get_weather],
)
# highlight-next-line
for token, metadata in agent.stream(
    {"messages": [{"role": "user", "content": "what is the weather in sf"}]},
    # highlight-next-line
    stream_mode="messages"
):
    print("Token", token)
    print("Metadata", metadata)
    print("\n")
```

=== "Async"

```python
agent = create_react_agent(
    model="anthropic:claude-3-7-sonnet-latest",
    tools=[get_weather],
)
# highlight-next-line
async for token, metadata in agent.astream(
    {"messages": [{"role": "user", "content": "what is the weather in sf"}]},
    # highlight-next-line
    stream_mode="messages"
):
    print("Token", token)
    print("Metadata", metadata)
    print("\n")
```

:::

:::js To stream tokens as they are produced by the LLM, use streamMode: "messages":

const agent = createReactAgent({
  llm: model,
  tools: [getWeather],
});

for await (const [token, metadata] of await agent.stream(
  { messages: [{ role: "user", content: "what is the weather in sf" }] },
  { streamMode: "messages" }
)) {
  console.log("Token", token);
  console.log("Metadata", metadata);
  console.log("\n");
}

:::

Tool updates

:::python To stream updates from tools as they are executed, you can use @[get_stream_writer][get_stream_writer].

=== "Sync"

```python
# highlight-next-line
from langgraph.config import get_stream_writer

def get_weather(city: str) -> str:
    """Get weather for a given city."""
    # highlight-next-line
    writer = get_stream_writer()
    # stream any arbitrary data
    # highlight-next-line
    writer(f"Looking up data for city: {city}")
    return f"It's always sunny in {city}!"

agent = create_react_agent(
    model="anthropic:claude-3-7-sonnet-latest",
    tools=[get_weather],
)

for chunk in agent.stream(
    {"messages": [{"role": "user", "content": "what is the weather in sf"}]},
    # highlight-next-line
    stream_mode="custom"
):
    print(chunk)
    print("\n")
```

=== "Async"

```python
# highlight-next-line
from langgraph.config import get_stream_writer

def get_weather(city: str) -> str:
    """Get weather for a given city."""
    # highlight-next-line
    writer = get_stream_writer()
    # stream any arbitrary data
    # highlight-next-line
    writer(f"Looking up data for city: {city}")
    return f"It's always sunny in {city}!"

agent = create_react_agent(
    model="anthropic:claude-3-7-sonnet-latest",
    tools=[get_weather],
)

async for chunk in agent.astream(
    {"messages": [{"role": "user", "content": "what is the weather in sf"}]},
    # highlight-next-line
    stream_mode="custom"
):
    print(chunk)
    print("\n")
```

!!! Note

  If you add `get_stream_writer` inside your tool, you won't be able to invoke the tool outside of a LangGraph execution context.

:::

:::js To stream updates from tools as they are executed, you can use the writer parameter from the configuration.

import { LangGraphRunnableConfig } from "@langchain/langgraph";

const getWeather = tool(
  async (input, config: LangGraphRunnableConfig) => {
    // Stream any arbitrary data
    config.writer?.("Looking up data for city: " + input.city);
    return `It's always sunny in ${input.city}!`;
  },
  {
    name: "get_weather",
    description: "Get weather for a given city.",
    schema: z.object({
      city: z.string().describe("The city to get weather for."),
    }),
  }
);

const agent = createReactAgent({
  llm: model,
  tools: [getWeather],
});

for await (const chunk of await agent.stream(
  { messages: [{ role: "user", content: "what is the weather in sf" }] },
  { streamMode: "custom" }
)) {
  console.log(chunk);
  console.log("\n");
}

!!! Note If you add the writer parameter to your tool, you won't be able to invoke the tool outside of a LangGraph execution context without providing a writer function. :::

Stream multiple modes

:::python You can specify multiple streaming modes by passing stream mode as a list: stream_mode=["updates", "messages", "custom"]:

=== "Sync"

```python
agent = create_react_agent(
    model="anthropic:claude-3-7-sonnet-latest",
    tools=[get_weather],
)

for stream_mode, chunk in agent.stream(
    {"messages": [{"role": "user", "content": "what is the weather in sf"}]},
    # highlight-next-line
    stream_mode=["updates", "messages", "custom"]
):
    print(chunk)
    print("\n")
```

=== "Async"

```python
agent = create_react_agent(
    model="anthropic:claude-3-7-sonnet-latest",
    tools=[get_weather],
)

async for stream_mode, chunk in agent.astream(
    {"messages": [{"role": "user", "content": "what is the weather in sf"}]},
    # highlight-next-line
    stream_mode=["updates", "messages", "custom"]
):
    print(chunk)
    print("\n")
```

:::

:::js You can specify multiple streaming modes by passing streamMode as an array: streamMode: ["updates", "messages", "custom"]:

const agent = createReactAgent({
  llm: model,
  tools: [getWeather],
});

for await (const chunk of await agent.stream(
  { messages: [{ role: "user", content: "what is the weather in sf" }] },
  { streamMode: ["updates", "messages", "custom"] }
)) {
  console.log(chunk);
  console.log("\n");
}

:::

Disable streaming

In some applications you might need to disable streaming of individual tokens for a given model. This is useful in multi-agent systems to control which agents stream their output.

See the Models guide to learn how to disable streaming.

Stream from a workflow

Basic usage example

:::python LangGraph graphs expose the @[.stream()][Pregel.stream] (sync) and @[.astream()][Pregel.astream] (async) methods to yield streamed outputs as iterators.

=== "Sync"

```python
for chunk in graph.stream(inputs, stream_mode="updates"):
    print(chunk)
```

=== "Async"

```python
async for chunk in graph.astream(inputs, stream_mode="updates"):
    print(chunk)
```

:::

:::js LangGraph graphs expose the @[.stream()][Pregel.stream] method to yield streamed outputs as iterators.

for await (const chunk of await graph.stream(inputs, {
  streamMode: "updates",
})) {
  console.log(chunk);
}

:::

??? example "Extended example: streaming updates"

  :::python
  ```python
  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()
  )

  # highlight-next-line
  for chunk in graph.stream( # (1)!
      {"topic": "ice cream"},
      # highlight-next-line
      stream_mode="updates", # (2)!
  ):
      print(chunk)
  ```

  1. The `stream()` method returns an iterator that yields streamed outputs.
  2. Set `stream_mode="updates"` to stream only the updates to the graph state after each node. Other stream modes are also available. See [supported stream modes](#supported-stream-modes) for details.
  :::

  :::js
  ```typescript
  import { StateGraph, START, END } from "@langchain/langgraph";
  import { z } from "zod";

  const State = z.object({
    topic: z.string(),
    joke: z.string(),
  });

  const graph = new StateGraph(State)
    .addNode("refineTopic", (state) => {
      return { topic: state.topic + " and cats" };
    })
    .addNode("generateJoke", (state) => {
      return { joke: `This is a joke about ${state.topic}` };
    })
    .addEdge(START, "refineTopic")
    .addEdge("refineTopic", "generateJoke")
    .addEdge("generateJoke", END)
    .compile();

  for await (const chunk of await graph.stream(
    { topic: "ice cream" },
    { streamMode: "updates" } // (1)!
  )) {
    console.log(chunk);
  }
  ```

  1. Set `streamMode: "updates"` to stream only the updates to the graph state after each node. Other stream modes are also available. See [supported stream modes](#supported-stream-modes) for details.
  :::

  ```output
  {'refineTopic': {'topic': 'ice cream and cats'}}
  {'generateJoke': {'joke': 'This is a joke about ice cream and cats'}}
  ```                                                                                                   |

Stream multiple modes

:::python You can pass a list as the stream_mode parameter to stream multiple modes at once.

The streamed outputs will be tuples of (mode, chunk) where mode is the name of the stream mode and chunk is the data streamed by that mode.

=== "Sync"

```python
for mode, chunk in graph.stream(inputs, stream_mode=["updates", "custom"]):
    print(chunk)
```

=== "Async"

```python
async for mode, chunk in graph.astream(inputs, stream_mode=["updates", "custom"]):
    print(chunk)
```

:::

:::js You can pass an array as the streamMode parameter to stream multiple modes at once.

The streamed outputs will be tuples of [mode, chunk] where mode is the name of the stream mode and chunk is the data streamed by that mode.

for await (const [mode, chunk] of await graph.stream(inputs, {
  streamMode: ["updates", "custom"],
})) {
  console.log(chunk);
}

:::

Stream graph state

Use the stream modes updates and values to stream the state of the graph as it executes.

  • updates streams the updates to the state after each step of the graph.
  • values streams the full value of the state after each step of the graph.

:::python

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()
)

:::

:::js

import { StateGraph, START, END } from "@langchain/langgraph";
import { z } from "zod";

const State = z.object({
  topic: z.string(),
  joke: z.string(),
});

const graph = new StateGraph(State)
  .addNode("refineTopic", (state) => {
    return { topic: state.topic + " and cats" };
  })
  .addNode("generateJoke", (state) => {
    return { joke: `This is a joke about ${state.topic}` };
  })
  .addEdge(START, "refineTopic")
  .addEdge("refineTopic", "generateJoke")
  .addEdge("generateJoke", END)
  .compile();

:::

=== "updates"

Use this to stream only the **state updates** returned by the nodes after each step. The streamed outputs include the name of the node as well as the update.

:::python
```python
for chunk in graph.stream(
    {"topic": "ice cream"},
    # highlight-next-line
    stream_mode="updates",
):
    print(chunk)
```
:::

:::js
```typescript
for await (const chunk of await graph.stream(
  { topic: "ice cream" },
  { streamMode: "updates" }
)) {
  console.log(chunk);
}
```
:::

=== "values"

Use this to stream the **full state** of the graph after each step.

:::python
```python
for chunk in graph.stream(
    {"topic": "ice cream"},
    # highlight-next-line
    stream_mode="values",
):
    print(chunk)
```
:::

:::js
```typescript
for await (const chunk of await graph.stream(
  { topic: "ice cream" },
  { streamMode: "values" }
)) {
  console.log(chunk);
}
```
:::

Stream subgraph outputs

:::python To include outputs from subgraphs in the streamed outputs, you can set subgraphs=True in the .stream() method of the parent graph. This will stream outputs from both the parent graph and any subgraphs.

The outputs will be streamed as tuples (namespace, data), where namespace is a tuple with the path to the node where a subgraph is invoked, e.g. ("parent_node:<task_id>", "child_node:<task_id>").

for chunk in graph.stream(
    {"foo": "foo"},
    # highlight-next-line
    subgraphs=True, # (1)!
    stream_mode="updates",
):
    print(chunk)
  1. Set subgraphs=True to stream outputs from subgraphs. :::

:::js To include outputs from subgraphs in the streamed outputs, you can set subgraphs: true in the .stream() method of the parent graph. This will stream outputs from both the parent graph and any subgraphs.

The outputs will be streamed as tuples [namespace, data], where namespace is a tuple with the path to the node where a subgraph is invoked, e.g. ["parent_node:<task_id>", "child_node:<task_id>"].

for await (const chunk of await graph.stream(
  { foo: "foo" },
  {
    subgraphs: true, // (1)!
    streamMode: "updates",
  }
)) {
  console.log(chunk);
}
  1. Set subgraphs: true to stream outputs from subgraphs. :::

??? example "Extended example: streaming from subgraphs"

  :::python
  ```python
  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()

  for chunk in graph.stream(
      {"foo": "foo"},
      stream_mode="updates",
      # highlight-next-line
      subgraphs=True, # (1)!
  ):
      print(chunk)
  ```

  1. Set `subgraphs=True` to stream outputs from subgraphs.
  :::

  :::js
  ```typescript
  import { StateGraph, START } from "@langchain/langgraph";
  import { z } from "zod";

  // Define subgraph
  const SubgraphState = z.object({
    foo: z.string(), // note that this key is shared with the parent graph state
    bar: z.string(),
  });

  const subgraphBuilder = new StateGraph(SubgraphState)
    .addNode("subgraphNode1", (state) => {
      return { bar: "bar" };
    })
    .addNode("subgraphNode2", (state) => {
      return { foo: state.foo + state.bar };
    })
    .addEdge(START, "subgraphNode1")
    .addEdge("subgraphNode1", "subgraphNode2");
  const subgraph = subgraphBuilder.compile();

  // Define parent graph
  const ParentState = z.object({
    foo: z.string(),
  });

  const builder = new StateGraph(ParentState)
    .addNode("node1", (state) => {
      return { foo: "hi! " + state.foo };
    })
    .addNode("node2", subgraph)
    .addEdge(START, "node1")
    .addEdge("node1", "node2");
  const graph = builder.compile();

  for await (const chunk of await graph.stream(
    { foo: "foo" },
    {
      streamMode: "updates",
      subgraphs: true, // (1)!
    }
  )) {
    console.log(chunk);
  }
  ```

  1. Set `subgraphs: true` to stream outputs from subgraphs.
  :::

  :::python
  ```
  ((), {'node_1': {'foo': 'hi! foo'}})
  (('node_2:dfddc4ba-c3c5-6887-5012-a243b5b377c2',), {'subgraph_node_1': {'bar': 'bar'}})
  (('node_2:dfddc4ba-c3c5-6887-5012-a243b5b377c2',), {'subgraph_node_2': {'foo': 'hi! foobar'}})
  ((), {'node_2': {'foo': 'hi! foobar'}})
  ```
  :::

  :::js
  ```
  [[], {'node1': {'foo': 'hi! foo'}}]
  [['node2:dfddc4ba-c3c5-6887-5012-a243b5b377c2'], {'subgraphNode1': {'bar': 'bar'}}]
  [['node2:dfddc4ba-c3c5-6887-5012-a243b5b377c2'], {'subgraphNode2': {'foo': 'hi! foobar'}}]
  [[], {'node2': {'foo': 'hi! foobar'}}]
  ```
  :::

  **Note** that we are receiving not just the node updates, but we also the namespaces which tell us what graph (or subgraph) we are streaming from.

Debugging

Use the debug streaming mode to stream as much information as possible throughout the execution of the graph. The streamed outputs include the name of the node as well as the full state.

:::python

for chunk in graph.stream(
    {"topic": "ice cream"},
    # highlight-next-line
    stream_mode="debug",
):
    print(chunk)

:::

:::js

for await (const chunk of await graph.stream(
  { topic: "ice cream" },
  { streamMode: "debug" }
)) {
  console.log(chunk);
}

:::

LLM tokens

Use the messages streaming mode to stream Large Language Model (LLM) outputs token by token from any part of your graph, including nodes, tools, subgraphs, or tasks.

:::python The streamed output from messages mode is a tuple (message_chunk, metadata) where:

  • message_chunk: the token or message segment from the LLM.
  • metadata: a dictionary containing details about the graph node and LLM invocation.

If your LLM is not available as a LangChain integration, you can stream its outputs using custom mode instead. See use with any LLM for details.

!!! warning "Manual config required for async in Python < 3.11"

When using Python < 3.11 with async code, you must explicitly pass `RunnableConfig` to `ainvoke()` to enable proper streaming. See [Async with Python < 3.11](#async) for details or upgrade to Python 3.11+.
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"""
    # highlight-next-line
    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()
)

for message_chunk, metadata in graph.stream( # (2)!
    {"topic": "ice cream"},
    # highlight-next-line
    stream_mode="messages",
):
    if message_chunk.content:
        print(message_chunk.content, end="|", flush=True)
  1. Note that the message events are emitted even when the LLM is run using .invoke rather than .stream.
  2. The "messages" stream mode returns an iterator of tuples (message_chunk, metadata) where message_chunk is the token streamed by the LLM and metadata is a dictionary with information about the graph node where the LLM was called and other information. :::

:::js The streamed output from messages mode is a tuple [message_chunk, metadata] where:

  • message_chunk: the token or message segment from the LLM.
  • metadata: a dictionary containing details about the graph node and LLM invocation.

If your LLM is not available as a LangChain integration, you can stream its outputs using custom mode instead. See use with any LLM for details.

import { ChatOpenAI } from "@langchain/openai";
import { StateGraph, START } from "@langchain/langgraph";
import { z } from "zod";

const MyState = z.object({
  topic: z.string(),
  joke: z.string().default(""),
});

const llm = new ChatOpenAI({ model: "gpt-4o-mini" });

const callModel = async (state: z.infer<typeof MyState>) => {
  // Call the LLM to generate a joke about a topic
  const llmResponse = await llm.invoke([
    { role: "user", content: `Generate a joke about ${state.topic}` },
  ]); // (1)!
  return { joke: llmResponse.content };
};

const graph = new StateGraph(MyState)
  .addNode("callModel", callModel)
  .addEdge(START, "callModel")
  .compile();

for await (const [messageChunk, metadata] of await graph.stream(
  // (2)!
  { topic: "ice cream" },
  { streamMode: "messages" }
)) {
  if (messageChunk.content) {
    console.log(messageChunk.content + "|");
  }
}
  1. Note that the message events are emitted even when the LLM is run using .invoke rather than .stream.
  2. The "messages" stream mode returns an iterator of tuples [messageChunk, metadata] where messageChunk is the token streamed by the LLM and metadata is a dictionary with information about the graph node where the LLM was called and other information. :::

Filter by LLM invocation

You can associate tags with LLM invocations to filter the streamed tokens by LLM invocation.

:::python

from langchain.chat_models import init_chat_model

llm_1 = init_chat_model(model="openai:gpt-4o-mini", tags=['joke']) # (1)!
llm_2 = init_chat_model(model="openai:gpt-4o-mini", tags=['poem']) # (2)!

graph = ... # define a graph that uses these LLMs

async for msg, metadata in graph.astream(  # (3)!
    {"topic": "cats"},
    # highlight-next-line
    stream_mode="messages",
):
    if metadata["tags"] == ["joke"]: # (4)!
        print(msg.content, end="|", flush=True)
  1. llm_1 is tagged with "joke".
  2. llm_2 is tagged with "poem".
  3. The stream_mode is set to "messages" to stream LLM tokens. The metadata contains information about the LLM invocation, including the tags.
  4. Filter the streamed tokens by the tags field in the metadata to only include the tokens from the LLM invocation with the "joke" tag. :::

:::js

import { ChatOpenAI } from "@langchain/openai";

const llm1 = new ChatOpenAI({
  model: "gpt-4o-mini",
  tags: ['joke'] // (1)!
});
const llm2 = new ChatOpenAI({
  model: "gpt-4o-mini",
  tags: ['poem'] // (2)!
});

const graph = // ... define a graph that uses these LLMs

for await (const [msg, metadata] of await graph.stream( // (3)!
  { topic: "cats" },
  { streamMode: "messages" }
)) {
  if (metadata.tags?.includes("joke")) { // (4)!
    console.log(msg.content + "|");
  }
}
  1. llm1 is tagged with "joke".
  2. llm2 is tagged with "poem".
  3. The streamMode is set to "messages" to stream LLM tokens. The metadata contains information about the LLM invocation, including the tags.
  4. Filter the streamed tokens by the tags field in the metadata to only include the tokens from the LLM invocation with the "joke" tag. :::

??? example "Extended example: filtering by tags"

  :::python
  ```python
  from typing import TypedDict

  from langchain.chat_models import init_chat_model
  from langgraph.graph import START, StateGraph

  joke_model = init_chat_model(model="openai:gpt-4o-mini", tags=["joke"]) # (1)!
  poem_model = init_chat_model(model="openai:gpt-4o-mini", tags=["poem"]) # (2)!


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


  async def call_model(state, config):
        topic = state["topic"]
        print("Writing joke...")
        # Note: Passing the config through explicitly is required for python < 3.11
        # Since context var support wasn't added before then: https://docs.python.org/3/library/asyncio-task.html#creating-tasks
        joke_response = await joke_model.ainvoke(
              [{"role": "user", "content": f"Write a joke about {topic}"}],
              config, # (3)!
        )
        print("\n\nWriting poem...")
        poem_response = await poem_model.ainvoke(
              [{"role": "user", "content": f"Write a short poem about {topic}"}],
              config, # (3)!
        )
        return {"joke": joke_response.content, "poem": poem_response.content}


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

  async for msg, metadata in graph.astream(
        {"topic": "cats"},
        # highlight-next-line
        stream_mode="messages", # (4)!
  ):
      if metadata["tags"] == ["joke"]: # (4)!
          print(msg.content, end="|", flush=True)
  ```

  1. The `joke_model` is tagged with "joke".
  2. The `poem_model` is tagged with "poem".
  3. The `config` is passed through explicitly to ensure the context vars are propagated correctly. This is required for Python < 3.11 when using async code. Please see the [async section](#async) for more details.
  4. The `stream_mode` is set to "messages" to stream LLM tokens. The `metadata` contains information about the LLM invocation, including the tags.
  :::

  :::js
  ```typescript
  import { ChatOpenAI } from "@langchain/openai";
  import { StateGraph, START } from "@langchain/langgraph";
  import { z } from "zod";

  const jokeModel = new ChatOpenAI({
    model: "gpt-4o-mini",
    tags: ["joke"] // (1)!
  });
  const poemModel = new ChatOpenAI({
    model: "gpt-4o-mini",
    tags: ["poem"] // (2)!
  });

  const State = z.object({
    topic: z.string(),
    joke: z.string(),
    poem: z.string(),
  });

  const graph = new StateGraph(State)
    .addNode("callModel", (state) => {
      const topic = state.topic;
      console.log("Writing joke...");

      const jokeResponse = await jokeModel.invoke([
        { role: "user", content: `Write a joke about ${topic}` }
      ]);

      console.log("\n\nWriting poem...");
      const poemResponse = await poemModel.invoke([
        { role: "user", content: `Write a short poem about ${topic}` }
      ]);

      return {
        joke: jokeResponse.content,
        poem: poemResponse.content
      };
    })
    .addEdge(START, "callModel")
    .compile();

  for await (const [msg, metadata] of await graph.stream(
    { topic: "cats" },
    { streamMode: "messages" } // (3)!
  )) {
    if (metadata.tags?.includes("joke")) { // (4)!
      console.log(msg.content + "|");
    }
  }
  ```

  1. The `jokeModel` is tagged with "joke".
  2. The `poemModel` is tagged with "poem".
  3. The `streamMode` is set to "messages" to stream LLM tokens. The `metadata` contains information about the LLM invocation, including the tags.
  4. Filter the streamed tokens by the `tags` field in the metadata to only include the tokens from the LLM invocation with the "joke" tag.
  :::

Filter by node

To stream tokens only from specific nodes, use stream_mode="messages" and filter the outputs by the langgraph_node field in the streamed metadata:

:::python

for msg, metadata in graph.stream( # (1)!
    inputs,
    # highlight-next-line
    stream_mode="messages",
):
    # highlight-next-line
    if msg.content and metadata["langgraph_node"] == "some_node_name": # (2)!
        ...
  1. The "messages" stream mode returns a tuple of (message_chunk, metadata) where message_chunk is the token streamed by the LLM and metadata is a dictionary with information about the graph node where the LLM was called and other information.
  2. Filter the streamed tokens by the langgraph_node field in the metadata to only include the tokens from the write_poem node. :::

:::js

for await (const [msg, metadata] of await graph.stream(
  // (1)!
  inputs,
  { streamMode: "messages" }
)) {
  if (msg.content && metadata.langgraph_node === "some_node_name") {
    // (2)!
    // ...
  }
}
  1. The "messages" stream mode returns a tuple of [messageChunk, metadata] where messageChunk is the token streamed by the LLM and metadata is a dictionary with information about the graph node where the LLM was called and other information.
  2. Filter the streamed tokens by the langgraph_node field in the metadata to only include the tokens from the writePoem node. :::

??? example "Extended example: streaming LLM tokens from specific nodes"

  :::python
  ```python
  from typing import TypedDict
  from langgraph.graph import START, StateGraph
  from langchain_openai import ChatOpenAI

  model = ChatOpenAI(model="gpt-4o-mini")


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


  def write_joke(state: State):
        topic = state["topic"]
        joke_response = model.invoke(
              [{"role": "user", "content": f"Write a joke about {topic}"}]
        )
        return {"joke": joke_response.content}


  def write_poem(state: State):
        topic = state["topic"]
        poem_response = model.invoke(
              [{"role": "user", "content": f"Write a short poem about {topic}"}]
        )
        return {"poem": poem_response.content}


  graph = (
        StateGraph(State)
        .add_node(write_joke)
        .add_node(write_poem)
        # write both the joke and the poem concurrently
        .add_edge(START, "write_joke")
        .add_edge(START, "write_poem")
        .compile()
  )

  # highlight-next-line
  for msg, metadata in graph.stream( # (1)!
      {"topic": "cats"},
      stream_mode="messages",
  ):
      # highlight-next-line
      if msg.content and metadata["langgraph_node"] == "write_poem": # (2)!
          print(msg.content, end="|", flush=True)
  ```

  1. The "messages" stream mode returns a tuple of `(message_chunk, metadata)` where `message_chunk` is the token streamed by the LLM and `metadata` is a dictionary with information about the graph node where the LLM was called and other information.
  2. Filter the streamed tokens by the `langgraph_node` field in the metadata to only include the tokens from the `write_poem` node.
  :::

  :::js
  ```typescript
  import { ChatOpenAI } from "@langchain/openai";
  import { StateGraph, START } from "@langchain/langgraph";
  import { z } from "zod";

  const model = new ChatOpenAI({ model: "gpt-4o-mini" });

  const State = z.object({
    topic: z.string(),
    joke: z.string(),
    poem: z.string(),
  });

  const graph = new StateGraph(State)
    .addNode("writeJoke", async (state) => {
      const topic = state.topic;
      const jokeResponse = await model.invoke([
        { role: "user", content: `Write a joke about ${topic}` }
      ]);
      return { joke: jokeResponse.content };
    })
    .addNode("writePoem", async (state) => {
      const topic = state.topic;
      const poemResponse = await model.invoke([
        { role: "user", content: `Write a short poem about ${topic}` }
      ]);
      return { poem: poemResponse.content };
    })
    // write both the joke and the poem concurrently
    .addEdge(START, "writeJoke")
    .addEdge(START, "writePoem")
    .compile();

  for await (const [msg, metadata] of await graph.stream( // (1)!
    { topic: "cats" },
    { streamMode: "messages" }
  )) {
    if (msg.content && metadata.langgraph_node === "writePoem") { // (2)!
      console.log(msg.content + "|");
    }
  }
  ```

  1. The "messages" stream mode returns a tuple of `[messageChunk, metadata]` where `messageChunk` is the token streamed by the LLM and `metadata` is a dictionary with information about the graph node where the LLM was called and other information.
  2. Filter the streamed tokens by the `langgraph_node` field in the metadata to only include the tokens from the `writePoem` node.
  :::

Stream custom data

:::python To send custom user-defined data from inside a LangGraph node or tool, follow these steps:

  1. Use get_stream_writer() to access the stream writer and emit custom data.
  2. Set stream_mode="custom" when calling .stream() or .astream() to get the custom data in the stream. You can combine multiple modes (e.g., ["updates", "custom"]), but at least one must be "custom".

!!! warning "No get_stream_writer() in async for Python < 3.11"

In async code running on Python < 3.11, `get_stream_writer()` will not work.
Instead, add a `writer` parameter to your node or tool and pass it manually.
See [Async with Python < 3.11](#async) for usage examples.

=== "node"

  ```python
  from typing import TypedDict
  from langgraph.config import get_stream_writer
  from langgraph.graph import StateGraph, START

  class State(TypedDict):
      query: str
      answer: str

  def node(state: State):
      writer = get_stream_writer()  # (1)!
      writer({"custom_key": "Generating custom data inside node"}) # (2)!
      return {"answer": "some data"}

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

  inputs = {"query": "example"}

  # Usage
  for chunk in graph.stream(inputs, stream_mode="custom"):  # (3)!
      print(chunk)
  ```

  1. Get the stream writer to send custom data.
  2. Emit a custom key-value pair (e.g., progress update).
  3. Set `stream_mode="custom"` to receive the custom data in the stream.

=== "tool"

  ```python
  from langchain_core.tools import tool
  from langgraph.config import get_stream_writer

  @tool
  def query_database(query: str) -> str:
      """Query the database."""
      writer = get_stream_writer() # (1)!
      # highlight-next-line
      writer({"data": "Retrieved 0/100 records", "type": "progress"}) # (2)!
      # perform query
      # highlight-next-line
      writer({"data": "Retrieved 100/100 records", "type": "progress"}) # (3)!
      return "some-answer"


  graph = ... # define a graph that uses this tool

  for chunk in graph.stream(inputs, stream_mode="custom"): # (4)!
      print(chunk)
  ```

  1. Access the stream writer to send custom data.
  2. Emit a custom key-value pair (e.g., progress update).
  3. Emit another custom key-value pair.
  4. Set `stream_mode="custom"` to receive the custom data in the stream.

:::

:::js To send custom user-defined data from inside a LangGraph node or tool, follow these steps:

  1. Use the writer parameter from the LangGraphRunnableConfig to emit custom data.
  2. Set streamMode: "custom" when calling .stream() to get the custom data in the stream. You can combine multiple modes (e.g., ["updates", "custom"]), but at least one must be "custom".

=== "node"

  ```typescript
  import { StateGraph, START, LangGraphRunnableConfig } from "@langchain/langgraph";
  import { z } from "zod";

  const State = z.object({
    query: z.string(),
    answer: z.string(),
  });

  const graph = new StateGraph(State)
    .addNode("node", async (state, config) => {
      config.writer({ custom_key: "Generating custom data inside node" }); // (1)!
      return { answer: "some data" };
    })
    .addEdge(START, "node")
    .compile();

  const inputs = { query: "example" };

  // Usage
  for await (const chunk of await graph.stream(inputs, { streamMode: "custom" })) { // (2)!
    console.log(chunk);
  }
  ```

  1. Use the writer to emit a custom key-value pair (e.g., progress update).
  2. Set `streamMode: "custom"` to receive the custom data in the stream.

=== "tool"

  ```typescript
  import { tool } from "@langchain/core/tools";
  import { LangGraphRunnableConfig } from "@langchain/langgraph";
  import { z } from "zod";

  const queryDatabase = tool(
    async (input, config: LangGraphRunnableConfig) => {
      config.writer({ data: "Retrieved 0/100 records", type: "progress" }); // (1)!
      // perform query
      config.writer({ data: "Retrieved 100/100 records", type: "progress" }); // (2)!
      return "some-answer";
    },
    {
      name: "query_database",
      description: "Query the database.",
      schema: z.object({
        query: z.string().describe("The query to execute."),
      }),
    }
  );

  const graph = // ... define a graph that uses this tool

  for await (const chunk of await graph.stream(inputs, { streamMode: "custom" })) { // (3)!
    console.log(chunk);
  }
  ```

  1. Use the writer to emit a custom key-value pair (e.g., progress update).
  2. Emit another custom key-value pair.
  3. Set `streamMode: "custom"` to receive the custom data in the stream.

:::

Use with any LLM

:::python You can use stream_mode="custom" to stream data from any LLM API — even if that API does not implement the LangChain chat model interface.

This lets you integrate raw LLM clients or external services that provide their own streaming interfaces, making LangGraph highly flexible for custom setups.

from langgraph.config import get_stream_writer

def call_arbitrary_model(state):
    """Example node that calls an arbitrary model and streams the output"""
    # highlight-next-line
    writer = get_stream_writer() # (1)!
    # Assume you have a streaming client that yields chunks
    for chunk in your_custom_streaming_client(state["topic"]): # (2)!
        # highlight-next-line
        writer({"custom_llm_chunk": chunk}) # (3)!
    return {"result": "completed"}

graph = (
    StateGraph(State)
    .add_node(call_arbitrary_model)
    # Add other nodes and edges as needed
    .compile()
)

for chunk in graph.stream(
    {"topic": "cats"},
    # highlight-next-line
    stream_mode="custom", # (4)!
):
    # The chunk will contain the custom data streamed from the llm
    print(chunk)
  1. Get the stream writer to send custom data.
  2. Generate LLM tokens using your custom streaming client.
  3. Use the writer to send custom data to the stream.
  4. Set stream_mode="custom" to receive the custom data in the stream. :::

:::js You can use streamMode: "custom" to stream data from any LLM API — even if that API does not implement the LangChain chat model interface.

This lets you integrate raw LLM clients or external services that provide their own streaming interfaces, making LangGraph highly flexible for custom setups.

import { LangGraphRunnableConfig } from "@langchain/langgraph";

const callArbitraryModel = async (
  state: any,
  config: LangGraphRunnableConfig
) => {
  // Example node that calls an arbitrary model and streams the output
  // Assume you have a streaming client that yields chunks
  for await (const chunk of yourCustomStreamingClient(state.topic)) {
    // (1)!
    config.writer({ custom_llm_chunk: chunk }); // (2)!
  }
  return { result: "completed" };
};

const graph = new StateGraph(State)
  .addNode("callArbitraryModel", callArbitraryModel)
  // Add other nodes and edges as needed
  .compile();

for await (const chunk of await graph.stream(
  { topic: "cats" },
  { streamMode: "custom" } // (3)!
)) {
  // The chunk will contain the custom data streamed from the llm
  console.log(chunk);
}
  1. Generate LLM tokens using your custom streaming client.
  2. Use the writer to send custom data to the stream.
  3. Set streamMode: "custom" to receive the custom data in the stream. :::

??? example "Extended example: streaming arbitrary chat model"

  :::python
  ```python
  import operator
  import json

  from typing import TypedDict
  from typing_extensions import Annotated
  from langgraph.graph import StateGraph, START

  from openai import AsyncOpenAI

  openai_client = AsyncOpenAI()
  model_name = "gpt-4o-mini"


  async def stream_tokens(model_name: str, messages: list[dict]):
      response = await openai_client.chat.completions.create(
          messages=messages, model=model_name, stream=True
      )
      role = None
      async for chunk in response:
          delta = chunk.choices[0].delta

          if delta.role is not None:
              role = delta.role

          if delta.content:
              yield {"role": role, "content": delta.content}


  # this is our tool
  async def get_items(place: str) -> str:
      """Use this tool to list items one might find in a place you're asked about."""
      writer = get_stream_writer()
      response = ""
      async for msg_chunk in stream_tokens(
          model_name,
          [
              {
                  "role": "user",
                  "content": (
                      "Can you tell me what kind of items "
                      f"i might find in the following place: '{place}'. "
                      "List at least 3 such items separating them by a comma. "
                      "And include a brief description of each item."
                  ),
              }
          ],
      ):
          response += msg_chunk["content"]
          writer(msg_chunk)

      return response


  class State(TypedDict):
      messages: Annotated[list[dict], operator.add]


  # this is the tool-calling graph node
  async def call_tool(state: State):
      ai_message = state["messages"][-1]
      tool_call = ai_message["tool_calls"][-1]

      function_name = tool_call["function"]["name"]
      if function_name != "get_items":
          raise ValueError(f"Tool {function_name} not supported")

      function_arguments = tool_call["function"]["arguments"]
      arguments = json.loads(function_arguments)

      function_response = await get_items(**arguments)
      tool_message = {
          "tool_call_id": tool_call["id"],
          "role": "tool",
          "name": function_name,
          "content": function_response,
      }
      return {"messages": [tool_message]}


  graph = (
      StateGraph(State)
      .add_node(call_tool)
      .add_edge(START, "call_tool")
      .compile()
  )
  ```

  Let's invoke the graph with an AI message that includes a tool call:

  ```python
  inputs = {
      "messages": [
          {
              "content": None,
              "role": "assistant",
              "tool_calls": [
                  {
                      "id": "1",
                      "function": {
                          "arguments": '{"place":"bedroom"}',
                          "name": "get_items",
                      },
                      "type": "function",
                  }
              ],
          }
      ]
  }

  async for chunk in graph.astream(
      inputs,
      stream_mode="custom",
  ):
      print(chunk["content"], end="|", flush=True)
  ```
  :::

  :::js
  ```typescript
  import { StateGraph, START, LangGraphRunnableConfig } from "@langchain/langgraph";
  import { z } from "zod";
  import OpenAI from "openai";

  const openaiClient = new OpenAI();
  const modelName = "gpt-4o-mini";

  async function* streamTokens(modelName: string, messages: any[]) {
    const response = await openaiClient.chat.completions.create({
      messages,
      model: modelName,
      stream: true,
    });

    let role: string | null = null;
    for await (const chunk of response) {
      const delta = chunk.choices[0]?.delta;

      if (delta?.role) {
        role = delta.role;
      }

      if (delta?.content) {
        yield { role, content: delta.content };
      }
    }
  }

  // this is our tool
  const getItems = tool(
    async (input, config: LangGraphRunnableConfig) => {
      let response = "";
      for await (const msgChunk of streamTokens(
        modelName,
        [
          {
            role: "user",
            content: `Can you tell me what kind of items i might find in the following place: '${input.place}'. List at least 3 such items separating them by a comma. And include a brief description of each item.`,
          },
        ]
      )) {
        response += msgChunk.content;
        config.writer?.(msgChunk);
      }
      return response;
    },
    {
      name: "get_items",
      description: "Use this tool to list items one might find in a place you're asked about.",
      schema: z.object({
        place: z.string().describe("The place to look up items for."),
      }),
    }
  );

  const State = z.object({
    messages: z.array(z.any()),
  });

  const graph = new StateGraph(State)
    // this is the tool-calling graph node
    .addNode("callTool", async (state) => {
      const aiMessage = state.messages.at(-1);
      const toolCall = aiMessage.tool_calls?.at(-1);

      const functionName = toolCall?.function?.name;
      if (functionName !== "get_items") {
        throw new Error(`Tool ${functionName} not supported`);
      }

      const functionArguments = toolCall?.function?.arguments;
      const args = JSON.parse(functionArguments);

      const functionResponse = await getItems.invoke(args);
      const toolMessage = {
        tool_call_id: toolCall.id,
        role: "tool",
        name: functionName,
        content: functionResponse,
      };
      return { messages: [toolMessage] };
    })
    .addEdge(START, "callTool")
    .compile();
  ```

  Let's invoke the graph with an AI message that includes a tool call:

  ```typescript
  const inputs = {
    messages: [
      {
        content: null,
        role: "assistant",
        tool_calls: [
          {
            id: "1",
            function: {
              arguments: '{"place":"bedroom"}',
              name: "get_items",
            },
            type: "function",
          }
        ],
      }
    ]
  };

  for await (const chunk of await graph.stream(
    inputs,
    { streamMode: "custom" }
  )) {
    console.log(chunk.content + "|");
  }
  ```
  :::

Disable streaming for specific chat models

If your application mixes models that support streaming with those that do not, you may need to explicitly disable streaming for models that do not support it.

:::python Set disable_streaming=True when initializing the model.

=== "init_chat_model"

  ```python
  from langchain.chat_models import init_chat_model

  model = init_chat_model(
      "anthropic:claude-3-7-sonnet-latest",
      # highlight-next-line
      disable_streaming=True # (1)!
  )
  ```

  1. Set `disable_streaming=True` to disable streaming for the chat model.

=== "chat model interface"

  ```python
  from langchain_openai import ChatOpenAI

  llm = ChatOpenAI(model="o1-preview", disable_streaming=True) # (1)!
  ```

  1. Set `disable_streaming=True` to disable streaming for the chat model.

:::

:::js Set streaming: false when initializing the model.

import { ChatOpenAI } from "@langchain/openai";

const model = new ChatOpenAI({
  model: "o1-preview",
  streaming: false, // (1)!
});

:::

:::python

Async with Python < 3.11

In Python versions < 3.11, asyncio tasks do not support the context parameter.
This limits LangGraph ability to automatically propagate context, and affects LangGraph's streaming mechanisms in two key ways:

  1. You must explicitly pass RunnableConfig into async LLM calls (e.g., ainvoke()), as callbacks are not automatically propagated.
  2. You cannot use get_stream_writer() in async nodes or tools — you must pass a writer argument directly.

??? example "Extended example: async LLM call with manual config"

  ```python
  from typing import TypedDict
  from langgraph.graph import START, StateGraph
  from langchain.chat_models import init_chat_model

  llm = init_chat_model(model="openai:gpt-4o-mini")

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

  async def call_model(state, config): # (1)!
      topic = state["topic"]
      print("Generating joke...")
      joke_response = await llm.ainvoke(
          [{"role": "user", "content": f"Write a joke about {topic}"}],
          # highlight-next-line
          config, # (2)!
      )
      return {"joke": joke_response.content}

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

  async for chunk, metadata in graph.astream(
      {"topic": "ice cream"},
      # highlight-next-line
      stream_mode="messages", # (3)!
  ):
      if chunk.content:
          print(chunk.content, end="|", flush=True)
  ```

  1. Accept `config` as an argument in the async node function.
  2. Pass `config` to `llm.ainvoke()` to ensure proper context propagation.
  3. Set `stream_mode="messages"` to stream LLM tokens.

??? example "Extended example: async custom streaming with stream writer"

  ```python
  from typing import TypedDict
  from langgraph.types import StreamWriter

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

  # highlight-next-line
  async def generate_joke(state: State, writer: StreamWriter): # (1)!
        writer({"custom_key": "Streaming custom data while generating a joke"})
        return {"joke": f"This is a joke about {state['topic']}"}

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

  async for chunk in graph.astream(
        {"topic": "ice cream"},
        # highlight-next-line
        stream_mode="custom", # (2)!
  ):
        print(chunk)
  ```

  1. Add `writer` as an argument in the function signature of the async node or tool. LangGraph will automatically pass the stream writer to the function.
  2. Set `stream_mode="custom"` to receive the custom data in the stream.

:::