diff --git a/libs/langchain/langchain/tools/base.py b/libs/langchain/langchain/tools/base.py index 2ae81d246b..e974070ece 100644 --- a/libs/langchain/langchain/tools/base.py +++ b/libs/langchain/langchain/tools/base.py @@ -17,6 +17,7 @@ from langchain.callbacks.manager import ( CallbackManagerForToolRun, Callbacks, ) +from langchain.load.serializable import Serializable from langchain.pydantic_v1 import ( BaseModel, Extra, @@ -165,7 +166,7 @@ class ChildTool(BaseTool): ] = False """Handle the content of the ToolException thrown.""" - class Config: + class Config(Serializable.Config): """Configuration for this pydantic object.""" arbitrary_types_allowed = True diff --git a/libs/langchain/langchain/tools/spark_sql/tool.py b/libs/langchain/langchain/tools/spark_sql/tool.py index c79bfd193a..ad1c6f5ba1 100644 --- a/libs/langchain/langchain/tools/spark_sql/tool.py +++ b/libs/langchain/langchain/tools/spark_sql/tool.py @@ -2,7 +2,7 @@ """Tools for interacting with Spark SQL.""" from typing import Any, Dict, Optional -from langchain.pydantic_v1 import BaseModel, Extra, Field, root_validator +from langchain.pydantic_v1 import BaseModel, Field, root_validator from langchain.schema.language_model import BaseLanguageModel from langchain.callbacks.manager import ( @@ -21,6 +21,9 @@ class BaseSparkSQLTool(BaseModel): db: SparkSQL = Field(exclude=True) + class Config(BaseTool.Config): + pass + class QuerySparkSQLTool(BaseSparkSQLTool, BaseTool): """Tool for querying a Spark SQL."""