diff --git a/libs/langchain/langchain/chains/base.py b/libs/langchain/langchain/chains/base.py index c4dd6989962..ce94a248b34 100644 --- a/libs/langchain/langchain/chains/base.py +++ b/libs/langchain/langchain/chains/base.py @@ -750,11 +750,11 @@ class Chain(RunnableSerializable[dict[str, Any], dict[str, Any]], ABC): Example: .. code-block:: python - chain.dict(exclude_unset=True) + chain.model_dump(exclude_unset=True) # -> {"_type": "foo", "verbose": False, ...} """ - _dict = super().dict(**kwargs) + _dict = super().model_dump(**kwargs) with contextlib.suppress(NotImplementedError): _dict["_type"] = self._chain_type return _dict @@ -779,7 +779,7 @@ class Chain(RunnableSerializable[dict[str, Any], dict[str, Any]], ABC): raise ValueError(msg) # Fetch dictionary to save - chain_dict = self.dict() + chain_dict = self.model_dump() if "_type" not in chain_dict: msg = f"Chain {self} does not support saving." raise NotImplementedError(msg)