From 311fe3970ca7c124f076df376c39860b9c5075a6 Mon Sep 17 00:00:00 2001
From: Eugene Yurtsev <eyurtsev@gmail.com>
Date: Tue, 14 Jan 2025 15:39:57 -0500
Subject: [PATCH] x

---
 libs/langgraph/langgraph/func/__init__.py  | 3 +--
 libs/langgraph/langgraph/pregel/algo.py    | 2 +-
 libs/langgraph/langgraph/utils/runnable.py | 6 +++---
 libs/langgraph/tests/test_pregel.py        | 1 -
 4 files changed, 5 insertions(+), 7 deletions(-)

diff --git a/libs/langgraph/langgraph/func/__init__.py b/libs/langgraph/langgraph/func/__init__.py
index 76e97d7e5..86c5403f9 100644
--- a/libs/langgraph/langgraph/func/__init__.py
+++ b/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")
diff --git a/libs/langgraph/langgraph/pregel/algo.py b/libs/langgraph/langgraph/pregel/algo.py
index f532b7bc8..0d2553240 100644
--- a/libs/langgraph/langgraph/pregel/algo.py
+++ b/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,
diff --git a/libs/langgraph/langgraph/utils/runnable.py b/libs/langgraph/langgraph/utils/runnable.py
index e5109443f..d5e6cf1d2 100644
--- a/libs/langgraph/langgraph/utils/runnable.py
+++ b/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] = (
diff --git a/libs/langgraph/tests/test_pregel.py b/libs/langgraph/tests/test_pregel.py
index 00568da18..8cf2e6b4f 100644
--- a/libs/langgraph/tests/test_pregel.py
+++ b/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 = []