diff --git a/libs/langgraph/langgraph/graph_engine/__init__.py b/libs/langgraph/langgraph/advanced_graph/__init__.py similarity index 81% rename from libs/langgraph/langgraph/graph_engine/__init__.py rename to libs/langgraph/langgraph/advanced_graph/__init__.py index 5e85971d7..95bb5f03f 100644 --- a/libs/langgraph/langgraph/graph_engine/__init__.py +++ b/libs/langgraph/langgraph/advanced_graph/__init__.py @@ -1,4 +1,4 @@ -from langgraph.graph_engine.state import ( +from langgraph.advanced_graph.state import ( AdvancedStateGraph, CompiledGraphEngine, publish_to_channel, diff --git a/libs/langgraph/langgraph/advanced-graph/evolve-extend-langgraph.md b/libs/langgraph/langgraph/advanced_graph/evolve-extend-langgraph.md similarity index 100% rename from libs/langgraph/langgraph/advanced-graph/evolve-extend-langgraph.md rename to libs/langgraph/langgraph/advanced_graph/evolve-extend-langgraph.md diff --git a/libs/langgraph/langgraph/graph_engine/state.py b/libs/langgraph/langgraph/advanced_graph/state.py similarity index 98% rename from libs/langgraph/langgraph/graph_engine/state.py rename to libs/langgraph/langgraph/advanced_graph/state.py index f6fc7636b..2e1b8487a 100644 --- a/libs/langgraph/langgraph/graph_engine/state.py +++ b/libs/langgraph/langgraph/advanced_graph/state.py @@ -12,7 +12,7 @@ from langgraph.types import Command, Send StateT = TypeVar("StateT") _CURRENT_RUN: contextvars.ContextVar[_GraphEngineRun | None] = contextvars.ContextVar( - "langgraph_graph_engine_run", default=None + "langgraph_advanced_graph_run", default=None ) @@ -324,7 +324,7 @@ def _normalize_goto(goto: Any, *, default_arg: Any) -> list[Send]: async def wait_for(channel: str, n: int = 1) -> Any: run = _CURRENT_RUN.get() if run is None: - raise RuntimeError("wait_for() can only be used inside graph_engine nodes") + raise RuntimeError("wait_for() can only be used inside advanced_graph nodes") return await run.wait_for(channel, n=n) @@ -332,7 +332,7 @@ def publish_to_channel(channel: str, value: Any) -> None: run = _CURRENT_RUN.get() if run is None: raise RuntimeError( - "publish_to_channel() can only be used inside graph_engine nodes" + "publish_to_channel() can only be used inside advanced_graph nodes" ) run.publish_nowait(channel, value) diff --git a/libs/langgraph/tests/advanced-graph/test_sub_agents.py b/libs/langgraph/tests/advanced-graph/test_sub_agents.py index a5704d9d0..c41eed8db 100644 --- a/libs/langgraph/tests/advanced-graph/test_sub_agents.py +++ b/libs/langgraph/tests/advanced-graph/test_sub_agents.py @@ -5,9 +5,9 @@ from typing import Any, Literal import pytest from typing_extensions import TypedDict +from langgraph.advanced_graph import AdvancedStateGraph, publish_to_channel, wait_for from langgraph.constants import END, START from langgraph.graph import StateGraph -from langgraph.graph_engine import AdvancedStateGraph, publish_to_channel, wait_for from langgraph.types import Command, Send pytestmark = pytest.mark.anyio