From 040bb2983d72eb6c309d608cc02c7982886362fd Mon Sep 17 00:00:00 2001
From: Nuno Campos <nuno@boringbits.io>
Date: Mon, 2 Oct 2023 10:11:26 +0100
Subject: [PATCH] Lint

---
 libs/langchain/langchain/tools/base.py           | 3 ++-
 libs/langchain/langchain/tools/spark_sql/tool.py | 5 ++++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/libs/langchain/langchain/tools/base.py b/libs/langchain/langchain/tools/base.py
index 2ae81d246bc..e974070ece9 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 c79bfd193ad..ad1c6f5ba15 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."""