Merge pull request #506 from langchain-ai/nc/20may/fix-stream-v2

Add compat with stream_events v2

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

poetry.lock

@@ -4077,4 +4077,4 @@ testing = ["big-O", "jaraco.functools", "jaraco.itertools", "more-itertools", "p
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.9.0,<4.0"
-content-hash = "84f249eca2884c8ef5a6728d78580deb29db78c8aa4f48998bc12b2a1f095733"
+content-hash = "396cb9a3938f1cc31440d45292757c13d3894d11374799e0c8ccfa1aa877fcb5"

pyproject.toml

@@ -9,7 +9,7 @@ repository = "https://www.github.com/langchain-ai/langgraph"
 
 [tool.poetry.dependencies]
 python = ">=3.9.0,<4.0"
-langchain-core = ">=0.1.52,<0.3"
+langchain-core = ">=0.2,<0.3"
 uuid6 = "^2024.1.12"