x

libs/langgraph/langgraph/func/__init__.py

@@ -2,7 +2,6 @@ import asyncio
 import concurrent
 import concurrent.futures
 import functools
-import inspect
 import types
 from typing import (
     Any,
@@ -25,7 +24,7 @@ from langgraph.pregel.call import get_runnable_for_func
 from langgraph.pregel.read import PregelNode
 from langgraph.pregel.write import ChannelWrite, ChannelWriteEntry
 from langgraph.store.base import BaseStore
-from langgraph.types import RetryPolicy, StreamMode, StreamWriter
+from langgraph.types import RetryPolicy, StreamMode
 
 P = ParamSpec("P")
 P1 = TypeVar("P1")

libs/langgraph/langgraph/pregel/algo.py

@@ -36,13 +36,13 @@ from langgraph.constants import (
     CONFIG_KEY_CHECKPOINT_MAP,
     CONFIG_KEY_CHECKPOINT_NS,
     CONFIG_KEY_CHECKPOINTER,
+    CONFIG_KEY_END,
     CONFIG_KEY_READ,
     CONFIG_KEY_SCRATCHPAD,
     CONFIG_KEY_SEND,
     CONFIG_KEY_STORE,
     CONFIG_KEY_TASK_ID,
     CONFIG_KEY_WRITES,
-    CONFIG_KEY_END,
     EMPTY_SEQ,
     ERROR,
     INTERRUPT,

libs/langgraph/langgraph/utils/runnable.py

@@ -36,8 +36,8 @@ from typing_extensions import TypeGuard
 
 from langgraph.constants import (
     CONF,
-    CONFIG_KEY_STORE,
     CONFIG_KEY_END,
+    CONFIG_KEY_STORE,
     CONFIG_KEY_STREAM_WRITER,
 )
 from langgraph.store.base import BaseStore
@@ -84,7 +84,7 @@ KWARGS_CONFIG_KEYS: tuple[tuple[str, tuple[Any, ...], str, Any], ...] = (
     ),
     (
         sys.intern("previous"),
-        (ANY_TYPE, ),
+        (ANY_TYPE,),
         CONFIG_KEY_END,
         inspect.Parameter.empty,
     ),
@@ -144,7 +144,7 @@ class RunnableCallable(Runnable):
         self.func_accepts: dict[str, bool] = {}
         for kw, typ, _, _ in KWARGS_CONFIG_KEYS:
             p = params.get(kw)
-            if typ == (ANY_TYPE, ):
+            if typ == (ANY_TYPE,):
                 self.func_accepts[kw] = p is not None and p.kind in VALID_KINDS
             else:
                 self.func_accepts[kw] = (

libs/langgraph/tests/test_pregel.py

@@ -5304,7 +5304,6 @@ def test_multiple_updates() -> None:
 
 def test_version_1_of_entrypoint() -> None:
     from langgraph.func import entrypoint
-    from typing import TypedDict, Annotated, NotRequired, Any
 
     states = []