diff --git a/langgraph/graph/__init__.py b/langgraph/graph/__init__.py index 1c0ef3262..1365f97ac 100644 --- a/langgraph/graph/__init__.py +++ b/langgraph/graph/__init__.py @@ -1,5 +1,12 @@ from langgraph.graph.graph import END, Graph -from langgraph.graph.message import MessageGraph, add_messages +from langgraph.graph.message import MessageGraph, MessagesState, add_messages from langgraph.graph.state import StateGraph -__all__ = ["END", "Graph", "StateGraph", "MessageGraph", "add_messages"] +__all__ = [ + "END", + "Graph", + "StateGraph", + "MessageGraph", + "add_messages", + "MessagesState", +] diff --git a/langgraph/graph/message.py b/langgraph/graph/message.py index 62fc39f2c..0b4db9152 100644 --- a/langgraph/graph/message.py +++ b/langgraph/graph/message.py @@ -1,5 +1,5 @@ import uuid -from typing import Annotated, Union +from typing import Annotated, TypedDict, Union from langchain_core.messages import ( AnyMessage, @@ -136,3 +136,7 @@ class MessageGraph(StateGraph): def __init__(self) -> None: super().__init__(Annotated[list[AnyMessage], add_messages]) + + +class MessagesState(TypedDict): + messages: Annotated[list[AnyMessage], add_messages]