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, ) diff --git a/poetry.lock b/poetry.lock index 686ba3333..567065424 100644 --- a/poetry.lock +++ b/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" diff --git a/pyproject.toml b/pyproject.toml index d9ed36408..5c9b089d5 100644 --- a/pyproject.toml +++ b/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"