sdk-py: add per-run context to thread stream run.start and run.respond

Add optional keyword-only `context` to `run.start` and `run.respond`
on both AsyncThreadStream and SyncThreadStream. The context is sent as
`params.context` on the `run.start` / `input.respond` commands when
non-None and omitted entirely otherwise, preserving the existing wire
payload for callers that don't pass it.

Co-authored-by: open-swe[bot] <open-swe@users.noreply.github.com>

.github/workflows/release.yml

@@ -139,10 +139,23 @@ jobs:
             echo EOF
           } >> "$GITHUB_OUTPUT"
 
+  test-pypi-publish:
+    needs:
+      - build
+      - release-notes
+    permissions:
+      contents: read
+      id-token: write
+    uses: ./.github/workflows/_test_release.yml
+    with:
+      working-directory: ${{ inputs.working-directory }}
+    secrets: inherit
+
   pre-release-checks:
     needs:
       - build
       - release-notes
+      - test-pypi-publish
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
@@ -167,20 +180,31 @@ jobs:
           enable-cache: false
           working-directory: ${{ inputs.working-directory }}
 
-      - uses: actions/download-artifact@3e5f45b2cfb9172054b4087a40e8e0b5a5461e7c # v8.0.1
-        with:
-          name: dist
-          path: ${{ inputs.working-directory }}/dist/
-
-      - name: Import dist package
+      - name: Import published package
         shell: bash
         working-directory: ${{ inputs.working-directory }}
         env:
           PKG_NAME: ${{ needs.build.outputs.pkg-name }}
           VERSION: ${{ needs.build.outputs.version }}
-        # Install directly from the locally-built wheel (no index resolution needed).
+        # Here we use:
+        # - The default regular PyPI index as the *primary* index, meaning
+        #   that it takes priority (https://pypi.org/simple)
+        # - The test PyPI index as an extra index, so that any dependencies that
+        #   are not found on test PyPI can be resolved and installed anyway.
+        #   (https://test.pypi.org/simple). This will include the PKG_NAME==VERSION
+        #   package because VERSION will not have been uploaded to regular PyPI yet.
+        # - attempt install again after 5 seconds if it fails because there is
+        #   sometimes a delay in availability on test pypi
         run: |
-          uv run pip install dist/*.whl
+          uv run pip install \
+            --extra-index-url https://test.pypi.org/simple/ \
+            "$PKG_NAME==$VERSION" || \
+          ( \
+            sleep 5 && \
+            uv run pip install \
+              --extra-index-url https://test.pypi.org/simple/ \
+              "$PKG_NAME==$VERSION" \
+          )
 
           if [[ "$PKG_NAME" == *prebuilt* ]]; then
             uv run pip install langgraph
@@ -202,7 +226,7 @@ jobs:
         run: uv sync --group test
         working-directory: ${{ inputs.working-directory }}
 
-      # Overwrite the local version of the package with the built version
+      # Overwrite the local version of the package with the test PyPI version.
       - name: Import published package (again)
         working-directory: ${{ inputs.working-directory }}
         shell: bash
@@ -210,25 +234,14 @@ jobs:
           PKG_NAME: ${{ needs.build.outputs.pkg-name }}
           VERSION: ${{ needs.build.outputs.version }}
         run: |
-          uv run pip install dist/*.whl
+          uv run pip install \
+            --extra-index-url https://test.pypi.org/simple/ \
+            "$PKG_NAME==$VERSION"
 
       - name: Run unit tests
         run: make test
         working-directory: ${{ inputs.working-directory }}
 
-  test-pypi-publish:
-    needs:
-      - build
-      - release-notes
-      - pre-release-checks
-    permissions:
-      contents: read
-      id-token: write
-    uses: ./.github/workflows/_test_release.yml
-    with:
-      working-directory: ${{ inputs.working-directory }}
-    secrets: inherit
-
   publish:
     needs:
       - build

CLAUDE.md

@@ -0,0 +1,65 @@
+# AGENTS Instructions
+
+This repository is a monorepo. Each library lives in a subdirectory under `libs/`.
+
+<corridor>
+
+## Corridor security analysis
+
+Before generating or modifying code, create a plan and use Corridor's `analyzePlan` tool to analyze it. Apply the resulting security guidance before writing code.
+
+</corridor>
+
+When you modify code in any library, run the following commands in that library's directory before creating a pull request:
+
+- `make format` – run code formatters
+- `make lint` – run the linter
+- `make test` – execute the test suite
+
+To run a particular test file or to pass additional pytest options you can specify the `TEST` variable:
+
+```
+TEST=path/to/test.py make test
+```
+
+Other pytest arguments can also be supplied inside the `TEST` variable.
+
+## Libraries
+
+The repository contains several Python and JavaScript/TypeScript libraries.
+Below is a high-level overview:
+
+- **checkpoint** – base interfaces for LangGraph checkpointers.
+- **checkpoint-postgres** – Postgres implementation of the checkpoint saver.
+- **checkpoint-sqlite** – SQLite implementation of the checkpoint saver.
+- **cli** – official command-line interface for LangGraph.
+- **langgraph** – core framework for building stateful, multi-actor agents.
+- **prebuilt** – high-level APIs for creating and running agents and tools.
+- **sdk-js** – JS/TS SDK for interacting with the LangGraph REST API.
+- **sdk-py** – Python SDK for the LangGraph Server API.
+
+### Dependency map
+
+The diagram below lists downstream libraries for each production dependency as
+declared in that library's `pyproject.toml` (or `package.json`).
+
+```text
+checkpoint
+├── checkpoint-postgres
+├── checkpoint-sqlite
+├── prebuilt
+└── langgraph
+
+prebuilt
+└── langgraph
+
+sdk-py
+├── langgraph
+└── cli
+
+sdk-js (standalone)
+```
+
+Changes to a library may impact all of its dependents shown above.
+
+- Do NOT use Sphinx-style double backtick formatting (` ``code`` `). Use single backticks (`` `code` ``) for inline code references in docstrings and comments.

libs/checkpoint-conformance/uv.lock

@@ -13,16 +13,16 @@ wheels = [
 
 [[package]]
 name = "anyio"
-version = "4.14.2"
+version = "4.13.0"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "exceptiongroup", marker = "python_full_version < '3.11'" },
     { name = "idna" },
     { name = "typing-extensions", marker = "python_full_version < '3.13'" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/61/cc/a381afa6efea9f496eff839d4a6a1aed3bfafc7b3ab4b0d1b243a12573dd/anyio-4.14.2.tar.gz", hash = "sha256:cfa139f3ed1a23ee8f88a145ddb5ac7605b8bbfd8592baacd7ce3d8bb4313c7f", size = 260176, upload-time = "2026-07-12T20:29:07.082Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/19/14/2c5dd9f512b66549ae92767a9c7b330ae88e1932ca57876909410251fe13/anyio-4.13.0.tar.gz", hash = "sha256:334b70e641fd2221c1505b3890c69882fe4a2df910cba14d97019b90b24439dc", size = 231622, upload-time = "2026-03-24T12:59:09.671Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/da/35/f2287558c17e29fafc8ef3daf819bb9834061cfa43bff8014f7df7f63bdc/anyio-4.14.2-py3-none-any.whl", hash = "sha256:9f505dda5ac9f0c8309b5e8bd445a8c2bf7246f3ce950121e45ea15bc41d1494", size = 125813, upload-time = "2026-07-12T20:29:05.763Z" },
+    { url = "https://files.pythonhosted.org/packages/da/42/e921fccf5015463e32a3cf6ee7f980a6ed0f395ceeaa45060b61d86486c2/anyio-4.13.0-py3-none-any.whl", hash = "sha256:08b310f9e24a9594186fd75b4f73f4a4152069e3853f1ed8bfbf58369f4ad708", size = 114353, upload-time = "2026-03-24T12:59:08.246Z" },
 ]
 
 [[package]]

libs/checkpoint-postgres/langgraph/checkpoint/postgres/__init__.py

@@ -448,12 +448,11 @@ class PostgresSaver(BasePostgresSaver):
 
         Two-stage query, both stages cover ALL requested channels:
 
-        * Stage 1 (paged): dynamic SELECT over `checkpoints` with three
-          columns per channel: its version, an `EXISTS` probe for a stored
-          blob at that version, and its inline value. Pages newest-first by
-          `checkpoint_id` with a cursor; page size is `_DELTA_PAGE_SIZE`.
-          Stops paging when every channel has found its seed or a page comes
-          back short.
+        * Stage 1 (paged): dynamic SELECT over `checkpoints` with K parallel
+          JSONB key lookups (one column pair per channel) — no subquery, no
+          aggregation. Pages newest-first by `checkpoint_id` with a cursor;
+          page size is `_DELTA_PAGE_SIZE`. Stops paging when every channel
+          has found its seed or the chain is exhausted.
 
         * Stage 2 (per-channel UNION ALL): one branch per channel reading
           `checkpoint_writes` filtered to that channel's specific

libs/checkpoint-postgres/langgraph/checkpoint/postgres/base.py

@@ -172,8 +172,30 @@ class _DeltaStage2Row(TypedDict, total=False):
     version: str | None  # "b" rows only
 
 
-# Delta history is rebuilt in two queries; `_build_delta_stage1_sql` and
-# `_build_delta_stage2_sql` document their shapes.
+# Multi-channel two-stage DeltaChannel reconstruction.
+#
+# Stage 1 scans checkpoint metadata (no blob bytes) and emits one row per
+# checkpoint with K parallel JSONB key lookups (one column pair per
+# requested delta channel: ver_i / hs_i).  No subqueries, no aggregation.
+# Python walks the parent chain once across all channels.
+#
+# Stage 2 fetches all writes and the seed blobs for ALL channels in a
+# single roundtrip via `channel = ANY(%s)` and chain/seed-version
+# filtering.
+#
+# Empirical comparison vs an alternative "ship full channel_versions /
+# channel_values JSONB and let Python pick" form (1000 checkpoints,
+# 8 total channels in graph, 3 delta channels requested):
+#
+#   Postgres execution:    A=0.24ms vs B=0.38ms   (both negligible)
+#   End-to-end latency:    A=6.83ms vs B=2.28ms   (B is 3.0x faster)
+#   Wire payload:          A=836KB  vs B=330KB    (61% smaller)
+#   Buffer hits:           identical (167 blocks)
+#
+# B (this dynamic-columns design) wins because it avoids JSONB
+# serialization on the wire and JSONB-to-dict deserialization in
+# psycopg.  Even at K=8 (8 delta channels = 16 dynamic columns), B
+# still beats A end-to-end (4.2ms vs 6.8ms).
 
 
 def _build_delta_stage1_sql(channels: Sequence[str], *, paged: bool) -> str:
@@ -313,8 +335,10 @@ def _build_delta_stage2_sql(
     return " UNION ALL ".join(branches)
 
 
-# Stage 1 rows are dicts keyed by the per-channel aliases
-# `_build_delta_stage1_sql` emits, so there is no static TypedDict.
+# Stage 1 rows are dynamic-shape dicts: {checkpoint_id, parent_checkpoint_id,
+# ver_0, hs_0, ver_1, hs_1, ...}.  Walking is parameterized by the channel
+# list to map indices back to channel names — no static TypedDict here.
+# `dict[str, Any]` is the practical signature.
 
 
 class BasePostgresSaver(BaseCheckpointSaver[str]):
@@ -407,11 +431,9 @@ class BasePostgresSaver(BaseCheckpointSaver[str]):
           (a) it found a stored value for its channel — a blob or an inline
               primitive (channel becomes seeded),
           (b) it reached a real root (parent_of[cid] is None — fully
-              materialized at this point),
+              materialized at this point), or
           (c) the next ancestor cid isn't in `parent_of` yet (waiting for
-              a later page; the cursor stays put), or
-          (d) the target's own row isn't in `parent_of` yet (the walk has
-              not started; no cursor is set, so a later page retries).
+              a later page; the cursor stays put).
 
         Mutates `chain_by_ch`, `seed_ver_by_ch`, `seed_inline_by_ch`,
         `walk_cursor_by_ch`, and `seeded` in place.
@@ -419,12 +441,9 @@ class BasePostgresSaver(BaseCheckpointSaver[str]):
         for i, ch in enumerate(channels):
             if ch in seeded:
                 continue
-            # Pages start at the thread head, so the target may not have
-            # loaded yet; a `None` cursor would read as "target is a root".
+            # First-time entry: cursor starts at the target's parent.
             if ch not in walk_cursor_by_ch:
-                if target_id not in parent_of:
-                    continue
-                walk_cursor_by_ch[ch] = parent_of[target_id]
+                walk_cursor_by_ch[ch] = parent_of.get(target_id)
             cur_cid = walk_cursor_by_ch[ch]
             ch_chain = chain_by_ch[ch]
             hb_i = hb_by_i_by_cid[i]

libs/checkpoint-postgres/tests/test_delta_pagination.py

@@ -1,132 +0,0 @@
-from __future__ import annotations
-
-from typing import Any
-from uuid import uuid4
-
-import pytest
-from langgraph.checkpoint.base import (
-    Checkpoint,
-    DeltaChannelHistory,
-    empty_checkpoint,
-)
-from langgraph.checkpoint.base.id import uuid6
-from langgraph.checkpoint.serde.types import _DeltaSnapshot
-
-from langgraph.checkpoint.postgres import PostgresSaver
-from langgraph.checkpoint.postgres.aio import AsyncPostgresSaver
-from langgraph.checkpoint.postgres.base import _DELTA_PAGE_SIZE
-from tests.conftest import DEFAULT_URI
-
-CHANNEL = "items"
-STEPS = 8
-SEED_STEP = 1
-SEED_VALUE = [10, 20]
-TARGET_STEP = 4
-
-# The real page size is the control; the rest leave the target off the first
-# page (three checkpoints are newer than it).
-PAGE_SIZES = [_DELTA_PAGE_SIZE, 3, 2, 1]
-
-
-def _step_args(
-    thread_id: str, step: int, parent: dict | None
-) -> tuple[dict, Checkpoint, dict[str, Any]]:
-    config: dict = {"configurable": {"thread_id": thread_id, "checkpoint_ns": ""}}
-    if parent is not None:
-        config["configurable"]["checkpoint_id"] = parent["configurable"][
-            "checkpoint_id"
-        ]
-    checkpoint: Checkpoint = empty_checkpoint()
-    checkpoint["id"] = str(uuid6(clock_seq=step))
-    checkpoint["channel_versions"][CHANNEL] = f"v{step}"
-    if step == SEED_STEP:
-        checkpoint["channel_values"][CHANNEL] = _DeltaSnapshot(list(SEED_VALUE))
-        return config, checkpoint, {CHANNEL: f"v{step}"}
-    return config, checkpoint, {}
-
-
-async def _abuild_chain(saver: AsyncPostgresSaver) -> list[dict]:
-    thread_id = str(uuid4())
-    parent: dict | None = None
-    configs: list[dict] = []
-    for step in range(STEPS):
-        config, checkpoint, new_versions = _step_args(thread_id, step, parent)
-        parent = await saver.aput(
-            config,
-            checkpoint,
-            {"source": "loop", "step": step, "parents": {}},
-            new_versions,
-        )
-        await saver.aput_writes(parent, [(CHANNEL, f"w{step}")], str(uuid4()))
-        configs.append(parent)
-    return configs
-
-
-def _build_chain(saver: PostgresSaver) -> list[dict]:
-    thread_id = str(uuid4())
-    parent: dict | None = None
-    configs: list[dict] = []
-    for step in range(STEPS):
-        config, checkpoint, new_versions = _step_args(thread_id, step, parent)
-        parent = saver.put(
-            config,
-            checkpoint,
-            {"source": "loop", "step": step, "parents": {}},
-            new_versions,
-        )
-        saver.put_writes(parent, [(CHANNEL, f"w{step}")], str(uuid4()))
-        configs.append(parent)
-    return configs
-
-
-def _assert_history(entry: DeltaChannelHistory, page_size: int) -> None:
-    seed = entry.get("seed")
-    assert isinstance(seed, _DeltaSnapshot), (
-        f"page_size={page_size}: expected a snapshot seed, "
-        f"got {entry.get('seed', '<missing>')!r}"
-    )
-    assert seed.value == SEED_VALUE
-    assert [w[2] for w in entry["writes"]] == ["w1", "w2", "w3"], (
-        f"page_size={page_size}: got {[w[2] for w in entry['writes']]}"
-    )
-
-
-@pytest.mark.parametrize("page_size", PAGE_SIZES)
-async def test_async_target_older_than_the_first_page(
-    page_size: int, monkeypatch: pytest.MonkeyPatch
-) -> None:
-    monkeypatch.setattr("langgraph.checkpoint.postgres.aio._DELTA_PAGE_SIZE", page_size)
-    async with AsyncPostgresSaver.from_conn_string(DEFAULT_URI) as saver:
-        await saver.setup()
-        configs = await _abuild_chain(saver)
-        result = await saver.aget_delta_channel_history(
-            config=configs[TARGET_STEP], channels=[CHANNEL]
-        )
-        _assert_history(result[CHANNEL], page_size)
-
-
-@pytest.mark.parametrize("page_size", PAGE_SIZES)
-def test_sync_target_older_than_the_first_page(
-    page_size: int, monkeypatch: pytest.MonkeyPatch
-) -> None:
-    monkeypatch.setattr("langgraph.checkpoint.postgres._DELTA_PAGE_SIZE", page_size)
-    with PostgresSaver.from_conn_string(DEFAULT_URI) as saver:
-        saver.setup()
-        configs = _build_chain(saver)
-        result = saver.get_delta_channel_history(
-            config=configs[TARGET_STEP], channels=[CHANNEL]
-        )
-        _assert_history(result[CHANNEL], page_size)
-
-
-async def test_root_target_has_no_history_and_still_terminates(
-    monkeypatch: pytest.MonkeyPatch,
-) -> None:
-    monkeypatch.setattr("langgraph.checkpoint.postgres.aio._DELTA_PAGE_SIZE", 1)
-    async with AsyncPostgresSaver.from_conn_string(DEFAULT_URI) as saver:
-        await saver.setup()
-        configs = await _abuild_chain(saver)
-        result = await saver.aget_delta_channel_history(
-            config=configs[0], channels=[CHANNEL]
-        )
-        assert result[CHANNEL] == {"writes": []}

libs/checkpoint-sqlite/langgraph/checkpoint/sqlite/__init__.py

@@ -507,12 +507,13 @@ class SqliteSaver(BaseCheckpointSaver[str]):
 
         Two-stage query:
 
-        * Stage 1 (streamed): recursive CTE over `checkpoints` following
-          `parent_checkpoint_id` from the target, returning
-          `(checkpoint_id, type, checkpoint)` per ancestor. Sqlite has no
-          JSONB, so we ship the full serialized checkpoint blob and inspect
-          `channel_values` in Python. Stops reading when every channel has
-          found its seed or the chain is exhausted.
+        * Stage 1 (paged): newest-first slice of `checkpoints` returning
+          `(checkpoint_id, parent_checkpoint_id, type, checkpoint)` per
+          ancestor. Sqlite has no JSONB, so we ship the full serialized
+          checkpoint blob and inspect `channel_values` in Python. Pages
+          newest-first by `checkpoint_id` with a `< cursor` predicate;
+          page size is `DELTA_PAGE_SIZE`. Stops paging when every channel
+          has found its seed or the chain is exhausted.
 
         * Stage 2 (per-channel UNION ALL): one branch per channel reading
           `writes` filtered to that channel's specific `chain_cids`. No
@@ -537,14 +538,12 @@ class SqliteSaver(BaseCheckpointSaver[str]):
         seeded: set[str] = set()
 
         with self.cursor(transaction=False) as cur:
-            cur.execute(
-                DELTA_STAGE1_SQL,
-                (thread_id, checkpoint_ns, checkpoint_id, thread_id, checkpoint_ns),
-            )
+            cur.execute(DELTA_STAGE1_SQL, (thread_id, checkpoint_ns, checkpoint_id))
             for row in cur:
-                cid, type_tag, blob = row
+                cid, parent_cid, type_tag, blob = row
                 if step_walk_with_row(
                     cid=cid,
+                    parent_cid=parent_cid,
                     type_tag=type_tag,
                     blob=blob,
                     target_id=checkpoint_id,

libs/checkpoint-sqlite/langgraph/checkpoint/sqlite/_delta.py

@@ -26,33 +26,16 @@ from typing import Any
 
 from langgraph.checkpoint.base import DeltaChannelHistory, PendingWrite
 
-# Stage 1 streams target, then its ancestors nearest-first, by following
-# `parent_checkpoint_id` rather than id order: ids are only monotonic within
-# one process, so a range scan by id can miss a parent whose id sorts above
-# its child's. Target is the anchor row; its own writes/seed are skipped
-# (matches the `BaseCheckpointSaver` contract).
-#
-# `put` is `INSERT OR REPLACE`, so re-putting an existing id under a
-# descendant's config makes the chain a loop. `step_walk_with_row` stops on a
-# repeated id; sqlite yields recursive rows lazily, so abandoning the cursor
-# ends the recursion.
-#
-# `CROSS JOIN` pins `ancestors` as the outer loop, so each step is one primary
-# key lookup. With a plain `JOIN` and no `ANALYZE` stats, sqlite can put
-# `checkpoints` outside and scan the whole thread per step.
+# Stage 1 streams ancestors of `target_cid` newest-first. The `<=`
+# predicate keeps target itself in the stream so we can read its
+# `parent_checkpoint_id` from the first row without a separate lookup;
+# the caller skips target's own writes/seed (matches the
+# `BaseCheckpointSaver` contract).
 DELTA_STAGE1_SQL = (
-    "WITH RECURSIVE ancestors(checkpoint_id, parent_checkpoint_id, type, "
-    "checkpoint) AS ("
     "SELECT checkpoint_id, parent_checkpoint_id, type, checkpoint "
     "FROM checkpoints "
-    "WHERE thread_id = ? AND checkpoint_ns = ? AND checkpoint_id = ? "
-    "UNION ALL "
-    "SELECT c.checkpoint_id, c.parent_checkpoint_id, c.type, c.checkpoint "
-    "FROM ancestors a CROSS JOIN checkpoints c "
-    "ON c.checkpoint_id = a.parent_checkpoint_id "
-    "WHERE c.thread_id = ? AND c.checkpoint_ns = ?"
-    ") "
-    "SELECT checkpoint_id, type, checkpoint FROM ancestors"
+    "WHERE thread_id = ? AND checkpoint_ns = ? AND checkpoint_id <= ? "
+    "ORDER BY checkpoint_id DESC"
 )
 
 
@@ -85,6 +68,7 @@ def build_delta_stage2_sql(*, chain_lens: Sequence[int]) -> str:
 def step_walk_with_row(
     *,
     cid: str,
+    parent_cid: str | None,
     type_tag: str,
     blob: bytes,
     target_id: str,
@@ -97,32 +81,36 @@ def step_walk_with_row(
 ) -> bool:
     """Process one streamed stage-1 row in the merged ancestor walk.
 
-    The cursor returns (cid, type, blob) rows in walk order starting at
-    target. The first row is target itself and is skipped (target's own
-    writes/seed are not part of the contract).
+    The cursor returns (cid, parent_cid, type, blob) rows in
+    `checkpoint_id` DESC order starting at target. The first row is
+    target itself; we read its parent_cid to seed the walk and otherwise
+    skip it (target's own writes/seed are not part of the contract).
 
-    For each subsequent row we deserialize the blob, append the cid to
-    every not-yet-seeded channel's chain, and check `channel_values` for
+    For each subsequent row, if `cid` matches the walk's current
+    position, we deserialize the blob, append the cid to every
+    not-yet-seeded channel's chain, and check `channel_values` for
     seeds. The deserialized checkpoint is dropped before advancing — no
     cross-row cache, so peak in-flight is one deserialized checkpoint.
 
-    Returns True when the caller can stop iterating and close the cursor:
-    every requested channel is seeded, or the chain revisited a checkpoint.
+    Off-path rows (different branch on the same thread) advance the
+    cursor without doing any work.
+
+    Returns True when every requested channel is seeded — the caller
+    can stop iterating and close the cursor.
     """
     if "started" not in walk_state:
         if cid == target_id:
             walk_state["started"] = True
+            walk_state["cur_cid"] = parent_cid
             walk_state["active"] = {ch for ch in channels if ch not in seeded}
-            walk_state["walked"] = {cid}
         # Not target yet (or target not present): keep streaming.
         return False
     active: set[str] = walk_state["active"]
     if not active:
         return True
-    walked: set[str] = walk_state["walked"]
-    if cid in walked:
-        return True
-    walked.add(cid)
+    if cid != walk_state["cur_cid"]:
+        # Off-path row from a sibling branch — skip without deserializing.
+        return False
     for ch in active:
         chain_by_ch[ch].append(cid)
     ckpt = serde.loads_typed((type_tag, blob))
@@ -132,6 +120,7 @@ def step_walk_with_row(
         seeded.add(ch)
         active.discard(ch)
     del ckpt, channel_values
+    walk_state["cur_cid"] = parent_cid
     return not active
 
 

libs/checkpoint-sqlite/langgraph/checkpoint/sqlite/aio.py

@@ -625,8 +625,8 @@ class AsyncSqliteSaver(BaseCheckpointSaver[str]):
         """Fast-path override of `BaseCheckpointSaver.aget_delta_channel_history`.
 
         See `SqliteSaver.get_delta_channel_history` for design notes; this
-        is the async equivalent using `aiosqlite` cursors. Stage 1 streams
-        the parent chain from the target and Python-deserializes each
+        is the async equivalent using `aiosqlite` cursors. Stage 1 pages
+        the parent chain newest-first and Python-deserializes each
         checkpoint blob to find per-channel snapshots; stage 2 fetches
         only the relevant writes via per-channel UNION ALL.
         """
@@ -650,13 +650,13 @@ class AsyncSqliteSaver(BaseCheckpointSaver[str]):
 
         async with self.lock, self.conn.cursor() as cur:
             await cur.execute(
-                DELTA_STAGE1_SQL,
-                (thread_id, checkpoint_ns, checkpoint_id, thread_id, checkpoint_ns),
+                DELTA_STAGE1_SQL, (thread_id, checkpoint_ns, checkpoint_id)
             )
             async for row in cur:
-                cid, type_tag, blob = row
+                cid, parent_cid, type_tag, blob = row
                 if step_walk_with_row(
                     cid=cid,
+                    parent_cid=parent_cid,
                     type_tag=type_tag,
                     blob=blob,
                     target_id=checkpoint_id,

libs/checkpoint-sqlite/tests/test_delta_parent_walk.py

@@ -1,104 +0,0 @@
-from __future__ import annotations
-
-from typing import Any
-
-import pytest
-from langgraph.checkpoint.base import (
-    BaseCheckpointSaver,
-    Checkpoint,
-    DeltaChannelHistory,
-    empty_checkpoint,
-)
-
-from langgraph.checkpoint.sqlite import SqliteSaver
-from langgraph.checkpoint.sqlite._delta import DELTA_STAGE1_SQL
-from langgraph.checkpoint.sqlite.aio import AsyncSqliteSaver
-
-CHANNEL = "ch"
-CONFIG: dict[str, Any] = {"configurable": {"thread_id": "t", "checkpoint_ns": ""}}
-EXPECTED: DeltaChannelHistory = {
-    "writes": [("task", CHANNEL, "write-root")],
-    "seed": "seed",
-}
-
-
-def _checkpoint(checkpoint_id: str, values: dict[str, Any]) -> Checkpoint:
-    value = empty_checkpoint()
-    value["id"] = checkpoint_id
-    value["channel_values"] = values
-    return value
-
-
-PARENT_ID_ORDERS = [
-    pytest.param("z-older", "a-newer", id="parent_id_sorts_above_child"),
-    pytest.param("a-older", "z-newer", id="parent_id_sorts_below_child"),
-]
-
-
-@pytest.mark.parametrize(("root_id", "child_id"), PARENT_ID_ORDERS)
-def test_sync_walk_reaches_parent_whatever_the_id_order(
-    root_id: str, child_id: str
-) -> None:
-    with SqliteSaver.from_conn_string(":memory:") as saver:
-        root = saver.put(CONFIG, _checkpoint(root_id, {CHANNEL: "seed"}), {}, {})
-        saver.put_writes(root, [(CHANNEL, "write-root")], "task")
-        child = saver.put(root, _checkpoint(child_id, {}), {}, {})
-
-        got = saver.get_delta_channel_history(config=child, channels=[CHANNEL])
-        reference = BaseCheckpointSaver.get_delta_channel_history(
-            saver, config=child, channels=[CHANNEL]
-        )
-        assert got[CHANNEL] == EXPECTED
-        assert got[CHANNEL] == reference[CHANNEL], "fast path disagrees with base"
-
-
-@pytest.mark.parametrize(("root_id", "child_id"), PARENT_ID_ORDERS)
-async def test_async_walk_reaches_parent_whatever_the_id_order(
-    root_id: str, child_id: str
-) -> None:
-    async with AsyncSqliteSaver.from_conn_string(":memory:") as saver:
-        root = await saver.aput(CONFIG, _checkpoint(root_id, {CHANNEL: "seed"}), {}, {})
-        await saver.aput_writes(root, [(CHANNEL, "write-root")], "task")
-        child = await saver.aput(root, _checkpoint(child_id, {}), {}, {})
-
-        got = await saver.aget_delta_channel_history(config=child, channels=[CHANNEL])
-        assert got[CHANNEL] == EXPECTED
-
-
-def test_walk_reaches_root_of_long_chain_with_descending_ids() -> None:
-    steps = 40
-    with SqliteSaver.from_conn_string(":memory:") as saver:
-        parent = saver.put(
-            CONFIG, _checkpoint(f"id-{steps:03d}", {CHANNEL: "seed"}), {}, {}
-        )
-        saver.put_writes(parent, [(CHANNEL, "write-root")], "task")
-        for step in range(steps - 1, 0, -1):
-            parent = saver.put(parent, _checkpoint(f"id-{step:03d}", {}), {}, {})
-
-        got = saver.get_delta_channel_history(config=parent, channels=[CHANNEL])
-        assert got[CHANNEL] == EXPECTED
-
-
-def test_walk_terminates_when_put_makes_the_parent_chain_cycle() -> None:
-    with SqliteSaver.from_conn_string(":memory:") as saver:
-        a = saver.put(CONFIG, _checkpoint("cid-a", {}), {}, {})
-        b = saver.put(a, _checkpoint("cid-b", {}), {}, {})
-        repoint_a_under_b = _checkpoint("cid-a", {})
-        saver.put(b, repoint_a_under_b, {}, {})
-
-        got = saver.get_delta_channel_history(config=b, channels=[CHANNEL])
-        assert got[CHANNEL] == {"writes": []}
-
-
-def test_walk_step_looks_up_the_parent_by_primary_key() -> None:
-    with SqliteSaver.from_conn_string(":memory:") as saver:
-        saver.setup()
-        plan = [
-            row[3]
-            for row in saver.conn.execute(
-                f"EXPLAIN QUERY PLAN {DELTA_STAGE1_SQL}", ("t", "", "id", "t", "")
-            )
-        ]
-    assert any(
-        step.startswith("SEARCH c ") and "checkpoint_id=?" in step for step in plan
-    ), f"recursive step should look up the parent by key, got {plan}"

libs/checkpoint-sqlite/uv.lock

@@ -26,16 +26,16 @@ wheels = [
 
 [[package]]
 name = "anyio"
-version = "4.14.2"
+version = "4.12.1"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "exceptiongroup", marker = "python_full_version < '3.11'" },
     { name = "idna" },
     { name = "typing-extensions", marker = "python_full_version < '3.13'" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/61/cc/a381afa6efea9f496eff839d4a6a1aed3bfafc7b3ab4b0d1b243a12573dd/anyio-4.14.2.tar.gz", hash = "sha256:cfa139f3ed1a23ee8f88a145ddb5ac7605b8bbfd8592baacd7ce3d8bb4313c7f", size = 260176, upload-time = "2026-07-12T20:29:07.082Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/96/f0/5eb65b2bb0d09ac6776f2eb54adee6abe8228ea05b20a5ad0e4945de8aac/anyio-4.12.1.tar.gz", hash = "sha256:41cfcc3a4c85d3f05c932da7c26d0201ac36f72abd4435ba90d0464a3ffed703", size = 228685, upload-time = "2026-01-06T11:45:21.246Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/da/35/f2287558c17e29fafc8ef3daf819bb9834061cfa43bff8014f7df7f63bdc/anyio-4.14.2-py3-none-any.whl", hash = "sha256:9f505dda5ac9f0c8309b5e8bd445a8c2bf7246f3ce950121e45ea15bc41d1494", size = 125813, upload-time = "2026-07-12T20:29:05.763Z" },
+    { url = "https://files.pythonhosted.org/packages/38/0e/27be9fdef66e72d64c0cdc3cc2823101b80585f8119b5c112c2e8f5f7dab/anyio-4.12.1-py3-none-any.whl", hash = "sha256:d405828884fc140aa80a3c667b8beed277f1dfedec42ba031bd6ac3db606ab6c", size = 113592, upload-time = "2026-01-06T11:45:19.497Z" },
 ]
 
 [[package]]

libs/checkpoint/uv.lock

@@ -19,16 +19,16 @@ wheels = [
 
 [[package]]
 name = "anyio"
-version = "4.14.2"
+version = "4.12.1"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "exceptiongroup", marker = "python_full_version < '3.11'" },
     { name = "idna" },
     { name = "typing-extensions", marker = "python_full_version < '3.13'" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/61/cc/a381afa6efea9f496eff839d4a6a1aed3bfafc7b3ab4b0d1b243a12573dd/anyio-4.14.2.tar.gz", hash = "sha256:cfa139f3ed1a23ee8f88a145ddb5ac7605b8bbfd8592baacd7ce3d8bb4313c7f", size = 260176, upload-time = "2026-07-12T20:29:07.082Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/96/f0/5eb65b2bb0d09ac6776f2eb54adee6abe8228ea05b20a5ad0e4945de8aac/anyio-4.12.1.tar.gz", hash = "sha256:41cfcc3a4c85d3f05c932da7c26d0201ac36f72abd4435ba90d0464a3ffed703", size = 228685, upload-time = "2026-01-06T11:45:21.246Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/da/35/f2287558c17e29fafc8ef3daf819bb9834061cfa43bff8014f7df7f63bdc/anyio-4.14.2-py3-none-any.whl", hash = "sha256:9f505dda5ac9f0c8309b5e8bd445a8c2bf7246f3ce950121e45ea15bc41d1494", size = 125813, upload-time = "2026-07-12T20:29:05.763Z" },
+    { url = "https://files.pythonhosted.org/packages/38/0e/27be9fdef66e72d64c0cdc3cc2823101b80585f8119b5c112c2e8f5f7dab/anyio-4.12.1-py3-none-any.whl", hash = "sha256:d405828884fc140aa80a3c667b8beed277f1dfedec42ba031bd6ac3db606ab6c", size = 113592, upload-time = "2026-01-06T11:45:19.497Z" },
 ]
 
 [[package]]

libs/cli/README.md

@@ -103,8 +103,6 @@ The CLI uses a `langgraph.json` configuration file with these key settings:
 }
 ```
 
-Git dependencies should use credential-free URLs. The CLI conservatively scans direct `langgraph.json` dependencies, common Python package files, uv project and lock files, and common Node.js package and lock files for HTTP Git URLs with userinfo. This check is not exhaustive: generated Docker builds can copy other files, including nested requirement or constraint files, into image layers without scanning them. For private dependencies, provide short-lived credentials through your build environment's secret-backed Git credential helper. Do not store credentials in copied files such as `langgraph.json` or `pip_config_file`.
-
 See the [full documentation](https://reference.langchain.com/python/langgraph-cli) for detailed configuration options.
 
 ## Development

libs/cli/examples/poetry.lock

@@ -0,0 +1,285 @@
+# This file is automatically @generated by Poetry 2.0.0 and should not be changed by hand.
+
+[[package]]
+name = "anyio"
+version = "4.4.0"
+description = "High level compatibility layer for multiple asynchronous event loop implementations"
+optional = false
+python-versions = ">=3.8"
+groups = ["main"]
+files = [
+    {file = "anyio-4.4.0-py3-none-any.whl", hash = "sha256:c1b2d8f46a8a812513012e1107cb0e68c17159a7a594208005a57dc776e1bdc7"},
+    {file = "anyio-4.4.0.tar.gz", hash = "sha256:5aadc6a1bbb7cdb0bede386cac5e2940f5e2ff3aa20277e991cf028e0585ce94"},
+]
+
+[package.dependencies]
+exceptiongroup = {version = ">=1.0.2", markers = "python_version < \"3.11\""}
+idna = ">=2.8"
+sniffio = ">=1.1"
+typing-extensions = {version = ">=4.1", markers = "python_version < \"3.11\""}
+
+[package.extras]
+doc = ["Sphinx (>=7)", "packaging", "sphinx-autodoc-typehints (>=1.2.0)", "sphinx-rtd-theme"]
+test = ["anyio[trio]", "coverage[toml] (>=7)", "exceptiongroup (>=1.2.0)", "hypothesis (>=4.0)", "psutil (>=5.9)", "pytest (>=7.0)", "pytest-mock (>=3.6.1)", "trustme", "uvloop (>=0.17)"]
+trio = ["trio (>=0.23)"]
+
+[[package]]
+name = "certifi"
+version = "2024.7.4"
+description = "Python package for providing Mozilla's CA Bundle."
+optional = false
+python-versions = ">=3.6"
+groups = ["main"]
+files = [
+    {file = "certifi-2024.7.4-py3-none-any.whl", hash = "sha256:c198e21b1289c2ab85ee4e67bb4b4ef3ead0892059901a8d5b622f24a1101e90"},
+    {file = "certifi-2024.7.4.tar.gz", hash = "sha256:5a1e7645bc0ec61a09e26c36f6106dd4cf40c6db3a1fb6352b0244e7fb057c7b"},
+]
+
+[[package]]
+name = "click"
+version = "8.1.7"
+description = "Composable command line interface toolkit"
+optional = false
+python-versions = ">=3.7"
+groups = ["main"]
+files = [
+    {file = "click-8.1.7-py3-none-any.whl", hash = "sha256:ae74fb96c20a0277a1d615f1e4d73c8414f5a98db8b799a7931d1582f3390c28"},
+    {file = "click-8.1.7.tar.gz", hash = "sha256:ca9853ad459e787e2192211578cc907e7594e294c7ccc834310722b41b9ca6de"},
+]
+
+[package.dependencies]
+colorama = {version = "*", markers = "platform_system == \"Windows\""}
+
+[[package]]
+name = "colorama"
+version = "0.4.6"
+description = "Cross-platform colored terminal text."
+optional = false
+python-versions = "!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*,!=3.4.*,!=3.5.*,!=3.6.*,>=2.7"
+groups = ["main"]
+markers = "platform_system == \"Windows\""
+files = [
+    {file = "colorama-0.4.6-py2.py3-none-any.whl", hash = "sha256:4f1d9991f5acc0ca119f9d443620b77f9d6b33703e51011c16baf57afb285fc6"},
+    {file = "colorama-0.4.6.tar.gz", hash = "sha256:08695f5cb7ed6e0531a20572697297273c47b8cae5a63ffc6d6ed5c201be6e44"},
+]
+
+[[package]]
+name = "exceptiongroup"
+version = "1.2.1"
+description = "Backport of PEP 654 (exception groups)"
+optional = false
+python-versions = ">=3.7"
+groups = ["main"]
+markers = "python_version < \"3.11\""
+files = [
+    {file = "exceptiongroup-1.2.1-py3-none-any.whl", hash = "sha256:5258b9ed329c5bbdd31a309f53cbfb0b155341807f6ff7606a1e801a891b29ad"},
+    {file = "exceptiongroup-1.2.1.tar.gz", hash = "sha256:a4785e48b045528f5bfe627b6ad554ff32def154f42372786903b7abcfe1aa16"},
+]
+
+[package.extras]
+test = ["pytest (>=6)"]
+
+[[package]]
+name = "h11"
+version = "0.16.0"
+description = "A pure-Python, bring-your-own-I/O implementation of HTTP/1.1"
+optional = false
+python-versions = ">=3.8"
+groups = ["main"]
+files = [
+    {file = "h11-0.16.0-py3-none-any.whl", hash = "sha256:63cf8bbe7522de3bf65932fda1d9c2772064ffb3dae62d55932da54b31cb6c86"},
+    {file = "h11-0.16.0.tar.gz", hash = "sha256:4e35b956cf45792e4caa5885e69fba00bdbc6ffafbfa020300e549b208ee5ff1"},
+]
+
+[[package]]
+name = "httpcore"
+version = "1.0.9"
+description = "A minimal low-level HTTP client."
+optional = false
+python-versions = ">=3.8"
+groups = ["main"]
+files = [
+    {file = "httpcore-1.0.9-py3-none-any.whl", hash = "sha256:2d400746a40668fc9dec9810239072b40b4484b640a8c38fd654a024c7a1bf55"},
+    {file = "httpcore-1.0.9.tar.gz", hash = "sha256:6e34463af53fd2ab5d807f399a9b45ea31c3dfa2276f15a2c3f00afff6e176e8"},
+]
+
+[package.dependencies]
+certifi = "*"
+h11 = ">=0.16"
+
+[package.extras]
+asyncio = ["anyio (>=4.0,<5.0)"]
+http2 = ["h2 (>=3,<5)"]
+socks = ["socksio (==1.*)"]
+trio = ["trio (>=0.22.0,<1.0)"]
+
+[[package]]
+name = "httpx"
+version = "0.28.1"
+description = "The next generation HTTP client."
+optional = false
+python-versions = ">=3.8"
+groups = ["main"]
+files = [
+    {file = "httpx-0.28.1-py3-none-any.whl", hash = "sha256:d909fcccc110f8c7faf814ca82a9a4d816bc5a6dbfea25d6591d6985b8ba59ad"},
+    {file = "httpx-0.28.1.tar.gz", hash = "sha256:75e98c5f16b0f35b567856f597f06ff2270a374470a5c2392242528e3e3e42fc"},
+]
+
+[package.dependencies]
+anyio = "*"
+certifi = "*"
+httpcore = "==1.*"
+idna = "*"
+
+[package.extras]
+brotli = ["brotli", "brotlicffi"]
+cli = ["click (==8.*)", "pygments (==2.*)", "rich (>=10,<14)"]
+http2 = ["h2 (>=3,<5)"]
+socks = ["socksio (==1.*)"]
+zstd = ["zstandard (>=0.18.0)"]
+
+[[package]]
+name = "httpx-sse"
+version = "0.4.0"
+description = "Consume Server-Sent Event (SSE) messages with HTTPX."
+optional = false
+python-versions = ">=3.8"
+groups = ["main"]
+files = [
+    {file = "httpx-sse-0.4.0.tar.gz", hash = "sha256:1e81a3a3070ce322add1d3529ed42eb5f70817f45ed6ec915ab753f961139721"},
+    {file = "httpx_sse-0.4.0-py3-none-any.whl", hash = "sha256:f329af6eae57eaa2bdfd962b42524764af68075ea87370a2de920af5341e318f"},
+]
+
+[[package]]
+name = "idna"
+version = "3.7"
+description = "Internationalized Domain Names in Applications (IDNA)"
+optional = false
+python-versions = ">=3.5"
+groups = ["main"]
+files = [
+    {file = "idna-3.7-py3-none-any.whl", hash = "sha256:82fee1fc78add43492d3a1898bfa6d8a904cc97d8427f683ed8e798d07761aa0"},
+    {file = "idna-3.7.tar.gz", hash = "sha256:028ff3aadf0609c1fd278d8ea3089299412a7a8b9bd005dd08b9f8285bcb5cfc"},
+]
+
+[[package]]
+name = "langgraph-cli"
+version = "0.1.52"
+description = "CLI for interacting with LangGraph API"
+optional = false
+python-versions = "^3.9.0,<4.0"
+groups = ["main"]
+files = []
+develop = true
+
+[package.dependencies]
+click = "^8.1.7"
+
+[package.source]
+type = "directory"
+url = ".."
+
+[[package]]
+name = "langgraph-sdk"
+version = "0.1.29"
+description = "SDK for interacting with LangGraph API"
+optional = false
+python-versions = "^3.9.0,<4.0"
+groups = ["main"]
+files = []
+develop = true
+
+[package.dependencies]
+httpx = ">=0.25.2"
+httpx-sse = ">=0.4.0"
+orjson = ">=3.10.1"
+
+[package.source]
+type = "directory"
+url = "../../sdk-py"
+
+[[package]]
+name = "orjson"
+version = "3.10.5"
+description = "Fast, correct Python JSON library supporting dataclasses, datetimes, and numpy"
+optional = false
+python-versions = ">=3.8"
+groups = ["main"]
+files = [
+    {file = "orjson-3.10.5-cp310-cp310-macosx_10_15_x86_64.macosx_11_0_arm64.macosx_10_15_universal2.whl", hash = "sha256:545d493c1f560d5ccfc134803ceb8955a14c3fcb47bbb4b2fee0232646d0b932"},
+    {file = "orjson-3.10.5-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:f4324929c2dd917598212bfd554757feca3e5e0fa60da08be11b4aa8b90013c1"},
+    {file = "orjson-3.10.5-cp310-cp310-manylinux_2_17_ppc64le.manylinux2014_ppc64le.whl", hash = "sha256:8c13ca5e2ddded0ce6a927ea5a9f27cae77eee4c75547b4297252cb20c4d30e6"},
+    {file = "orjson-3.10.5-cp310-cp310-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:b6c8e30adfa52c025f042a87f450a6b9ea29649d828e0fec4858ed5e6caecf63"},
+    {file = "orjson-3.10.5-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:338fd4f071b242f26e9ca802f443edc588fa4ab60bfa81f38beaedf42eda226c"},
+    {file = "orjson-3.10.5-cp310-cp310-musllinux_1_2_aarch64.whl", hash = "sha256:6970ed7a3126cfed873c5d21ece1cd5d6f83ca6c9afb71bbae21a0b034588d96"},
+    {file = "orjson-3.10.5-cp310-cp310-musllinux_1_2_x86_64.whl", hash = "sha256:235dadefb793ad12f7fa11e98a480db1f7c6469ff9e3da5e73c7809c700d746b"},
+    {file = "orjson-3.10.5-cp310-none-win32.whl", hash = "sha256:be79e2393679eda6a590638abda16d167754393f5d0850dcbca2d0c3735cebe2"},
+    {file = "orjson-3.10.5-cp310-none-win_amd64.whl", hash = "sha256:c4a65310ccb5c9910c47b078ba78e2787cb3878cdded1702ac3d0da71ddc5228"},
+    {file = "orjson-3.10.5-cp311-cp311-macosx_10_15_x86_64.macosx_11_0_arm64.macosx_10_15_universal2.whl", hash = "sha256:cdf7365063e80899ae3a697def1277c17a7df7ccfc979990a403dfe77bb54d40"},
+    {file = "orjson-3.10.5-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:6b68742c469745d0e6ca5724506858f75e2f1e5b59a4315861f9e2b1df77775a"},
+    {file = "orjson-3.10.5-cp311-cp311-manylinux_2_17_ppc64le.manylinux2014_ppc64le.whl", hash = "sha256:7d10cc1b594951522e35a3463da19e899abe6ca95f3c84c69e9e901e0bd93d38"},
+    {file = "orjson-3.10.5-cp311-cp311-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:dcbe82b35d1ac43b0d84072408330fd3295c2896973112d495e7234f7e3da2e1"},
+    {file = "orjson-3.10.5-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:10c0eb7e0c75e1e486c7563fe231b40fdd658a035ae125c6ba651ca3b07936f5"},
+    {file = "orjson-3.10.5-cp311-cp311-musllinux_1_2_aarch64.whl", hash = "sha256:53ed1c879b10de56f35daf06dbc4a0d9a5db98f6ee853c2dbd3ee9d13e6f302f"},
+    {file = "orjson-3.10.5-cp311-cp311-musllinux_1_2_x86_64.whl", hash = "sha256:099e81a5975237fda3100f918839af95f42f981447ba8f47adb7b6a3cdb078fa"},
+    {file = "orjson-3.10.5-cp311-none-win32.whl", hash = "sha256:1146bf85ea37ac421594107195db8bc77104f74bc83e8ee21a2e58596bfb2f04"},
+    {file = "orjson-3.10.5-cp311-none-win_amd64.whl", hash = "sha256:36a10f43c5f3a55c2f680efe07aa93ef4a342d2960dd2b1b7ea2dd764fe4a37c"},
+    {file = "orjson-3.10.5-cp312-cp312-macosx_10_15_x86_64.macosx_11_0_arm64.macosx_10_15_universal2.whl", hash = "sha256:68f85ecae7af14a585a563ac741b0547a3f291de81cd1e20903e79f25170458f"},
+    {file = "orjson-3.10.5-cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:28afa96f496474ce60d3340fe8d9a263aa93ea01201cd2bad844c45cd21f5268"},
+    {file = "orjson-3.10.5-cp312-cp312-manylinux_2_17_ppc64le.manylinux2014_ppc64le.whl", hash = "sha256:9cd684927af3e11b6e754df80b9ffafd9fb6adcaa9d3e8fdd5891be5a5cad51e"},
+    {file = "orjson-3.10.5-cp312-cp312-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:3d21b9983da032505f7050795e98b5d9eee0df903258951566ecc358f6696969"},
+    {file = "orjson-3.10.5-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:1ad1de7fef79736dde8c3554e75361ec351158a906d747bd901a52a5c9c8d24b"},
+    {file = "orjson-3.10.5-cp312-cp312-musllinux_1_2_aarch64.whl", hash = "sha256:2d97531cdfe9bdd76d492e69800afd97e5930cb0da6a825646667b2c6c6c0211"},
+    {file = "orjson-3.10.5-cp312-cp312-musllinux_1_2_x86_64.whl", hash = "sha256:d69858c32f09c3e1ce44b617b3ebba1aba030e777000ebdf72b0d8e365d0b2b3"},
+    {file = "orjson-3.10.5-cp312-none-win32.whl", hash = "sha256:64c9cc089f127e5875901ac05e5c25aa13cfa5dbbbd9602bda51e5c611d6e3e2"},
+    {file = "orjson-3.10.5-cp312-none-win_amd64.whl", hash = "sha256:b2efbd67feff8c1f7728937c0d7f6ca8c25ec81373dc8db4ef394c1d93d13dc5"},
+    {file = "orjson-3.10.5-cp38-cp38-macosx_10_15_x86_64.macosx_11_0_arm64.macosx_10_15_universal2.whl", hash = "sha256:03b565c3b93f5d6e001db48b747d31ea3819b89abf041ee10ac6988886d18e01"},
+    {file = "orjson-3.10.5-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:584c902ec19ab7928fd5add1783c909094cc53f31ac7acfada817b0847975f26"},
+    {file = "orjson-3.10.5-cp38-cp38-manylinux_2_17_ppc64le.manylinux2014_ppc64le.whl", hash = "sha256:5a35455cc0b0b3a1eaf67224035f5388591ec72b9b6136d66b49a553ce9eb1e6"},
+    {file = "orjson-3.10.5-cp38-cp38-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:1670fe88b116c2745a3a30b0f099b699a02bb3482c2591514baf5433819e4f4d"},
+    {file = "orjson-3.10.5-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:185c394ef45b18b9a7d8e8f333606e2e8194a50c6e3c664215aae8cf42c5385e"},
+    {file = "orjson-3.10.5-cp38-cp38-musllinux_1_2_aarch64.whl", hash = "sha256:ca0b3a94ac8d3886c9581b9f9de3ce858263865fdaa383fbc31c310b9eac07c9"},
+    {file = "orjson-3.10.5-cp38-cp38-musllinux_1_2_x86_64.whl", hash = "sha256:dfc91d4720d48e2a709e9c368d5125b4b5899dced34b5400c3837dadc7d6271b"},
+    {file = "orjson-3.10.5-cp38-none-win32.whl", hash = "sha256:c05f16701ab2a4ca146d0bca950af254cb7c02f3c01fca8efbbad82d23b3d9d4"},
+    {file = "orjson-3.10.5-cp38-none-win_amd64.whl", hash = "sha256:8a11d459338f96a9aa7f232ba95679fc0c7cedbd1b990d736467894210205c09"},
+    {file = "orjson-3.10.5-cp39-cp39-macosx_10_15_x86_64.macosx_11_0_arm64.macosx_10_15_universal2.whl", hash = "sha256:85c89131d7b3218db1b24c4abecea92fd6c7f9fab87441cfc342d3acc725d807"},
+    {file = "orjson-3.10.5-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:fb66215277a230c456f9038d5e2d84778141643207f85336ef8d2a9da26bd7ca"},
+    {file = "orjson-3.10.5-cp39-cp39-manylinux_2_17_ppc64le.manylinux2014_ppc64le.whl", hash = "sha256:51bbcdea96cdefa4a9b4461e690c75ad4e33796530d182bdd5c38980202c134a"},
+    {file = "orjson-3.10.5-cp39-cp39-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:dbead71dbe65f959b7bd8cf91e0e11d5338033eba34c114f69078d59827ee139"},
+    {file = "orjson-3.10.5-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:5df58d206e78c40da118a8c14fc189207fffdcb1f21b3b4c9c0c18e839b5a214"},
+    {file = "orjson-3.10.5-cp39-cp39-musllinux_1_2_aarch64.whl", hash = "sha256:c4057c3b511bb8aef605616bd3f1f002a697c7e4da6adf095ca5b84c0fd43595"},
+    {file = "orjson-3.10.5-cp39-cp39-musllinux_1_2_x86_64.whl", hash = "sha256:b39e006b00c57125ab974362e740c14a0c6a66ff695bff44615dcf4a70ce2b86"},
+    {file = "orjson-3.10.5-cp39-none-win32.whl", hash = "sha256:eded5138cc565a9d618e111c6d5c2547bbdd951114eb822f7f6309e04db0fb47"},
+    {file = "orjson-3.10.5-cp39-none-win_amd64.whl", hash = "sha256:cc28e90a7cae7fcba2493953cff61da5a52950e78dc2dacfe931a317ee3d8de7"},
+    {file = "orjson-3.10.5.tar.gz", hash = "sha256:7a5baef8a4284405d96c90c7c62b755e9ef1ada84c2406c24a9ebec86b89f46d"},
+]
+
+[[package]]
+name = "sniffio"
+version = "1.3.1"
+description = "Sniff out which async library your code is running under"
+optional = false
+python-versions = ">=3.7"
+groups = ["main"]
+files = [
+    {file = "sniffio-1.3.1-py3-none-any.whl", hash = "sha256:2f6da418d1f1e0fddd844478f41680e794e6051915791a034ff65e5f100525a2"},
+    {file = "sniffio-1.3.1.tar.gz", hash = "sha256:f4324edc670a0f49750a81b895f35c3adb843cca46f0530f79fc1babb23789dc"},
+]
+
+[[package]]
+name = "typing-extensions"
+version = "4.12.2"
+description = "Backported and Experimental Type Hints for Python 3.8+"
+optional = false
+python-versions = ">=3.8"
+groups = ["main"]
+markers = "python_version < \"3.11\""
+files = [
+    {file = "typing_extensions-4.12.2-py3-none-any.whl", hash = "sha256:04e5ca0351e0f3f85c6853954072df659d0d13fac324d0072316b67d7794700d"},
+    {file = "typing_extensions-4.12.2.tar.gz", hash = "sha256:1a7ead55c7e559dd4dee8856e3a88b41225abfe1ce8df57b7c13915fe121ffb8"},
+]
+
+[metadata]
+lock-version = "2.1"
+python-versions = "^3.9.0,<4.0"
+content-hash = "ec5109729f30d2033a10a10e8f8d3ed94c7d96d5d31025b4815b0123664bb063"

libs/cli/langgraph_cli/__init__.py

@@ -1 +1 @@
-__version__ = "0.4.32"
+__version__ = "0.4.31"

libs/cli/langgraph_cli/config.py

@@ -6,7 +6,6 @@ import re
 import shlex
 import textwrap
 from collections import Counter
-from collections.abc import Iterable
 from typing import Literal, NamedTuple
 
 import click
@@ -37,10 +36,6 @@ DISALLOWED_BUILD_COMMAND_CHARS = [
 # This blocks background execution (cmd &) while allowing command
 # chaining (cmd1 && cmd2) which is common in build commands.
 _SINGLE_AMPERSAND_RE = re.compile(r"(?<!&)&(?:&&)*(?!&)")
-_GIT_HTTP_AUTHORITY_RES = (
-    re.compile(r"git\+https?://(?P<authority>[^/\s\"']+)", re.I),
-    re.compile(r"\bgit\s*=\s*[\"']https?://(?P<authority>[^/\s\"']+)", re.I),
-)
 _API_VERSION_PATTERN = re.compile(
     r"^(?P<major>\d+)"
     r"(?:\.(?P<minor>\d+))?"
@@ -83,62 +78,6 @@ def has_disallowed_build_command_content(command: str) -> bool:
     return False
 
 
-def _has_git_http_url_userinfo(dependency: str) -> bool:
-    """Check whether a Git HTTP URL contains userinfo."""
-    return any(
-        "@" in match.group("authority")
-        for pattern in _GIT_HTTP_AUTHORITY_RES
-        for match in pattern.finditer(dependency)
-    )
-
-
-def _validate_git_http_url_userinfo(
-    values: Iterable[str], *, source: pathlib.Path | None = None
-) -> None:
-    """Reject credential-bearing Git HTTP URLs without echoing their values."""
-    if not any(_has_git_http_url_userinfo(value) for value in values):
-        return
-    message = (
-        "Git dependency URLs must not contain credentials or other URL "
-        "userinfo because generated Dockerfiles and image layers can retain "
-        "them. Use a credential-free Git URL and provide short-lived "
-        "credentials through your build environment's secret-backed Git "
-        "credential helper."
-    )
-    if source is not None:
-        message += f" Found in: {source}"
-    raise click.UsageError(message)
-
-
-def _validate_git_http_url_userinfo_files(paths: Iterable[pathlib.Path]) -> None:
-    """Reject credential-bearing Git HTTP URLs in dependency files."""
-    for path in paths:
-        path = path.resolve()
-        if not path.is_file():
-            continue
-        try:
-            contents = path.read_text(encoding="utf-8", errors="replace")
-        except OSError:
-            raise click.UsageError(
-                f"Could not inspect dependency file for embedded credentials: {path}"
-            ) from None
-        _validate_git_http_url_userinfo([contents], source=path)
-
-
-def _validate_local_dependency_files(config_path: pathlib.Path, config: Config) -> None:
-    """Validate dependency files copied into a non-uv Python image."""
-    paths: list[pathlib.Path] = []
-    for dependency in config["dependencies"]:
-        if not isinstance(dependency, str) or not dependency.startswith("."):
-            continue
-        root = (config_path.parent / dependency).resolve()
-        paths.extend(
-            root / name
-            for name in ("requirements.txt", "pyproject.toml", "setup.py", "setup.cfg")
-        )
-    _validate_git_http_url_userinfo_files(paths)
-
-
 MIN_PYTHON_VERSION = "3.11"
 DEFAULT_PYTHON_VERSION = "3.11"
 
@@ -381,9 +320,7 @@ def _get_source_kind(config: Config) -> str | None:
     return kind if isinstance(kind, str) else None
 
 
-def validate_config(
-    config: Config, *, source_path: pathlib.Path | None = None
-) -> Config:
+def validate_config(config: Config) -> Config:
     """Validate a configuration dictionary."""
 
     graphs = config.get("graphs", {})
@@ -478,15 +415,6 @@ def validate_config(
                 '  "source": {"kind": "uv", "root": ".."}'
             )
 
-    _validate_git_http_url_userinfo(
-        (
-            dependency
-            for dependency in config["dependencies"]
-            if isinstance(dependency, str)
-        ),
-        source=source_path,
-    )
-
     source = config.get("source")
     source_kind = _get_source_kind(config)
     if source is not None and not isinstance(source, dict):
@@ -681,7 +609,7 @@ def validate_config_file(config_path: pathlib.Path) -> Config:
     """Load and validate a configuration file."""
     with open(config_path) as f:
         config = json.load(f)
-    validated = validate_config(config, source_path=config_path.resolve())
+    validated = validate_config(config)
     # Enforce the package.json doesn't enforce an
     # incompatible Node.js version
     if validated.get("node_version"):
@@ -1352,7 +1280,6 @@ def python_config_to_docker(
             api_version=api_version,
             build_tools_to_uninstall=build_tools_to_uninstall,
         )
-    _validate_local_dependency_files(config_path, config)
     if pip_installer == "auto":
         if _image_supports_uv(base_image):
             pip_installer = "uv"
@@ -1563,18 +1490,7 @@ def node_config_to_docker(
 ) -> tuple[str, dict[str, str]]:
     # Calculate paths for monorepo support
     install_root = (
-        pathlib.Path(build_context).resolve()
-        if build_context
-        else config_path.parent.resolve()
-    )
-    config_root = config_path.parent.resolve()
-    dependency_roots = (
-        (install_root, config_root) if install_root != config_root else (install_root,)
-    )
-    _validate_git_http_url_userinfo_files(
-        root / name
-        for root in dependency_roots
-        for name in ("package.json", "package-lock.json", "yarn.lock", "pnpm-lock.yaml")
+        pathlib.Path(build_context).resolve() if build_context else config_path.parent
     )
     install_cmd = install_command or _get_node_pm_install_cmd(install_root)
     if build_context:

libs/cli/langgraph_cli/exec.py

@@ -1,18 +1,12 @@
 import asyncio
 import signal
 import sys
-from collections.abc import Callable, Coroutine
+from collections.abc import Callable
 from contextlib import contextmanager
-from typing import Any, Protocol, TypeVar, cast
+from typing import cast
 
 import click.exceptions
 
-_T = TypeVar("_T")
-
-
-class CommandRunner(Protocol):
-    def run(self, coro: Coroutine[Any, Any, _T]) -> _T: ...
-
 
 @contextmanager
 def Runner():

libs/cli/langgraph_cli/host_backend.py

@@ -2,125 +2,18 @@
 
 from __future__ import annotations
 
-from dataclasses import dataclass
-from typing import Any, Literal
-from urllib.parse import urlparse
+from typing import Any
 
 import click
 import httpx
 
-CLOUD_CONTROL_PLANE_URL = "https://api.host.langchain.com"
-CLOUD_DASHBOARD_URL = "https://smith.langchain.com"
-CLOUD_DOMAIN = "langchain.com"
-CLOUD_API_HOST = "api.smith.langchain.com"
-CLOUD_CONTROL_PLANE_HOST = "api.host.langchain.com"
-CLOUD_DASHBOARD_HOST = "smith.langchain.com"
-CONTROL_PLANE_PATH = "/api-host"
-LANGSMITH_API_PATHS = ("/api/v1", "/api")
-LOCAL_HOSTNAMES = ("localhost", "127.0.0.1")
-MAX_PAGE_SIZE = 100
-SourceName = Literal["internal_docker", "internal_source", "external_docker"]
-
-
-@dataclass(frozen=True, slots=True)
-class ControlPlaneEndpoints:
-    control_plane_url: str
-    dashboard_url: str
-
-    @classmethod
-    def resolve(
-        cls, host_url: str | None, langsmith_endpoint: str | None
-    ) -> ControlPlaneEndpoints:
-        if host_url:
-            return cls.from_control_plane_url(host_url)
-        if langsmith_endpoint:
-            return cls.from_langsmith_endpoint(langsmith_endpoint)
-        return cls(CLOUD_CONTROL_PLANE_URL, CLOUD_DASHBOARD_URL)
-
-    @property
-    def is_cloud(self) -> bool:
-        hostname = urlparse(self.control_plane_url).hostname or ""
-        return hostname == CLOUD_CONTROL_PLANE_HOST or hostname.endswith(
-            f".{CLOUD_CONTROL_PLANE_HOST}"
-        )
-
-    @classmethod
-    def from_control_plane_url(cls, url: str) -> ControlPlaneEndpoints:
-        control_plane_url = url.rstrip("/")
-        hostname = urlparse(control_plane_url).hostname or ""
-        if control_plane_url.endswith(CONTROL_PLANE_PATH):
-            return cls(control_plane_url, control_plane_url[: -len(CONTROL_PLANE_PATH)])
-        if hostname in LOCAL_HOSTNAMES:
-            return cls(control_plane_url, control_plane_url)
-        return cls(control_plane_url, _cloud_dashboard_for(hostname))
-
-    @classmethod
-    def from_langsmith_endpoint(cls, endpoint: str) -> ControlPlaneEndpoints:
-        parsed = urlparse(endpoint.rstrip("/"))
-        hostname = parsed.hostname or ""
-        if _is_cloud_host(hostname):
-            return cls.from_control_plane_url(
-                f"https://{_cloud_control_plane_host_for(hostname)}"
-            )
-        root = f"{parsed.scheme}://{parsed.netloc}{_without_api_path(parsed.path)}"
-        return cls(f"{root}{CONTROL_PLANE_PATH}", root)
-
-
-def _is_cloud_host(hostname: str) -> bool:
-    return hostname == CLOUD_DOMAIN or hostname.endswith(f".{CLOUD_DOMAIN}")
-
-
-def _cloud_control_plane_host_for(langsmith_api_host: str) -> str:
-    if langsmith_api_host.endswith(f".{CLOUD_API_HOST}"):
-        region = langsmith_api_host[: -len(CLOUD_API_HOST)]
-        return f"{region}{CLOUD_CONTROL_PLANE_HOST}"
-    return CLOUD_CONTROL_PLANE_HOST
-
-
-def _cloud_dashboard_for(control_plane_host: str) -> str:
-    if control_plane_host.endswith(f".{CLOUD_CONTROL_PLANE_HOST}"):
-        region = control_plane_host[: -len(CLOUD_CONTROL_PLANE_HOST) - 1]
-        return f"https://{region}.{CLOUD_DASHBOARD_HOST}"
-    return CLOUD_DASHBOARD_URL
-
-
-def _without_api_path(path: str) -> str:
-    for api_path in LANGSMITH_API_PATHS:
-        if path.endswith(api_path):
-            return path[: -len(api_path)]
-    return path
-
-
-def _resources(payload: object) -> list[dict[str, Any]]:
-    if not isinstance(payload, dict):
-        return []
-    resources = payload.get("resources")
-    if not isinstance(resources, list):
-        return []
-    return [item for item in resources if isinstance(item, dict)]
-
 
 class HostBackendError(click.ClickException):
     """Raised when the host backend returns an error response."""
 
-    def __init__(
-        self,
-        message: str,
-        status_code: int | None = None,
-        detail: str | None = None,
-    ):
+    def __init__(self, message: str, status_code: int | None = None):
         super().__init__(message)
         self.status_code = status_code
-        self.detail = detail
-
-
-def _error_detail(response: httpx.Response) -> str | None:
-    try:
-        body = response.json()
-    except ValueError:
-        return None
-    detail = body.get("detail") if isinstance(body, dict) else None
-    return detail if isinstance(detail, str) else None
 
 
 class HostBackendClient:
@@ -131,37 +24,24 @@ class HostBackendClient:
         base_url: str,
         api_key: str,
         tenant_id: str | None = None,
-        *,
-        transport: httpx.BaseTransport | None = None,
     ):
         if not base_url:
             raise click.UsageError("Host backend URL is required")
+        transport = httpx.HTTPTransport(retries=3)
         headers: dict[str, str] = {
             "X-Api-Key": api_key,
             "Accept": "application/json",
         }
         if tenant_id:
             headers["X-Tenant-ID"] = tenant_id
-        self._endpoints = ControlPlaneEndpoints.from_control_plane_url(base_url)
-        self._base_url = self._endpoints.control_plane_url
+        self._base_url = base_url.rstrip("/")
         self._client = httpx.Client(
             base_url=self._base_url,
             headers=headers,
-            transport=transport or httpx.HTTPTransport(retries=3),
+            transport=transport,
             timeout=30,
         )
 
-    @property
-    def base_url(self) -> str:
-        return self._base_url
-
-    @property
-    def endpoints(self) -> ControlPlaneEndpoints:
-        return self._endpoints
-
-    def set_tenant(self, tenant_id: str) -> None:
-        self._client.headers["X-Tenant-ID"] = tenant_id
-
     def _request(
         self,
         method: str,
@@ -173,12 +53,10 @@ class HostBackendClient:
             resp = self._client.request(method, path, json=payload, params=params)
             resp.raise_for_status()
         except httpx.HTTPStatusError as err:
-            detail = _error_detail(err.response)
-            reason = detail or err.response.text or str(err.response.status_code)
+            detail = err.response.text or str(err.response.status_code)
             raise HostBackendError(
-                f"{method} {path} failed with status {err.response.status_code}: {reason}",
+                f"{method} {path} failed with status {err.response.status_code}: {detail}",
                 status_code=err.response.status_code,
-                detail=detail,
             ) from None
         except httpx.TransportError as err:
             raise HostBackendError(str(err)) from None
@@ -194,52 +72,30 @@ class HostBackendClient:
 
     def create_deployment(
         self,
-        *,
-        name: str | None,
-        source: SourceName,
-        source_config: dict[str, object],
-        source_revision_config: dict[str, object],
+        name: str,
+        deployment_type: str,
+        source: str,
+        config_path: str | None = None,
         secrets: list[dict[str, str]] | None = None,
-        agent: dict[str, str] | None = None,
     ) -> dict[str, Any]:
+        """Create a deployment."""
         payload: dict[str, Any] = {
+            "name": name,
             "source": source,
-            "source_config": source_config,
-            "source_revision_config": source_revision_config,
+            "source_config": {"deployment_type": deployment_type},
+            "source_revision_config": {},
         }
-        if agent is not None:
-            payload["agent"] = agent
-        else:
-            payload["name"] = name
+        if source == "internal_source" and config_path:
+            payload["source_revision_config"]["langgraph_config_path"] = config_path
         if secrets is not None:
             payload["secrets"] = secrets
         return self._request("POST", "/v2/deployments", payload)
 
-    def list_deployments(
-        self,
-        *,
-        name: str | None = None,
-        name_contains: str | None = None,
-        limit: int | None = None,
-        agent_id: str | None = None,
-        agent_environment: str | None = None,
-    ) -> list[dict[str, Any]]:
-        given = (
-            ("name", name),
-            ("name_contains", name_contains),
-            ("limit", limit),
-            ("agent_id", agent_id),
-            ("agent_environment", agent_environment),
-        )
-        params = {key: value for key, value in given if value is not None}
-        return _resources(self._request("GET", "/v2/deployments", params=params))
-
-    def get_listener(self, listener_id: str) -> dict[str, Any]:
-        return self._request("GET", f"/v2/listeners/{listener_id}")
-
-    def list_listeners(self) -> list[dict[str, Any]]:
-        return _resources(
-            self._request("GET", "/v2/listeners", params={"limit": MAX_PAGE_SIZE})
+    def list_deployments(self, name_contains: str = "") -> dict[str, Any]:
+        return self._request(
+            "GET",
+            "/v2/deployments",
+            params={"name_contains": name_contains},
         )
 
     def get_deployment(self, deployment_id: str) -> dict[str, Any]:
@@ -265,21 +121,22 @@ class HostBackendClient:
         self,
         deployment_id: str,
         image_uri: str,
-        *,
-        revision_source: SourceName | None,
         secrets: list[dict[str, str]] | None = None,
         tracked_packages: list[str] | None = None,
     ) -> dict[str, Any]:
         payload: dict[str, Any] = {
+            "revision_source": "internal_docker",
             "source_revision_config": {"image_uri": image_uri},
         }
-        if revision_source is not None:
-            payload["revision_source"] = revision_source
         if tracked_packages:
             payload["tracked_packages"] = tracked_packages
         if secrets is not None:
             payload["secrets"] = secrets
-        return self._request("PATCH", f"/v2/deployments/{deployment_id}", payload)
+        return self._request(
+            "PATCH",
+            f"/v2/deployments/{deployment_id}",
+            payload,
+        )
 
     def update_deployment_internal_source(
         self,
@@ -314,15 +171,10 @@ class HostBackendClient:
             payload["secrets"] = secrets
         return self._request("PATCH", f"/v2/deployments/{deployment_id}", payload)
 
-    def list_revisions(
-        self, deployment_id: str, limit: int = 1
-    ) -> list[dict[str, Any]]:
-        return _resources(
-            self._request(
-                "GET",
-                f"/v2/deployments/{deployment_id}/revisions",
-                params={"limit": limit},
-            )
+    def list_revisions(self, deployment_id: str, limit: int = 1) -> dict[str, Any]:
+        return self._request(
+            "GET",
+            f"/v2/deployments/{deployment_id}/revisions?limit={limit}",
         )
 
     def get_revision(self, deployment_id: str, revision_id: str) -> dict[str, Any]:

libs/cli/langgraph_cli/image_reference.py

@@ -1,35 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import dataclass, replace
-
-DIGEST_SEPARATOR = "@sha256:"
-DIGEST_MARKER = "@"
-TAG_SEPARATOR = ":"
-PATH_SEPARATOR = "/"
-
-
-@dataclass(frozen=True, slots=True)
-class ImageReference:
-    repository: str
-    tag: str | None = None
-
-    @classmethod
-    def parse(cls, reference: str) -> ImageReference:
-        if DIGEST_MARKER in reference:
-            raise ValueError(f"{reference!r} carries a digest and cannot be tagged")
-        path_start = reference.rfind(PATH_SEPARATOR) + 1
-        name, separator, tag = reference[path_start:].partition(TAG_SEPARATOR)
-        if not separator:
-            return cls(reference)
-        return cls(reference[:path_start] + name, tag)
-
-    def with_tag(self, tag: str) -> ImageReference:
-        return replace(self, tag=tag)
-
-    def matches_digest(self, repo_digest: str) -> bool:
-        return repo_digest.startswith(f"{self.repository}{DIGEST_SEPARATOR}")
-
-    def __str__(self) -> str:
-        if self.tag is None:
-            return self.repository
-        return f"{self.repository}{TAG_SEPARATOR}{self.tag}"

libs/cli/langgraph_cli/schemas.py

@@ -650,8 +650,7 @@ class Config(TypedDict, total=False):
 
     pip_config_file: str | None
     """Optional. Path to a pip config file (e.g., "/etc/pip.conf" or "pip.ini") for controlling
-    package installation (custom indices, timeouts, etc.). The file is copied into the
-    generated image, so it must not contain credentials or other secrets.
+    package installation (custom indices, credentials, etc.).
 
     Only relevant if Python dependencies are installed via pip. If omitted, default pip settings are used.
     """
@@ -690,9 +689,6 @@ class Config(TypedDict, total=False):
       - "." or "./src" if you have a local Python package
       - str (aka "anthropic") for a PyPI package
       - "git+https://github.com/org/repo.git@main" for a Git-based package
-    Git HTTP URLs must not contain userinfo such as a username or token. For private
-    dependencies, provide short-lived credentials through the build environment's
-    secret-backed Git credential helper.
     Defaults to an empty list, meaning no additional packages installed beyond your base environment.
 
     This field is not supported when `source.kind` is `uv`.

libs/cli/langgraph_cli/uv_lock.py

@@ -880,7 +880,6 @@ def python_config_to_docker_uv_lock(
         _get_node_pm_install_cmd,
         _get_pip_cleanup_lines,
         _image_supports_uv,
-        _validate_git_http_url_userinfo_files,
         docker_tag,
     )
 
@@ -891,20 +890,11 @@ def python_config_to_docker_uv_lock(
         )
 
     config_root = config_path.parent.resolve()
-    source_root = config["source"].get("root", ".")
-    project_root = (config_root / source_root).resolve()
-    _validate_git_http_url_userinfo_files(
-        [project_root / "pyproject.toml", project_root / "uv.lock"]
-    )
-
     install_cmd = "uv pip install --system"
     _, global_reqs_pip_install, pip_config_file_str = _build_python_install_commands(
         config, install_cmd
     )
     plan = _plan_uv_lock_workspace(config_path, config)
-    _validate_git_http_url_userinfo_files(
-        package.pyproject_path for package in plan.install_order
-    )
 
     _update_uv_lock_graph_paths(config_path, config, plan)
     for section, key in [

libs/cli/schemas/schema.json

@@ -28,7 +28,7 @@
                   "type": "null"
                 }
               ],
-              "description": "Optional. Path to a pip config file (e.g., \"/etc/pip.conf\" or \"pip.ini\") for controlling\npackage installation (custom indices, timeouts, etc.). The file is copied into the\ngenerated image, so it must not contain credentials or other secrets.\n\nOnly relevant if Python dependencies are installed via pip. If omitted, default pip settings are used.\n"
+              "description": "Optional. Path to a pip config file (e.g., \"/etc/pip.conf\" or \"pip.ini\") for controlling\npackage installation (custom indices, credentials, etc.).\n\nOnly relevant if Python dependencies are installed via pip. If omitted, default pip settings are used.\n"
             },
             "_INTERNAL_docker_tag": {
               "anyOf": [
@@ -270,7 +270,7 @@
                   "type": "null"
                 }
               ],
-              "description": "Optional. Path to a pip config file (e.g., \"/etc/pip.conf\" or \"pip.ini\") for controlling\npackage installation (custom indices, timeouts, etc.). The file is copied into the\ngenerated image, so it must not contain credentials or other secrets.\n\nOnly relevant if Python dependencies are installed via pip. If omitted, default pip settings are used.\n"
+              "description": "Optional. Path to a pip config file (e.g., \"/etc/pip.conf\" or \"pip.ini\") for controlling\npackage installation (custom indices, credentials, etc.).\n\nOnly relevant if Python dependencies are installed via pip. If omitted, default pip settings are used.\n"
             },
             "_INTERNAL_docker_tag": {
               "anyOf": [

libs/cli/schemas/schema.v0.json

@@ -28,7 +28,7 @@
                   "type": "null"
                 }
               ],
-              "description": "Optional. Path to a pip config file (e.g., \"/etc/pip.conf\" or \"pip.ini\") for controlling\npackage installation (custom indices, timeouts, etc.). The file is copied into the\ngenerated image, so it must not contain credentials or other secrets.\n\nOnly relevant if Python dependencies are installed via pip. If omitted, default pip settings are used.\n"
+              "description": "Optional. Path to a pip config file (e.g., \"/etc/pip.conf\" or \"pip.ini\") for controlling\npackage installation (custom indices, credentials, etc.).\n\nOnly relevant if Python dependencies are installed via pip. If omitted, default pip settings are used.\n"
             },
             "_INTERNAL_docker_tag": {
               "anyOf": [
@@ -270,7 +270,7 @@
                   "type": "null"
                 }
               ],
-              "description": "Optional. Path to a pip config file (e.g., \"/etc/pip.conf\" or \"pip.ini\") for controlling\npackage installation (custom indices, timeouts, etc.). The file is copied into the\ngenerated image, so it must not contain credentials or other secrets.\n\nOnly relevant if Python dependencies are installed via pip. If omitted, default pip settings are used.\n"
+              "description": "Optional. Path to a pip config file (e.g., \"/etc/pip.conf\" or \"pip.ini\") for controlling\npackage installation (custom indices, credentials, etc.).\n\nOnly relevant if Python dependencies are installed via pip. If omitted, default pip settings are used.\n"
             },
             "_INTERNAL_docker_tag": {
               "anyOf": [

libs/cli/tests/unit_tests/cli/test_cli.py

@@ -382,18 +382,20 @@ def test_deploy_list_command(monkeypatch) -> None:
 
         def list_deployments(self, name_contains: str = ""):
             captured["name_contains"] = name_contains
-            return [
-                {
-                    "id": "dep-123",
-                    "name": "alpha",
-                    "source_config": {"custom_url": "https://alpha.example.com"},
-                },
-                {
-                    "id": "dep-456",
-                    "name": "beta",
-                    "source_config": {"custom_url": "https://beta.example.com"},
-                },
-            ]
+            return {
+                "resources": [
+                    {
+                        "id": "dep-123",
+                        "name": "alpha",
+                        "source_config": {"custom_url": "https://alpha.example.com"},
+                    },
+                    {
+                        "id": "dep-456",
+                        "name": "beta",
+                        "source_config": {"custom_url": "https://beta.example.com"},
+                    },
+                ]
+            }
 
     monkeypatch.setattr(deploy_module, "HostBackendClient", FakeClient)
 
@@ -433,7 +435,7 @@ def test_deploy_list_command_no_results(monkeypatch) -> None:
             pass
 
         def list_deployments(self, name_contains: str = ""):
-            return []
+            return {"resources": []}
 
     monkeypatch.setattr(deploy_module, "HostBackendClient", FakeClient)
 
@@ -466,18 +468,20 @@ def test_deploy_revisions_list_command(monkeypatch) -> None:
         def list_revisions(self, deployment_id: str, limit: int = 1):
             captured["deployment_id"] = deployment_id
             captured["limit"] = str(limit)
-            return [
-                {
-                    "id": "rev-123",
-                    "status": "CREATING",
-                    "created_at": "2023-11-07T05:31:56Z",
-                },
-                {
-                    "id": "rev-456",
-                    "status": "DEPLOYED",
-                    "created_at": "2023-11-08T10:00:00Z",
-                },
-            ]
+            return {
+                "resources": [
+                    {
+                        "id": "rev-123",
+                        "status": "CREATING",
+                        "created_at": "2023-11-07T05:31:56Z",
+                    },
+                    {
+                        "id": "rev-456",
+                        "status": "DEPLOYED",
+                        "created_at": "2023-11-08T10:00:00Z",
+                    },
+                ]
+            }
 
     monkeypatch.setattr(deploy_module, "HostBackendClient", FakeClient)
 
@@ -518,7 +522,7 @@ def test_deploy_revisions_list_command_no_results(monkeypatch) -> None:
             pass
 
         def list_revisions(self, deployment_id: str, limit: int = 1):
-            return []
+            return {"resources": []}
 
     monkeypatch.setattr(deploy_module, "HostBackendClient", FakeClient)
 
@@ -551,7 +555,7 @@ def test_deploy_revisions_list_command_with_explicit_limit(monkeypatch) -> None:
         def list_revisions(self, deployment_id: str, limit: int = 1):
             captured["deployment_id"] = deployment_id
             captured["limit"] = str(limit)
-            return []
+            return {"resources": []}
 
     monkeypatch.setattr(deploy_module, "HostBackendClient", FakeClient)
 

libs/cli/tests/unit_tests/conftest.py

@@ -12,5 +12,5 @@ def disable_analytics_env() -> None:
     if "LANGGRAPH_CLI_NO_ANALYTICS" in os.environ:
         print("⚠️ LANGGRAPH_CLI_NO_ANALYTICS is set. Overriding it for the test.")
 
-    with patch.dict(os.environ, {"LANGGRAPH_CLI_NO_ANALYTICS": "1"}):
+    with patch.dict(os.environ, {"LANGGRAPH_CLI_NO_ANALYTICS": "0"}):
         yield

libs/cli/tests/unit_tests/test_config.py

@@ -255,243 +255,6 @@ def test_validate_config():
         )
 
 
-@pytest.mark.parametrize(
-    "dependency",
-    [
-        "git+https://user:secret-token@github.com/org/private.git@main",
-        "private-package @ git+http://token@github.com/org/private.git",
-        "git+HTTPS://user%40example.com:secret%2Ftoken@github.com/org/private.git",
-        "git+https://${GIT_TOKEN}@github.com/org/private.git",
-    ],
-)
-def test_validate_config_rejects_git_http_url_userinfo(dependency: str):
-    with pytest.raises(click.UsageError) as exc_info:
-        validate_config(
-            {
-                "python_version": "3.11",
-                "dependencies": [dependency],
-                "graphs": {"agent": "./agent.py:graph"},
-            }
-        )
-
-    message = str(exc_info.value)
-    assert "must not contain credentials or other URL userinfo" in message
-    assert "secret-token" not in message
-    assert "secret%2Ftoken" not in message
-
-
-def test_validate_config_file_reports_source_for_git_http_url_userinfo(
-    tmp_path: pathlib.Path,
-):
-    config_path = tmp_path / "langgraph.json"
-    config_path.write_text(
-        json.dumps(
-            {
-                "python_version": "3.11",
-                "dependencies": ["git+https://secret-token@github.com/org/private.git"],
-                "graphs": {"agent": "./agent.py:graph"},
-            }
-        )
-    )
-
-    with pytest.raises(click.UsageError) as exc_info:
-        validate_config_file(config_path)
-
-    message = str(exc_info.value)
-    assert "secret-token" not in message
-    assert f"Found in: {config_path.resolve()}" in message
-
-
-@pytest.mark.parametrize(
-    "manifest", ["package.json", "package-lock.json", "yarn.lock", "pnpm-lock.yaml"]
-)
-def test_config_to_docker_rejects_git_http_url_userinfo_in_node_files(
-    tmp_path: pathlib.Path, manifest: str
-):
-    config_path = tmp_path / "langgraph.json"
-    config_path.write_text("{}\n")
-    (tmp_path / "agent.js").write_text("export const graph = {};\n")
-    (tmp_path / "package.json").write_text('{"name":"agent"}\n')
-    (tmp_path / manifest).write_text(
-        '"priv": "git+https://user:secret-token@github.com/org/private.git"\n'
-    )
-    config = validate_config(
-        {
-            "node_version": "20",
-            "graphs": {"agent": "./agent.js:graph"},
-        }
-    )
-
-    with pytest.raises(click.UsageError) as exc_info:
-        config_to_docker(
-            config_path,
-            config,
-            base_image="langchain/langgraphjs-api",
-        )
-
-    message = str(exc_info.value)
-    assert "must not contain credentials or other URL userinfo" in message
-    assert "secret-token" not in message
-    assert f"Found in: {(tmp_path / manifest).resolve()}" in message
-
-
-def test_config_to_docker_allows_node_git_urls_without_http_userinfo(
-    tmp_path: pathlib.Path,
-):
-    config_path = tmp_path / "langgraph.json"
-    config_path.write_text("{}\n")
-    (tmp_path / "agent.js").write_text("export const graph = {};\n")
-    (tmp_path / "package.json").write_text(
-        '{"dependencies":{"public":"git+https://github.com/org/public.git"}}\n'
-    )
-    config = validate_config(
-        {
-            "node_version": "20",
-            "graphs": {"agent": "./agent.js:graph"},
-        }
-    )
-
-    docker, _ = config_to_docker(
-        config_path,
-        config,
-        base_image="langchain/langgraphjs-api",
-    )
-
-    assert f"ADD . /deps/{tmp_path.name}" in docker
-
-
-def test_config_to_docker_rejects_git_http_url_userinfo_in_node_workspace(
-    tmp_path: pathlib.Path,
-):
-    config_root = tmp_path / "apps" / "agent"
-    config_root.mkdir(parents=True)
-    config_path = config_root / "langgraph.json"
-    config_path.write_text("{}\n")
-    (config_root / "agent.js").write_text("export const graph = {};\n")
-    (config_root / "package.json").write_text(
-        '{"dependencies":{"priv":"git+https://secret-token@github.com/org/private.git"}}\n'
-    )
-    (tmp_path / "package.json").write_text('{"name":"workspace"}\n')
-    config = validate_config(
-        {
-            "node_version": "20",
-            "graphs": {"agent": "./agent.js:graph"},
-        }
-    )
-
-    with pytest.raises(click.UsageError) as exc_info:
-        config_to_docker(
-            config_path,
-            config,
-            base_image="langchain/langgraphjs-api",
-            build_context=str(tmp_path),
-        )
-
-    message = str(exc_info.value)
-    assert "secret-token" not in message
-    assert f"Found in: {(config_root / 'package.json').resolve()}" in message
-
-
-@pytest.mark.parametrize(
-    "dependency",
-    [
-        "git+https://github.com/org/public.git@main",
-        "private-package @ git+https://github.com/org/private.git@main",
-        "git+ssh://git@github.com/org/private.git@main",
-    ],
-)
-def test_validate_config_allows_git_urls_without_http_userinfo(dependency: str):
-    config = validate_config(
-        {
-            "python_version": "3.11",
-            "dependencies": [dependency],
-            "graphs": {"agent": "./agent.py:graph"},
-        }
-    )
-
-    assert config["dependencies"] == [dependency]
-
-
-def test_config_to_docker_rejects_git_http_url_userinfo_in_requirements(
-    tmp_path: pathlib.Path,
-):
-    config_path = tmp_path / "langgraph.json"
-    config_path.write_text("{}\n")
-    (tmp_path / "agent.py").write_text("graph = object()\n")
-    (tmp_path / "requirements.txt").write_text(
-        "private @ git+https://secret-token@github.com/org/private.git\n"
-    )
-    config = validate_config(
-        {
-            "python_version": "3.11",
-            "dependencies": ["."],
-            "graphs": {"agent": "./agent.py:graph"},
-        }
-    )
-
-    with pytest.raises(click.UsageError) as exc_info:
-        config_to_docker(
-            config_path,
-            config,
-            base_image="langchain/langgraph-api:0.2.47",
-        )
-
-    message = str(exc_info.value)
-    assert "must not contain credentials or other URL userinfo" in message
-    assert "secret-token" not in message
-    assert f"Found in: {(tmp_path / 'requirements.txt').resolve()}" in message
-
-
-@pytest.mark.parametrize("manifest", ["pyproject.toml", "uv.lock"])
-def test_config_to_docker_rejects_git_http_url_userinfo_in_uv_files(
-    tmp_path: pathlib.Path, manifest: str
-):
-    config_path = tmp_path / "langgraph.json"
-    config_path.write_text("{}\n")
-    (tmp_path / "src").mkdir()
-    (tmp_path / "src" / "agent.py").write_text("graph = object()\n")
-    pyproject = textwrap.dedent(
-        """
-        [project]
-        name = "agent"
-        version = "0.1.0"
-        dependencies = ["private"]
-
-        [tool.uv.sources]
-        private = { git = "https://github.com/org/private.git" }
-        """
-    ).strip()
-    uv_lock = "# uv lock file\n"
-    if manifest == "pyproject.toml":
-        pyproject = pyproject.replace(
-            "https://github.com", "https://secret-token@github.com"
-        )
-    else:
-        uv_lock += (
-            'source = { git = "https://secret-token@github.com/org/private.git" }\n'
-        )
-    (tmp_path / "pyproject.toml").write_text(pyproject + "\n")
-    (tmp_path / "uv.lock").write_text(uv_lock)
-    config = validate_config(
-        {
-            "python_version": "3.11",
-            "graphs": {"agent": "./src/agent.py:graph"},
-            "source": {"kind": "uv"},
-        }
-    )
-
-    with pytest.raises(click.UsageError) as exc_info:
-        config_to_docker(
-            config_path,
-            config,
-            base_image="langchain/langgraph-api:0.2.47",
-        )
-
-    message = str(exc_info.value)
-    assert "must not contain credentials or other URL userinfo" in message
-    assert "secret-token" not in message
-
-
 def test_validate_config_image_distro():
     """Test validation of image_distro field."""
     # Valid image_distro values should work

libs/cli/tests/unit_tests/test_deploy_agents.py

@@ -1,119 +0,0 @@
-import json
-from unittest.mock import Mock
-
-import httpx
-import pytest
-from click.testing import CliRunner
-
-import langgraph_cli.deploy as deploy
-from langgraph_cli.cli import cli
-from langgraph_cli.host_backend import HostBackendClient
-
-
-@pytest.fixture
-def deployment_api(monkeypatch, tmp_path):
-    monkeypatch.chdir(tmp_path)
-    monkeypatch.delenv("LANGSMITH_DEPLOYMENT_NAME", raising=False)
-    monkeypatch.setattr(deploy, "_emitter", None)
-    monkeypatch.setattr(deploy, "_no_input", False)
-    (tmp_path / "langgraph.json").write_text(
-        json.dumps({"dependencies": ["."], "graphs": {"agent": "./agent.py:graph"}})
-    )
-    (tmp_path / ".env").write_text("LANGSMITH_DEPLOYMENT_NAME=legacy\n")
-    requests = []
-    state = {"enabled": True, "resources": []}
-
-    def handler(request):
-        requests.append(request)
-        assert request.url.path == "/v2/deployments"
-        if request.method == "GET":
-            if not state["enabled"] and (
-                "agent_id" in request.url.params
-                or "agent_environment" in request.url.params
-            ):
-                return httpx.Response(
-                    400, text="Agent filters are not available for this tenant."
-                )
-            return httpx.Response(200, json={"resources": state["resources"]})
-        assert request.method == "POST"
-        return httpx.Response(200, json={"id": "runtime-id", "name": "server-name"})
-
-    client = HostBackendClient("https://api.example.com", "test-key")
-    client._client.close()
-    client._client = httpx.Client(
-        base_url="https://api.example.com",
-        transport=httpx.MockTransport(handler),
-        headers={"X-Api-Key": "test-key"},
-    )
-    monkeypatch.setattr(deploy, "_create_host_backend_client", lambda *a, **kw: client)
-    monkeypatch.setattr(deploy, "find_tracked_packages", lambda *a: [])
-    remote_build = Mock(return_value=deploy.BuildResult())
-    monkeypatch.setattr(deploy, "_run_remote_build", remote_build)
-    monkeypatch.setattr(deploy, "_resolve_build_mode", lambda flag, **kw: (flag, None))
-    yield state, requests, remote_build
-    client._client.close()
-
-
-AGENT_ARGS = [
-    "deploy",
-    "--agent-id",
-    "customer-support",
-    "--agent-environment",
-    "staging",
-    "--remote",
-    "--no-wait",
-    "--no-input",
-]
-
-
-def test_agent_create(deployment_api, tmp_path, monkeypatch):
-    monkeypatch.setenv("LANGSMITH_DEPLOYMENT_NAME", "legacy")
-    _, requests, build = deployment_api
-    result = CliRunner().invoke(cli, AGENT_ARGS)
-    assert result.exit_code == 0, result.output
-    assert dict(requests[0].url.params) == {
-        "agent_id": "customer-support",
-        "agent_environment": "staging",
-        "limit": "100",
-    }
-    payload = json.loads(requests[1].content)
-    assert payload["agent"] == {
-        "agent_id": "customer-support",
-        "environment": "staging",
-    }
-    assert "name" not in payload
-    assert build.call_args.kwargs["deployment_id"] == "runtime-id"
-    assert "server-name" in result.output
-    assert (tmp_path / ".env").read_text() == "LANGSMITH_DEPLOYMENT_NAME=legacy\n"
-
-
-def test_agent_update(deployment_api):
-    state, requests, build = deployment_api
-    state["resources"] = [{"id": "existing-id", "is_preview": False}]
-    result = CliRunner().invoke(cli, AGENT_ARGS)
-    assert result.exit_code == 0, result.output
-    assert len(requests) == 1
-    assert build.call_args.kwargs["deployment_id"] == "existing-id"
-
-
-def test_agent_rejects_explicit_name(deployment_api, monkeypatch):
-    monkeypatch.setenv("LANGSMITH_DEPLOYMENT_NAME", "legacy")
-    _, requests, _ = deployment_api
-    result = CliRunner().invoke(cli, [*AGENT_ARGS, "--name", "legacy"])
-    assert result.exit_code == 2
-    assert "cannot be combined" in result.output
-    assert not requests
-
-
-def test_agent_lookup_refuses_a_control_plane_that_ignores_the_filter(deployment_api):
-    state, requests, _ = deployment_api
-    state["resources"] = [
-        {"id": "someone-elses", "is_preview": False},
-        {"id": "another", "is_preview": False},
-    ]
-
-    result = CliRunner().invoke(cli, AGENT_ARGS)
-
-    assert result.exit_code != 0
-    assert "does not filter deployments by agent" in result.output
-    assert len(requests) == 1

libs/cli/tests/unit_tests/test_deploy_helpers.py

@@ -13,17 +13,6 @@ import pytest
 
 import langgraph_cli.deploy as deploy_mod
 from langgraph_cli.deploy import (
-    ById,
-    ByName,
-    CustomerRegistrySource,
-    DockerBuildCommand,
-    ExistingDeployment,
-    Listener,
-    ManagedRegistrySource,
-    OnListener,
-    RemoteBuildSource,
-    RequestedPlacement,
-    Unplaced,
     _call_host_backend_with_optional_tenant,
     _create_host_backend_client,
     _docker_config_for_token,
@@ -32,14 +21,12 @@ from langgraph_cli.deploy import (
     _parse_env_from_config,
     _resolve_env_path,
     _resolve_pushed_image_digest,
-    _select_source,
+    _smith_dashboard_base_url,
     _validate_prebuilt_image,
-    find_deployment_by_name,
     normalize_image_tag,
     normalize_name,
 )
 from langgraph_cli.host_backend import HostBackendClient, HostBackendError
-from langgraph_cli.image_reference import ImageReference
 
 
 class TestDockerConfigForToken:
@@ -272,29 +259,31 @@ class TestEnvWithoutDeploymentName:
 
 class TestCallHostBackendWithOptionalTenant:
     def _make_client(self, handler):
-        c = HostBackendClient(
-            "https://api.example.com",
-            "test-key",
+        c = HostBackendClient("https://api.example.com", "test-key")
+        c._client = httpx.Client(
+            base_url="https://api.example.com",
             transport=httpx.MockTransport(handler),
+            headers={"X-Api-Key": "test-key", "Accept": "application/json"},
+            timeout=30,
         )
         return c
 
     def _make_eu_client(self, handler):
-        c = HostBackendClient(
-            "https://eu.api.host.langchain.com",
-            "test-key",
+        c = HostBackendClient("https://eu.api.host.langchain.com", "test-key")
+        c._client = httpx.Client(
+            base_url="https://eu.api.host.langchain.com",
             transport=httpx.MockTransport(handler),
+            headers={"X-Api-Key": "test-key", "Accept": "application/json"},
+            timeout=30,
         )
         return c
 
     def test_success_passes_through(self):
-        client = self._make_client(
-            lambda req: httpx.Response(200, json={"resources": [{"id": "dep-1"}]})
-        )
+        client = self._make_client(lambda req: httpx.Response(200, json={"ok": True}))
         result = _call_host_backend_with_optional_tenant(
             client, lambda c: c.list_deployments()
         )
-        assert result == [{"id": "dep-1"}]
+        assert result == {"ok": True}
 
     def test_403_not_enabled_gives_actionable_error(self):
         detail = (
@@ -345,6 +334,7 @@ class TestCallHostBackendWithOptionalTenant:
         assert exc_info.value.status_code == 403
         assert "smith.langchain.com" in exc_info.value.message
         assert seen_tenant_ids == [None, "workspace-123"]
+        assert client._client.headers["X-Tenant-ID"] == "workspace-123"
 
     def test_other_403_re_raises_original(self):
         client = self._make_client(
@@ -550,226 +540,60 @@ class TestCreateHostBackendClientNoInput:
         assert client is not None
 
 
-class TestCreateHostBackendClientEndpoint:
-    def test_langsmith_endpoint_from_project_env_selects_self_hosted_control_plane(
-        self, monkeypatch
-    ):
-        monkeypatch.setenv("LANGSMITH_API_KEY", "lsv2_test")
-        monkeypatch.delenv("LANGSMITH_ENDPOINT", raising=False)
+class TestSmithDashboardBaseUrl:
+    def test_none_returns_default(self):
+        assert _smith_dashboard_base_url(None) == "https://smith.langchain.com"
 
-        client = _create_host_backend_client(
-            host_url=None,
-            api_key=None,
-            env_vars={"LANGSMITH_ENDPOINT": "https://smith.example.com/api/v1"},
+    def test_empty_returns_default(self):
+        assert _smith_dashboard_base_url("") == "https://smith.langchain.com"
+
+    def test_prod_host_url(self):
+        assert (
+            _smith_dashboard_base_url("https://api.host.langchain.com")
+            == "https://smith.langchain.com"
         )
 
-        assert client.base_url == "https://smith.example.com/api-host"
-
-    def test_explicit_host_url_wins_over_langsmith_endpoint(self, monkeypatch):
-        monkeypatch.setenv("LANGSMITH_API_KEY", "lsv2_test")
-        monkeypatch.setenv("LANGSMITH_ENDPOINT", "https://smith.example.com/api/v1")
-
-        client = _create_host_backend_client(
-            host_url="https://custom.host.com", api_key=None, env_vars={}
+    def test_dev_host_url(self):
+        assert (
+            _smith_dashboard_base_url("https://dev.api.host.langchain.com")
+            == "https://dev.smith.langchain.com"
         )
 
-        assert client.base_url == "https://custom.host.com"
-
-
-class TestDockerBuildCommand:
-    @pytest.mark.parametrize(
-        ("machine", "verbose", "expected"),
-        [
-            pytest.param(
-                "x86_64",
-                False,
-                DockerBuildCommand(("docker", "build"), ()),
-                id="amd64_host_builds_natively",
-            ),
-            pytest.param(
-                "arm64",
-                False,
-                DockerBuildCommand(
-                    ("docker", "buildx", "build"),
-                    ("--platform", "linux/amd64", "--load", "--progress=quiet"),
-                ),
-                id="other_hosts_cross_build_quietly",
-            ),
-            pytest.param(
-                "arm64",
-                True,
-                DockerBuildCommand(
-                    ("docker", "buildx", "build"),
-                    ("--platform", "linux/amd64", "--load"),
-                ),
-                id="verbose_cross_build_keeps_progress_output",
-            ),
-        ],
-    )
-    def test_for_host_targets_the_deployment_platform(self, machine, verbose, expected):
-        assert DockerBuildCommand.for_host(machine, verbose=verbose) == expected
-
-
-class TestSelectSource:
-    OPTIONS = {
-        "push_to": None,
-        "image": None,
-        "image_name": None,
-        "tag": None,
-        "remote_build_flag": None,
-        "placement": RequestedPlacement(),
-        "selector": ByName("my-app"),
-    }
-    REPOSITORY = "registry.example.com/app"
-
-    @pytest.mark.parametrize(
-        ("flags", "docker_available", "expected"),
-        [
-            pytest.param(
-                {"push_to": REPOSITORY},
-                True,
-                CustomerRegistrySource(
-                    reference=ImageReference(REPOSITORY, "latest"),
-                    prebuilt_image=None,
-                    requested_placement=RequestedPlacement(),
-                ),
-                id="push_to_selects_the_external_source_with_the_default_tag",
-            ),
-            pytest.param(
-                {"push_to": f"{REPOSITORY}:v2"},
-                True,
-                CustomerRegistrySource(
-                    reference=ImageReference(REPOSITORY, "v2"),
-                    prebuilt_image=None,
-                    requested_placement=RequestedPlacement(),
-                ),
-                id="push_to_keeps_a_tag_given_in_the_reference",
-            ),
-            pytest.param(
-                {"push_to": REPOSITORY, "tag": "v3"},
-                True,
-                CustomerRegistrySource(
-                    reference=ImageReference(REPOSITORY, "v3"),
-                    prebuilt_image=None,
-                    requested_placement=RequestedPlacement(),
-                ),
-                id="tag_flag_composes_with_push_to",
-            ),
-            pytest.param(
-                {"push_to": REPOSITORY, "image": "app:dev"},
-                False,
-                CustomerRegistrySource(
-                    reference=ImageReference(REPOSITORY, "latest"),
-                    prebuilt_image="app:dev",
-                    requested_placement=RequestedPlacement(),
-                ),
-                id="prebuilt_image_is_retagged_for_push_to_without_docker_checks",
-            ),
-            pytest.param(
-                {
-                    "push_to": REPOSITORY,
-                    "placement": RequestedPlacement("listener-1", "agents"),
-                },
-                True,
-                CustomerRegistrySource(
-                    reference=ImageReference(REPOSITORY, "latest"),
-                    prebuilt_image=None,
-                    requested_placement=RequestedPlacement("listener-1", "agents"),
-                ),
-                id="push_to_carries_the_requested_placement",
-            ),
-            pytest.param(
-                {"remote_build_flag": True},
-                True,
-                RemoteBuildSource(),
-                id="remote_flag_selects_the_source_upload",
-            ),
-            pytest.param(
-                {},
-                False,
-                RemoteBuildSource(),
-                id="no_local_docker_falls_back_to_the_source_upload",
-            ),
-            pytest.param(
-                {},
-                True,
-                ManagedRegistrySource(
-                    prebuilt_image=None, image_name=None, tag="latest"
-                ),
-                id="local_docker_selects_the_internal_docker_source",
-            ),
-            pytest.param(
-                {"image": "app:dev", "tag": "v1"},
-                False,
-                ManagedRegistrySource(
-                    prebuilt_image="app:dev", image_name=None, tag="v1"
-                ),
-                id="prebuilt_image_forces_the_internal_docker_source",
-            ),
-        ],
-    )
-    def test_flags_select_one_source(
-        self, monkeypatch, mocker, flags, docker_available, expected
-    ):
-        mocker.patch(
-            "langgraph_cli.deploy._get_emitter", return_value=mocker.MagicMock()
-        )
-        monkeypatch.setattr(
-            deploy_mod,
-            "can_build_locally",
-            lambda: (True, None) if docker_available else (False, "Docker is required"),
+    def test_eu_host_url(self):
+        assert (
+            _smith_dashboard_base_url("https://eu.api.host.langchain.com")
+            == "https://eu.smith.langchain.com"
         )
 
-        assert _select_source(**{**self.OPTIONS, **flags}) == expected
-
-    def test_push_to_build_requires_local_docker(self, monkeypatch):
-        monkeypatch.setattr(
-            deploy_mod, "can_build_locally", lambda: (False, "Docker is required")
+    def test_staging_host_url(self):
+        assert (
+            _smith_dashboard_base_url("https://staging.api.host.langchain.com")
+            == "https://staging.smith.langchain.com"
         )
 
-        with pytest.raises(click.UsageError, match="Docker is required"):
-            _select_source(**{**self.OPTIONS, "push_to": self.REPOSITORY})
+    def test_localhost(self):
+        assert (
+            _smith_dashboard_base_url("http://localhost:8080")
+            == "http://localhost:8080"
+        )
 
-    @pytest.mark.parametrize(
-        ("flags", "message"),
-        [
-            pytest.param(
-                {"push_to": REPOSITORY, "remote_build_flag": True},
-                "--push-to cannot be combined with --remote.",
-                id="push_to_with_remote",
-            ),
-            pytest.param(
-                {"push_to": f"{REPOSITORY}:v1", "tag": "v2"},
-                "already includes a tag",
-                id="push_to_with_a_tag_and_the_tag_flag",
-            ),
-            pytest.param(
-                {"push_to": f"{REPOSITORY}@sha256:abc"},
-                "not a digest",
-                id="push_to_with_a_digest",
-            ),
-            pytest.param(
-                {"image": "app:dev", "remote_build_flag": True},
-                "--image cannot be combined with --remote builds.",
-                id="image_with_remote",
-            ),
-            pytest.param(
-                {"placement": RequestedPlacement(listener_id="listener-1")},
-                "only apply when creating a deployment with --push-to",
-                id="listener_without_push_to",
-            ),
-            pytest.param(
-                {"placement": RequestedPlacement(k8s_namespace="agents")},
-                "only apply when creating a deployment with --push-to",
-                id="namespace_without_push_to",
-            ),
-        ],
-    )
-    def test_conflicting_flags_are_rejected(self, monkeypatch, flags, message):
-        monkeypatch.setattr(deploy_mod, "can_build_locally", lambda: (True, None))
+    def test_localhost_trailing_slash(self):
+        assert (
+            _smith_dashboard_base_url("http://localhost:8080/")
+            == "http://localhost:8080"
+        )
 
-        with pytest.raises(click.UsageError, match=message):
-            _select_source(**{**self.OPTIONS, **flags})
+    def test_127_0_0_1(self):
+        assert (
+            _smith_dashboard_base_url("http://127.0.0.1:3000")
+            == "http://127.0.0.1:3000"
+        )
+
+    def test_unknown_domain_returns_default(self):
+        assert (
+            _smith_dashboard_base_url("https://custom.example.com")
+            == "https://smith.langchain.com"
+        )
 
 
 class TestResolvePushedImageDigest:
@@ -820,16 +644,6 @@ class TestResolvePushedImageDigest:
         )
         assert out == "us-central1-docker.pkg.dev/proj/repo@sha256:abc123"
 
-    def test_registry_port_without_tag_still_resolves_the_digest(self):
-        runner = self._runner('["localhost:5000/repo@sha256:abc123"]')
-        out = _resolve_pushed_image_digest(
-            runner,
-            remote_image="localhost:5000/repo",
-            docker_config_dir=None,
-            verbose=False,
-        )
-        assert out == "localhost:5000/repo@sha256:abc123"
-
     def test_empty_repodigests_falls_back_with_warning(self, mocker):
         emitter = mocker.MagicMock()
         mocker.patch("langgraph_cli.deploy._get_emitter", return_value=emitter)
@@ -933,289 +747,3 @@ class TestResolvePushedImageDigest:
         frame_locals = captured["coro"].cr_frame.f_locals
         assert "--config" not in frame_locals["args"]
         captured["coro"].close()
-
-
-class TestListener:
-    @pytest.mark.parametrize(
-        ("resource", "expected"),
-        [
-            pytest.param(
-                {
-                    "id": "listener-1",
-                    "compute_id": "prod-cluster",
-                    "compute_config": {"k8s_namespaces": ["agents", "agents-staging"]},
-                },
-                Listener("listener-1", "prod-cluster", ("agents", "agents-staging")),
-                id="reads_id_cluster_and_namespaces",
-            ),
-            pytest.param(
-                {"id": "listener-1", "compute_id": "c", "compute_config": {}},
-                Listener("listener-1", "c", ()),
-                id="missing_namespaces",
-            ),
-            pytest.param(
-                {"id": "listener-1", "compute_id": "c", "compute_config": None},
-                Listener("listener-1", "c", ()),
-                id="null_compute_config",
-            ),
-            pytest.param(
-                {"id": "listener-1"},
-                Listener("listener-1", "", ()),
-                id="only_an_id",
-            ),
-        ],
-    )
-    def test_from_resource_reads_the_control_plane_shape(self, resource, expected):
-        assert Listener.from_resource(resource) == expected
-
-
-ONE_NAMESPACE = Listener("listener-1", "prod-cluster", ("agents",))
-TWO_NAMESPACES = Listener("listener-2", "multi-cluster", ("agents", "agents-staging"))
-NO_NAMESPACE = Listener("listener-3", "broken-cluster", ())
-
-
-class TestRequestedPlacement:
-    @pytest.mark.parametrize(
-        ("request_", "listeners", "expected"),
-        [
-            pytest.param(
-                RequestedPlacement(), (), Unplaced(), id="no_listeners_no_request"
-            ),
-            pytest.param(
-                RequestedPlacement(),
-                (ONE_NAMESPACE,),
-                OnListener("listener-1", "agents"),
-                id="uses_the_only_possible_answer",
-            ),
-            pytest.param(
-                RequestedPlacement(k8s_namespace="agents-staging"),
-                (TWO_NAMESPACES,),
-                OnListener("listener-2", "agents-staging"),
-                id="namespace_alone_picks_the_only_listener",
-            ),
-        ],
-    )
-    def test_resolves_to_a_placement(self, request_, listeners, expected):
-        assert request_.among(listeners) == expected
-
-    @pytest.mark.parametrize(
-        ("request_", "listeners", "message"),
-        [
-            pytest.param(
-                RequestedPlacement(listener_id="listener-1"),
-                (),
-                "no listeners",
-                id="workspace_has_no_listeners",
-            ),
-            pytest.param(
-                RequestedPlacement(),
-                (ONE_NAMESPACE, TWO_NAMESPACES),
-                "--listener-id",
-                id="several_listeners_need_a_choice",
-            ),
-            pytest.param(
-                RequestedPlacement(k8s_namespace="agents"),
-                (ONE_NAMESPACE, TWO_NAMESPACES),
-                "--listener-id",
-                id="namespace_alone_is_ambiguous_with_several_listeners",
-            ),
-            pytest.param(
-                RequestedPlacement(k8s_namespace="agents"),
-                (),
-                "no listeners",
-                id="namespace_without_any_listener",
-            ),
-            pytest.param(
-                RequestedPlacement(),
-                (TWO_NAMESPACES,),
-                "--k8s-namespace",
-                id="several_namespaces_need_a_choice",
-            ),
-        ],
-    )
-    def test_refuses_and_names_the_choices(self, request_, listeners, message):
-        with pytest.raises(click.UsageError, match=message):
-            request_.among(listeners)
-
-    def test_the_error_lists_every_listener_with_its_cluster_and_namespaces(self):
-        with pytest.raises(click.UsageError) as error:
-            RequestedPlacement().among((ONE_NAMESPACE, TWO_NAMESPACES))
-
-        assert "listener-1" in error.value.message
-        assert "prod-cluster" in error.value.message
-        assert "agents-staging" in error.value.message
-
-    @pytest.mark.parametrize(
-        ("placement", "expected"),
-        [
-            pytest.param(Unplaced(), {}, id="unplaced_adds_nothing"),
-            pytest.param(
-                OnListener("listener-1", "agents"),
-                {
-                    "listener_id": "listener-1",
-                    "listener_config": {"k8s_namespace": "agents"},
-                },
-                id="placed_carries_listener_and_namespace",
-            ),
-        ],
-    )
-    def test_source_config_matches_the_control_plane_shape(self, placement, expected):
-        assert placement.source_config() == expected
-
-
-def test_finding_a_deployment_by_name_narrows_the_search_for_every_server_version():
-    seen: dict = {}
-
-    def handler(req: httpx.Request) -> httpx.Response:
-        seen["params"] = dict(req.url.params)
-        return httpx.Response(
-            200,
-            json={"resources": [{"id": "dep-1", "name": "agent", "source": "github"}]},
-        )
-
-    client = HostBackendClient(
-        "https://api.example.com", "key", transport=httpx.MockTransport(handler)
-    )
-
-    found = find_deployment_by_name(client, "agent")
-
-    assert seen["params"] == {
-        "name": "agent",
-        "name_contains": "agent",
-        "limit": "100",
-    }
-    assert found == ExistingDeployment("dep-1", "github")
-
-
-def test_a_server_that_ignores_the_exact_name_filter_never_matches_another_deployment():
-    client = HostBackendClient(
-        "https://api.example.com",
-        "key",
-        transport=httpx.MockTransport(
-            lambda req: httpx.Response(
-                200,
-                json={
-                    "resources": [
-                        {
-                            "id": "dep-other",
-                            "name": "another-teams-agent",
-                            "source": "external_docker",
-                        }
-                    ]
-                },
-            )
-        ),
-    )
-
-    assert find_deployment_by_name(client, "brand-new-agent") is None
-
-
-def test_a_full_page_without_a_match_refuses_to_claim_the_name_is_free():
-    page = [
-        {"id": f"dep-{index}", "name": f"other-agent-{index}"} for index in range(100)
-    ]
-    client = HostBackendClient(
-        "https://api.example.com",
-        "key",
-        transport=httpx.MockTransport(
-            lambda req: httpx.Response(200, json={"resources": page})
-        ),
-    )
-
-    with pytest.raises(click.ClickException, match="--deployment-id"):
-        find_deployment_by_name(client, "brand-new-agent")
-
-
-def test_a_partial_page_without_a_match_means_the_name_is_free():
-    client = HostBackendClient(
-        "https://api.example.com",
-        "key",
-        transport=httpx.MockTransport(
-            lambda req: httpx.Response(
-                200, json={"resources": [{"id": "dep-1", "name": "other"}]}
-            )
-        ),
-    )
-
-    assert find_deployment_by_name(client, "brand-new-agent") is None
-
-
-@pytest.mark.parametrize(
-    "resource",
-    [
-        pytest.param({"compute_id": "c"}, id="no_id"),
-        pytest.param({"id": ""}, id="empty_id"),
-    ],
-)
-def test_a_listener_without_an_id_is_refused(resource):
-    with pytest.raises(HostBackendError, match="without an id"):
-        Listener.from_resource(resource)
-
-
-def test_a_deployment_id_with_listener_flags_is_refused_without_probing_docker(
-    monkeypatch,
-):
-    def explode() -> tuple[bool, str | None]:
-        raise AssertionError("docker must not be probed for an argv-only conflict")
-
-    monkeypatch.setattr(deploy_mod, "can_build_locally", explode)
-
-    with pytest.raises(click.UsageError, match="--deployment-id"):
-        _select_source(
-            push_to="registry.example.com/app",
-            image=None,
-            image_name=None,
-            tag=None,
-            remote_build_flag=None,
-            placement=RequestedPlacement(listener_id="listener-1"),
-            selector=ById("dep-1"),
-        )
-
-
-class TestPlacementOnAKnownListener:
-    @pytest.mark.parametrize(
-        ("request_", "listener", "expected"),
-        [
-            pytest.param(
-                RequestedPlacement(listener_id="listener-1"),
-                ONE_NAMESPACE,
-                OnListener("listener-1", "agents"),
-                id="the_only_namespace_is_used",
-            ),
-            pytest.param(
-                RequestedPlacement(listener_id="listener-2", k8s_namespace="agents"),
-                TWO_NAMESPACES,
-                OnListener("listener-2", "agents"),
-                id="the_chosen_namespace_is_used",
-            ),
-        ],
-    )
-    def test_places_on_the_listener(self, request_, listener, expected):
-        assert request_.on(listener) == expected
-
-    @pytest.mark.parametrize(
-        ("request_", "listener", "message"),
-        [
-            pytest.param(
-                RequestedPlacement(listener_id="listener-2"),
-                TWO_NAMESPACES,
-                "--k8s-namespace",
-                id="several_namespaces_need_a_choice",
-            ),
-            pytest.param(
-                RequestedPlacement(listener_id="listener-2", k8s_namespace="nope"),
-                TWO_NAMESPACES,
-                "does not serve namespace",
-                id="unknown_namespace",
-            ),
-            pytest.param(
-                RequestedPlacement(listener_id="listener-3"),
-                NO_NAMESPACE,
-                "serves no namespaces",
-                id="listener_without_namespaces",
-            ),
-        ],
-    )
-    def test_refuses_and_names_the_namespaces(self, request_, listener, message):
-        with pytest.raises(click.UsageError, match=message):
-            request_.on(listener)

libs/cli/tests/unit_tests/test_host_backend.py

@@ -3,16 +3,29 @@ import json
 import httpx
 import pytest
 
-from langgraph_cli.host_backend import (
-    ControlPlaneEndpoints,
-    HostBackendClient,
-    HostBackendError,
-)
+from langgraph_cli.host_backend import HostBackendClient, HostBackendError
+
+
+@pytest.fixture
+def mock_transport():
+    return httpx.MockTransport(lambda req: httpx.Response(200, json={"ok": True}))
+
+
+@pytest.fixture
+def client(mock_transport):
+    c = HostBackendClient("https://api.example.com", "test-key")
+    c._client = httpx.Client(
+        base_url="https://api.example.com",
+        transport=mock_transport,
+        headers={"X-Api-Key": "test-key", "Accept": "application/json"},
+        timeout=30,
+    )
+    return c
 
 
 def test_constructor_strips_trailing_slash():
     c = HostBackendClient("https://api.example.com/", "key")
-    assert c.base_url == "https://api.example.com"
+    assert str(c._client.base_url) == "https://api.example.com"
 
 
 def test_constructor_empty_url_raises():
@@ -26,8 +39,12 @@ def test_request_sends_headers():
         assert req.headers["accept"] == "application/json"
         return httpx.Response(200, json={"ok": True})
 
-    c = HostBackendClient(
-        "https://api.example.com", "test-key", transport=httpx.MockTransport(handler)
+    c = HostBackendClient("https://api.example.com", "test-key")
+    c._client = httpx.Client(
+        base_url="https://api.example.com",
+        transport=httpx.MockTransport(handler),
+        headers={"X-Api-Key": "test-key", "Accept": "application/json"},
+        timeout=30,
     )
     result = c._request("GET", "/test")
     assert result == {"ok": True}
@@ -39,8 +56,12 @@ def test_request_sends_json_payload():
         assert req.content == b'{"key":"value"}'
         return httpx.Response(200, json={"created": True})
 
-    c = HostBackendClient(
-        "https://api.example.com", "test-key", transport=httpx.MockTransport(handler)
+    c = HostBackendClient("https://api.example.com", "test-key")
+    c._client = httpx.Client(
+        base_url="https://api.example.com",
+        transport=httpx.MockTransport(handler),
+        headers={"X-Api-Key": "test-key", "Accept": "application/json"},
+        timeout=30,
     )
     result = c._request("POST", "/test", {"key": "value"})
     assert result == {"created": True}
@@ -48,13 +69,25 @@ def test_request_sends_json_payload():
 
 def test_request_empty_body_returns_none():
     transport = httpx.MockTransport(lambda req: httpx.Response(200, content=b""))
-    c = HostBackendClient("https://api.example.com", "test-key", transport=transport)
+    c = HostBackendClient("https://api.example.com", "test-key")
+    c._client = httpx.Client(
+        base_url="https://api.example.com",
+        transport=transport,
+        headers={"X-Api-Key": "test-key", "Accept": "application/json"},
+        timeout=30,
+    )
     assert c._request("DELETE", "/test") is None
 
 
 def test_request_http_error_raises():
     transport = httpx.MockTransport(lambda req: httpx.Response(404, text="not found"))
-    c = HostBackendClient("https://api.example.com", "test-key", transport=transport)
+    c = HostBackendClient("https://api.example.com", "test-key")
+    c._client = httpx.Client(
+        base_url="https://api.example.com",
+        transport=transport,
+        headers={"X-Api-Key": "test-key", "Accept": "application/json"},
+        timeout=30,
+    )
     with pytest.raises(HostBackendError, match="404"):
         c._request("GET", "/missing")
 
@@ -63,7 +96,13 @@ def test_request_invalid_json_raises():
     transport = httpx.MockTransport(
         lambda req: httpx.Response(200, content=b"not json")
     )
-    c = HostBackendClient("https://api.example.com", "test-key", transport=transport)
+    c = HostBackendClient("https://api.example.com", "test-key")
+    c._client = httpx.Client(
+        base_url="https://api.example.com",
+        transport=transport,
+        headers={"X-Api-Key": "test-key", "Accept": "application/json"},
+        timeout=30,
+    )
     with pytest.raises(HostBackendError, match="Failed to decode"):
         c._request("GET", "/bad-json")
 
@@ -72,20 +111,84 @@ def test_request_transport_error_raises():
     def handler(req: httpx.Request) -> httpx.Response:
         raise httpx.ConnectError("connection refused")
 
-    c = HostBackendClient(
-        "https://api.example.com", "test-key", transport=httpx.MockTransport(handler)
+    c = HostBackendClient("https://api.example.com", "test-key")
+    c._client = httpx.Client(
+        base_url="https://api.example.com",
+        transport=httpx.MockTransport(handler),
+        headers={"X-Api-Key": "test-key", "Accept": "application/json"},
+        timeout=30,
     )
     with pytest.raises(HostBackendError, match="connection refused"):
         c._request("GET", "/test")
 
 
+def test_create_deployment(client):
+    result = client.create_deployment(
+        name="my-deploy", deployment_type="dev", source="internal_docker"
+    )
+    assert result == {"ok": True}
+
+
+def test_get_deployment(client):
+    result = client.get_deployment("dep-123")
+    assert result == {"ok": True}
+
+
+def test_list_deployments(client):
+    result = client.list_deployments("my-app")
+    assert result == {"ok": True}
+
+
+def test_list_deployments_sends_query_params():
+    def handler(req: httpx.Request) -> httpx.Response:
+        assert req.url.path == "/v2/deployments"
+        assert req.url.params["name_contains"] == "my app"
+        return httpx.Response(200, json={"ok": True})
+
+    c = HostBackendClient("https://api.example.com", "test-key")
+    c._client = httpx.Client(
+        base_url="https://api.example.com",
+        transport=httpx.MockTransport(handler),
+        headers={"X-Api-Key": "test-key", "Accept": "application/json"},
+        timeout=30,
+    )
+    result = c.list_deployments("my app")
+    assert result == {"ok": True}
+
+
+def test_delete_deployment(client):
+    result = client.delete_deployment("dep-123")
+    assert result == {"ok": True}
+
+
+def test_request_push_token(client):
+    result = client.request_push_token("dep-123")
+    assert result == {"ok": True}
+
+
+def test_update_deployment(client):
+    result = client.update_deployment(
+        "dep-123", "image:latest", secrets=[{"name": "KEY", "value": "val"}]
+    )
+    assert result == {"ok": True}
+
+
+def test_update_deployment_no_secrets(client):
+    result = client.update_deployment("dep-123", "image:latest")
+    assert result == {"ok": True}
+
+
 def _capturing_client(captured: dict) -> HostBackendClient:
     def handler(req: httpx.Request) -> httpx.Response:
         captured["body"] = req.read()
         return httpx.Response(200, json={"ok": True})
 
-    c = HostBackendClient(
-        "https://api.example.com", "key", transport=httpx.MockTransport(handler)
+    c = HostBackendClient("https://api.example.com", "key")
+    c._client = httpx.Client(
+        base_url="https://api.example.com",
+        transport=httpx.MockTransport(handler),
+        headers={"X-Api-Key": "key", "Accept": "application/json"},
+        timeout=30,
     )
     return c
 
@@ -96,7 +199,6 @@ def test_update_deployment_forwards_tracked_packages():
     c.update_deployment(
         "dep-123",
         "image:latest",
-        revision_source="internal_docker",
         tracked_packages=["google-adk:1.0.0"],
     )
     body = json.loads(captured["body"])
@@ -107,7 +209,7 @@ def test_update_deployment_forwards_tracked_packages():
 def test_update_deployment_omits_tracked_packages_when_absent():
     captured: dict = {}
     c = _capturing_client(captured)
-    c.update_deployment("dep-123", "image:latest", revision_source="internal_docker")
+    c.update_deployment("dep-123", "image:latest")
     body = json.loads(captured["body"])
     assert "tracked_packages" not in body
 
@@ -139,14 +241,33 @@ def test_update_deployment_internal_source_omits_tracked_packages_when_absent():
     assert "tracked_packages" not in body
 
 
+def test_list_revisions(client):
+    result = client.list_revisions("dep-123", limit=5)
+    assert result == {"ok": True}
+
+
+def test_get_revision(client):
+    result = client.get_revision("dep-123", "rev-456")
+    assert result == {"ok": True}
+
+
+def test_get_build_logs(client):
+    result = client.get_build_logs("proj-1", "rev-1", {"limit": 10})
+    assert result == {"ok": True}
+
+
 def test_get_deploy_logs_all_revisions():
     def handler(req: httpx.Request) -> httpx.Response:
         assert "/v1/projects/proj-1/deploy_logs" in str(req.url)
         assert "/revisions/" not in str(req.url)
         return httpx.Response(200, json={"logs": [{"message": "running"}]})
 
-    c = HostBackendClient(
-        "https://api.example.com", "key", transport=httpx.MockTransport(handler)
+    c = HostBackendClient("https://api.example.com", "key")
+    c._client = httpx.Client(
+        base_url="https://api.example.com",
+        transport=httpx.MockTransport(handler),
+        headers={"X-Api-Key": "key", "Accept": "application/json"},
+        timeout=30,
     )
     result = c.get_deploy_logs("proj-1", {"limit": 10})
     assert result == {"logs": [{"message": "running"}]}
@@ -157,520 +278,12 @@ def test_get_deploy_logs_specific_revision():
         assert "/v1/projects/proj-1/revisions/rev-2/deploy_logs" in str(req.url)
         return httpx.Response(200, json={"logs": []})
 
-    c = HostBackendClient(
-        "https://api.example.com", "key", transport=httpx.MockTransport(handler)
+    c = HostBackendClient("https://api.example.com", "key")
+    c._client = httpx.Client(
+        base_url="https://api.example.com",
+        transport=httpx.MockTransport(handler),
+        headers={"X-Api-Key": "key", "Accept": "application/json"},
+        timeout=30,
     )
     result = c.get_deploy_logs("proj-1", {"limit": 10}, revision_id="rev-2")
     assert result == {"logs": []}
-
-
-def _routing_client(seen: dict) -> HostBackendClient:
-    def handler(req: httpx.Request) -> httpx.Response:
-        seen["method"] = req.method
-        seen["url"] = str(req.url)
-        return httpx.Response(200, json={"ok": True})
-
-    c = HostBackendClient(
-        "https://api.example.com/prefix", "key", transport=httpx.MockTransport(handler)
-    )
-    return c
-
-
-@pytest.mark.parametrize(
-    ("call", "expected_body"),
-    [
-        pytest.param(
-            lambda c: c.create_deployment(
-                name="my-deploy",
-                source="internal_docker",
-                source_config={"deployment_type": "dev"},
-                source_revision_config={},
-            ),
-            {
-                "name": "my-deploy",
-                "source": "internal_docker",
-                "source_config": {"deployment_type": "dev"},
-                "source_revision_config": {},
-            },
-            id="internal_docker_create_omits_secrets_key_when_not_given",
-        ),
-        pytest.param(
-            lambda c: c.create_deployment(
-                name="my-deploy",
-                source="internal_docker",
-                source_config={"deployment_type": "prod"},
-                source_revision_config={},
-                secrets=[{"name": "KEY", "value": "val"}],
-            ),
-            {
-                "name": "my-deploy",
-                "source": "internal_docker",
-                "source_config": {"deployment_type": "prod"},
-                "source_revision_config": {},
-                "secrets": [{"name": "KEY", "value": "val"}],
-            },
-            id="internal_docker_create_forwards_secrets",
-        ),
-        pytest.param(
-            lambda c: c.update_deployment(
-                "dep-123",
-                "registry.example.com/app@sha256:abc",
-                revision_source="internal_docker",
-                secrets=[{"name": "KEY", "value": "val"}],
-            ),
-            {
-                "revision_source": "internal_docker",
-                "source_revision_config": {
-                    "image_uri": "registry.example.com/app@sha256:abc"
-                },
-                "secrets": [{"name": "KEY", "value": "val"}],
-            },
-            id="internal_docker_revision_names_its_source",
-        ),
-        pytest.param(
-            lambda c: c.update_deployment_internal_source(
-                "dep-123",
-                source_tarball_path="tarballs/src.tgz",
-                config_path="langgraph.json",
-                secrets=[],
-                install_command="yarn install",
-                build_command="yarn build",
-            ),
-            {
-                "revision_source": "internal_source",
-                "source_revision_config": {
-                    "source_tarball_path": "tarballs/src.tgz",
-                    "langgraph_config_path": "langgraph.json",
-                },
-                "source_config": {
-                    "install_command": "yarn install",
-                    "build_command": "yarn build",
-                },
-                "secrets": [],
-            },
-            id="internal_source_revision_sends_js_build_commands",
-        ),
-        pytest.param(
-            lambda c: c.update_deployment_internal_source(
-                "dep-123",
-                source_tarball_path="tarballs/src.tgz",
-                config_path="langgraph.json",
-            ),
-            {
-                "revision_source": "internal_source",
-                "source_revision_config": {
-                    "source_tarball_path": "tarballs/src.tgz",
-                    "langgraph_config_path": "langgraph.json",
-                },
-            },
-            id="internal_source_revision_omits_source_config_without_commands",
-        ),
-        pytest.param(
-            lambda c: c.create_deployment(
-                name="agent",
-                source="external_docker",
-                source_config={"resource_spec": {}},
-                source_revision_config={
-                    "image_uri": "registry.example.com/agent@sha256:1"
-                },
-                secrets=[],
-            ),
-            {
-                "name": "agent",
-                "source": "external_docker",
-                "source_config": {"resource_spec": {}},
-                "source_revision_config": {
-                    "image_uri": "registry.example.com/agent@sha256:1"
-                },
-                "secrets": [],
-            },
-            id="create_sends_the_source_configs_as_given",
-        ),
-        pytest.param(
-            lambda c: c.update_deployment(
-                "dep-1", "registry.example.com/agent@sha256:2", revision_source=None
-            ),
-            {
-                "source_revision_config": {
-                    "image_uri": "registry.example.com/agent@sha256:2"
-                }
-            },
-            id="revision_without_source_override_omits_revision_source",
-        ),
-        pytest.param(
-            lambda c: c.update_deployment(
-                "dep-1",
-                "registry.example.com/agent@sha256:2",
-                revision_source="internal_docker",
-                tracked_packages=["langgraph:1.0.0"],
-            ),
-            {
-                "revision_source": "internal_docker",
-                "source_revision_config": {
-                    "image_uri": "registry.example.com/agent@sha256:2"
-                },
-                "tracked_packages": ["langgraph:1.0.0"],
-            },
-            id="revision_with_source_override_names_it",
-        ),
-    ],
-)
-def test_request_body_matches_control_plane_contract(call, expected_body):
-    captured: dict = {}
-    call(_capturing_client(captured))
-    assert json.loads(captured["body"]) == expected_body
-
-
-@pytest.mark.parametrize(
-    ("call", "method", "route"),
-    [
-        pytest.param(
-            lambda c: c.create_deployment(
-                name="n",
-                source="internal_docker",
-                source_config={"deployment_type": "dev"},
-                source_revision_config={},
-            ),
-            "POST",
-            "/v2/deployments",
-            id="create_deployment",
-        ),
-        pytest.param(
-            lambda c: c.get_deployment("dep-1"),
-            "GET",
-            "/v2/deployments/dep-1",
-            id="get_deployment",
-        ),
-        pytest.param(
-            lambda c: c.delete_deployment("dep-1"),
-            "DELETE",
-            "/v2/deployments/dep-1",
-            id="delete_deployment",
-        ),
-        pytest.param(
-            lambda c: c.update_deployment("dep-1", "img", revision_source=None),
-            "PATCH",
-            "/v2/deployments/dep-1",
-            id="patch_deployment",
-        ),
-        pytest.param(
-            lambda c: c.request_push_token("dep-1"),
-            "POST",
-            "/v2/deployments/dep-1/push-token",
-            id="push_token",
-        ),
-        pytest.param(
-            lambda c: c.request_upload_url("dep-1"),
-            "POST",
-            "/v2/deployments/dep-1/upload-url",
-            id="upload_url",
-        ),
-        pytest.param(
-            lambda c: c.list_revisions("dep-1", limit=5),
-            "GET",
-            "/v2/deployments/dep-1/revisions?limit=5",
-            id="list_revisions_puts_limit_in_query",
-        ),
-        pytest.param(
-            lambda c: c.get_revision("dep-1", "rev-2"),
-            "GET",
-            "/v2/deployments/dep-1/revisions/rev-2",
-            id="get_revision",
-        ),
-        pytest.param(
-            lambda c: c.get_build_logs("dep-1", "rev-2", {"limit": 10}),
-            "POST",
-            "/v1/projects/dep-1/revisions/rev-2/build_logs",
-            id="build_logs",
-        ),
-    ],
-)
-def test_request_targets_control_plane_route_under_base_url(call, method, route):
-    seen: dict = {}
-    call(_routing_client(seen))
-    assert (seen["method"], seen["url"]) == (
-        method,
-        f"https://api.example.com/prefix{route}",
-    )
-
-
-def test_injected_transport_receives_requests_under_the_prefixed_base_url():
-    seen: dict = {}
-
-    def handler(req: httpx.Request) -> httpx.Response:
-        seen["url"] = str(req.url)
-        seen["api_key"] = req.headers["x-api-key"]
-        return httpx.Response(200, json={"ok": True})
-
-    c = HostBackendClient(
-        "https://smith.example.com/api-host",
-        "key",
-        transport=httpx.MockTransport(handler),
-    )
-
-    assert c.list_revisions("dep-1", limit=2) == []
-    assert seen == {
-        "url": "https://smith.example.com/api-host/v2/deployments/dep-1/revisions?limit=2",
-        "api_key": "key",
-    }
-
-
-CLOUD = ("https://api.host.langchain.com", "https://smith.langchain.com")
-
-
-@pytest.mark.parametrize(
-    ("host_url", "langsmith_endpoint", "expected"),
-    [
-        pytest.param(None, None, CLOUD, id="nothing_configured_targets_cloud"),
-        pytest.param(
-            None, "https://api.smith.langchain.com", CLOUD, id="cloud_langsmith_api"
-        ),
-        pytest.param(
-            None,
-            "https://api.smith.langchain.com/api/v1",
-            CLOUD,
-            id="cloud_langsmith_api_with_versioned_path",
-        ),
-        pytest.param(
-            None, "https://api.langchain.com", CLOUD, id="cloud_langchain_api_alias"
-        ),
-        pytest.param(
-            None,
-            "https://xapi.smith.langchain.com",
-            CLOUD,
-            id="lookalike_cloud_host_is_not_rewritten_into_a_control_plane",
-        ),
-        pytest.param(
-            None,
-            "https://eu.api.smith.langchain.com",
-            ("https://eu.api.host.langchain.com", "https://eu.smith.langchain.com"),
-            id="eu_cloud_maps_to_eu_control_plane",
-        ),
-        pytest.param(
-            None,
-            "https://dev.api.smith.langchain.com",
-            ("https://dev.api.host.langchain.com", "https://dev.smith.langchain.com"),
-            id="dev_cloud_maps_to_dev_control_plane",
-        ),
-        pytest.param(
-            None,
-            "https://aks.smith.langchain.dev/api",
-            (
-                "https://aks.smith.langchain.dev/api-host",
-                "https://aks.smith.langchain.dev",
-            ),
-            id="self_hosted_api_path_becomes_api_host",
-        ),
-        pytest.param(
-            None,
-            "https://smith.example.com/api/v1",
-            ("https://smith.example.com/api-host", "https://smith.example.com"),
-            id="self_hosted_versioned_api_path_becomes_api_host",
-        ),
-        pytest.param(
-            None,
-            "https://smith.example.com",
-            ("https://smith.example.com/api-host", "https://smith.example.com"),
-            id="self_hosted_origin_gets_api_host_appended",
-        ),
-        pytest.param(
-            None,
-            "https://corp.example.com/langsmith/api/v1",
-            (
-                "https://corp.example.com/langsmith/api-host",
-                "https://corp.example.com/langsmith",
-            ),
-            id="self_hosted_path_prefix_is_kept",
-        ),
-        pytest.param(
-            "https://custom.host.example",
-            "https://aks.smith.langchain.dev/api",
-            ("https://custom.host.example", "https://smith.langchain.com"),
-            id="explicit_host_url_beats_langsmith_endpoint",
-        ),
-        pytest.param(
-            "https://api.host.langchain.com",
-            "https://aks.smith.langchain.dev/api",
-            CLOUD,
-            id="explicit_cloud_host_url_beats_self_hosted_endpoint",
-        ),
-        pytest.param(
-            "https://smith.example.com/api-host/",
-            None,
-            ("https://smith.example.com/api-host", "https://smith.example.com"),
-            id="explicit_api_host_url_derives_dashboard_root",
-        ),
-        pytest.param(
-            "https://corp.example.com/langsmith/api-host",
-            None,
-            (
-                "https://corp.example.com/langsmith/api-host",
-                "https://corp.example.com/langsmith",
-            ),
-            id="explicit_api_host_url_keeps_path_prefix_in_dashboard",
-        ),
-        pytest.param(
-            "http://localhost:8080",
-            None,
-            ("http://localhost:8080", "http://localhost:8080"),
-            id="localhost_dashboard_is_the_same_origin",
-        ),
-        pytest.param(
-            "http://localhost:8080/api-host",
-            None,
-            ("http://localhost:8080/api-host", "http://localhost:8080"),
-            id="localhost_api_host_dashboard_is_the_origin",
-        ),
-        pytest.param(
-            "https://eu.api.host.langchain.com",
-            None,
-            ("https://eu.api.host.langchain.com", "https://eu.smith.langchain.com"),
-            id="regional_control_plane_maps_to_regional_dashboard",
-        ),
-    ],
-)
-def test_control_plane_endpoints_resolve(host_url, langsmith_endpoint, expected):
-    endpoints = ControlPlaneEndpoints.resolve(host_url, langsmith_endpoint)
-
-    assert (endpoints.control_plane_url, endpoints.dashboard_url) == expected
-
-
-@pytest.mark.parametrize(
-    ("payload", "expected"),
-    [
-        pytest.param(
-            {"resources": [{"id": "a"}, {"id": "b"}]},
-            [{"id": "a"}, {"id": "b"}],
-            id="list_returns_the_resources",
-        ),
-        pytest.param({"resources": []}, [], id="empty_list"),
-        pytest.param({}, [], id="missing_key"),
-        pytest.param({"resources": None}, [], id="null_resources"),
-        pytest.param(
-            {"resources": ["nope", {"id": "a"}]}, [{"id": "a"}], id="skips_non_objects"
-        ),
-        pytest.param([], [], id="unexpected_envelope"),
-    ],
-)
-def test_list_endpoints_return_resource_objects(payload, expected):
-    def handler(req: httpx.Request) -> httpx.Response:
-        return httpx.Response(200, json=payload)
-
-    c = HostBackendClient(
-        "https://api.example.com", "key", transport=httpx.MockTransport(handler)
-    )
-
-    assert c.list_deployments() == expected
-
-
-def test_list_listeners_asks_for_a_full_page():
-    seen: dict = {}
-
-    def handler(req: httpx.Request) -> httpx.Response:
-        seen["url"] = str(req.url)
-        return httpx.Response(200, json={"resources": [{"id": "listener-1"}]})
-
-    c = HostBackendClient(
-        "https://api.example.com", "key", transport=httpx.MockTransport(handler)
-    )
-
-    assert c.list_listeners() == [{"id": "listener-1"}]
-    assert seen["url"] == "https://api.example.com/v2/listeners?limit=100"
-
-
-@pytest.mark.parametrize(
-    ("control_plane_url", "expected"),
-    [
-        pytest.param("https://api.host.langchain.com", True, id="cloud"),
-        pytest.param("https://eu.api.host.langchain.com", True, id="cloud_region"),
-        pytest.param("https://dev.api.host.langchain.com", True, id="cloud_dev"),
-        pytest.param("https://smith.example.com/api-host", False, id="self_hosted"),
-        pytest.param(
-            "https://corp.example.com/langsmith/api-host",
-            False,
-            id="self_hosted_prefix",
-        ),
-        pytest.param("http://localhost:8080/api-host", False, id="local"),
-        pytest.param(
-            "https://evil-api.host.langchain.com", False, id="lookalike_needs_a_dot"
-        ),
-    ],
-)
-def test_is_cloud_recognises_the_managed_control_plane(control_plane_url, expected):
-    endpoints = ControlPlaneEndpoints.from_control_plane_url(control_plane_url)
-
-    assert endpoints.is_cloud is expected
-
-
-@pytest.mark.parametrize(
-    ("call", "expected_params"),
-    [
-        pytest.param(
-            lambda c: c.list_deployments(name="agent"),
-            {"name": "agent"},
-            id="exact_name_filters_server_side",
-        ),
-        pytest.param(
-            lambda c: c.list_deployments(name_contains="age"),
-            {"name_contains": "age"},
-            id="substring_search_keeps_its_own_parameter",
-        ),
-        pytest.param(
-            lambda c: c.list_deployments(),
-            {},
-            id="no_filter_sends_no_parameters",
-        ),
-        pytest.param(
-            lambda c: c.list_deployments(
-                name="agent", name_contains="agent", limit=100
-            ),
-            {"name": "agent", "name_contains": "agent", "limit": "100"},
-            id="both_filters_travel_together_for_older_servers",
-        ),
-    ],
-)
-def test_list_deployments_sends_one_name_filter(call, expected_params):
-    seen: dict = {}
-
-    def handler(req: httpx.Request) -> httpx.Response:
-        seen.update(dict(req.url.params))
-        return httpx.Response(200, json={"resources": []})
-
-    call(
-        HostBackendClient(
-            "https://api.example.com", "key", transport=httpx.MockTransport(handler)
-        )
-    )
-
-    assert seen == expected_params
-
-
-@pytest.mark.parametrize(
-    ("body", "expected"),
-    [
-        pytest.param(
-            {"detail": "Source configuration error: bad listener"},
-            "Source configuration error: bad listener",
-            id="fastapi_detail_is_unwrapped",
-        ),
-        pytest.param(
-            {"detail": {"loc": ["body"], "msg": "nope"}},
-            None,
-            id="a_structured_detail_is_left_alone",
-        ),
-        pytest.param({"other": "shape"}, None, id="an_unknown_shape_is_left_alone"),
-    ],
-)
-def test_error_detail_is_readable(body, expected):
-    c = HostBackendClient(
-        "https://api.example.com",
-        "key",
-        transport=httpx.MockTransport(lambda req: httpx.Response(400, json=body)),
-    )
-
-    with pytest.raises(HostBackendError) as error:
-        c.get_deployment("dep-1")
-
-    assert error.value.detail == expected
-    if expected is not None:
-        assert error.value.message.endswith(expected)

libs/cli/tests/unit_tests/test_image_reference.py

@@ -1,71 +0,0 @@
-import pytest
-
-from langgraph_cli.image_reference import ImageReference
-
-
-@pytest.mark.parametrize(
-    ("reference", "repository", "tag"),
-    [
-        pytest.param(
-            "registry.example.com/team/app:v1",
-            "registry.example.com/team/app",
-            "v1",
-            id="tag_after_last_slash",
-        ),
-        pytest.param(
-            "registry.example.com/team/app",
-            "registry.example.com/team/app",
-            None,
-            id="no_tag",
-        ),
-        pytest.param(
-            "localhost:5000/app",
-            "localhost:5000/app",
-            None,
-            id="registry_port_is_not_a_tag",
-        ),
-        pytest.param(
-            "localhost:5000/app:latest",
-            "localhost:5000/app",
-            "latest",
-            id="registry_port_with_tag",
-        ),
-        pytest.param("app:dev", "app", "dev", id="bare_name_with_tag"),
-    ],
-)
-def test_parse_splits_repository_and_tag(reference, repository, tag):
-    assert ImageReference.parse(reference) == ImageReference(repository, tag)
-
-
-def test_with_tag_replaces_the_tag():
-    assert ImageReference("r/app", "v1").with_tag("v2") == ImageReference("r/app", "v2")
-
-
-@pytest.mark.parametrize(
-    ("reference", "expected"),
-    [
-        pytest.param(ImageReference("r/app", "v1"), "r/app:v1", id="tagged"),
-        pytest.param(ImageReference("r/app"), "r/app", id="untagged"),
-    ],
-)
-def test_str_renders_the_docker_reference(reference, expected):
-    assert str(reference) == expected
-
-
-@pytest.mark.parametrize(
-    ("repo_digest", "expected"),
-    [
-        pytest.param("localhost:5000/app@sha256:abc", True, id="same_repository"),
-        pytest.param("localhost:5000/app-2@sha256:abc", False, id="other_repository"),
-        pytest.param("mirror.example.com/app@sha256:abc", False, id="other_registry"),
-    ],
-)
-def test_matches_digest_only_for_the_same_repository(repo_digest, expected):
-    assert ImageReference("localhost:5000/app", "v1").matches_digest(repo_digest) is (
-        expected
-    )
-
-
-def test_parse_rejects_a_digest_reference():
-    with pytest.raises(ValueError, match="digest"):
-        ImageReference.parse("registry.example.com/app@sha256:abc")

libs/cli/uv-examples/monorepo/uv.lock

@@ -39,15 +39,15 @@ wheels = [
 
 [[package]]
 name = "anyio"
-version = "4.14.2"
+version = "4.13.0"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "idna" },
     { name = "typing-extensions", marker = "python_full_version < '3.13'" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/61/cc/a381afa6efea9f496eff839d4a6a1aed3bfafc7b3ab4b0d1b243a12573dd/anyio-4.14.2.tar.gz", hash = "sha256:cfa139f3ed1a23ee8f88a145ddb5ac7605b8bbfd8592baacd7ce3d8bb4313c7f", size = 260176, upload-time = "2026-07-12T20:29:07.082Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/19/14/2c5dd9f512b66549ae92767a9c7b330ae88e1932ca57876909410251fe13/anyio-4.13.0.tar.gz", hash = "sha256:334b70e641fd2221c1505b3890c69882fe4a2df910cba14d97019b90b24439dc", size = 231622, upload-time = "2026-03-24T12:59:09.671Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/da/35/f2287558c17e29fafc8ef3daf819bb9834061cfa43bff8014f7df7f63bdc/anyio-4.14.2-py3-none-any.whl", hash = "sha256:9f505dda5ac9f0c8309b5e8bd445a8c2bf7246f3ce950121e45ea15bc41d1494", size = 125813, upload-time = "2026-07-12T20:29:05.763Z" },
+    { url = "https://files.pythonhosted.org/packages/da/42/e921fccf5015463e32a3cf6ee7f980a6ed0f395ceeaa45060b61d86486c2/anyio-4.13.0-py3-none-any.whl", hash = "sha256:08b310f9e24a9594186fd75b4f73f4a4152069e3853f1ed8bfbf58369f4ad708", size = 114353, upload-time = "2026-03-24T12:59:08.246Z" },
 ]
 
 [[package]]

libs/cli/uv-examples/simple/uv.lock

@@ -13,15 +13,15 @@ wheels = [
 
 [[package]]
 name = "anyio"
-version = "4.14.2"
+version = "4.13.0"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "idna" },
     { name = "typing-extensions", marker = "python_full_version < '3.13'" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/61/cc/a381afa6efea9f496eff839d4a6a1aed3bfafc7b3ab4b0d1b243a12573dd/anyio-4.14.2.tar.gz", hash = "sha256:cfa139f3ed1a23ee8f88a145ddb5ac7605b8bbfd8592baacd7ce3d8bb4313c7f", size = 260176, upload-time = "2026-07-12T20:29:07.082Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/19/14/2c5dd9f512b66549ae92767a9c7b330ae88e1932ca57876909410251fe13/anyio-4.13.0.tar.gz", hash = "sha256:334b70e641fd2221c1505b3890c69882fe4a2df910cba14d97019b90b24439dc", size = 231622, upload-time = "2026-03-24T12:59:09.671Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/da/35/f2287558c17e29fafc8ef3daf819bb9834061cfa43bff8014f7df7f63bdc/anyio-4.14.2-py3-none-any.whl", hash = "sha256:9f505dda5ac9f0c8309b5e8bd445a8c2bf7246f3ce950121e45ea15bc41d1494", size = 125813, upload-time = "2026-07-12T20:29:05.763Z" },
+    { url = "https://files.pythonhosted.org/packages/da/42/e921fccf5015463e32a3cf6ee7f980a6ed0f395ceeaa45060b61d86486c2/anyio-4.13.0-py3-none-any.whl", hash = "sha256:08b310f9e24a9594186fd75b4f73f4a4152069e3853f1ed8bfbf58369f4ad708", size = 114353, upload-time = "2026-03-24T12:59:08.246Z" },
 ]
 
 [[package]]

libs/cli/uv.lock

@@ -19,16 +19,16 @@ wheels = [
 
 [[package]]
 name = "anyio"
-version = "4.14.2"
+version = "4.13.0"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "exceptiongroup", marker = "python_full_version < '3.11'" },
     { name = "idna" },
     { name = "typing-extensions", marker = "python_full_version < '3.13'" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/61/cc/a381afa6efea9f496eff839d4a6a1aed3bfafc7b3ab4b0d1b243a12573dd/anyio-4.14.2.tar.gz", hash = "sha256:cfa139f3ed1a23ee8f88a145ddb5ac7605b8bbfd8592baacd7ce3d8bb4313c7f", size = 260176, upload-time = "2026-07-12T20:29:07.082Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/19/14/2c5dd9f512b66549ae92767a9c7b330ae88e1932ca57876909410251fe13/anyio-4.13.0.tar.gz", hash = "sha256:334b70e641fd2221c1505b3890c69882fe4a2df910cba14d97019b90b24439dc", size = 231622, upload-time = "2026-03-24T12:59:09.671Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/da/35/f2287558c17e29fafc8ef3daf819bb9834061cfa43bff8014f7df7f63bdc/anyio-4.14.2-py3-none-any.whl", hash = "sha256:9f505dda5ac9f0c8309b5e8bd445a8c2bf7246f3ce950121e45ea15bc41d1494", size = 125813, upload-time = "2026-07-12T20:29:05.763Z" },
+    { url = "https://files.pythonhosted.org/packages/da/42/e921fccf5015463e32a3cf6ee7f980a6ed0f395ceeaa45060b61d86486c2/anyio-4.13.0-py3-none-any.whl", hash = "sha256:08b310f9e24a9594186fd75b4f73f4a4152069e3853f1ed8bfbf58369f4ad708", size = 114353, upload-time = "2026-03-24T12:59:08.246Z" },
 ]
 
 [[package]]

libs/langgraph/langgraph/pregel/main.py

@@ -1950,7 +1950,7 @@ class Pregel(
             run_tasks: list[PregelTaskWrites] = []
             run_task_ids: list[str] = []
 
-            for i, (as_node, values, provided_task_id) in enumerate(valid_updates):
+            for as_node, values, provided_task_id in valid_updates:
                 # create task to run all writers of the chosen node
                 writers = self.nodes[as_node].flat_writers
                 if not writers:
@@ -1964,7 +1964,7 @@ class Pregel(
                 task_id = provided_task_id or (
                     prepared_task_ids.popleft()
                     if prepared_task_ids
-                    else _update_task_id(checkpoint["id"], i)
+                    else str(uuid5(UUID(checkpoint["id"]), INTERRUPT))
                 )
                 run_tasks.append(task)
                 run_task_ids.append(task_id)
@@ -2410,7 +2410,7 @@ class Pregel(
             run_tasks: list[PregelTaskWrites] = []
             run_task_ids: list[str] = []
 
-            for i, (as_node, values, provided_task_id) in enumerate(valid_updates):
+            for as_node, values, provided_task_id in valid_updates:
                 # create task to run all writers of the chosen node
                 writers = self.nodes[as_node].flat_writers
                 if not writers:
@@ -2424,7 +2424,7 @@ class Pregel(
                 task_id = provided_task_id or (
                     prepared_task_ids.popleft()
                     if prepared_task_ids
-                    else _update_task_id(checkpoint["id"], i)
+                    else str(uuid5(UUID(checkpoint["id"]), INTERRUPT))
                 )
                 run_tasks.append(task)
                 run_task_ids.append(task_id)
@@ -4172,16 +4172,6 @@ class Pregel(
         await self.cache.aclear(namespaces)
 
 
-def _update_task_id(checkpoint_id: str, i: int) -> str:
-    """Task id for the `i`th update of a superstep that has no task to reuse.
-
-    Savers keep one write per `(task_id, idx)`, so updates sharing an id lose
-    all but the first one's writes, which a `DeltaChannel` replays from. The
-    first update keeps the id a lone update has always had.
-    """
-    return str(uuid5(UUID(checkpoint_id), INTERRUPT if i == 0 else f"{INTERRUPT}:{i}"))
-
-
 def _trigger_to_nodes(nodes: dict[str, PregelNode]) -> Mapping[str, Sequence[str]]:
     """Index from a trigger to nodes that depend on it."""
     trigger_to_nodes: defaultdict[str, list[str]] = defaultdict(list)

libs/langgraph/pyproject.toml

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

libs/langgraph/tests/test_delta_channel_update_state.py

@@ -20,7 +20,6 @@ from typing import Annotated, Any
 
 import pytest
 from langchain_core.messages import HumanMessage
-from langgraph.checkpoint.base import BaseCheckpointSaver
 from langgraph.checkpoint.memory import InMemorySaver
 from langgraph.checkpoint.serde.types import _DeltaSnapshot
 from typing_extensions import TypedDict
@@ -28,17 +27,16 @@ from typing_extensions import TypedDict
 from langgraph.channels.delta import DeltaChannel
 from langgraph.graph import START, StateGraph
 from langgraph.graph.message import _messages_delta_reducer
-from langgraph.types import StateSnapshot, StateUpdate
+from langgraph.types import StateUpdate
 
 pytestmark = pytest.mark.anyio
 
 
 def _build_graph(
-    checkpointer: BaseCheckpointSaver,
+    checkpointer: InMemorySaver,
     *,
     two_nodes: bool = False,
     snapshot_frequency: int = 1000,
-    interrupt_before: list[str] | None = None,
 ) -> Any:
     """Compile a minimal DeltaChannel-backed `messages` graph.
 
@@ -65,7 +63,7 @@ def _build_graph(
         builder.set_finish_point("assistant")
     else:
         builder.set_finish_point("model")
-    return builder.compile(checkpointer=checkpointer, interrupt_before=interrupt_before)
+    return builder.compile(checkpointer=checkpointer)
 
 
 # ---------------------------------------------------------------------------
@@ -275,6 +273,10 @@ def test_bulk_update_state_multi_task_per_superstep_delta_channel() -> None:
     that each call `put_writes`. Guards the regression where moving
     `put_writes` outside the per-task loop would persist only the last
     task's writes.
+
+    Explicit `task_id`s are required to disambiguate writes belonging to
+    different `StateUpdate`s targeting the same node — otherwise both share
+    the deterministic interrupt-derived id and collide in the saver.
     """
 
     saver = InMemorySaver()
@@ -308,92 +310,6 @@ def test_bulk_update_state_multi_task_per_superstep_delta_channel() -> None:
     assert sorted(ids) == ["m1", "m2"]
 
 
-def _update(content: str, as_node: str) -> StateUpdate:
-    return StateUpdate(
-        values={"messages": [HumanMessage(content=content, id=content)]},
-        as_node=as_node,
-    )
-
-
-def _contents(state: StateSnapshot) -> list[str]:
-    return [m.content for m in state.values["messages"]]
-
-
-def test_bulk_update_state_keeps_every_update_without_task_ids(
-    sync_checkpointer: BaseCheckpointSaver,
-) -> None:
-    graph = _build_graph(sync_checkpointer, two_nodes=True)
-    config = {"configurable": {"thread_id": "bulk-no-task-ids"}}
-    graph.invoke({"messages": [HumanMessage(content="hi", id="hi")]}, config)
-
-    graph.bulk_update_state(
-        config,
-        [
-            [
-                _update("first", "model"),
-                _update("second", "model"),
-                _update("third", "assistant"),
-            ]
-        ],
-    )
-
-    contents = _contents(graph.get_state(config))
-    assert sorted(contents) == ["first", "hi", "second", "third"], (
-        f"every update's writes must persist; got {contents}"
-    )
-
-
-async def test_abulk_update_state_keeps_every_update_without_task_ids(
-    async_checkpointer: BaseCheckpointSaver,
-) -> None:
-    graph = _build_graph(async_checkpointer, two_nodes=True)
-    config = {"configurable": {"thread_id": "bulk-no-task-ids"}}
-    await graph.ainvoke({"messages": [HumanMessage(content="hi", id="hi")]}, config)
-
-    await graph.abulk_update_state(
-        config,
-        [
-            [
-                _update("first", "model"),
-                _update("second", "model"),
-                _update("third", "assistant"),
-            ]
-        ],
-    )
-
-    contents = _contents(await graph.aget_state(config))
-    assert sorted(contents) == ["first", "hi", "second", "third"], (
-        f"every update's writes must persist; got {contents}"
-    )
-
-
-def test_bulk_update_state_keeps_every_update_next_to_a_pending_task(
-    sync_checkpointer: BaseCheckpointSaver,
-) -> None:
-    graph = _build_graph(
-        sync_checkpointer, two_nodes=True, interrupt_before=["assistant"]
-    )
-    config = {"configurable": {"thread_id": "bulk-pending-task"}}
-    graph.invoke({"messages": [HumanMessage(content="hi", id="hi")]}, config)
-    assert graph.get_state(config).next == ("assistant",)
-
-    graph.bulk_update_state(
-        config,
-        [
-            [
-                _update("first", "assistant"),
-                _update("second", "model"),
-                _update("third", "model"),
-            ]
-        ],
-    )
-
-    contents = _contents(graph.get_state(config))
-    assert sorted(contents) == ["first", "hi", "second", "third"], (
-        f"every update's writes must persist; got {contents}"
-    )
-
-
 # ---------------------------------------------------------------------------
 # Public-API observation of fresh-thread checkpoint shape
 # ---------------------------------------------------------------------------

libs/langgraph/uv.lock

@@ -1437,7 +1437,7 @@ wheels = [
 
 [[package]]
 name = "langgraph"
-version = "1.2.12"
+version = "1.2.11"
 source = { editable = "." }
 dependencies = [
     { name = "langchain-core" },

libs/prebuilt/uv.lock

@@ -286,7 +286,7 @@ wheels = [
 
 [[package]]
 name = "langgraph"
-version = "1.2.12"
+version = "1.2.11"
 source = { editable = "../langgraph" }
 dependencies = [
     { name = "langchain-core" },

libs/sdk-py/langgraph_sdk/__init__.py

@@ -3,7 +3,7 @@ from langgraph_sdk.client import get_client, get_sync_client
 from langgraph_sdk.encryption import Encryption
 from langgraph_sdk.encryption.types import DecryptResult, EncryptionContext
 
-__version__ = "0.4.5"
+__version__ = "0.4.4"
 
 __all__ = [
     "Auth",

libs/sdk-py/langgraph_sdk/_async/stream.py

@@ -173,8 +173,18 @@ class RunModule:
         config: dict[str, Any] | None = None,
         metadata: dict[str, Any] | None = None,
         langsmith_tracing: LangSmithTracing | None = None,
+        context: dict[str, Any] | None = None,
     ) -> dict[str, Any]:
-        """Send `run.start` to the server. Returns the result (`{"run_id": ...}`)."""
+        """Send `run.start` to the server. Returns the result (`{"run_id": ...}`).
+
+        Args:
+            input: the run input; omitted from the wire payload when None.
+            config: the run config; omitted when None.
+            metadata: run metadata; omitted when None.
+            langsmith_tracing: tracing options; omitted when None.
+            context: per-run static context; omitted from the wire payload
+                when None (server applies its default context behavior).
+        """
         params: dict[str, Any] = {"assistant_id": self._owner.assistant_id}
         if input is not None:
             params["input"] = input
@@ -184,6 +194,8 @@ class RunModule:
             params["metadata"] = metadata
         if langsmith_tracing is not None:
             params["langsmith_tracer"] = langsmith_tracing
+        if context is not None:
+            params["context"] = context
         loop = asyncio.get_running_loop()
         gate: asyncio.Future[None] = loop.create_future()
         self._owner._run_start_ready = gate
@@ -216,6 +228,7 @@ class RunModule:
         response: Any,
         *,
         interrupt_id: str | None = None,
+        context: dict[str, Any] | None = None,
     ) -> dict[str, Any]:
         """Reply to a server-side interrupt and resume the run.
 
@@ -224,6 +237,8 @@ class RunModule:
                 wire (protocol field name).
             interrupt_id: optional explicit id. When omitted, requires exactly
                 one outstanding interrupt and uses its id.
+            context: optional per-run static context for the resumed run;
+                forwarded with the `input.respond` command when non-None.
 
         Raises:
             RuntimeError: no outstanding interrupts; `interrupt_id` is None but
@@ -266,6 +281,8 @@ class RunModule:
                 "namespace": match["namespace"],
                 "response": response,
             }
+            if context is not None:
+                params["context"] = context
             return await self._owner._send_command("input.respond", params)
 
 

libs/sdk-py/langgraph_sdk/_sync/stream.py

@@ -216,8 +216,18 @@ class SyncRunModule:
         config: dict[str, Any] | None = None,
         metadata: dict[str, Any] | None = None,
         langsmith_tracing: LangSmithTracing | None = None,
+        context: dict[str, Any] | None = None,
     ) -> dict[str, Any]:
-        """Send `run.start` to the server. Returns the result (`{"run_id": ...}`)."""
+        """Send `run.start` to the server. Returns the result (`{"run_id": ...}`).
+
+        Args:
+            input: the run input; omitted from the wire payload when None.
+            config: the run config; omitted when None.
+            metadata: run metadata; omitted when None.
+            langsmith_tracing: tracing options; omitted when None.
+            context: per-run static context; omitted from the wire payload
+                when None (server applies its default context behavior).
+        """
         params: dict[str, Any] = {"assistant_id": self._owner.assistant_id}
         if input is not None:
             params["input"] = input
@@ -227,6 +237,8 @@ class SyncRunModule:
             params["metadata"] = metadata
         if langsmith_tracing is not None:
             params["langsmith_tracer"] = langsmith_tracing
+        if context is not None:
+            params["context"] = context
         result = self._owner._send_command("run.start", params)
         self._owner._run_seen = True
         controller = self._owner._controller
@@ -239,6 +251,7 @@ class SyncRunModule:
         response: Any,
         *,
         interrupt_id: str | None = None,
+        context: dict[str, Any] | None = None,
     ) -> dict[str, Any]:
         """Reply to a server-side interrupt and resume the run.
 
@@ -246,6 +259,8 @@ class SyncRunModule:
             response: the response value forwarded as `params.response` on the wire.
             interrupt_id: optional explicit id. When omitted, requires exactly one
                 outstanding interrupt.
+            context: optional per-run static context for the resumed run;
+                forwarded with the `input.respond` command when non-None.
 
         Raises:
             RuntimeError: no outstanding interrupts; `interrupt_id` is None but
@@ -282,6 +297,8 @@ class SyncRunModule:
             "namespace": match["namespace"],
             "response": response,
         }
+        if context is not None:
+            params["context"] = context
         return self._owner._send_command("input.respond", params)
 
 

libs/sdk-py/tests/streaming/test_sync_extensions_projection.py

@@ -29,10 +29,7 @@ def test_sync_extension_projection_yields_matching_custom_payloads():
         {"name": "progress", "step": 1},
         {"name": "progress", "step": 2},
     ]
-    assert any(
-        "custom:progress" in body.get("channels", [])
-        for body in fake.stream_request_bodies
-    )
+    assert "custom:progress" in fake.stream_request_bodies[-1]["channels"]
 
 
 def test_sync_extension_projection_supports_namespace_scope_on_subgraph_handle():

libs/sdk-py/tests/streaming/test_sync_thread_stream.py

@@ -439,6 +439,64 @@ def test_sync_run_start_sends_command():
     }
 
 
+def test_sync_run_start_forwards_context():
+    fake = SyncFakeServer()
+    fake.script([lifecycle_completed_event(seq=1)])
+    with httpx.Client(transport=fake.transport, base_url="http://test") as raw:
+        threads = SyncThreadsClient(SyncHttpClient(raw))
+        with threads.stream(thread_id="t-1", assistant_id="agent") as thread:
+            thread.run.start(input={"x": 1}, context={"user_id": "u-1"})
+
+    assert fake.received_commands[0]["params"]["context"] == {"user_id": "u-1"}
+
+
+def test_sync_run_start_omits_context_when_not_provided():
+    fake = SyncFakeServer()
+    fake.script([lifecycle_completed_event(seq=1)])
+    with httpx.Client(transport=fake.transport, base_url="http://test") as raw:
+        threads = SyncThreadsClient(SyncHttpClient(raw))
+        with threads.stream(thread_id="t-1", assistant_id="agent") as thread:
+            thread.run.start(input={"x": 1})
+
+    assert "context" not in fake.received_commands[0]["params"]
+
+
+def test_sync_run_respond_forwards_context():
+    fake = SyncFakeServer()
+    fake.script([lifecycle_completed_event(seq=1)])
+    with httpx.Client(transport=fake.transport, base_url="http://test") as raw:
+        threads = SyncThreadsClient(SyncHttpClient(raw))
+        with threads.stream(thread_id="t-1", assistant_id="agent") as thread:
+            thread.run.start(input={})
+            thread.interrupts.append(
+                {"interrupt_id": "i-1", "value": None, "namespace": []}
+            )
+            thread.interrupted = True
+            thread.run.respond("yes", context={"user_id": "u-1"})
+
+    command = fake.received_commands[-1]
+    assert command["method"] == "input.respond"
+    assert command["params"]["context"] == {"user_id": "u-1"}
+
+
+def test_sync_run_respond_omits_context_when_not_provided():
+    fake = SyncFakeServer()
+    fake.script([lifecycle_completed_event(seq=1)])
+    with httpx.Client(transport=fake.transport, base_url="http://test") as raw:
+        threads = SyncThreadsClient(SyncHttpClient(raw))
+        with threads.stream(thread_id="t-1", assistant_id="agent") as thread:
+            thread.run.start(input={})
+            thread.interrupts.append(
+                {"interrupt_id": "i-1", "value": None, "namespace": []}
+            )
+            thread.interrupted = True
+            thread.run.respond("yes")
+
+    command = fake.received_commands[-1]
+    assert command["method"] == "input.respond"
+    assert "context" not in command["params"]
+
+
 def test_sync_events_iterates_raw_events():
 
     fake = SyncFakeServer()

libs/sdk-py/tests/streaming/test_thread_stream.py

@@ -311,6 +311,28 @@ async def test_run_start_forwards_config_metadata_and_langsmith_tracing():
     }
 
 
+async def test_run_start_forwards_context():
+    fake = FakeServer()
+    transport = httpx.ASGITransport(app=fake.app)
+    async with httpx.AsyncClient(transport=transport, base_url="http://test") as raw:
+        threads = ThreadsClient(HttpClient(raw))
+        async with threads.stream(thread_id="t-1", assistant_id="agent") as thread:
+            await thread.run.start(input={"x": 1}, context={"user_id": "u-1"})
+    params = fake.received_commands[0]["params"]
+    assert params["context"] == {"user_id": "u-1"}
+
+
+async def test_run_start_omits_context_when_not_provided():
+    fake = FakeServer()
+    transport = httpx.ASGITransport(app=fake.app)
+    async with httpx.AsyncClient(transport=transport, base_url="http://test") as raw:
+        threads = ThreadsClient(HttpClient(raw))
+        async with threads.stream(thread_id="t-1", assistant_id="agent") as thread:
+            await thread.run.start(input={"x": 1})
+    params = fake.received_commands[0]["params"]
+    assert "context" not in params
+
+
 async def test_run_start_raises_outside_context_manager():
 
     async with httpx.AsyncClient(base_url="http://test") as raw:
@@ -616,6 +638,38 @@ async def test_run_respond_dispatches_input_respond_command():
     assert command["params"]["namespace"] == []
 
 
+async def test_run_respond_forwards_context():
+    fake = FakeServer()
+    asgi = httpx.ASGITransport(app=fake.app)
+    async with httpx.AsyncClient(transport=asgi, base_url="http://test") as raw:
+        threads = ThreadsClient(HttpClient(raw))
+        async with threads.stream(thread_id="t-1", assistant_id="agent") as thread:
+            await thread.run.start(input={})
+            thread.interrupts.append(
+                {"interrupt_id": "i-1", "value": None, "namespace": []}
+            )
+            thread.interrupted = True
+            await thread.run.respond("yes", context={"user_id": "u-1"})
+    params = fake.received_commands[-1]["params"]
+    assert params["context"] == {"user_id": "u-1"}
+
+
+async def test_run_respond_omits_context_when_not_provided():
+    fake = FakeServer()
+    asgi = httpx.ASGITransport(app=fake.app)
+    async with httpx.AsyncClient(transport=asgi, base_url="http://test") as raw:
+        threads = ThreadsClient(HttpClient(raw))
+        async with threads.stream(thread_id="t-1", assistant_id="agent") as thread:
+            await thread.run.start(input={})
+            thread.interrupts.append(
+                {"interrupt_id": "i-1", "value": None, "namespace": []}
+            )
+            thread.interrupted = True
+            await thread.run.respond("yes")
+    params = fake.received_commands[-1]["params"]
+    assert "context" not in params
+
+
 async def test_run_respond_with_explicit_interrupt_id():
     fake = FakeServer()
     asgi = httpx.ASGITransport(app=fake.app)

libs/sdk-py/uv.lock

@@ -17,16 +17,16 @@ wheels = [
 
 [[package]]
 name = "anyio"
-version = "4.15.1"
+version = "4.12.1"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "exceptiongroup", marker = "python_full_version < '3.11'" },
     { name = "idna" },
-    { name = "typing-extensions", version = "4.16.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.15'" },
+    { name = "typing-extensions", marker = "python_full_version < '3.13'" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/a9/d2/f4d173e22df740bc37b1db102b386ba719b66e95b0f0d751f556b387e6d2/anyio-4.15.1.tar.gz", hash = "sha256:9f28306018cbd6d329e64a36d58256edff76dd996fe423bc957326e578b82a94", size = 276966, upload-time = "2026-09-05T10:42:39.44Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/96/f0/5eb65b2bb0d09ac6776f2eb54adee6abe8228ea05b20a5ad0e4945de8aac/anyio-4.12.1.tar.gz", hash = "sha256:41cfcc3a4c85d3f05c932da7c26d0201ac36f72abd4435ba90d0464a3ffed703", size = 228685, upload-time = "2026-01-06T11:45:21.246Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/12/b8/4bd346e22b28902df4d651910f5242c28d84e4a5c2435ca5c3f797ed7e2e/anyio-4.15.1-py3-none-any.whl", hash = "sha256:6152fdbbf9a77fdec97731721bebf7c4c44f7c29b424b0065826173efc7ed101", size = 132079, upload-time = "2026-09-05T10:42:37.923Z" },
+    { url = "https://files.pythonhosted.org/packages/38/0e/27be9fdef66e72d64c0cdc3cc2823101b80585f8119b5c112c2e8f5f7dab/anyio-4.12.1-py3-none-any.whl", hash = "sha256:d405828884fc140aa80a3c667b8beed277f1dfedec42ba031bd6ac3db606ab6c", size = 113592, upload-time = "2026-01-06T11:45:19.497Z" },
 ]
 
 [[package]]
@@ -181,7 +181,7 @@ name = "exceptiongroup"
 version = "1.3.1"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
-    { name = "typing-extensions", version = "4.16.0", source = { registry = "https://pypi.org/simple" } },
+    { name = "typing-extensions" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/50/79/66800aadf48771f6b62f7eb014e352e5d06856655206165d775e675a02c9/exceptiongroup-1.3.1.tar.gz", hash = "sha256:8b412432c6055b0b7d14c310000ae93352ed6754f70fa8f7c34141f91c4e3219", size = 30371, upload-time = "2025-11-21T23:01:54.787Z" }
 wheels = [
@@ -277,8 +277,7 @@ dependencies = [
     { name = "pydantic" },
     { name = "pyyaml" },
     { name = "tenacity" },
-    { name = "typing-extensions", version = "4.15.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.15'" },
-    { name = "typing-extensions", version = "4.16.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.15'" },
+    { name = "typing-extensions" },
     { name = "uuid-utils" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/90/12/aff76ca89c219ebe6f9dd3c5dbc4e3b1cf5450e9fc7037dccad23d45cd7a/langchain_core-1.6.1.tar.gz", hash = "sha256:1b156cb395aac4f009a8a1b38a574c7d948fe2d5f74c96e0d8a5017b4149e04f", size = 1003359, upload-time = "2026-08-27T19:31:14.956Z" }
@@ -291,8 +290,7 @@ name = "langchain-protocol"
 version = "0.0.19"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
-    { name = "typing-extensions", version = "4.15.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.15'" },
-    { name = "typing-extensions", version = "4.16.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.15'" },
+    { name = "typing-extensions" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/14/56/913599f2f9cec8524868929f12d72b2ede377a6056ca8a40a32bdadfa535/langchain_protocol-0.0.19.tar.gz", hash = "sha256:79d90a1425122ac87e8052e2ec054fbd09c3edbf341bdfb6397112a495c7bf8c", size = 6265, upload-time = "2026-08-26T21:12:00.703Z" }
 wheels = [
@@ -301,7 +299,7 @@ wheels = [
 
 [[package]]
 name = "langgraph"
-version = "1.2.12"
+version = "1.2.11"
 source = { editable = "../langgraph" }
 dependencies = [
     { name = "langchain-core" },
@@ -728,8 +726,7 @@ source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "annotated-types" },
     { name = "pydantic-core" },
-    { name = "typing-extensions", version = "4.15.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.15'" },
-    { name = "typing-extensions", version = "4.16.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.15'" },
+    { name = "typing-extensions" },
     { name = "typing-inspection" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/53/ef/fc4f868f4e2cee79f863883abffceff107875f569b848507319842d2a681/pydantic-2.13.5.tar.gz", hash = "sha256:51a9c5f7b2f8e636f04c6cada605d9b6a3bf1348fdf945a3d8869b19bba0ee08", size = 845750, upload-time = "2026-08-28T14:04:00.916Z" }
@@ -742,8 +739,7 @@ name = "pydantic-core"
 version = "2.46.5"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
-    { name = "typing-extensions", version = "4.15.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.15'" },
-    { name = "typing-extensions", version = "4.16.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.15'" },
+    { name = "typing-extensions" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/af/f9/8a06bea35ef8daf588f707784c973a7046e0034c8d8cfb08828eeffb8b75/pydantic_core-2.46.5.tar.gz", hash = "sha256:10416c15b8839ecc4ef4d0885da76da6fd0f67333a0eb8aff6d93c4b8f2910fc", size = 472262, upload-time = "2026-08-28T10:01:31.677Z" }
 wheels = [
@@ -888,7 +884,7 @@ source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "backports-asyncio-runner", marker = "python_full_version < '3.11'" },
     { name = "pytest" },
-    { name = "typing-extensions", version = "4.16.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.13'" },
+    { name = "typing-extensions", marker = "python_full_version < '3.13'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/43/7c/d36d04db312ecf4298932ef77e6e4a9e8ad017906e24e34f0b0c361a2473/pytest_asyncio-1.4.0.tar.gz", hash = "sha256:c6c0d2259945122819f171a32ecea2c349ead889ee28176caaf492143424be42", size = 58514, upload-time = "2026-05-26T09:56:04.083Z" }
 wheels = [
@@ -1041,7 +1037,7 @@ version = "1.6.0"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "anyio" },
-    { name = "typing-extensions", version = "4.16.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.13'" },
+    { name = "typing-extensions", marker = "python_full_version < '3.13'" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/b5/b4/205b0d5241d934e8add0c38aa924c4f9fb7330834ff11e5444db964ec3f9/starlette-1.6.0.tar.gz", hash = "sha256:d4e3ac5e546444960c710297a3c9fc3f7ebae1b7e963f3d36173b49da535be9b", size = 2716969, upload-time = "2026-08-08T18:27:57.512Z" }
 wheels = [
@@ -1140,33 +1136,17 @@ wheels = [
 name = "typing-extensions"
 version = "4.15.0"
 source = { registry = "https://pypi.org/simple" }
-resolution-markers = [
-    "python_full_version >= '3.15'",
-]
 sdist = { url = "https://files.pythonhosted.org/packages/72/94/1a15dd82efb362ac84269196e94cf00f187f7ed21c242792a923cdb1c61f/typing_extensions-4.15.0.tar.gz", hash = "sha256:0cea48d173cc12fa28ecabc3b837ea3cf6f38c6d1136f85cbaaf598984861466", size = 109391, upload-time = "2025-08-25T13:49:26.313Z" }
 wheels = [
     { url = "https://files.pythonhosted.org/packages/18/67/36e9267722cc04a6b9f15c7f3441c2363321a3ea07da7ae0c0707beb2a9c/typing_extensions-4.15.0-py3-none-any.whl", hash = "sha256:f0fa19c6845758ab08074a0cfa8b7aecb71c999ca73d62883bc25cc018c4e548", size = 44614, upload-time = "2025-08-25T13:49:24.86Z" },
 ]
 
-[[package]]
-name = "typing-extensions"
-version = "4.16.0"
-source = { registry = "https://pypi.org/simple" }
-resolution-markers = [
-    "python_full_version < '3.15'",
-]
-sdist = { url = "https://files.pythonhosted.org/packages/f6/cc/6253133b5bb138fc3306cebfbda2c520f545d36b5be2c7255cc528bb45d6/typing_extensions-4.16.0.tar.gz", hash = "sha256:dc983d19a509c94dba722ee6abd33940f7c05a89e243c47e907eb4db6f1a43e5", size = 113555, upload-time = "2026-07-02T08:40:05.92Z" }
-wheels = [
-    { url = "https://files.pythonhosted.org/packages/49/d3/b8441a820a491ddfc024b0b0cf0393375b75ea13866d9c66727e54c2fc80/typing_extensions-4.16.0-py3-none-any.whl", hash = "sha256:481caa481374e813c1b176ada14e97f1f67a4539ce9cfeb3f350d78d6370c2e8", size = 45571, upload-time = "2026-07-02T08:40:04.659Z" },
-]
-
 [[package]]
 name = "typing-inspection"
 version = "0.4.2"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
-    { name = "typing-extensions", version = "4.15.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version >= '3.15'" },
-    { name = "typing-extensions", version = "4.16.0", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.15'" },
+    { name = "typing-extensions" },
 ]
 sdist = { url = "https://files.pythonhosted.org/packages/55/e3/70399cb7dd41c10ac53367ae42139cf4b1ca5f36bb3dc6c9d33acdb43655/typing_inspection-0.4.2.tar.gz", hash = "sha256:ba561c48a67c5958007083d386c3295464928b01faa735ab8547c5692e87f464", size = 75949, upload-time = "2025-10-01T02:14:41.687Z" }
 wheels = [