fix(langgraph): split task function overloads to resolve mypy incompatible type error

.github/workflows/_integration_test.yml

@@ -14,19 +14,6 @@ jobs:
         python-version:
           - "3.10"
           - "3.11"
-        example:
-          - name: A
-            workdir: libs/cli/examples
-            tag: langgraph-test-a
-          - name: B
-            workdir: libs/cli/examples/graphs
-            tag: langgraph-test-b
-          - name: C
-            workdir: libs/cli/examples/graphs_reqs_a
-            tag: langgraph-test-c
-          - name: D
-            workdir: libs/cli/examples/graphs_reqs_b
-            tag: langgraph-test-d
     name: "CLI integration test"
     defaults:
       run:
@@ -46,24 +33,54 @@ jobs:
           enable-cache: true
           cache-suffix: "cli-integration-test"
           ignore-nothing-to-cache: true
+      - name: Setup env
+        if: steps.changed-files.outputs.all
+        working-directory: libs/cli/examples
+        run: cat .env.example > .env
       - name: Install cli globally
         if: steps.changed-files.outputs.all
         run: pip install -e .
-      - name: Build and test service ${{ matrix.example.name }}
+      - name: Build and test service A
         if: steps.changed-files.outputs.all
-        working-directory: ${{ matrix.example.workdir }}
+        working-directory: libs/cli/examples
         env:
           LANGSMITH_API_KEY: ${{ secrets.LANGSMITH_API_KEY }}
         run: |
-          # Build the image for this example
-          langgraph build -t ${{ matrix.example.tag }}
-          # Prepare environment file from local or parent example directory
-          if [ -f .env.example ]; then cp .env.example .env; elif [ -f ../.env.example ]; then cp ../.env.example .env; fi
+          # The build-arg isn't used; just testing that we accept other args
+          langgraph build -t langgraph-test-a
+          cp .env.example .env
           if [ -n "${{ secrets.LANGSMITH_API_KEY }}" ]; then echo "LANGSMITH_API_KEY=${{ secrets.LANGSMITH_API_KEY }}" >> .env; fi
-          # Run the integration test using the built tag
-          # Compute repo root to reference the shared script robustly
-          REPO_ROOT=$(git rev-parse --show-toplevel)
-          timeout 60 python "$REPO_ROOT/.github/scripts/run_langgraph_cli_test.py" -t ${{ matrix.example.tag }}
+          timeout 60 python ../../../.github/scripts/run_langgraph_cli_test.py -c langgraph.json -t langgraph-test-a
+      - name: Build and test service B
+        if: steps.changed-files.outputs.all
+        working-directory: libs/cli/examples/graphs
+        env:
+          LANGSMITH_API_KEY: ${{ secrets.LANGSMITH_API_KEY }}
+        run: |
+          langgraph build -t langgraph-test-b
+          cp ../.env.example .env
+          if [ -n "${{ secrets.LANGSMITH_API_KEY }}" ]; then echo "LANGSMITH_API_KEY=${{ secrets.LANGSMITH_API_KEY }}" >> .env; fi
+          timeout 60 python ../../../../.github/scripts/run_langgraph_cli_test.py -t langgraph-test-b
+      - name: Build and test service C
+        if: steps.changed-files.outputs.all
+        working-directory: libs/cli/examples/graphs_reqs_a
+        env:
+          LANGSMITH_API_KEY: ${{ secrets.LANGSMITH_API_KEY }}
+        run: |
+          langgraph build -t langgraph-test-c
+          cp ../.env.example .env
+          if [ -n "${{ secrets.LANGSMITH_API_KEY }}" ]; then echo "LANGSMITH_API_KEY=${{ secrets.LANGSMITH_API_KEY }}" >> .env; fi
+          timeout 60 python ../../../../.github/scripts/run_langgraph_cli_test.py -t langgraph-test-c
+      - name: Build and test service D
+        if: steps.changed-files.outputs.all
+        working-directory: libs/cli/examples/graphs_reqs_b
+        env:
+          LANGSMITH_API_KEY: ${{ secrets.LANGSMITH_API_KEY }}
+        run: |
+          langgraph build -t langgraph-test-d
+          cp ../.env.example .env
+          if [ -n "${{ secrets.LANGSMITH_API_KEY }}" ]; then echo "LANGSMITH_API_KEY=${{ secrets.LANGSMITH_API_KEY }}" >> .env; fi
+          timeout 60 python ../../../../.github/scripts/run_langgraph_cli_test.py -t langgraph-test-d
 
       - name: Build JS service
         if: steps.changed-files.outputs.all

libs/langgraph/langgraph/_internal/_config.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from collections import ChainMap
-from collections.abc import Mapping, Sequence
+from collections.abc import Sequence
 from os import getenv
 from typing import Any, cast
 
@@ -312,22 +312,11 @@ def ensure_config(*configs: RunnableConfig | None) -> RunnableConfig:
         for k, v in config.items():
             if _is_not_empty(v) and k not in CONFIG_KEYS:
                 empty[CONF][k] = v
-    _empty_metadata = empty["metadata"]
     for key, value in empty[CONF].items():
-        if _exclude_as_metadata(key, value, _empty_metadata):
-            continue
-        _empty_metadata[key] = value
+        if (
+            not key.startswith("__")
+            and isinstance(value, (str, int, float, bool))
+            and key not in empty["metadata"]
+        ):
+            empty["metadata"][key] = value
     return empty
-
-
-_OMIT = ("key", "token", "secret", "password", "auth")
-
-
-def _exclude_as_metadata(key: str, value: Any, metadata: Mapping[str, Any]) -> bool:
-    key_lower = key.casefold()
-    return (
-        key.startswith("__")
-        or not isinstance(value, (str, int, float, bool))
-        or key in metadata
-        or any(substr in key_lower for substr in _OMIT)
-    )

libs/langgraph/langgraph/func/__init__.py

@@ -100,16 +100,25 @@ def task(
     retry_policy: RetryPolicy | Sequence[RetryPolicy] | None = None,
     cache_policy: CachePolicy[Callable[P, str | bytes]] | None = None,
     **kwargs: Unpack[DeprecatedKwargs],
-) -> Callable[
-    [Callable[P, Awaitable[T]] | Callable[P, T]],
-    _TaskFunction[P, T],
-]: ...
+) -> Callable[[Callable[P, T]], _TaskFunction[P, T]]: ...
 
 
 @overload
 def task(
-    __func_or_none__: Callable[P, Awaitable[T]] | Callable[P, T],
-) -> _TaskFunction[P, T]: ...
+    *,
+    name: str | None = None,
+    retry_policy: RetryPolicy | Sequence[RetryPolicy] | None = None,
+    cache_policy: CachePolicy[Callable[P, str | bytes]] | None = None,
+    **kwargs: Unpack[DeprecatedKwargs],
+) -> Callable[[Callable[P, Awaitable[T]]], _TaskFunction[P, T]]: ...
+
+
+@overload
+def task(__func_or_none__: Callable[P, T]) -> _TaskFunction[P, T]: ...
+
+
+@overload
+def task(__func_or_none__: Callable[P, Awaitable[T]]) -> _TaskFunction[P, T]: ...
 
 
 def task(
@@ -120,7 +129,8 @@ def task(
     cache_policy: CachePolicy[Callable[P, str | bytes]] | None = None,
     **kwargs: Unpack[DeprecatedKwargs],
 ) -> (
-    Callable[[Callable[P, Awaitable[T]] | Callable[P, T]], _TaskFunction[P, T]]
+    Callable[[Callable[P, T]], _TaskFunction[P, T]]
+    | Callable[[Callable[P, Awaitable[T]]], _TaskFunction[P, T]]
     | _TaskFunction[P, T]
 ):
     """Define a LangGraph task using the `task` decorator.

libs/langgraph/pyproject.toml

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

libs/langgraph/tests/test_utils.py

@@ -17,13 +17,16 @@ import langsmith
 import pytest
 from typing_extensions import NotRequired, Required, TypedDict
 
-from langgraph._internal._config import _is_not_empty, ensure_config
+from langgraph._internal._config import _is_not_empty
 from langgraph._internal._fields import (
     _is_optional_type,
     get_enhanced_type_hints,
     get_field_default,
 )
-from langgraph._internal._runnable import is_async_callable, is_async_generator
+from langgraph._internal._runnable import (
+    is_async_callable,
+    is_async_generator,
+)
 from langgraph.constants import END
 from langgraph.graph import StateGraph
 from langgraph.graph.state import CompiledStateGraph
@@ -298,24 +301,3 @@ def test_is_not_empty() -> None:
     assert not _is_not_empty([])
     assert not _is_not_empty(())
     assert not _is_not_empty({})
-
-
-def test_configurable_metadata():
-    config = {
-        "configurable": {
-            "a-key": "foo",
-            "somesecretval": "bar",
-            "sometoken": "thetoken",
-            "__dontinclude": "bar",
-            "includeme": "hi",
-            "andme": 42,
-            "nested": {"foo": "bar"},
-            "nooverride": -2,
-        },
-        "metadata": {"nooverride": 18},
-    }
-    expected = {"includeme", "andme", "nooverride"}
-    merged = ensure_config(config)
-    metadata = merged["metadata"]
-    assert metadata.keys() == expected
-    assert metadata["nooverride"] == 18

libs/langgraph/uv.lock

@@ -1,5 +1,5 @@
 version = 1
-revision = 3
+revision = 2
 requires-python = ">=3.9"
 resolution-markers = [
     "python_full_version >= '3.14'",
@@ -1269,7 +1269,7 @@ wheels = [
 
 [[package]]
 name = "langgraph"
-version = "0.6.8"
+version = "0.6.7"
 source = { editable = "." }
 dependencies = [
     { name = "langchain-core" },

libs/prebuilt/uv.lock

@@ -1,5 +1,5 @@
 version = 1
-revision = 3
+revision = 2
 requires-python = ">=3.9"
 
 [[package]]
@@ -316,7 +316,7 @@ wheels = [
 
 [[package]]
 name = "langgraph"
-version = "0.6.8"
+version = "0.6.7"
 source = { editable = "../langgraph" }
 dependencies = [
     { name = "langchain-core" },