diff --git a/libs/langgraph/langgraph/managed/base.py b/libs/langgraph/langgraph/managed/base.py index 41a009908..38f0f755a 100644 --- a/libs/langgraph/langgraph/managed/base.py +++ b/libs/langgraph/langgraph/managed/base.py @@ -30,4 +30,3 @@ def is_managed_value(value: Any) -> TypeGuard[ManagedValueSpec]: ManagedValueMapping = dict[str, ManagedValueSpec] - diff --git a/libs/langgraph/langgraph/managed/is_last_step.py b/libs/langgraph/langgraph/managed/is_last_step.py index 427361351..a5bf1598b 100644 --- a/libs/langgraph/langgraph/managed/is_last_step.py +++ b/libs/langgraph/langgraph/managed/is_last_step.py @@ -22,4 +22,3 @@ class RemainingStepsManager(ManagedValue[int]): RemainingSteps = Annotated[int, RemainingStepsManager] - diff --git a/libs/langgraph/langgraph/pregel/_algo.py b/libs/langgraph/langgraph/pregel/_algo.py index 66000ed8f..e8cb077da 100644 --- a/libs/langgraph/langgraph/pregel/_algo.py +++ b/libs/langgraph/langgraph/pregel/_algo.py @@ -69,7 +69,6 @@ from langgraph.pregel._call import get_runnable_for_task, identifier from langgraph.pregel._io import read_channels from langgraph.pregel._log import logger from langgraph.pregel._read import INPUT_CACHE_KEY_TYPE, PregelNode -from langgraph.types import PregelScratchpad from langgraph.runtime import DEFAULT_RUNTIME, Runtime from langgraph.store.base import BaseStore from langgraph.types import ( @@ -77,6 +76,7 @@ from langgraph.types import ( CacheKey, CachePolicy, PregelExecutableTask, + PregelScratchpad, PregelTask, RetryPolicy, Send, @@ -1105,4 +1105,3 @@ class LazyAtomicCounter: if self._counter is None: self._counter = itertools.count(0).__next__ return self._counter() - diff --git a/libs/langgraph/langgraph/pregel/_loop.py b/libs/langgraph/langgraph/pregel/_loop.py index ccfdbb7b7..d7fd553d5 100644 --- a/libs/langgraph/langgraph/pregel/_loop.py +++ b/libs/langgraph/langgraph/pregel/_loop.py @@ -100,7 +100,6 @@ from langgraph.pregel._io import ( read_channels, ) from langgraph.pregel._read import PregelNode -from langgraph.types import PregelScratchpad from langgraph.pregel._utils import get_new_channel_versions, is_xxh3_128_hexdigest from langgraph.pregel.debug import ( map_debug_checkpoint, @@ -115,6 +114,7 @@ from langgraph.types import ( Command, Durability, PregelExecutableTask, + PregelScratchpad, RetryPolicy, StreamMode, ) @@ -1233,4 +1233,3 @@ class AsyncPregelLoop(PregelLoop, AbstractAsyncContextManager): # consumer to await it before e.g., reusing the DB connection. e.args = (*e.args, exit_task) raise - diff --git a/libs/langgraph/langgraph/pregel/_runner.py b/libs/langgraph/langgraph/pregel/_runner.py index 25935250c..5c4dd98a1 100644 --- a/libs/langgraph/langgraph/pregel/_runner.py +++ b/libs/langgraph/langgraph/pregel/_runner.py @@ -36,10 +36,10 @@ from langgraph.errors import GraphBubbleUp, GraphInterrupt from langgraph.pregel._algo import Call from langgraph.pregel._executor import Submit from langgraph.pregel._retry import arun_with_retry, run_with_retry -from langgraph.types import PregelScratchpad from langgraph.types import ( CachePolicy, PregelExecutableTask, + PregelScratchpad, RetryPolicy, ) @@ -758,4 +758,3 @@ async def _acall_impl( destination.set_exception(RuntimeError("Task not scheduled")) except Exception as exc: destination.set_exception(exc) - diff --git a/libs/langgraph/langgraph/types.py b/libs/langgraph/langgraph/types.py index 6f23645ef..0f6618916 100644 --- a/libs/langgraph/langgraph/types.py +++ b/libs/langgraph/langgraph/types.py @@ -538,6 +538,3 @@ def interrupt(value: Any) -> Any: ), ) ) - - -