diff --git a/libs/langchain/langchain/tools/sql_database/tool.py b/libs/langchain/langchain/tools/sql_database/tool.py index eba921c163..99289e4a97 100644 --- a/libs/langchain/langchain/tools/sql_database/tool.py +++ b/libs/langchain/langchain/tools/sql_database/tool.py @@ -21,6 +21,9 @@ class BaseSQLDatabaseTool(BaseModel): db: SQLDatabase = Field(exclude=True) + class Config(BaseTool.Config): + pass + class QuerySQLDataBaseTool(BaseSQLDatabaseTool, BaseTool): """Tool for querying a SQL database."""