From a32fe442f0eb9ceff5d70a3f4412a98cde512900 Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Mon, 29 Apr 2024 14:59:31 -0700 Subject: [PATCH] cr --- langgraph/graph/graph.py | 4 +++- langgraph/pregel/__init__.py | 12 ++++++++---- langgraph/serde/base.py | 6 ++++-- langgraph/version.py | 1 - 4 files changed, 15 insertions(+), 8 deletions(-) diff --git a/langgraph/graph/graph.py b/langgraph/graph/graph.py index 36cc01cc0..7291b4e15 100644 --- a/langgraph/graph/graph.py +++ b/langgraph/graph/graph.py @@ -15,7 +15,9 @@ from typing import ( from langchain_core.runnables import Runnable from langchain_core.runnables.base import RunnableLike from langchain_core.runnables.config import RunnableConfig -from langchain_core.runnables.graph import Node as RunnableGraphNode +from langchain_core.runnables.graph import ( + Node as RunnableGraphNode, +) from langgraph.channels.ephemeral_value import EphemeralValue from langgraph.checkpoint import BaseCheckpointSaver diff --git a/langgraph/pregel/__init__.py b/langgraph/pregel/__init__.py index c26b05241..5d5f90633 100644 --- a/langgraph/pregel/__init__.py +++ b/langgraph/pregel/__init__.py @@ -116,7 +116,8 @@ class Channel: *, key: Optional[str] = None, tags: Optional[list[str]] = None, - ) -> PregelNode: ... + ) -> PregelNode: + ... @overload @classmethod @@ -126,7 +127,8 @@ class Channel: *, key: None = None, tags: Optional[list[str]] = None, - ) -> PregelNode: ... + ) -> PregelNode: + ... @classmethod def subscribe_to( @@ -1210,7 +1212,8 @@ def _prepare_next_tasks( processes: Mapping[str, PregelNode], channels: Mapping[str, BaseChannel], for_execution: Literal[False], -) -> tuple[Checkpoint, list[PregelTaskDescription]]: ... +) -> tuple[Checkpoint, list[PregelTaskDescription]]: + ... @overload @@ -1219,7 +1222,8 @@ def _prepare_next_tasks( processes: Mapping[str, PregelNode], channels: Mapping[str, BaseChannel], for_execution: Literal[True], -) -> tuple[Checkpoint, list[PregelExecutableTask]]: ... +) -> tuple[Checkpoint, list[PregelExecutableTask]]: + ... def _prepare_next_tasks( diff --git a/langgraph/serde/base.py b/langgraph/serde/base.py index 5f1250d1e..2fbb0ab71 100644 --- a/langgraph/serde/base.py +++ b/langgraph/serde/base.py @@ -10,6 +10,8 @@ class SerializerProtocol(Protocol): Valid implementations include the `pickle`, `json` and `orjson` modules. """ - def dumps(self, obj: Any) -> bytes: ... + def dumps(self, obj: Any) -> bytes: + ... - def loads(self, data: bytes) -> Any: ... + def loads(self, data: bytes) -> Any: + ... diff --git a/langgraph/version.py b/langgraph/version.py index 3368893c0..ac7aeef6f 100644 --- a/langgraph/version.py +++ b/langgraph/version.py @@ -1,5 +1,4 @@ """Main entrypoint into package.""" - from importlib import metadata try: