From 313e3eed3e499af811ec3ca9e0576dff4332ccd7 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Sat, 21 Sep 2024 17:26:29 -0700 Subject: [PATCH] Fix circular import --- libs/langgraph/langgraph/constants.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/libs/langgraph/langgraph/constants.py b/libs/langgraph/langgraph/constants.py index eb0f703be..e17ce7de8 100644 --- a/libs/langgraph/langgraph/constants.py +++ b/libs/langgraph/langgraph/constants.py @@ -1,9 +1,21 @@ from types import MappingProxyType from typing import Any, Mapping -from langgraph.pregel.types import Interrupt, Send # noqa: F401 # Interrupt, Send re-exported for backwards compatibility +def __getattr__(name: str) -> Any: + if name in globals(): + return globals()[name] + elif name == "Interrupt": + from langgraph.pregel.types import Interrupt + + return Interrupt + elif name == "Send": + from langgraph.pregel.types import Send + + return Send + raise AttributeError(f"module {__name__} has no attribute {name}") + # --- Empty read-only containers --- EMPTY_MAP: Mapping[str, Any] = MappingProxyType({})