From 2e3a690b3466d60ec12c3253e46355d4a49d88ec Mon Sep 17 00:00:00 2001 From: Tat Dat Duong Date: Tue, 30 Apr 2024 22:18:10 +0200 Subject: [PATCH] feat(pregel): mark graph as serializable --- langgraph/pregel/__init__.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/langgraph/pregel/__init__.py b/langgraph/pregel/__init__.py index 552021e4d..1586308e9 100644 --- a/langgraph/pregel/__init__.py +++ b/langgraph/pregel/__init__.py @@ -211,6 +211,11 @@ class Pregel( class Config: arbitrary_types_allowed = True + @classmethod + def is_lc_serializable(cls) -> bool: + """Return whether the graph can be serialized by Langchain.""" + return True + @root_validator(skip_on_failure=True) def validate_on_init(cls, values: dict[str, Any]) -> dict[str, Any]: if not values["auto_validate"]: