From fa4f50c952ec47ad4f7bb9a23576f3611a300447 Mon Sep 17 00:00:00 2001
From: Nuno Campos <nuno@langchain.dev>
Date: Mon, 20 May 2024 13:39:28 -0700
Subject: [PATCH] Add compat with stream_events v2

---
 langgraph/pregel/__init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/langgraph/pregel/__init__.py b/langgraph/pregel/__init__.py
index 746a1e1ed..d7b03a499 100644
--- a/langgraph/pregel/__init__.py
+++ b/langgraph/pregel/__init__.py
@@ -45,7 +45,7 @@ from langchain_core.runnables.utils import (
     create_model,
     get_unique_config_specs,
 )
-from langchain_core.tracers.log_stream import LogStreamCallbackHandler
+from langchain_core.tracers._streaming import _StreamingCallbackHandler
 from typing_extensions import Self
 
 from langgraph.channels.base import (
@@ -960,7 +960,7 @@ class Pregel(
             (
                 h
                 for h in run_manager.handlers
-                if isinstance(h, LogStreamCallbackHandler)
+                if isinstance(h, _StreamingCallbackHandler)
             ),
             None,
         )