WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-08-21 23:22:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Caspar Broekhuizen
2025-09-09 14:03:14 -07:00
parent 682f39e0d3
commit 08d47daa90
1 changed files with 17 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/langgraph/func/__init__.py
+17 -7
View File
@@ -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.