From c04802a3440e998a6ba6710a708f2a2029895bff Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Fri, 24 Jan 2025 09:31:27 -0800 Subject: [PATCH 1/2] 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 55ea88e02..1d4bf7679 100644 --- a/libs/langgraph/langgraph/func/__init__.py +++ b/libs/langgraph/langgraph/func/__init__.py @@ -309,7 +309,7 @@ class entrypoint: 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: From be9cb03dfa60a296762121ff245fa1a9b3733eeb Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Fri, 24 Jan 2025 13:51:16 -0800 Subject: [PATCH 2/2] Lint --- libs/langgraph/langgraph/func/__init__.py | 1 - 1 file changed, 1 deletion(-) diff --git a/libs/langgraph/langgraph/func/__init__.py b/libs/langgraph/langgraph/func/__init__.py index 1d4bf7679..3e6f4647e 100644 --- a/libs/langgraph/langgraph/func/__init__.py +++ b/libs/langgraph/langgraph/func/__init__.py @@ -2,7 +2,6 @@ import asyncio import concurrent.futures import functools import inspect -import types from dataclasses import dataclass from typing import ( Any,