github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-04-23 20:23:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(agents): use DeltaChannel for AgentState messages

Browse Source 
Reduces checkpoint storage for messages from O(N²) to O(N) by
replacing the bare add_messages reducer with DeltaChannel(add_messages).
Same change applied to _OutputAgentState.
This commit is contained in:
Sydney Runkle
2026-04-22 09:35:52 -04:00
parent 0545feead7
commit e5791e9835
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
libs/langchain_v1/langchain/agents/middleware/types.py
View File

@@ -30,6 +30,7 @@ from langchain_core.messages import (
SystemMessage,
ToolMessage,
)
from langgraph.channels.delta import DeltaChannel
from langgraph.channels.ephemeral_value import EphemeralValue
from langgraph.graph.message import add_messages
from langgraph.prebuilt.tool_node import ToolCallRequest, ToolCallWrapper
@@ -350,7 +351,7 @@ PrivateStateAttr = OmitFromSchema(input=True, output=True)
class AgentState(TypedDict, Generic[ResponseT]):
"""State schema for the agent."""
messages: Required[Annotated[list[AnyMessage], add_messages]]
messages: Required[Annotated[list[AnyMessage], DeltaChannel(add_messages)]]
jump_to: NotRequired[Annotated[JumpTo | None, EphemeralValue, PrivateStateAttr]]
structured_response: NotRequired[Annotated[ResponseT, OmitFromInput]]
@@ -364,7 +365,7 @@ class _InputAgentState(TypedDict): # noqa: PYI049
class _OutputAgentState(TypedDict, Generic[ResponseT]): # noqa: PYI049
"""Output state schema for the agent."""
messages: Required[Annotated[list[AnyMessage], add_messages]]
messages: Required[Annotated[list[AnyMessage], DeltaChannel(add_messages)]]
structured_response: NotRequired[ResponseT]