diff --git a/libs/langgraph/langgraph/pregel/__init__.py b/libs/langgraph/langgraph/pregel/__init__.py index acb225090..8c7a05e13 100644 --- a/libs/langgraph/langgraph/pregel/__init__.py +++ b/libs/langgraph/langgraph/pregel/__init__.py @@ -249,7 +249,7 @@ def __init__( def copy(self, update: dict[str, Any]) -> Self: attrs = {**self.__dict__, **update} - return Pregel(**attrs) + return self.__class__(**attrs) def with_config(self, config: RunnableConfig | None = None, **kwargs: Any) -> Self: return self.copy({"config": merge_configs(self.config, config, kwargs)})