Add compat with stream_events v2

This commit is contained in:
Nuno Campos
2024-05-20 13:39:28 -07:00
parent 9754602a56
commit fa4f50c952
+2 -2
View File
@@ -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,
)