mirror of
https://github.com/langchain-ai/langgraph.git
synced 2026-09-11 04:07:52 +02:00
David DuongandGitHub
81935a73d8
Sample usage:
```python
from typing import Annotated, Sequence, TypedDict
from langchain_core.messages import BaseMessage
from langchain_openai import ChatOpenAI
from langgraph.graph import StateGraph
from langgraph.graph.message import add_messages
from langgraph.graph.ui import AnyUIMessage, ui_message_reducer, push_ui_message
class AgentState(TypedDict): # noqa: D101
messages: Annotated[Sequence[BaseMessage], add_messages]
ui: Annotated[Sequence[AnyUIMessage], ui_message_reducer]
async def agent(state: AgentState): # noqa: D103
message = await ChatOpenAI(model="gpt-4o-mini", temperature=0).ainvoke(
state["messages"]
)
# Also directly writes the message to `ui`
push_ui_message("simple", {"hello": "123"}, message=message, state_key="ui")
return {"messages": [message]}
```