checkpoint: Fix memory leak in checkpoint serializer (#3481)

- Packer instances can't be kept in memory as they retain memory after
used

docs/_scripts/third_party_page/packages.yml

@@ -9,3 +9,6 @@ packages:
   - name: "langgraph-supervisor"
     repo: "langchain-ai/langgraph-supervisor"
     description: "Build supervisor multi-agent systems with LangGraph"
+  - name: "langmem"
+    repo: "langchain-ai/langmem"
+    description: "Build agents that learn and adapt from interactions over time."

docs/docs/cloud/how-tos/use_stream_react.md

@@ -15,7 +15,7 @@ Key features:
 
 Let's explore how to use `useStream()` in your React application.
 
-The `useStream()` provides a solid foundation for creating bespoke chat experiences. For pre-built chat components and interfaces, we recommend checking out [CopilotKit](https://docs.copilotkit.ai/coagents/quickstart/langgraph) and [assistant-ui](https://github.com/langchain-ai/assistant-ui).
+The `useStream()` provides a solid foundation for creating bespoke chat experiences. For pre-built chat components and interfaces, we recommend checking out [CopilotKit](https://docs.copilotkit.ai/coagents/quickstart/langgraph) and [assistant-ui](https://www.assistant-ui.com/docs/runtimes/langgraph).
 
 ## Example
 

docs/docs/concepts/functional_api.md

@@ -1,8 +1,5 @@
 # Functional API
 
-!!! warning "Beta"
-    The Functional API is currently in **beta** and is subject to change. Please [report any issues](https://github.com/langchain-ai/langgraph/issues) or feedback to the LangGraph team.
-
 ## Overview
 
 The **Functional API** allows you to add LangGraph's key features -- [persistence](./persistence.md), [memory](./memory.md), [human-in-the-loop](./human_in_the_loop.md), and [streaming](./streaming.md) — to your applications with minimal changes to your existing code.

docs/docs/concepts/index.md

@@ -28,7 +28,7 @@ The conceptual guide does not cover step-by-step instructions or specific implem
 - [Persistence](persistence.md): LangGraph has a built-in persistence layer, implemented through checkpointers. This persistence layer helps to support powerful capabilities like human-in-the-loop, memory, time travel, and fault-tolerance.
 - [Memory](memory.md): Memory in AI applications refers to the ability to process, store, and effectively recall information from past interactions. With memory, your agents can learn from feedback and adapt to users' preferences.
 - [Streaming](streaming.md): Streaming is crucial for enhancing the responsiveness of applications built on LLMs. By displaying output progressively, even before a complete response is ready, streaming significantly improves user experience (UX), particularly when dealing with the latency of LLMs.
-- [Functional API (beta)](functional_api.md): An alternative to [Graph API (StateGraph)](low_level.md#stategraph) for development in LangGraph.
+- [Functional API](functional_api.md): `@entrypoint` and `@task` decorators that allow you to add LangGraph functionality to an existing codebase.
 - [FAQ](faq.md): Frequently asked questions about LangGraph.
 
 ## LangGraph Platform

docs/docs/how-tos/index.md

@@ -39,8 +39,7 @@ execution of your graph.
 - [How to use MongoDB checkpointer for persistence](persistence_mongodb.ipynb)
 - [How to create a custom checkpointer using Redis](persistence_redis.ipynb)
 
-See the below guides for how-to add persistence to your workflow using the (beta)
-[Functional API](../concepts/functional_api.md):
+See the below guides for how-to add persistence to your workflow using the [Functional API](../concepts/functional_api.md):
 
 - [How to add thread-level persistence (functional API)](persistence-functional.ipynb)
 - [How to add cross-thread persistence (functional API)](cross-thread-persistence-functional.ipynb)
@@ -73,7 +72,7 @@ Other methods:
 - [How to edit graph state](human_in_the_loop/edit-graph-state.ipynb): Edit graph state using `graph.update_state` method. Use this if implementing a **human-in-the-loop** workflow via **static breakpoints**.
 - [How to add dynamic breakpoints with `NodeInterrupt`](human_in_the_loop/dynamic_breakpoints.ipynb): **Not recommended**: Use the [`interrupt` function](../concepts/human_in_the_loop.md) instead.
 
-See the below guides for how-to implement human-in-the-loop workflows with the (beta)
+See the below guides for how-to implement human-in-the-loop workflows with the
 [Functional API](../concepts/functional_api.md):
 
 - [How to wait for user input (Functional API)](wait-user-input-functional.ipynb)
@@ -130,8 +129,7 @@ These how-to guides show common patterns for tool calling with LangGraph:
 
 See the [multi-agent tutorials](../tutorials/index.md#multi-agent-systems) for implementations of other multi-agent architectures.
 
-See the below guides for how to implement multi-agent workflows with the (beta)
-[Functional API](../concepts/functional_api.md):
+See the below guides for how to implement multi-agent workflows with the [Functional API](../concepts/functional_api.md):
 
 - [How to build a multi-agent network (functional API)](multi-agent-network-functional.ipynb)
 - [How to add multi-turn conversation in a multi-agent application (functional API)](multi-agent-multi-turn-convo-functional.ipynb)
@@ -149,8 +147,7 @@ See the below guides for how to implement multi-agent workflows with the (beta)
 - [How to pass custom LangSmith run ID for graph runs](run-id-langsmith.ipynb)
 - [How to integrate LangGraph with AutoGen, CrewAI, and other frameworks](autogen-integration.ipynb)
 
-See the below guide for how to integrate with other frameworks using the (beta)
-[Functional API](../concepts/functional_api.md):
+See the below guide for how to integrate with other frameworks using the [Functional API](../concepts/functional_api.md):
 
 - [How to integrate LangGraph (functional API) with AutoGen, CrewAI, and other frameworks](autogen-integration-functional.ipynb)
 
@@ -174,8 +171,7 @@ overview of its underlying implementation to help you customize for your own nee
 
 - [How to create prebuilt ReAct agent from scratch](react-agent-from-scratch.ipynb)
 
-See the below guide for how-to build ReAct agents with the (beta)
-[Functional API](../concepts/functional_api.md):
+See the below guide for how-to build ReAct agents with the [Functional API](../concepts/functional_api.md):
 
 - [How to create a ReAct agent from scratch (Functional API)](react-agent-from-scratch-functional.ipynb)
 

libs/checkpoint-postgres/poetry.lock

@@ -348,7 +348,7 @@ typing-extensions = ">=4.7"
 
 [[package]]
 name = "langgraph-checkpoint"
-version = "2.0.10"
+version = "2.0.15"
 description = "Library with base interfaces for LangGraph checkpoint savers."
 optional = false
 python-versions = "^3.9.0,<4.0"
@@ -1255,4 +1255,4 @@ watchmedo = ["PyYAML (>=3.10)"]
 [metadata]
 lock-version = "2.1"
 python-versions = "^3.9.0,<4.0"
-content-hash = "ca18ea480430714e47685ff0f25da9990dbbdecd38e9c1a3ca41d42be7069067"
+content-hash = "369bfffecb9489835b43b8255932e043176a11d2f639aad2d055ffd89263ca1e"

libs/checkpoint-postgres/pyproject.toml

@@ -10,7 +10,7 @@ packages = [{ include = "langgraph" }]
 
 [tool.poetry.dependencies]
 python = "^3.9.0,<4.0"
-langgraph-checkpoint = "^2.0.10"
+langgraph-checkpoint = "^2.0.15"
 orjson = ">=3.10.1"
 psycopg = "^3.2.0"
 psycopg-pool = "^3.2.0"

libs/checkpoint-sqlite/poetry.lock

@@ -350,7 +350,7 @@ typing-extensions = ">=4.7"
 
 [[package]]
 name = "langgraph-checkpoint"
-version = "2.0.10"
+version = "2.0.15"
 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 = "03c697eae6f550f3c7e29f1d61f4c409dabe04ae8d43281728e549174d2fc670"
+content-hash = "e6d3ca9bce723c05f4c5ae9dc4bee872f7581b7763680b34112f1d280f5a9b0a"

libs/checkpoint-sqlite/pyproject.toml

@@ -10,7 +10,7 @@ packages = [{ include = "langgraph" }]
 
 [tool.poetry.dependencies]
 python = "^3.9.0"
-langgraph-checkpoint = "^2.0.10"
+langgraph-checkpoint = "^2.0.15"
 aiosqlite = "^0.20.0"
 
 [tool.poetry.group.dev.dependencies]

libs/checkpoint/langgraph/checkpoint/serde/jsonplus.py

@@ -5,6 +5,7 @@ import json
 import pathlib
 import re
 from collections import deque
+from collections.abc import Sequence
 from datetime import date, datetime, time, timedelta, timezone
 from enum import Enum
 from inspect import isclass
@@ -16,7 +17,7 @@ from ipaddress import (
     IPv6Interface,
     IPv6Network,
 )
-from typing import Any, Callable, Optional, Sequence, Union, cast
+from typing import Any, Callable, Optional, Union, cast
 from uuid import UUID
 
 import msgpack  # type: ignore[import-untyped]
@@ -502,15 +503,5 @@ def _msgpack_ext_hook(code: int, data: bytes) -> Any:
             return
 
 
-ENC_POOL: deque[msgpack.Packer] = deque(maxlen=32)
-
-
 def _msgpack_enc(data: Any) -> bytes:
-    try:
-        enc = ENC_POOL.popleft()
-    except IndexError:
-        enc = msgpack.Packer(default=_msgpack_default)
-    try:
-        return enc.pack(data)
-    finally:
-        ENC_POOL.append(enc)
+    return msgpack.packb(data, default=_msgpack_default)

libs/checkpoint/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-checkpoint"
-version = "2.0.15"
+version = "2.0.16"
 description = "Library with base interfaces for LangGraph checkpoint savers."
 authors = []
 license = "MIT"

libs/cli/langgraph_cli/cli.py

@@ -303,7 +303,15 @@ def _build(
         tag,
     ]
     # apply config
-    stdin = langgraph_cli.config.config_to_docker(config, config_json, base_image)
+    stdin, additional_contexts = langgraph_cli.config.config_to_docker(
+        config, config_json, base_image
+    )
+    # add additional_contexts
+    if additional_contexts:
+        additional_contexts_str = ",".join(
+            f"{k}={v}" for k, v in additional_contexts.items()
+        )
+        args.extend(["--build-context", additional_contexts_str])
     # run docker build
     runner.run(
         subp_exec(
@@ -439,20 +447,28 @@ def dockerfile(save_path: str, config: pathlib.Path, add_docker_compose: bool) -
     secho("✅ Configuration validated!", fg="green")
 
     secho(f"📝 Generating Dockerfile at {save_path}", fg="yellow")
+    dockerfile, additional_contexts = langgraph_cli.config.config_to_docker(
+        config,
+        config_json,
+        (
+            "langchain/langgraphjs-api"
+            if config_json.get("node_version")
+            else "langchain/langgraph-api"
+        ),
+    )
     with open(str(save_path), "w", encoding="utf-8") as f:
-        f.write(
-            langgraph_cli.config.config_to_docker(
-                config,
-                config_json,
-                (
-                    "langchain/langgraphjs-api"
-                    if config_json.get("node_version")
-                    else "langchain/langgraph-api"
-                ),
-            )
-        )
+        f.write(dockerfile)
     secho("✅ Created: Dockerfile", fg="green")
 
+    if additional_contexts:
+        additional_contexts_str = ",".join(
+            f"{k}={v}" for k, v in additional_contexts.items()
+        )
+        secho(
+            f"""📝 Run docker build with these additional build contexts `--build-context {additional_contexts_str}`""",
+            fg="yellow",
+        )
+
     if add_docker_compose:
         # Add docker compose and related files
         # Add .dockerignore file in the same directory as the Dockerfile
@@ -575,7 +591,7 @@ def dev(
 ):
     """CLI entrypoint for running the LangGraph API server."""
     try:
-        from langgraph_api.cli import run_server
+        from langgraph_api.cli import run_server  # type: ignore
     except ImportError:
         py_version_msg = ""
         if sys.version_info < (3, 11):
@@ -662,6 +678,7 @@ def prepare_args_and_stdin(
     debugger_base_url: Optional[str] = None,
     postgres_uri: Optional[str] = None,
 ) -> Tuple[List[str], str]:
+    assert config_path.exists(), f"Config file not found: {config_path}"
     # prepare args
     stdin = langgraph_cli.docker.compose(
         capabilities,

libs/cli/langgraph_cli/config.py

@@ -2,6 +2,7 @@ import json
 import os
 import pathlib
 import textwrap
+from collections import Counter
 from typing import NamedTuple, Optional, TypedDict, Union
 
 import click
@@ -294,10 +295,10 @@ class LocalDeps(NamedTuple):
             tuples. Each entry points to a local `requirements.txt` file and where
             it should be placed inside the Docker container before running `pip install`.
 
-        real_pkgs: A dictionary mapping a local directory path (host side) to the
-            same dependency string from the config. These directories contain the
-            necessary files (e.g., `pyproject.toml` or `setup.py`) to be installed
-            as a standard Python package with pip.
+        real_pkgs: A dictionary mapping a local directory path (host side) to a
+            tuple of (dependency_string, container_package_path). These directories
+            contain the necessary files (e.g., `pyproject.toml` or `setup.py`) to be
+            installed as a standard Python package with pip.
 
         faux_pkgs: A dictionary mapping a local directory path (host side) to a
             tuple of (dependency_string, container_package_path). For these
@@ -310,16 +311,23 @@ class LocalDeps(NamedTuple):
             directory. If the local dependency `"."` is present in the config, this
             field captures the path where that dependency will appear in the
             container (e.g., `/deps/<name>` or similar). Otherwise, it may be `None`.
+
+        additional_contexts: A list of paths to directories that contain local
+            dependencies in parent directories. These directories are added to the
+            Docker build context to ensure that the Dockerfile can access them.
     """
 
-    pip_reqs: list[tuple[str, str]]
-    real_pkgs: dict[pathlib.Path, str]
+    pip_reqs: list[tuple[pathlib.Path, str]]
+    real_pkgs: dict[pathlib.Path, tuple[str, str]]
     faux_pkgs: dict[pathlib.Path, tuple[str, str]]
     # if . is in dependencies, use it as working_dir
     working_dir: Optional[str] = None
+    # if there are local dependencies in parent directories, use additional_contexts
+    additional_contexts: list[pathlib.Path] = None
 
 
 def _assemble_local_deps(config_path: pathlib.Path, config: Config) -> LocalDeps:
+    config_path = config_path.resolve()
     # ensure reserved package names are not used
     reserved = {
         "src",
@@ -336,6 +344,7 @@ def _assemble_local_deps(config_path: pathlib.Path, config: Config) -> LocalDeps
         "httpx",
         "langsmith",
     }
+    counter = Counter()
 
     def check_reserved(name: str, ref: str):
         if name in reserved:
@@ -348,7 +357,8 @@ def _assemble_local_deps(config_path: pathlib.Path, config: Config) -> LocalDeps
     pip_reqs = []
     real_pkgs = {}
     faux_pkgs = {}
-    working_dir = None
+    working_dir: Optional[str] = None
+    additional_contexts: list[pathlib.Path] = []
 
     for local_dep in config["dependencies"]:
         if not local_dep.startswith("."):
@@ -357,7 +367,7 @@ def _assemble_local_deps(config_path: pathlib.Path, config: Config) -> LocalDeps
 
         # Verify that the local dependency can be resolved
         # (e.g., this would raise an informative error if a user mistyped a path).
-        resolved = config_path.parent / local_dep
+        resolved = (config_path.parent / local_dep).resolve()
 
         # validate local dependency
         if not resolved.exists():
@@ -366,25 +376,28 @@ def _assemble_local_deps(config_path: pathlib.Path, config: Config) -> LocalDeps
             raise NotADirectoryError(
                 f"Local dependency must be a directory: {resolved}"
             )
-        elif not resolved.is_relative_to(config_path.parent):
-            raise ValueError(
-                f"Local dependency '{resolved}' must be a subdirectory of '{config_path.parent}'"
-            )
+        elif resolved == config_path.parent:
+            pass
+        elif config_path.parent not in resolved.parents:
+            additional_contexts.append(resolved)
 
         # Check for pyproject.toml or setup.py
         # If found, treat as a real package, if not treat as a faux package.
         # For faux packages, we'll also check for presence of requirements.txt.
         files = os.listdir(resolved)
-        if "pyproject.toml" in files:
+        if "pyproject.toml" in files or "setup.py" in files:
             # real package
-            real_pkgs[resolved] = local_dep
+
+            # assign a unique folder name
+            container_name = resolved.name
+            if counter[container_name] > 0:
+                container_name += f"_{counter[container_name]}"
+            counter[container_name] += 1
+            # add to deps
+            real_pkgs[resolved] = (local_dep, container_name)
+            # set working_dir
             if local_dep == ".":
-                working_dir = f"/deps/{resolved.name}"
-        elif "setup.py" in files:
-            # real package
-            real_pkgs[resolved] = local_dep
-            if local_dep == ".":
-                working_dir = f"/deps/{resolved.name}"
+                working_dir = f"/deps/{container_name}"
         else:
             # We could not find a pyproject.toml or setup.py, so treat as a faux package
             if any(file == "__init__.py" for file in files):
@@ -423,12 +436,12 @@ def _assemble_local_deps(config_path: pathlib.Path, config: Config) -> LocalDeps
                 rfile = resolved / "requirements.txt"
                 pip_reqs.append(
                     (
-                        rfile.relative_to(config_path.parent).as_posix(),
+                        rfile,
                         f"{container_path}/requirements.txt",
                     )
                 )
 
-    return LocalDeps(pip_reqs, real_pkgs, faux_pkgs, working_dir)
+    return LocalDeps(pip_reqs, real_pkgs, faux_pkgs, working_dir, additional_contexts)
 
 
 def _update_graph_paths(
@@ -556,7 +569,7 @@ def _update_auth_path(
 
 def python_config_to_docker(
     config_path: pathlib.Path, config: Config, base_image: str
-) -> str:
+) -> tuple[str, dict[str, str]]:
     """Generate a Dockerfile from the configuration."""
     # configure pip
     pip_install = (
@@ -581,9 +594,15 @@ def python_config_to_docker(
     pip_pkgs_str = f"RUN {pip_install} {' '.join(pypi_deps)}" if pypi_deps else ""
     if local_deps.pip_reqs:
         pip_reqs_str = os.linesep.join(
-            f"ADD {reqpath} {destpath}" for reqpath, destpath in local_deps.pip_reqs
+            f"COPY --from=__outer_{reqpath.name} requirements.txt {destpath}"
+            if reqpath.parent in local_deps.additional_contexts
+            else f"ADD {reqpath.relative_to(config_path.parent)} {destpath}"
+            for reqpath, destpath in local_deps.pip_reqs
         )
         pip_reqs_str += f'{os.linesep}RUN {pip_install} {" ".join("-r " + r for _,r in local_deps.pip_reqs)}'
+        pip_reqs_str = f"""# -- Installing local requirements --
+{pip_reqs_str}
+# -- End of local requirements install --"""
 
     else:
         pip_reqs_str = ""
@@ -591,7 +610,14 @@ def python_config_to_docker(
     # https://setuptools.pypa.io/en/latest/userguide/datafiles.html#package-data
     # https://til.simonwillison.net/python/pyproject
     faux_pkgs_str = f"{os.linesep}{os.linesep}".join(
-        f"""ADD {relpath} {destpath}
+        (
+            f"""# -- Adding non-package dependency {fullpath.name} --
+COPY --from=__outer_{fullpath.name} . {destpath}"""
+            if fullpath in local_deps.additional_contexts
+            else f"""# -- Adding non-package dependency {fullpath.name} --
+ADD {relpath} {destpath}"""
+        )
+        + f"""
 RUN set -ex && \\
     for line in '[project]' \\
                 'name = "{fullpath.name}"' \\
@@ -599,12 +625,20 @@ RUN set -ex && \\
                 '[tool.setuptools.package-data]' \\
                 '"*" = ["**/*"]'; do \\
         echo "$line" >> /deps/__outer_{fullpath.name}/pyproject.toml; \\
-    done"""
+    done
+# -- End of non-package dependency {fullpath.name} --"""
         for fullpath, (relpath, destpath) in local_deps.faux_pkgs.items()
     )
+
     local_pkgs_str = os.linesep.join(
-        f"ADD {relpath} /deps/{fullpath.name}"
-        for fullpath, relpath in local_deps.real_pkgs.items()
+        f"""# -- Adding local package {relpath} --
+COPY --from={name} . /deps/{name}
+# -- End of local package {relpath} --"""
+        if fullpath in local_deps.additional_contexts
+        else f"""# -- Adding local package {relpath} --
+ADD {relpath} /deps/{name}
+# -- End of local package {relpath} --"""
+        for fullpath, (relpath, name) in local_deps.real_pkgs.items()
     )
 
     installs = f"{os.linesep}{os.linesep}".join(
@@ -638,15 +672,30 @@ RUN set -ex && \\
         "",
         installs,
         "",
+        "# -- Installing all local dependencies --",
         f"RUN {pip_install} -e /deps/*",
+        "# -- End of local dependencies install --",
         os.linesep.join(env_vars),
         "",
         f"WORKDIR {local_deps.working_dir}" if local_deps.working_dir else "",
     ]
-    return os.linesep.join(docker_file_contents)
+
+    additional_contexts: dict[str, str] = {}
+    for p in local_deps.additional_contexts:
+        if p in local_deps.real_pkgs:
+            name = local_deps.real_pkgs[p][1]
+        elif p in local_deps.faux_pkgs:
+            name = f"__outer_{p.name}"
+        else:
+            raise RuntimeError(f"Unknown additional context: {p}")
+        additional_contexts[name] = str(p)
+
+    return os.linesep.join(docker_file_contents), additional_contexts
 
 
-def node_config_to_docker(config_path: pathlib.Path, config: Config, base_image: str):
+def node_config_to_docker(
+    config_path: pathlib.Path, config: Config, base_image: str
+) -> tuple[str, dict[str, str]]:
     faux_path = f"/deps/{config_path.parent.name}"
 
     def test_file(file_name):
@@ -686,7 +735,8 @@ ENV LANGGRAPH_STORE='{json.dumps(store_config)}'
 ENV LANGGRAPH_AUTH='{json.dumps(auth_config)}'
 """
 
-    return f"""FROM {base_image}:{config['node_version']}
+    return (
+        f"""FROM {base_image}:{config['node_version']}
 
 {os.linesep.join(config["dockerfile_lines"])}
 
@@ -698,10 +748,14 @@ ENV LANGSERVE_GRAPHS='{json.dumps(config["graphs"])}'
 
 WORKDIR {faux_path}
 
-RUN (test ! -f /api/langgraph_api/js/build.mts && echo "Prebuild script not found, skipping") || tsx /api/langgraph_api/js/build.mts"""
+RUN (test ! -f /api/langgraph_api/js/build.mts && echo "Prebuild script not found, skipping") || tsx /api/langgraph_api/js/build.mts""",
+        {},
+    )
 
 
-def config_to_docker(config_path: pathlib.Path, config: Config, base_image: str):
+def config_to_docker(
+    config_path: pathlib.Path, config: Config, base_image: str
+) -> tuple[str, dict[str, str]]:
     if config.get("node_version"):
         return node_config_to_docker(config_path, config, base_image)
 
@@ -737,13 +791,24 @@ def config_to_compose(
     else:
         watch_str = ""
 
+    dockerfile, additional_contexts = config_to_docker(config_path, config, base_image)
+
+    additional_contexts_str = "\n".join(
+        f"                - {name}: {path}"
+        for name, path in additional_contexts.items()
+    )
+    if additional_contexts_str:
+        additional_contexts_str = f"""
+            additional_contexts:
+{additional_contexts_str}"""
+
     return f"""
 {textwrap.indent(env_vars_str, "            ")}
         {env_file_str}
         pull_policy: build
         build:
-            context: .
+            context: .{additional_contexts_str}
             dockerfile_inline: |
-{textwrap.indent(config_to_docker(config_path, config, base_image), "                ")}
+{textwrap.indent(dockerfile, "                ")}
         {watch_str}
 """

libs/cli/langgraph_cli/docker.py

@@ -49,7 +49,9 @@ def check_capabilities(runner) -> DockerCapabilities:
         raise click.UsageError("Docker not installed") from None
 
     try:
-        stdout, _ = runner.run(subp_exec("docker", "info", "-f", "json", collect=True))
+        stdout, _ = runner.run(
+            subp_exec("docker", "info", "-f", "{{json .}}", collect=True)
+        )
         info = json.loads(stdout)
     except (click.exceptions.Exit, json.JSONDecodeError):
         raise click.UsageError("Docker not installed or not running") from None

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

@@ -40,9 +40,9 @@ def temporary_config_folder(config_content: dict):
 
 def test_prepare_args_and_stdin() -> None:
     # this basically serves as an end-to-end test for using config and docker helpers
-    config_path = pathlib.Path("./langgraph.json")
+    config_path = pathlib.Path(__file__).parent / "langgraph.json"
     config = validate_config(
-        Config(dependencies=["."], graphs={"agent": "agent.py:graph"})
+        Config(dependencies=[".", "../../.."], graphs={"agent": "agent.py:graph"})
     )
     port = 8000
     debugger_port = 8001
@@ -61,7 +61,7 @@ def test_prepare_args_and_stdin() -> None:
 
     expected_args = [
         "--project-directory",
-        ".",
+        str(pathlib.Path(__file__).parent.absolute()),
         "-f",
         "custom-docker-compose.yml",
         "-f",
@@ -129,18 +129,29 @@ services:
         pull_policy: build
         build:
             context: .
+            additional_contexts:
+                - cli_1: {str(pathlib.Path(__file__).parent.parent.parent.parent.absolute())}
             dockerfile_inline: |
                 FROM langchain/langgraph-api:3.11
-                ADD . /deps/
+                # -- Adding local package . --
+                ADD . /deps/cli
+                # -- End of local package . --
+                # -- Adding local package ../../.. --
+                COPY --from=cli_1 . /deps/cli_1
+                # -- End of local package ../../.. --
+                # -- Installing all local dependencies --
                 RUN PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -c /api/constraints.txt -e /deps/*
+                # -- End of local dependencies install --
                 ENV LANGSERVE_GRAPHS='{{"agent": "agent.py:graph"}}'
-                WORKDIR /deps/
+                WORKDIR /deps/cli
         
         develop:
             watch:
                 - path: langgraph.json
                   action: rebuild
                 - path: .
+                  action: rebuild
+                - path: ../../..
                   action: rebuild\
 """
     assert actual_args == expected_args

libs/cli/tests/unit_tests/test_config.py

@@ -177,13 +177,20 @@ def test_validate_config_file():
 # config_to_docker
 def test_config_to_docker_simple():
     graphs = {"agent": "./agent.py:graph"}
-    actual_docker_stdin = config_to_docker(
+    actual_docker_stdin, additional_contexts = config_to_docker(
         PATH_TO_CONFIG,
-        validate_config({"dependencies": ["."], "graphs": graphs}),
+        validate_config(
+            {"dependencies": [".", "../../examples/graphs_reqs_a"], "graphs": graphs}
+        ),
         "langchain/langgraph-api",
     )
     expected_docker_stdin = """\
 FROM langchain/langgraph-api:3.11
+# -- Installing local requirements --
+COPY --from=__outer_requirements.txt requirements.txt /deps/__outer_graphs_reqs_a/graphs_reqs_a/requirements.txt
+RUN PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -c /api/constraints.txt -r /deps/__outer_graphs_reqs_a/graphs_reqs_a/requirements.txt
+# -- End of local requirements install --
+# -- Adding non-package dependency unit_tests --
 ADD . /deps/__outer_unit_tests/unit_tests
 RUN set -ex && \\
     for line in '[project]' \\
@@ -193,16 +200,79 @@ RUN set -ex && \\
                 '"*" = ["**/*"]'; do \\
         echo "$line" >> /deps/__outer_unit_tests/pyproject.toml; \\
     done
+# -- End of non-package dependency unit_tests --
+# -- Adding non-package dependency graphs_reqs_a --
+COPY --from=__outer_graphs_reqs_a . /deps/__outer_graphs_reqs_a/graphs_reqs_a
+RUN set -ex && \\
+    for line in '[project]' \\
+                'name = "graphs_reqs_a"' \\
+                'version = "0.1"' \\
+                '[tool.setuptools.package-data]' \\
+                '"*" = ["**/*"]'; do \\
+        echo "$line" >> /deps/__outer_graphs_reqs_a/pyproject.toml; \\
+    done
+# -- End of non-package dependency graphs_reqs_a --
+# -- Installing all local dependencies --
 RUN PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -c /api/constraints.txt -e /deps/*
+# -- End of local dependencies install --
 ENV LANGSERVE_GRAPHS='{"agent": "/deps/__outer_unit_tests/unit_tests/agent.py:graph"}'
 WORKDIR /deps/__outer_unit_tests/unit_tests\
 """
     assert clean_empty_lines(actual_docker_stdin) == expected_docker_stdin
 
+    assert additional_contexts == {
+        "__outer_graphs_reqs_a": str(
+            (pathlib.Path(__file__).parent / "../../examples/graphs_reqs_a").resolve()
+        )
+    }
+
+
+def test_config_to_docker_outside_path():
+    graphs = {"agent": "./agent.py:graph"}
+    actual_docker_stdin, additional_contexts = config_to_docker(
+        PATH_TO_CONFIG,
+        validate_config({"dependencies": [".", ".."], "graphs": graphs}),
+        "langchain/langgraph-api",
+    )
+    expected_docker_stdin = """\
+FROM langchain/langgraph-api:3.11
+# -- Adding non-package dependency unit_tests --
+ADD . /deps/__outer_unit_tests/unit_tests
+RUN set -ex && \\
+    for line in '[project]' \\
+                'name = "unit_tests"' \\
+                'version = "0.1"' \\
+                '[tool.setuptools.package-data]' \\
+                '"*" = ["**/*"]'; do \\
+        echo "$line" >> /deps/__outer_unit_tests/pyproject.toml; \\
+    done
+# -- End of non-package dependency unit_tests --
+# -- Adding non-package dependency tests --
+COPY --from=__outer_tests . /deps/__outer_tests/tests
+RUN set -ex && \\
+    for line in '[project]' \\
+                'name = "tests"' \\
+                'version = "0.1"' \\
+                '[tool.setuptools.package-data]' \\
+                '"*" = ["**/*"]'; do \\
+        echo "$line" >> /deps/__outer_tests/pyproject.toml; \\
+    done
+# -- End of non-package dependency tests --
+# -- Installing all local dependencies --
+RUN PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -c /api/constraints.txt -e /deps/*
+# -- End of local dependencies install --
+ENV LANGSERVE_GRAPHS='{"agent": "/deps/__outer_unit_tests/unit_tests/agent.py:graph"}'
+WORKDIR /deps/__outer_unit_tests/unit_tests\
+"""
+    assert clean_empty_lines(actual_docker_stdin) == expected_docker_stdin
+    assert additional_contexts == {
+        "__outer_tests": str(pathlib.Path(__file__).parent.parent.absolute()),
+    }
+
 
 def test_config_to_docker_pipconfig():
     graphs = {"agent": "./agent.py:graph"}
-    actual_docker_stdin = config_to_docker(
+    actual_docker_stdin, additional_contexts = config_to_docker(
         PATH_TO_CONFIG,
         validate_config(
             {
@@ -216,6 +286,7 @@ def test_config_to_docker_pipconfig():
     expected_docker_stdin = """\
 FROM langchain/langgraph-api:3.11
 ADD pipconfig.txt /pipconfig.txt
+# -- Adding non-package dependency unit_tests --
 ADD . /deps/__outer_unit_tests/unit_tests
 RUN set -ex && \\
     for line in '[project]' \\
@@ -225,11 +296,15 @@ RUN set -ex && \\
                 '"*" = ["**/*"]'; do \\
         echo "$line" >> /deps/__outer_unit_tests/pyproject.toml; \\
     done
+# -- End of non-package dependency unit_tests --
+# -- Installing all local dependencies --
 RUN PIP_CONFIG_FILE=/pipconfig.txt PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -c /api/constraints.txt -e /deps/*
+# -- End of local dependencies install --
 ENV LANGSERVE_GRAPHS='{"agent": "/deps/__outer_unit_tests/unit_tests/agent.py:graph"}'
 WORKDIR /deps/__outer_unit_tests/unit_tests\
 """
     assert clean_empty_lines(actual_docker_stdin) == expected_docker_stdin
+    assert additional_contexts == {}
 
 
 def test_config_to_docker_invalid_inputs():
@@ -254,7 +329,7 @@ def test_config_to_docker_invalid_inputs():
 
 def test_config_to_docker_local_deps():
     graphs = {"agent": "./graphs/agent.py:graph"}
-    actual_docker_stdin = config_to_docker(
+    actual_docker_stdin, additional_contexts = config_to_docker(
         PATH_TO_CONFIG,
         validate_config(
             {
@@ -266,6 +341,7 @@ def test_config_to_docker_local_deps():
     )
     expected_docker_stdin = """\
 FROM langchain/langgraph-api-custom:3.11
+# -- Adding non-package dependency graphs --
 ADD ./graphs /deps/__outer_graphs/src
 RUN set -ex && \\
     for line in '[project]' \\
@@ -275,10 +351,14 @@ RUN set -ex && \\
                 '"*" = ["**/*"]'; do \\
         echo "$line" >> /deps/__outer_graphs/pyproject.toml; \\
     done
+# -- End of non-package dependency graphs --
+# -- Installing all local dependencies --
 RUN PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -c /api/constraints.txt -e /deps/*
+# -- End of local dependencies install --
 ENV LANGSERVE_GRAPHS='{"agent": "/deps/__outer_graphs/src/agent.py:graph"}'\
 """
     assert clean_empty_lines(actual_docker_stdin) == expected_docker_stdin
+    assert additional_contexts == {}
 
 
 def test_config_to_docker_pyproject():
@@ -291,7 +371,7 @@ dependencies = ["langchain"]"""
         f.write(pyproject_str)
 
     graphs = {"agent": "./graphs/agent.py:graph"}
-    actual_docker_stdin = config_to_docker(
+    actual_docker_stdin, additional_contexts = config_to_docker(
         PATH_TO_CONFIG,
         validate_config(
             {
@@ -303,16 +383,21 @@ dependencies = ["langchain"]"""
     )
     os.remove(pyproject_path)
     expected_docker_stdin = """FROM langchain/langgraph-api:3.11
+# -- Adding local package . --
 ADD . /deps/unit_tests
+# -- End of local package . --
+# -- Installing all local dependencies --
 RUN PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -c /api/constraints.txt -e /deps/*
+# -- End of local dependencies install --
 ENV LANGSERVE_GRAPHS='{"agent": "/deps/unit_tests/graphs/agent.py:graph"}'
 WORKDIR /deps/unit_tests"""
     assert clean_empty_lines(actual_docker_stdin) == expected_docker_stdin
+    assert additional_contexts == {}
 
 
 def test_config_to_docker_end_to_end():
     graphs = {"agent": "./graphs/agent.py:graph"}
-    actual_docker_stdin = config_to_docker(
+    actual_docker_stdin, additional_contexts = config_to_docker(
         PATH_TO_CONFIG,
         validate_config(
             {
@@ -330,6 +415,7 @@ ARG meow
 ARG foo
 ADD pipconfig.txt /pipconfig.txt
 RUN PIP_CONFIG_FILE=/pipconfig.txt PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -c /api/constraints.txt langchain langchain_openai
+# -- Adding non-package dependency graphs --
 ADD ./graphs/ /deps/__outer_graphs/src
 RUN set -ex && \\
     for line in '[project]' \\
@@ -339,15 +425,19 @@ RUN set -ex && \\
                 '"*" = ["**/*"]'; do \\
         echo "$line" >> /deps/__outer_graphs/pyproject.toml; \\
     done
+# -- End of non-package dependency graphs --
+# -- Installing all local dependencies --
 RUN PIP_CONFIG_FILE=/pipconfig.txt PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -c /api/constraints.txt -e /deps/*
+# -- End of local dependencies install --
 ENV LANGSERVE_GRAPHS='{"agent": "/deps/__outer_graphs/src/agent.py:graph"}'"""
     assert clean_empty_lines(actual_docker_stdin) == expected_docker_stdin
+    assert additional_contexts == {}
 
 
 # node.js build used for LangGraph Cloud
 def test_config_to_docker_nodejs():
     graphs = {"agent": "./graphs/agent.js:graph"}
-    actual_docker_stdin = config_to_docker(
+    actual_docker_stdin, additional_contexts = config_to_docker(
         PATH_TO_CONFIG,
         validate_config(
             {
@@ -368,6 +458,7 @@ WORKDIR /deps/unit_tests
 RUN (test ! -f /api/langgraph_api/js/build.mts && echo "Prebuild script not found, skipping") || tsx /api/langgraph_api/js/build.mts"""
 
     assert clean_empty_lines(actual_docker_stdin) == expected_docker_stdin
+    assert additional_contexts == {}
 
 
 # config_to_compose
@@ -380,6 +471,7 @@ def test_config_to_compose_simple_config():
             context: .
             dockerfile_inline: |
                 FROM langchain/langgraph-api:3.11
+                # -- Adding non-package dependency unit_tests --
                 ADD . /deps/__outer_unit_tests/unit_tests
                 RUN set -ex && \\
                     for line in '[project]' \\
@@ -389,7 +481,10 @@ def test_config_to_compose_simple_config():
                                 '"*" = ["**/*"]'; do \\
                         echo "$line" >> /deps/__outer_unit_tests/pyproject.toml; \\
                     done
+                # -- End of non-package dependency unit_tests --
+                # -- Installing all local dependencies --
                 RUN PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -c /api/constraints.txt -e /deps/*
+                # -- End of local dependencies install --
                 ENV LANGSERVE_GRAPHS='{"agent": "/deps/__outer_unit_tests/unit_tests/agent.py:graph"}'
                 WORKDIR /deps/__outer_unit_tests/unit_tests
         """
@@ -410,6 +505,7 @@ def test_config_to_compose_env_vars():
             context: .
             dockerfile_inline: |
                 FROM langchain/langgraph-api-custom:3.11
+                # -- Adding non-package dependency unit_tests --
                 ADD . /deps/__outer_unit_tests/unit_tests
                 RUN set -ex && \\
                     for line in '[project]' \\
@@ -419,7 +515,10 @@ def test_config_to_compose_env_vars():
                                 '"*" = ["**/*"]'; do \\
                         echo "$line" >> /deps/__outer_unit_tests/pyproject.toml; \\
                     done
+                # -- End of non-package dependency unit_tests --
+                # -- Installing all local dependencies --
                 RUN PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -c /api/constraints.txt -e /deps/*
+                # -- End of local dependencies install --
                 ENV LANGSERVE_GRAPHS='{"agent": "/deps/__outer_unit_tests/unit_tests/agent.py:graph"}'
                 WORKDIR /deps/__outer_unit_tests/unit_tests
         """
@@ -447,6 +546,7 @@ def test_config_to_compose_env_file():
             context: .
             dockerfile_inline: |
                 FROM langchain/langgraph-api:3.11
+                # -- Adding non-package dependency unit_tests --
                 ADD . /deps/__outer_unit_tests/unit_tests
                 RUN set -ex && \\
                     for line in '[project]' \\
@@ -456,7 +556,10 @@ def test_config_to_compose_env_file():
                                 '"*" = ["**/*"]'; do \\
                         echo "$line" >> /deps/__outer_unit_tests/pyproject.toml; \\
                     done
+                # -- End of non-package dependency unit_tests --
+                # -- Installing all local dependencies --
                 RUN PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -c /api/constraints.txt -e /deps/*
+                # -- End of local dependencies install --
                 ENV LANGSERVE_GRAPHS='{"agent": "/deps/__outer_unit_tests/unit_tests/agent.py:graph"}'
                 WORKDIR /deps/__outer_unit_tests/unit_tests
         """
@@ -477,6 +580,7 @@ def test_config_to_compose_watch():
             context: .
             dockerfile_inline: |
                 FROM langchain/langgraph-api:3.11
+                # -- Adding non-package dependency unit_tests --
                 ADD . /deps/__outer_unit_tests/unit_tests
                 RUN set -ex && \\
                     for line in '[project]' \\
@@ -486,7 +590,10 @@ def test_config_to_compose_watch():
                                 '"*" = ["**/*"]'; do \\
                         echo "$line" >> /deps/__outer_unit_tests/pyproject.toml; \\
                     done
+                # -- End of non-package dependency unit_tests --
+                # -- Installing all local dependencies --
                 RUN PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -c /api/constraints.txt -e /deps/*
+                # -- End of local dependencies install --
                 ENV LANGSERVE_GRAPHS='{"agent": "/deps/__outer_unit_tests/unit_tests/agent.py:graph"}'
                 WORKDIR /deps/__outer_unit_tests/unit_tests
         
@@ -516,6 +623,7 @@ def test_config_to_compose_end_to_end():
             context: .
             dockerfile_inline: |
                 FROM langchain/langgraph-api:3.11
+                # -- Adding non-package dependency unit_tests --
                 ADD . /deps/__outer_unit_tests/unit_tests
                 RUN set -ex && \\
                     for line in '[project]' \\
@@ -525,7 +633,10 @@ def test_config_to_compose_end_to_end():
                                 '"*" = ["**/*"]'; do \\
                         echo "$line" >> /deps/__outer_unit_tests/pyproject.toml; \\
                     done
+                # -- End of non-package dependency unit_tests --
+                # -- Installing all local dependencies --
                 RUN PYTHONDONTWRITEBYTECODE=1 pip install --no-cache-dir -c /api/constraints.txt -e /deps/*
+                # -- End of local dependencies install --
                 ENV LANGSERVE_GRAPHS='{"agent": "/deps/__outer_unit_tests/unit_tests/agent.py:graph"}'
                 WORKDIR /deps/__outer_unit_tests/unit_tests
         

libs/langgraph/langgraph/constants.py

@@ -81,6 +81,8 @@ CONFIG_KEY_SCRATCHPAD = sys.intern("__pregel_scratchpad")
 # holds a mutable dict for temporary storage scoped to the current task
 CONFIG_KEY_PREVIOUS = sys.intern("__pregel_previous")
 # holds the previous return value from a stateful Pregel graph.
+CONFIG_KEY_RUNNER_SUBMIT = sys.intern("__pregel_runner_submit")
+# holds a function that receives tasks from runner, executes them and returns results
 
 # --- Other constants ---
 PUSH = sys.intern("__pregel_push")

libs/langgraph/langgraph/func/__init__.py

@@ -57,9 +57,6 @@ def task(
 ]:
     """Define a LangGraph task using the `task` decorator.
 
-    !!! warning "Beta"
-        The Functional API is currently in beta and is subject to change.
-
     !!! important "Requires python 3.11 or higher for async functions"
         The `task` decorator supports both sync and async functions. To use async
         functions, ensure that you are using Python 3.11 or higher.
@@ -153,10 +150,6 @@ S = TypeVar("S")
 class entrypoint:
     """Define a LangGraph workflow using the `entrypoint` decorator.
 
-    !!! warning "Beta"
-        The Functional API is currently in beta and is subject to change.
-
-
     ### Function signature
 
     The decorated function must accept a **single parameter**, which serves as the input

libs/langgraph/langgraph/graph/state.py

@@ -286,7 +286,7 @@ class StateGraph(Graph):
         Will take the name of the function/runnable as the node name.
 
         Args:
-            node (Union[str, RunnableLike)]: The function or runnable this node will run.
+            node (Union[str, RunnableLike]): The function or runnable this node will run.
             action (Optional[RunnableLike]): The action associated with the node. (default: None)
             metadata (Optional[dict[str, Any]]): The metadata associated with the node. (default: None)
             input (Optional[Type[Any]]): The input schema for the node. (default: the graph's input schema)

libs/langgraph/langgraph/prebuilt/chat_agent_executor.py

@@ -12,7 +12,11 @@ from typing import (
     cast,
 )
 
-from langchain_core.language_models import BaseChatModel, LanguageModelLike
+from langchain_core.language_models import (
+    BaseChatModel,
+    LanguageModelInput,
+    LanguageModelLike,
+)
 from langchain_core.messages import AIMessage, BaseMessage, SystemMessage, ToolMessage
 from langchain_core.runnables import (
     Runnable,
@@ -63,15 +67,15 @@ PROMPT_RUNNABLE_NAME = "Prompt"
 MessagesModifier = Union[
     SystemMessage,
     str,
-    Callable[[Sequence[BaseMessage]], Sequence[BaseMessage]],
-    Runnable[Sequence[BaseMessage], Sequence[BaseMessage]],
+    Callable[[Sequence[BaseMessage]], LanguageModelInput],
+    Runnable[Sequence[BaseMessage], LanguageModelInput],
 ]
 
 Prompt = Union[
     SystemMessage,
     str,
-    Callable[[StateSchema], Sequence[BaseMessage]],
-    Runnable[StateSchema, Sequence[BaseMessage]],
+    Callable[[StateSchema], LanguageModelInput],
+    Runnable[StateSchema, LanguageModelInput],
 ]
 
 

libs/langgraph/langgraph/pregel/__init__.py

@@ -59,6 +59,7 @@ from langgraph.constants import (
     CONFIG_KEY_NODE_FINISHED,
     CONFIG_KEY_READ,
     CONFIG_KEY_RESUMING,
+    CONFIG_KEY_RUNNER_SUBMIT,
     CONFIG_KEY_SEND,
     CONFIG_KEY_STORE,
     CONFIG_KEY_STREAM,
@@ -249,25 +250,6 @@ class Pregel(PregelProtocol):
 
     Repeat until no chains are planned for execution, or a maximum number of steps
     is reached.
-
-    Example:
-        ```python
-        from langgraph import Channel, Pregel
-
-        grow_value = (
-            Channel.subscribe_to("value")
-            | (lambda x: x + x)
-            | Channel.write_to(value=lambda x: x if len(x) < 10 else None)
-        )
-
-        app = Pregel(
-            chains={"grow_value": grow_value},
-            input="value",
-            output="value",
-        )
-
-        assert app.invoke("a") == "aaaaaaaa"
-        ```
     """
 
     nodes: dict[str, PregelNode]
@@ -1755,7 +1737,7 @@ class Pregel(PregelProtocol):
             ) as loop:
                 # create runner
                 runner = PregelRunner(
-                    submit=loop.submit,
+                    submit=config[CONF].get(CONFIG_KEY_RUNNER_SUBMIT, loop.submit),
                     put_writes=loop.put_writes,
                     schedule_task=loop.accept_push,
                     node_finished=config[CONF].get(CONFIG_KEY_NODE_FINISHED),
@@ -2047,7 +2029,7 @@ class Pregel(PregelProtocol):
             ) as loop:
                 # create runner
                 runner = PregelRunner(
-                    submit=loop.submit,
+                    submit=config[CONF].get(CONFIG_KEY_RUNNER_SUBMIT, loop.submit),
                     put_writes=loop.put_writes,
                     schedule_task=loop.accept_push,
                     use_astream=do_stream is not None,

libs/langgraph/langgraph/pregel/remote.py

@@ -323,11 +323,15 @@ class RemoteGraph(PregelProtocol):
             if k not in reserved_configurable_keys and not k.startswith("__pregel_")
         }
 
-        return {
+        sanitized: RunnableConfig = {
             "tags": config.get("tags") or [],
             "metadata": config.get("metadata") or {},
             "configurable": new_configurable,
         }
+        if "recursion_limit" in config:
+            sanitized["recursion_limit"] = config["recursion_limit"]
+
+        return sanitized
 
     def get_state(
         self, config: RunnableConfig, *, subgraphs: bool = False

libs/langgraph/langgraph/utils/runnable.py

@@ -13,6 +13,7 @@ from typing import (
     Coroutine,
     Iterator,
     Optional,
+    Protocol,
     Sequence,
     Tuple,
     Union,
@@ -35,7 +36,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 Concatenate, ParamSpec, TypeGuard
+from typing_extensions import TypeGuard
 
 from langgraph.constants import (
     CONF,
@@ -132,12 +133,51 @@ Each tuple contains:
 VALID_KINDS = (inspect.Parameter.POSITIONAL_OR_KEYWORD, inspect.Parameter.KEYWORD_ONLY)
 
 
-P = ParamSpec("P")  # to handle injected kwargs like `writer` / `store`
+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: ...
+
 
 RunnableLike = Union[
     LCRunnableLike,
-    Callable[Concatenate[Input, P], Output],
-    Callable[Concatenate[Input, P], Awaitable[Output]],
+    _RunnableWithWriter[Input, Output],
+    _RunnableWithStore[Input, Output],
+    _RunnableWithWriterStore[Input, Output],
+    _RunnableWithConfigWriter[Input, Output],
+    _RunnableWithConfigStore[Input, Output],
+    _RunnableWithConfigWriterStore[Input, Output],
 ]
 
 

libs/langgraph/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph"
-version = "0.2.72"
+version = "0.2.73"
 description = "Building stateful, multi-actor applications with LLMs"
 authors = []
 license = "MIT"

libs/sdk-py/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph-sdk"
-version = "0.1.51"
+version = "0.1.52"
 description = "SDK for interacting with LangGraph API"
 authors = []
 license = "MIT"