From cee6a450dc9bd4fcecd32bee14a7ba279939fdd8 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Tue, 8 Apr 2025 10:52:28 -0700 Subject: [PATCH] Lint --- libs/langgraph/langgraph/pregel/__init__.py | 4 ++-- libs/langgraph/langgraph/pregel/messages.py | 2 +- libs/langgraph/langgraph/utils/runnable.py | 6 +++--- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/libs/langgraph/langgraph/pregel/__init__.py b/libs/langgraph/langgraph/pregel/__init__.py index cb0a89a79..4711965f6 100644 --- a/libs/langgraph/langgraph/pregel/__init__.py +++ b/libs/langgraph/langgraph/pregel/__init__.py @@ -127,7 +127,7 @@ from langgraph.utils.queue import AsyncQueue, SyncQueue # type: ignore[attr-def try: from langchain_core.tracers._streaming import _StreamingCallbackHandler except ImportError: - _StreamingCallbackHandler = None + _StreamingCallbackHandler = None # type: ignore WriteValue = Union[Callable[[Input], Output], Any] @@ -2536,7 +2536,7 @@ class Pregel(PregelProtocol): do_stream = ( next( ( - cast(_StreamingCallbackHandler, h) # type: ignore + cast(_StreamingCallbackHandler, h) for h in run_manager.handlers if isinstance(h, _StreamingCallbackHandler) ), diff --git a/libs/langgraph/langgraph/pregel/messages.py b/libs/langgraph/langgraph/pregel/messages.py index d53598220..5766c2b63 100644 --- a/libs/langgraph/langgraph/pregel/messages.py +++ b/libs/langgraph/langgraph/pregel/messages.py @@ -23,7 +23,7 @@ from langgraph.types import StreamChunk try: from langchain_core.tracers._streaming import _StreamingCallbackHandler except ImportError: - _StreamingCallbackHandler = object + _StreamingCallbackHandler = object # type: ignore T = TypeVar("T") Meta = tuple[tuple[str, ...], dict[str, Any]] diff --git a/libs/langgraph/langgraph/utils/runnable.py b/libs/langgraph/langgraph/utils/runnable.py index f51da12d9..4dab44023 100644 --- a/libs/langgraph/langgraph/utils/runnable.py +++ b/libs/langgraph/langgraph/utils/runnable.py @@ -56,7 +56,7 @@ from langgraph.utils.config import ( try: from langchain_core.tracers._streaming import _StreamingCallbackHandler except ImportError: - _StreamingCallbackHandler = None + _StreamingCallbackHandler = None # type: ignore def _set_config_context( @@ -690,7 +690,7 @@ class RunnableSeq(Runnable): if _StreamingCallbackHandler is not None and ( stream_handler := next( ( - cast(_StreamingCallbackHandler, h) # type: ignore + cast(_StreamingCallbackHandler, h) for h in run_manager.handlers if isinstance(h, _StreamingCallbackHandler) ), @@ -758,7 +758,7 @@ class RunnableSeq(Runnable): if _StreamingCallbackHandler is not None and ( stream_handler := next( ( - cast(_StreamingCallbackHandler, h) # type: ignore + cast(_StreamingCallbackHandler, h) for h in run_manager.handlers if isinstance(h, _StreamingCallbackHandler) ),