From 3d79c7552d07bb042b739e2d955255f1b8ab2b55 Mon Sep 17 00:00:00 2001
From: Nuno Campos <nuno@langchain.dev>
Date: Tue, 16 Jan 2024 13:14:00 -0800
Subject: [PATCH] Lint

---
 langgraph/graph/state.py       | 2 +-
 langgraph/prebuilt/__init__.py | 9 +++++++--
 langgraph/pregel/__init__.py   | 6 +++---
 langgraph/pregel/read.py       | 2 +-
 langgraph/pregel/validate.py   | 2 +-
 5 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/langgraph/graph/state.py b/langgraph/graph/state.py
index b0cb163d0..45fb5ec65 100644
--- a/langgraph/graph/state.py
+++ b/langgraph/graph/state.py
@@ -88,7 +88,7 @@ def _coerce_state(schema: Type[Any], input: dict[str, Any]) -> dict[str, Any]:
     return schema(**input)
 
 
-def _update_state(input: dict[str, Any], config: RunnableConfig):
+def _update_state(input: dict[str, Any], config: RunnableConfig) -> dict[str, Any]:
     if input is not None:
         ChannelWrite.do_write(config, **input)
     return input
diff --git a/langgraph/prebuilt/__init__.py b/langgraph/prebuilt/__init__.py
index fb2340d42..db31f778b 100644
--- a/langgraph/prebuilt/__init__.py
+++ b/langgraph/prebuilt/__init__.py
@@ -1,5 +1,10 @@
-from langgraph.prebuilt.agent_executor import create_agent_executor
 from langgraph.prebuilt import chat_agent_executor
+from langgraph.prebuilt.agent_executor import create_agent_executor
 from langgraph.prebuilt.tool_executor import ToolExecutor, ToolInvocation
 
-__all__ = ["create_agent_executor", "chat_agent_executor", "ToolExecutor", ToolInvocation]
+__all__ = [
+    "create_agent_executor",
+    "chat_agent_executor",
+    "ToolExecutor",
+    "ToolInvocation",
+]
diff --git a/langgraph/pregel/__init__.py b/langgraph/pregel/__init__.py
index 4ec34ae32..f29c0b92a 100644
--- a/langgraph/pregel/__init__.py
+++ b/langgraph/pregel/__init__.py
@@ -87,7 +87,7 @@ class Channel:
         *,
         key: Optional[str] = None,
         when: Optional[Callable[[Any], bool]] = None,
-        tags: Optional[Sequence[str]] = None,
+        tags: Optional[list[str]] = None,
     ) -> ChannelInvoke:
         ...
 
@@ -99,7 +99,7 @@ class Channel:
         *,
         key: None = None,
         when: Optional[Callable[[Any], bool]] = None,
-        tags: Optional[Sequence[str]] = None,
+        tags: Optional[list[str]] = None,
     ) -> ChannelInvoke:
         ...
 
@@ -110,7 +110,7 @@ class Channel:
         *,
         key: Optional[str] = None,
         when: Optional[Callable[[Any], bool]] = None,
-        tags: Optional[Sequence[str]] = None,
+        tags: Optional[list[str]] = None,
     ) -> ChannelInvoke:
         """Runs process.invoke() each time channels are updated,
         with a dict of the channel values as input."""
diff --git a/langgraph/pregel/read.py b/langgraph/pregel/read.py
index ebba367b9..2baea3fa9 100644
--- a/langgraph/pregel/read.py
+++ b/langgraph/pregel/read.py
@@ -90,7 +90,7 @@ class ChannelInvoke(RunnableBindingBase):
         channels: Mapping[None, str] | Mapping[str, str],
         triggers: Sequence[str],
         when: Optional[Callable[[Any], bool]] = None,
-        tags: Optional[Sequence[str]] = None,
+        tags: Optional[list[str]] = None,
         *,
         bound: Optional[Runnable[Any, Any]] = None,
         kwargs: Optional[Mapping[str, Any]] = None,
diff --git a/langgraph/pregel/validate.py b/langgraph/pregel/validate.py
index 1186bb8be..8283d093c 100644
--- a/langgraph/pregel/validate.py
+++ b/langgraph/pregel/validate.py
@@ -61,7 +61,7 @@ def validate_graph(
 
 def validate_keys(
     keys: Union[str, Sequence[str]],
-    channels: dict[str, BaseChannel],
+    channels: Mapping[str, BaseChannel],
 ) -> None:
     if isinstance(keys, str):
         if keys not in channels: