langgraph 1.2.1__tar.gz → 1.2.2__tar.gz

{langgraph-1.2.1 → langgraph-1.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph
-Version: 1.2.1
+Version: 1.2.2
 Summary: Building stateful, multi-actor applications with LLMs
 Project-URL: Homepage, https://docs.langchain.com/oss/python/langgraph/overview
 Project-URL: Documentation, https://reference.langchain.com/python/langgraph/

{langgraph-1.2.1 → langgraph-1.2.2}/langgraph/pregel/_loop.py

@@ -115,6 +115,7 @@ from langgraph.pregel._io import (
     map_output_values,
     read_channels,
 )
+from langgraph.pregel._messages import ensure_message_ids
 from langgraph.pregel._read import PregelNode
 from langgraph.pregel._utils import get_new_channel_versions, is_xxh3_128_hexdigest
 from langgraph.pregel.debug import (
@@ -447,6 +448,14 @@ class PregelLoop:
 
         # save writes
         self.checkpoint_pending_writes.extend((task_id, c, v) for c, v in writes)
+        # Assign stable IDs to any id=None BaseMessages in DeltaChannel writes
+        # before the background thread serialises them. Without this, reducers
+        # that assign IDs inside apply_writes() race with serialisation and
+        # store id=None, causing get_state() replays to produce a different UUID
+        # on every call.
+        for c, v in writes_to_save:
+            if isinstance(self.specs.get(c), DeltaChannel):
+                ensure_message_ids(v)
         if self.durability != "exit" and self.checkpointer_put_writes is not None:
             config = patch_configurable(
                 self.checkpoint_config,

{langgraph-1.2.1 → langgraph-1.2.2}/langgraph/pregel/_messages.py

@@ -11,6 +11,7 @@ from uuid import UUID, uuid4
 
 from langchain_core.callbacks import BaseCallbackHandler
 from langchain_core.messages import BaseMessage, ToolMessage
+from langchain_core.messages.utils import convert_to_messages
 from langchain_core.outputs import ChatGeneration, ChatGenerationChunk, LLMResult
 from pydantic import BaseModel
 
@@ -405,3 +406,56 @@ class StreamMessagesHandlerV2(StreamMessagesHandler, _V2StreamingCallbackHandler
                     self.seen.add(msg_id)
             v2_meta = {**meta[1], "run_id": str(run_id)}
             self.stream((meta[0], "messages", (event, v2_meta)))
+
+
+# Known role values (OpenAI-style) and type values (LangChain serialisation)
+# that identify a dict as a message. Checked before coercing to BaseMessage so
+# we don't accidentally touch unrelated dicts that happen to have a "role" key.
+_MESSAGE_ROLES: frozenset[str] = frozenset(
+    {"user", "human", "assistant", "ai", "tool", "system", "function"}
+)
+_MESSAGE_TYPES: frozenset[str] = frozenset(
+    {"human", "ai", "tool", "system", "function", "remove"}
+)
+
+
+def _is_message_dict(item: dict) -> bool:
+    return item.get("role") in _MESSAGE_ROLES or item.get("type") in _MESSAGE_TYPES
+
+
+def ensure_message_ids(value: Any) -> None:
+    """Coerce message-like write values to typed BaseMessages with stable IDs.
+
+    Called in put_writes() before DeltaChannel writes are submitted to the
+    checkpointer. Without this the checkpoint may store raw dicts or id=None
+    BaseMessages; every get_state() replay then produces a different UUID and
+    the same message appears with a different ID in each LangSmith trace.
+
+    Handles three input shapes:
+    - BaseMessage objects: assign a UUID if id is None.
+    - Dicts with a known "role" (OpenAI-style) or "type" (LangChain format) at
+      the root level: stamp "id" into the dict in-place. The reducer's
+      convert_to_messages call will forward the id to the resulting BaseMessage.
+    - Lists of the above: apply the same logic to each element, replacing dict
+      items with coerced BaseMessages so the shared list reference seen by
+      checkpoint_pending_writes and the background thread both get typed messages.
+
+    Mutating synchronously here (before the background thread is submitted) is
+    safe: the serialised bytes always reflect the post-coercion state.
+    """
+    if isinstance(value, BaseMessage):
+        if value.id is None:
+            value.id = str(uuid4())
+    elif isinstance(value, dict) and _is_message_dict(value):
+        if not value.get("id"):
+            value["id"] = str(uuid4())
+    elif isinstance(value, list):
+        for i, item in enumerate(value):
+            if isinstance(item, BaseMessage):
+                if item.id is None:
+                    item.id = str(uuid4())
+            elif isinstance(item, dict) and _is_message_dict(item):
+                msg = convert_to_messages([item])[0]
+                if msg.id is None:
+                    msg.id = str(uuid4())
+                value[i] = msg

{langgraph-1.2.1 → langgraph-1.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langgraph"
-version = "1.2.1"
+version = "1.2.2"
 description = "Building stateful, multi-actor applications with LLMs"
 authors = []
 requires-python = ">=3.10"

langgraph-1.2.2/tests/test_delta_channel_id_stability.py

@@ -0,0 +1,140 @@
+"""ensure_message_ids() assigns stable UUIDs to id=None BaseMessages
+before DeltaChannel writes are serialised to the checkpoint.
+
+Without this, the checkpoint stores id=None and every get_state() replay
+produces a different UUID — the same HumanMessage appears with a different
+ID in each LangSmith trace / on every resumed invocation.
+"""
+
+from __future__ import annotations
+
+from typing import Annotated, Any
+
+import pytest
+from langchain_core.messages import AIMessage, AnyMessage, HumanMessage
+from langgraph.checkpoint.memory import InMemorySaver
+from typing_extensions import TypedDict
+
+from langgraph.channels.delta import DeltaChannel
+from langgraph.graph import END, START, StateGraph
+
+pytestmark = pytest.mark.anyio
+
+
+def _append_reducer(
+    state: list[AnyMessage], writes: list[list[AnyMessage]]
+) -> list[AnyMessage]:
+    """Simple append — no ID assignment. IDs come from ensure_message_ids()."""
+    result = list(state)
+    for w in writes:
+        if isinstance(w, list):
+            result.extend(w)
+        else:
+            result.append(w)  # type: ignore[arg-type]
+    return result
+
+
+def _build_graph(checkpointer: Any) -> Any:
+    State = TypedDict(  # noqa: UP013
+        "State",
+        {
+            "messages": Annotated[
+                list, DeltaChannel(_append_reducer, snapshot_frequency=50)
+            ]
+        },
+    )  # type: ignore[call-overload]
+
+    def agent(state: dict) -> dict:
+        return {"messages": [AIMessage(content="reply", id="ai-1")]}
+
+    return (
+        StateGraph(State)
+        .add_node("agent", agent)
+        .add_edge(START, "agent")
+        .add_edge("agent", END)
+        .compile(checkpointer=checkpointer)
+    )
+
+
+def test_delta_channel_message_gets_id_and_stays_stable() -> None:
+    """Messages written with id=None must receive a stable UUID.
+
+    ensure_message_ids() is called in put_writes() before the background
+    thread serialises DeltaChannel writes. The checkpoint stores the
+    assigned UUID, so every get_state() replay sees the same ID.
+    """
+    saver = InMemorySaver()
+    graph = _build_graph(saver)
+    config = {"configurable": {"thread_id": "id-stability"}}
+
+    graph.invoke({"messages": [HumanMessage(content="hello")]}, config)
+
+    ids = [
+        next(
+            m.id
+            for m in graph.get_state(config).values["messages"]
+            if isinstance(m, HumanMessage)
+        )
+        for _ in range(3)
+    ]
+
+    assert ids[0] is not None, "ensure_message_ids should have assigned a UUID"
+    assert len(set(ids)) == 1, (
+        f"HumanMessage id must be stable across get_state() calls; "
+        f"got {ids}. The checkpoint is storing id=None."
+    )
+
+
+async def test_delta_channel_message_gets_id_and_stays_stable_async() -> None:
+    """Same check via ainvoke (AsyncPregelLoop path)."""
+    saver = InMemorySaver()
+    graph = _build_graph(saver)
+    config = {"configurable": {"thread_id": "id-stability-async"}}
+
+    await graph.ainvoke({"messages": [HumanMessage(content="hello")]}, config)
+
+    ids = [
+        next(
+            m.id
+            for m in (await graph.aget_state(config)).values["messages"]
+            if isinstance(m, HumanMessage)
+        )
+        for _ in range(3)
+    ]
+
+    assert ids[0] is not None, "ensure_message_ids should have assigned a UUID"
+    assert len(set(ids)) == 1, (
+        f"Async path: HumanMessage id unstable across aget_state() calls: {ids}"
+    )
+
+
+def test_delta_channel_dict_style_message_gets_stable_id() -> None:
+    """Dict-style inputs (API / over-the-wire format) must also get stable IDs.
+
+    When the graph is invoked via the LangGraph API the input arrives as a raw
+    dict {"role": "user", "content": "..."} rather than a BaseMessage object.
+    ensure_message_ids() must coerce those dicts to typed BaseMessages and
+    stamp a UUID so the checkpoint never stores an id-less message.
+    """
+    saver = InMemorySaver()
+    graph = _build_graph(saver)
+    config = {"configurable": {"thread_id": "dict-id-stability"}}
+
+    # Invoke with a raw dict (the format LangGraph API sends)
+    graph.invoke({"messages": [{"role": "user", "content": "hello"}]}, config)
+
+    ids = [
+        next(
+            m.id
+            for m in graph.get_state(config).values["messages"]
+            if isinstance(m, HumanMessage)
+        )
+        for _ in range(3)
+    ]
+
+    assert ids[0] is not None, (
+        "dict-style message should have been coerced and assigned a UUID"
+    )
+    assert len(set(ids)) == 1, (
+        f"dict-style HumanMessage id must be stable across get_state() calls; got {ids}"
+    )

{langgraph-1.2.1 → langgraph-1.2.2}/uv.lock

@@ -1382,7 +1382,7 @@ wheels = [
 
 [[package]]
 name = "langgraph"
-version = "1.2.1"
+version = "1.2.2"
 source = { editable = "." }
 dependencies = [
     { name = "langchain-core" },
@@ -1563,7 +1563,7 @@ wheels = [
 
 [[package]]
 name = "langgraph-checkpoint"
-version = "4.1.0"
+version = "4.1.1"
 source = { editable = "../checkpoint" }
 dependencies = [
     { name = "langchain-core" },