From c04802a3440e998a6ba6710a708f2a2029895bff Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Fri, 24 Jan 2025 09:31:27 -0800 Subject: [PATCH] Fix type hints for entrypoint decorator --- libs/langgraph/langgraph/func/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/langgraph/langgraph/func/__init__.py b/libs/langgraph/langgraph/func/__init__.py index 55ea88e025..1d4bf76798 100644 --- a/libs/langgraph/langgraph/func/__init__.py +++ b/libs/langgraph/langgraph/func/__init__.py @@ -309,7 +309,7 @@ def my_workflow(number: int, *, previous: Any = None) -> entrypoint.final[int, i A value will always be saved even if it is None. """ - def __call__(self, func: types.FunctionType) -> Pregel: + def __call__(self, func: Callable[..., Any]) -> Pregel: """Convert a function into a Pregel graph. Args: