github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-23 11:30:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: update method calls from dict to model_dump in Chain (#33035)

Browse Source
This commit is contained in:
Mason Daugherty
2025-09-20 23:47:44 -04:00
committed by GitHub
parent 043a7560a5
commit 7dcf6a515e
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libs/langchain/langchain/chains/base.py
View File

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