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) ),