harness-sdk-python 0.24.0__tar.gz → 0.25.1__tar.gz

{harness_sdk_python-0.24.0 → harness_sdk_python-0.25.1}/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.5
 Name: harness-sdk-python
-Version: 0.24.0
+Version: 0.25.1
 Summary: RunManager: the harness-sdk runs subsystem for Python Statewire hosts
 Project-URL: Repository, https://github.com/assistant-ui/harness-sdk
 License-Expression: MIT
 Requires-Python: <4.0,>=3.12
 Requires-Dist: pinned<0.15,>=0.14.0
-Requires-Dist: statewire<0.12,>=0.11.0
+Requires-Dist: statewire<0.13,>=0.12.0
 Provides-Extra: deepagents
 Requires-Dist: deepagents>=0.6.12; extra == 'deepagents'
 Requires-Dist: langchain-core>=0.3; extra == 'deepagents'

{harness_sdk_python-0.24.0 → harness_sdk_python-0.25.1}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "harness-sdk-python"
-version = "0.24.0"
+version = "0.25.1"
 description = "RunManager: the harness-sdk runs subsystem for Python Statewire hosts"
 readme = "README.md"
 license = "MIT"
 requires-python = ">=3.12,<4.0"
-dependencies = ["statewire>=0.11.0,<0.12", "pinned>=0.14.0,<0.15"]
+dependencies = ["statewire>=0.12.0,<0.13", "pinned>=0.14.0,<0.15"]
 
 [project.optional-dependencies]
 postgres = ["langgraph-checkpoint-postgres>=2.0.0"]

{harness_sdk_python-0.24.0 → harness_sdk_python-0.25.1}/src/harness_sdk/__init__.py

@@ -8,6 +8,7 @@ from .checkpoints import (
     Snapshot,
     checkpoint_thread,
     langgraph_checkpoints,
+    langgraph_snapshot,
     langgraph_switch_values,
 )
 from .linear_thread import linear_thread
@@ -25,6 +26,7 @@ __all__ = [
     "Snapshot",
     "checkpoint_thread",
     "langgraph_checkpoints",
+    "langgraph_snapshot",
     "langgraph_switch_values",
     "linear_thread",
     "run_manager",

{harness_sdk_python-0.24.0 → harness_sdk_python-0.25.1}/src/harness_sdk/checkpoints.py

@@ -295,15 +295,31 @@ def _message_dict(message: Any) -> dict[str, Any]:
     return message
 
 
+def _values_snapshot(
+    checkpoint_id: str,
+    values: dict[str, Any],
+    messages_key: str,
+    lineage_key: str,
+) -> Snapshot:
+    lineage = values.get(lineage_key)
+    if lineage is not None and not isinstance(lineage, dict):
+        raise ValueError(f"lineage channel must be an object, got {lineage!r}")
+    return Snapshot(
+        checkpoint_id=checkpoint_id,
+        messages=[_message_dict(m) for m in values.get(messages_key, [])],
+        lineage=lineage,
+    )
+
+
 def langgraph_checkpoints(
-    saver: Any,
+    graph: Any,
     *,
     thread_id: str,
     checkpoint_ns: str = "",
     messages_key: str = "messages",
     lineage_key: str = "lineage",
 ) -> CheckpointStore:
-    """A CheckpointStore over a langgraph BaseCheckpointSaver's ``aget_tuple``."""
+    """A CheckpointStore over a compiled langgraph graph's ``aget_state``."""
 
     class _Store:
         async def get(self, checkpoint_id: str | None) -> Snapshot | None:
@@ -313,22 +329,34 @@ def langgraph_checkpoints(
             }
             if checkpoint_id is not None:
                 configurable["checkpoint_id"] = checkpoint_id
-            saved = await saver.aget_tuple({"configurable": configurable})
-            if saved is None:
+            state = await graph.aget_state({"configurable": configurable})
+            if state.metadata is None:
                 return None
-            values = saved.checkpoint["channel_values"]
-            lineage = values.get(lineage_key)
-            if lineage is not None and not isinstance(lineage, dict):
-                raise ValueError(f"lineage channel must be an object, got {lineage!r}")
-            return Snapshot(
-                checkpoint_id=saved.checkpoint["id"],
-                messages=[_message_dict(m) for m in values.get(messages_key, [])],
-                lineage=lineage,
+            return _values_snapshot(
+                state.config["configurable"]["checkpoint_id"],
+                state.values,
+                messages_key,
+                lineage_key,
             )
 
     return _Store()
 
 
+def langgraph_snapshot(
+    payload: dict[str, Any],
+    *,
+    messages_key: str = "messages",
+    lineage_key: str = "lineage",
+) -> Snapshot:
+    """A Snapshot from one ``stream_mode="checkpoints"`` event payload, for ingesting a run's own output."""
+    return _values_snapshot(
+        payload["config"]["configurable"]["checkpoint_id"],
+        payload["values"],
+        messages_key,
+        lineage_key,
+    )
+
+
 def langgraph_switch_values(
     transcript: list[dict[str, Any]],
     lineage: dict[str, Any],

{harness_sdk_python-0.24.0 → harness_sdk_python-0.25.1}/tests/run_helpers.py

@@ -189,11 +189,11 @@ class RunDriver:
     async def side(self, method: str, params: Any = None, *, client_id: str = "c2"):
         """A second client's command: its own seq stream, its verdicts riding
         the other stream."""
-        await attach(self._app, client_id=client_id)
         seq = self._side_seqs[client_id] = self._side_seqs.get(client_id, 0) + 1
-        await self._submit(
-            {"cmd": [_statement(method, params, seq)]}, client_id=client_id
-        )
+        async with attach(self._app, client_id=client_id):
+            await self._submit(
+                {"cmd": [_statement(method, params, seq)]}, client_id=client_id
+            )
 
     async def pump(self, timeout: float = 5) -> None:
         env = await self._stream.next_event(timeout=timeout)

{harness_sdk_python-0.24.0 → harness_sdk_python-0.25.1}/tests/test_checkpoints.py

@@ -506,57 +506,90 @@ async def test_locate_loads_the_own_branch_tip_across_forks():
 # ─── langgraph adapter ──────────────────────────────────────
 
 
-async def _saved(saver, checkpoint_id, channel_values):
-    from langgraph.checkpoint.base import empty_checkpoint
+def _graph(state_type):
+    from langgraph.checkpoint.memory import InMemorySaver
+    from langgraph.graph import START, StateGraph
+
+    builder = StateGraph(state_type)
+    builder.add_node("node", lambda state: {})
+    builder.add_edge(START, "node")
+    return builder.compile(checkpointer=InMemorySaver())
+
+
+def _messages_graph():
+    from typing import Annotated, Any, TypedDict
+
+    from langgraph.graph.message import add_messages
+
+    class State(TypedDict):
+        messages: Annotated[list, add_messages]
+        lineage: dict[str, Any] | None
+
+    return _graph(State)
 
-    versions = {key: "1" for key in channel_values}
-    checkpoint = empty_checkpoint()
-    checkpoint["id"] = checkpoint_id
-    checkpoint["channel_values"] = channel_values
-    checkpoint["channel_versions"] = versions
-    config = {"configurable": {"thread_id": THREAD, "checkpoint_ns": ""}}
-    await saver.aput(config, checkpoint, {"source": "loop", "step": 1}, versions)
+
+CONFIG = {"configurable": {"thread_id": THREAD}}
 
 
 async def test_langgraph_adapter_reads_latest_and_by_id():
-    from langchain_core.messages import HumanMessage
-    from langgraph.checkpoint.memory import InMemorySaver
+    pytest.importorskip("langgraph.graph")
+    from langchain_core.messages import AIMessage, HumanMessage
 
-    saver = InMemorySaver()
+    graph = _messages_graph()
+    await graph.ainvoke({"messages": [HumanMessage(id="u1", content="hi")]}, CONFIG)
+    store = langgraph_checkpoints(graph, thread_id=THREAD)
+    first = await store.get(None)
+    assert [m["id"] for m in first.messages] == ["u1"]
+    assert first.lineage is None
     lineage = {"branch_ids": [0], "parent_checkpoint_id": None}
-    await _saved(saver, "1", {"messages": [HumanMessage(id="u1", content="hi")]})
-    await _saved(
-        saver,
-        "2",
-        {
-            "messages": [HumanMessage(id="u1", content="hi"), msg("a1", "ai")],
-            "lineage": lineage,
-        },
+    await graph.ainvoke(
+        {"messages": [AIMessage(id="a1", content="yo")], "lineage": lineage}, CONFIG
     )
-    store = langgraph_checkpoints(saver, thread_id=THREAD)
     latest = await store.get(None)
-    assert latest.checkpoint_id == "2"
+    assert latest.checkpoint_id != first.checkpoint_id
     assert [m["id"] for m in latest.messages] == ["u1", "a1"]
     assert latest.lineage == lineage
-    first = await store.get("1")
-    assert first.checkpoint_id == "1"
-    assert first.messages[0]["id"] == "u1"
-    assert first.lineage is None
-    other = langgraph_checkpoints(saver, thread_id="absent")
-    assert await other.get(None) is None
+    assert await store.get(first.checkpoint_id) == first
+    assert await store.get("absent") is None
+    assert await langgraph_checkpoints(graph, thread_id="other").get(None) is None
 
 
 async def test_langgraph_adapter_rejects_a_message_without_an_id():
-    from langchain_core.messages import HumanMessage
-    from langgraph.checkpoint.memory import InMemorySaver
+    pytest.importorskip("langgraph.graph")
+    from typing import TypedDict
 
-    saver = InMemorySaver()
-    await _saved(saver, "1", {"messages": [HumanMessage(content="no id")]})
-    store = langgraph_checkpoints(saver, thread_id=THREAD)
+    class State(TypedDict):
+        messages: list
+
+    graph = _graph(State)
+    await graph.ainvoke({"messages": [{"type": "human", "content": "no id"}]}, CONFIG)
+    store = langgraph_checkpoints(graph, thread_id=THREAD)
     with pytest.raises(ValueError, match="without an id"):
         await store.get(None)
 
 
+async def test_langgraph_snapshot_ingests_the_run_output():
+    pytest.importorskip("langgraph.graph")
+    from langchain_core.messages import HumanMessage
+
+    from harness_sdk import langgraph_snapshot
+
+    graph = _messages_graph()
+    checkpoints = [
+        event
+        async for namespace, mode, event in graph.astream(
+            {"messages": [HumanMessage(id="u1", content="hi")]},
+            CONFIG,
+            stream_mode=["checkpoints"],
+            subgraphs=True,
+        )
+        if namespace == ()
+    ]
+    snapshot = langgraph_snapshot(checkpoints[-1])
+    store = langgraph_checkpoints(graph, thread_id=THREAD)
+    assert snapshot == await store.get(None)
+
+
 async def test_langgraph_switch_values_replaces_the_messages_channel():
     pytest.importorskip("langgraph.graph")
     from langchain_core.messages import AIMessage, HumanMessage, RemoveMessage