add missing build step

bump prebuilt.md
fix cassette hashing
2026-10-09 18:05:10 +02:00 · 2025-02-13 23:03:58 -08:00 · 2025-02-13 22:57:15 -08:00 · 2025-02-13 22:56:16 -08:00 · 2025-02-13 21:57:45 -08:00 · 2025-02-13 21:57:44 -08:00
47 changed files with 6998 additions and 779 deletions
@@ -56,6 +56,12 @@ jobs:
        with:
          fetch-depth: 0

+      - uses: actions/checkout@v4
+        with:
+          repository: langchain-ai/langchainjs
+          token: ${{ secrets.MKDOCS_GITHUB_TOKEN }}
+          path: docs/langchainjs
+
      - name: Set up Python + Poetry ${{ env.POETRY_VERSION }}
        uses: "./.github/actions/poetry_setup"
        with:
@@ -72,6 +78,7 @@ jobs:

      - name: Install dependencies
        run: |
+          cd langchainjs && yarn && yarn build && cd ..
          yarn
          poetry install --with test --with docs --no-root
          poetry run pip install -U \
@@ -0,0 +1,3 @@
+nodeLinker: node-modules
+
+yarnPath: .yarn/releases/yarn-3.5.1.cjs
@@ -13,7 +13,22 @@ build-prebuilt:
 	poetry run python -m _scripts.third_party_page.get_download_stats stats.yml
 	poetry run python -m _scripts.third_party_page.create_third_party_page stats.yml docs/prebuilt.md --language python

-build-docs: build-typedoc build-prebuilt
+grab-langgraphjs:
+	if [ -d "langgraphjs" ]; then \
+		if [ ! -d "langgraphjs/.git" ]; then \
+			rm -rf langgraphjs; \
+		fi \
+	fi
+	if [ ! -d "langgraphjs" ]; then \
+		git clone https://github.com/langchain-ai/langgraphjs.git; \
+	else \
+		cd langgraphjs && git checkout main && git pull; \
+	fi
+	cd langgraphjs && yarn
+	cd langgraphjs && yarn build
+	yarn
+
+build-docs: build-typedoc build-prebuilt grab-langgraphjs
 	poetry run python -m mkdocs build --clean -f mkdocs.yml --strict

 llms-text:
@@ -32,7 +47,6 @@ install-vercel-deps:
 	poetry run tslab install --python=python3
 	poetry run jupyter kernelspec list

-
 tests:
 	# RUn unit tests
 	poetry run pytest tests/unit_tests
@@ -45,7 +59,7 @@ vercel-build-docs: install-vercel-deps
 serve-clean-docs: clean-docs
 	poetry run python -m mkdocs serve -c -f mkdocs.yml --strict -w ../libs/langgraph

-serve-docs: build-typedoc
+serve-docs: build-typedoc grab-langgraphjs
 	poetry run python -m mkdocs serve -f mkdocs.yml -w ../libs/langgraph  -w ../libs/checkpoint -w ../libs/sdk-py --dirty

 clean-docs:
@@ -24,14 +24,16 @@ function decompressData(compressedString: string): NockCassetteData {
  return msgpack.decode(decompressed) as NockCassetteData;
 }

-// deno-lint-ignore no-unused-vars
 class HashedCassette {
+  hash: string;
  private recording = true;

  constructor(
    private readonly cassettePath: string,
-    private readonly hash: string
-  ) {}
+    hash: string
+  ) {
+    this.hash = hash;
+  }

  async enter() {
    try {
@@ -63,7 +63,7 @@ class HashedCassette:
                content = f.read()
            try:
                cassette_data = serializer.deserialize(content)
-            except Exception as e:
+            except Exception:
                os.remove(self.cassette_path)
            else:
                existing_hash = cassette_data.get("cassette_hash")
@@ -96,7 +96,7 @@ class HashedCassette:
                content = f.read()
            try:
                cassette_data = serializer.deserialize(content)
-            except Exception as e:
+            except Exception:
                return result
            # Update the cassette data with the expected hash.
            if cassette_data.get("cassette_hash") != self.hash_value:
@@ -0,0 +1,4 @@
+hook_state = {
+    "document_filename": "__UNKNOWN__",
+    "document_content": "__UNKNOWN__",
+}
@@ -6,14 +6,17 @@ import traceback
 from typing import Any, Callable, Dict

 from markdown import Markdown
-from markdown_exec.hooks import SessionHistoryEntry
 from mkdocs.structure.files import Files, File
 from mkdocs.structure.pages import Page
 from pymdownx.superfences import SuperFencesException

+from _scripts.hook_state import hook_state
+from markdown_exec.hooks import SessionHistoryEntry
+
+
 from _scripts.generate_api_reference_links import update_markdown_with_imports
 from _scripts.notebook_convert import convert_notebook
-from _scripts.setup_vcr import load_postamble, load_preamble, _hash_string
+from _scripts.setup_vcr import get_hash_for_session, load_postamble, load_preamble, _hash_string

 logger = logging.getLogger(__name__)
 logging.basicConfig()
@@ -62,29 +65,6 @@ def on_files(files: Files, **kwargs: Dict[str, Any]):
    return new_files


-def _add_path_to_code_blocks(markdown: str, page: Page) -> str:
-    """Add the path to the code blocks."""
-    code_block_pattern = re.compile(
-        r"(?P<indent>[ \t]*)```(?P<language>\w+)[ ]*(?P<attributes>[^\n]*)\n"
-        r"(?P<code>((?:.*\n)*?))"  # Capture the code inside the block using named group
-        r"(?P=indent)```"  # Match closing backticks with the same indentation
-    )
-
-    def replace_code_block_header(match: re.Match) -> str:
-        indent = match.group("indent")
-        language = match.group("language")
-        attributes = match.group("attributes").rstrip()
-
-        if 'exec="on"' not in attributes:
-            # Return original code block
-            return match.group(0)
-
-        code = match.group("code")
-        return f'{indent}```{language} {attributes} path="{page.file.src_path}"\n{code}{indent}```'
-
-    return code_block_pattern.sub(replace_code_block_header, markdown)
-
-
 def _highlight_code_blocks(markdown: str) -> str:
    """Find code blocks with highlight comments and add hl_lines attribute.

@@ -162,7 +142,6 @@ def _highlight_code_blocks(markdown: str) -> str:
    markdown = code_block_pattern.sub(replace_highlight_comments, markdown)
    return markdown

-
 def handle_vcr_setup(
    *,
    formatter: Callable,
@@ -174,26 +153,25 @@ def handle_vcr_setup(
    **kwargs: Dict[str, Any],
 ) -> Dict[str, Any]:
    """Handle VCR setup in markdown content if necessary."""
+    logger.info(f"handle_vcr_setup: {hook_state['document_filename']}")
    try:
-        if kwargs.get("extra", None) is None:
+        if hook_state['document_filename'] == '__UNKNOWN__':
            raise SuperFencesException(
-                f"error while processing {language} block: extra dict is required"
+                f"error while processing {language} block: document filename hasn't been set yet"
            )
-
-        if kwargs["extra"].get("path", None) is None:
+        
+        if hook_state['document_content'] == '__UNKNOWN__':
            raise SuperFencesException(
-                f"error while processing {language} block: path is required"
+                f"error while processing {language} block: document content hasn't been set yet"
            )

-        document_filename = kwargs["extra"]["path"]
-
        if session is None or session == "" and id is None or id == "":
            id = _hash_string(code)

        if session is not None and session != "":
-            logger.info(f"new {language} session {session} on page {document_filename}")
+            logger.info(f"new {language} session {session} on page {hook_state['document_filename']}")

-        cassette_prefix = document_filename.replace(".md", "").replace(os.path.sep, "_")
+        cassette_prefix = hook_state['document_filename'].replace(".md", "").replace(os.path.sep, "_")

        cassette_dir = os.path.abspath(
            os.path.join(os.path.dirname(os.path.dirname(__file__)), "cassettes")
@@ -208,14 +186,15 @@ def handle_vcr_setup(

        # Add context manager at start with explicit __enter__ and __exit__ calls

+        hash_ = get_hash_for_session(language, session, hook_state['document_content'])
        wrapped_lines = [
-            load_preamble(language, code, cassette_name),
+            load_preamble(language, hash_, cassette_name),
            code,
        ]

        if session is None or session == "":
            logger.info(
-                f"no session, adding postamble for {language} in {document_filename}"
+                f"no session, adding postamble for {language} in {hook_state['document_filename']}"
            )
            wrapped_lines.append(load_postamble(language))

@@ -234,7 +213,7 @@ def handle_vcr_setup(
        return dict(
            transform_source=lambda code: (transformed_source, code),
            id=id,
-            extra=keep_extras,
+            extra={ **keep_extras, "path": hook_state['document_filename'] },
        )
    except Exception as e:
        raise SuperFencesException(traceback.format_exc()) from e
@@ -247,24 +226,17 @@ def handle_vcr_teardown(
    session: str,
    history: list[SessionHistoryEntry],
 ):
-    last_inputs = dict(history[-1].inputs)
    code = load_postamble(language)
-    md = last_inputs["md"]
    html = False
    update_toc = False

-    document_filename = last_inputs.get("extra", {}).get("path", None)
-
-    if document_filename is None:
-        logger.warning(f"no document filename found while tearing down {session}!")
-    else:
-        logger.info(f"tearing down {language} {session} on {document_filename}")
+    logger.info(f"tearing down {language} {session} on {hook_state['document_filename']}")

    kwargs = dict(
        code=code,
        session=session,
        id=f"{id}_vcr_end",
-        md=md,
+        md=None, # md is unused by the formatter, but it's a required argument
        html=html,
        update_toc=update_toc,
        extra={},
@@ -296,11 +268,6 @@ def _on_page_markdown_with_config(
    # Apply highlight comments to code blocks
    markdown = _highlight_code_blocks(markdown)

-    # Add file path as an attribute to code blocks that are executable.
-    # This file path is used to associate fixtures with the executable code
-    # which can be used in CI to test the docs without making network requests.
-    markdown = _add_path_to_code_blocks(markdown, page)
-
    if remove_base64_images:
        # Remove base64 encoded images from markdown
        markdown = re.sub(r"!\[.*?\]\(data:image/+;base64,[^\)]+\)", "", markdown)
@@ -309,6 +276,8 @@ def _on_page_markdown_with_config(


 def on_page_markdown(markdown: str, page: Page, **kwargs: Dict[str, Any]):
+    logger.info(f"on_page_markdown: {page.file.src_path}")
+    hook_state['document_filename'] = page.file.src_path
    return _on_page_markdown_with_config(
        markdown,
        page,
@@ -370,3 +339,9 @@ def on_post_build(config):
            + suffix
        )
        write_html(config["site_dir"], old_html_path, new_html_path)
+
+def on_pre_page(page: Page, **kwargs: Dict[str, Any]):
+    logger.info(f"on_pre_page: {page.file.src_path}")
+    hook_state['document_filename'] = page.file.src_path
+    hook_state['document_content'] = page.file.content_string
+    return page
@@ -1,7 +1,12 @@
 # A list of patterns that, if found in a code block, will cause us to leave that block unchanged.
 import hashlib
+import json
 import os
-from textwrap import dedent
+import re
+from textwrap import dedent, indent
+
+from mistune import BlockParser, BlockState, Markdown, create_markdown
+from mistune.renderers.markdown import MarkdownRenderer

 preambles = {
    "python": "vcr_setup_preamble.py",
@@ -50,21 +55,19 @@ preamble_cleanups = {
 }


-def load_preamble(language: str, code: str, cassette_name: str) -> str:
+def load_preamble(language: str, hash_: str, cassette_name: str) -> str:
    """Load the source code for the preamble for a given language."""
    _assets_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "assets")

    preamble_path = os.path.join(_assets_dir, preambles[language])
    with open(preamble_path, "r") as f:
        lines = f.readlines()
-        hash_ = _hash_string(code)
        lines.append(preamble_inits[language](cassette_name, hash_))
        return "\n".join(lines).strip()


 def load_postamble(language: str) -> str:
    """Load the source code for the postamble for a given language."""
-
    return preamble_cleanups[language]()


@@ -75,3 +78,63 @@ def _hash_string(input_string: str) -> str:
    sha256_hash = hashlib.sha256(encoded_string)
    # Get the hexadecimal digest of the hash
    return sha256_hash.hexdigest()
+
+
+def extract_code_blocks_for_session(language: str, session: str, content: str) -> str:
+    code_blocks_for_session = []
+
+    TAB_REGEX = r"^===!? \"(?P<title>[^\"]+)\"\n(?P<content>(?:(?P<indent>    )+[^\n]*\n)+)"
+    def parse_tabs(block: BlockParser, m: re.Match, state: BlockState) -> str:
+        state.append_token(
+            {
+                "raw": m.group(0),
+                "type": "block_tab",
+                "attrs": {
+                    "title": m.group("title"),
+                    "level": len(m.group("indent")) // 4,
+                    "content": dedent(m.group("content")).strip(),
+                },
+            }
+        )
+        return m.end()
+
+    def render_tabs(self, token: dict, state: BlockState):
+        recursive_transformer = create_markdown(renderer=DocumentRenderer())
+        recursive_transformer.block.register("block_tab", TAB_REGEX, parse_tabs, before='list')
+        recursive_transformer.renderer.register("block_tab", render_tabs)
+        return (
+            f'=== "{token["attrs"]["title"]}"\n'
+            f'{indent(recursive_transformer(token["attrs"]["content"]), "    " * token["attrs"]["level"])}\n'
+        )
+
+    class DocumentRenderer(MarkdownRenderer):
+        def block_code(self, token: dict, state: BlockState):
+            if token["style"] == "fenced":
+                if token["attrs"]["info"]:
+                    attributes = {}
+                    block_language = token["attrs"]["info"].split()[0]
+                    for match in re.finditer(r'(?P<key>\w+)=(?:(?P<value>(?:[\w]+))|"(?P<value_quoted>(?:[^"\s]+))")', token["attrs"]["info"]):
+                        attributes[match.group("key")] = match.group("value") or match.group("value_quoted")
+                    if block_language == language and "session" in attributes and attributes["session"] == session:
+                        code_blocks_for_session.append(token["raw"].rstrip())
+            return super().block_code(token, state)
+
+    transformer: Markdown = create_markdown(renderer=DocumentRenderer())
+    transformer.block.register("block_tab", TAB_REGEX, parse_tabs, before='list')
+    transformer.renderer.register("block_tab", render_tabs)
+    
+    # Parses the page content, which causes the code blocks to be added to the code_blocks_for_session list.
+    # There's probably some way to do this by using the renderer as a filter, but I would've had to NO-OP
+    # all of the default behavior, and this was easier.
+    transformer(content)
+
+    return code_blocks_for_session
+
+
+def get_hash_for_session(language: str, session: str, content: str) -> str:
+    # include the preamble in the hash so we invalidate if it changes
+    preamble_hash = _hash_string(load_preamble(language, session, "test"))
+
+    code_blocks_for_session = [preamble_hash, *extract_code_blocks_for_session(language, session, content)]
+    
+    return _hash_string("\n".join(code_blocks_for_session))
@@ -12,9 +12,9 @@ below. These libraries can extend LangGraph's functionality in various ways.
 [//]: # (This file is automatically generated using a script in docs/_scripts. Do not edit this file directly!)
 | Name | GitHub URL | Description | Weekly Downloads |
 | --- | --- | --- | --- |
-| **trustcall** | [hinthornw/trustcall](https://github.com/hinthornw/trustcall) | Tenacious tool calling built on LangGraph | 8803 |
-| **langgraph-supervisor** | [langchain-ai/langgraph-supervisor](https://github.com/langchain-ai/langgraph-supervisor) | Build supervisor multi-agent systems with LangGraph | 636 |
-| **breeze-agent** | [andrestorres123/breeze-agent](https://github.com/andrestorres123/breeze-agent) | A streamlined research system built inspired on STORM and built on LangGraph | 184 |
+| **trustcall** | [hinthornw/trustcall](https://github.com/hinthornw/trustcall) | Tenacious tool calling built on LangGraph | 11189 |
+| **langgraph-supervisor** | [langchain-ai/langgraph-supervisor](https://github.com/langchain-ai/langgraph-supervisor) | Build supervisor multi-agent systems with LangGraph | 1291 |
+| **breeze-agent** | [andrestorres123/breeze-agent](https://github.com/andrestorres123/breeze-agent) | A streamlined research system built inspired on STORM and built on LangGraph | 226 |

 ## ✨ Contributing Your Library

@@ -468,16 +468,6 @@ markdown_extensions:
 hooks:
  - _scripts/notebook_hooks.py
 extra:
-  consent:
-    title: Cookie consent
-    actions:
-      - accept
-      - reject
-    description: >- 
-      We use cookies to recognize your repeated visits and preferences, as well
-      as to measure the effectiveness of our documentation and whether users
-      find what they're searching for. <strong>Clicking "Accept" makes our
-      documentation better. Thank you!</strong> ❤️
  social:
    - icon: fontawesome/brands/js
      link: https://langchain-ai.github.io/langgraphjs/
@@ -486,9 +476,9 @@ extra:
    - icon: fontawesome/brands/twitter
      link: https://twitter.com/LangChainAI
  analytics:
-    provider: google
-    property: G-WR87FQLG9F
-    feedback:
+    - provider: google
+    - property: G-G8X6ELZYE0
+    - feedback:
      title: Was this page helpful?
      ratings:
        - icon: material/emoticon-happy-outline
@@ -519,5 +509,3 @@ validation:
  anchors: info
  # this is needed to handle headers with anchors for nav
  not_found: info
-copyright: >
-  Copyright &copy; 2025 LangChain, Inc | <a href="#__consent">Consent Preferences</a>
@@ -2,11 +2,18 @@
  "name": "docs",
  "version": "1.0.0",
  "license": "MIT",
+  "packageManager": "yarn@3.5.1",
  "scripts": {
-    "build": "echo 'export OPENAI_API_KEY=\"sk-proj-1234567890\"' >> ~/.bashrc && echo 'export ANTHROPIC_API_KEY=\"sk-ant-api03-1234567890\"' >> ~/.bashrc && echo 'export PATH=$PATH:/vercel/.local/bin:$PATH' >> ~/.bashrc && source ~/.bashrc && make vercel-build-docs"
+    "build": "make build-docs"
  },
  "dependencies": {
    "@langchain/core": "^0.3.38",
+    "@langchain/langgraph": "portal:./langgraphjs/libs/langgraph",
+    "@langchain/langgraph-checkpoint": "portal:./langgraphjs/libs/checkpoint",
+    "@langchain/langgraph-checkpoint-mongodb": "portal:./langgraphjs/libs/checkpoint-mongodb",
+    "@langchain/langgraph-checkpoint-postgres": "portal:./langgraphjs/libs/checkpoint-postgres",
+    "@langchain/langgraph-checkpoint-sqlite": "portal:./langgraphjs/libs/checkpoint-sqlite",
+    "@langchain/langgraph-checkpoint-validation": "portal:./langgraphjs/libs/checkpoint-validation",
    "@langchain/openai": "^0.4.2",
    "msgpack-lite": "^0.1.26",
    "nock": "^14.0.1"
@@ -1016,12 +1016,12 @@ 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 = ["docs", "test"]
+groups = ["main", "docs", "test"]
 files = [
    {file = "colorama-0.4.6-py2.py3-none-any.whl", hash = "sha256:4f1d9991f5acc0ca119f9d443620b77f9d6b33703e51011c16baf57afb285fc6"},
    {file = "colorama-0.4.6.tar.gz", hash = "sha256:08695f5cb7ed6e0531a20572697297273c47b8cae5a63ffc6d6ed5c201be6e44"},
 ]
-markers = {docs = "python_version <= \"3.11\" or python_version >= \"3.12\"", test = "(platform_system == \"Windows\" or sys_platform == \"win32\" or os_name == \"nt\") and (python_version <= \"3.11\" or python_version >= \"3.12\")"}
+markers = {main = "sys_platform == \"win32\" and (python_version <= \"3.11\" or python_version >= \"3.12\")", docs = "python_version <= \"3.11\" or python_version >= \"3.12\"", test = "(platform_system == \"Windows\" or sys_platform == \"win32\" or os_name == \"nt\") and (python_version <= \"3.11\" or python_version >= \"3.12\")"}

 [[package]]
 name = "coloredlogs"
@@ -1416,7 +1416,7 @@ version = "1.2.2"
 description = "Backport of PEP 654 (exception groups)"
 optional = false
 python-versions = ">=3.7"
-groups = ["docs", "test"]
+groups = ["main", "docs", "test"]
 markers = "python_version < \"3.11\""
 files = [
    {file = "exceptiongroup-1.2.2-py3-none-any.whl", hash = "sha256:3111b9d131c238bec2f8f516e123e14ba243563fb135d3fe885990585aa7795b"},
@@ -2462,6 +2462,19 @@ enabler = ["pytest-enabler (>=2.2)"]
 test = ["jaraco.test (>=5.4)", "pytest (>=6,!=8.1.*)", "zipp (>=3.17)"]
 type = ["pytest-mypy"]

+[[package]]
+name = "iniconfig"
+version = "2.0.0"
+description = "brain-dead simple config-ini parsing"
+optional = false
+python-versions = ">=3.7"
+groups = ["main"]
+markers = "python_version <= \"3.11\" or python_version >= \"3.12\""
+files = [
+    {file = "iniconfig-2.0.0-py3-none-any.whl", hash = "sha256:b6a85871a79d2e3b22d2d1b94ac2824226a63c6b741c88f7ae975f18b6778374"},
+    {file = "iniconfig-2.0.0.tar.gz", hash = "sha256:2d91e135bf72d31a410b17c16da610a82cb55f6b0477d1a902134b24a455b8b3"},
+]
+
 [[package]]
 name = "ipykernel"
 version = "6.29.5"
@@ -5234,7 +5247,7 @@ version = "24.2"
 description = "Core utilities for Python packages"
 optional = false
 python-versions = ">=3.8"
-groups = ["docs", "test"]
+groups = ["main", "docs", "test"]
 markers = "python_version <= \"3.11\" or python_version >= \"3.12\""
 files = [
    {file = "packaging-24.2-py3-none-any.whl", hash = "sha256:09abb1bccd265c01f4a3aa3f7a7db064b36514d2cba19a2f694fe6150451a759"},
@@ -5522,6 +5535,23 @@ docs = ["furo (>=2024.8.6)", "proselint (>=0.14)", "sphinx (>=8.0.2)", "sphinx-a
 test = ["appdirs (==1.4.4)", "covdefaults (>=2.3)", "pytest (>=8.3.2)", "pytest-cov (>=5)", "pytest-mock (>=3.14)"]
 type = ["mypy (>=1.11.2)"]

+[[package]]
+name = "pluggy"
+version = "1.5.0"
+description = "plugin and hook calling mechanisms for python"
+optional = false
+python-versions = ">=3.8"
+groups = ["main"]
+markers = "python_version <= \"3.11\" or python_version >= \"3.12\""
+files = [
+    {file = "pluggy-1.5.0-py3-none-any.whl", hash = "sha256:44e1ad92c8ca002de6377e165f3e0f1be63266ab4d554740532335b9d75ea669"},
+    {file = "pluggy-1.5.0.tar.gz", hash = "sha256:2cffa88e94fdc978c4c574f15f9e59b7f4201d439195c3715ca9e2486f1d0cf1"},
+]
+
+[package.extras]
+dev = ["pre-commit", "tox"]
+testing = ["pytest", "pytest-benchmark"]
+
 [[package]]
 name = "posthog"
 version = "3.12.1"
@@ -6382,6 +6412,30 @@ files = [
 [package.extras]
 dev = ["build", "flake8", "mypy", "pytest", "twine"]

+[[package]]
+name = "pytest"
+version = "8.3.4"
+description = "pytest: simple powerful testing with Python"
+optional = false
+python-versions = ">=3.8"
+groups = ["main"]
+markers = "python_version <= \"3.11\" or python_version >= \"3.12\""
+files = [
+    {file = "pytest-8.3.4-py3-none-any.whl", hash = "sha256:50e16d954148559c9a74109af1eaf0c945ba2d8f30f0a3d3335edde19788b6f6"},
+    {file = "pytest-8.3.4.tar.gz", hash = "sha256:965370d062bce11e73868e0335abac31b4d3de0e82f4007408d242b4f8610761"},
+]
+
+[package.dependencies]
+colorama = {version = "*", markers = "sys_platform == \"win32\""}
+exceptiongroup = {version = ">=1.0.0rc8", markers = "python_version < \"3.11\""}
+iniconfig = "*"
+packaging = "*"
+pluggy = ">=1.5,<2"
+tomli = {version = ">=1", markers = "python_version < \"3.11\""}
+
+[package.extras]
+dev = ["argcomplete", "attrs (>=19.2)", "hypothesis (>=3.56)", "mock", "pygments (>=2.7.2)", "requests", "setuptools", "xmlschema"]
+
 [[package]]
 name = "python-dateutil"
 version = "2.9.0.post0"
@@ -7755,7 +7809,7 @@ version = "2.2.1"
 description = "A lil' TOML parser"
 optional = false
 python-versions = ">=3.8"
-groups = ["docs", "test"]
+groups = ["main", "docs", "test"]
 markers = "python_version < \"3.11\""
 files = [
    {file = "tomli-2.2.1-cp311-cp311-macosx_10_9_x86_64.whl", hash = "sha256:678e4fa69e4575eb77d103de3df8a895e1591b48e740211bd1067378c69e8249"},
@@ -8634,4 +8688,4 @@ type = ["pytest-mypy"]
 [metadata]
 lock-version = "2.1"
 python-versions = "^3.10"
-content-hash = "06debb82135affdb2baf1fdcc028c062c236121508d787588cd0de1db2da11e4"
+content-hash = "cbcc30bb9bdead3545070eedf47fb83fe971d4914613e2188323fb3e91035e76"
@@ -10,6 +10,7 @@ readme = "README.md"
 python = "^3.10"
 aiohappyeyeballs = "2.4.3"
 hub = "^3.0.1"
+pytest = "^8.3.4"

 [tool.poetry.group.docs.dependencies]
 langgraph = { path = "../libs/langgraph/", develop = true }
@@ -0,0 +1,117 @@
+import re
+from textwrap import dedent
+
+import pytest
+from _scripts.hook_state import hook_state
+from _scripts.notebook_hooks import handle_vcr_setup
+from _scripts.setup_vcr import extract_code_blocks_for_session, get_hash_for_session
+
+INITIAL_DOCUMENT_CONTENT = dedent(
+    """
+    Blah blah blah
+
+    ```python exec="on" source="above" session="1" result="ansi"
+    print("FIRST_CODE_BLOCK")
+    ```
+
+    Blah blah blah!
+
+    ```python exec="on" source="above" session="1" result="ansi"
+    print("SECOND_CODE_BLOCK")
+    ```
+    
+    more blah blah blah
+    """
+)
+
+
+@pytest.mark.parametrize(
+    "replace_string",
+    [
+        "FIRST_CODE_BLOCK",
+        "SECOND_CODE_BLOCK",
+    ],
+)
+def test_changing_block_in_session_invalidates_hash(replace_string: str):
+    hook_state['document_filename'] = 'test.md'
+    hook_state['document_content'] = INITIAL_DOCUMENT_CONTENT
+
+    code = "print('Hello, world!')"
+
+    result1 = handle_vcr_setup(
+        formatter=lambda **kwargs: None,
+        language="python",
+        session="1",
+        id="test",
+        code=code,
+        md=None,
+        extra={},
+    )
+
+    cassette_init_expr = re.compile(r"^_cassette = HashedCassette\('[^']+', '(?P<hash>[^']+)'\)$")
+    assert result1['transform_source']
+    execute_source, display_source = result1['transform_source'](code)
+    assert display_source == code
+    cassette_init_line = [line for line in execute_source.splitlines() if line.startswith("_cassette = HashedCassette(")][0]
+    assert cassette_init_line
+    match = cassette_init_expr.match(cassette_init_line)
+    assert match
+    hash_ = str(match.group('hash'))
+
+    # change the content of the second block of code
+    hook_state['document_content'] = INITIAL_DOCUMENT_CONTENT.replace(replace_string, "world")
+    assert hook_state['document_content'] != INITIAL_DOCUMENT_CONTENT
+
+    result2 = handle_vcr_setup(
+        formatter=lambda **kwargs: None,
+        language="python",
+        session="1",
+        id="test",
+        code=code,
+        md=None,
+        extra={},
+    )
+
+    assert result2['transform_source']
+    execute_source, display_source = result2['transform_source'](code)
+    assert display_source == code
+    cassette_init_line = [line for line in execute_source.splitlines() if line.startswith("_cassette = HashedCassette(")][0]
+    assert cassette_init_line
+    match = cassette_init_expr.match(cassette_init_line)
+    assert match
+
+    # this is the important part
+    assert str(match.group('hash')) != hash_
+
+@pytest.mark.parametrize(
+    "replace_string",
+    [
+        "FIRST_CODE_BLOCK",
+        "SECOND_CODE_BLOCK",
+    ],
+)
+def test_get_hash_for_session(replace_string: str):
+    hash_ = get_hash_for_session(
+        language="python",
+        session="1",
+        content=INITIAL_DOCUMENT_CONTENT,
+    )
+    
+    content = INITIAL_DOCUMENT_CONTENT.replace(replace_string, "world")
+    assert content != INITIAL_DOCUMENT_CONTENT
+    assert get_hash_for_session(
+        language="python",
+        session="1",
+        content=INITIAL_DOCUMENT_CONTENT.replace(replace_string, "world"),
+    ) != hash_
+
+def test_get_code_blocks_for_session():
+    code_blocks = extract_code_blocks_for_session(
+        language="python",
+        session="1",
+        content=INITIAL_DOCUMENT_CONTENT,
+    )
+    assert code_blocks
+    assert len(code_blocks) == 2
+    assert code_blocks[0] == 'print("FIRST_CODE_BLOCK")'
+    assert code_blocks[1] == 'print("SECOND_CODE_BLOCK")'
@@ -111,7 +111,7 @@ print("Hello, World!")


 END_TO_END_INPUT_HIGHLIGHT_1_EXPECT = """\
-```python exec="on" source="below" hl_lines="2" path="dummy.md"
+```python exec="on" source="below" hl_lines="2"
 print("Hello, World!")
 print("Hello, World!")
 ```
@@ -1,4 +1,5 @@
 {
-  "buildCommand": "yarn build",
-  "outputDirectory": "site"
+  "buildCommand": "echo 'export OPENAI_API_KEY=\"sk-proj-1234567890\"' >> ~/.bashrc && echo 'export ANTHROPIC_API_KEY=\"sk-ant-api03-1234567890\"' >> ~/.bashrc && echo 'export PATH=$PATH:/vercel/.local/bin:$PATH' >> ~/.bashrc && source ~/.bashrc && make vercel-build-docs",
+  "outputDirectory": "site",
+  "installCommand": "echo done"
 }
@@ -16,7 +16,6 @@ from langgraph.checkpoint.base import (
    CheckpointMetadata,
    CheckpointTuple,
    get_checkpoint_id,
-    get_checkpoint_metadata,
 )
 from langgraph.checkpoint.postgres import _internal
 from langgraph.checkpoint.postgres.base import BasePostgresSaver
@@ -318,7 +317,17 @@ class PostgresSaver(BasePostgresSaver):
                    checkpoint["id"],
                    checkpoint_id,
                    Jsonb(self._dump_checkpoint(copy)),
-                    self._dump_metadata(get_checkpoint_metadata(config, metadata)),
+                    self._dump_metadata(
+                        {
+                            **{
+                                k: v
+                                for k, v in config["configurable"].items()
+                                if not k.startswith("__")
+                            },
+                            **config.get("metadata", {}),
+                            **metadata,
+                        }
+                    ),
                ),
            )
        return next_config
@@ -16,7 +16,6 @@ from langgraph.checkpoint.base import (
    CheckpointMetadata,
    CheckpointTuple,
    get_checkpoint_id,
-    get_checkpoint_metadata,
 )
 from langgraph.checkpoint.postgres import _ainternal
 from langgraph.checkpoint.postgres.base import BasePostgresSaver
@@ -276,7 +275,17 @@ class AsyncPostgresSaver(BasePostgresSaver):
                    checkpoint["id"],
                    checkpoint_id,
                    Jsonb(self._dump_checkpoint(copy)),
-                    self._dump_metadata(get_checkpoint_metadata(config, metadata)),
+                    self._dump_metadata(
+                        {
+                            **{
+                                k: v
+                                for k, v in config["configurable"].items()
+                                if not k.startswith("__")
+                            },
+                            **config.get("metadata", {}),
+                            **metadata,
+                        }
+                    ),
                ),
            )
        return next_config
@@ -24,7 +24,6 @@ from langgraph.checkpoint.base import (
    Checkpoint,
    CheckpointMetadata,
    CheckpointTuple,
-    get_checkpoint_metadata,
 )
 from langgraph.checkpoint.postgres import _ainternal, _internal
 from langgraph.checkpoint.postgres.base import BasePostgresSaver
@@ -424,7 +423,17 @@ class ShallowPostgresSaver(BasePostgresSaver):
                    thread_id,
                    checkpoint_ns,
                    Jsonb(self._dump_checkpoint(copy)),
-                    self._dump_metadata(get_checkpoint_metadata(config, metadata)),
+                    self._dump_metadata(
+                        {
+                            **{
+                                k: v
+                                for k, v in config["configurable"].items()
+                                if not k.startswith("__")
+                            },
+                            **config.get("metadata", {}),
+                            **metadata,
+                        }
+                    ),
                ),
            )
        return next_config
@@ -743,7 +752,17 @@ class AsyncShallowPostgresSaver(BasePostgresSaver):
                    thread_id,
                    checkpoint_ns,
                    Jsonb(self._dump_checkpoint(copy)),
-                    self._dump_metadata(get_checkpoint_metadata(config, metadata)),
+                    self._dump_metadata(
+                        {
+                            **{
+                                k: v
+                                for k, v in config["configurable"].items()
+                                if not k.startswith("__")
+                            },
+                            **config.get("metadata", {}),
+                            **metadata,
+                        }
+                    ),
                ),
            )
        return next_config
@@ -187,12 +187,22 @@ 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 = ["dev"]
-markers = "sys_platform == \"win32\""
 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 = "docopt"
+version = "0.6.2"
+description = "Pythonic argument parser, that will make you smile"
+optional = false
+python-versions = "*"
+groups = ["dev"]
+files = [
+    {file = "docopt-0.6.2.tar.gz", hash = "sha256:49b3a825280bd66b3aa83585ef59c4a8c82f2c8a522dbe754a8bc8d08c85c491"},
+]
+
 [[package]]
 name = "exceptiongroup"
 version = "1.2.2"
@@ -348,7 +358,7 @@ typing-extensions = ">=4.7"

 [[package]]
 name = "langgraph-checkpoint"
-version = "2.0.15"
+version = "2.0.10"
 description = "Library with base interfaces for LangGraph checkpoint savers."
 optional = false
 python-versions = "^3.9.0,<4.0"
@@ -952,20 +962,21 @@ pytest = ">=6.2.5"
 dev = ["pre-commit", "pytest-asyncio", "tox"]

 [[package]]
-name = "pytest-watcher"
-version = "0.4.3"
-description = "Automatically rerun your tests on file modifications"
+name = "pytest-watch"
+version = "4.2.0"
+description = "Local continuous test runner with pytest and watchdog."
 optional = false
-python-versions = "<4.0.0,>=3.7.0"
+python-versions = "*"
 groups = ["dev"]
 files = [
-    {file = "pytest_watcher-0.4.3-py3-none-any.whl", hash = "sha256:d59b1e1396f33a65ea4949b713d6884637755d641646960056a90b267c3460f9"},
-    {file = "pytest_watcher-0.4.3.tar.gz", hash = "sha256:0cb0e4661648c8c0ff2b2d25efa5a8e421784b9e4c60fcecbf9b7c30b2d731b3"},
+    {file = "pytest-watch-4.2.0.tar.gz", hash = "sha256:06136f03d5b361718b8d0d234042f7b2f203910d8568f63df2f866b547b3d4b9"},
 ]

 [package.dependencies]
-tomli = {version = ">=2.0.1,<3.0.0", markers = "python_version < \"3.11\""}
-watchdog = ">=2.0.0"
+colorama = ">=0.3.3"
+docopt = ">=0.4.0"
+pytest = ">=2.6.4"
+watchdog = ">=0.6.0"

 [[package]]
 name = "pyyaml"
@@ -1255,4 +1266,4 @@ watchmedo = ["PyYAML (>=3.10)"]
 [metadata]
 lock-version = "2.1"
 python-versions = "^3.9.0,<4.0"
-content-hash = "369bfffecb9489835b43b8255932e043176a11d2f639aad2d055ffd89263ca1e"
+content-hash = "61326e4e81a4e8854763a119f39d4f5d0a54cee868b4dbc91b95ce7d2cebba5b"
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-checkpoint-postgres"
-version = "2.0.15"
+version = "2.0.14"
 description = "Library with a Postgres implementation of LangGraph checkpoint saver."
 authors = []
 license = "MIT"
@@ -10,7 +10,7 @@ packages = [{ include = "langgraph" }]

 [tool.poetry.dependencies]
 python = "^3.9.0,<4.0"
-langgraph-checkpoint = "^2.0.15"
+langgraph-checkpoint = "^2.0.10"
 orjson = ">=3.10.1"
 psycopg = "^3.2.0"
 psycopg-pool = "^3.2.0"
@@ -22,10 +22,10 @@ pytest = "^7.2.1"
 anyio = "^4.4.0"
 pytest-asyncio = "^0.21.1"
 pytest-mock = "^3.11.1"
+pytest-watch = "^4.2.0"
 mypy = "^1.10.0"
 psycopg = {extras = ["binary"], version = ">=3.0.0"}
 langgraph-checkpoint = {path = "../checkpoint", develop = true}
-pytest-watcher = "^0.4.3"

 [tool.pytest.ini_options]
 # --strict-markers will raise errors on unknown marks.
@@ -61,9 +61,3 @@ warn_unused_ignores = "True"
 warn_redundant_casts = "True"
 allow_redefinition = "True"
 disable_error_code = "typeddict-item, return-value"
-
-[tool.pytest-watcher]
-now = true
-delay = 0.1
-runner_args = ["--ff", "-x", "-v", "--tb", "short"]
-patterns = ["*.py"]
@@ -11,7 +11,6 @@ from psycopg.rows import dict_row
 from psycopg_pool import AsyncConnectionPool

 from langgraph.checkpoint.base import (
-    EXCLUDED_METADATA_KEYS,
    Checkpoint,
    CheckpointMetadata,
    create_checkpoint,
@@ -24,10 +23,6 @@ from langgraph.checkpoint.postgres.aio import (
 from tests.conftest import DEFAULT_POSTGRES_URI


-def _exclude_keys(config: dict[str, Any]) -> dict[str, Any]:
-    return {k: v for k, v in config.items() if k not in EXCLUDED_METADATA_KEYS}
-
-
@asynccontextmanager
 async def _pool_saver():
    """Fixture for pool mode testing."""
@@ -228,6 +223,7 @@ async def test_combined_metadata(saver_name: str, test_data) -> None:
        assert checkpoint.metadata == {
            **metadata,
            "thread_id": "thread-2",
+            "checkpoint_ns": "",
            "run_id": "my_run_id",
        }

@@ -255,14 +251,14 @@ async def test_asearch(saver_name: str, test_data) -> None:
        search_results_1 = [c async for c in saver.alist(None, filter=query_1)]
        assert len(search_results_1) == 1
        assert search_results_1[0].metadata == {
-            **_exclude_keys(configs[0]["configurable"]),
+            **configs[0]["configurable"],
            **metadata[0],
        }

        search_results_2 = [c async for c in saver.alist(None, filter=query_2)]
        assert len(search_results_2) == 1
        assert search_results_2[0].metadata == {
-            **_exclude_keys(configs[1]["configurable"]),
+            **configs[1]["configurable"],
            **metadata[1],
        }

@@ -12,7 +12,6 @@ from psycopg.rows import dict_row
 from psycopg_pool import ConnectionPool

 from langgraph.checkpoint.base import (
-    EXCLUDED_METADATA_KEYS,
    Checkpoint,
    CheckpointMetadata,
    create_checkpoint,
@@ -22,10 +21,6 @@ from langgraph.checkpoint.postgres import PostgresSaver, ShallowPostgresSaver
 from tests.conftest import DEFAULT_POSTGRES_URI


-def _exclude_keys(config: dict[str, Any]) -> dict[str, Any]:
-    return {k: v for k, v in config.items() if k not in EXCLUDED_METADATA_KEYS}
-
-
@contextmanager
 def _pool_saver():
    """Fixture for pool mode testing."""
@@ -210,6 +205,7 @@ def test_combined_metadata(saver_name: str, test_data) -> None:
        assert checkpoint.metadata == {
            **metadata,
            "thread_id": "thread-2",
+            "checkpoint_ns": "",
            "run_id": "my_run_id",
        }

@@ -237,14 +233,14 @@ def test_search(saver_name: str, test_data) -> None:
        search_results_1 = list(saver.list(None, filter=query_1))
        assert len(search_results_1) == 1
        assert search_results_1[0].metadata == {
-            **_exclude_keys(configs[0]["configurable"]),
+            **configs[0]["configurable"],
            **metadata[0],
        }

        search_results_2 = list(saver.list(None, filter=query_2))
        assert len(search_results_2) == 1
        assert search_results_2[0].metadata == {
-            **_exclude_keys(configs[1]["configurable"]),
+            **configs[1]["configurable"],
            **metadata[1],
        }

@@ -15,7 +15,6 @@ from langgraph.checkpoint.base import (
    CheckpointTuple,
    SerializerProtocol,
    get_checkpoint_id,
-    get_checkpoint_metadata,
 )
 from langgraph.checkpoint.serde.jsonplus import JsonPlusSerializer
 from langgraph.checkpoint.serde.types import ChannelProtocol
@@ -399,7 +398,15 @@ class SqliteSaver(BaseCheckpointSaver[str]):
        checkpoint_ns = config["configurable"]["checkpoint_ns"]
        type_, serialized_checkpoint = self.serde.dumps_typed(checkpoint)
        serialized_metadata = self.jsonplus_serde.dumps(
-            get_checkpoint_metadata(config, metadata)
+            {
+                **{
+                    k: v
+                    for k, v in config["configurable"].items()
+                    if not k.startswith("__")
+                },
+                **config.get("metadata", {}),
+                **metadata,
+            }
        )
        with self.cursor() as cur:
            cur.execute(
@@ -16,7 +16,6 @@ from langgraph.checkpoint.base import (
    CheckpointTuple,
    SerializerProtocol,
    get_checkpoint_id,
-    get_checkpoint_metadata,
 )
 from langgraph.checkpoint.serde.jsonplus import JsonPlusSerializer
 from langgraph.checkpoint.serde.types import ChannelProtocol
@@ -465,7 +464,15 @@ class AsyncSqliteSaver(BaseCheckpointSaver[str]):
        checkpoint_ns = config["configurable"]["checkpoint_ns"]
        type_, serialized_checkpoint = self.serde.dumps_typed(checkpoint)
        serialized_metadata = self.jsonplus_serde.dumps(
-            get_checkpoint_metadata(config, metadata)
+            {
+                **{
+                    k: v
+                    for k, v in config["configurable"].items()
+                    if not k.startswith("__")
+                },
+                **config.get("metadata", {}),
+                **metadata,
+            }
        )
        async with (
            self.lock,
@@ -350,7 +350,7 @@ typing-extensions = ">=4.7"

 [[package]]
 name = "langgraph-checkpoint"
-version = "2.0.15"
+version = "2.0.10"
 description = "Library with base interfaces for LangGraph checkpoint savers."
 optional = false
 python-versions = "^3.9.0,<4.0"
@@ -1043,4 +1043,4 @@ watchmedo = ["PyYAML (>=3.10)"]
 [metadata]
 lock-version = "2.1"
 python-versions = "^3.9.0"
-content-hash = "e6d3ca9bce723c05f4c5ae9dc4bee872f7581b7763680b34112f1d280f5a9b0a"
+content-hash = "03c697eae6f550f3c7e29f1d61f4c409dabe04ae8d43281728e549174d2fc670"
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-checkpoint-sqlite"
-version = "2.0.5"
+version = "2.0.4"
 description = "Library with a SQLite implementation of LangGraph checkpoint saver."
 authors = []
 license = "MIT"
@@ -10,7 +10,7 @@ packages = [{ include = "langgraph" }]

 [tool.poetry.dependencies]
 python = "^3.9.0"
-langgraph-checkpoint = "^2.0.15"
+langgraph-checkpoint = "^2.0.10"
 aiosqlite = "^0.20.0"

 [tool.poetry.group.dev.dependencies]
@@ -72,6 +72,7 @@ class TestAsyncSqliteSaver:
            assert checkpoint.metadata == {
                **self.metadata_2,
                "thread_id": "thread-2",
+                "checkpoint_ns": "",
                "run_id": "my_run_id",
            }

@@ -93,15 +94,14 @@ class TestAsyncSqliteSaver:
            search_results_1 = [c async for c in saver.alist(None, filter=query_1)]
            assert len(search_results_1) == 1
            assert search_results_1[0].metadata == {
-                "thread_id": "thread-1",
-                "thread_ts": "1",
+                **self.config_1["configurable"],
                **self.metadata_1,
            }

            search_results_2 = [c async for c in saver.alist(None, filter=query_2)]
            assert len(search_results_2) == 1
            assert search_results_2[0].metadata == {
-                "thread_id": "thread-2",
+                **self.config_2["configurable"],
                **self.metadata_2,
            }

@@ -73,6 +73,7 @@ class TestSqliteSaver:
            assert checkpoint.metadata == {
                **self.metadata_2,
                "thread_id": "thread-2",
+                "checkpoint_ns": "",
                "run_id": "my_run_id",
            }

@@ -96,15 +97,14 @@ class TestSqliteSaver:
            search_results_1 = list(saver.list(None, filter=query_1))
            assert len(search_results_1) == 1
            assert search_results_1[0].metadata == {
-                "thread_id": "thread-1",
-                "thread_ts": "1",
+                **self.config_1["configurable"],
                **self.metadata_1,
            }

            search_results_2 = list(saver.list(None, filter=query_2))
            assert len(search_results_2) == 1
            assert search_results_2[0].metadata == {
-                "thread_id": "thread-2",
+                **self.config_2["configurable"],
                **self.metadata_2,
            }

@@ -446,23 +446,6 @@ def get_checkpoint_id(config: RunnableConfig) -> Optional[str]:
    )


-def get_checkpoint_metadata(
-    config: RunnableConfig, metadata: CheckpointMetadata
-) -> CheckpointMetadata:
-    """Get checkpoint metadata in a backwards-compatible manner."""
-    metadata = metadata.copy()
-    for obj in (config.get("metadata"), config.get("configurable")):
-        if not obj:
-            continue
-        for key in obj:
-            if key in metadata or key in EXCLUDED_METADATA_KEYS or key.startswith("__"):
-                continue
-            v = obj[key]
-            if isinstance(v, (str, int, bool, float)):
-                metadata[key] = v  # type: ignore[literal-required]
-    return metadata
-
-
 """
 Mapping from error type to error index.
 Regular writes just map to their index in the list of writes being saved.
@@ -471,9 +454,3 @@ conflicting with regular writes.
 Each Checkpointer implementation should use this mapping in put_writes.
 """
 WRITES_IDX_MAP = {ERROR: -1, SCHEDULED: -2, INTERRUPT: -3, RESUME: -4}
-
-EXCLUDED_METADATA_KEYS = {
-    "checkpoint_id",
-    "checkpoint_ns",
-    "checkpoint_map",
-}
@@ -20,7 +20,6 @@ from langgraph.checkpoint.base import (
    CheckpointTuple,
    SerializerProtocol,
    get_checkpoint_id,
-    get_checkpoint_metadata,
 )
 from langgraph.checkpoint.serde.types import TASKS, ChannelProtocol

@@ -357,7 +356,17 @@ class InMemorySaver(
            {
                checkpoint["id"]: (
                    self.serde.dumps_typed(c),
-                    self.serde.dumps_typed(get_checkpoint_metadata(config, metadata)),
+                    self.serde.dumps_typed(
+                        {
+                            **{
+                                k: v
+                                for k, v in config["configurable"].items()
+                                if not k.startswith("__")
+                            },
+                            **config.get("metadata", {}),
+                            **metadata,
+                        }
+                    ),
                    config["configurable"].get("checkpoint_id"),  # parent
                )
            }
@@ -1,8 +1,7 @@
 import asyncio
 import functools
 import weakref
-from collections.abc import Iterable
-from typing import Any, Callable, Literal, Optional, TypeVar, Union
+from typing import Any, Callable, Iterable, Literal, Optional, TypeVar, Union

 from langgraph.store.base import (
    BaseStore,
@@ -55,23 +54,19 @@ class AsyncBatchedBaseStore(BaseStore):
    def __init__(self) -> None:
        super().__init__()
        self._loop = asyncio.get_running_loop()
-        self._aqueue: asyncio.Queue[tuple[asyncio.Future, Op]] = asyncio.Queue()
+        self._aqueue: dict[asyncio.Future, Op] = {}
        self._task = self._loop.create_task(_run(self._aqueue, weakref.ref(self)))

    def __del__(self) -> None:
-        try:
-            self._task.cancel()
-        except RuntimeError:
-            pass
+        self._task.cancel()

    async def aget(
        self,
        namespace: tuple[str, ...],
        key: str,
    ) -> Optional[Item]:
-        assert not self._task.done()
        fut = self._loop.create_future()
-        self._aqueue.put_nowait((fut, GetOp(namespace, key)))
+        self._aqueue[fut] = GetOp(namespace, key)
        return await fut

    async def asearch(
@@ -84,11 +79,8 @@ class AsyncBatchedBaseStore(BaseStore):
        limit: int = 10,
        offset: int = 0,
    ) -> list[SearchItem]:
-        assert not self._task.done()
        fut = self._loop.create_future()
-        self._aqueue.put_nowait(
-            (fut, SearchOp(namespace_prefix, filter, limit, offset, query))
-        )
+        self._aqueue[fut] = SearchOp(namespace_prefix, filter, limit, offset, query)
        return await fut

    async def aput(
@@ -98,10 +90,9 @@ class AsyncBatchedBaseStore(BaseStore):
        value: dict[str, Any],
        index: Optional[Union[Literal[False], list[str]]] = None,
    ) -> None:
-        assert not self._task.done()
        _validate_namespace(namespace)
        fut = self._loop.create_future()
-        self._aqueue.put_nowait((fut, PutOp(namespace, key, value, index)))
+        self._aqueue[fut] = PutOp(namespace, key, value, index)
        return await fut

    async def adelete(
@@ -109,9 +100,8 @@ class AsyncBatchedBaseStore(BaseStore):
        namespace: tuple[str, ...],
        key: str,
    ) -> None:
-        assert not self._task.done()
        fut = self._loop.create_future()
-        self._aqueue.put_nowait((fut, PutOp(namespace, key, None)))
+        self._aqueue[fut] = PutOp(namespace, key, None)
        return await fut

    async def alist_namespaces(
@@ -123,7 +113,6 @@ class AsyncBatchedBaseStore(BaseStore):
        limit: int = 100,
        offset: int = 0,
    ) -> list[tuple[str, ...]]:
-        assert not self._task.done()
        fut = self._loop.create_future()
        match_conditions = []
        if prefix:
@@ -137,7 +126,7 @@ class AsyncBatchedBaseStore(BaseStore):
            limit=limit,
            offset=offset,
        )
-        self._aqueue.put_nowait((fut, op))
+        self._aqueue[fut] = op
        return await fut

    @_check_loop
@@ -261,38 +250,34 @@ def _dedupe_ops(values: list[Op]) -> tuple[Optional[list[int]], list[Op]]:


 async def _run(
-    aqueue: asyncio.Queue[tuple[asyncio.Future, Op]],
+    aqueue: dict[asyncio.Future, Op],
    store: weakref.ReferenceType[BaseStore],
 ) -> None:
-    while item := await aqueue.get():
-        # check if store is still alive
+    while True:
+        await asyncio.sleep(0)
+        if not aqueue:
+            continue
        if s := store():
+            # get the operations to run
+            taken = aqueue.copy()
+            # action each operation
            try:
-                # accumulate operations scheduled in same tick
-                items = [item]
-                try:
-                    while item := aqueue.get_nowait():
-                        items.append(item)
-                except asyncio.QueueEmpty:
-                    pass
-                # get the operations to run
-                futs = [item[0] for item in items]
-                values = [item[1] for item in items]
-                # action each operation
-                try:
-                    listen, dedupped = _dedupe_ops(values)
-                    results = await s.abatch(dedupped)
-                    if listen is not None:
-                        results = [results[ix] for ix in listen]
+                values = list(taken.values())
+                listen, dedupped = _dedupe_ops(values)
+                results = await s.abatch(dedupped)
+                if listen is not None:
+                    results = [results[ix] for ix in listen]

-                    # set the results of each operation
-                    for fut, result in zip(futs, results):
-                        fut.set_result(result)
-                except Exception as e:
-                    for fut in futs:
-                        fut.set_exception(e)
-            finally:
-                # remove strong ref to store
-                del s
+                # set the results of each operation
+                for fut, result in zip(taken, results):
+                    fut.set_result(result)
+            except Exception as e:
+                for fut in taken:
+                    fut.set_exception(e)
+            # remove the operations from the queue
+            for fut in taken:
+                del aqueue[fut]
        else:
            break
+        # remove strong ref to store
+        del s
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-checkpoint"
-version = "2.0.15"
+version = "2.0.13"
 description = "Library with base interfaces for LangGraph checkpoint savers."
 authors = []
 license = "MIT"
@@ -60,7 +60,7 @@ class TestMemorySaver:
        self.metadata_3: CheckpointMetadata = {}

    def test_combined_metadata(self) -> None:
-        config: RunnableConfig = {
+        config = {
            "configurable": {
                "thread_id": "thread-2",
                "checkpoint_ns": "",
@@ -70,10 +70,10 @@ class TestMemorySaver:
        }
        self.memory_saver.put(config, self.chkpnt_2, self.metadata_2, {})
        checkpoint = self.memory_saver.get_tuple(config)
-        assert checkpoint is not None
        assert checkpoint.metadata == {
            **self.metadata_2,
            "thread_id": "thread-2",
+            "checkpoint_ns": "",
            "run_id": "my_run_id",
        }

@@ -96,15 +96,14 @@ class TestMemorySaver:
        search_results_1 = list(self.memory_saver.list(None, filter=query_1))
        assert len(search_results_1) == 1
        assert search_results_1[0].metadata == {
-            "thread_id": "thread-1",
-            "thread_ts": "1",
+            **self.config_1["configurable"],
            **self.metadata_1,
        }

        search_results_2 = list(self.memory_saver.list(None, filter=query_2))
        assert len(search_results_2) == 1
        assert search_results_2[0].metadata == {
-            "thread_id": "thread-2",
+            **self.config_2["configurable"],
            **self.metadata_2,
        }

@@ -147,8 +146,7 @@ class TestMemorySaver:
        ]
        assert len(search_results_1) == 1
        assert search_results_1[0].metadata == {
-            "thread_id": "thread-1",
-            "thread_ts": "1",
+            **self.config_1["configurable"],
            **self.metadata_1,
        }

@@ -157,7 +155,7 @@ class TestMemorySaver:
        ]
        assert len(search_results_2) == 1
        assert search_results_2[0].metadata == {
-            "thread_id": "thread-2",
+            **self.config_2["configurable"],
            **self.metadata_2,
        }

@@ -13,7 +13,6 @@ from typing import (
    Coroutine,
    Iterator,
    Optional,
-    Protocol,
    Sequence,
    Tuple,
    Union,
@@ -36,7 +35,7 @@ from langchain_core.runnables.config import (
 )
 from langchain_core.runnables.utils import Input, Output
 from langchain_core.tracers._streaming import _StreamingCallbackHandler
-from typing_extensions import TypeGuard
+from typing_extensions import Concatenate, ParamSpec, TypeGuard

 from langgraph.constants import (
    CONF,
@@ -133,51 +132,12 @@ Each tuple contains:
 VALID_KINDS = (inspect.Parameter.POSITIONAL_OR_KEYWORD, inspect.Parameter.KEYWORD_ONLY)


-class _RunnableWithWriter(Protocol[Input, Output]):
-    def __call__(self, state: Input, *, writer: StreamWriter) -> Output: ...
-
-
-class _RunnableWithStore(Protocol[Input, Output]):
-    def __call__(self, state: Input, *, store: BaseStore) -> Output: ...
-
-
-class _RunnableWithWriterStore(Protocol[Input, Output]):
-    def __call__(
-        self, state: Input, *, writer: StreamWriter, store: BaseStore
-    ) -> Output: ...
-
-
-class _RunnableWithConfigWriter(Protocol[Input, Output]):
-    def __call__(
-        self, state: Input, *, config: RunnableConfig, writer: StreamWriter
-    ) -> Output: ...
-
-
-class _RunnableWithConfigStore(Protocol[Input, Output]):
-    def __call__(
-        self, state: Input, *, config: RunnableConfig, store: BaseStore
-    ) -> Output: ...
-
-
-class _RunnableWithConfigWriterStore(Protocol[Input, Output]):
-    def __call__(
-        self,
-        state: Input,
-        *,
-        config: RunnableConfig,
-        writer: StreamWriter,
-        store: BaseStore,
-    ) -> Output: ...
-
+P = ParamSpec("P")  # to handle injected kwargs like `writer` / `store`

 RunnableLike = Union[
    LCRunnableLike,
-    _RunnableWithWriter[Input, Output],
-    _RunnableWithStore[Input, Output],
-    _RunnableWithWriterStore[Input, Output],
-    _RunnableWithConfigWriter[Input, Output],
-    _RunnableWithConfigStore[Input, Output],
-    _RunnableWithConfigWriterStore[Input, Output],
+    Callable[Concatenate[Input, P], Output],
+    Callable[Concatenate[Input, P], Awaitable[Output]],
 ]


@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph"
-version = "0.2.73"
+version = "0.2.72"
 description = "Building stateful, multi-actor applications with LLMs"
 authors = []
 license = "MIT"
@@ -133,6 +133,8 @@ async def test_invoke_two_processes_in_out_interrupt(
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "step": 6,
@@ -156,6 +158,8 @@ async def test_invoke_two_processes_in_out_interrupt(
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "step": 5,
@@ -179,6 +183,8 @@ async def test_invoke_two_processes_in_out_interrupt(
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "input",
                    "step": 4,
@@ -200,6 +206,8 @@ async def test_invoke_two_processes_in_out_interrupt(
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "step": 3,
@@ -223,6 +231,8 @@ async def test_invoke_two_processes_in_out_interrupt(
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "input",
                    "step": 2,
@@ -244,6 +254,8 @@ async def test_invoke_two_processes_in_out_interrupt(
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "step": 1,
@@ -267,6 +279,8 @@ async def test_invoke_two_processes_in_out_interrupt(
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "step": 0,
@@ -290,6 +304,7 @@ async def test_invoke_two_processes_in_out_interrupt(
                    }
                },
                metadata={
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "input",
                    "step": -1,
@@ -366,6 +381,8 @@ async def test_fork_always_re_runs_nodes(
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "step": 5,
@@ -387,6 +404,8 @@ async def test_fork_always_re_runs_nodes(
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "step": 4,
@@ -408,6 +427,8 @@ async def test_fork_always_re_runs_nodes(
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "step": 3,
@@ -429,6 +450,8 @@ async def test_fork_always_re_runs_nodes(
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "step": 2,
@@ -450,6 +473,8 @@ async def test_fork_always_re_runs_nodes(
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "step": 1,
@@ -471,6 +496,8 @@ async def test_fork_always_re_runs_nodes(
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "step": 0,
@@ -494,6 +521,7 @@ async def test_fork_always_re_runs_nodes(
                    }
                },
                metadata={
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "input",
                    "step": -1,
@@ -834,6 +862,8 @@ async def test_conditional_graph(checkpointer_name: str) -> None:
                await app_w_interrupt.checkpointer.aget_tuple(config)
            ).checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 0,
@@ -894,6 +924,8 @@ async def test_conditional_graph(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 1,
@@ -1018,6 +1050,8 @@ async def test_conditional_graph(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 4,
@@ -1100,6 +1134,8 @@ async def test_conditional_graph(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 0,
@@ -1160,6 +1196,8 @@ async def test_conditional_graph(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 1,
@@ -1284,6 +1322,8 @@ async def test_conditional_graph(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 4,
@@ -1366,6 +1406,8 @@ async def test_conditional_graph(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 0,
@@ -1798,6 +1840,8 @@ async def test_conditional_graph_state(
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 1,
@@ -1853,6 +1897,8 @@ async def test_conditional_graph_state(
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 2,
@@ -1943,6 +1989,8 @@ async def test_conditional_graph_state(
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 5,
@@ -2010,6 +2058,8 @@ async def test_conditional_graph_state(
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 1,
@@ -2064,6 +2114,8 @@ async def test_conditional_graph_state(
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 2,
@@ -2152,6 +2204,8 @@ async def test_conditional_graph_state(
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 5,
@@ -2768,6 +2822,8 @@ async def test_state_graph_packets(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 1,
@@ -2827,6 +2883,8 @@ async def test_state_graph_packets(checkpointer_name: str) -> None:
            config=tup.config,
            created_at=tup.checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 2,
@@ -2936,6 +2994,8 @@ async def test_state_graph_packets(checkpointer_name: str) -> None:
            config=tup.config,
            created_at=tup.checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 4,
@@ -3005,6 +3065,8 @@ async def test_state_graph_packets(checkpointer_name: str) -> None:
            config=tup.config,
            created_at=tup.checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 5,
@@ -3079,6 +3141,8 @@ async def test_state_graph_packets(checkpointer_name: str) -> None:
            config=tup.config,
            created_at=tup.checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 1,
@@ -3140,6 +3204,8 @@ async def test_state_graph_packets(checkpointer_name: str) -> None:
            config=tup.config,
            created_at=tup.checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 2,
@@ -3249,6 +3315,8 @@ async def test_state_graph_packets(checkpointer_name: str) -> None:
            config=tup.config,
            created_at=tup.checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 4,
@@ -3318,6 +3386,8 @@ async def test_state_graph_packets(checkpointer_name: str) -> None:
            config=tup.config,
            created_at=tup.checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 5,
@@ -3575,6 +3645,8 @@ async def test_message_graph(checkpointer_name: str) -> None:
            config=tup.config,
            created_at=tup.checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 1,
@@ -3629,6 +3701,8 @@ async def test_message_graph(checkpointer_name: str) -> None:
            config=tup.config,
            created_at=tup.checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 2,
@@ -3719,6 +3793,8 @@ async def test_message_graph(checkpointer_name: str) -> None:
            config=tup.config,
            created_at=tup.checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 4,
@@ -3779,6 +3855,8 @@ async def test_message_graph(checkpointer_name: str) -> None:
            config=tup.config,
            created_at=tup.checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 5,
@@ -4087,6 +4165,8 @@ async def test_start_branch_then(checkpointer_name: str) -> None:
        if "shallow" not in checkpointer_name:
            assert [c.metadata async for c in tool_two.checkpointer.alist(thread1)] == [
                {
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "step": 0,
@@ -4095,6 +4175,7 @@ async def test_start_branch_then(checkpointer_name: str) -> None:
                    "thread_id": "1",
                },
                {
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "input",
                    "step": -1,
@@ -4117,6 +4198,8 @@ async def test_start_branch_then(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 0,
@@ -4150,6 +4233,8 @@ async def test_start_branch_then(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 1,
@@ -4185,6 +4270,8 @@ async def test_start_branch_then(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 0,
@@ -4218,6 +4305,8 @@ async def test_start_branch_then(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 1,
@@ -4253,6 +4342,8 @@ async def test_start_branch_then(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 0,
@@ -4283,6 +4374,8 @@ async def test_start_branch_then(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 1,
@@ -4316,6 +4409,8 @@ async def test_start_branch_then(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 2,
@@ -4856,6 +4951,8 @@ async def test_branch_then(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 1,
@@ -4888,6 +4985,8 @@ async def test_branch_then(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 3,
@@ -4922,6 +5021,8 @@ async def test_branch_then(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 1,
@@ -4954,6 +5055,8 @@ async def test_branch_then(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 3,
@@ -4996,6 +5099,8 @@ async def test_branch_then(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 1,
@@ -5028,6 +5133,8 @@ async def test_branch_then(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 3,
@@ -5062,6 +5169,8 @@ async def test_branch_then(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 1,
@@ -5094,6 +5203,8 @@ async def test_branch_then(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 3,
@@ -5122,6 +5233,7 @@ async def test_branch_then(checkpointer_name: str) -> None:
            config=uconfig,
            created_at=AnyStr(),
            metadata={
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 0,
@@ -5149,6 +5261,8 @@ async def test_branch_then(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 1,
@@ -5175,6 +5289,8 @@ async def test_branch_then(checkpointer_name: str) -> None:
            },
            created_at=AnyStr(),
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 3,
@@ -5266,6 +5382,8 @@ async def test_nested_graph_state(checkpointer_name: str) -> None:
                }
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "writes": {"outer_1": {"my_key": "hi my value"}},
@@ -5317,6 +5435,8 @@ async def test_nested_graph_state(checkpointer_name: str) -> None:
                            }
                        },
                        metadata={
+                            "checkpoint_id": AnyStr(),
+                            "checkpoint_map": AnyDict({"": AnyStr()}),
                            "parents": {
                                "": AnyStr(),
                            },
@@ -5363,6 +5483,8 @@ async def test_nested_graph_state(checkpointer_name: str) -> None:
                }
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "writes": {"outer_1": {"my_key": "hi my value"}},
@@ -5409,6 +5531,8 @@ async def test_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "writes": {"outer_1": {"my_key": "hi my value"}},
@@ -5447,6 +5571,8 @@ async def test_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "writes": None,
@@ -5481,6 +5607,7 @@ async def test_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "input",
                    "writes": {"__start__": {"my_key": "my value"}},
@@ -5516,6 +5643,8 @@ async def test_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_map": AnyDict({"": AnyStr()}),
                    "source": "loop",
                    "writes": {
                        "inner_1": {
@@ -5564,6 +5693,8 @@ async def test_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_map": AnyDict({"": AnyStr()}),
                    "source": "loop",
                    "writes": None,
                    "step": 0,
@@ -5613,6 +5744,8 @@ async def test_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": None,
+                    "checkpoint_map": AnyDict({"": AnyStr()}),
                    "source": "input",
                    "writes": {"__start__": {"my_key": "hi my value"}},
                    "step": -1,
@@ -5658,6 +5791,8 @@ async def test_nested_graph_state(checkpointer_name: str) -> None:
                }
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "writes": {
@@ -5694,6 +5829,8 @@ async def test_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "writes": {
@@ -5736,6 +5873,8 @@ async def test_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "writes": {"inner": {"my_key": "hi my value here and there"}},
@@ -5776,6 +5915,8 @@ async def test_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "writes": {"outer_1": {"my_key": "hi my value"}},
@@ -5810,6 +5951,8 @@ async def test_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "writes": None,
@@ -5844,6 +5987,7 @@ async def test_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "input",
                    "writes": {"__start__": {"my_key": "my value"}},
@@ -5953,6 +6097,8 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                }
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "writes": {"parent_1": {"my_key": "hi my value"}},
@@ -5999,6 +6145,8 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {"": AnyStr()},
                    "source": "loop",
                    "writes": None,
@@ -6045,6 +6193,8 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                }
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_map": AnyDict({"": AnyStr(), AnyStr("child:"): AnyStr()}),
                "parents": AnyDict(
                    {
                        "": AnyStr(),
@@ -6124,6 +6274,10 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                                        }
                                    },
                                    metadata={
+                                        "checkpoint_id": AnyStr(),
+                                        "checkpoint_map": AnyDict(
+                                            {"": AnyStr(), AnyStr("child:"): AnyStr()}
+                                        ),
                                        "parents": AnyDict(
                                            {
                                                "": AnyStr(),
@@ -6186,6 +6340,8 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                            }
                        },
                        metadata={
+                            "checkpoint_id": AnyStr(),
+                            "checkpoint_map": AnyDict({"": AnyStr()}),
                            "parents": {"": AnyStr()},
                            "source": "loop",
                            "writes": None,
@@ -6225,6 +6381,8 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                }
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "writes": {"parent_1": {"my_key": "hi my value"}},
@@ -6273,6 +6431,8 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "writes": {
@@ -6318,6 +6478,8 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                        }
                    },
                    metadata={
+                        "checkpoint_id": AnyStr(),
+                        "checkpoint_ns": "",
                        "parents": {},
                        "source": "loop",
                        "writes": {
@@ -6348,6 +6510,8 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                        }
                    },
                    metadata={
+                        "checkpoint_id": AnyStr(),
+                        "checkpoint_ns": "",
                        "parents": {},
                        "source": "loop",
                        "writes": {"child": {"my_key": "hi my value here and there"}},
@@ -6392,6 +6556,8 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                        }
                    },
                    metadata={
+                        "checkpoint_id": AnyStr(),
+                        "checkpoint_ns": "",
                        "parents": {},
                        "source": "loop",
                        "writes": {"parent_1": {"my_key": "hi my value"}},
@@ -6449,6 +6615,7 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                        }
                    },
                    metadata={
+                        "checkpoint_ns": "",
                        "parents": {},
                        "source": "input",
                        "writes": {"my_key": "my value"},
@@ -6484,6 +6651,8 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_map": AnyDict({"": AnyStr()}),
                    "source": "loop",
                    "writes": {"child_1": {"my_key": "hi my value here and there"}},
                    "step": 1,
@@ -6523,6 +6692,8 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_map": AnyDict({"": AnyStr()}),
                    "source": "loop",
                    "writes": None,
                    "step": 0,
@@ -6575,6 +6746,8 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": None,
+                    "checkpoint_map": AnyDict({"": AnyStr()}),
                    "source": "input",
                    "writes": {"__start__": {"my_key": "hi my value"}},
                    "step": -1,
@@ -6622,6 +6795,10 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_map": AnyDict(
+                        {"": AnyStr(), AnyStr("child:"): AnyStr()}
+                    ),
                    "source": "loop",
                    "writes": {
                        "grandchild_2": {"my_key": "hi my value here and there"}
@@ -6679,6 +6856,10 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_map": AnyDict(
+                        {"": AnyStr(), AnyStr("child:"): AnyStr()}
+                    ),
                    "source": "loop",
                    "writes": {"grandchild_1": {"my_key": "hi my value here"}},
                    "step": 1,
@@ -6741,6 +6922,10 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_map": AnyDict(
+                        {"": AnyStr(), AnyStr("child:"): AnyStr()}
+                    ),
                    "source": "loop",
                    "writes": None,
                    "step": 0,
@@ -6803,6 +6988,10 @@ async def test_doubly_nested_graph_state(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": None,
+                    "checkpoint_map": AnyDict(
+                        {"": AnyStr(), AnyStr("child:"): AnyStr()}
+                    ),
                    "source": "input",
                    "writes": {"__start__": {"my_key": "hi my value"}},
                    "step": -1,
@@ -7081,6 +7270,8 @@ async def test_weather_subgraph(
                }
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "source": "loop",
                "writes": {"router_node": {"route": "weather"}},
                "step": 1,
@@ -7177,6 +7368,8 @@ async def test_weather_subgraph(
                }
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "source": "loop",
                "writes": {"router_node": {"route": "weather"}},
                "step": 1,
@@ -7222,6 +7415,8 @@ async def test_weather_subgraph(
                            }
                        },
                        metadata={
+                            "checkpoint_id": AnyStr(),
+                            "checkpoint_map": AnyDict({"": AnyStr()}),
                            "source": "loop",
                            "writes": {"model_node": {"city": "San Francisco"}},
                            "step": 1,
@@ -7285,6 +7480,8 @@ async def test_weather_subgraph(
                }
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "source": "loop",
                "writes": {"router_node": {"route": "weather"}},
                "step": 1,
@@ -7331,6 +7528,7 @@ async def test_weather_subgraph(
                            }
                        },
                        metadata={
+                            "checkpoint_map": AnyDict({"": AnyStr()}),
                            "step": 2,
                            "source": "update",
                            "writes": {
@@ -63,6 +63,7 @@ from langgraph.store.base import BaseStore
 from langgraph.store.memory import InMemoryStore
 from langgraph.types import Command, Interrupt, interrupt
 from langgraph.utils.config import get_stream_writer
+from tests.any_str import AnyStr
 from tests.conftest import (
    ALL_CHECKPOINTERS_ASYNC,
    ALL_CHECKPOINTERS_SYNC,
@@ -182,6 +183,8 @@ def test_no_prompt(
            "agent": "agent",
        }
        assert saved.metadata == {
+            "checkpoint_id": AnyStr(),
+            "checkpoint_ns": "",
            "parents": {},
            "source": "loop",
            "writes": {"agent": {"messages": [AIMessage(content="hi?", id="0")]}},
@@ -214,6 +217,8 @@ async def test_no_prompt_async(checkpointer_name: str) -> None:
                "agent": "agent",
            }
            assert saved.metadata == {
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "writes": {"agent": {"messages": [AIMessage(content="hi?", id="0")]}},
@@ -1112,6 +1112,8 @@ def test_pending_writes_resume(
        PregelTask(AnyStr(), "two", (PULL, "two"), 'ConnectionError("I\'m not good")'),
    )
    assert state.metadata == {
+        "checkpoint_id": AnyStr(),
+        "checkpoint_ns": "",
        "parents": {},
        "source": "loop",
        "step": 0,
@@ -1210,6 +1212,8 @@ def test_pending_writes_resume(
            "channel_values": {"one": "one", "two": "two", "value": 6},
        },
        metadata={
+            "checkpoint_id": AnyStr(),
+            "checkpoint_ns": "",
            "parents": {},
            "step": 1,
            "source": "loop",
@@ -1260,6 +1264,8 @@ def test_pending_writes_resume(
            },
        },
        metadata={
+            "checkpoint_id": AnyStr(),
+            "checkpoint_ns": "",
            "parents": {},
            "step": 0,
            "source": "loop",
@@ -1301,6 +1307,7 @@ def test_pending_writes_resume(
            "channel_values": {"__start__": {"value": 1}},
        },
        metadata={
+            "checkpoint_ns": "",
            "parents": {},
            "step": -1,
            "source": "input",
@@ -2391,6 +2398,8 @@ def test_in_one_fan_out_state_graph_waiting_edge(
        },
        created_at=AnyStr(),
        metadata={
+            "checkpoint_id": AnyStr(),
+            "checkpoint_ns": "",
            "parents": {},
            "source": "update",
            "step": 4,
@@ -4826,6 +4835,8 @@ def test_parent_command(request: pytest.FixtureRequest, checkpointer_name: str)
            }
        },
        metadata={
+            "checkpoint_id": AnyStr(),
+            "checkpoint_ns": "",
            "source": "loop",
            "writes": {
                "alice": {
@@ -606,6 +606,8 @@ async def test_dynamic_interrupt(checkpointer_name: str) -> None:
        if "shallow" not in checkpointer_name:
            assert [c.metadata async for c in tool_two.checkpointer.alist(thread1)] == [
                {
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "step": 0,
@@ -613,6 +615,7 @@ async def test_dynamic_interrupt(checkpointer_name: str) -> None:
                    "thread_id": "1",
                },
                {
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "input",
                    "step": -1,
@@ -641,6 +644,8 @@ async def test_dynamic_interrupt(checkpointer_name: str) -> None:
            config=tup.config,
            created_at=tup.checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 0,
@@ -667,6 +672,8 @@ async def test_dynamic_interrupt(checkpointer_name: str) -> None:
            config=tup.config,
            created_at=tup.checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 1,
@@ -791,6 +798,8 @@ async def test_dynamic_interrupt_subgraph(checkpointer_name: str) -> None:
                c.metadata async for c in tool_two.checkpointer.alist(thread1root)
            ] == [
                {
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "step": 0,
@@ -798,6 +807,7 @@ async def test_dynamic_interrupt_subgraph(checkpointer_name: str) -> None:
                    "thread_id": "1",
                },
                {
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "input",
                    "step": -1,
@@ -832,6 +842,8 @@ async def test_dynamic_interrupt_subgraph(checkpointer_name: str) -> None:
            config=tup.config,
            created_at=tup.checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 0,
@@ -858,6 +870,8 @@ async def test_dynamic_interrupt_subgraph(checkpointer_name: str) -> None:
            config=tup.config,
            created_at=tup.checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "update",
                "step": 1,
@@ -976,6 +990,8 @@ async def test_copy_checkpoint(checkpointer_name: str) -> None:
        if "shallow" not in checkpointer_name:
            assert [c.metadata async for c in tool_two.checkpointer.alist(thread1)] == [
                {
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "loop",
                    "step": 0,
@@ -983,6 +999,7 @@ async def test_copy_checkpoint(checkpointer_name: str) -> None:
                    "thread_id": "1",
                },
                {
+                    "checkpoint_ns": "",
                    "parents": {},
                    "source": "input",
                    "step": -1,
@@ -1021,6 +1038,8 @@ async def test_copy_checkpoint(checkpointer_name: str) -> None:
            config=tup.config,
            created_at=tup.checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 0,
@@ -1064,6 +1083,8 @@ async def test_copy_checkpoint(checkpointer_name: str) -> None:
            config=tup.config,
            created_at=tup.checkpoint["ts"],
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "fork",
                "step": 1,
@@ -1230,6 +1251,8 @@ async def test_cancel_graph_astream(checkpointer_name: str) -> None:
            assert state.values == {"value": 3}  # 1 + 2
            assert state.next == ("aparallelwhile",)
            assert state.metadata == {
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 0,
@@ -1307,6 +1330,8 @@ async def test_cancel_graph_astream_events_v2(checkpointer_name: Optional[str])
            assert state.values == {"value": 2}
            assert state.next == ("awhile",)
            assert state.metadata == {
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "step": 1,
@@ -2007,6 +2032,8 @@ async def test_pending_writes_resume(
            ),
        )
        assert state.metadata == {
+            "checkpoint_id": AnyStr(),
+            "checkpoint_ns": "",
            "parents": {},
            "source": "loop",
            "step": 0,
@@ -2105,6 +2132,8 @@ async def test_pending_writes_resume(
                "channel_values": {"one": "one", "two": "two", "value": 6},
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "step": 1,
                "source": "loop",
@@ -2157,6 +2186,8 @@ async def test_pending_writes_resume(
                },
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "step": 0,
                "source": "loop",
@@ -2200,6 +2231,7 @@ async def test_pending_writes_resume(
                "channel_values": {"__start__": {"value": 1}},
            },
            metadata={
+                "checkpoint_ns": "",
                "parents": {},
                "step": -1,
                "source": "input",
@@ -2778,6 +2810,8 @@ async def test_send_dedupe_on_resume(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "source": "loop",
                    "writes": {"3": ["3"]},
                    "thread_id": "1",
@@ -2814,6 +2848,8 @@ async def test_send_dedupe_on_resume(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "source": "loop",
                    "writes": {"2": ["2|3"], "3": ["3"], "flaky": ["flaky|4"]},
                    "thread_id": "1",
@@ -2857,6 +2893,8 @@ async def test_send_dedupe_on_resume(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "source": "loop",
                    "writes": {
                        "2": [
@@ -2922,6 +2960,8 @@ async def test_send_dedupe_on_resume(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "source": "loop",
                    "writes": {"1": ["1"]},
                    "thread_id": "1",
@@ -2977,6 +3017,8 @@ async def test_send_dedupe_on_resume(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_id": AnyStr(),
+                    "checkpoint_ns": "",
                    "source": "loop",
                    "writes": None,
                    "thread_id": "1",
@@ -3014,6 +3056,7 @@ async def test_send_dedupe_on_resume(checkpointer_name: str) -> None:
                    }
                },
                metadata={
+                    "checkpoint_ns": "",
                    "source": "input",
                    "writes": {"__start__": ["0"]},
                    "thread_id": "1",
@@ -3186,6 +3229,8 @@ async def test_send_react_interrupt(checkpointer_name: str) -> None:
                }
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "step": 1,
                "source": "loop",
                "writes": {
@@ -3257,6 +3302,8 @@ async def test_send_react_interrupt(checkpointer_name: str) -> None:
                }
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "step": 2,
                "source": "update",
                "writes": {
@@ -3344,6 +3391,8 @@ async def test_send_react_interrupt(checkpointer_name: str) -> None:
                }
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "step": 1,
                "source": "loop",
                "writes": {
@@ -3436,6 +3485,8 @@ async def test_send_react_interrupt(checkpointer_name: str) -> None:
                }
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "step": 2,
                "source": "update",
                "writes": {
@@ -3651,6 +3702,8 @@ async def test_send_react_interrupt_control(
                }
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "step": 1,
                "source": "loop",
                "writes": {
@@ -3722,6 +3775,8 @@ async def test_send_react_interrupt_control(
                }
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "step": 2,
                "source": "update",
                "writes": {
@@ -4671,6 +4726,8 @@ async def test_in_one_fan_out_state_graph_waiting_edge_custom_state_class(
                }
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "parents": {},
                "source": "loop",
                "writes": {"qa": {"answer": "doc1,doc2,doc3,doc4"}},
@@ -6145,6 +6202,8 @@ async def test_parent_command(checkpointer_name: str) -> None:
                }
            },
            metadata={
+                "checkpoint_id": AnyStr(),
+                "checkpoint_ns": "",
                "source": "loop",
                "writes": {
                    "alice": {
Author	SHA1	Message	Date
Ben Burns	b9f26d28ea	add missing build step	2025-02-13 23:03:58 -08:00
Ben Burns	06d4ba7fa0	bump prebuilt.md	2025-02-13 22:57:15 -08:00
Ben Burns	2dbdb36743	fix cassette hashing	2025-02-13 22:56:16 -08:00
Ben Burns	2b72fbd5de	state-reducers.md WIP	2025-02-13 21:57:45 -08:00
Ben Burns	c681545c97	make CodeQL happy, fix Makefile	2025-02-13 21:57:44 -08:00
Ben Burns	5ebdefba63	ts exec works - back out half-baked TS code snippet for now	2025-02-13 21:57:44 -08:00
Ben Burns	441923282c	docs: get ts execution working in build pipeline	2025-02-13 21:57:37 -08:00